FD_CLOEXEC
[platform/upstream/openconnect.git] / tun.c
1 /*
2  * Open AnyConnect (SSL + DTLS) client
3  *
4  * © 2008 David Woodhouse <dwmw2@infradead.org>
5  *
6  * Permission to use, copy, modify, and/or distribute this software
7  * for any purpose with or without fee is hereby granted, provided
8  * that the above copyright notice and this permission notice appear
9  * in all copies.
10  *
11  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL
12  * WARRANTIES WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED
13  * WARRANTIES OF MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE
14  * AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT, INDIRECT, OR
15  * CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS
16  * OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT,
17  * NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN
18  * CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
19  */
20
21 #include <string.h>
22 #include <net/if.h>
23 #include <sys/ioctl.h>
24 #include <linux/if_tun.h>
25 #include <sys/types.h>
26 #include <sys/stat.h>
27 #include <fcntl.h>
28 #include <unistd.h>
29 #include <netinet/in.h>
30 #include <arpa/inet.h>
31
32 #include "anyconnect.h"
33
34 /* Set up a tuntap device. */
35 int setup_tun(struct anyconnect_info *vpninfo)
36 {
37         struct vpn_option *cstp_opt = vpninfo->cstp_options;
38         struct ifreq ifr;
39         struct sockaddr_in *addr;
40         int tun_fd, net_fd;
41         int pfd;
42
43         tun_fd = open("/dev/net/tun", O_RDWR);
44         if (tun_fd < 0) {
45                 perror("open tun");
46                 exit(1);
47         }
48         memset(&ifr, 0, sizeof(ifr));
49         ifr.ifr_flags = IFF_TUN | IFF_NO_PI;
50         strncpy(ifr.ifr_name, vpninfo->ifname, sizeof(ifr.ifr_name) - 1);
51         if (ioctl(tun_fd, TUNSETIFF, (void *) &ifr) < 0) {
52                 perror("TUNSETIFF");
53                 exit(1);
54         }
55
56         fcntl(tun_fd, F_SETFD, FD_CLOEXEC);
57
58         net_fd = socket(PF_INET, SOCK_DGRAM, 0);
59         if (net_fd < 0) {
60                 perror("open net");
61                 goto proceed;
62         }
63         memset(&ifr, 0, sizeof(ifr));
64         strncpy(ifr.ifr_name, vpninfo->ifname, sizeof(ifr.ifr_name) - 1);
65         if (ioctl(net_fd, SIOCGIFFLAGS, &ifr) < 0)
66                 perror("SIOCGIFFLAGS");
67         ifr.ifr_flags |= IFF_UP;
68         if (ioctl(net_fd, SIOCSIFFLAGS, &ifr) < 0)
69                 perror("SIOCSIFFLAGS");
70
71         addr = (struct sockaddr_in *) &ifr.ifr_addr;
72         while (cstp_opt) {
73                 printf("CSTP option %s : %s\n", cstp_opt->option, cstp_opt->value);
74                 if (!strcmp(cstp_opt->option, "X-CSTP-Address")) {
75                         addr->sin_family = AF_INET;
76                         addr->sin_addr.s_addr = inet_addr(cstp_opt->value);
77                         if (ioctl(net_fd, SIOCSIFADDR, &ifr) < 0)
78                                 perror("SIOCSIFADDR");
79                 } else if (!strcmp(cstp_opt->option, "X-CSTP-Netmask")) {
80                         addr->sin_family = AF_INET;
81                         addr->sin_addr.s_addr = inet_addr(cstp_opt->value);
82                         if (ioctl(net_fd, SIOCSIFNETMASK, &ifr) < 0)
83                                 perror("SIOCSIFNETMASK");
84                 } else if (!strcmp(cstp_opt->option, "X-CSTP-MTU")) {
85                         ifr.ifr_mtu = atol(cstp_opt->value);
86                         if (ioctl(net_fd, SIOCSIFMTU, &ifr) < 0)
87                                 perror("SIOCSIFMTU");
88                 }
89                 cstp_opt = cstp_opt->next;
90         }
91         close(net_fd);
92
93 proceed:
94         /* Better still, use lwip and just provide a SOCKS server rather than
95            telling the kernel about it at all */
96         vpninfo->tun_fd = tun_fd;
97         pfd = vpn_add_pollfd(vpninfo, vpninfo->tun_fd, POLLIN);
98
99         fcntl(vpninfo->tun_fd, F_SETFL, fcntl(vpninfo->tun_fd, F_GETFL) | O_NONBLOCK);
100
101         return 0;
102 }
103
104 int tun_mainloop(struct anyconnect_info *vpninfo, int *timeout)
105 {
106         char buf[2000];
107         int len;
108         int work_done = 0;
109
110         while ( (len = read(vpninfo->tun_fd, buf, sizeof(buf))) > 0) {
111                 queue_new_packet(&vpninfo->outgoing_queue, AF_INET, buf, len);
112                 work_done = 1;
113         }
114
115         /* The kernel returns -ENOMEM when the queue is full, so theoretically
116            we could handle that and retry... but it doesn't let us poll() for
117            the no-longer-full situation, so let's not bother. */
118         while (vpninfo->incoming_queue) {
119                 struct pkt *this = vpninfo->incoming_queue;
120                 vpninfo->incoming_queue = this->next;
121                 write(vpninfo->tun_fd, this->data, this->len);
122         }
123         /* Work is not done if we just got rid of packets off the queue */
124         return work_done;
125 }