Set script environment earlier, so it applies to script_tun too
[platform/upstream/openconnect.git] / tun.c
1 /*
2  * OpenConnect (SSL + DTLS) VPN client
3  *
4  * Copyright © 2008-2010 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 #include <signal.h>
31 #include <fcntl.h>
32 #include <unistd.h>
33 #include <netdb.h>
34 #include <netinet/in_systm.h>
35 #include <netinet/in.h>
36 #include <netinet/ip.h>
37 #include <net/if.h>
38 #include <arpa/inet.h>
39 #include <errno.h>
40 #if defined(__sun__)
41 #include <stropts.h>
42 #include <sys/sockio.h>
43 #endif
44
45 #include "openconnect.h"
46
47 /*
48  * If an if_tun.h include file was found anywhere (by the Makefile), it's 
49  * included. Else, we end up assuming that we have BSD-style devices such
50  * as /dev/tun0 etc.
51  */
52 #ifdef IF_TUN_HDR
53 #include IF_TUN_HDR
54 #endif
55
56 /*
57  * The OS X tun/tap driver doesn't provide a header file; you're expected
58  * to define this for yourself.
59  */
60 #ifdef __APPLE__
61 #define TUNSIFHEAD  _IOW('t', 96, int)
62 #endif
63
64 /*
65  * OpenBSD always puts the protocol family prefix onto packets. Other
66  * systems let us enable that with the TUNSIFHEAD ioctl, and some of them
67  * (e.g. FreeBSD) _need_ it otherwise they'll interpret IPv6 packets as IPv4.
68  */
69 #if defined(__OpenBSD__) || defined(TUNSIFHEAD)
70 #define TUN_HAS_AF_PREFIX 1
71 #endif
72
73 #ifdef __sun__
74 static int local_config_tun(struct openconnect_info *vpninfo, int mtu_only)
75 {
76         if (!mtu_only)
77                 vpninfo->progress(vpninfo, PRG_ERR,
78                                   "No vpnc-script configured. Need Solaris IP-setting code\n");
79         return 0;
80 }
81 #else
82 static int local_config_tun(struct openconnect_info *vpninfo, int mtu_only)
83 {
84         struct ifreq ifr;
85         int net_fd;
86
87         net_fd = socket(PF_INET, SOCK_DGRAM, 0);
88         if (net_fd < 0) {
89                 perror("open net");
90                 return -EINVAL;
91         }
92         memset(&ifr, 0, sizeof(ifr));
93         strncpy(ifr.ifr_name, vpninfo->ifname, sizeof(ifr.ifr_name) - 1);
94
95         if (!mtu_only) {
96                 struct sockaddr_in addr;
97
98                 if (ioctl(net_fd, SIOCGIFFLAGS, &ifr) < 0)
99                         perror("SIOCGIFFLAGS");
100
101                 ifr.ifr_flags |= IFF_UP | IFF_POINTOPOINT;
102                 if (ioctl(net_fd, SIOCSIFFLAGS, &ifr) < 0)
103                         perror("SIOCSIFFLAGS");
104
105                 addr.sin_family = AF_INET;
106                 addr.sin_addr.s_addr = inet_addr(vpninfo->vpn_addr);
107                 memcpy(&ifr.ifr_addr, &addr, sizeof(addr));
108                 if (ioctl(net_fd, SIOCSIFADDR, &ifr) < 0)
109                         perror("SIOCSIFADDR");
110         }
111
112         ifr.ifr_mtu = vpninfo->mtu;
113         if (ioctl(net_fd, SIOCSIFMTU, &ifr) < 0)
114                 perror("SIOCSIFMTU");
115
116         close(net_fd);
117
118         return 0;
119 }
120 #endif
121
122 static int setenv_int(const char *opt, int value)
123 {
124         char buf[16];
125         sprintf(buf, "%d", value);
126         return setenv(opt, buf, 1);
127 }
128
129 static int netmasklen(struct in_addr addr)
130 {
131         int masklen;
132
133         for (masklen = 0; masklen < 32; masklen++) {
134                 if (ntohl(addr.s_addr) >= (0xffffffff << masklen))
135                         break;
136         }
137         return 32 - masklen;
138 }
139
140 static int process_split_xxclude(struct openconnect_info *vpninfo,
141                                  char *in_ex, char *route, int *v4_incs,
142                                  int *v6_incs)
143 {
144         struct in_addr addr;
145         char envname[80];
146         char *slash;
147
148         slash = strchr(route, '/');
149         if (!slash) {
150         badinc:
151                 vpninfo->progress(vpninfo, PRG_ERR,
152                                   "Discard bad split %sclude: \"%s\"\n",
153                                   in_ex, route);
154                 return -EINVAL;
155         }
156
157         *slash = 0;
158
159         if (strchr(route, ':')) {
160                 snprintf(envname, 79, "CISCO_IPV6_SPLIT_%sC_%d_ADDR", in_ex,
161                          *v6_incs);
162                 setenv(envname, route, 1);
163
164                 snprintf(envname, 79, "CISCO_IPV6_SPLIT_%sC_%d_MASKLEN", in_ex,
165                          *v6_incs);
166                 setenv(envname, slash+1, 1);
167
168                 (*v6_incs)++;
169                 return 0;
170         }
171                 
172         if (!inet_aton(route, &addr)) {
173                 *slash = '/';
174                 goto badinc;
175         }
176
177         envname[79] = 0;
178         snprintf(envname, 79, "CISCO_SPLIT_%sC_%d_ADDR", in_ex, *v4_incs);
179         setenv(envname, route, 1);
180
181         /* Put it back how we found it */
182         *slash = '/';
183
184         if (!inet_aton(slash+1, &addr))
185                 goto badinc;
186
187         snprintf(envname, 79, "CISCO_SPLIT_%sC_%d_MASK", in_ex, *v4_incs);
188         setenv(envname, slash+1, 1);
189
190         snprintf(envname, 79, "CISCO_SPLIT_%sC_%d_MASKLEN", in_ex, *v4_incs);
191         setenv_int(envname, netmasklen(addr));
192
193         (*v4_incs)++;
194         return 0;
195 }
196
197 static int appendenv(const char *opt, const char *new)
198 {
199         char buf[1024];
200         char *old = getenv(opt);
201
202         buf[1023] = 0;
203         if (old)
204                 snprintf(buf, 1023, "%s %s", old, new);
205         else
206                 snprintf(buf, 1023, "%s", new);
207
208         return setenv(opt, buf, 1);
209 }
210
211 static void setenv_cstp_opts(struct openconnect_info *vpninfo)
212 {
213         char *env_buf;
214         int buflen = 0;
215         int bufofs = 0;
216         struct vpn_option *opt;
217
218         for (opt = vpninfo->cstp_options; opt; opt = opt->next)
219                 buflen += 2 + strlen(opt->option) + strlen(opt->value);
220
221         env_buf = malloc(buflen + 1);
222         if (!env_buf)
223                 return;
224
225         env_buf[buflen] = 0;
226
227         for (opt = vpninfo->cstp_options; opt; opt = opt->next)
228                 bufofs += snprintf(env_buf + bufofs, buflen - bufofs,
229                                    "%s=%s\n", opt->option, opt->value);
230
231         setenv("CISCO_CSTP_OPTIONS", env_buf, 1);
232         free(env_buf);
233 }
234
235 static void set_script_env(struct openconnect_info *vpninfo)
236 {
237         char host[80];
238         int ret = getnameinfo(vpninfo->peer_addr, vpninfo->peer_addrlen, host,
239                               sizeof(host), NULL, 0, NI_NUMERICHOST);
240         if (!ret)
241                 setenv("VPNGATEWAY", host, 1);
242         setenv("TUNDEV", vpninfo->ifname, 1);
243         setenv("reason", "connect", 1);
244         unsetenv("CISCO_BANNER");
245         unsetenv("CISCO_SPLIT_INC");
246         unsetenv("CISCO_SPLIT_EXC");
247
248         setenv_int("INTERNAL_IP4_MTU", vpninfo->mtu);
249
250         if (vpninfo->vpn_addr) {
251                 setenv("INTERNAL_IP4_ADDRESS", vpninfo->vpn_addr, 1);
252                 if (vpninfo->vpn_netmask) {
253                         struct in_addr addr;
254                         struct in_addr mask;
255
256                         if (inet_aton(vpninfo->vpn_addr, &addr) &&
257                             inet_aton(vpninfo->vpn_netmask, &mask)) {
258                                 char *netaddr;
259
260                                 addr.s_addr &= mask.s_addr;
261                                 netaddr = inet_ntoa(addr);
262
263                                 setenv("INTERNAL_IP4_NETADDR", netaddr, 1);
264                                 setenv("INTERNAL_IP4_NETMASK", vpninfo->vpn_netmask, 1);
265                                 setenv_int("INTERNAL_IP4_NETMASKLEN", netmasklen(mask));
266                         }
267                 }
268         }
269         if (vpninfo->vpn_addr6) {
270                 setenv("INTERNAL_IP6_ADDRESS", vpninfo->vpn_addr6, 1);
271                 setenv("INTERNAL_IP6_NETMASK", vpninfo->vpn_netmask6, 1);
272         }
273
274         if (vpninfo->vpn_dns[0])
275                 setenv("INTERNAL_IP4_DNS", vpninfo->vpn_dns[0], 1);
276         else
277                 unsetenv("INTERNAL_IP4_DNS");
278         if (vpninfo->vpn_dns[1])
279                 appendenv("INTERNAL_IP4_DNS", vpninfo->vpn_dns[1]);
280         if (vpninfo->vpn_dns[2])
281                 appendenv("INTERNAL_IP4_DNS", vpninfo->vpn_dns[2]);
282
283         if (vpninfo->vpn_nbns[0])
284                 setenv("INTERNAL_IP4_NBNS", vpninfo->vpn_nbns[0], 1);
285         else
286                 unsetenv("INTERNAL_IP4_NBNS");
287         if (vpninfo->vpn_nbns[1])
288                 appendenv("INTERNAL_IP4_NBNS", vpninfo->vpn_nbns[1]);
289         if (vpninfo->vpn_nbns[2])
290                 appendenv("INTERNAL_IP4_NBNS", vpninfo->vpn_nbns[2]);
291
292         if (vpninfo->vpn_domain)
293                 setenv("CISCO_DEF_DOMAIN", vpninfo->vpn_domain, 1);
294         else unsetenv ("CISCO_DEF_DOMAIN");
295
296         if (vpninfo->vpn_proxy_pac)
297                 setenv("CISCO_PROXY_PAC", vpninfo->vpn_proxy_pac, 1);
298
299         if (vpninfo->split_includes) {
300                 struct split_include *this = vpninfo->split_includes;
301                 int nr_split_includes = 0;
302                 int nr_v6_split_includes = 0;
303
304                 while (this) {
305                         process_split_xxclude(vpninfo, "IN", this->route,
306                                               &nr_split_includes,
307                                               &nr_v6_split_includes);
308                         this = this->next;
309                 }
310                 if (nr_split_includes)
311                         setenv_int("CISCO_SPLIT_INC", nr_split_includes);
312                 if (nr_v6_split_includes)
313                         setenv_int("CISCO_IPV6_SPLIT_INC", nr_v6_split_includes);
314         }
315         if (vpninfo->split_excludes) {
316                 struct split_include *this = vpninfo->split_excludes;
317                 int nr_split_excludes = 0;
318                 int nr_v6_split_excludes = 0;
319
320                 while (this) {
321                         process_split_xxclude(vpninfo, "EX", this->route,
322                                               &nr_split_excludes,
323                                               &nr_v6_split_excludes);
324                         this = this->next;
325                 }
326                 if (nr_split_excludes)
327                         setenv_int("CISCO_SPLIT_EXC", nr_split_excludes);
328                 if (nr_v6_split_excludes)
329                         setenv_int("CISCO_IPV6_SPLIT_EXC", nr_v6_split_excludes);
330         }
331         setenv_cstp_opts(vpninfo);
332 }
333
334 static int script_config_tun(struct openconnect_info *vpninfo)
335 {
336         if (system(vpninfo->vpnc_script)) {
337                 int e = errno;
338                 vpninfo->progress(vpninfo, PRG_ERR,
339                                   "Failed to spawn script '%s': %s\n",
340                                   vpninfo->vpnc_script, strerror(e));
341                 return -e;
342         }
343         return 0;
344 }
345
346
347 /* Set up a tuntap device. */
348 int setup_tun(struct openconnect_info *vpninfo)
349 {
350         int tun_fd;
351
352         set_script_env(vpninfo);
353
354         if (vpninfo->script_tun) {
355                 pid_t child;
356                 int fds[2];
357
358                 if (socketpair(AF_UNIX, SOCK_DGRAM, 0, fds)) {
359                         perror("socketpair");
360                         exit(1);
361                 }
362                 tun_fd = fds[0];
363                 child = fork();
364                 if (child < 0) {
365                         perror("fork");
366                         exit(1);
367                 } else if (!child) {
368                         close(tun_fd);
369                         setenv_int("VPNFD", fds[1]);
370                         execl("/bin/sh", "/bin/sh", "-c", vpninfo->vpnc_script, NULL);
371                         perror("execl");
372                         exit(1);
373                 }
374                 close(fds[1]);
375                 vpninfo->script_tun = child;
376                 vpninfo->ifname = "(script)";
377         } else {
378 #ifdef IFF_TUN /* Linux */
379                 struct ifreq ifr;
380
381                 tun_fd = open("/dev/net/tun", O_RDWR);
382                 if (tun_fd < 0) {
383                         vpninfo->progress(vpninfo, PRG_ERR,
384                                           "Failed to open tun device: %s\n",
385                                           strerror(errno));
386                         exit(1);
387                 }
388                 memset(&ifr, 0, sizeof(ifr));
389                 ifr.ifr_flags = IFF_TUN | IFF_NO_PI;
390                 if (vpninfo->ifname)
391                         strncpy(ifr.ifr_name, vpninfo->ifname,
392                                 sizeof(ifr.ifr_name) - 1);
393                 if (ioctl(tun_fd, TUNSETIFF, (void *) &ifr) < 0) {
394                         vpninfo->progress(vpninfo, PRG_ERR,
395                                           "TUNSETIFF failed: %s\n",
396                                           strerror(errno));
397                         exit(1);
398                 }
399                 if (!vpninfo->ifname)
400                         vpninfo->ifname = strdup(ifr.ifr_name);
401 #elif defined (__sun__)
402                 static char tun_name[80];
403                 int tun2_fd, ip_fd = open("/dev/ip", O_RDWR);
404                 int unit_nr, mux_id;
405                 struct ifreq ifr;
406
407                 if (ip_fd < 0) {
408                         perror("open /dev/ip");
409                         return -EIO;
410                 }
411
412                 tun_fd = open("/dev/tun", O_RDWR);
413                 if (tun_fd < 0) {
414                         perror("open /dev/tun");
415                         close(ip_fd);
416                         return -EIO;
417                 }
418
419                 unit_nr = ioctl(tun_fd, TUNNEWPPA, -1);
420                 if (unit_nr < 0) {
421                         perror("Failed to create new tun");
422                         close(tun_fd);
423                         close(ip_fd);
424                         return -EIO;
425                 }
426                 
427                 tun2_fd = open("/dev/tun", O_RDWR);
428                 if (tun2_fd < 0) {
429                         perror("open /dev/tun again");
430                         close(tun_fd);
431                         close(ip_fd);
432                         return -EIO;
433                 }
434                 if (ioctl(tun2_fd, I_PUSH, "ip") < 0) {
435                         perror("Can't push IP");
436                         close(tun2_fd);
437                         close(tun_fd);
438                         close(ip_fd);
439                         return -EIO;
440                 }
441                 if (ioctl(tun2_fd, IF_UNITSEL, &unit_nr) < 0) {
442                         perror("Can't select unit");
443                         close(tun2_fd);
444                         close(tun_fd);
445                         close(ip_fd);
446                         return -EIO;
447                 }
448                 mux_id = ioctl(ip_fd, I_PLINK, tun2_fd);
449                 if (mux_id < 0) {
450                         perror("Can't link tun to IP");
451                         close(tun2_fd);
452                         close(tun_fd);
453                         close(ip_fd);
454                         return -EIO;
455                 }
456                 close(tun2_fd);
457
458                 sprintf(tun_name, "tun%d", unit_nr);
459                 vpninfo->ifname = tun_name;
460
461                 memset(&ifr, 0, sizeof(ifr));
462                 strcpy(ifr.ifr_name, tun_name);
463                 ifr.ifr_ip_muxid = mux_id;
464
465                 if (ioctl(ip_fd, SIOCSIFMUXID, &ifr) < 0) {
466                         perror("Set mux id");
467                         close(tun_fd);
468                         ioctl(ip_fd, I_PUNLINK, mux_id);
469                         close(ip_fd);
470                         return -EIO;
471                 }
472                 /* Solaris tunctl needs this in order to tear it down */
473                 vpninfo->progress(vpninfo, PRG_DEBUG, "mux id is %d\n", mux_id);
474                 vpninfo->tun_muxid = mux_id;
475                 vpninfo->ip_fd = ip_fd;
476
477 #else /* BSD et al have /dev/tun$x devices */
478                 static char tun_name[80];
479                 int i;
480                 for (i = 0; i < 255; i++) {
481                         sprintf(tun_name, "/dev/tun%d", i);
482                         tun_fd = open(tun_name, O_RDWR);
483                         if (tun_fd >= 0)
484                                 break;
485                 }
486                 if (tun_fd < 0) {
487                         perror("open tun");
488                         exit(1);
489                 }
490                 vpninfo->ifname = tun_name + 5;
491 #ifdef TUNSIFHEAD
492                 i = 1;
493                 if (ioctl(tun_fd, TUNSIFHEAD, &i) < 0) {
494                         perror("TUNSIFHEAD");
495                         exit(1);
496                 }
497 #endif
498 #endif
499                 if (vpninfo->vpnc_script) {
500                         script_config_tun(vpninfo);
501                         /* We have to set the MTU for ourselves, because the script doesn't */
502                         local_config_tun(vpninfo, 1);
503                 } else
504                         local_config_tun(vpninfo, 0);
505         }
506
507         fcntl(tun_fd, F_SETFD, FD_CLOEXEC);
508
509         vpninfo->tun_fd = tun_fd;
510
511         if (vpninfo->select_nfds <= tun_fd)
512                 vpninfo->select_nfds = tun_fd + 1;
513
514         FD_SET(tun_fd, &vpninfo->select_rfds);
515
516         fcntl(vpninfo->tun_fd, F_SETFL, fcntl(vpninfo->tun_fd, F_GETFL) | O_NONBLOCK);
517
518         return 0;
519 }
520
521 int tun_mainloop(struct openconnect_info *vpninfo, int *timeout)
522 {
523         unsigned char buf[2000];
524         int len;
525         int work_done = 0;
526
527         if (FD_ISSET(vpninfo->tun_fd, &vpninfo->select_rfds)) {
528                 while ((len = read(vpninfo->tun_fd, buf, sizeof(buf))) > 0) {
529                         unsigned char *pkt = buf;
530 #ifdef TUN_HAS_AF_PREFIX
531                         pkt += 4;
532                         len -= 4;
533 #endif
534                         if (queue_new_packet(&vpninfo->outgoing_queue, pkt,
535                                              len))
536                                 break;
537
538                         work_done = 1;
539                         vpninfo->outgoing_qlen++;
540                         if (vpninfo->outgoing_qlen == vpninfo->max_qlen) {
541                                 FD_CLR(vpninfo->tun_fd, &vpninfo->select_rfds);
542                                 break;
543                         }
544                 }
545         } else if (vpninfo->outgoing_qlen < vpninfo->max_qlen) {
546                 FD_SET(vpninfo->tun_fd, &vpninfo->select_rfds);
547         }
548
549         /* The kernel returns -ENOMEM when the queue is full, so theoretically
550            we could handle that and retry... but it doesn't let us poll() for
551            the no-longer-full situation, so let's not bother. */
552         while (vpninfo->incoming_queue) {
553                 struct pkt *this = vpninfo->incoming_queue;
554                 unsigned char *data = this->data;
555                 int len = this->len;
556
557 #ifdef TUN_HAS_AF_PREFIX
558                 struct ip *iph = (void *)data;
559                 int type;
560
561                 if (iph->ip_v == 6)
562                         type = AF_INET6;
563                 else if (iph->ip_v == 4)
564                         type = AF_INET;
565                 else {
566                         static int complained = 0;
567                         if (!complained) {
568                                 complained = 1;
569                                 vpninfo->progress(vpninfo, PRG_ERR,
570                                                   "Unknown packet (len %d) received: %02x %02x %02x %02x...\n",
571                                                   len, data[0], data[1], data[2], data[3]);
572                         }
573                         free(this);
574                         continue;
575                 }
576                 data -= 4;
577                 len += 4;
578                 *(int *)data = htonl(type);
579 #endif
580                 vpninfo->incoming_queue = this->next;
581
582                 if (write(vpninfo->tun_fd, data, len) < 0 &&
583                     errno == ENOTCONN) {
584                         vpninfo->quit_reason = "Client connection terminated";
585                         return 1;
586                 }
587                 free(this);
588         }
589         /* Work is not done if we just got rid of packets off the queue */
590         return work_done;
591 }
592
593 void shutdown_tun(struct openconnect_info *vpninfo)
594 {       
595         if (vpninfo->script_tun) {
596                 kill(vpninfo->script_tun, SIGHUP);
597         } else {
598                 if (vpninfo->vpnc_script) {
599                         setenv("TUNDEV", vpninfo->ifname, 1);
600                         setenv("reason", "disconnect", 1);
601                         if (system(vpninfo->vpnc_script) == -1) {
602                                 vpninfo->progress(vpninfo, PRG_ERR,
603                                                   "Failed to spawn script '%s': %s\n",
604                                                   vpninfo->vpnc_script,
605                                                   strerror(errno));
606                         }
607                 }
608 #ifdef __sun__
609                 if (ioctl(vpninfo->ip_fd, I_PUNLINK, vpninfo->tun_muxid) < 0)
610                         perror("ioctl(I_PUNLINK)");
611
612                 close(vpninfo->ip_fd);
613                 vpninfo->ip_fd = -1;
614 #endif
615         }
616
617         close(vpninfo->tun_fd);
618         vpninfo->tun_fd = -1;
619 }