1 | /* $Id: socket.c 96407 2022-08-22 17:43:14Z vboxsync $ */
|
---|
2 | /** @file
|
---|
3 | * NAT - socket handling.
|
---|
4 | */
|
---|
5 |
|
---|
6 | /*
|
---|
7 | * Copyright (C) 2006-2022 Oracle and/or its affiliates.
|
---|
8 | *
|
---|
9 | * This file is part of VirtualBox base platform packages, as
|
---|
10 | * available from https://www.virtualbox.org.
|
---|
11 | *
|
---|
12 | * This program is free software; you can redistribute it and/or
|
---|
13 | * modify it under the terms of the GNU General Public License
|
---|
14 | * as published by the Free Software Foundation, in version 3 of the
|
---|
15 | * License.
|
---|
16 | *
|
---|
17 | * This program is distributed in the hope that it will be useful, but
|
---|
18 | * WITHOUT ANY WARRANTY; without even the implied warranty of
|
---|
19 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
---|
20 | * General Public License for more details.
|
---|
21 | *
|
---|
22 | * You should have received a copy of the GNU General Public License
|
---|
23 | * along with this program; if not, see <https://www.gnu.org/licenses>.
|
---|
24 | *
|
---|
25 | * SPDX-License-Identifier: GPL-3.0-only
|
---|
26 | */
|
---|
27 |
|
---|
28 | /*
|
---|
29 | * This code is based on:
|
---|
30 | *
|
---|
31 | * Copyright (c) 1995 Danny Gasparovski.
|
---|
32 | *
|
---|
33 | * Please read the file COPYRIGHT for the
|
---|
34 | * terms and conditions of the copyright.
|
---|
35 | */
|
---|
36 |
|
---|
37 | #include <slirp.h>
|
---|
38 | #include "ip_icmp.h"
|
---|
39 | #include "main.h"
|
---|
40 | #ifdef __sun__
|
---|
41 | #include <sys/filio.h>
|
---|
42 | #endif
|
---|
43 | #include <VBox/vmm/pdmdrv.h>
|
---|
44 | #if defined (RT_OS_WINDOWS)
|
---|
45 | #include <iprt/win/iphlpapi.h>
|
---|
46 | #include <icmpapi.h>
|
---|
47 | #endif
|
---|
48 | #include <alias.h>
|
---|
49 |
|
---|
50 | #if defined(DECLARE_IOVEC) && defined(RT_OS_WINDOWS)
|
---|
51 | AssertCompileMembersSameSizeAndOffset(struct iovec, iov_base, WSABUF, buf);
|
---|
52 | AssertCompileMembersSameSizeAndOffset(struct iovec, iov_len, WSABUF, len);
|
---|
53 | #endif
|
---|
54 |
|
---|
55 | #ifdef VBOX_WITH_NAT_SEND2HOME
|
---|
56 | DECLINLINE(bool) slirpSend2Home(PNATState pData, struct socket *pSo, const void *pvBuf, uint32_t cbBuf, int iFlags)
|
---|
57 | {
|
---|
58 | int idxAddr;
|
---|
59 | int ret = 0;
|
---|
60 | bool fSendDone = false;
|
---|
61 | LogFlowFunc(("Enter pSo:%R[natsock] pvBuf: %p, cbBuf: %d, iFlags: %d\n", pSo, pvBuf, cbBuf, iFlags));
|
---|
62 | for (idxAddr = 0; idxAddr < pData->cInHomeAddressSize; ++idxAddr)
|
---|
63 | {
|
---|
64 |
|
---|
65 | struct socket *pNewSocket = soCloneUDPSocketWithForegnAddr(pData, pSo, pData->pInSockAddrHomeAddress[idxAddr].sin_addr);
|
---|
66 | AssertReturn((pNewSocket, false));
|
---|
67 | pData->pInSockAddrHomeAddress[idxAddr].sin_port = pSo->so_fport;
|
---|
68 | /** @todo more verbose on errors,
|
---|
69 | * @note: we shouldn't care if this send fail or not (we're in broadcast).
|
---|
70 | */
|
---|
71 | LogFunc(("send %d bytes to %RTnaipv4 from %R[natsock]\n", cbBuf, pData->pInSockAddrHomeAddress[idxAddr].sin_addr.s_addr, pNewSocket));
|
---|
72 | ret = sendto(pNewSocket->s, pvBuf, cbBuf, iFlags, (struct sockaddr *)&pData->pInSockAddrHomeAddress[idxAddr], sizeof(struct sockaddr_in));
|
---|
73 | if (ret < 0)
|
---|
74 | LogFunc(("Failed to send %d bytes to %RTnaipv4\n", cbBuf, pData->pInSockAddrHomeAddress[idxAddr].sin_addr.s_addr));
|
---|
75 | fSendDone |= ret > 0;
|
---|
76 | }
|
---|
77 | LogFlowFunc(("Leave %RTbool\n", fSendDone));
|
---|
78 | return fSendDone;
|
---|
79 | }
|
---|
80 | #endif /* !VBOX_WITH_NAT_SEND2HOME */
|
---|
81 |
|
---|
82 | #if !defined(RT_OS_WINDOWS)
|
---|
83 | static void send_icmp_to_guest(PNATState, char *, size_t, const struct sockaddr_in *);
|
---|
84 | static void sorecvfrom_icmp_unix(PNATState, struct socket *);
|
---|
85 | #endif /* !RT_OS_WINDOWS */
|
---|
86 |
|
---|
87 | void
|
---|
88 | so_init(void)
|
---|
89 | {
|
---|
90 | }
|
---|
91 |
|
---|
92 | struct socket *
|
---|
93 | solookup(struct socket *head, struct in_addr laddr,
|
---|
94 | u_int lport, struct in_addr faddr, u_int fport)
|
---|
95 | {
|
---|
96 | struct socket *so;
|
---|
97 |
|
---|
98 | for (so = head->so_next; so != head; so = so->so_next)
|
---|
99 | {
|
---|
100 | if ( so->so_lport == lport
|
---|
101 | && so->so_laddr.s_addr == laddr.s_addr
|
---|
102 | && so->so_faddr.s_addr == faddr.s_addr
|
---|
103 | && so->so_fport == fport)
|
---|
104 | return so;
|
---|
105 | }
|
---|
106 |
|
---|
107 | return (struct socket *)NULL;
|
---|
108 | }
|
---|
109 |
|
---|
110 | /*
|
---|
111 | * Create a new socket, initialise the fields
|
---|
112 | * It is the responsibility of the caller to
|
---|
113 | * insque() it into the correct linked-list
|
---|
114 | */
|
---|
115 | struct socket *
|
---|
116 | socreate(void)
|
---|
117 | {
|
---|
118 | struct socket *so;
|
---|
119 |
|
---|
120 | so = (struct socket *)RTMemAllocZ(sizeof(struct socket));
|
---|
121 | if (so)
|
---|
122 | {
|
---|
123 | so->so_state = SS_NOFDREF;
|
---|
124 | so->s = -1;
|
---|
125 | #if !defined(RT_OS_WINDOWS)
|
---|
126 | so->so_poll_index = -1;
|
---|
127 | #endif
|
---|
128 | }
|
---|
129 | return so;
|
---|
130 | }
|
---|
131 |
|
---|
132 | /*
|
---|
133 | * remque and free a socket, clobber cache
|
---|
134 | */
|
---|
135 | void
|
---|
136 | sofree(PNATState pData, struct socket *so)
|
---|
137 | {
|
---|
138 | LogFlowFunc(("ENTER:%R[natsock]\n", so));
|
---|
139 | /*
|
---|
140 | * We should not remove socket when polling routine do the polling
|
---|
141 | * instead we mark it for deletion.
|
---|
142 | */
|
---|
143 | if (so->fUnderPolling)
|
---|
144 | {
|
---|
145 | so->fShouldBeRemoved = 1;
|
---|
146 | LogFlowFunc(("LEAVE:%R[natsock] postponed deletion\n", so));
|
---|
147 | return;
|
---|
148 | }
|
---|
149 | /**
|
---|
150 | * Check that we don't freeng socket with tcbcb
|
---|
151 | */
|
---|
152 | Assert(!sototcpcb(so));
|
---|
153 | /* udp checks */
|
---|
154 | Assert(!so->so_timeout);
|
---|
155 | Assert(!so->so_timeout_arg);
|
---|
156 | if (so == tcp_last_so)
|
---|
157 | tcp_last_so = &tcb;
|
---|
158 | else if (so == udp_last_so)
|
---|
159 | udp_last_so = &udb;
|
---|
160 |
|
---|
161 | /* check if mbuf haven't been already freed */
|
---|
162 | if (so->so_m != NULL)
|
---|
163 | {
|
---|
164 | m_freem(pData, so->so_m);
|
---|
165 | so->so_m = NULL;
|
---|
166 | }
|
---|
167 |
|
---|
168 | if (so->so_ohdr != NULL)
|
---|
169 | {
|
---|
170 | RTMemFree(so->so_ohdr);
|
---|
171 | so->so_ohdr = NULL;
|
---|
172 | }
|
---|
173 |
|
---|
174 | if (so->so_next && so->so_prev)
|
---|
175 | {
|
---|
176 | remque(pData, so); /* crashes if so is not in a queue */
|
---|
177 | NSOCK_DEC();
|
---|
178 | }
|
---|
179 |
|
---|
180 | RTMemFree(so);
|
---|
181 | LogFlowFuncLeave();
|
---|
182 | }
|
---|
183 |
|
---|
184 |
|
---|
185 | /*
|
---|
186 | * Worker for sobind() below.
|
---|
187 | */
|
---|
188 | static int
|
---|
189 | sobindto(struct socket *so, uint32_t addr, uint16_t port)
|
---|
190 | {
|
---|
191 | struct sockaddr_in self;
|
---|
192 | int status;
|
---|
193 |
|
---|
194 | if (addr == INADDR_ANY && port == 0 && so->so_type != IPPROTO_UDP)
|
---|
195 | {
|
---|
196 | /* TCP sockets without constraints don't need to be bound */
|
---|
197 | Log2(("NAT: sobind: %s guest %RTnaipv4:%d - nothing to do\n",
|
---|
198 | so->so_type == IPPROTO_UDP ? "udp" : "tcp",
|
---|
199 | so->so_laddr.s_addr, ntohs(so->so_lport)));
|
---|
200 | return 0;
|
---|
201 | }
|
---|
202 |
|
---|
203 | RT_ZERO(self);
|
---|
204 | #ifdef RT_OS_DARWIN
|
---|
205 | self.sin_len = sizeof(self);
|
---|
206 | #endif
|
---|
207 | self.sin_family = AF_INET;
|
---|
208 | self.sin_addr.s_addr = addr;
|
---|
209 | self.sin_port = port;
|
---|
210 |
|
---|
211 | status = bind(so->s, (struct sockaddr *)&self, sizeof(self));
|
---|
212 | if (status == 0)
|
---|
213 | {
|
---|
214 | Log2(("NAT: sobind: %s guest %RTnaipv4:%d to host %RTnaipv4:%d\n",
|
---|
215 | so->so_type == IPPROTO_UDP ? "udp" : "tcp",
|
---|
216 | so->so_laddr.s_addr, ntohs(so->so_lport), addr, ntohs(port)));
|
---|
217 | return 0;
|
---|
218 | }
|
---|
219 |
|
---|
220 | Log2(("NAT: sobind: %s guest %RTnaipv4:%d to host %RTnaipv4:%d error %d%s\n",
|
---|
221 | so->so_type == IPPROTO_UDP ? "udp" : "tcp",
|
---|
222 | so->so_laddr.s_addr, ntohs(so->so_lport),
|
---|
223 | addr, ntohs(port),
|
---|
224 | errno, port ? " (will retry with random port)" : ""));
|
---|
225 |
|
---|
226 | if (port) /* retry without */
|
---|
227 | status = sobindto(so, addr, 0);
|
---|
228 |
|
---|
229 | if (addr)
|
---|
230 | return status;
|
---|
231 | else
|
---|
232 | return 0;
|
---|
233 | }
|
---|
234 |
|
---|
235 |
|
---|
236 | /*
|
---|
237 | * Bind the socket to specific host address and/or port if necessary.
|
---|
238 | * We also always bind udp sockets to force the local port to be
|
---|
239 | * allocated and known in advance.
|
---|
240 | */
|
---|
241 | int
|
---|
242 | sobind(PNATState pData, struct socket *so)
|
---|
243 | {
|
---|
244 | uint32_t addr = pData->bindIP.s_addr; /* may be INADDR_ANY */
|
---|
245 | bool fSamePorts = !!(pData->i32AliasMode & PKT_ALIAS_SAME_PORTS);
|
---|
246 | uint16_t port;
|
---|
247 | int status;
|
---|
248 |
|
---|
249 | if (fSamePorts)
|
---|
250 | {
|
---|
251 | int opt = 1;
|
---|
252 | setsockopt(so->s, SOL_SOCKET, SO_REUSEADDR, (char *)&opt, sizeof(opt));
|
---|
253 | port = so->so_lport;
|
---|
254 | }
|
---|
255 | else
|
---|
256 | {
|
---|
257 | port = 0;
|
---|
258 | }
|
---|
259 |
|
---|
260 | status = sobindto(so, addr, port);
|
---|
261 | return status;
|
---|
262 | }
|
---|
263 |
|
---|
264 |
|
---|
265 | /*
|
---|
266 | * Read from so's socket into sb_snd, updating all relevant sbuf fields
|
---|
267 | * NOTE: This will only be called if it is select()ed for reading, so
|
---|
268 | * a read() of 0 (or less) means it's disconnected
|
---|
269 | */
|
---|
270 | int
|
---|
271 | soread(PNATState pData, struct socket *so)
|
---|
272 | {
|
---|
273 | int n, nn, lss, total;
|
---|
274 | struct sbuf *sb = &so->so_snd;
|
---|
275 | u_int len = sb->sb_datalen - sb->sb_cc;
|
---|
276 | struct iovec iov[2];
|
---|
277 | int mss = so->so_tcpcb->t_maxseg;
|
---|
278 | int sockerr;
|
---|
279 |
|
---|
280 | STAM_PROFILE_START(&pData->StatIOread, a);
|
---|
281 | STAM_COUNTER_RESET(&pData->StatIORead_in_1);
|
---|
282 | STAM_COUNTER_RESET(&pData->StatIORead_in_2);
|
---|
283 |
|
---|
284 | QSOCKET_LOCK(tcb);
|
---|
285 | SOCKET_LOCK(so);
|
---|
286 | QSOCKET_UNLOCK(tcb);
|
---|
287 |
|
---|
288 | LogFlow(("soread: so = %R[natsock]\n", so));
|
---|
289 | Log2(("%s: so = %R[natsock] so->so_snd = %R[sbuf]\n", RT_GCC_EXTENSION __PRETTY_FUNCTION__, so, sb));
|
---|
290 |
|
---|
291 | /*
|
---|
292 | * No need to check if there's enough room to read.
|
---|
293 | * soread wouldn't have been called if there weren't
|
---|
294 | */
|
---|
295 |
|
---|
296 | len = sb->sb_datalen - sb->sb_cc;
|
---|
297 |
|
---|
298 | iov[0].iov_base = sb->sb_wptr;
|
---|
299 | iov[1].iov_base = 0;
|
---|
300 | iov[1].iov_len = 0;
|
---|
301 | if (sb->sb_wptr < sb->sb_rptr)
|
---|
302 | {
|
---|
303 | iov[0].iov_len = sb->sb_rptr - sb->sb_wptr;
|
---|
304 | /* Should never succeed, but... */
|
---|
305 | if (iov[0].iov_len > len)
|
---|
306 | iov[0].iov_len = len;
|
---|
307 | if (iov[0].iov_len > mss)
|
---|
308 | iov[0].iov_len -= iov[0].iov_len%mss;
|
---|
309 | n = 1;
|
---|
310 | }
|
---|
311 | else
|
---|
312 | {
|
---|
313 | iov[0].iov_len = (sb->sb_data + sb->sb_datalen) - sb->sb_wptr;
|
---|
314 | /* Should never succeed, but... */
|
---|
315 | if (iov[0].iov_len > len)
|
---|
316 | iov[0].iov_len = len;
|
---|
317 | len -= iov[0].iov_len;
|
---|
318 | if (len)
|
---|
319 | {
|
---|
320 | iov[1].iov_base = sb->sb_data;
|
---|
321 | iov[1].iov_len = sb->sb_rptr - sb->sb_data;
|
---|
322 | if (iov[1].iov_len > len)
|
---|
323 | iov[1].iov_len = len;
|
---|
324 | total = iov[0].iov_len + iov[1].iov_len;
|
---|
325 | if (total > mss)
|
---|
326 | {
|
---|
327 | lss = total % mss;
|
---|
328 | if (iov[1].iov_len > lss)
|
---|
329 | {
|
---|
330 | iov[1].iov_len -= lss;
|
---|
331 | n = 2;
|
---|
332 | }
|
---|
333 | else
|
---|
334 | {
|
---|
335 | lss -= iov[1].iov_len;
|
---|
336 | iov[0].iov_len -= lss;
|
---|
337 | n = 1;
|
---|
338 | }
|
---|
339 | }
|
---|
340 | else
|
---|
341 | n = 2;
|
---|
342 | }
|
---|
343 | else
|
---|
344 | {
|
---|
345 | if (iov[0].iov_len > mss)
|
---|
346 | iov[0].iov_len -= iov[0].iov_len%mss;
|
---|
347 | n = 1;
|
---|
348 | }
|
---|
349 | }
|
---|
350 |
|
---|
351 | #ifdef HAVE_READV
|
---|
352 | nn = readv(so->s, (struct iovec *)iov, n);
|
---|
353 | #else
|
---|
354 | nn = recv(so->s, iov[0].iov_base, iov[0].iov_len, (so->so_tcpcb->t_force? MSG_OOB:0));
|
---|
355 | #endif
|
---|
356 | if (nn < 0)
|
---|
357 | sockerr = errno; /* save it, as it may be clobbered by logging */
|
---|
358 | else
|
---|
359 | sockerr = 0;
|
---|
360 |
|
---|
361 | Log2(("%s: read(1) nn = %d bytes\n", RT_GCC_EXTENSION __PRETTY_FUNCTION__, nn));
|
---|
362 | Log2(("%s: so = %R[natsock] so->so_snd = %R[sbuf]\n", RT_GCC_EXTENSION __PRETTY_FUNCTION__, so, sb));
|
---|
363 | if (nn <= 0)
|
---|
364 | {
|
---|
365 | #ifdef RT_OS_WINDOWS
|
---|
366 | /*
|
---|
367 | * Windows reports ESHUTDOWN after SHUT_RD (SD_RECEIVE)
|
---|
368 | * instead of just returning EOF indication.
|
---|
369 | */
|
---|
370 | if (nn < 0 && sockerr == ESHUTDOWN)
|
---|
371 | {
|
---|
372 | nn = 0;
|
---|
373 | sockerr = 0;
|
---|
374 | }
|
---|
375 | #endif
|
---|
376 |
|
---|
377 | if (nn == 0) /* XXX: should this be inside #if defined(RT_OS_WINDOWS)? */
|
---|
378 | {
|
---|
379 | /*
|
---|
380 | * Special case for WSAEnumNetworkEvents: If we receive 0 bytes that
|
---|
381 | * _could_ mean that the connection is closed. But we will receive an
|
---|
382 | * FD_CLOSE event later if the connection was _really_ closed. With
|
---|
383 | * www.youtube.com I see this very often. Closing the socket too early
|
---|
384 | * would be dangerous.
|
---|
385 | */
|
---|
386 | int status;
|
---|
387 | unsigned long pending = 0;
|
---|
388 | status = ioctlsocket(so->s, FIONREAD, &pending);
|
---|
389 | if (status < 0)
|
---|
390 | Log(("NAT:%s: error in WSAIoctl: %d\n", RT_GCC_EXTENSION __PRETTY_FUNCTION__, errno));
|
---|
391 | if (pending != 0)
|
---|
392 | {
|
---|
393 | SOCKET_UNLOCK(so);
|
---|
394 | STAM_PROFILE_STOP(&pData->StatIOread, a);
|
---|
395 | return 0;
|
---|
396 | }
|
---|
397 | }
|
---|
398 |
|
---|
399 | if ( nn < 0
|
---|
400 | && soIgnorableErrorCode(sockerr))
|
---|
401 | {
|
---|
402 | SOCKET_UNLOCK(so);
|
---|
403 | STAM_PROFILE_STOP(&pData->StatIOread, a);
|
---|
404 | return 0;
|
---|
405 | }
|
---|
406 | else
|
---|
407 | {
|
---|
408 | int fUninitializedTemplate = 0;
|
---|
409 | int shuterr;
|
---|
410 |
|
---|
411 | fUninitializedTemplate = RT_BOOL(( sototcpcb(so)
|
---|
412 | && ( sototcpcb(so)->t_template.ti_src.s_addr == INADDR_ANY
|
---|
413 | || sototcpcb(so)->t_template.ti_dst.s_addr == INADDR_ANY)));
|
---|
414 | /* nn == 0 means peer has performed an orderly shutdown */
|
---|
415 | Log2(("%s: disconnected, nn = %d, errno = %d (%s)\n",
|
---|
416 | RT_GCC_EXTENSION __PRETTY_FUNCTION__, nn, sockerr, strerror(sockerr)));
|
---|
417 |
|
---|
418 | shuterr = sofcantrcvmore(so);
|
---|
419 | if (!sockerr && !shuterr && !fUninitializedTemplate)
|
---|
420 | tcp_sockclosed(pData, sototcpcb(so));
|
---|
421 | else
|
---|
422 | {
|
---|
423 | LogRel2(("NAT: sockerr %d, shuterr %d - %R[natsock]\n", sockerr, shuterr, so));
|
---|
424 | tcp_drop(pData, sototcpcb(so), sockerr);
|
---|
425 | }
|
---|
426 | SOCKET_UNLOCK(so);
|
---|
427 | STAM_PROFILE_STOP(&pData->StatIOread, a);
|
---|
428 | return -1;
|
---|
429 | }
|
---|
430 | }
|
---|
431 | STAM_STATS(
|
---|
432 | if (n == 1)
|
---|
433 | {
|
---|
434 | STAM_COUNTER_INC(&pData->StatIORead_in_1);
|
---|
435 | STAM_COUNTER_ADD(&pData->StatIORead_in_1_bytes, nn);
|
---|
436 | }
|
---|
437 | else
|
---|
438 | {
|
---|
439 | STAM_COUNTER_INC(&pData->StatIORead_in_2);
|
---|
440 | STAM_COUNTER_ADD(&pData->StatIORead_in_2_1st_bytes, nn);
|
---|
441 | }
|
---|
442 | );
|
---|
443 |
|
---|
444 | #ifndef HAVE_READV
|
---|
445 | /*
|
---|
446 | * If there was no error, try and read the second time round
|
---|
447 | * We read again if n = 2 (ie, there's another part of the buffer)
|
---|
448 | * and we read as much as we could in the first read
|
---|
449 | * We don't test for <= 0 this time, because there legitimately
|
---|
450 | * might not be any more data (since the socket is non-blocking),
|
---|
451 | * a close will be detected on next iteration.
|
---|
452 | * A return of -1 wont (shouldn't) happen, since it didn't happen above
|
---|
453 | */
|
---|
454 | if (n == 2 && (unsigned)nn == iov[0].iov_len)
|
---|
455 | {
|
---|
456 | int ret;
|
---|
457 | ret = recv(so->s, iov[1].iov_base, iov[1].iov_len, 0);
|
---|
458 | if (ret > 0)
|
---|
459 | nn += ret;
|
---|
460 | STAM_STATS(
|
---|
461 | if (ret > 0)
|
---|
462 | {
|
---|
463 | STAM_COUNTER_INC(&pData->StatIORead_in_2);
|
---|
464 | STAM_COUNTER_ADD(&pData->StatIORead_in_2_2nd_bytes, ret);
|
---|
465 | }
|
---|
466 | );
|
---|
467 | }
|
---|
468 |
|
---|
469 | Log2(("%s: read(2) nn = %d bytes\n", RT_GCC_EXTENSION __PRETTY_FUNCTION__, nn));
|
---|
470 | #endif
|
---|
471 |
|
---|
472 | /* Update fields */
|
---|
473 | sb->sb_cc += nn;
|
---|
474 | sb->sb_wptr += nn;
|
---|
475 | Log2(("%s: update so_snd (readed nn = %d) %R[sbuf]\n", RT_GCC_EXTENSION __PRETTY_FUNCTION__, nn, sb));
|
---|
476 | if (sb->sb_wptr >= (sb->sb_data + sb->sb_datalen))
|
---|
477 | {
|
---|
478 | sb->sb_wptr -= sb->sb_datalen;
|
---|
479 | Log2(("%s: alter sb_wptr so_snd = %R[sbuf]\n", RT_GCC_EXTENSION __PRETTY_FUNCTION__, sb));
|
---|
480 | }
|
---|
481 | STAM_PROFILE_STOP(&pData->StatIOread, a);
|
---|
482 | SOCKET_UNLOCK(so);
|
---|
483 | return nn;
|
---|
484 | }
|
---|
485 |
|
---|
486 | /*
|
---|
487 | * Get urgent data
|
---|
488 | *
|
---|
489 | * When the socket is created, we set it SO_OOBINLINE,
|
---|
490 | * so when OOB data arrives, we soread() it and everything
|
---|
491 | * in the send buffer is sent as urgent data
|
---|
492 | */
|
---|
493 | void
|
---|
494 | sorecvoob(PNATState pData, struct socket *so)
|
---|
495 | {
|
---|
496 | struct tcpcb *tp = sototcpcb(so);
|
---|
497 | ssize_t ret;
|
---|
498 |
|
---|
499 | LogFlowFunc(("sorecvoob: so = %R[natsock]\n", so));
|
---|
500 |
|
---|
501 | /*
|
---|
502 | * We take a guess at how much urgent data has arrived.
|
---|
503 | * In most situations, when urgent data arrives, the next
|
---|
504 | * read() should get all the urgent data. This guess will
|
---|
505 | * be wrong however if more data arrives just after the
|
---|
506 | * urgent data, or the read() doesn't return all the
|
---|
507 | * urgent data.
|
---|
508 | */
|
---|
509 | ret = soread(pData, so);
|
---|
510 | if (RT_LIKELY(ret > 0))
|
---|
511 | {
|
---|
512 | /*
|
---|
513 | * @todo for now just scrub the URG pointer. To faithfully
|
---|
514 | * proxy URG we need to read the srteam until SIOCATMARK, and
|
---|
515 | * then mark the first byte of the next read ar urgent.
|
---|
516 | */
|
---|
517 | #if 0
|
---|
518 | tp->snd_up = tp->snd_una + SBUF_LEN(&so->so_snd);
|
---|
519 | #endif
|
---|
520 | tp->t_force = 1;
|
---|
521 | tcp_output(pData, tp);
|
---|
522 | tp->t_force = 0;
|
---|
523 | }
|
---|
524 | }
|
---|
525 |
|
---|
526 | /*
|
---|
527 | * Send urgent data
|
---|
528 | * There's a lot duplicated code here, but...
|
---|
529 | */
|
---|
530 | int
|
---|
531 | sosendoob(struct socket *so)
|
---|
532 | {
|
---|
533 | struct sbuf *sb = &so->so_rcv;
|
---|
534 | char buff[2048]; /* XXX Shouldn't be sending more oob data than this */
|
---|
535 |
|
---|
536 | int n, len;
|
---|
537 |
|
---|
538 | LogFlowFunc(("sosendoob so = %R[natsock]\n", so));
|
---|
539 |
|
---|
540 | if (so->so_urgc > sizeof(buff))
|
---|
541 | so->so_urgc = sizeof(buff); /* XXX */
|
---|
542 |
|
---|
543 | if (sb->sb_rptr < sb->sb_wptr)
|
---|
544 | {
|
---|
545 | /* We can send it directly */
|
---|
546 | n = send(so->s, sb->sb_rptr, so->so_urgc, (MSG_OOB)); /* |MSG_DONTWAIT)); */
|
---|
547 | so->so_urgc -= n;
|
---|
548 |
|
---|
549 | Log2((" --- sent %d bytes urgent data, %d urgent bytes left\n",
|
---|
550 | n, so->so_urgc));
|
---|
551 | }
|
---|
552 | else
|
---|
553 | {
|
---|
554 | /*
|
---|
555 | * Since there's no sendv or sendtov like writev,
|
---|
556 | * we must copy all data to a linear buffer then
|
---|
557 | * send it all
|
---|
558 | */
|
---|
559 | len = (sb->sb_data + sb->sb_datalen) - sb->sb_rptr;
|
---|
560 | if (len > so->so_urgc)
|
---|
561 | len = so->so_urgc;
|
---|
562 | memcpy(buff, sb->sb_rptr, len);
|
---|
563 | so->so_urgc -= len;
|
---|
564 | if (so->so_urgc)
|
---|
565 | {
|
---|
566 | n = sb->sb_wptr - sb->sb_data;
|
---|
567 | if (n > so->so_urgc)
|
---|
568 | n = so->so_urgc;
|
---|
569 | memcpy(buff + len, sb->sb_data, n);
|
---|
570 | so->so_urgc -= n;
|
---|
571 | len += n;
|
---|
572 | }
|
---|
573 | n = send(so->s, buff, len, (MSG_OOB)); /* |MSG_DONTWAIT)); */
|
---|
574 | #ifdef DEBUG
|
---|
575 | if (n != len)
|
---|
576 | Log(("Didn't send all data urgently XXXXX\n"));
|
---|
577 | #endif
|
---|
578 | Log2((" ---2 sent %d bytes urgent data, %d urgent bytes left\n",
|
---|
579 | n, so->so_urgc));
|
---|
580 | }
|
---|
581 |
|
---|
582 | sb->sb_cc -= n;
|
---|
583 | sb->sb_rptr += n;
|
---|
584 | if (sb->sb_rptr >= (sb->sb_data + sb->sb_datalen))
|
---|
585 | sb->sb_rptr -= sb->sb_datalen;
|
---|
586 |
|
---|
587 | return n;
|
---|
588 | }
|
---|
589 |
|
---|
590 | /*
|
---|
591 | * Write data from so_rcv to so's socket,
|
---|
592 | * updating all sbuf field as necessary
|
---|
593 | */
|
---|
594 | int
|
---|
595 | sowrite(PNATState pData, struct socket *so)
|
---|
596 | {
|
---|
597 | int n, nn;
|
---|
598 | struct sbuf *sb = &so->so_rcv;
|
---|
599 | u_int len = sb->sb_cc;
|
---|
600 | struct iovec iov[2];
|
---|
601 |
|
---|
602 | STAM_PROFILE_START(&pData->StatIOwrite, a);
|
---|
603 | STAM_COUNTER_RESET(&pData->StatIOWrite_in_1);
|
---|
604 | STAM_COUNTER_RESET(&pData->StatIOWrite_in_1_bytes);
|
---|
605 | STAM_COUNTER_RESET(&pData->StatIOWrite_in_2);
|
---|
606 | STAM_COUNTER_RESET(&pData->StatIOWrite_in_2_1st_bytes);
|
---|
607 | STAM_COUNTER_RESET(&pData->StatIOWrite_in_2_2nd_bytes);
|
---|
608 | STAM_COUNTER_RESET(&pData->StatIOWrite_no_w);
|
---|
609 | STAM_COUNTER_RESET(&pData->StatIOWrite_rest);
|
---|
610 | STAM_COUNTER_RESET(&pData->StatIOWrite_rest_bytes);
|
---|
611 | LogFlowFunc(("so = %R[natsock]\n", so));
|
---|
612 | Log2(("%s: so = %R[natsock] so->so_rcv = %R[sbuf]\n", RT_GCC_EXTENSION __PRETTY_FUNCTION__, so, sb));
|
---|
613 | QSOCKET_LOCK(tcb);
|
---|
614 | SOCKET_LOCK(so);
|
---|
615 | QSOCKET_UNLOCK(tcb);
|
---|
616 | if (so->so_urgc)
|
---|
617 | {
|
---|
618 | sosendoob(so);
|
---|
619 | if (sb->sb_cc == 0)
|
---|
620 | {
|
---|
621 | SOCKET_UNLOCK(so);
|
---|
622 | STAM_PROFILE_STOP(&pData->StatIOwrite, a);
|
---|
623 | return 0;
|
---|
624 | }
|
---|
625 | }
|
---|
626 |
|
---|
627 | /*
|
---|
628 | * No need to check if there's something to write,
|
---|
629 | * sowrite wouldn't have been called otherwise
|
---|
630 | */
|
---|
631 |
|
---|
632 | len = sb->sb_cc;
|
---|
633 |
|
---|
634 | iov[0].iov_base = sb->sb_rptr;
|
---|
635 | iov[1].iov_base = 0;
|
---|
636 | iov[1].iov_len = 0;
|
---|
637 | if (sb->sb_rptr < sb->sb_wptr)
|
---|
638 | {
|
---|
639 | iov[0].iov_len = sb->sb_wptr - sb->sb_rptr;
|
---|
640 | /* Should never succeed, but... */
|
---|
641 | if (iov[0].iov_len > len)
|
---|
642 | iov[0].iov_len = len;
|
---|
643 | n = 1;
|
---|
644 | }
|
---|
645 | else
|
---|
646 | {
|
---|
647 | iov[0].iov_len = (sb->sb_data + sb->sb_datalen) - sb->sb_rptr;
|
---|
648 | if (iov[0].iov_len > len)
|
---|
649 | iov[0].iov_len = len;
|
---|
650 | len -= iov[0].iov_len;
|
---|
651 | if (len)
|
---|
652 | {
|
---|
653 | iov[1].iov_base = sb->sb_data;
|
---|
654 | iov[1].iov_len = sb->sb_wptr - sb->sb_data;
|
---|
655 | if (iov[1].iov_len > len)
|
---|
656 | iov[1].iov_len = len;
|
---|
657 | n = 2;
|
---|
658 | }
|
---|
659 | else
|
---|
660 | n = 1;
|
---|
661 | }
|
---|
662 | STAM_STATS({
|
---|
663 | if (n == 1)
|
---|
664 | {
|
---|
665 | STAM_COUNTER_INC(&pData->StatIOWrite_in_1);
|
---|
666 | STAM_COUNTER_ADD(&pData->StatIOWrite_in_1_bytes, iov[0].iov_len);
|
---|
667 | }
|
---|
668 | else
|
---|
669 | {
|
---|
670 | STAM_COUNTER_INC(&pData->StatIOWrite_in_2);
|
---|
671 | STAM_COUNTER_ADD(&pData->StatIOWrite_in_2_1st_bytes, iov[0].iov_len);
|
---|
672 | STAM_COUNTER_ADD(&pData->StatIOWrite_in_2_2nd_bytes, iov[1].iov_len);
|
---|
673 | }
|
---|
674 | });
|
---|
675 | /* Check if there's urgent data to send, and if so, send it */
|
---|
676 | #ifdef HAVE_READV
|
---|
677 | nn = writev(so->s, (const struct iovec *)iov, n);
|
---|
678 | #else
|
---|
679 | nn = send(so->s, iov[0].iov_base, iov[0].iov_len, 0);
|
---|
680 | #endif
|
---|
681 | Log2(("%s: wrote(1) nn = %d bytes\n", RT_GCC_EXTENSION __PRETTY_FUNCTION__, nn));
|
---|
682 | /* This should never happen, but people tell me it does *shrug* */
|
---|
683 | if ( nn < 0
|
---|
684 | && soIgnorableErrorCode(errno))
|
---|
685 | {
|
---|
686 | SOCKET_UNLOCK(so);
|
---|
687 | STAM_PROFILE_STOP(&pData->StatIOwrite, a);
|
---|
688 | return 0;
|
---|
689 | }
|
---|
690 |
|
---|
691 | if (nn < 0 || (nn == 0 && iov[0].iov_len > 0))
|
---|
692 | {
|
---|
693 | Log2(("%s: disconnected, so->so_state = %x, errno = %d\n",
|
---|
694 | RT_GCC_EXTENSION __PRETTY_FUNCTION__, so->so_state, errno));
|
---|
695 | sofcantsendmore(so);
|
---|
696 | tcp_sockclosed(pData, sototcpcb(so));
|
---|
697 | SOCKET_UNLOCK(so);
|
---|
698 | STAM_PROFILE_STOP(&pData->StatIOwrite, a);
|
---|
699 | return -1;
|
---|
700 | }
|
---|
701 |
|
---|
702 | #ifndef HAVE_READV
|
---|
703 | if (n == 2 && (unsigned)nn == iov[0].iov_len)
|
---|
704 | {
|
---|
705 | int ret;
|
---|
706 | ret = send(so->s, iov[1].iov_base, iov[1].iov_len, 0);
|
---|
707 | if (ret > 0)
|
---|
708 | nn += ret;
|
---|
709 | # ifdef VBOX_WITH_STATISTICS
|
---|
710 | if (ret > 0 && ret != (ssize_t)iov[1].iov_len)
|
---|
711 | {
|
---|
712 | STAM_COUNTER_INC(&pData->StatIOWrite_rest);
|
---|
713 | STAM_COUNTER_ADD(&pData->StatIOWrite_rest_bytes, (iov[1].iov_len - ret));
|
---|
714 | }
|
---|
715 | #endif
|
---|
716 | }
|
---|
717 | Log2(("%s: wrote(2) nn = %d bytes\n", RT_GCC_EXTENSION __PRETTY_FUNCTION__, nn));
|
---|
718 | #endif
|
---|
719 |
|
---|
720 | /* Update sbuf */
|
---|
721 | sb->sb_cc -= nn;
|
---|
722 | sb->sb_rptr += nn;
|
---|
723 | Log2(("%s: update so_rcv (written nn = %d) %R[sbuf]\n", RT_GCC_EXTENSION __PRETTY_FUNCTION__, nn, sb));
|
---|
724 | if (sb->sb_rptr >= (sb->sb_data + sb->sb_datalen))
|
---|
725 | {
|
---|
726 | sb->sb_rptr -= sb->sb_datalen;
|
---|
727 | Log2(("%s: alter sb_rptr of so_rcv %R[sbuf]\n", RT_GCC_EXTENSION __PRETTY_FUNCTION__, sb));
|
---|
728 | }
|
---|
729 |
|
---|
730 | /*
|
---|
731 | * If in DRAIN mode, and there's no more data, set
|
---|
732 | * it CANTSENDMORE
|
---|
733 | */
|
---|
734 | if ((so->so_state & SS_FWDRAIN) && sb->sb_cc == 0)
|
---|
735 | sofcantsendmore(so);
|
---|
736 |
|
---|
737 | SOCKET_UNLOCK(so);
|
---|
738 | STAM_PROFILE_STOP(&pData->StatIOwrite, a);
|
---|
739 | return nn;
|
---|
740 | }
|
---|
741 |
|
---|
742 | /*
|
---|
743 | * recvfrom() a UDP socket
|
---|
744 | */
|
---|
745 | void
|
---|
746 | sorecvfrom(PNATState pData, struct socket *so)
|
---|
747 | {
|
---|
748 | LogFlowFunc(("sorecvfrom: so = %p\n", so));
|
---|
749 |
|
---|
750 | #ifdef RT_OS_WINDOWS
|
---|
751 | /* ping is handled with ICMP API in ip_icmpwin.c */
|
---|
752 | Assert(so->so_type == IPPROTO_UDP);
|
---|
753 | #else
|
---|
754 | if (so->so_type == IPPROTO_ICMP)
|
---|
755 | {
|
---|
756 | /* This is a "ping" reply */
|
---|
757 | sorecvfrom_icmp_unix(pData, so);
|
---|
758 | udp_detach(pData, so);
|
---|
759 | }
|
---|
760 | else
|
---|
761 | #endif /* !RT_OS_WINDOWS */
|
---|
762 | {
|
---|
763 | static char achBuf[64 * 1024];
|
---|
764 |
|
---|
765 | /* A "normal" UDP packet */
|
---|
766 | struct sockaddr_in addr;
|
---|
767 | socklen_t addrlen = sizeof(struct sockaddr_in);
|
---|
768 | struct iovec iov[2];
|
---|
769 | ssize_t nread;
|
---|
770 | struct mbuf *m;
|
---|
771 |
|
---|
772 | QSOCKET_LOCK(udb);
|
---|
773 | SOCKET_LOCK(so);
|
---|
774 | QSOCKET_UNLOCK(udb);
|
---|
775 |
|
---|
776 | m = m_getjcl(pData, M_NOWAIT, MT_HEADER, M_PKTHDR, slirp_size(pData));
|
---|
777 | if (m == NULL)
|
---|
778 | {
|
---|
779 | SOCKET_UNLOCK(so);
|
---|
780 | return;
|
---|
781 | }
|
---|
782 |
|
---|
783 | m->m_data += ETH_HLEN;
|
---|
784 | m->m_pkthdr.header = mtod(m, void *);
|
---|
785 |
|
---|
786 | m->m_data += sizeof(struct udpiphdr);
|
---|
787 |
|
---|
788 | /* small packets will fit without copying */
|
---|
789 | iov[0].iov_base = mtod(m, char *);
|
---|
790 | iov[0].iov_len = M_TRAILINGSPACE(m);
|
---|
791 |
|
---|
792 | /* large packets will spill into a temp buffer */
|
---|
793 | iov[1].iov_base = achBuf;
|
---|
794 | iov[1].iov_len = sizeof(achBuf);
|
---|
795 |
|
---|
796 | #if !defined(RT_OS_WINDOWS)
|
---|
797 | {
|
---|
798 | struct msghdr mh;
|
---|
799 | memset(&mh, 0, sizeof(mh));
|
---|
800 |
|
---|
801 | mh.msg_iov = iov;
|
---|
802 | mh.msg_iovlen = 2;
|
---|
803 | mh.msg_name = &addr;
|
---|
804 | mh.msg_namelen = addrlen;
|
---|
805 |
|
---|
806 | nread = recvmsg(so->s, &mh, 0);
|
---|
807 | }
|
---|
808 | #else /* RT_OS_WINDOWS */
|
---|
809 | {
|
---|
810 | DWORD nbytes; /* NB: can't use nread b/c of different size */
|
---|
811 | DWORD flags = 0;
|
---|
812 | int status;
|
---|
813 | AssertCompile(sizeof(WSABUF) == sizeof(struct iovec));
|
---|
814 | AssertCompileMembersSameSizeAndOffset(WSABUF, len, struct iovec, iov_len);
|
---|
815 | AssertCompileMembersSameSizeAndOffset(WSABUF, buf, struct iovec, iov_base);
|
---|
816 | status = WSARecvFrom(so->s, (WSABUF *)&iov[0], 2, &nbytes, &flags,
|
---|
817 | (struct sockaddr *)&addr, &addrlen,
|
---|
818 | NULL, NULL);
|
---|
819 | if (status != SOCKET_ERROR)
|
---|
820 | nread = nbytes;
|
---|
821 | else
|
---|
822 | nread = -1;
|
---|
823 | }
|
---|
824 | #endif
|
---|
825 | if (nread >= 0)
|
---|
826 | {
|
---|
827 | if (nread <= iov[0].iov_len)
|
---|
828 | m->m_len = nread;
|
---|
829 | else
|
---|
830 | {
|
---|
831 | m->m_len = iov[0].iov_len;
|
---|
832 | m_append(pData, m, nread - iov[0].iov_len, iov[1].iov_base);
|
---|
833 | }
|
---|
834 | Assert(m_length(m, NULL) == (size_t)nread);
|
---|
835 |
|
---|
836 | /*
|
---|
837 | * Hack: domain name lookup will be used the most for UDP,
|
---|
838 | * and since they'll only be used once there's no need
|
---|
839 | * for the 4 minute (or whatever) timeout... So we time them
|
---|
840 | * out much quicker (10 seconds for now...)
|
---|
841 | */
|
---|
842 | if (so->so_expire)
|
---|
843 | {
|
---|
844 | if (so->so_fport != RT_H2N_U16_C(53))
|
---|
845 | so->so_expire = curtime + SO_EXPIRE;
|
---|
846 | }
|
---|
847 |
|
---|
848 | /*
|
---|
849 | * DNS proxy requests are forwarded to the real resolver,
|
---|
850 | * but its socket's so_faddr is that of the DNS proxy
|
---|
851 | * itself.
|
---|
852 | *
|
---|
853 | * last argument should be changed if Slirp will inject IP attributes
|
---|
854 | */
|
---|
855 | if ( pData->fUseDnsProxy
|
---|
856 | && so->so_fport == RT_H2N_U16_C(53)
|
---|
857 | && CTL_CHECK(so->so_faddr.s_addr, CTL_DNS))
|
---|
858 | dnsproxy_answer(pData, so, m);
|
---|
859 |
|
---|
860 | /* packets definetly will be fragmented, could confuse receiver peer. */
|
---|
861 | if (nread > if_mtu)
|
---|
862 | m->m_flags |= M_SKIP_FIREWALL;
|
---|
863 |
|
---|
864 | /*
|
---|
865 | * If this packet was destined for CTL_ADDR,
|
---|
866 | * make it look like that's where it came from, done by udp_output
|
---|
867 | */
|
---|
868 | udp_output(pData, so, m, &addr);
|
---|
869 | }
|
---|
870 | else
|
---|
871 | {
|
---|
872 | m_freem(pData, m);
|
---|
873 |
|
---|
874 | if (!soIgnorableErrorCode(errno))
|
---|
875 | {
|
---|
876 | u_char code;
|
---|
877 | if (errno == EHOSTUNREACH)
|
---|
878 | code = ICMP_UNREACH_HOST;
|
---|
879 | else if (errno == ENETUNREACH)
|
---|
880 | code = ICMP_UNREACH_NET;
|
---|
881 | else
|
---|
882 | code = ICMP_UNREACH_PORT;
|
---|
883 |
|
---|
884 | Log2((" rx error, tx icmp ICMP_UNREACH:%i\n", code));
|
---|
885 | icmp_error(pData, so->so_m, ICMP_UNREACH, code, 0, strerror(errno));
|
---|
886 | so->so_m = NULL;
|
---|
887 | }
|
---|
888 | }
|
---|
889 |
|
---|
890 | SOCKET_UNLOCK(so);
|
---|
891 | }
|
---|
892 | }
|
---|
893 |
|
---|
894 | /*
|
---|
895 | * sendto() a socket
|
---|
896 | */
|
---|
897 | int
|
---|
898 | sosendto(PNATState pData, struct socket *so, struct mbuf *m)
|
---|
899 | {
|
---|
900 | int ret;
|
---|
901 | struct sockaddr_in *paddr;
|
---|
902 | struct sockaddr addr;
|
---|
903 | #if 0
|
---|
904 | struct sockaddr_in host_addr;
|
---|
905 | #endif
|
---|
906 | caddr_t buf = 0;
|
---|
907 | int mlen;
|
---|
908 |
|
---|
909 | LogFlowFunc(("sosendto: so = %R[natsock], m = %p\n", so, m));
|
---|
910 |
|
---|
911 | memset(&addr, 0, sizeof(struct sockaddr));
|
---|
912 | #ifdef RT_OS_DARWIN
|
---|
913 | addr.sa_len = sizeof(struct sockaddr_in);
|
---|
914 | #endif
|
---|
915 | paddr = (struct sockaddr_in *)&addr;
|
---|
916 | paddr->sin_family = AF_INET;
|
---|
917 | if ((so->so_faddr.s_addr & RT_H2N_U32(pData->netmask)) == pData->special_addr.s_addr)
|
---|
918 | {
|
---|
919 | /* It's an alias */
|
---|
920 | uint32_t last_byte = RT_N2H_U32(so->so_faddr.s_addr) & ~pData->netmask;
|
---|
921 | switch(last_byte)
|
---|
922 | {
|
---|
923 | #if 0
|
---|
924 | /* handle this case at 'default:' */
|
---|
925 | case CTL_BROADCAST:
|
---|
926 | addr.sin_addr.s_addr = INADDR_BROADCAST;
|
---|
927 | /* Send the packet to host to fully emulate broadcast */
|
---|
928 | /** @todo r=klaus: on Linux host this causes the host to receive
|
---|
929 | * the packet twice for some reason. And I cannot find any place
|
---|
930 | * in the man pages which states that sending a broadcast does not
|
---|
931 | * reach the host itself. */
|
---|
932 | host_addr.sin_family = AF_INET;
|
---|
933 | host_addr.sin_port = so->so_fport;
|
---|
934 | host_addr.sin_addr = our_addr;
|
---|
935 | sendto(so->s, m->m_data, m->m_len, 0,
|
---|
936 | (struct sockaddr *)&host_addr, sizeof (struct sockaddr));
|
---|
937 | break;
|
---|
938 | #endif
|
---|
939 | case CTL_DNS:
|
---|
940 | case CTL_ALIAS:
|
---|
941 | default:
|
---|
942 | if (last_byte == ~pData->netmask)
|
---|
943 | paddr->sin_addr.s_addr = INADDR_BROADCAST;
|
---|
944 | else
|
---|
945 | paddr->sin_addr = loopback_addr;
|
---|
946 | break;
|
---|
947 | }
|
---|
948 | }
|
---|
949 | else
|
---|
950 | paddr->sin_addr = so->so_faddr;
|
---|
951 | paddr->sin_port = so->so_fport;
|
---|
952 |
|
---|
953 | Log2((" sendto()ing, addr.sin_port=%d, addr.sin_addr.s_addr=%.16s\n",
|
---|
954 | RT_N2H_U16(paddr->sin_port), inet_ntoa(paddr->sin_addr)));
|
---|
955 |
|
---|
956 | /* Don't care what port we get */
|
---|
957 | /*
|
---|
958 | * > nmap -sV -T4 -O -A -v -PU3483 255.255.255.255
|
---|
959 | * generates bodyless messages, annoying memmory management system.
|
---|
960 | */
|
---|
961 | mlen = m_length(m, NULL);
|
---|
962 | if (mlen > 0)
|
---|
963 | {
|
---|
964 | buf = RTMemAlloc(mlen);
|
---|
965 | if (buf == NULL)
|
---|
966 | {
|
---|
967 | return -1;
|
---|
968 | }
|
---|
969 | m_copydata(m, 0, mlen, buf);
|
---|
970 | }
|
---|
971 | ret = sendto(so->s, buf, mlen, 0,
|
---|
972 | (struct sockaddr *)&addr, sizeof (struct sockaddr));
|
---|
973 | #ifdef VBOX_WITH_NAT_SEND2HOME
|
---|
974 | if (slirpIsWideCasting(pData, so->so_faddr.s_addr))
|
---|
975 | {
|
---|
976 | slirpSend2Home(pData, so, buf, mlen, 0);
|
---|
977 | }
|
---|
978 | #endif
|
---|
979 | if (buf)
|
---|
980 | RTMemFree(buf);
|
---|
981 | if (ret < 0)
|
---|
982 | {
|
---|
983 | Log2(("UDP: sendto fails (%s)\n", strerror(errno)));
|
---|
984 | return -1;
|
---|
985 | }
|
---|
986 |
|
---|
987 | /*
|
---|
988 | * Kill the socket if there's no reply in 4 minutes,
|
---|
989 | * but only if it's an expirable socket
|
---|
990 | */
|
---|
991 | if (so->so_expire)
|
---|
992 | so->so_expire = curtime + SO_EXPIRE;
|
---|
993 | so->so_state = SS_ISFCONNECTED; /* So that it gets select()ed */
|
---|
994 | return 0;
|
---|
995 | }
|
---|
996 |
|
---|
997 | /*
|
---|
998 | * XXX This should really be tcp_listen
|
---|
999 | */
|
---|
1000 | struct socket *
|
---|
1001 | solisten(PNATState pData, u_int32_t bind_addr, u_int port, u_int32_t laddr, u_int lport, int flags)
|
---|
1002 | {
|
---|
1003 | struct sockaddr_in addr;
|
---|
1004 | struct socket *so;
|
---|
1005 | socklen_t addrlen = sizeof(addr);
|
---|
1006 | int s, opt = 1;
|
---|
1007 | int status;
|
---|
1008 |
|
---|
1009 | LogFlowFunc(("solisten: port = %d, laddr = %x, lport = %d, flags = %x\n", port, laddr, lport, flags));
|
---|
1010 |
|
---|
1011 | if ((so = socreate()) == NULL)
|
---|
1012 | {
|
---|
1013 | /* RTMemFree(so); Not sofree() ??? free(NULL) == NOP */
|
---|
1014 | return NULL;
|
---|
1015 | }
|
---|
1016 |
|
---|
1017 | /* Don't tcp_attach... we don't need so_snd nor so_rcv */
|
---|
1018 | if ((so->so_tcpcb = tcp_newtcpcb(pData, so)) == NULL)
|
---|
1019 | {
|
---|
1020 | RTMemFree(so);
|
---|
1021 | return NULL;
|
---|
1022 | }
|
---|
1023 |
|
---|
1024 | SOCKET_LOCK_CREATE(so);
|
---|
1025 | SOCKET_LOCK(so);
|
---|
1026 | QSOCKET_LOCK(tcb);
|
---|
1027 | insque(pData, so,&tcb);
|
---|
1028 | NSOCK_INC();
|
---|
1029 | QSOCKET_UNLOCK(tcb);
|
---|
1030 |
|
---|
1031 | /*
|
---|
1032 | * SS_FACCEPTONCE sockets must time out.
|
---|
1033 | */
|
---|
1034 | if (flags & SS_FACCEPTONCE)
|
---|
1035 | so->so_tcpcb->t_timer[TCPT_KEEP] = TCPTV_KEEP_INIT*2;
|
---|
1036 |
|
---|
1037 | so->so_state = (SS_FACCEPTCONN|flags);
|
---|
1038 | so->so_lport = lport; /* Kept in network format */
|
---|
1039 | so->so_laddr.s_addr = laddr; /* Ditto */
|
---|
1040 |
|
---|
1041 | memset(&addr, 0, sizeof(addr));
|
---|
1042 | #ifdef RT_OS_DARWIN
|
---|
1043 | addr.sin_len = sizeof(addr);
|
---|
1044 | #endif
|
---|
1045 | addr.sin_family = AF_INET;
|
---|
1046 | addr.sin_addr.s_addr = bind_addr;
|
---|
1047 | addr.sin_port = port;
|
---|
1048 |
|
---|
1049 | /**
|
---|
1050 | * changing listen(,1->SOMAXCONN) shouldn't be harmful for NAT's TCP/IP stack,
|
---|
1051 | * kernel will choose the optimal value for requests queue length.
|
---|
1052 | * @note: MSDN recommends low (2-4) values for bluetooth networking devices.
|
---|
1053 | */
|
---|
1054 | if ( ((s = socket(AF_INET, SOCK_STREAM, 0)) < 0)
|
---|
1055 | || (setsockopt(s, SOL_SOCKET, SO_REUSEADDR,(char *)&opt, sizeof(int)) < 0)
|
---|
1056 | || (bind(s,(struct sockaddr *)&addr, sizeof(addr)) < 0)
|
---|
1057 | || (listen(s, pData->soMaxConn) < 0))
|
---|
1058 | {
|
---|
1059 | #ifdef RT_OS_WINDOWS
|
---|
1060 | int tmperrno = WSAGetLastError(); /* Don't clobber the real reason we failed */
|
---|
1061 | closesocket(s);
|
---|
1062 | QSOCKET_LOCK(tcb);
|
---|
1063 | sofree(pData, so);
|
---|
1064 | QSOCKET_UNLOCK(tcb);
|
---|
1065 | /* Restore the real errno */
|
---|
1066 | WSASetLastError(tmperrno);
|
---|
1067 | #else
|
---|
1068 | int tmperrno = errno; /* Don't clobber the real reason we failed */
|
---|
1069 | close(s);
|
---|
1070 | if (sototcpcb(so))
|
---|
1071 | tcp_close(pData, sototcpcb(so));
|
---|
1072 | else
|
---|
1073 | sofree(pData, so);
|
---|
1074 | /* Restore the real errno */
|
---|
1075 | errno = tmperrno;
|
---|
1076 | #endif
|
---|
1077 | return NULL;
|
---|
1078 | }
|
---|
1079 | fd_nonblock(s);
|
---|
1080 | setsockopt(s, SOL_SOCKET, SO_OOBINLINE,(char *)&opt, sizeof(int));
|
---|
1081 |
|
---|
1082 | getsockname(s,(struct sockaddr *)&addr,&addrlen);
|
---|
1083 | so->so_fport = addr.sin_port;
|
---|
1084 | /* set socket buffers */
|
---|
1085 | opt = pData->socket_rcv;
|
---|
1086 | status = setsockopt(s, SOL_SOCKET, SO_RCVBUF, (char *)&opt, sizeof(int));
|
---|
1087 | if (status < 0)
|
---|
1088 | {
|
---|
1089 | LogRel(("NAT: Error(%d) while setting RCV capacity to (%d)\n", errno, opt));
|
---|
1090 | goto no_sockopt;
|
---|
1091 | }
|
---|
1092 | opt = pData->socket_snd;
|
---|
1093 | status = setsockopt(s, SOL_SOCKET, SO_SNDBUF, (char *)&opt, sizeof(int));
|
---|
1094 | if (status < 0)
|
---|
1095 | {
|
---|
1096 | LogRel(("NAT: Error(%d) while setting SND capacity to (%d)\n", errno, opt));
|
---|
1097 | goto no_sockopt;
|
---|
1098 | }
|
---|
1099 | no_sockopt:
|
---|
1100 | if (addr.sin_addr.s_addr == 0 || addr.sin_addr.s_addr == loopback_addr.s_addr)
|
---|
1101 | so->so_faddr = alias_addr;
|
---|
1102 | else
|
---|
1103 | so->so_faddr = addr.sin_addr;
|
---|
1104 |
|
---|
1105 | so->s = s;
|
---|
1106 | SOCKET_UNLOCK(so);
|
---|
1107 | return so;
|
---|
1108 | }
|
---|
1109 |
|
---|
1110 | /*
|
---|
1111 | * Data is available in so_rcv
|
---|
1112 | * Just write() the data to the socket
|
---|
1113 | * XXX not yet...
|
---|
1114 | * @todo do we really need this function, what it's intended to do?
|
---|
1115 | */
|
---|
1116 | void
|
---|
1117 | sorwakeup(struct socket *so)
|
---|
1118 | {
|
---|
1119 | NOREF(so);
|
---|
1120 | #if 0
|
---|
1121 | sowrite(so);
|
---|
1122 | FD_CLR(so->s,&writefds);
|
---|
1123 | #endif
|
---|
1124 | }
|
---|
1125 |
|
---|
1126 | /*
|
---|
1127 | * Data has been freed in so_snd
|
---|
1128 | * We have room for a read() if we want to
|
---|
1129 | * For now, don't read, it'll be done in the main loop
|
---|
1130 | */
|
---|
1131 | void
|
---|
1132 | sowwakeup(struct socket *so)
|
---|
1133 | {
|
---|
1134 | NOREF(so);
|
---|
1135 | }
|
---|
1136 |
|
---|
1137 | /*
|
---|
1138 | * Various session state calls
|
---|
1139 | * XXX Should be #define's
|
---|
1140 | * The socket state stuff needs work, these often get call 2 or 3
|
---|
1141 | * times each when only 1 was needed
|
---|
1142 | */
|
---|
1143 | void
|
---|
1144 | soisfconnecting(struct socket *so)
|
---|
1145 | {
|
---|
1146 | so->so_state &= ~(SS_NOFDREF|SS_ISFCONNECTED|SS_FCANTRCVMORE|
|
---|
1147 | SS_FCANTSENDMORE|SS_FWDRAIN);
|
---|
1148 | so->so_state |= SS_ISFCONNECTING; /* Clobber other states */
|
---|
1149 | }
|
---|
1150 |
|
---|
1151 | void
|
---|
1152 | soisfconnected(struct socket *so)
|
---|
1153 | {
|
---|
1154 | LogFlowFunc(("ENTER: so:%R[natsock]\n", so));
|
---|
1155 | so->so_state &= ~(SS_ISFCONNECTING|SS_FWDRAIN|SS_NOFDREF);
|
---|
1156 | so->so_state |= SS_ISFCONNECTED; /* Clobber other states */
|
---|
1157 | LogFlowFunc(("LEAVE: so:%R[natsock]\n", so));
|
---|
1158 | }
|
---|
1159 |
|
---|
1160 | int
|
---|
1161 | sofcantrcvmore(struct socket *so)
|
---|
1162 | {
|
---|
1163 | int err = 0;
|
---|
1164 |
|
---|
1165 | LogFlowFunc(("ENTER: so:%R[natsock]\n", so));
|
---|
1166 | if ((so->so_state & SS_NOFDREF) == 0)
|
---|
1167 | {
|
---|
1168 | /*
|
---|
1169 | * If remote closes first and then sends an RST, the recv() in
|
---|
1170 | * soread() will keep reporting EOF without any error
|
---|
1171 | * indication. As far as I can tell the only way to detect
|
---|
1172 | * this on Linux is to check if shutdown() succeeds here (but
|
---|
1173 | * see below).
|
---|
1174 | *
|
---|
1175 | * OTOH on OS X shutdown() "helpfully" checks if remote has
|
---|
1176 | * already closed and then always returns ENOTCONN
|
---|
1177 | * immediately.
|
---|
1178 | */
|
---|
1179 | int status = shutdown(so->s, SHUT_RD);
|
---|
1180 | #if defined(RT_OS_LINUX)
|
---|
1181 | if (status < 0)
|
---|
1182 | err = errno;
|
---|
1183 | #else
|
---|
1184 | RT_NOREF(status);
|
---|
1185 | #endif
|
---|
1186 | }
|
---|
1187 | so->so_state &= ~(SS_ISFCONNECTING);
|
---|
1188 | if (so->so_state & SS_FCANTSENDMORE)
|
---|
1189 | {
|
---|
1190 | #if defined(RT_OS_LINUX)
|
---|
1191 | /*
|
---|
1192 | * If we have closed first, and remote closes, shutdown will
|
---|
1193 | * return ENOTCONN, but this is expected. Don't tell the
|
---|
1194 | * caller there was an error.
|
---|
1195 | */
|
---|
1196 | if (err == ENOTCONN)
|
---|
1197 | err = 0;
|
---|
1198 | #endif
|
---|
1199 | so->so_state = SS_NOFDREF; /* Don't select it */
|
---|
1200 | /* XXX close() here as well? */
|
---|
1201 | }
|
---|
1202 | else
|
---|
1203 | so->so_state |= SS_FCANTRCVMORE;
|
---|
1204 |
|
---|
1205 | LogFlowFunc(("LEAVE: %d\n", err));
|
---|
1206 | return err;
|
---|
1207 | }
|
---|
1208 |
|
---|
1209 | void
|
---|
1210 | sofcantsendmore(struct socket *so)
|
---|
1211 | {
|
---|
1212 | LogFlowFunc(("ENTER: so:%R[natsock]\n", so));
|
---|
1213 | if ((so->so_state & SS_NOFDREF) == 0)
|
---|
1214 | shutdown(so->s, 1); /* send FIN to fhost */
|
---|
1215 |
|
---|
1216 | so->so_state &= ~(SS_ISFCONNECTING);
|
---|
1217 | if (so->so_state & SS_FCANTRCVMORE)
|
---|
1218 | so->so_state = SS_NOFDREF; /* as above */
|
---|
1219 | else
|
---|
1220 | so->so_state |= SS_FCANTSENDMORE;
|
---|
1221 | LogFlowFuncLeave();
|
---|
1222 | }
|
---|
1223 |
|
---|
1224 | void
|
---|
1225 | soisfdisconnected(struct socket *so)
|
---|
1226 | {
|
---|
1227 | NOREF(so);
|
---|
1228 | #if 0
|
---|
1229 | so->so_state &= ~(SS_ISFCONNECTING|SS_ISFCONNECTED);
|
---|
1230 | close(so->s);
|
---|
1231 | so->so_state = SS_ISFDISCONNECTED;
|
---|
1232 | /*
|
---|
1233 | * XXX Do nothing ... ?
|
---|
1234 | */
|
---|
1235 | #endif
|
---|
1236 | }
|
---|
1237 |
|
---|
1238 | /*
|
---|
1239 | * Set write drain mode
|
---|
1240 | * Set CANTSENDMORE once all data has been write()n
|
---|
1241 | */
|
---|
1242 | void
|
---|
1243 | sofwdrain(struct socket *so)
|
---|
1244 | {
|
---|
1245 | if (SBUF_LEN(&so->so_rcv))
|
---|
1246 | so->so_state |= SS_FWDRAIN;
|
---|
1247 | else
|
---|
1248 | sofcantsendmore(so);
|
---|
1249 | }
|
---|
1250 |
|
---|
1251 | #if !defined(RT_OS_WINDOWS)
|
---|
1252 | static void
|
---|
1253 | send_icmp_to_guest(PNATState pData, char *buff, size_t len, const struct sockaddr_in *addr)
|
---|
1254 | {
|
---|
1255 | struct ip *ip;
|
---|
1256 | uint32_t dst, src;
|
---|
1257 | char ip_copy[256];
|
---|
1258 | struct icmp *icp;
|
---|
1259 | int old_ip_len = 0;
|
---|
1260 | int hlen, original_hlen = 0;
|
---|
1261 | struct mbuf *m;
|
---|
1262 | struct icmp_msg *icm;
|
---|
1263 | uint8_t proto;
|
---|
1264 | int type = 0;
|
---|
1265 |
|
---|
1266 | ip = (struct ip *)buff;
|
---|
1267 | /* Fix ip->ip_len to contain the total packet length including the header
|
---|
1268 | * in _host_ byte order for all OSes. On Darwin, that value already is in
|
---|
1269 | * host byte order. Solaris and Darwin report only the payload. */
|
---|
1270 | #ifndef RT_OS_DARWIN
|
---|
1271 | ip->ip_len = RT_N2H_U16(ip->ip_len);
|
---|
1272 | #endif
|
---|
1273 | hlen = (ip->ip_hl << 2);
|
---|
1274 | #if defined(RT_OS_SOLARIS) || defined(RT_OS_DARWIN)
|
---|
1275 | ip->ip_len += hlen;
|
---|
1276 | #endif
|
---|
1277 | if (ip->ip_len < hlen + ICMP_MINLEN)
|
---|
1278 | {
|
---|
1279 | Log(("send_icmp_to_guest: ICMP header is too small to understand which type/subtype of the datagram\n"));
|
---|
1280 | return;
|
---|
1281 | }
|
---|
1282 | icp = (struct icmp *)((char *)ip + hlen);
|
---|
1283 |
|
---|
1284 | Log(("ICMP:received msg(t:%d, c:%d)\n", icp->icmp_type, icp->icmp_code));
|
---|
1285 | if ( icp->icmp_type != ICMP_ECHOREPLY
|
---|
1286 | && icp->icmp_type != ICMP_TIMXCEED
|
---|
1287 | && icp->icmp_type != ICMP_UNREACH)
|
---|
1288 | {
|
---|
1289 | return;
|
---|
1290 | }
|
---|
1291 |
|
---|
1292 | /*
|
---|
1293 | * ICMP_ECHOREPLY, ICMP_TIMXCEED, ICMP_UNREACH minimal header size is
|
---|
1294 | * ICMP_ECHOREPLY assuming data 0
|
---|
1295 | * icmp_{type(8), code(8), cksum(16),identifier(16),seqnum(16)}
|
---|
1296 | */
|
---|
1297 | if (ip->ip_len < hlen + 8)
|
---|
1298 | {
|
---|
1299 | Log(("send_icmp_to_guest: NAT accept ICMP_{ECHOREPLY, TIMXCEED, UNREACH} the minimum size is 64 (see rfc792)\n"));
|
---|
1300 | return;
|
---|
1301 | }
|
---|
1302 |
|
---|
1303 | type = icp->icmp_type;
|
---|
1304 | if ( type == ICMP_TIMXCEED
|
---|
1305 | || type == ICMP_UNREACH)
|
---|
1306 | {
|
---|
1307 | /*
|
---|
1308 | * ICMP_TIMXCEED, ICMP_UNREACH minimal header size is
|
---|
1309 | * icmp_{type(8), code(8), cksum(16),unused(32)} + IP header + 64 bit of original datagram
|
---|
1310 | */
|
---|
1311 | if (ip->ip_len < hlen + 2*8 + sizeof(struct ip))
|
---|
1312 | {
|
---|
1313 | Log(("send_icmp_to_guest: NAT accept ICMP_{TIMXCEED, UNREACH} the minimum size of ipheader + 64 bit of data (see rfc792)\n"));
|
---|
1314 | return;
|
---|
1315 | }
|
---|
1316 | ip = &icp->icmp_ip;
|
---|
1317 | }
|
---|
1318 |
|
---|
1319 | icm = icmp_find_original_mbuf(pData, ip);
|
---|
1320 | if (icm == NULL)
|
---|
1321 | {
|
---|
1322 | Log(("NAT: Can't find the corresponding packet for the received ICMP\n"));
|
---|
1323 | return;
|
---|
1324 | }
|
---|
1325 |
|
---|
1326 | m = icm->im_m;
|
---|
1327 | if (!m)
|
---|
1328 | {
|
---|
1329 | LogFunc(("%R[natsock] hasn't stored it's mbuf on sent\n", icm->im_so));
|
---|
1330 | goto done;
|
---|
1331 | }
|
---|
1332 |
|
---|
1333 | src = addr->sin_addr.s_addr;
|
---|
1334 | if (type == ICMP_ECHOREPLY)
|
---|
1335 | {
|
---|
1336 | struct ip *ip0 = mtod(m, struct ip *);
|
---|
1337 | struct icmp *icp0 = (struct icmp *)((char *)ip0 + (ip0->ip_hl << 2));
|
---|
1338 | if (icp0->icmp_type != ICMP_ECHO)
|
---|
1339 | {
|
---|
1340 | Log(("NAT: we haven't found echo for this reply\n"));
|
---|
1341 | goto done;
|
---|
1342 | }
|
---|
1343 | /*
|
---|
1344 | * while combining buffer to send (see ip_icmp.c) we control ICMP header only,
|
---|
1345 | * IP header combined by OS network stack, our local copy of IP header contians values
|
---|
1346 | * in host byte order so no byte order conversion is required. IP headers fields are converting
|
---|
1347 | * in ip_output0 routine only.
|
---|
1348 | */
|
---|
1349 | if ( (ip->ip_len - hlen)
|
---|
1350 | != (ip0->ip_len - (ip0->ip_hl << 2)))
|
---|
1351 | {
|
---|
1352 | Log(("NAT: ECHO(%d) lenght doesn't match ECHOREPLY(%d)\n",
|
---|
1353 | (ip->ip_len - hlen), (ip0->ip_len - (ip0->ip_hl << 2))));
|
---|
1354 | goto done;
|
---|
1355 | }
|
---|
1356 | }
|
---|
1357 |
|
---|
1358 | /* ip points on origianal ip header */
|
---|
1359 | ip = mtod(m, struct ip *);
|
---|
1360 | proto = ip->ip_p;
|
---|
1361 | /* Now ip is pointing on header we've sent from guest */
|
---|
1362 | if ( icp->icmp_type == ICMP_TIMXCEED
|
---|
1363 | || icp->icmp_type == ICMP_UNREACH)
|
---|
1364 | {
|
---|
1365 | old_ip_len = (ip->ip_hl << 2) + 64;
|
---|
1366 | if (old_ip_len > sizeof(ip_copy))
|
---|
1367 | old_ip_len = sizeof(ip_copy);
|
---|
1368 | memcpy(ip_copy, ip, old_ip_len);
|
---|
1369 | }
|
---|
1370 |
|
---|
1371 | /* source address from original IP packet*/
|
---|
1372 | dst = ip->ip_src.s_addr;
|
---|
1373 |
|
---|
1374 | /* overide ther tail of old packet */
|
---|
1375 | ip = mtod(m, struct ip *); /* ip is from mbuf we've overrided */
|
---|
1376 | original_hlen = ip->ip_hl << 2;
|
---|
1377 | /* saves original ip header and options */
|
---|
1378 | m_copyback(pData, m, original_hlen, len - hlen, buff + hlen);
|
---|
1379 | ip->ip_len = m_length(m, NULL);
|
---|
1380 | ip->ip_p = IPPROTO_ICMP; /* the original package could be whatever, but we're response via ICMP*/
|
---|
1381 |
|
---|
1382 | icp = (struct icmp *)((char *)ip + (ip->ip_hl << 2));
|
---|
1383 | type = icp->icmp_type;
|
---|
1384 | if ( type == ICMP_TIMXCEED
|
---|
1385 | || type == ICMP_UNREACH)
|
---|
1386 | {
|
---|
1387 | /* according RFC 793 error messages required copy of initial IP header + 64 bit */
|
---|
1388 | memcpy(&icp->icmp_ip, ip_copy, old_ip_len);
|
---|
1389 |
|
---|
1390 | /* undo byte order conversions done in ip_input() */
|
---|
1391 | HTONS(icp->icmp_ip.ip_len);
|
---|
1392 | HTONS(icp->icmp_ip.ip_id);
|
---|
1393 | HTONS(icp->icmp_ip.ip_off);
|
---|
1394 |
|
---|
1395 | ip->ip_tos = ((ip->ip_tos & 0x1E) | 0xC0); /* high priority for errors */
|
---|
1396 | }
|
---|
1397 |
|
---|
1398 | ip->ip_src.s_addr = src;
|
---|
1399 | ip->ip_dst.s_addr = dst;
|
---|
1400 | icmp_reflect(pData, m);
|
---|
1401 | /* m was freed */
|
---|
1402 | icm->im_m = NULL;
|
---|
1403 |
|
---|
1404 | done:
|
---|
1405 | icmp_msg_delete(pData, icm);
|
---|
1406 | }
|
---|
1407 |
|
---|
1408 | static void sorecvfrom_icmp_unix(PNATState pData, struct socket *so)
|
---|
1409 | {
|
---|
1410 | struct sockaddr_in addr;
|
---|
1411 | socklen_t addrlen = sizeof(struct sockaddr_in);
|
---|
1412 | struct ip ip;
|
---|
1413 | char *buff;
|
---|
1414 | int len = 0;
|
---|
1415 |
|
---|
1416 | /* 1- step: read the ip header */
|
---|
1417 | len = recvfrom(so->s, &ip, sizeof(struct ip), MSG_PEEK,
|
---|
1418 | (struct sockaddr *)&addr, &addrlen);
|
---|
1419 | if ( len < 0
|
---|
1420 | && ( soIgnorableErrorCode(errno)
|
---|
1421 | || errno == ENOTCONN))
|
---|
1422 | {
|
---|
1423 | Log(("sorecvfrom_icmp_unix: 1 - step can't read IP datagramm (would block)\n"));
|
---|
1424 | return;
|
---|
1425 | }
|
---|
1426 |
|
---|
1427 | if ( len < sizeof(struct ip)
|
---|
1428 | || len < 0
|
---|
1429 | || len == 0)
|
---|
1430 | {
|
---|
1431 | u_char code;
|
---|
1432 | code = ICMP_UNREACH_PORT;
|
---|
1433 |
|
---|
1434 | if (errno == EHOSTUNREACH)
|
---|
1435 | code = ICMP_UNREACH_HOST;
|
---|
1436 | else if (errno == ENETUNREACH)
|
---|
1437 | code = ICMP_UNREACH_NET;
|
---|
1438 |
|
---|
1439 | LogRel(("NAT: UDP ICMP rx errno=%d (%s)\n", errno, strerror(errno)));
|
---|
1440 | icmp_error(pData, so->so_m, ICMP_UNREACH, code, 0, strerror(errno));
|
---|
1441 | so->so_m = NULL;
|
---|
1442 | Log(("sorecvfrom_icmp_unix: 1 - step can't read IP datagramm\n"));
|
---|
1443 | return;
|
---|
1444 | }
|
---|
1445 | /* basic check of IP header */
|
---|
1446 | if ( ip.ip_v != IPVERSION
|
---|
1447 | # ifndef RT_OS_DARWIN
|
---|
1448 | || ip.ip_p != IPPROTO_ICMP
|
---|
1449 | # endif
|
---|
1450 | )
|
---|
1451 | {
|
---|
1452 | Log(("sorecvfrom_icmp_unix: 1 - step IP isn't IPv4\n"));
|
---|
1453 | return;
|
---|
1454 | }
|
---|
1455 | # ifndef RT_OS_DARWIN
|
---|
1456 | /* Darwin reports the IP length already in host byte order. */
|
---|
1457 | ip.ip_len = RT_N2H_U16(ip.ip_len);
|
---|
1458 | # endif
|
---|
1459 | # if defined(RT_OS_SOLARIS) || defined(RT_OS_DARWIN)
|
---|
1460 | /* Solaris and Darwin report the payload only */
|
---|
1461 | ip.ip_len += (ip.ip_hl << 2);
|
---|
1462 | # endif
|
---|
1463 | /* Note: ip->ip_len in host byte order (all OS) */
|
---|
1464 | len = ip.ip_len;
|
---|
1465 | buff = RTMemAlloc(len);
|
---|
1466 | if (buff == NULL)
|
---|
1467 | {
|
---|
1468 | Log(("sorecvfrom_icmp_unix: 1 - step can't allocate enought room for datagram\n"));
|
---|
1469 | return;
|
---|
1470 | }
|
---|
1471 | /* 2 - step: we're reading rest of the datagramm to the buffer */
|
---|
1472 | addrlen = sizeof(struct sockaddr_in);
|
---|
1473 | memset(&addr, 0, addrlen);
|
---|
1474 | len = recvfrom(so->s, buff, len, 0,
|
---|
1475 | (struct sockaddr *)&addr, &addrlen);
|
---|
1476 | if ( len < 0
|
---|
1477 | && ( soIgnorableErrorCode(errno)
|
---|
1478 | || errno == ENOTCONN))
|
---|
1479 | {
|
---|
1480 | Log(("sorecvfrom_icmp_unix: 2 - step can't read IP body (would block expected:%d)\n",
|
---|
1481 | ip.ip_len));
|
---|
1482 | RTMemFree(buff);
|
---|
1483 | return;
|
---|
1484 | }
|
---|
1485 | if ( len < 0
|
---|
1486 | || len == 0)
|
---|
1487 | {
|
---|
1488 | Log(("sorecvfrom_icmp_unix: 2 - step read of the rest of datagramm is fallen (errno:%d, len:%d expected: %d)\n",
|
---|
1489 | errno, len, (ip.ip_len - sizeof(struct ip))));
|
---|
1490 | RTMemFree(buff);
|
---|
1491 | return;
|
---|
1492 | }
|
---|
1493 | /* len is modified in 2nd read, when the rest of the datagramm was read */
|
---|
1494 | send_icmp_to_guest(pData, buff, len, &addr);
|
---|
1495 | RTMemFree(buff);
|
---|
1496 | }
|
---|
1497 | #endif /* !RT_OS_WINDOWS */
|
---|