Merge branch 'master' into 1.1
[tinc] / src / mingw / device.c
1 /*
2     device.c -- Interaction with Windows tap driver in a MinGW environment
3     Copyright (C) 2002-2005 Ivo Timmermans,
4                   2002-2009 Guus Sliepen <guus@tinc-vpn.org>
5
6     This program is free software; you can redistribute it and/or modify
7     it under the terms of the GNU General Public License as published by
8     the Free Software Foundation; either version 2 of the License, or
9     (at your option) any later version.
10
11     This program is distributed in the hope that it will be useful,
12     but WITHOUT ANY WARRANTY; without even the implied warranty of
13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14     GNU General Public License for more details.
15
16     You should have received a copy of the GNU General Public License
17     along with this program; if not, write to the Free Software
18     Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
19
20     $Id$
21 */
22
23 #include "system.h"
24
25 #include <windows.h>
26 #include <winioctl.h>
27
28 #include "conf.h"
29 #include "logger.h"
30 #include "net.h"
31 #include "route.h"
32 #include "utils.h"
33 #include "xalloc.h"
34
35 #include "mingw/common.h"
36
37 int device_fd = 0;
38 static HANDLE device_handle = INVALID_HANDLE_VALUE;
39 char *device = NULL;
40 char *iface = NULL;
41 static char *device_info = NULL;
42
43 static int device_total_in = 0;
44 static int device_total_out = 0;
45
46 extern char *myport;
47
48 static struct packetbuf {
49         uint8_t data[MTU];
50         length_t len;
51 } *bufs;
52
53 static int nbufs = 64;
54
55 static DWORD WINAPI tapreader(void *bla) {
56         int sock, err, status;
57         struct addrinfo *ai;
58         struct addrinfo hint = {
59                 .ai_family = AF_UNSPEC,
60                 .ai_socktype = SOCK_STREAM,
61                 .ai_protocol = IPPROTO_TCP,
62                 .ai_flags = 0,
63         };
64         unsigned char bufno = 0;
65         long len;
66         OVERLAPPED overlapped;
67
68         /* Open a socket to the parent process */
69
70         err = getaddrinfo(NULL, myport, &hint, &ai);
71
72         if(err || !ai) {
73                 logger(LOG_ERR, _("System call `%s' failed: %s"), "getaddrinfo", gai_strerror(errno));
74                 return -1;
75         }
76
77         sock = socket(ai->ai_family, SOCK_STREAM, IPPROTO_TCP);
78
79         if(sock < 0) {
80                 logger(LOG_ERR, _("System call `%s' failed: %s"), "socket", strerror(errno));
81                 freeaddrinfo(ai);
82                 return -1;
83         }
84
85         if(connect(sock, ai->ai_addr, ai->ai_addrlen)) {
86                 logger(LOG_ERR, _("System call `%s' failed: %s"), "connect", strerror(errno));
87                 freeaddrinfo(ai);
88                 return -1;
89         }
90
91         freeaddrinfo(ai);
92
93         logger(LOG_DEBUG, _("Tap reader running"));
94
95         /* Read from tap device and send to parent */
96
97         overlapped.hEvent = CreateEvent(NULL, TRUE, FALSE, NULL);
98         
99         for(;;) {
100                 overlapped.Offset = 0;
101                 overlapped.OffsetHigh = 0;
102                 ResetEvent(overlapped.hEvent);
103
104                 status = ReadFile(device_handle, bufs[bufno].data, MTU, &len, &overlapped);
105
106                 if(!status) {
107                         if(GetLastError() == ERROR_IO_PENDING) {
108                                 WaitForSingleObject(overlapped.hEvent, INFINITE);
109                                 if(!GetOverlappedResult(device_handle, &overlapped, &len, FALSE))
110                                         continue;
111                         } else {
112                                 logger(LOG_ERR, _("Error while reading from %s %s: %s"), device_info,
113                                            device, strerror(errno));
114                                 return -1;
115                         }
116                 }
117
118                 bufs[bufno].len = len;
119                 if(send(sock, &bufno, 1, 0) <= 0)
120                         return -1;
121                 if(++bufno >= nbufs)
122                         bufno = 0;
123         }
124 }
125
126 bool setup_device(void) {
127         HKEY key, key2;
128         int i;
129
130         char regpath[1024];
131         char adapterid[1024];
132         char adaptername[1024];
133         char tapname[1024];
134         long len;
135         unsigned long status;
136
137         bool found = false;
138
139         int sock, err;
140         HANDLE thread;
141
142         struct addrinfo *ai;
143         struct addrinfo hint = {
144                 .ai_family = AF_UNSPEC,
145                 .ai_socktype = SOCK_STREAM,
146                 .ai_protocol = IPPROTO_TCP,
147                 .ai_flags = 0,
148         };
149
150         cp();
151
152         get_config_string(lookup_config(config_tree, "Device"), &device);
153         get_config_string(lookup_config(config_tree, "Interface"), &iface);
154
155         /* Open registry and look for network adapters */
156
157         if(RegOpenKeyEx(HKEY_LOCAL_MACHINE, NETWORK_CONNECTIONS_KEY, 0, KEY_READ, &key)) {
158                 logger(LOG_ERR, _("Unable to read registry: %s"), winerror(GetLastError()));
159                 return false;
160         }
161
162         for (i = 0; ; i++) {
163                 len = sizeof adapterid;
164                 if(RegEnumKeyEx(key, i, adapterid, &len, 0, 0, 0, NULL))
165                         break;
166
167                 /* Find out more about this adapter */
168
169                 snprintf(regpath, sizeof regpath, "%s\\%s\\Connection", NETWORK_CONNECTIONS_KEY, adapterid);
170
171                 if(RegOpenKeyEx(HKEY_LOCAL_MACHINE, regpath, 0, KEY_READ, &key2))
172                         continue;
173
174                 len = sizeof adaptername;
175                 err = RegQueryValueEx(key2, "Name", 0, 0, adaptername, &len);
176
177                 RegCloseKey(key2);
178
179                 if(err)
180                         continue;
181
182                 if(device) {
183                         if(!strcmp(device, adapterid)) {
184                                 found = true;
185                                 break;
186                         } else
187                                 continue;
188                 }
189
190                 if(iface) {
191                         if(!strcmp(iface, adaptername)) {
192                                 found = true;
193                                 break;
194                         } else
195                                 continue;
196                 }
197
198                 snprintf(tapname, sizeof tapname, USERMODEDEVICEDIR "%s" TAPSUFFIX, adapterid);
199                 device_handle = CreateFile(tapname, GENERIC_WRITE | GENERIC_READ, 0, 0, OPEN_EXISTING, FILE_ATTRIBUTE_SYSTEM | FILE_FLAG_OVERLAPPED, 0);
200                 if(device_handle != INVALID_HANDLE_VALUE) {
201                         found = true;
202                         break;
203                 }
204         }
205
206         RegCloseKey(key);
207
208         if(!found) {
209                 logger(LOG_ERR, _("No Windows tap device found!"));
210                 return false;
211         }
212
213         if(!device)
214                 device = xstrdup(adapterid);
215
216         if(!iface)
217                 iface = xstrdup(adaptername);
218
219         /* Try to open the corresponding tap device */
220
221         if(device_handle == INVALID_HANDLE_VALUE) {
222                 snprintf(tapname, sizeof tapname, USERMODEDEVICEDIR "%s" TAPSUFFIX, device);
223                 device_handle = CreateFile(tapname, GENERIC_WRITE | GENERIC_READ, 0, 0, OPEN_EXISTING, FILE_ATTRIBUTE_SYSTEM | FILE_FLAG_OVERLAPPED, 0);
224         }
225         
226         if(device_handle == INVALID_HANDLE_VALUE) {
227                 logger(LOG_ERR, _("%s (%s) is not a usable Windows tap device: %s"), device, iface, winerror(GetLastError()));
228                 return false;
229         }
230
231         /* Get MAC address from tap device */
232
233         if(!DeviceIoControl(device_handle, TAP_IOCTL_GET_MAC, mymac.x, sizeof mymac.x, mymac.x, sizeof mymac.x, &len, 0)) {
234                 logger(LOG_ERR, _("Could not get MAC address from Windows tap device %s (%s): %s"), device, iface, winerror(GetLastError()));
235                 return false;
236         }
237
238         if(routing_mode == RMODE_ROUTER) {
239                 overwrite_mac = 1;
240         }
241
242         /* Set up ringbuffer */
243
244         get_config_int(lookup_config(config_tree, "RingBufferSize"), &nbufs);
245         if(nbufs <= 1)
246                 nbufs = 1;
247         else if(nbufs > 256)
248                 nbufs = 256;
249         
250         bufs = xmalloc_and_zero(nbufs * sizeof *bufs);
251
252         /* Create a listening socket */
253
254         err = getaddrinfo(NULL, myport, &hint, &ai);
255
256         if(err || !ai) {
257                 logger(LOG_ERR, _("System call `%s' failed: %s"), "getaddrinfo", gai_strerror(errno));
258                 return false;
259         }
260
261         sock = socket(ai->ai_family, SOCK_STREAM, IPPROTO_TCP);
262
263         if(sock < 0) {
264                 logger(LOG_ERR, _("System call `%s' failed: %s"), "socket", strerror(errno));
265                 return false;
266         }
267
268         if(bind(sock, ai->ai_addr, ai->ai_addrlen)) {
269                 logger(LOG_ERR, _("System call `%s' failed: %s"), "bind", strerror(errno));
270                 return false;
271         }
272
273         freeaddrinfo(ai);
274
275         if(listen(sock, 1)) {
276                 logger(LOG_ERR, _("System call `%s' failed: %s"), "listen", strerror(errno));
277                 return false;
278         }
279
280         /* Start the tap reader */
281
282         thread = CreateThread(NULL, 0, tapreader, NULL, 0, NULL);
283
284         if(!thread) {
285                 logger(LOG_ERR, _("System call `%s' failed: %s"), "CreateThread", winerror(GetLastError()));
286                 return false;
287         }
288
289         /* Wait for the tap reader to connect back to us */
290
291         if((device_fd = accept(sock, NULL, 0)) == -1) {
292                 logger(LOG_ERR, _("System call `%s' failed: %s"), "accept", strerror(errno));
293                 return false;
294         }
295
296         closesocket(sock);
297
298         /* Set media status for newer TAP-Win32 devices */
299
300         status = true;
301         DeviceIoControl(device_handle, TAP_IOCTL_SET_MEDIA_STATUS, &status, sizeof status, &status, sizeof status, &len, NULL);
302
303         device_info = _("Windows tap device");
304
305         logger(LOG_INFO, _("%s (%s) is a %s"), device, iface, device_info);
306
307         return true;
308 }
309
310 void close_device(void) {
311         cp();
312
313         CloseHandle(device_handle);
314
315         free(device);
316         free(iface);
317 }
318
319 bool read_packet(vpn_packet_t *packet) {
320         unsigned char bufno;
321
322         cp();
323
324         if((recv(device_fd, &bufno, 1, 0)) <= 0) {
325                 logger(LOG_ERR, _("Error while reading from %s %s: %s"), device_info,
326                            device, strerror(errno));
327                 return false;
328         }
329         
330         packet->len = bufs[bufno].len;
331         memcpy(packet->data, bufs[bufno].data, bufs[bufno].len);
332
333         device_total_in += packet->len;
334
335         ifdebug(TRAFFIC) logger(LOG_DEBUG, _("Read packet of %d bytes from %s"), packet->len,
336                            device_info);
337
338         return true;
339 }
340
341 bool write_packet(vpn_packet_t *packet) {
342         long outlen;
343         OVERLAPPED overlapped = {0};
344
345         cp();
346
347         ifdebug(TRAFFIC) logger(LOG_DEBUG, _("Writing packet of %d bytes to %s"),
348                            packet->len, device_info);
349
350         if(!WriteFile(device_handle, packet->data, packet->len, &outlen, &overlapped)) {
351                 logger(LOG_ERR, _("Error while writing to %s %s: %s"), device_info, device, winerror(GetLastError()));
352                 return false;
353         }
354
355         device_total_out += packet->len;
356
357         return true;
358 }
359
360 void dump_device_stats(void) {
361         cp();
362
363         logger(LOG_DEBUG, _("Statistics for %s %s:"), device_info, device);
364         logger(LOG_DEBUG, _(" total bytes in:  %10d"), device_total_in);
365         logger(LOG_DEBUG, _(" total bytes out: %10d"), device_total_out);
366 }