Fix compiler warnings
[platform/upstream/openconnect.git] / tun.c
1 /*
2  * OpenConnect (SSL + DTLS) VPN client
3  *
4  * Copyright © 2008 Intel Corporation.
5  *
6  * Author: David Woodhouse <dwmw2@infradead.org>
7  *
8  * This program is free software; you can redistribute it and/or
9  * modify it under the terms of the GNU Lesser General Public License
10  * version 2.1, as published by the Free Software Foundation.
11  *
12  * This program is distributed in the hope that it will be useful, but
13  * WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with this library; if not, write to:
19  *
20  *   Free Software Foundation, Inc.
21  *   51 Franklin Street, Fifth Floor,
22  *   Boston, MA 02110-1301 USA
23  */
24
25 #include <sys/types.h>
26 #include <sys/stat.h>
27 #include <sys/socket.h>
28 #include <sys/ioctl.h>
29 #include <string.h>
30 #ifdef __linux__
31 #include <linux/if_tun.h>
32 #endif
33 #include <fcntl.h>
34 #include <unistd.h>
35 #include <netinet/in_systm.h>
36 #include <netinet/in.h>
37 #include <netinet/ip.h>
38 #include <net/if.h>
39 #include <arpa/inet.h>
40 #include <errno.h>
41
42 #include "openconnect.h"
43
44 #ifdef BSD
45 #define TUN_HAS_AF_PREFIX 1
46 #endif
47
48 static int local_config_tun(struct openconnect_info *vpninfo, int mtu_only)
49 {
50         struct ifreq ifr;
51         int net_fd;
52
53         net_fd = socket(PF_INET, SOCK_DGRAM, 0);
54         if (net_fd < 0) {
55                 perror("open net");
56                 return -EINVAL;
57         }
58         memset(&ifr, 0, sizeof(ifr));
59         strncpy(ifr.ifr_name, vpninfo->ifname, sizeof(ifr.ifr_name) - 1);
60
61         if (!mtu_only) {
62                 struct sockaddr_in addr;
63
64                 if (ioctl(net_fd, SIOCGIFFLAGS, &ifr) < 0)
65                         perror("SIOCGIFFLAGS");
66
67                 ifr.ifr_flags |= IFF_UP | IFF_POINTOPOINT;
68                 if (ioctl(net_fd, SIOCSIFFLAGS, &ifr) < 0)
69                         perror("SIOCSIFFLAGS");
70
71                 addr.sin_family = AF_INET;
72                 addr.sin_addr.s_addr = inet_addr(vpninfo->vpn_addr);
73                 memcpy(&ifr.ifr_addr, &addr, sizeof(addr));
74                 if (ioctl(net_fd, SIOCSIFADDR, &ifr) < 0)
75                         perror("SIOCSIFADDR");
76         }
77
78         ifr.ifr_mtu = vpninfo->mtu;
79         if (ioctl(net_fd, SIOCSIFMTU, &ifr) < 0)
80                 perror("SIOCSIFMTU");
81
82         close(net_fd);
83
84         return 0;
85 }
86
87 static int setenv_int(const char *opt, int value)
88 {
89         char buf[16];
90         sprintf(buf, "%d", value);
91         return setenv(opt, buf, 1);
92 }
93
94 static int process_split_xxclude(struct openconnect_info *vpninfo,
95                                  char *in_ex, char *route, int *nr_incs)
96 {
97         struct in_addr addr;
98         int masklen;
99         char envname[80];
100         char *slash;
101
102         slash = strchr(route, '/');
103         if (!slash) {
104         badinc:
105                 vpninfo->progress(vpninfo, PRG_ERR,
106                                   "Discard bad split %sclude: \"%s\"\n",
107                                   in_ex, route);
108                 return -EINVAL;
109         }
110
111         *slash = 0;
112         if (!inet_aton(route, &addr)) {
113                 *slash = '/';
114                 goto badinc;
115         }
116
117         envname[79] = 0;
118         snprintf(envname, 79, "CISCO_SPLIT_%sC_%d_ADDR", in_ex, *nr_incs);
119         setenv(envname, route, 1);
120
121         /* Put it back how we found it */
122         *slash = '/';
123
124         if (!inet_aton(slash+1, &addr))
125                 goto badinc;
126
127         snprintf(envname, 79, "CISCO_SPLIT_%sC_%d_MASK", in_ex, *nr_incs);
128         setenv(envname, slash+1, 1);
129
130         for (masklen = 0; masklen < 32; masklen++) {
131                 if (ntohl(addr.s_addr) >= (0xffffffff << masklen))
132                         break;
133         }
134         masklen = 32 - masklen;
135
136         snprintf(envname, 79, "CISCO_SPLIT_%sC_%d_MASKLEN", in_ex, *nr_incs);
137         setenv_int(envname, masklen);
138
139         (*nr_incs)++;
140         return 0;
141 }
142
143 static int appendenv(const char *opt, const char *new)
144 {
145         char buf[1024];
146         char *old = getenv(opt);
147
148         buf[1023] = 0;
149         if (old)
150                 snprintf(buf, 1023, "%s %s", old, new);
151         else
152                 snprintf(buf, 1023, "%s", new);
153
154         return setenv(opt, buf, 1);
155 }
156
157 static void setenv_cstp_opts(struct openconnect_info *vpninfo)
158 {
159         char *env_buf;
160         int buflen = 0;
161         int bufofs = 0;
162         struct vpn_option *opt;
163
164         for (opt = vpninfo->cstp_options; opt; opt = opt->next)
165                 buflen += 2 + strlen(opt->option) + strlen(opt->value);
166
167         env_buf = malloc(buflen + 1);
168         if (!env_buf)
169                 return;
170
171         env_buf[buflen] = 0;
172
173         for (opt = vpninfo->cstp_options; opt; opt = opt->next)
174                 bufofs += snprintf(env_buf + bufofs, buflen - bufofs,
175                                    "%s=%s\n", opt->option, opt->value);
176
177         setenv("CISCO_CSTP_OPTIONS", env_buf, 1);
178         free(env_buf);
179 }
180
181 static void set_script_env(struct openconnect_info *vpninfo)
182 {
183         struct sockaddr_in *sin = (void *)vpninfo->peer_addr;
184
185         setenv("VPNGATEWAY", inet_ntoa(sin->sin_addr), 1);
186         setenv("TUNDEV", vpninfo->ifname, 1);
187         setenv("reason", "connect", 1);
188         unsetenv("CISCO_BANNER");
189         unsetenv("CISCO_SPLIT_INC");
190         unsetenv("CISCO_SPLIT_EXC");
191
192         setenv_int("INTERNAL_IP4_MTU", vpninfo->mtu);
193
194         setenv("INTERNAL_IP4_ADDRESS", vpninfo->vpn_addr, 1);
195         setenv("INTERNAL_IP4_NETMASK", vpninfo->vpn_netmask, 1);
196
197         if (vpninfo->vpn_dns[0])
198                 setenv("INTERNAL_IP4_DNS", vpninfo->vpn_dns[0], 1);
199         else
200                 unsetenv("INTERNAL_IP4_DNS");
201         if (vpninfo->vpn_dns[1])
202                 appendenv("INTERNAL_IP4_DNS", vpninfo->vpn_dns[1]);
203         if (vpninfo->vpn_dns[2])
204                 appendenv("INTERNAL_IP4_DNS", vpninfo->vpn_dns[2]);
205
206         if (vpninfo->vpn_nbns[0])
207                 setenv("INTERNAL_IP4_NBNS", vpninfo->vpn_nbns[0], 1);
208         else
209                 unsetenv("INTERNAL_IP4_NBNS");
210         if (vpninfo->vpn_nbns[1])
211                 appendenv("INTERNAL_IP4_NBNS", vpninfo->vpn_nbns[1]);
212         if (vpninfo->vpn_nbns[2])
213                 appendenv("INTERNAL_IP4_NBNS", vpninfo->vpn_nbns[2]);
214
215         if (vpninfo->vpn_domain)
216                 setenv("CISCO_DEF_DOMAIN", vpninfo->vpn_domain, 1);
217         else unsetenv ("CISCO_DEF_DOMAIN");
218
219         if (vpninfo->vpn_proxy_pac)
220                 setenv("CISCO_PROXY_PAC", vpninfo->vpn_proxy_pac, 1);
221
222         if (vpninfo->split_includes) {
223                 struct split_include *this = vpninfo->split_includes;
224                 int nr_split_includes = 0;
225
226                 while (this) {
227                         process_split_xxclude(vpninfo, "IN", this->route,
228                                               &nr_split_includes);
229                         this = this->next;
230                 }
231                 setenv_int("CISCO_SPLIT_INC", nr_split_includes);
232         }
233         if (vpninfo->split_excludes) {
234                 struct split_include *this = vpninfo->split_excludes;
235                 int nr_split_excludes = 0;
236
237                 while (this) {
238                         process_split_xxclude(vpninfo, "EX", this->route,
239                                               &nr_split_excludes);
240                         this = this->next;
241                 }
242                 setenv_int("CISCO_SPLIT_EXC", nr_split_excludes);
243         }
244         setenv_cstp_opts(vpninfo);
245 }
246
247 static int script_config_tun(struct openconnect_info *vpninfo)
248 {
249         if (vpninfo->peer_addr->sa_family != AF_INET) {
250                 vpninfo->progress(vpninfo, PRG_ERR, "Script cannot handle anything but Legacy IP\n");
251                 return -EINVAL;
252         }
253
254         set_script_env(vpninfo);
255
256         system(vpninfo->vpnc_script);
257         return 0;
258 }
259
260
261 /* Set up a tuntap device. */
262 int setup_tun(struct openconnect_info *vpninfo)
263 {
264         int tun_fd;
265
266         if (vpninfo->script_tun) {
267                 pid_t child;
268                 int fds[2];
269
270                 if (socketpair(AF_UNIX, SOCK_DGRAM, 0, fds)) {
271                         perror("socketpair");
272                         exit(1);
273                 }
274                 tun_fd = fds[0];
275                 child = fork();
276                 if (child < 0) {
277                         perror("fork");
278                         exit(1);
279                 } else if (!child) {
280                         close(tun_fd);
281                         setenv_int("VPNFD", fds[1]);
282                         execl("/bin/sh", "/bin/sh", "-c", vpninfo->vpnc_script, NULL);
283                         perror("execl");
284                         exit(1);
285                 }
286                 close(fds[1]);
287                 vpninfo->script_tun = child;
288                 vpninfo->ifname = "(script)";
289         } else {
290 #ifdef IFF_TUN /* Linux */
291                 struct ifreq ifr;
292
293                 tun_fd = open("/dev/net/tun", O_RDWR);
294                 if (tun_fd < 0) {
295                         vpninfo->progress(vpninfo, PRG_ERR,
296                                           "Failed to open tun device: %s\n",
297                                           strerror(errno));
298                         exit(1);
299                 }
300                 memset(&ifr, 0, sizeof(ifr));
301                 ifr.ifr_flags = IFF_TUN | IFF_NO_PI;
302                 if (vpninfo->ifname)
303                         strncpy(ifr.ifr_name, vpninfo->ifname,
304                                 sizeof(ifr.ifr_name) - 1);
305                 if (ioctl(tun_fd, TUNSETIFF, (void *) &ifr) < 0) {
306                         vpninfo->progress(vpninfo, PRG_ERR,
307                                           "TUNSETIFF failed: %s\n",
308                                           strerror(errno));
309                         exit(1);
310                 }
311                 if (!vpninfo->ifname)
312                         vpninfo->ifname = strdup(ifr.ifr_name);
313
314 #else /* BSD et al have /dev/tun$x devices */
315                 static char tun_name[80];
316                 int i;
317                 for (i = 0; i < 255; i++) {
318                         sprintf(tun_name, "/dev/tun%d", i);
319                         tun_fd = open(tun_name, O_RDWR);
320                         if (tun_fd >= 0)
321                                 break;
322                 }
323                 if (tun_fd < 0) {
324                         perror("open tun");
325                         exit(1);
326                 }
327                 vpninfo->ifname = tun_name + 5;
328 #endif
329                 if (vpninfo->vpnc_script) {
330                         script_config_tun(vpninfo);
331                         /* We have to set the MTU for ourselves, because the script doesn't */
332                         local_config_tun(vpninfo, 1);
333                 } else
334                         local_config_tun(vpninfo, 0);
335         }
336
337         fcntl(tun_fd, F_SETFD, FD_CLOEXEC);
338
339         vpninfo->tun_fd = tun_fd;
340
341         if (vpninfo->select_nfds <= tun_fd)
342                 vpninfo->select_nfds = tun_fd + 1;
343
344         FD_SET(tun_fd, &vpninfo->select_rfds);
345
346         fcntl(vpninfo->tun_fd, F_SETFL, fcntl(vpninfo->tun_fd, F_GETFL) | O_NONBLOCK);
347
348         return 0;
349 }
350
351 int tun_mainloop(struct openconnect_info *vpninfo, int *timeout)
352 {
353         unsigned char buf[2000];
354         int len;
355         int work_done = 0;
356
357         if (FD_ISSET(vpninfo->tun_fd, &vpninfo->select_rfds)) {
358                 while ((len = read(vpninfo->tun_fd, buf, sizeof(buf))) > 0) {
359                         unsigned char *pkt = buf;
360                         int type;
361 #ifdef __OpenBSD__
362                         type = ntohl(*(int *)buf);
363                         if (type != AF_INET && type != AF_INET6)
364                                 continue;
365                         pkt += 4;
366                         len -= 4;
367 #else
368                         struct ip *iph = (void *)buf;
369                         if (iph->ip_v == 6)
370                                 type = AF_INET6;
371                         else if (iph->ip_v == 4)
372                                 type = AF_INET;
373                         else
374                                 continue;
375 #endif
376                         if (queue_new_packet(&vpninfo->outgoing_queue, type,
377                                              pkt, len))
378                                 break;
379
380                         work_done = 1;
381                         vpninfo->outgoing_qlen++;
382                         if (vpninfo->outgoing_qlen == vpninfo->max_qlen) {
383                                 FD_CLR(vpninfo->tun_fd, &vpninfo->select_rfds);
384                                 break;
385                         }
386                 }
387         } else if (vpninfo->outgoing_qlen < vpninfo->max_qlen) {
388                 FD_SET(vpninfo->tun_fd, &vpninfo->select_rfds);
389         }
390
391         /* The kernel returns -ENOMEM when the queue is full, so theoretically
392            we could handle that and retry... but it doesn't let us poll() for
393            the no-longer-full situation, so let's not bother. */
394         while (vpninfo->incoming_queue) {
395                 struct pkt *this = vpninfo->incoming_queue;
396                 unsigned char *data = this->data;
397                 int len = this->len;
398
399                 vpninfo->incoming_queue = this->next;
400
401 #ifdef __OpenBSD__
402                 data -= 4;
403                 len += 4;
404                 *(int *)data = htonl(this->type);
405 #endif
406
407                 if (write(vpninfo->tun_fd, data, len) < 0 &&
408                     errno == ENOTCONN) {
409                         vpninfo->quit_reason = "Client connection terminated";
410                         return 1;
411                 }
412                 free(this);
413         }
414         /* Work is not done if we just got rid of packets off the queue */
415         return work_done;
416 }