slirp.c@ 34209

Last change on this file since 34209 was 34209, checked in by vboxsync, 14 years ago
NAT: added 'info nat%d' handler
Property svn:eol-style set to `native` Property svn:keywords set to `Author Date Id Revision`
File size: 64.7 KB

Line
1	/* $Id: slirp.c 34209 2010-11-19 16:04:55Z vboxsync $ */
2	/** @file
3	* NAT - slirp glue.
4	*/
5
6	/*
7	* Copyright (C) 2006-2010 Oracle Corporation
8	*
9	* This file is part of VirtualBox Open Source Edition (OSE), as
10	* available from http://www.virtualbox.org. This file is free software;
11	* you can redistribute it and/or modify it under the terms of the GNU
12	* General Public License (GPL) as published by the Free Software
13	* Foundation, in version 2 as it comes in the "COPYING" file of the
14	* VirtualBox OSE distribution. VirtualBox OSE is distributed in the
15	* hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
16	*/
17
18	/*
19	* This code is based on:
20	*
21	* libslirp glue
22	*
23	* Copyright (c) 2004-2008 Fabrice Bellard
24	*
25	* Permission is hereby granted, free of charge, to any person obtaining a copy
26	* of this software and associated documentation files (the "Software"), to deal
27	* in the Software without restriction, including without limitation the rights
28	* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
29	* copies of the Software, and to permit persons to whom the Software is
30	* furnished to do so, subject to the following conditions:
31	*
32	* The above copyright notice and this permission notice shall be included in
33	* all copies or substantial portions of the Software.
34	*
35	* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
36	* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
37	* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
38	* THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
39	* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
40	* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
41	* THE SOFTWARE.
42	*/
43
44	#include "slirp.h"
45	#ifdef RT_OS_OS2
46	# include <paths.h>
47	#endif
48
49	#include <VBox/err.h>
50	#include <VBox/pdmdrv.h>
51	#include <iprt/assert.h>
52	#include <iprt/file.h>
53	#ifndef RT_OS_WINDOWS
54	# include <sys/ioctl.h>
55	# include <poll.h>
56	#else
57	# include <Winnls.h>
58	# define _WINSOCK2API_
59	# include <IPHlpApi.h>
60	#endif
61	#include <alias.h>
62
63	#ifndef RT_OS_WINDOWS
64
65	# define DO_ENGAGE_EVENT1(so, fdset, label) \
66	do { \
67	if ( so->so_poll_index != -1 \
68	&& so->s == polls[so->so_poll_index].fd) \
69	{ \
70	polls[so->so_poll_index].events \|= N_(fdset ## _poll); \
71	break; \
72	} \
73	AssertRelease(poll_index < (nfds)); \
74	AssertRelease(poll_index >= 0 && poll_index < (nfds)); \
75	polls[poll_index].fd = (so)->s; \
76	(so)->so_poll_index = poll_index; \
77	polls[poll_index].events = N_(fdset ## _poll); \
78	polls[poll_index].revents = 0; \
79	poll_index++; \
80	} while (0)
81
82	# define DO_ENGAGE_EVENT2(so, fdset1, fdset2, label) \
83	do { \
84	if ( so->so_poll_index != -1 \
85	&& so->s == polls[so->so_poll_index].fd) \
86	{ \
87	polls[so->so_poll_index].events \|= \
88	N_(fdset1 ## _poll) \| N_(fdset2 ## _poll); \
89	break; \
90	} \
91	AssertRelease(poll_index < (nfds)); \
92	polls[poll_index].fd = (so)->s; \
93	(so)->so_poll_index = poll_index; \
94	polls[poll_index].events = \
95	N_(fdset1 ## _poll) \| N_(fdset2 ## _poll); \
96	poll_index++; \
97	} while (0)
98
99	# define DO_POLL_EVENTS(rc, error, so, events, label) do {} while (0)
100
101	/*
102	* DO_CHECK_FD_SET is used in dumping events on socket, including POLLNVAL.
103	* gcc warns about attempts to log POLLNVAL so construction in a last to lines
104	* used to catch POLLNVAL while logging and return false in case of error while
105	* normal usage.
106	*/
107	# define DO_CHECK_FD_SET(so, events, fdset) \
108	( ((so)->so_poll_index != -1) \
109	&& ((so)->so_poll_index <= ndfs) \
110	&& ((so)->s == polls[so->so_poll_index].fd) \
111	&& (polls[(so)->so_poll_index].revents & N_(fdset ## _poll)) \
112	&& ( N_(fdset ## _poll) == POLLNVAL \
113	\|\| !(polls[(so)->so_poll_index].revents & POLLNVAL)))
114
115	/* specific for Unix API */
116	# define DO_UNIX_CHECK_FD_SET(so, events, fdset) DO_CHECK_FD_SET((so), (events), fdset)
117	/* specific for Windows Winsock API */
118	# define DO_WIN_CHECK_FD_SET(so, events, fdset) 0
119
120	# ifndef RT_OS_LINUX
121	# define readfds_poll (POLLRDNORM)
122	# define writefds_poll (POLLWRNORM)
123	# else
124	# define readfds_poll (POLLIN)
125	# define writefds_poll (POLLOUT)
126	# endif
127	# define xfds_poll (POLLPRI)
128	# define closefds_poll (POLLHUP)
129	# define rderr_poll (POLLERR)
130	# define rdhup_poll (POLLHUP)
131	# define nval_poll (POLLNVAL)
132
133	# define ICMP_ENGAGE_EVENT(so, fdset) \
134	do { \
135	if (pData->icmp_socket.s != -1) \
136	DO_ENGAGE_EVENT1((so), fdset, ICMP); \
137	} while (0)
138
139	#else /* RT_OS_WINDOWS */
140
141	/*
142	* On Windows, we will be notified by IcmpSendEcho2() when the response arrives.
143	* So no call to WSAEventSelect necessary.
144	*/
145	# define ICMP_ENGAGE_EVENT(so, fdset) do {} while (0)
146
147	/*
148	* On Windows we use FD_ALL_EVENTS to ensure that we don't miss any event.
149	*/
150	# define DO_ENGAGE_EVENT1(so, fdset1, label) \
151	do { \
152	rc = WSAEventSelect((so)->s, VBOX_SOCKET_EVENT, FD_ALL_EVENTS); \
153	if (rc == SOCKET_ERROR) \
154	{ \
155	/* This should not happen */ \
156	error = WSAGetLastError(); \
157	LogRel(("WSAEventSelect (" #label ") error %d (so=%x, socket=%s, event=%x)\n", \
158	error, (so), (so)->s, VBOX_SOCKET_EVENT)); \
159	} \
160	} while (0); \
161	CONTINUE(label)
162
163	# define DO_ENGAGE_EVENT2(so, fdset1, fdset2, label) \
164	DO_ENGAGE_EVENT1((so), (fdset1), label)
165
166	# define DO_POLL_EVENTS(rc, error, so, events, label) \
167	(rc) = WSAEnumNetworkEvents((so)->s, VBOX_SOCKET_EVENT, (events)); \
168	if ((rc) == SOCKET_ERROR) \
169	{ \
170	(error) = WSAGetLastError(); \
171	LogRel(("WSAEnumNetworkEvents " #label " error %d\n", (error))); \
172	CONTINUE(label); \
173	}
174
175	# define acceptds_win FD_ACCEPT
176	# define acceptds_win_bit FD_ACCEPT_BIT
177	# define readfds_win FD_READ
178	# define readfds_win_bit FD_READ_BIT
179	# define writefds_win FD_WRITE
180	# define writefds_win_bit FD_WRITE_BIT
181	# define xfds_win FD_OOB
182	# define xfds_win_bit FD_OOB_BIT
183	# define closefds_win FD_CLOSE
184	# define closefds_win_bit FD_CLOSE_BIT
185
186	# define closefds_win FD_CLOSE
187	# define closefds_win_bit FD_CLOSE_BIT
188
189	# define DO_CHECK_FD_SET(so, events, fdset) \
190	(((events).lNetworkEvents & fdset ## _win) && ((events).iErrorCode[fdset ## _win_bit] == 0))
191
192	# define DO_WIN_CHECK_FD_SET(so, events, fdset) DO_CHECK_FD_SET((so), (events), fdset)
193	# define DO_UNIX_CHECK_FD_SET(so, events, fdset) 1 /specific for Unix API /
194
195	#endif /* RT_OS_WINDOWS */
196
197	#define TCP_ENGAGE_EVENT1(so, fdset) \
198	DO_ENGAGE_EVENT1((so), fdset, tcp)
199
200	#define TCP_ENGAGE_EVENT2(so, fdset1, fdset2) \
201	DO_ENGAGE_EVENT2((so), fdset1, fdset2, tcp)
202
203	#define UDP_ENGAGE_EVENT(so, fdset) \
204	DO_ENGAGE_EVENT1((so), fdset, udp)
205
206	#define POLL_TCP_EVENTS(rc, error, so, events) \
207	DO_POLL_EVENTS((rc), (error), (so), (events), tcp)
208
209	#define POLL_UDP_EVENTS(rc, error, so, events) \
210	DO_POLL_EVENTS((rc), (error), (so), (events), udp)
211
212	#define CHECK_FD_SET(so, events, set) \
213	(DO_CHECK_FD_SET((so), (events), set))
214
215	#define WIN_CHECK_FD_SET(so, events, set) \
216	(DO_WIN_CHECK_FD_SET((so), (events), set))
217
218	#define UNIX_CHECK_FD_SET(so, events, set) \
219	(DO_UNIX_CHECK_FD_SET(so, events, set))
220
221	/*
222	* Loging macros
223	*/
224	#if VBOX_WITH_DEBUG_NAT_SOCKETS
225	# if defined(RT_OS_WINDOWS)
226	# define DO_LOG_NAT_SOCK(so, proto, winevent, r_fdset, w_fdset, x_fdset) \
227	do { \
228	LogRel((" " #proto " %R[natsock] %R[natwinnetevents]\n", (so), (winevent))); \
229	} while (0)
230	# else /* !RT_OS_WINDOWS */
231	# define DO_LOG_NAT_SOCK(so, proto, winevent, r_fdset, w_fdset, x_fdset) \
232	do { \
233	LogRel((" " #proto " %R[natsock] %s %s %s er: %s, %s, %s\n", (so), \
234	CHECK_FD_SET(so, ign ,r_fdset) ? "READ":"", \
235	CHECK_FD_SET(so, ign, w_fdset) ? "WRITE":"", \
236	CHECK_FD_SET(so, ign, x_fdset) ? "OOB":"", \
237	CHECK_FD_SET(so, ign, rderr) ? "RDERR":"", \
238	CHECK_FD_SET(so, ign, rdhup) ? "RDHUP":"", \
239	CHECK_FD_SET(so, ign, nval) ? "RDNVAL":"")); \
240	} while (0)
241	# endif /* !RT_OS_WINDOWS */
242	#else /* !VBOX_WITH_DEBUG_NAT_SOCKETS */
243	# define DO_LOG_NAT_SOCK(so, proto, winevent, r_fdset, w_fdset, x_fdset) do {} while (0)
244	#endif /* !VBOX_WITH_DEBUG_NAT_SOCKETS */
245
246	#define LOG_NAT_SOCK(so, proto, winevent, r_fdset, w_fdset, x_fdset) \
247	DO_LOG_NAT_SOCK((so), proto, (winevent), r_fdset, w_fdset, x_fdset)
248
249	static void activate_port_forwarding(PNATState, const uint8_t *pEther);
250
251	static const uint8_t special_ethaddr[6] =
252	{
253	0x52, 0x54, 0x00, 0x12, 0x35, 0x00
254	};
255
256	static const uint8_t broadcast_ethaddr[6] =
257	{
258	0xff, 0xff, 0xff, 0xff, 0xff, 0xff
259	};
260
261	const uint8_t zerro_ethaddr[6] =
262	{
263	0x0, 0x0, 0x0, 0x0, 0x0, 0x0
264	};
265
266	#ifdef RT_OS_WINDOWS
267	static int get_dns_addr_domain(PNATState pData, bool fVerbose,
268	struct in_addr *pdns_addr,
269	const char **ppszDomain)
270	{
271	ULONG flags = GAA_FLAG_INCLUDE_PREFIX; /GAA_FLAG_INCLUDE_ALL_INTERFACES;/ /* all interfaces registered in NDIS */
272	PIP_ADAPTER_ADDRESSES pAdapterAddr = NULL;
273	PIP_ADAPTER_ADDRESSES pAddr = NULL;
274	PIP_ADAPTER_DNS_SERVER_ADDRESS pDnsAddr = NULL;
275	ULONG size;
276	int wlen = 0;
277	char *pszSuffix;
278	struct dns_domain_entry *pDomain = NULL;
279	ULONG ret = ERROR_SUCCESS;
280
281	/* @todo add SKIPing flags to get only required information */
282
283	/* determine size of buffer */
284	size = 0;
285	ret = pData->pfGetAdaptersAddresses(AF_INET, 0, NULL /* reserved */, pAdapterAddr, &size);
286	if (ret != ERROR_BUFFER_OVERFLOW)
287	{
288	Log(("NAT: error %lu occurred on capacity detection operation\n", ret));
289	return -1;
290	}
291	if (size == 0)
292	{
293	Log(("NAT: Win socket API returns non capacity\n"));
294	return -1;
295	}
296
297	pAdapterAddr = RTMemAllocZ(size);
298	if (!pAdapterAddr)
299	{
300	Log(("NAT: No memory available\n"));
301	return -1;
302	}
303	ret = pData->pfGetAdaptersAddresses(AF_INET, 0, NULL /* reserved */, pAdapterAddr, &size);
304	if (ret != ERROR_SUCCESS)
305	{
306	Log(("NAT: error %lu occurred on fetching adapters info\n", ret));
307	RTMemFree(pAdapterAddr);
308	return -1;
309	}
310
311	for (pAddr = pAdapterAddr; pAddr != NULL; pAddr = pAddr->Next)
312	{
313	int found;
314	if (pAddr->OperStatus != IfOperStatusUp)
315	continue;
316
317	for (pDnsAddr = pAddr->FirstDnsServerAddress; pDnsAddr != NULL; pDnsAddr = pDnsAddr->Next)
318	{
319	struct sockaddr *SockAddr = pDnsAddr->Address.lpSockaddr;
320	struct in_addr InAddr;
321	struct dns_entry *pDns;
322
323	if (SockAddr->sa_family != AF_INET)
324	continue;
325
326	InAddr = ((struct sockaddr_in *)SockAddr)->sin_addr;
327
328	/* add dns server to list */
329	pDns = RTMemAllocZ(sizeof(struct dns_entry));
330	if (!pDns)
331	{
332	Log(("NAT: Can't allocate buffer for DNS entry\n"));
333	RTMemFree(pAdapterAddr);
334	return VERR_NO_MEMORY;
335	}
336
337	Log(("NAT: adding %R[IP4] to DNS server list\n", &InAddr));
338	if ((InAddr.s_addr & RT_H2N_U32_C(IN_CLASSA_NET)) == RT_N2H_U32_C(INADDR_LOOPBACK & IN_CLASSA_NET))
339	pDns->de_addr.s_addr = RT_H2N_U32(RT_N2H_U32(pData->special_addr.s_addr) \| CTL_ALIAS);
340	else
341	pDns->de_addr.s_addr = InAddr.s_addr;
342
343	TAILQ_INSERT_HEAD(&pData->pDnsList, pDns, de_list);
344
345	if (pAddr->DnsSuffix == NULL)
346	continue;
347
348	/* uniq */
349	RTUtf16ToUtf8(pAddr->DnsSuffix, &pszSuffix);
350	if (!pszSuffix \|\| strlen(pszSuffix) == 0)
351	{
352	RTStrFree(pszSuffix);
353	continue;
354	}
355
356	found = 0;
357	LIST_FOREACH(pDomain, &pData->pDomainList, dd_list)
358	{
359	if ( pDomain->dd_pszDomain != NULL
360	&& strcmp(pDomain->dd_pszDomain, pszSuffix) == 0)
361	{
362	found = 1;
363	RTStrFree(pszSuffix);
364	break;
365	}
366	}
367	if (!found)
368	{
369	pDomain = RTMemAllocZ(sizeof(struct dns_domain_entry));
370	if (!pDomain)
371	{
372	Log(("NAT: not enough memory\n"));
373	RTStrFree(pszSuffix);
374	RTMemFree(pAdapterAddr);
375	return VERR_NO_MEMORY;
376	}
377	pDomain->dd_pszDomain = pszSuffix;
378	Log(("NAT: adding domain name %s to search list\n", pDomain->dd_pszDomain));
379	LIST_INSERT_HEAD(&pData->pDomainList, pDomain, dd_list);
380	}
381	}
382	}
383	RTMemFree(pAdapterAddr);
384	return 0;
385	}
386
387	#else /* !RT_OS_WINDOWS */
388
389	static int RTFileGets(RTFILE File, void pvBuf, size_t cbBufSize, size_t pcbRead)
390	{
391	size_t cbRead;
392	char bTest;
393	int rc = VERR_NO_MEMORY;
394	char pu8Buf = (char )pvBuf;
395	*pcbRead = 0;
396
397	while ( RT_SUCCESS(rc = RTFileRead(File, &bTest, 1, &cbRead))
398	&& (pu8Buf - (char *)pvBuf) < cbBufSize)
399	{
400	if (cbRead == 0)
401	return VERR_EOF;
402
403	if (bTest == '\r' \|\| bTest == '\n')
404	{
405	*pu8Buf = 0;
406	return VINF_SUCCESS;
407	}
408	*pu8Buf = bTest;
409	pu8Buf++;
410	(*pcbRead)++;
411	}
412	return rc;
413	}
414
415	static int get_dns_addr_domain(PNATState pData, bool fVerbose,
416	struct in_addr *pdns_addr,
417	const char **ppszDomain)
418	{
419	char buff[512];
420	char buff2[256];
421	RTFILE f;
422	int cNameserversFound = 0;
423	bool fWarnTooManyDnsServers = false;
424	struct in_addr tmp_addr;
425	int rc;
426	size_t bytes;
427
428	# ifdef RT_OS_OS2
429	/* Try various locations. */
430	char *etc = getenv("ETC");
431	if (etc)
432	{
433	RTStrmPrintf(buff, sizeof(buff), "%s/RESOLV2", etc);
434	rc = RTFileOpen(&f, buff, RTFILE_O_READ \| RTFILE_O_OPEN \| RTFILE_O_DENY_NONE);
435	}
436	if (RT_FAILURE(rc))
437	{
438	RTStrmPrintf(buff, sizeof(buff), "%s/RESOLV2", _PATH_ETC);
439	rc = RTFileOpen(&f, buff, RTFILE_O_READ \| RTFILE_O_OPEN \| RTFILE_O_DENY_NONE);
440	}
441	if (RT_FAILURE(rc))
442	{
443	RTStrmPrintf(buff, sizeof(buff), "%s/resolv.conf", _PATH_ETC);
444	rc = RTFileOpen(&f, buff, RTFILE_O_READ \| RTFILE_O_OPEN \| RTFILE_O_DENY_NONE);
445	}
446	# else /* !RT_OS_OS2 */
447	# ifndef DEBUG_vvl
448	rc = RTFileOpen(&f, "/etc/resolv.conf", RTFILE_O_READ \| RTFILE_O_OPEN \| RTFILE_O_DENY_NONE);
449	# else
450	char *home = getenv("HOME");
451	RTStrPrintf(buff, sizeof(buff), "%s/resolv.conf", home);
452	rc = RTFileOpen(&f, buff, RTFILE_O_READ \| RTFILE_O_OPEN \| RTFILE_O_DENY_NONE);
453	if (RT_SUCCESS(rc))
454	{
455	Log(("NAT: DNS we're using %s\n", buff));
456	}
457	else
458	{
459	rc = RTFileOpen(&f, "/etc/resolv.conf", RTFILE_O_READ \| RTFILE_O_OPEN \| RTFILE_O_DENY_NONE);
460	Log(("NAT: DNS we're using %s\n", buff));
461	}
462	# endif
463	# endif /* !RT_OS_OS2 */
464	if (RT_FAILURE(rc))
465	return -1;
466
467	if (ppszDomain)
468	*ppszDomain = NULL;
469
470	Log(("NAT: DNS Servers:\n"));
471	while ( RT_SUCCESS(rc = RTFileGets(f, buff, sizeof(buff), &bytes))
472	&& rc != VERR_EOF)
473	{
474	struct dns_entry *pDns = NULL;
475	if ( cNameserversFound == 4
476	&& !fWarnTooManyDnsServers
477	&& sscanf(buff, "nameserver%*[ \t]%255s", buff2) == 1)
478	{
479	fWarnTooManyDnsServers = true;
480	LogRel(("NAT: too many nameservers registered.\n"));
481	}
482	if ( sscanf(buff, "nameserver%*[ \t]%255s", buff2) == 1
483	&& cNameserversFound < 4) /* Unix doesn't accept more than 4 name servers*/
484	{
485	if (!inet_aton(buff2, &tmp_addr))
486	continue;
487
488	/* localhost mask */
489	pDns = RTMemAllocZ(sizeof (struct dns_entry));
490	if (!pDns)
491	{
492	Log(("can't alloc memory for DNS entry\n"));
493	return -1;
494	}
495
496	/* check */
497	pDns->de_addr.s_addr = tmp_addr.s_addr;
498	if ((pDns->de_addr.s_addr & RT_H2N_U32_C(IN_CLASSA_NET)) == RT_N2H_U32_C(INADDR_LOOPBACK & IN_CLASSA_NET))
499	{
500	pDns->de_addr.s_addr = RT_H2N_U32(RT_N2H_U32(pData->special_addr.s_addr) \| CTL_ALIAS);
501	}
502	TAILQ_INSERT_HEAD(&pData->pDnsList, pDns, de_list);
503	cNameserversFound++;
504	}
505	if ((!strncmp(buff, "domain", 6) \|\| !strncmp(buff, "search", 6)))
506	{
507	char *tok;
508	char *saveptr;
509	struct dns_domain_entry *pDomain = NULL;
510	int fFoundDomain = 0;
511	tok = strtok_r(&buff[6], " \t\n", &saveptr);
512	LIST_FOREACH(pDomain, &pData->pDomainList, dd_list)
513	{
514	if ( tok != NULL
515	&& strcmp(tok, pDomain->dd_pszDomain) == 0)
516	{
517	fFoundDomain = 1;
518	break;
519	}
520	}
521	if (tok != NULL && !fFoundDomain)
522	{
523	pDomain = RTMemAllocZ(sizeof(struct dns_domain_entry));
524	if (!pDomain)
525	{
526	Log(("NAT: not enought memory to add domain list\n"));
527	return VERR_NO_MEMORY;
528	}
529	pDomain->dd_pszDomain = RTStrDup(tok);
530	Log(("NAT: adding domain name %s to search list\n", pDomain->dd_pszDomain));
531	LIST_INSERT_HEAD(&pData->pDomainList, pDomain, dd_list);
532	}
533	}
534	}
535	RTFileClose(f);
536	if (!cNameserversFound)
537	return -1;
538	return 0;
539	}
540
541	#endif /* !RT_OS_WINDOWS */
542
543	int slirp_init_dns_list(PNATState pData)
544	{
545	TAILQ_INIT(&pData->pDnsList);
546	LIST_INIT(&pData->pDomainList);
547	return get_dns_addr_domain(pData, true, NULL, NULL);
548	}
549
550	void slirp_release_dns_list(PNATState pData)
551	{
552	struct dns_entry *pDns = NULL;
553	struct dns_domain_entry *pDomain = NULL;
554
555	while (!TAILQ_EMPTY(&pData->pDnsList))
556	{
557	pDns = TAILQ_FIRST(&pData->pDnsList);
558	TAILQ_REMOVE(&pData->pDnsList, pDns, de_list);
559	RTMemFree(pDns);
560	}
561
562	while (!LIST_EMPTY(&pData->pDomainList))
563	{
564	pDomain = LIST_FIRST(&pData->pDomainList);
565	LIST_REMOVE(pDomain, dd_list);
566	if (pDomain->dd_pszDomain != NULL)
567	RTStrFree(pDomain->dd_pszDomain);
568	RTMemFree(pDomain);
569	}
570	}
571
572	int get_dns_addr(PNATState pData, struct in_addr *pdns_addr)
573	{
574	return get_dns_addr_domain(pData, false, pdns_addr, NULL);
575	}
576
577	int slirp_init(PNATState *ppData, uint32_t u32NetAddr, uint32_t u32Netmask,
578	bool fPassDomain, bool fUseHostResolver, int i32AliasMode, void *pvUser)
579	{
580	int fNATfailed = 0;
581	int rc;
582	PNATState pData = RTMemAllocZ(sizeof(NATState));
583	*ppData = pData;
584	if (!pData)
585	return VERR_NO_MEMORY;
586	if (u32Netmask & 0x1f)
587	/* CTL is x.x.x.15, bootp passes up to 16 IPs (15..31) */
588	return VERR_INVALID_PARAMETER;
589	pData->fPassDomain = !fUseHostResolver ? fPassDomain : false;
590	pData->fUseHostResolver = fUseHostResolver;
591	pData->pvUser = pvUser;
592	pData->netmask = u32Netmask;
593
594	/* sockets & TCP defaults */
595	pData->socket_rcv = 64 * _1K;
596	pData->socket_snd = 64 * _1K;
597	tcp_sndspace = 64 * _1K;
598	tcp_rcvspace = 64 * _1K;
599
600	#ifdef RT_OS_WINDOWS
601	{
602	WSADATA Data;
603	WSAStartup(MAKEWORD(2, 0), &Data);
604	}
605	pData->phEvents[VBOX_SOCKET_EVENT_INDEX] = CreateEvent(NULL, FALSE, FALSE, NULL);
606	#endif
607	#ifdef VBOX_WITH_SLIRP_MT
608	QSOCKET_LOCK_CREATE(tcb);
609	QSOCKET_LOCK_CREATE(udb);
610	rc = RTReqCreateQueue(&pData->pReqQueue);
611	AssertReleaseRC(rc);
612	#endif
613
614	link_up = 1;
615
616	rc = bootp_dhcp_init(pData);
617	if (rc != 0)
618	{
619	Log(("NAT: DHCP server initialization was failed\n"));
620	return VINF_NAT_DNS;
621	}
622	debug_init();
623	if_init(pData);
624	ip_init(pData);
625	icmp_init(pData);
626
627	/* Initialise mbufs after setting the MTU */
628	mbuf_init(pData);
629
630	pData->special_addr.s_addr = u32NetAddr;
631	pData->slirp_ethaddr = &special_ethaddr[0];
632	alias_addr.s_addr = pData->special_addr.s_addr \| RT_H2N_U32_C(CTL_ALIAS);
633	/* @todo: add ability to configure this staff */
634
635	/* set default addresses */
636	inet_aton("127.0.0.1", &loopback_addr);
637	if (!pData->fUseHostResolver)
638	{
639	if (slirp_init_dns_list(pData) < 0)
640	fNATfailed = 1;
641
642	dnsproxy_init(pData);
643	}
644	if (i32AliasMode & ~(PKT_ALIAS_LOG\|PKT_ALIAS_SAME_PORTS\|PKT_ALIAS_PROXY_ONLY))
645	{
646	Log(("NAT: alias mode %x is ignored\n", i32AliasMode));
647	i32AliasMode = 0;
648	}
649	pData->i32AliasMode = i32AliasMode;
650	getouraddr(pData);
651	{
652	int flags = 0;
653	struct in_addr proxy_addr;
654	pData->proxy_alias = LibAliasInit(pData, NULL);
655	if (pData->proxy_alias == NULL)
656	{
657	Log(("NAT: LibAlias default rule wasn't initialized\n"));
658	AssertMsgFailed(("NAT: LibAlias default rule wasn't initialized\n"));
659	}
660	flags = LibAliasSetMode(pData->proxy_alias, 0, 0);
661	#ifndef NO_FW_PUNCH
662	flags \|= PKT_ALIAS_PUNCH_FW;
663	#endif
664	flags \|= pData->i32AliasMode; /* do transparent proxying */
665	flags = LibAliasSetMode(pData->proxy_alias, flags, ~0);
666	proxy_addr.s_addr = RT_H2N_U32(RT_N2H_U32(pData->special_addr.s_addr) \| CTL_ALIAS);
667	LibAliasSetAddress(pData->proxy_alias, proxy_addr);
668	ftp_alias_load(pData);
669	nbt_alias_load(pData);
670	if (pData->fUseHostResolver)
671	dns_alias_load(pData);
672	}
673	return fNATfailed ? VINF_NAT_DNS : VINF_SUCCESS;
674	}
675
676	/**
677	* Register statistics.
678	*/
679	void slirp_register_statistics(PNATState pData, PPDMDRVINS pDrvIns)
680	{
681	#ifdef VBOX_WITH_STATISTICS
682	# define PROFILE_COUNTER(name, dsc) REGISTER_COUNTER(name, pData, STAMTYPE_PROFILE, STAMUNIT_TICKS_PER_CALL, dsc)
683	# define COUNTING_COUNTER(name, dsc) REGISTER_COUNTER(name, pData, STAMTYPE_COUNTER, STAMUNIT_COUNT, dsc)
684	# include "counters.h"
685	# undef COUNTER
686	/** @todo register statistics for the variables dumped by:
687	* ipstats(pData); tcpstats(pData); udpstats(pData); icmpstats(pData);
688	* mbufstats(pData); sockstats(pData); */
689	#endif /* VBOX_WITH_STATISTICS */
690	}
691
692	/**
693	* Deregister statistics.
694	*/
695	void slirp_deregister_statistics(PNATState pData, PPDMDRVINS pDrvIns)
696	{
697	if (pData == NULL)
698	return;
699	#ifdef VBOX_WITH_STATISTICS
700	# define PROFILE_COUNTER(name, dsc) DEREGISTER_COUNTER(name, pData)
701	# define COUNTING_COUNTER(name, dsc) DEREGISTER_COUNTER(name, pData)
702	# include "counters.h"
703	#endif /* VBOX_WITH_STATISTICS */
704	}
705
706	/**
707	* Marks the link as up, making it possible to establish new connections.
708	*/
709	void slirp_link_up(PNATState pData)
710	{
711	struct arp_cache_entry *ac;
712	link_up = 1;
713
714	if (LIST_EMPTY(&pData->arp_cache))
715	return;
716
717	LIST_FOREACH(ac, &pData->arp_cache, list)
718	{
719	activate_port_forwarding(pData, ac->ether);
720	}
721	}
722
723	/**
724	* Marks the link as down and cleans up the current connections.
725	*/
726	void slirp_link_down(PNATState pData)
727	{
728	struct socket *so;
729	struct port_forward_rule *rule;
730
731	while ((so = tcb.so_next) != &tcb)
732	{
733	if (so->so_state & SS_NOFDREF \|\| so->s == -1)
734	sofree(pData, so);
735	else
736	tcp_drop(pData, sototcpcb(so), 0);
737	}
738
739	while ((so = udb.so_next) != &udb)
740	udp_detach(pData, so);
741
742	/*
743	* Clear the active state of port-forwarding rules to force
744	* re-setup on restoration of communications.
745	*/
746	LIST_FOREACH(rule, &pData->port_forward_rule_head, list)
747	{
748	rule->activated = 0;
749	}
750	pData->cRedirectionsActive = 0;
751
752	link_up = 0;
753	}
754
755	/**
756	* Terminates the slirp component.
757	*/
758	void slirp_term(PNATState pData)
759	{
760	if (pData == NULL)
761	return;
762	#ifdef RT_OS_WINDOWS
763	pData->pfIcmpCloseHandle(pData->icmp_socket.sh);
764	FreeLibrary(pData->hmIcmpLibrary);
765	RTMemFree(pData->pvIcmpBuffer);
766	#else
767	closesocket(pData->icmp_socket.s);
768	#endif
769
770	slirp_link_down(pData);
771	slirp_release_dns_list(pData);
772	ftp_alias_unload(pData);
773	nbt_alias_unload(pData);
774	if (pData->fUseHostResolver)
775	dns_alias_unload(pData);
776	while (!LIST_EMPTY(&instancehead))
777	{
778	struct libalias *la = LIST_FIRST(&instancehead);
779	/* libalias do all clean up */
780	LibAliasUninit(la);
781	}
782	while (!LIST_EMPTY(&pData->arp_cache))
783	{
784	struct arp_cache_entry *ac = LIST_FIRST(&pData->arp_cache);
785	LIST_REMOVE(ac, list);
786	RTMemFree(ac);
787	}
788	bootp_dhcp_fini(pData);
789	m_fini(pData);
790	#ifdef RT_OS_WINDOWS
791	WSACleanup();
792	#endif
793	#ifndef VBOX_WITH_SLIRP_BSD_SBUF
794	#ifdef LOG_ENABLED
795	Log(("\n"
796	"NAT statistics\n"
797	"--------------\n"
798	"\n"));
799	ipstats(pData);
800	tcpstats(pData);
801	udpstats(pData);
802	icmpstats(pData);
803	mbufstats(pData);
804	sockstats(pData);
805	Log(("\n"
806	"\n"
807	"\n"));
808	#endif
809	#endif
810	RTMemFree(pData);
811	}
812
813
814	#define CONN_CANFSEND(so) (((so)->so_state & (SS_FCANTSENDMORE\|SS_ISFCONNECTED)) == SS_ISFCONNECTED)
815	#define CONN_CANFRCV(so) (((so)->so_state & (SS_FCANTRCVMORE\|SS_ISFCONNECTED)) == SS_ISFCONNECTED)
816
817	/*
818	* curtime kept to an accuracy of 1ms
819	*/
820	static void updtime(PNATState pData)
821	{
822	#ifdef RT_OS_WINDOWS
823	struct _timeb tb;
824
825	_ftime(&tb);
826	curtime = (u_int)tb.time * (u_int)1000;
827	curtime += (u_int)tb.millitm;
828	#else
829	gettimeofday(&tt, 0);
830
831	curtime = (u_int)tt.tv_sec * (u_int)1000;
832	curtime += (u_int)tt.tv_usec / (u_int)1000;
833
834	if ((tt.tv_usec % 1000) >= 500)
835	curtime++;
836	#endif
837	}
838
839	#ifdef RT_OS_WINDOWS
840	void slirp_select_fill(PNATState pData, int *pnfds)
841	#else /* RT_OS_WINDOWS */
842	void slirp_select_fill(PNATState pData, int pnfds, struct pollfd polls)
843	#endif /* !RT_OS_WINDOWS */
844	{
845	struct socket so, so_next;
846	int nfds;
847	#if defined(RT_OS_WINDOWS)
848	int rc;
849	int error;
850	#else
851	int poll_index = 0;
852	#endif
853	int i;
854
855	STAM_PROFILE_START(&pData->StatFill, a);
856
857	nfds = *pnfds;
858
859	/*
860	* First, TCP sockets
861	*/
862	do_slowtimo = 0;
863	if (!link_up)
864	goto done;
865
866	/*
867	* *_slowtimo needs calling if there are IP fragments
868	* in the fragment queue, or there are TCP connections active
869	*/
870	/* XXX:
871	* triggering of fragment expiration should be the same but use new macroses
872	*/
873	do_slowtimo = (tcb.so_next != &tcb);
874	if (!do_slowtimo)
875	{
876	for (i = 0; i < IPREASS_NHASH; i++)
877	{
878	if (!TAILQ_EMPTY(&ipq[i]))
879	{
880	do_slowtimo = 1;
881	break;
882	}
883	}
884	}
885	/* always add the ICMP socket */
886	#ifndef RT_OS_WINDOWS
887	pData->icmp_socket.so_poll_index = -1;
888	#endif
889	ICMP_ENGAGE_EVENT(&pData->icmp_socket, readfds);
890
891	STAM_COUNTER_RESET(&pData->StatTCP);
892	STAM_COUNTER_RESET(&pData->StatTCPHot);
893
894	QSOCKET_FOREACH(so, so_next, tcp)
895	/* { */
896	#if !defined(RT_OS_WINDOWS)
897	so->so_poll_index = -1;
898	#endif
899	STAM_COUNTER_INC(&pData->StatTCP);
900
901	/*
902	* See if we need a tcp_fasttimo
903	*/
904	if ( time_fasttimo == 0
905	&& so->so_tcpcb != NULL
906	&& so->so_tcpcb->t_flags & TF_DELACK)
907	{
908	time_fasttimo = curtime; /* Flag when we want a fasttimo */
909	}
910
911	/*
912	* NOFDREF can include still connecting to local-host,
913	* newly socreated() sockets etc. Don't want to select these.
914	*/
915	if (so->so_state & SS_NOFDREF \|\| so->s == -1)
916	CONTINUE(tcp);
917
918	/*
919	* Set for reading sockets which are accepting
920	*/
921	if (so->so_state & SS_FACCEPTCONN)
922	{
923	STAM_COUNTER_INC(&pData->StatTCPHot);
924	TCP_ENGAGE_EVENT1(so, readfds);
925	CONTINUE(tcp);
926	}
927
928	/*
929	* Set for writing sockets which are connecting
930	*/
931	if (so->so_state & SS_ISFCONNECTING)
932	{
933	Log2(("connecting %R[natsock] engaged\n",so));
934	STAM_COUNTER_INC(&pData->StatTCPHot);
935	TCP_ENGAGE_EVENT1(so, writefds);
936	}
937
938	/*
939	* Set for writing if we are connected, can send more, and
940	* we have something to send
941	*/
942	if (CONN_CANFSEND(so) && SBUF_LEN(&so->so_rcv))
943	{
944	STAM_COUNTER_INC(&pData->StatTCPHot);
945	TCP_ENGAGE_EVENT1(so, writefds);
946	}
947
948	/*
949	* Set for reading (and urgent data) if we are connected, can
950	* receive more, and we have room for it XXX /2 ?
951	*/
952	/* @todo: vvl - check which predicat here will be more useful here in rerm of new sbufs. */
953	if (CONN_CANFRCV(so) && (SBUF_LEN(&so->so_snd) < (SBUF_SIZE(&so->so_snd)/2)))
954	{
955	STAM_COUNTER_INC(&pData->StatTCPHot);
956	TCP_ENGAGE_EVENT2(so, readfds, xfds);
957	}
958	LOOP_LABEL(tcp, so, so_next);
959	}
960
961	/*
962	* UDP sockets
963	*/
964	STAM_COUNTER_RESET(&pData->StatUDP);
965	STAM_COUNTER_RESET(&pData->StatUDPHot);
966
967	QSOCKET_FOREACH(so, so_next, udp)
968	/* { */
969
970	STAM_COUNTER_INC(&pData->StatUDP);
971	#if !defined(RT_OS_WINDOWS)
972	so->so_poll_index = -1;
973	#endif
974
975	/*
976	* See if it's timed out
977	*/
978	if (so->so_expire)
979	{
980	if (so->so_expire <= curtime)
981	{
982	Log2(("NAT: %R[natsock] expired\n", so));
983	if (so->so_timeout != NULL)
984	{
985	so->so_timeout(pData, so, so->so_timeout_arg);
986	}
987	#ifdef VBOX_WITH_SLIRP_MT
988	/* we need so_next for continue our cycle*/
989	so_next = so->so_next;
990	#endif
991	UDP_DETACH(pData, so, so_next);
992	CONTINUE_NO_UNLOCK(udp);
993	}
994	else
995	{
996	do_slowtimo = 1; /* Let socket expire */
997	}
998	}
999
1000	/*
1001	* When UDP packets are received from over the link, they're
1002	* sendto()'d straight away, so no need for setting for writing
1003	* Limit the number of packets queued by this session to 4.
1004	* Note that even though we try and limit this to 4 packets,
1005	* the session could have more queued if the packets needed
1006	* to be fragmented.
1007	*
1008	* (XXX <= 4 ?)
1009	*/
1010	if ((so->so_state & SS_ISFCONNECTED) && so->so_queued <= 4)
1011	{
1012	STAM_COUNTER_INC(&pData->StatUDPHot);
1013	UDP_ENGAGE_EVENT(so, readfds);
1014	}
1015	LOOP_LABEL(udp, so, so_next);
1016	}
1017	done:
1018
1019	#if defined(RT_OS_WINDOWS)
1020	*pnfds = VBOX_EVENT_COUNT;
1021	#else /* RT_OS_WINDOWS */
1022	AssertRelease(poll_index <= *pnfds);
1023	*pnfds = poll_index;
1024	#endif /* !RT_OS_WINDOWS */
1025
1026	STAM_PROFILE_STOP(&pData->StatFill, a);
1027	}
1028
1029	#if defined(RT_OS_WINDOWS)
1030	void slirp_select_poll(PNATState pData, int fTimeout, int fIcmp)
1031	#else /* RT_OS_WINDOWS */
1032	void slirp_select_poll(PNATState pData, struct pollfd *polls, int ndfs)
1033	#endif /* !RT_OS_WINDOWS */
1034	{
1035	struct socket so, so_next;
1036	int ret;
1037	#if defined(RT_OS_WINDOWS)
1038	WSANETWORKEVENTS NetworkEvents;
1039	int rc;
1040	int error;
1041	#else
1042	int poll_index = 0;
1043	#endif
1044
1045	STAM_PROFILE_START(&pData->StatPoll, a);
1046
1047	/* Update time */
1048	updtime(pData);
1049
1050	/*
1051	* See if anything has timed out
1052	*/
1053	if (link_up)
1054	{
1055	if (time_fasttimo && ((curtime - time_fasttimo) >= 2))
1056	{
1057	STAM_PROFILE_START(&pData->StatFastTimer, b);
1058	tcp_fasttimo(pData);
1059	time_fasttimo = 0;
1060	STAM_PROFILE_STOP(&pData->StatFastTimer, b);
1061	}
1062	if (do_slowtimo && ((curtime - last_slowtimo) >= 499))
1063	{
1064	STAM_PROFILE_START(&pData->StatSlowTimer, c);
1065	ip_slowtimo(pData);
1066	tcp_slowtimo(pData);
1067	last_slowtimo = curtime;
1068	STAM_PROFILE_STOP(&pData->StatSlowTimer, c);
1069	}
1070	}
1071	#if defined(RT_OS_WINDOWS)
1072	if (fTimeout)
1073	return; /* only timer update */
1074	#endif
1075
1076	/*
1077	* Check sockets
1078	*/
1079	if (!link_up)
1080	goto done;
1081	#if defined(RT_OS_WINDOWS)
1082	/*XXX: before renaming please make see define
1083	* fIcmp in slirp_state.h
1084	*/
1085	if (fIcmp)
1086	sorecvfrom(pData, &pData->icmp_socket);
1087	#else
1088	if ( (pData->icmp_socket.s != -1)
1089	&& CHECK_FD_SET(&pData->icmp_socket, ignored, readfds))
1090	sorecvfrom(pData, &pData->icmp_socket);
1091	#endif
1092	/*
1093	* Check TCP sockets
1094	*/
1095	QSOCKET_FOREACH(so, so_next, tcp)
1096	/* { */
1097
1098	#ifdef VBOX_WITH_SLIRP_MT
1099	if ( so->so_state & SS_NOFDREF
1100	&& so->so_deleted == 1)
1101	{
1102	struct socket son, sop = NULL;
1103	QSOCKET_LOCK(tcb);
1104	if (so->so_next != NULL)
1105	{
1106	if (so->so_next != &tcb)
1107	SOCKET_LOCK(so->so_next);
1108	son = so->so_next;
1109	}
1110	if ( so->so_prev != &tcb
1111	&& so->so_prev != NULL)
1112	{
1113	SOCKET_LOCK(so->so_prev);
1114	sop = so->so_prev;
1115	}
1116	QSOCKET_UNLOCK(tcb);
1117	remque(pData, so);
1118	NSOCK_DEC();
1119	SOCKET_UNLOCK(so);
1120	SOCKET_LOCK_DESTROY(so);
1121	RTMemFree(so);
1122	so_next = son;
1123	if (sop != NULL)
1124	SOCKET_UNLOCK(sop);
1125	CONTINUE_NO_UNLOCK(tcp);
1126	}
1127	#endif
1128	/*
1129	* FD_ISSET is meaningless on these sockets
1130	* (and they can crash the program)
1131	*/
1132	if (so->so_state & SS_NOFDREF \|\| so->s == -1)
1133	CONTINUE(tcp);
1134
1135	POLL_TCP_EVENTS(rc, error, so, &NetworkEvents);
1136
1137	LOG_NAT_SOCK(so, TCP, &NetworkEvents, readfds, writefds, xfds);
1138
1139
1140	/*
1141	* Check for URG data
1142	* This will soread as well, so no need to
1143	* test for readfds below if this succeeds
1144	*/
1145
1146	/* out-of-band data */
1147	if ( CHECK_FD_SET(so, NetworkEvents, xfds)
1148	#ifdef RT_OS_DARWIN
1149	/* Darwin and probably BSD hosts generates POLLPRI\|POLLHUP event on receiving TCP.flags.{ACK\|URG\|FIN} this
1150	* combination on other Unixs hosts doesn't enter to this branch
1151	*/
1152	&& !CHECK_FD_SET(so, NetworkEvents, closefds)
1153	#endif
1154	)
1155	{
1156	sorecvoob(pData, so);
1157	}
1158
1159	/*
1160	* Check sockets for reading
1161	*/
1162	else if ( CHECK_FD_SET(so, NetworkEvents, readfds)
1163	\|\| WIN_CHECK_FD_SET(so, NetworkEvents, acceptds))
1164	{
1165	/*
1166	* Check for incoming connections
1167	*/
1168	if (so->so_state & SS_FACCEPTCONN)
1169	{
1170	TCP_CONNECT(pData, so);
1171	if (!CHECK_FD_SET(so, NetworkEvents, closefds))
1172	CONTINUE(tcp);
1173	}
1174
1175	ret = soread(pData, so);
1176	/* Output it if we read something */
1177	if (RT_LIKELY(ret > 0))
1178	TCP_OUTPUT(pData, sototcpcb(so));
1179	}
1180
1181	/*
1182	* Check for FD_CLOSE events.
1183	* in some cases once FD_CLOSE engaged on socket it could be flashed latter (for some reasons)
1184	*/
1185	if ( CHECK_FD_SET(so, NetworkEvents, closefds)
1186	\|\| (so->so_close == 1))
1187	{
1188	/*
1189	* drain the socket
1190	*/
1191	for (;;)
1192	{
1193	ret = soread(pData, so);
1194	if (ret > 0)
1195	TCP_OUTPUT(pData, sototcpcb(so));
1196	else
1197	{
1198	Log2(("%R[natsock] errno %d (%s)\n", so, errno, strerror(errno)));
1199	break;
1200	}
1201	}
1202	/* mark the socket for termination _after_ it was drained */
1203	so->so_close = 1;
1204	/* No idea about Windows but on Posix, POLLHUP means that we can't send more */
1205	#ifndef RT_OS_WINDOWS
1206	sofcantsendmore(so);
1207	#endif
1208	CONTINUE(tcp);
1209	}
1210
1211	/*
1212	* Check sockets for writing
1213	*/
1214	if (CHECK_FD_SET(so, NetworkEvents, writefds))
1215	{
1216	/*
1217	* Check for non-blocking, still-connecting sockets
1218	*/
1219	if (so->so_state & SS_ISFCONNECTING)
1220	{
1221	Log2(("connecting %R[natsock] catched\n", so));
1222	/* Connected */
1223	so->so_state &= ~SS_ISFCONNECTING;
1224
1225	/*
1226	* This should be probably guarded by PROBE_CONN too. Anyway,
1227	* we disable it on OS/2 because the below send call returns
1228	* EFAULT which causes the opened TCP socket to close right
1229	* after it has been opened and connected.
1230	*/
1231	#ifndef RT_OS_OS2
1232	ret = send(so->s, (const char *)&ret, 0, 0);
1233	if (ret < 0)
1234	{
1235	/* XXXXX Must fix, zero bytes is a NOP */
1236	if ( errno == EAGAIN
1237	\|\| errno == EWOULDBLOCK
1238	\|\| errno == EINPROGRESS
1239	\|\| errno == ENOTCONN)
1240	CONTINUE(tcp);
1241
1242	/* else failed */
1243	so->so_state = SS_NOFDREF;
1244	}
1245	/* else so->so_state &= ~SS_ISFCONNECTING; */
1246	#endif
1247
1248	/*
1249	* Continue tcp_input
1250	*/
1251	TCP_INPUT(pData, (struct mbuf *)NULL, sizeof(struct ip), so);
1252	/* continue; */
1253	}
1254	else
1255	SOWRITE(ret, pData, so);
1256	/*
1257	* XXX If we wrote something (a lot), there could be the need
1258	* for a window update. In the worst case, the remote will send
1259	* a window probe to get things going again.
1260	*/
1261	}
1262
1263	/*
1264	* Probe a still-connecting, non-blocking socket
1265	* to check if it's still alive
1266	*/
1267	#ifdef PROBE_CONN
1268	if (so->so_state & SS_ISFCONNECTING)
1269	{
1270	ret = recv(so->s, (char *)&ret, 0, 0);
1271
1272	if (ret < 0)
1273	{
1274	/* XXX */
1275	if ( errno == EAGAIN
1276	\|\| errno == EWOULDBLOCK
1277	\|\| errno == EINPROGRESS
1278	\|\| errno == ENOTCONN)
1279	{
1280	CONTINUE(tcp); /* Still connecting, continue */
1281	}
1282
1283	/* else failed */
1284	so->so_state = SS_NOFDREF;
1285
1286	/* tcp_input will take care of it */
1287	}
1288	else
1289	{
1290	ret = send(so->s, &ret, 0, 0);
1291	if (ret < 0)
1292	{
1293	/* XXX */
1294	if ( errno == EAGAIN
1295	\|\| errno == EWOULDBLOCK
1296	\|\| errno == EINPROGRESS
1297	\|\| errno == ENOTCONN)
1298	{
1299	CONTINUE(tcp);
1300	}
1301	/* else failed */
1302	so->so_state = SS_NOFDREF;
1303	}
1304	else
1305	so->so_state &= ~SS_ISFCONNECTING;
1306
1307	}
1308	TCP_INPUT((struct mbuf *)NULL, sizeof(struct ip),so);
1309	} /* SS_ISFCONNECTING */
1310	#endif
1311	LOOP_LABEL(tcp, so, so_next);
1312	}
1313
1314	/*
1315	* Now UDP sockets.
1316	* Incoming packets are sent straight away, they're not buffered.
1317	* Incoming UDP data isn't buffered either.
1318	*/
1319	QSOCKET_FOREACH(so, so_next, udp)
1320	/* { */
1321	#ifdef VBOX_WITH_SLIRP_MT
1322	if ( so->so_state & SS_NOFDREF
1323	&& so->so_deleted == 1)
1324	{
1325	struct socket son, sop = NULL;
1326	QSOCKET_LOCK(udb);
1327	if (so->so_next != NULL)
1328	{
1329	if (so->so_next != &udb)
1330	SOCKET_LOCK(so->so_next);
1331	son = so->so_next;
1332	}
1333	if ( so->so_prev != &udb
1334	&& so->so_prev != NULL)
1335	{
1336	SOCKET_LOCK(so->so_prev);
1337	sop = so->so_prev;
1338	}
1339	QSOCKET_UNLOCK(udb);
1340	remque(pData, so);
1341	NSOCK_DEC();
1342	SOCKET_UNLOCK(so);
1343	SOCKET_LOCK_DESTROY(so);
1344	RTMemFree(so);
1345	so_next = son;
1346	if (sop != NULL)
1347	SOCKET_UNLOCK(sop);
1348	CONTINUE_NO_UNLOCK(udp);
1349	}
1350	#endif
1351	POLL_UDP_EVENTS(rc, error, so, &NetworkEvents);
1352
1353	LOG_NAT_SOCK(so, UDP, &NetworkEvents, readfds, writefds, xfds);
1354
1355	if (so->s != -1 && CHECK_FD_SET(so, NetworkEvents, readfds))
1356	{
1357	SORECVFROM(pData, so);
1358	}
1359	LOOP_LABEL(udp, so, so_next);
1360	}
1361
1362	done:
1363
1364	STAM_PROFILE_STOP(&pData->StatPoll, a);
1365	}
1366
1367
1368	struct arphdr
1369	{
1370	unsigned short ar_hrd; /* format of hardware address */
1371	unsigned short ar_pro; /* format of protocol address */
1372	unsigned char ar_hln; /* length of hardware address */
1373	unsigned char ar_pln; /* length of protocol address */
1374	unsigned short ar_op; /* ARP opcode (command) */
1375
1376	/*
1377	* Ethernet looks like this : This bit is variable sized however...
1378	*/
1379	unsigned char ar_sha[ETH_ALEN]; /* sender hardware address */
1380	unsigned char ar_sip[4]; /* sender IP address */
1381	unsigned char ar_tha[ETH_ALEN]; /* target hardware address */
1382	unsigned char ar_tip[4]; /* target IP address */
1383	};
1384	AssertCompileSize(struct arphdr, 28);
1385
1386	/**
1387	* @note This function will free m!
1388	*/
1389	static void arp_input(PNATState pData, struct mbuf *m)
1390	{
1391	struct ethhdr *eh;
1392	struct ethhdr *reh;
1393	struct arphdr *ah;
1394	struct arphdr *rah;
1395	int ar_op;
1396	uint32_t htip;
1397	uint32_t tip;
1398	struct mbuf *mr;
1399	eh = mtod(m, struct ethhdr *);
1400	ah = (struct arphdr *)&eh[1];
1401	htip = RT_N2H_U32((uint32_t)ah->ar_tip);
1402	tip = (uint32_t)ah->ar_tip;
1403
1404	ar_op = RT_N2H_U16(ah->ar_op);
1405
1406	switch (ar_op)
1407	{
1408	case ARPOP_REQUEST:
1409	mr = m_getcl(pData, M_NOWAIT, MT_HEADER, M_PKTHDR);
1410	if (!mr)
1411	break;
1412	reh = mtod(mr, struct ethhdr *);
1413	mr->m_data += ETH_HLEN;
1414	rah = mtod(mr, struct arphdr *);
1415	mr->m_len = sizeof(struct arphdr);
1416	memcpy(reh->h_source, eh->h_source, ETH_ALEN); /* XXX: if_encap will swap src and dst*/
1417	if ( 0
1418	#ifdef VBOX_WITH_NAT_SERVICE
1419	\|\| (tip == pData->special_addr.s_addr)
1420	#endif
1421	\|\| ( ((htip & pData->netmask) == RT_N2H_U32(pData->special_addr.s_addr))
1422	&& ( CTL_CHECK(htip, CTL_DNS)
1423	\|\| CTL_CHECK(htip, CTL_ALIAS)
1424	\|\| CTL_CHECK(htip, CTL_TFTP))
1425	)
1426	)
1427	{
1428	rah->ar_hrd = RT_H2N_U16_C(1);
1429	rah->ar_pro = RT_H2N_U16_C(ETH_P_IP);
1430	rah->ar_hln = ETH_ALEN;
1431	rah->ar_pln = 4;
1432	rah->ar_op = RT_H2N_U16_C(ARPOP_REPLY);
1433	memcpy(rah->ar_sha, special_ethaddr, ETH_ALEN);
1434
1435	switch (htip & ~pData->netmask)
1436	{
1437	case CTL_DNS:
1438	case CTL_ALIAS:
1439	rah->ar_sha[5] = (uint8_t)(htip & ~pData->netmask);
1440	break;
1441	default:;
1442	}
1443
1444	memcpy(rah->ar_sip, ah->ar_tip, 4);
1445	memcpy(rah->ar_tha, ah->ar_sha, ETH_ALEN);
1446	memcpy(rah->ar_tip, ah->ar_sip, 4);
1447	if_encap(pData, ETH_P_ARP, mr, ETH_ENCAP_URG);
1448	}
1449	else
1450	m_freem(pData, mr);
1451
1452	/* Gratuitous ARP */
1453	if ( (uint32_t )ah->ar_sip == (uint32_t )ah->ar_tip
1454	&& memcmp(ah->ar_tha, broadcast_ethaddr, ETH_ALEN) == 0
1455	&& memcmp(eh->h_dest, broadcast_ethaddr, ETH_ALEN) == 0)
1456	{
1457	/* We've received an announce about address assignment,
1458	* let's do an ARP cache update
1459	*/
1460	slirp_arp_cache_update_or_add(pData, (uint32_t )ah->ar_tip, &eh->h_dest[0]);
1461	}
1462	break;
1463
1464	case ARPOP_REPLY:
1465	slirp_arp_cache_update_or_add(pData, (uint32_t )ah->ar_sip, &ah->ar_sha[0]);
1466	break;
1467
1468	default:
1469	break;
1470	}
1471
1472	m_freem(pData, m);
1473	}
1474
1475	/**
1476	* Feed a packet into the slirp engine.
1477	*
1478	* @param m Data buffer, m_len is not valid.
1479	* @param cbBuf The length of the data in m.
1480	*/
1481	void slirp_input(PNATState pData, struct mbuf *m, size_t cbBuf)
1482	{
1483	int proto;
1484	static bool fWarnedIpv6;
1485	struct ethhdr *eh;
1486	uint8_t au8Ether[ETH_ALEN];
1487
1488	m->m_len = cbBuf;
1489	if (cbBuf < ETH_HLEN)
1490	{
1491	Log(("NAT: packet having size %d has been ignored\n", m->m_len));
1492	m_freem(pData, m);
1493	return;
1494	}
1495	eh = mtod(m, struct ethhdr *);
1496	proto = RT_N2H_U16(eh->h_proto);
1497
1498	memcpy(au8Ether, eh->h_source, ETH_ALEN);
1499
1500	switch(proto)
1501	{
1502	case ETH_P_ARP:
1503	arp_input(pData, m);
1504	break;
1505
1506	case ETH_P_IP:
1507	/* Update time. Important if the network is very quiet, as otherwise
1508	* the first outgoing connection gets an incorrect timestamp. */
1509	updtime(pData);
1510	m_adj(m, ETH_HLEN);
1511	M_ASSERTPKTHDR(m);
1512	m->m_pkthdr.header = mtod(m, void *);
1513	ip_input(pData, m);
1514	break;
1515
1516	case ETH_P_IPV6:
1517	m_freem(pData, m);
1518	if (!fWarnedIpv6)
1519	{
1520	LogRel(("NAT: IPv6 not supported\n"));
1521	fWarnedIpv6 = true;
1522	}
1523	break;
1524
1525	default:
1526	Log(("NAT: Unsupported protocol %x\n", proto));
1527	m_freem(pData, m);
1528	break;
1529	}
1530
1531	if (pData->cRedirectionsActive != pData->cRedirectionsStored)
1532	activate_port_forwarding(pData, au8Ether);
1533	}
1534
1535	/**
1536	* Output the IP packet to the ethernet device.
1537	*
1538	* @note This function will free m!
1539	*/
1540	void if_encap(PNATState pData, uint16_t eth_proto, struct mbuf *m, int flags)
1541	{
1542	struct ethhdr *eh;
1543	uint8_t *buf = NULL;
1544	uint8_t *mbuf = NULL;
1545	size_t mlen = 0;
1546	STAM_PROFILE_START(&pData->StatIF_encap, a);
1547
1548	M_ASSERTPKTHDR(m);
1549	m->m_data -= ETH_HLEN;
1550	m->m_len += ETH_HLEN;
1551	eh = mtod(m, struct ethhdr *);
1552	mlen = m->m_len;
1553
1554	if (memcmp(eh->h_source, special_ethaddr, ETH_ALEN) != 0)
1555	{
1556	memcpy(eh->h_dest, eh->h_source, ETH_ALEN);
1557	memcpy(eh->h_source, special_ethaddr, ETH_ALEN);
1558	Assert(memcmp(eh->h_dest, special_ethaddr, ETH_ALEN) != 0);
1559	if (memcmp(eh->h_dest, zerro_ethaddr, ETH_ALEN) == 0)
1560	{
1561	/* don't do anything */
1562	m_freem(pData, m);
1563	goto done;
1564	}
1565	}
1566	/*
1567	* we're processing the chain, that isn't not expected.
1568	*/
1569	Assert((!m->m_next));
1570	if (m->m_next)
1571	{
1572	Log(("NAT: if_encap's recived the chain, dropping...\n"));
1573	m_freem(pData, m);
1574	goto done;
1575	}
1576	mbuf = mtod(m, uint8_t *);
1577	eh->h_proto = RT_H2N_U16(eth_proto);
1578	if (flags & ETH_ENCAP_URG)
1579	slirp_urg_output(pData->pvUser, m, mbuf, mlen);
1580	else
1581	slirp_output(pData->pvUser, m, mbuf, mlen);
1582	done:
1583	STAM_PROFILE_STOP(&pData->StatIF_encap, a);
1584	}
1585
1586	/**
1587	* Still we're using dhcp server leasing to map ether to IP
1588	* @todo see rt_lookup_in_cache
1589	*/
1590	static uint32_t find_guest_ip(PNATState pData, const uint8_t *eth_addr)
1591	{
1592	uint32_t ip = INADDR_ANY;
1593	int rc;
1594
1595	if (eth_addr == NULL)
1596	return INADDR_ANY;
1597
1598	if ( memcmp(eth_addr, zerro_ethaddr, ETH_ALEN) == 0
1599	\|\| memcmp(eth_addr, broadcast_ethaddr, ETH_ALEN) == 0)
1600	return INADDR_ANY;
1601
1602	rc = slirp_arp_lookup_ip_by_ether(pData, eth_addr, &ip);
1603	if (RT_SUCCESS(rc))
1604	return ip;
1605
1606	bootp_cache_lookup_ip_by_ether(pData, eth_addr, &ip);
1607	/* ignore return code, ip will be set to INADDR_ANY on error */
1608	return ip;
1609	}
1610
1611	/**
1612	* We need check if we've activated port forwarding
1613	* for specific machine ... that of course relates to
1614	* service mode
1615	* @todo finish this for service case
1616	*/
1617	static void activate_port_forwarding(PNATState pData, const uint8_t *h_source)
1618	{
1619	struct port_forward_rule rule, tmp;
1620
1621	/* check mac here */
1622	LIST_FOREACH_SAFE(rule, &pData->port_forward_rule_head, list, tmp)
1623	{
1624	struct socket *so;
1625	struct alias_link *alias_link;
1626	struct libalias *lib;
1627	int flags;
1628	struct sockaddr sa;
1629	struct sockaddr_in *psin;
1630	socklen_t socketlen;
1631	struct in_addr alias;
1632	int rc;
1633	uint32_t guest_addr; /* need to understand if we already give address to guest */
1634
1635	if (rule->activated)
1636	continue;
1637
1638	#ifdef VBOX_WITH_NAT_SERVICE
1639	if (memcmp(rule->mac_address, h_source, ETH_ALEN) != 0)
1640	continue; /not right mac, @todo: it'd be better do the list port forwarding per mac /
1641	guest_addr = find_guest_ip(pData, h_source);
1642	#else
1643	#if 0
1644	if (memcmp(client_ethaddr, h_source, ETH_ALEN) != 0)
1645	continue;
1646	#endif
1647	guest_addr = find_guest_ip(pData, h_source);
1648	#endif
1649	if (guest_addr == INADDR_ANY)
1650	{
1651	/* the address wasn't granted */
1652	return;
1653	}
1654
1655	#if !defined(VBOX_WITH_NAT_SERVICE)
1656	if ( rule->guest_addr.s_addr != guest_addr
1657	&& rule->guest_addr.s_addr != INADDR_ANY)
1658	continue;
1659	if (rule->guest_addr.s_addr == INADDR_ANY)
1660	rule->guest_addr.s_addr = guest_addr;
1661	#endif
1662
1663	LogRel(("NAT: set redirect %s host port %d => guest port %d @ %R[IP4]\n",
1664	rule->proto == IPPROTO_UDP ? "UDP" : "TCP", rule->host_port, rule->guest_port, &guest_addr));
1665
1666	if (rule->proto == IPPROTO_UDP)
1667	so = udp_listen(pData, rule->bind_ip.s_addr, RT_H2N_U16(rule->host_port), guest_addr,
1668	RT_H2N_U16(rule->guest_port), 0);
1669	else
1670	so = solisten(pData, rule->bind_ip.s_addr, RT_H2N_U16(rule->host_port), guest_addr,
1671	RT_H2N_U16(rule->guest_port), 0);
1672
1673	if (so == NULL)
1674	goto remove_port_forwarding;
1675
1676	psin = (struct sockaddr_in *)&sa;
1677	psin->sin_family = AF_INET;
1678	psin->sin_port = 0;
1679	psin->sin_addr.s_addr = INADDR_ANY;
1680	socketlen = sizeof(struct sockaddr);
1681
1682	rc = getsockname(so->s, &sa, &socketlen);
1683	if (rc < 0 \|\| sa.sa_family != AF_INET)
1684	goto remove_port_forwarding;
1685
1686	psin = (struct sockaddr_in *)&sa;
1687
1688	lib = LibAliasInit(pData, NULL);
1689	flags = LibAliasSetMode(lib, 0, 0);
1690	flags \|= pData->i32AliasMode;
1691	flags \|= PKT_ALIAS_REVERSE; /* set reverse */
1692	flags = LibAliasSetMode(lib, flags, ~0);
1693
1694	alias.s_addr = RT_H2N_U32(RT_N2H_U32(guest_addr) \| CTL_ALIAS);
1695	alias_link = LibAliasRedirectPort(lib, psin->sin_addr, RT_H2N_U16(rule->host_port),
1696	alias, RT_H2N_U16(rule->guest_port),
1697	pData->special_addr, -1, /* not very clear for now */
1698	rule->proto);
1699	if (!alias_link)
1700	goto remove_port_forwarding;
1701
1702	so->so_la = lib;
1703	rule->activated = 1;
1704	rule->so = so;
1705	pData->cRedirectionsActive++;
1706	continue;
1707
1708	remove_port_forwarding:
1709	LogRel(("NAT: failed to redirect %s %d => %d\n",
1710	(rule->proto == IPPROTO_UDP?"UDP":"TCP"), rule->host_port, rule->guest_port));
1711	LIST_REMOVE(rule, list);
1712	pData->cRedirectionsStored--;
1713	RTMemFree(rule);
1714	}
1715	}
1716
1717	/**
1718	* Changes in 3.1 instead of opening new socket do the following:
1719	* gain more information:
1720	* 1. bind IP
1721	* 2. host port
1722	* 3. guest port
1723	* 4. proto
1724	* 5. guest MAC address
1725	* the guest's MAC address is rather important for service, but we easily
1726	* could get it from VM configuration in DrvNAT or Service, the idea is activating
1727	* corresponding port-forwarding
1728	*/
1729	int slirp_add_redirect(PNATState pData, int is_udp, struct in_addr host_addr, int host_port,
1730	struct in_addr guest_addr, int guest_port, const uint8_t *ethaddr)
1731	{
1732	struct port_forward_rule *rule = NULL;
1733	Assert(ethaddr);
1734	LIST_FOREACH(rule, &pData->port_forward_rule_head, list)
1735	{
1736	if ( rule->proto == (is_udp ? IPPROTO_UDP : IPPROTO_TCP)
1737	&& rule->host_port == host_port
1738	&& rule->bind_ip.s_addr == host_addr.s_addr
1739	&& rule->guest_port == guest_port
1740	#ifndef VBOX_WITH_NAT_SERVICE
1741	&& rule->guest_addr.s_addr == guest_addr.s_addr
1742	#endif
1743	)
1744	return 0; /* rule has been already registered */
1745	}
1746
1747	rule = RTMemAllocZ(sizeof(struct port_forward_rule));
1748	if (rule == NULL)
1749	return 1;
1750
1751	rule->proto = (is_udp ? IPPROTO_UDP : IPPROTO_TCP);
1752	rule->host_port = host_port;
1753	rule->guest_port = guest_port;
1754	#ifndef VBOX_WITH_NAT_SERVICE
1755	rule->guest_addr.s_addr = guest_addr.s_addr;
1756	#endif
1757	rule->bind_ip.s_addr = host_addr.s_addr;
1758	memcpy(rule->mac_address, ethaddr, ETH_ALEN);
1759	/* @todo add mac address */
1760	LIST_INSERT_HEAD(&pData->port_forward_rule_head, rule, list);
1761	pData->cRedirectionsStored++;
1762	/* activate port-forwarding if guest has already got assigned IP */
1763	if (memcmp(ethaddr, zerro_ethaddr, ETH_ALEN))
1764	activate_port_forwarding(pData, ethaddr);
1765	return 0;
1766	}
1767
1768	int slirp_remove_redirect(PNATState pData, int is_udp, struct in_addr host_addr, int host_port,
1769	struct in_addr guest_addr, int guest_port)
1770	{
1771	struct port_forward_rule *rule = NULL;
1772	LIST_FOREACH(rule, &pData->port_forward_rule_head, list)
1773	{
1774	if ( rule->proto == (is_udp ? IPPROTO_UDP : IPPROTO_TCP)
1775	&& rule->host_port == host_port
1776	&& rule->guest_port == guest_port
1777	&& rule->bind_ip.s_addr == host_addr.s_addr
1778	#ifndef VBOX_WITH_NAT_SERVICE
1779	&& rule->guest_addr.s_addr == guest_addr.s_addr
1780	#endif
1781	&& rule->activated)
1782	{
1783	LogRel(("NAT: remove redirect %s host port %d => guest port %d @ %R[IP4]\n",
1784	rule->proto == IPPROTO_UDP ? "UDP" : "TCP", rule->host_port, rule->guest_port, &guest_addr));
1785
1786	LibAliasUninit(rule->so->so_la);
1787	if (is_udp)
1788	udp_detach(pData, rule->so);
1789	else
1790	tcp_close(pData, sototcpcb(rule->so));
1791	LIST_REMOVE(rule, list);
1792	RTMemFree(rule);
1793	pData->cRedirectionsStored--;
1794	break;
1795	}
1796
1797	}
1798	return 0;
1799	}
1800
1801	void slirp_set_ethaddr_and_activate_port_forwarding(PNATState pData, const uint8_t *ethaddr, uint32_t GuestIP)
1802	{
1803	#ifndef VBOX_WITH_NAT_SERVICE
1804	memcpy(client_ethaddr, ethaddr, ETH_ALEN);
1805	#endif
1806	if (GuestIP != INADDR_ANY)
1807	{
1808	slirp_arp_cache_update_or_add(pData, GuestIP, ethaddr);
1809	activate_port_forwarding(pData, ethaddr);
1810	}
1811	}
1812
1813	#if defined(RT_OS_WINDOWS)
1814	HANDLE *slirp_get_events(PNATState pData)
1815	{
1816	return pData->phEvents;
1817	}
1818	void slirp_register_external_event(PNATState pData, HANDLE hEvent, int index)
1819	{
1820	pData->phEvents[index] = hEvent;
1821	}
1822	#endif
1823
1824	unsigned int slirp_get_timeout_ms(PNATState pData)
1825	{
1826	if (link_up)
1827	{
1828	if (time_fasttimo)
1829	return 2;
1830	if (do_slowtimo)
1831	return 500; /* see PR_SLOWHZ */
1832	}
1833	return 36001000; / one hour */
1834	}
1835
1836	#ifndef RT_OS_WINDOWS
1837	int slirp_get_nsock(PNATState pData)
1838	{
1839	return pData->nsock;
1840	}
1841	#endif
1842
1843	/*
1844	* this function called from NAT thread
1845	*/
1846	void slirp_post_sent(PNATState pData, void *pvArg)
1847	{
1848	struct socket *so = 0;
1849	struct tcpcb *tp = 0;
1850	struct mbuf m = (struct mbuf )pvArg;
1851	m_freem(pData, m);
1852	}
1853	#ifdef VBOX_WITH_SLIRP_MT
1854	void slirp_process_queue(PNATState pData)
1855	{
1856	RTReqProcess(pData->pReqQueue, RT_INDEFINITE_WAIT);
1857	}
1858	void *slirp_get_queue(PNATState pData)
1859	{
1860	return pData->pReqQueue;
1861	}
1862	#endif
1863
1864	void slirp_set_dhcp_TFTP_prefix(PNATState pData, const char *tftpPrefix)
1865	{
1866	Log2(("tftp_prefix: %s\n", tftpPrefix));
1867	tftp_prefix = tftpPrefix;
1868	}
1869
1870	void slirp_set_dhcp_TFTP_bootfile(PNATState pData, const char *bootFile)
1871	{
1872	Log2(("bootFile: %s\n", bootFile));
1873	bootp_filename = bootFile;
1874	}
1875
1876	void slirp_set_dhcp_next_server(PNATState pData, const char *next_server)
1877	{
1878	Log2(("next_server: %s\n", next_server));
1879	if (next_server == NULL)
1880	pData->tftp_server.s_addr = RT_H2N_U32(RT_N2H_U32(pData->special_addr.s_addr) \| CTL_TFTP);
1881	else
1882	inet_aton(next_server, &pData->tftp_server);
1883	}
1884
1885	int slirp_set_binding_address(PNATState pData, char *addr)
1886	{
1887	if (addr == NULL \|\| (inet_aton(addr, &pData->bindIP) == 0))
1888	{
1889	pData->bindIP.s_addr = INADDR_ANY;
1890	return 1;
1891	}
1892	return 0;
1893	}
1894
1895	void slirp_set_dhcp_dns_proxy(PNATState pData, bool fDNSProxy)
1896	{
1897	if (!pData->fUseHostResolver)
1898	{
1899	Log2(("NAT: DNS proxy switched %s\n", (fDNSProxy ? "on" : "off")));
1900	pData->fUseDnsProxy = fDNSProxy;
1901	}
1902	else
1903	LogRel(("NAT: Host Resolver conflicts with DNS proxy, the last one was forcely ignored\n"));
1904	}
1905
1906	#define CHECK_ARG(name, val, lim_min, lim_max) \
1907	do { \
1908	if ((val) < (lim_min) \|\| (val) > (lim_max)) \
1909	{ \
1910	LogRel(("NAT: (" #name ":%d) has been ignored, " \
1911	"because out of range (%d, %d)\n", (val), (lim_min), (lim_max))); \
1912	return; \
1913	} \
1914	else \
1915	LogRel(("NAT: (" #name ":%d)\n", (val))); \
1916	} while (0)
1917
1918	/* don't allow user set less 8kB and more than 1M values */
1919	#define _8K_1M_CHECK_ARG(name, val) CHECK_ARG(name, (val), 8, 1024)
1920	void slirp_set_rcvbuf(PNATState pData, int kilobytes)
1921	{
1922	_8K_1M_CHECK_ARG("SOCKET_RCVBUF", kilobytes);
1923	pData->socket_rcv = kilobytes;
1924	}
1925	void slirp_set_sndbuf(PNATState pData, int kilobytes)
1926	{
1927	_8K_1M_CHECK_ARG("SOCKET_SNDBUF", kilobytes);
1928	pData->socket_snd = kilobytes * _1K;
1929	}
1930	void slirp_set_tcp_rcvspace(PNATState pData, int kilobytes)
1931	{
1932	_8K_1M_CHECK_ARG("TCP_RCVSPACE", kilobytes);
1933	tcp_rcvspace = kilobytes * _1K;
1934	}
1935	void slirp_set_tcp_sndspace(PNATState pData, int kilobytes)
1936	{
1937	_8K_1M_CHECK_ARG("TCP_SNDSPACE", kilobytes);
1938	tcp_sndspace = kilobytes * _1K;
1939	}
1940
1941	/*
1942	* Looking for Ether by ip in ARP-cache
1943	* Note: it´s responsible of caller to allocate buffer for result
1944	* @returns iprt status code
1945	*/
1946	int slirp_arp_lookup_ether_by_ip(PNATState pData, uint32_t ip, uint8_t *ether)
1947	{
1948	struct arp_cache_entry *ac;
1949
1950	if (ether == NULL)
1951	return VERR_INVALID_PARAMETER;
1952
1953	if (LIST_EMPTY(&pData->arp_cache))
1954	return VERR_NOT_FOUND;
1955
1956	LIST_FOREACH(ac, &pData->arp_cache, list)
1957	{
1958	if (ac->ip == ip)
1959	{
1960	memcpy(ether, ac->ether, ETH_ALEN);
1961	return VINF_SUCCESS;
1962	}
1963	}
1964	return VERR_NOT_FOUND;
1965	}
1966
1967	/*
1968	* Looking for IP by Ether in ARP-cache
1969	* Note: it´s responsible of caller to allocate buffer for result
1970	* @returns 0 - if found, 1 - otherwise
1971	*/
1972	int slirp_arp_lookup_ip_by_ether(PNATState pData, const uint8_t ether, uint32_t ip)
1973	{
1974	struct arp_cache_entry *ac;
1975	*ip = INADDR_ANY;
1976
1977	if (LIST_EMPTY(&pData->arp_cache))
1978	return VERR_NOT_FOUND;
1979
1980	LIST_FOREACH(ac, &pData->arp_cache, list)
1981	{
1982	if (memcmp(ether, ac->ether, ETH_ALEN) == 0)
1983	{
1984	*ip = ac->ip;
1985	return VINF_SUCCESS;
1986	}
1987	}
1988	return VERR_NOT_FOUND;
1989	}
1990
1991	void slirp_arp_who_has(PNATState pData, uint32_t dst)
1992	{
1993	struct mbuf *m;
1994	struct ethhdr *ehdr;
1995	struct arphdr *ahdr;
1996
1997	m = m_getcl(pData, M_NOWAIT, MT_HEADER, M_PKTHDR);
1998	if (m == NULL)
1999	{
2000	Log(("NAT: Can't alloc mbuf for ARP request\n"));
2001	return;
2002	}
2003	ehdr = mtod(m, struct ethhdr *);
2004	memset(ehdr->h_source, 0xff, ETH_ALEN);
2005	ahdr = (struct arphdr *)&ehdr[1];
2006	ahdr->ar_hrd = RT_H2N_U16_C(1);
2007	ahdr->ar_pro = RT_H2N_U16_C(ETH_P_IP);
2008	ahdr->ar_hln = ETH_ALEN;
2009	ahdr->ar_pln = 4;
2010	ahdr->ar_op = RT_H2N_U16_C(ARPOP_REQUEST);
2011	memcpy(ahdr->ar_sha, special_ethaddr, ETH_ALEN);
2012	(uint32_t )ahdr->ar_sip = RT_H2N_U32(RT_N2H_U32(pData->special_addr.s_addr) \| CTL_ALIAS);
2013	memset(ahdr->ar_tha, 0xff, ETH_ALEN); /broadcast/
2014	(uint32_t )ahdr->ar_tip = dst;
2015	/* warn!!! should falls in mbuf minimal size */
2016	m->m_len = sizeof(struct arphdr) + ETH_HLEN;
2017	m->m_data += ETH_HLEN;
2018	m->m_len -= ETH_HLEN;
2019	if_encap(pData, ETH_P_ARP, m, ETH_ENCAP_URG);
2020	}
2021
2022	int slirp_arp_cache_update_or_add(PNATState pData, uint32_t dst, const uint8_t *mac)
2023	{
2024	if (slirp_arp_cache_update(pData, dst, mac))
2025	slirp_arp_cache_add(pData, dst, mac);
2026
2027	return 0;
2028	}
2029
2030	/* updates the arp cache
2031	* @returns 0 - if has found and updated
2032	* 1 - if hasn't found.
2033	*/
2034	int slirp_arp_cache_update(PNATState pData, uint32_t dst, const uint8_t *mac)
2035	{
2036	struct arp_cache_entry *ac;
2037	LIST_FOREACH(ac, &pData->arp_cache, list)
2038	{
2039	if (memcmp(ac->ether, mac, ETH_ALEN) == 0)
2040	{
2041	ac->ip = dst;
2042	return 0;
2043	}
2044	}
2045	return 1;
2046	}
2047
2048	void slirp_arp_cache_add(PNATState pData, uint32_t ip, const uint8_t *ether)
2049	{
2050	struct arp_cache_entry *ac = NULL;
2051	ac = RTMemAllocZ(sizeof(struct arp_cache_entry));
2052	if (ac == NULL)
2053	{
2054	Log(("NAT: Can't allocate arp cache entry\n"));
2055	return;
2056	}
2057	ac->ip = ip;
2058	memcpy(ac->ether, ether, ETH_ALEN);
2059	LIST_INSERT_HEAD(&pData->arp_cache, ac, list);
2060	}
2061
2062	void slirp_set_mtu(PNATState pData, int mtu)
2063	{
2064	if (mtu < 20 \|\| mtu >= 16000)
2065	{
2066	LogRel(("NAT: mtu(%d) is out of range (20;16000] mtu forcely assigned to 1500\n", mtu));
2067	mtu = 1500;
2068	}
2069	/* MTU is maximum transition unit on */
2070	if_mtu =
2071	if_mru = mtu;
2072	}
2073
2074	/**
2075	* Info handler.
2076	*/
2077	void slirp_info(PNATState pData, PCDBGFINFOHLP pHlp, const char *pszArgs)
2078	{
2079	struct socket so, so_next;
2080	struct arp_cache_entry *ac;
2081	struct port_forward_rule *rule;
2082
2083	pHlp->pfnPrintf(pHlp, "NAT parameters: MTU=%d\n", if_mtu);
2084	pHlp->pfnPrintf(pHlp, "NAT TCP ports:\n");
2085	QSOCKET_FOREACH(so, so_next, tcp)
2086	/* { */
2087	pHlp->pfnPrintf(pHlp, " %R[natsock]\n", so);
2088	}
2089
2090	pHlp->pfnPrintf(pHlp, "NAT UDP ports:\n");
2091	QSOCKET_FOREACH(so, so_next, udp)
2092	/* { */
2093	pHlp->pfnPrintf(pHlp, " %R[natsock]\n", so);
2094	}
2095
2096	pHlp->pfnPrintf(pHlp, "NAT ARP cache:\n");
2097	LIST_FOREACH(ac, &pData->arp_cache, list)
2098	{
2099	pHlp->pfnPrintf(pHlp, " %R[IP4] %R[ether]\n", &ac->ip, &ac->ether);
2100	}
2101
2102	pHlp->pfnPrintf(pHlp, "NAT rules:\n");
2103	LIST_FOREACH(rule, &pData->port_forward_rule_head, list)
2104	{
2105	pHlp->pfnPrintf(pHlp, " %s %d => %R[IP4]:%d %c\n",
2106	rule->proto == IPPROTO_UDP ? "UDP" : "TCP",
2107	rule->host_port, &rule->guest_addr.s_addr, rule->guest_port,
2108	rule->activated ? ' ' : '*');
2109	}
2110	}

Note: See TracBrowser for help on using the repository browser.

source: vbox/trunk/src/VBox/Devices/Network/slirp/slirp.c@ 34209

Download in other formats: