1 |
/* |
2 |
device.c -- Interaction with Windows tap driver in a MinGW environment |
3 |
Copyright (C) 2002-2003 Ivo Timmermans <ivo@o2w.nl>, |
4 |
2002-2003 Guus Sliepen <guus@sliepen.eu.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: device.c,v 1.1.2.13 2003/08/08 19:49:47 guus Exp $ |
21 |
*/ |
22 |
|
23 |
|
24 |
#include <windows.h> |
25 |
#include <winioctl.h> |
26 |
|
27 |
|
28 |
#define REG_CONTROL_NET "SYSTEM\\CurrentControlSet\\Control\\Network\\{4D36E972-E325-11CE-BFC1-08002BE10318}" |
29 |
|
30 |
#define USERMODEDEVICEDIR "\\\\.\\" |
31 |
#define USERDEVICEDIR "\\??\\" |
32 |
#define TAPSUFFIX ".tap" |
33 |
|
34 |
#define TAP_CONTROL_CODE(request,method) CTL_CODE(FILE_DEVICE_PHYSICAL_NETCARD | 8000, request, method, FILE_ANY_ACCESS) |
35 |
|
36 |
#define TAP_IOCTL_GET_LASTMAC TAP_CONTROL_CODE(0, METHOD_BUFFERED) |
37 |
#define TAP_IOCTL_GET_MAC TAP_CONTROL_CODE(1, METHOD_BUFFERED) |
38 |
#define TAP_IOCTL_SET_STATISTICS TAP_CONTROL_CODE(2, METHOD_BUFFERED) |
39 |
|
40 |
int device_fd = 0; |
41 |
HANDLE device_handle = INVALID_HANDLE_VALUE; |
42 |
char *device = NULL; |
43 |
char *iface = NULL; |
44 |
char *device_info = NULL; |
45 |
|
46 |
int device_total_in = 0; |
47 |
int device_total_out = 0; |
48 |
|
49 |
extern char *myport; |
50 |
|
51 |
DWORD WINAPI tapreader(void *bla) { |
52 |
int sock, err, status; |
53 |
struct addrinfo *ai; |
54 |
struct addrinfo hint = { |
55 |
.ai_family = AF_UNSPEC, |
56 |
.ai_socktype = SOCK_DGRAM, |
57 |
.ai_protocol = IPPROTO_UDP, |
58 |
.ai_flags = 0, |
59 |
}; |
60 |
char buf[MTU]; |
61 |
long len; |
62 |
OVERLAPPED overlapped; |
63 |
|
64 |
/* Open a socket to the parent process */ |
65 |
|
66 |
err = getaddrinfo(NULL, myport, &hint, &ai); |
67 |
|
68 |
if(err || !ai) { |
69 |
logger(LOG_ERR, _("System call `%s' failed: %s"), "getaddrinfo", gai_strerror(errno)); |
70 |
return -1; |
71 |
} |
72 |
|
73 |
sock = socket(ai->ai_family, ai->ai_socktype, ai->ai_protocol); |
74 |
|
75 |
freeaddrinfo(ai); |
76 |
|
77 |
if(sock < 0) { |
78 |
logger(LOG_ERR, _("System call `%s' failed: %s"), "socket", strerror(errno)); |
79 |
return -1; |
80 |
} |
81 |
|
82 |
if(connect(sock, ai->ai_addr, ai->ai_addrlen)) { |
83 |
logger(LOG_ERR, _("System call `%s' failed: %s"), "connect", strerror(errno)); |
84 |
return -1; |
85 |
} |
86 |
|
87 |
logger(LOG_DEBUG, _("Tap reader running")); |
88 |
|
89 |
/* Read from tap device and send to parent */ |
90 |
|
91 |
overlapped.hEvent = CreateEvent(NULL, TRUE, FALSE, NULL); |
92 |
|
93 |
for(;;) { |
94 |
overlapped.Offset = 0; |
95 |
overlapped.OffsetHigh = 0; |
96 |
ResetEvent(overlapped.hEvent); |
97 |
|
98 |
status = ReadFile(device_handle, buf, sizeof(buf), &len, &overlapped); |
99 |
|
100 |
if(!status) { |
101 |
if(GetLastError() == ERROR_IO_PENDING) { |
102 |
WaitForSingleObject(overlapped.hEvent, INFINITE); |
103 |
if(!GetOverlappedResult(device_handle, &overlapped, &len, FALSE)) |
104 |
continue; |
105 |
} else { |
106 |
logger(LOG_ERR, _("Error while reading from %s %s: %s"), device_info, |
107 |
device, strerror(errno)); |
108 |
return -1; |
109 |
} |
110 |
} |
111 |
|
112 |
if(send(sock, buf, len, 0) <= 0) |
113 |
return -1; |
114 |
} |
115 |
} |
116 |
|
117 |
bool setup_device(void) |
118 |
{ |
119 |
HKEY key, key2; |
120 |
int i; |
121 |
|
122 |
char regpath[1024]; |
123 |
char adapterid[1024]; |
124 |
char adaptername[1024]; |
125 |
char tapname[1024]; |
126 |
long len; |
127 |
|
128 |
bool found = false; |
129 |
|
130 |
int sock, err; |
131 |
HANDLE thread; |
132 |
|
133 |
struct addrinfo *ai; |
134 |
struct addrinfo hint = { |
135 |
.ai_family = AF_UNSPEC, |
136 |
.ai_socktype = SOCK_DGRAM, |
137 |
.ai_protocol = IPPROTO_UDP, |
138 |
.ai_flags = 0, |
139 |
}; |
140 |
|
141 |
cp(); |
142 |
|
143 |
get_config_string(lookup_config(config_tree, "Device"), &device); |
144 |
get_config_string(lookup_config(config_tree, "Interface"), &iface); |
145 |
|
146 |
/* Open registry and look for network adapters */ |
147 |
|
148 |
if(RegOpenKeyEx(HKEY_LOCAL_MACHINE, REG_CONTROL_NET, 0, KEY_READ, &key)) { |
149 |
logger(LOG_ERR, _("Unable to read registry: %s"), winerror(GetLastError())); |
150 |
return false; |
151 |
} |
152 |
|
153 |
for (i = 0; ; i++) { |
154 |
len = sizeof(adapterid); |
155 |
if(RegEnumKeyEx(key, i, adapterid, &len, 0, 0, 0, NULL)) |
156 |
break; |
157 |
|
158 |
/* Find out more about this adapter */ |
159 |
|
160 |
snprintf(regpath, sizeof(regpath), "%s\\%s\\Connection", REG_CONTROL_NET, adapterid); |
161 |
|
162 |
if(RegOpenKeyEx(HKEY_LOCAL_MACHINE, regpath, 0, KEY_READ, &key2)) |
163 |
continue; |
164 |
|
165 |
len = sizeof(adaptername); |
166 |
err = RegQueryValueEx(key2, "Name", 0, 0, adaptername, &len); |
167 |
|
168 |
RegCloseKey(key2); |
169 |
|
170 |
if(err) |
171 |
continue; |
172 |
|
173 |
if(device) { |
174 |
if(!strcmp(device, adapterid)) { |
175 |
found = true; |
176 |
break; |
177 |
} else |
178 |
continue; |
179 |
} |
180 |
|
181 |
if(iface) { |
182 |
if(!strcmp(iface, adaptername)) { |
183 |
found = true; |
184 |
break; |
185 |
} else |
186 |
continue; |
187 |
} |
188 |
|
189 |
snprintf(tapname, sizeof(tapname), USERMODEDEVICEDIR "%s" TAPSUFFIX, adapterid); |
190 |
device_handle = CreateFile(tapname, GENERIC_WRITE | GENERIC_READ, 0, 0, OPEN_EXISTING, FILE_ATTRIBUTE_SYSTEM | FILE_FLAG_OVERLAPPED, 0); |
191 |
if(device_handle != INVALID_HANDLE_VALUE) { |
192 |
found = true; |
193 |
break; |
194 |
} |
195 |
} |
196 |
|
197 |
RegCloseKey(key); |
198 |
|
199 |
if(!found) { |
200 |
logger(LOG_ERR, _("No Windows tap device found!")); |
201 |
return false; |
202 |
} |
203 |
|
204 |
if(!device) |
205 |
device = xstrdup(adapterid); |
206 |
|
207 |
if(!iface) |
208 |
iface = xstrdup(adaptername); |
209 |
|
210 |
/* Try to open the corresponding tap device */ |
211 |
|
212 |
if(device_handle == INVALID_HANDLE_VALUE) { |
213 |
snprintf(tapname, sizeof(tapname), USERMODEDEVICEDIR "%s" TAPSUFFIX, device); |
214 |
device_handle = CreateFile(tapname, GENERIC_WRITE | GENERIC_READ, 0, 0, OPEN_EXISTING, FILE_ATTRIBUTE_SYSTEM | FILE_FLAG_OVERLAPPED, 0); |
215 |
} |
216 |
|
217 |
if(device_handle == INVALID_HANDLE_VALUE) { |
218 |
logger(LOG_ERR, _("%s (%s) is not a usable Windows tap device: %s"), device, iface, winerror(GetLastError())); |
219 |
return false; |
220 |
} |
221 |
|
222 |
/* Get MAC address from tap device */ |
223 |
|
224 |
if(!DeviceIoControl(device_handle, TAP_IOCTL_GET_MAC, mymac.x, sizeof(mymac.x), mymac.x, sizeof(mymac.x), &len, 0)) { |
225 |
logger(LOG_ERR, _("Could not get MAC address from Windows tap device %s (%s): %s"), device, iface, winerror(GetLastError())); |
226 |
return false; |
227 |
} |
228 |
|
229 |
if(routing_mode == RMODE_ROUTER) { |
230 |
overwrite_mac = 1; |
231 |
} |
232 |
|
233 |
/* Create a listening socket */ |
234 |
|
235 |
err = getaddrinfo(NULL, myport, &hint, &ai); |
236 |
|
237 |
if(err || !ai) { |
238 |
logger(LOG_ERR, _("System call `%s' failed: %s"), "getaddrinfo", gai_strerror(errno)); |
239 |
return false; |
240 |
} |
241 |
|
242 |
sock = socket(ai->ai_family, ai->ai_socktype, ai->ai_protocol); |
243 |
|
244 |
if(sock < 0) { |
245 |
logger(LOG_ERR, _("System call `%s' failed: %s"), "socket", strerror(errno)); |
246 |
return false; |
247 |
} |
248 |
|
249 |
if(bind(sock, ai->ai_addr, ai->ai_addrlen)) { |
250 |
logger(LOG_ERR, _("System call `%s' failed: %s"), "bind", strerror(errno)); |
251 |
return false; |
252 |
} |
253 |
|
254 |
freeaddrinfo(ai); |
255 |
|
256 |
if(listen(sock, 1)) { |
257 |
logger(LOG_ERR, _("System call `%s' failed: %s"), "listen", strerror(errno)); |
258 |
return false; |
259 |
} |
260 |
|
261 |
/* Start the tap reader */ |
262 |
|
263 |
thread = CreateThread(NULL, 0, tapreader, NULL, 0, NULL); |
264 |
|
265 |
if(!thread) { |
266 |
logger(LOG_ERR, _("System call `%s' failed: %s"), "CreateThread", winerror(GetLastError())); |
267 |
return false; |
268 |
} |
269 |
|
270 |
/* Wait for the tap reader to connect back to us */ |
271 |
|
272 |
if((device_fd = accept(sock, NULL, 0)) == -1) { |
273 |
logger(LOG_ERR, _("System call `%s' failed: %s"), "accept", strerror(errno)); |
274 |
return false; |
275 |
} |
276 |
|
277 |
closesocket(sock); |
278 |
|
279 |
device_info = _("Windows tap device"); |
280 |
|
281 |
logger(LOG_INFO, _("%s (%s) is a %s"), device, iface, device_info); |
282 |
|
283 |
return true; |
284 |
} |
285 |
|
286 |
void close_device(void) |
287 |
{ |
288 |
cp(); |
289 |
|
290 |
CloseHandle(device_handle); |
291 |
} |
292 |
|
293 |
bool read_packet(vpn_packet_t *packet) |
294 |
{ |
295 |
int lenin; |
296 |
|
297 |
cp(); |
298 |
|
299 |
if((lenin = recv(device_fd, packet->data, MTU, 0)) <= 0) { |
300 |
logger(LOG_ERR, _("Error while reading from %s %s: %s"), device_info, |
301 |
device, strerror(errno)); |
302 |
return false; |
303 |
} |
304 |
|
305 |
packet->len = lenin; |
306 |
|
307 |
device_total_in += packet->len; |
308 |
|
309 |
ifdebug(TRAFFIC) logger(LOG_DEBUG, _("Read packet of %d bytes from %s"), packet->len, |
310 |
device_info); |
311 |
|
312 |
return true; |
313 |
} |
314 |
|
315 |
bool write_packet(vpn_packet_t *packet) |
316 |
{ |
317 |
long lenout; |
318 |
OVERLAPPED overlapped = {0}; |
319 |
|
320 |
cp(); |
321 |
|
322 |
ifdebug(TRAFFIC) logger(LOG_DEBUG, _("Writing packet of %d bytes to %s"), |
323 |
packet->len, device_info); |
324 |
|
325 |
if(!WriteFile(device_handle, packet->data, packet->len, &lenout, &overlapped)) { |
326 |
logger(LOG_ERR, _("Error while writing to %s %s: %s"), device_info, device, winerror(GetLastError())); |
327 |
return false; |
328 |
} |
329 |
|
330 |
device_total_out += packet->len; |
331 |
|
332 |
return true; |
333 |
} |
334 |
|
335 |
void dump_device_stats(void) |
336 |
{ |
337 |
cp(); |
338 |
|
339 |
logger(LOG_DEBUG, _("Statistics for %s %s:"), device_info, device); |
340 |
logger(LOG_DEBUG, _(" total bytes in: %10d"), device_total_in); |
341 |
logger(LOG_DEBUG, _(" total bytes out: %10d"), device_total_out); |
342 |
} |