1 | /* $Id: pxudp.c 63277 2016-08-10 14:31:08Z vboxsync $ */
|
---|
2 | /** @file
|
---|
3 | * NAT Network - UDP proxy.
|
---|
4 | */
|
---|
5 |
|
---|
6 | /*
|
---|
7 | * Copyright (C) 2013-2016 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 | #define LOG_GROUP LOG_GROUP_NAT_SERVICE
|
---|
19 |
|
---|
20 | #include "winutils.h"
|
---|
21 | #include "proxy.h"
|
---|
22 | #include "proxy_pollmgr.h"
|
---|
23 | #include "pxremap.h"
|
---|
24 |
|
---|
25 | #ifndef RT_OS_WINDOWS
|
---|
26 | #include <sys/types.h>
|
---|
27 | #include <sys/socket.h>
|
---|
28 | #ifdef RT_OS_DARWIN
|
---|
29 | # define __APPLE_USE_RFC_3542
|
---|
30 | #endif
|
---|
31 | #include <netinet/in.h>
|
---|
32 | #include <stdlib.h>
|
---|
33 | #include <stdint.h>
|
---|
34 | #include <stdio.h>
|
---|
35 | #include <poll.h>
|
---|
36 |
|
---|
37 | #include <err.h> /* BSD'ism */
|
---|
38 | #else
|
---|
39 | #include <stdlib.h>
|
---|
40 | #include <iprt/stdint.h>
|
---|
41 | #include <stdio.h>
|
---|
42 | #include "winpoll.h"
|
---|
43 | #endif
|
---|
44 |
|
---|
45 | #include "lwip/opt.h"
|
---|
46 |
|
---|
47 | #include "lwip/sys.h"
|
---|
48 | #include "lwip/tcpip.h"
|
---|
49 | #include "lwip/udp.h"
|
---|
50 | #include "lwip/icmp.h"
|
---|
51 |
|
---|
52 | struct pxudp {
|
---|
53 | /**
|
---|
54 | * Our poll manager handler.
|
---|
55 | */
|
---|
56 | struct pollmgr_handler pmhdl;
|
---|
57 |
|
---|
58 | /**
|
---|
59 | * lwIP ("internal") side of the proxied connection.
|
---|
60 | */
|
---|
61 | struct udp_pcb *pcb;
|
---|
62 |
|
---|
63 | /**
|
---|
64 | * Host ("external") side of the proxied connection.
|
---|
65 | */
|
---|
66 | SOCKET sock;
|
---|
67 |
|
---|
68 | /**
|
---|
69 | * Is this pcb a mapped host loopback?
|
---|
70 | */
|
---|
71 | int is_mapped;
|
---|
72 |
|
---|
73 | /**
|
---|
74 | * Cached value of TTL socket option.
|
---|
75 | */
|
---|
76 | int ttl;
|
---|
77 |
|
---|
78 | /**
|
---|
79 | * Cached value of TOS socket option.
|
---|
80 | */
|
---|
81 | int tos;
|
---|
82 |
|
---|
83 | /**
|
---|
84 | * Cached value of "don't fragment" socket option.
|
---|
85 | */
|
---|
86 | int df;
|
---|
87 |
|
---|
88 | /**
|
---|
89 | * For some protocols (notably: DNS) we know we are getting just
|
---|
90 | * one reply, so we don't want the pcb and the socket to sit there
|
---|
91 | * waiting to be g/c'ed by timeout. This field counts request and
|
---|
92 | * replies for them.
|
---|
93 | */
|
---|
94 | int count;
|
---|
95 |
|
---|
96 | /**
|
---|
97 | * Mailbox for inbound pbufs.
|
---|
98 | *
|
---|
99 | * XXX: since we have single producer and single consumer we can
|
---|
100 | * use lockless ringbuf like for pxtcp.
|
---|
101 | */
|
---|
102 | sys_mbox_t inmbox;
|
---|
103 |
|
---|
104 | /**
|
---|
105 | * lwIP thread's strong reference to us.
|
---|
106 | */
|
---|
107 | struct pollmgr_refptr *rp;
|
---|
108 |
|
---|
109 | /*
|
---|
110 | * We use static messages to void malloc/free overhead.
|
---|
111 | */
|
---|
112 | struct tcpip_msg msg_delete; /* delete pxudp */
|
---|
113 | struct tcpip_msg msg_inbound; /* trigger send of inbound data */
|
---|
114 | };
|
---|
115 |
|
---|
116 |
|
---|
117 | static struct pxudp *pxudp_allocate(void);
|
---|
118 | static void pxudp_drain_inmbox(struct pxudp *);
|
---|
119 | static void pxudp_free(struct pxudp *);
|
---|
120 |
|
---|
121 | static struct udp_pcb *pxudp_pcb_dissociate(struct pxudp *);
|
---|
122 |
|
---|
123 | /* poll manager callbacks for pxudp related channels */
|
---|
124 | static int pxudp_pmgr_chan_add(struct pollmgr_handler *, SOCKET, int);
|
---|
125 | static int pxudp_pmgr_chan_del(struct pollmgr_handler *, SOCKET, int);
|
---|
126 |
|
---|
127 | /* helper functions for sending/receiving pxudp over poll manager channels */
|
---|
128 | static ssize_t pxudp_chan_send(enum pollmgr_slot_t, struct pxudp *);
|
---|
129 | static ssize_t pxudp_chan_send_weak(enum pollmgr_slot_t, struct pxudp *);
|
---|
130 | static struct pxudp *pxudp_chan_recv(struct pollmgr_handler *, SOCKET, int);
|
---|
131 | static struct pxudp *pxudp_chan_recv_strong(struct pollmgr_handler *, SOCKET, int);
|
---|
132 |
|
---|
133 | /* poll manager callbacks for individual sockets */
|
---|
134 | static int pxudp_pmgr_pump(struct pollmgr_handler *, SOCKET, int);
|
---|
135 |
|
---|
136 | /* convenience function for poll manager callback */
|
---|
137 | static int pxudp_schedule_delete(struct pxudp *);
|
---|
138 |
|
---|
139 | /* lwip thread callbacks called via proxy_lwip_post() */
|
---|
140 | static void pxudp_pcb_delete_pxudp(void *);
|
---|
141 |
|
---|
142 | /* outbound ttl check */
|
---|
143 | static int pxudp_ttl_expired(struct pbuf *);
|
---|
144 |
|
---|
145 | /* udp pcb callbacks &c */
|
---|
146 | static void pxudp_pcb_accept(void *, struct udp_pcb *, struct pbuf *, ip_addr_t *, u16_t);
|
---|
147 | static void pxudp_pcb_recv(void *, struct udp_pcb *, struct pbuf *, ip_addr_t *, u16_t);
|
---|
148 | static void pxudp_pcb_forward_outbound(struct pxudp *, struct pbuf *, ip_addr_t *, u16_t);
|
---|
149 | static void pxudp_pcb_expired(struct pxudp *);
|
---|
150 | static void pxudp_pcb_write_inbound(void *);
|
---|
151 | static void pxudp_pcb_forward_inbound(struct pxudp *);
|
---|
152 |
|
---|
153 | /* poll manager handlers for pxudp channels */
|
---|
154 | static struct pollmgr_handler pxudp_pmgr_chan_add_hdl;
|
---|
155 | static struct pollmgr_handler pxudp_pmgr_chan_del_hdl;
|
---|
156 |
|
---|
157 |
|
---|
158 | void
|
---|
159 | pxudp_init(void)
|
---|
160 | {
|
---|
161 | /*
|
---|
162 | * Create channels.
|
---|
163 | */
|
---|
164 | pxudp_pmgr_chan_add_hdl.callback = pxudp_pmgr_chan_add;
|
---|
165 | pxudp_pmgr_chan_add_hdl.data = NULL;
|
---|
166 | pxudp_pmgr_chan_add_hdl.slot = -1;
|
---|
167 | pollmgr_add_chan(POLLMGR_CHAN_PXUDP_ADD, &pxudp_pmgr_chan_add_hdl);
|
---|
168 |
|
---|
169 | pxudp_pmgr_chan_del_hdl.callback = pxudp_pmgr_chan_del;
|
---|
170 | pxudp_pmgr_chan_del_hdl.data = NULL;
|
---|
171 | pxudp_pmgr_chan_del_hdl.slot = -1;
|
---|
172 | pollmgr_add_chan(POLLMGR_CHAN_PXUDP_DEL, &pxudp_pmgr_chan_del_hdl);
|
---|
173 |
|
---|
174 | udp_proxy_accept(pxudp_pcb_accept);
|
---|
175 | }
|
---|
176 |
|
---|
177 |
|
---|
178 | /**
|
---|
179 | * Syntactic sugar for sending pxudp pointer over poll manager
|
---|
180 | * channel. Used by lwip thread functions.
|
---|
181 | */
|
---|
182 | static ssize_t
|
---|
183 | pxudp_chan_send(enum pollmgr_slot_t chan, struct pxudp *pxudp)
|
---|
184 | {
|
---|
185 | return pollmgr_chan_send(chan, &pxudp, sizeof(pxudp));
|
---|
186 | }
|
---|
187 |
|
---|
188 |
|
---|
189 | /**
|
---|
190 | * Syntactic sugar for sending weak reference to pxudp over poll
|
---|
191 | * manager channel. Used by lwip thread functions.
|
---|
192 | */
|
---|
193 | static ssize_t
|
---|
194 | pxudp_chan_send_weak(enum pollmgr_slot_t chan, struct pxudp *pxudp)
|
---|
195 | {
|
---|
196 | pollmgr_refptr_weak_ref(pxudp->rp);
|
---|
197 | return pollmgr_chan_send(chan, &pxudp->rp, sizeof(pxudp->rp));
|
---|
198 | }
|
---|
199 |
|
---|
200 |
|
---|
201 | /**
|
---|
202 | * Counterpart of pxudp_chan_send().
|
---|
203 | */
|
---|
204 | static struct pxudp *
|
---|
205 | pxudp_chan_recv(struct pollmgr_handler *handler, SOCKET fd, int revents)
|
---|
206 | {
|
---|
207 | struct pxudp *pxudp;
|
---|
208 |
|
---|
209 | pxudp = (struct pxudp *)pollmgr_chan_recv_ptr(handler, fd, revents);
|
---|
210 | return pxudp;
|
---|
211 | }
|
---|
212 |
|
---|
213 |
|
---|
214 | /**
|
---|
215 | * Counterpart of pxudp_chan_send_weak().
|
---|
216 | */
|
---|
217 | struct pxudp *
|
---|
218 | pxudp_chan_recv_strong(struct pollmgr_handler *handler, SOCKET fd, int revents)
|
---|
219 | {
|
---|
220 | struct pollmgr_refptr *rp;
|
---|
221 | struct pollmgr_handler *base;
|
---|
222 | struct pxudp *pxudp;
|
---|
223 |
|
---|
224 | rp = (struct pollmgr_refptr *)pollmgr_chan_recv_ptr(handler, fd, revents);
|
---|
225 | base = (struct pollmgr_handler *)pollmgr_refptr_get(rp);
|
---|
226 | pxudp = (struct pxudp *)base;
|
---|
227 |
|
---|
228 | return pxudp;
|
---|
229 | }
|
---|
230 |
|
---|
231 |
|
---|
232 | /**
|
---|
233 | * POLLMGR_CHAN_PXUDP_ADD handler.
|
---|
234 | *
|
---|
235 | * Get new pxudp from lwip thread and start polling its socket.
|
---|
236 | */
|
---|
237 | static int
|
---|
238 | pxudp_pmgr_chan_add(struct pollmgr_handler *handler, SOCKET fd, int revents)
|
---|
239 | {
|
---|
240 | struct pxudp *pxudp;
|
---|
241 | int status;
|
---|
242 |
|
---|
243 | pxudp = pxudp_chan_recv(handler, fd, revents);
|
---|
244 | DPRINTF(("pxudp_add: new pxudp %p; pcb %p\n",
|
---|
245 | (void *)pxudp, (void *)pxudp->pcb));
|
---|
246 |
|
---|
247 | LWIP_ASSERT1(pxudp != NULL);
|
---|
248 | LWIP_ASSERT1(pxudp->pmhdl.callback != NULL);
|
---|
249 | LWIP_ASSERT1(pxudp->pmhdl.data = (void *)pxudp);
|
---|
250 | LWIP_ASSERT1(pxudp->pmhdl.slot < 0);
|
---|
251 |
|
---|
252 |
|
---|
253 | status = pollmgr_add(&pxudp->pmhdl, pxudp->sock, POLLIN);
|
---|
254 | if (status < 0) {
|
---|
255 | pxudp_schedule_delete(pxudp);
|
---|
256 | }
|
---|
257 |
|
---|
258 | return POLLIN;
|
---|
259 | }
|
---|
260 |
|
---|
261 |
|
---|
262 | /**
|
---|
263 | * POLLMGR_CHAN_PXUDP_DEL handler.
|
---|
264 | */
|
---|
265 | static int
|
---|
266 | pxudp_pmgr_chan_del(struct pollmgr_handler *handler, SOCKET fd, int revents)
|
---|
267 | {
|
---|
268 | struct pxudp *pxudp;
|
---|
269 |
|
---|
270 | pxudp = pxudp_chan_recv_strong(handler, fd, revents);
|
---|
271 | if (pxudp == NULL) {
|
---|
272 | return POLLIN;
|
---|
273 | }
|
---|
274 |
|
---|
275 | DPRINTF(("pxudp_del: pxudp %p; socket %d\n", (void *)pxudp, pxudp->sock));
|
---|
276 |
|
---|
277 | pollmgr_del_slot(pxudp->pmhdl.slot);
|
---|
278 |
|
---|
279 | /*
|
---|
280 | * Go back to lwip thread to delete after any pending callbacks
|
---|
281 | * for unprocessed inbound traffic are drained.
|
---|
282 | */
|
---|
283 | pxudp_schedule_delete(pxudp);
|
---|
284 |
|
---|
285 | return POLLIN;
|
---|
286 | }
|
---|
287 |
|
---|
288 |
|
---|
289 | static struct pxudp *
|
---|
290 | pxudp_allocate(void)
|
---|
291 | {
|
---|
292 | struct pxudp *pxudp;
|
---|
293 | err_t error;
|
---|
294 |
|
---|
295 | pxudp = (struct pxudp *)malloc(sizeof(*pxudp));
|
---|
296 | if (pxudp == NULL) {
|
---|
297 | return NULL;
|
---|
298 | }
|
---|
299 |
|
---|
300 | pxudp->pmhdl.callback = NULL;
|
---|
301 | pxudp->pmhdl.data = (void *)pxudp;
|
---|
302 | pxudp->pmhdl.slot = -1;
|
---|
303 |
|
---|
304 | pxudp->pcb = NULL;
|
---|
305 | pxudp->sock = INVALID_SOCKET;
|
---|
306 | pxudp->df = -1;
|
---|
307 | pxudp->ttl = -1;
|
---|
308 | pxudp->tos = -1;
|
---|
309 | pxudp->count = 0;
|
---|
310 |
|
---|
311 | pxudp->rp = pollmgr_refptr_create(&pxudp->pmhdl);
|
---|
312 | if (pxudp->rp == NULL) {
|
---|
313 | free(pxudp);
|
---|
314 | return NULL;
|
---|
315 | }
|
---|
316 |
|
---|
317 | error = sys_mbox_new(&pxudp->inmbox, 16);
|
---|
318 | if (error != ERR_OK) {
|
---|
319 | pollmgr_refptr_unref(pxudp->rp);
|
---|
320 | free(pxudp);
|
---|
321 | return NULL;
|
---|
322 | }
|
---|
323 |
|
---|
324 | #define CALLBACK_MSG(MSG, FUNC) \
|
---|
325 | do { \
|
---|
326 | pxudp->MSG.type = TCPIP_MSG_CALLBACK_STATIC; \
|
---|
327 | pxudp->MSG.sem = NULL; \
|
---|
328 | pxudp->MSG.msg.cb.function = FUNC; \
|
---|
329 | pxudp->MSG.msg.cb.ctx = (void *)pxudp; \
|
---|
330 | } while (0)
|
---|
331 |
|
---|
332 | CALLBACK_MSG(msg_delete, pxudp_pcb_delete_pxudp);
|
---|
333 | CALLBACK_MSG(msg_inbound, pxudp_pcb_write_inbound);
|
---|
334 |
|
---|
335 | return pxudp;
|
---|
336 | }
|
---|
337 |
|
---|
338 |
|
---|
339 | static void
|
---|
340 | pxudp_drain_inmbox(struct pxudp *pxudp)
|
---|
341 | {
|
---|
342 | void *ptr;
|
---|
343 |
|
---|
344 | if (!sys_mbox_valid(&pxudp->inmbox)) {
|
---|
345 | return;
|
---|
346 | }
|
---|
347 |
|
---|
348 | while (sys_mbox_tryfetch(&pxudp->inmbox, &ptr) != SYS_MBOX_EMPTY) {
|
---|
349 | struct pbuf *p = (struct pbuf *)ptr;
|
---|
350 | pbuf_free(p);
|
---|
351 | }
|
---|
352 |
|
---|
353 | sys_mbox_free(&pxudp->inmbox);
|
---|
354 | sys_mbox_set_invalid(&pxudp->inmbox);
|
---|
355 | }
|
---|
356 |
|
---|
357 |
|
---|
358 | static void
|
---|
359 | pxudp_free(struct pxudp *pxudp)
|
---|
360 | {
|
---|
361 | pxudp_drain_inmbox(pxudp);
|
---|
362 | free(pxudp);
|
---|
363 | }
|
---|
364 |
|
---|
365 |
|
---|
366 | /**
|
---|
367 | * Dissociate pxudp and its udp_pcb.
|
---|
368 | *
|
---|
369 | * Unlike its TCP cousin returns the pcb since UDP pcbs need to be
|
---|
370 | * actively deleted, so save callers the trouble of saving a copy
|
---|
371 | * before calling us.
|
---|
372 | */
|
---|
373 | static struct udp_pcb *
|
---|
374 | pxudp_pcb_dissociate(struct pxudp *pxudp)
|
---|
375 | {
|
---|
376 | struct udp_pcb *pcb;
|
---|
377 |
|
---|
378 | if (pxudp == NULL || pxudp->pcb == NULL) {
|
---|
379 | return NULL;
|
---|
380 | }
|
---|
381 |
|
---|
382 | pcb = pxudp->pcb;
|
---|
383 |
|
---|
384 | udp_recv(pxudp->pcb, NULL, NULL);
|
---|
385 | pxudp->pcb = NULL;
|
---|
386 |
|
---|
387 | return pcb;
|
---|
388 | }
|
---|
389 |
|
---|
390 |
|
---|
391 | /**
|
---|
392 | * Lwip thread callback invoked via pxudp::msg_delete
|
---|
393 | *
|
---|
394 | * Since we use static messages to communicate to the lwip thread, we
|
---|
395 | * cannot delete pxudp without making sure there are no unprocessed
|
---|
396 | * messages in the lwip thread mailbox.
|
---|
397 | *
|
---|
398 | * The easiest way to ensure that is to send this "delete" message as
|
---|
399 | * the last one and when it's processed we know there are no more and
|
---|
400 | * it's safe to delete pxudp.
|
---|
401 | *
|
---|
402 | * Channel callback should use pxudp_schedule_delete() convenience
|
---|
403 | * function defined below.
|
---|
404 | */
|
---|
405 | static void
|
---|
406 | pxudp_pcb_delete_pxudp(void *arg)
|
---|
407 | {
|
---|
408 | struct pxudp *pxudp = (struct pxudp *)arg;
|
---|
409 | struct udp_pcb *pcb;
|
---|
410 |
|
---|
411 | LWIP_ASSERT1(pxudp != NULL);
|
---|
412 |
|
---|
413 | if (pxudp->sock != INVALID_SOCKET) {
|
---|
414 | closesocket(pxudp->sock);
|
---|
415 | pxudp->sock = INVALID_SOCKET;
|
---|
416 | }
|
---|
417 |
|
---|
418 | pcb = pxudp_pcb_dissociate(pxudp);
|
---|
419 | if (pcb != NULL) {
|
---|
420 | udp_remove(pcb);
|
---|
421 | }
|
---|
422 |
|
---|
423 | pollmgr_refptr_unref(pxudp->rp);
|
---|
424 | pxudp_free(pxudp);
|
---|
425 | }
|
---|
426 |
|
---|
427 |
|
---|
428 | /**
|
---|
429 | * Poll manager callback should use this convenience wrapper to
|
---|
430 | * schedule pxudp deletion on the lwip thread and to deregister from
|
---|
431 | * the poll manager.
|
---|
432 | */
|
---|
433 | static int
|
---|
434 | pxudp_schedule_delete(struct pxudp *pxudp)
|
---|
435 | {
|
---|
436 | /*
|
---|
437 | * If pollmgr_refptr_get() is called by any channel before
|
---|
438 | * scheduled deletion happens, let them know we are gone.
|
---|
439 | */
|
---|
440 | pxudp->pmhdl.slot = -1;
|
---|
441 |
|
---|
442 | /*
|
---|
443 | * Schedule deletion. Since poll manager thread may be pre-empted
|
---|
444 | * right after we send the message, the deletion may actually
|
---|
445 | * happen on the lwip thread before we return from this function,
|
---|
446 | * so it's not safe to refer to pxudp after this call.
|
---|
447 | */
|
---|
448 | proxy_lwip_post(&pxudp->msg_delete);
|
---|
449 |
|
---|
450 | /* tell poll manager to deregister us */
|
---|
451 | return -1;
|
---|
452 | }
|
---|
453 |
|
---|
454 |
|
---|
455 | /**
|
---|
456 | * Outbound TTL/HOPL check.
|
---|
457 | */
|
---|
458 | static int
|
---|
459 | pxudp_ttl_expired(struct pbuf *p)
|
---|
460 | {
|
---|
461 | int ttl;
|
---|
462 |
|
---|
463 | if (ip_current_is_v6()) {
|
---|
464 | ttl = IP6H_HOPLIM(ip6_current_header());
|
---|
465 | }
|
---|
466 | else {
|
---|
467 | ttl = IPH_TTL(ip_current_header());
|
---|
468 | }
|
---|
469 |
|
---|
470 | if (RT_UNLIKELY(ttl <= 1)) {
|
---|
471 | int status = pbuf_header(p, ip_current_header_tot_len() + UDP_HLEN);
|
---|
472 | if (RT_LIKELY(status == 0)) {
|
---|
473 | if (ip_current_is_v6()) {
|
---|
474 | icmp6_time_exceeded(p, ICMP6_TE_HL);
|
---|
475 | }
|
---|
476 | else {
|
---|
477 | icmp_time_exceeded(p, ICMP_TE_TTL);
|
---|
478 | }
|
---|
479 | }
|
---|
480 | pbuf_free(p);
|
---|
481 | return 1;
|
---|
482 | }
|
---|
483 |
|
---|
484 | return 0;
|
---|
485 | }
|
---|
486 |
|
---|
487 |
|
---|
488 | /**
|
---|
489 | * New proxied UDP conversation created.
|
---|
490 | * Global callback for udp_proxy_accept().
|
---|
491 | */
|
---|
492 | static void
|
---|
493 | pxudp_pcb_accept(void *arg, struct udp_pcb *newpcb, struct pbuf *p,
|
---|
494 | ip_addr_t *addr, u16_t port)
|
---|
495 | {
|
---|
496 | struct pxudp *pxudp;
|
---|
497 | ipX_addr_t dst_addr;
|
---|
498 | int mapping;
|
---|
499 | int sdom;
|
---|
500 | SOCKET sock;
|
---|
501 |
|
---|
502 | LWIP_ASSERT1(newpcb != NULL);
|
---|
503 | LWIP_ASSERT1(p != NULL);
|
---|
504 | LWIP_UNUSED_ARG(arg);
|
---|
505 |
|
---|
506 | mapping = pxremap_outbound_ipX(PCB_ISIPV6(newpcb), &dst_addr, &newpcb->local_ip);
|
---|
507 | if (mapping != PXREMAP_MAPPED && pxudp_ttl_expired(p)) {
|
---|
508 | udp_remove(newpcb);
|
---|
509 | return;
|
---|
510 | }
|
---|
511 |
|
---|
512 | pxudp = pxudp_allocate();
|
---|
513 | if (pxudp == NULL) {
|
---|
514 | DPRINTF(("pxudp_allocate: failed\n"));
|
---|
515 | udp_remove(newpcb);
|
---|
516 | pbuf_free(p);
|
---|
517 | return;
|
---|
518 | }
|
---|
519 |
|
---|
520 | sdom = PCB_ISIPV6(newpcb) ? PF_INET6 : PF_INET;
|
---|
521 | pxudp->is_mapped = (mapping == PXREMAP_MAPPED);
|
---|
522 |
|
---|
523 | #if 0 /* XXX: DNS IPv6->IPv4 remapping hack */
|
---|
524 | if (pxudp->is_mapped
|
---|
525 | && newpcb->local_port == 53
|
---|
526 | && PCB_ISIPV6(newpcb))
|
---|
527 | {
|
---|
528 | /*
|
---|
529 | * "Remap" DNS over IPv6 to IPv4 since Ubuntu dnsmasq does not
|
---|
530 | * listen on IPv6.
|
---|
531 | */
|
---|
532 | sdom = PF_INET;
|
---|
533 | ipX_addr_set_loopback(0, &dst_addr);
|
---|
534 | }
|
---|
535 | #endif /* DNS IPv6->IPv4 remapping hack */
|
---|
536 |
|
---|
537 | sock = proxy_connected_socket(sdom, SOCK_DGRAM,
|
---|
538 | &dst_addr, newpcb->local_port);
|
---|
539 | if (sock == INVALID_SOCKET) {
|
---|
540 | udp_remove(newpcb);
|
---|
541 | pbuf_free(p);
|
---|
542 | return;
|
---|
543 | }
|
---|
544 |
|
---|
545 | pxudp->sock = sock;
|
---|
546 | pxudp->pcb = newpcb;
|
---|
547 | udp_recv(newpcb, pxudp_pcb_recv, pxudp);
|
---|
548 |
|
---|
549 | pxudp->pmhdl.callback = pxudp_pmgr_pump;
|
---|
550 | pxudp_chan_send(POLLMGR_CHAN_PXUDP_ADD, pxudp);
|
---|
551 |
|
---|
552 | /* dispatch directly instead of calling pxudp_pcb_recv() */
|
---|
553 | pxudp_pcb_forward_outbound(pxudp, p, addr, port);
|
---|
554 | }
|
---|
555 |
|
---|
556 |
|
---|
557 | /**
|
---|
558 | * udp_recv() callback.
|
---|
559 | */
|
---|
560 | static void
|
---|
561 | pxudp_pcb_recv(void *arg, struct udp_pcb *pcb, struct pbuf *p,
|
---|
562 | ip_addr_t *addr, u16_t port)
|
---|
563 | {
|
---|
564 | struct pxudp *pxudp = (struct pxudp *)arg;
|
---|
565 |
|
---|
566 | LWIP_ASSERT1(pxudp != NULL);
|
---|
567 | LWIP_ASSERT1(pcb == pxudp->pcb);
|
---|
568 | LWIP_UNUSED_ARG(pcb);
|
---|
569 |
|
---|
570 | if (p != NULL) {
|
---|
571 | pxudp_pcb_forward_outbound(pxudp, p, addr, port);
|
---|
572 | }
|
---|
573 | else {
|
---|
574 | pxudp_pcb_expired(pxudp);
|
---|
575 | }
|
---|
576 | }
|
---|
577 |
|
---|
578 |
|
---|
579 | static void
|
---|
580 | pxudp_pcb_forward_outbound(struct pxudp *pxudp, struct pbuf *p,
|
---|
581 | ip_addr_t *addr, u16_t port)
|
---|
582 | {
|
---|
583 | int status;
|
---|
584 |
|
---|
585 | LWIP_UNUSED_ARG(addr);
|
---|
586 | LWIP_UNUSED_ARG(port);
|
---|
587 |
|
---|
588 | if (!pxudp->is_mapped && pxudp_ttl_expired(p)) {
|
---|
589 | return;
|
---|
590 | }
|
---|
591 |
|
---|
592 | if (!ip_current_is_v6()) { /* IPv4 */
|
---|
593 | const struct ip_hdr *iph = ip_current_header();
|
---|
594 | int ttl, tos, df;
|
---|
595 |
|
---|
596 | /*
|
---|
597 | * Different OSes have different socket options for DF.
|
---|
598 | * Unlike pxping.c, we can't use IP_HDRINCL here as it's only
|
---|
599 | * valid for SOCK_RAW.
|
---|
600 | */
|
---|
601 | # define USE_DF_OPTION(_Optname) \
|
---|
602 | const int dfopt = _Optname; \
|
---|
603 | const char * const dfoptname = #_Optname; \
|
---|
604 | RT_NOREF_PV(dfoptname)
|
---|
605 | #if defined(IP_MTU_DISCOVER) /* Linux */
|
---|
606 | USE_DF_OPTION(IP_MTU_DISCOVER);
|
---|
607 | #elif defined(IP_DONTFRAG) /* Solaris 11+, FreeBSD */
|
---|
608 | USE_DF_OPTION(IP_DONTFRAG);
|
---|
609 | #elif defined(IP_DONTFRAGMENT) /* Windows */
|
---|
610 | USE_DF_OPTION(IP_DONTFRAGMENT);
|
---|
611 | #else
|
---|
612 | USE_DF_OPTION(0);
|
---|
613 | #endif
|
---|
614 |
|
---|
615 | ttl = IPH_TTL(iph);
|
---|
616 | if (!pxudp->is_mapped) {
|
---|
617 | LWIP_ASSERT1(ttl > 1);
|
---|
618 | --ttl;
|
---|
619 | }
|
---|
620 |
|
---|
621 | if (ttl != pxudp->ttl) {
|
---|
622 | status = setsockopt(pxudp->sock, IPPROTO_IP, IP_TTL,
|
---|
623 | (char *)&ttl, sizeof(ttl));
|
---|
624 | if (RT_LIKELY(status == 0)) {
|
---|
625 | pxudp->ttl = ttl;
|
---|
626 | }
|
---|
627 | else {
|
---|
628 | DPRINTF(("IP_TTL: %R[sockerr]\n", SOCKERRNO()));
|
---|
629 | }
|
---|
630 | }
|
---|
631 |
|
---|
632 | tos = IPH_TOS(iph);
|
---|
633 | if (tos != pxudp->tos) {
|
---|
634 | status = setsockopt(pxudp->sock, IPPROTO_IP, IP_TOS,
|
---|
635 | (char *)&tos, sizeof(tos));
|
---|
636 | if (RT_LIKELY(status == 0)) {
|
---|
637 | pxudp->tos = tos;
|
---|
638 | }
|
---|
639 | else {
|
---|
640 | DPRINTF(("IP_TOS: %R[sockerr]\n", SOCKERRNO()));
|
---|
641 | }
|
---|
642 | }
|
---|
643 |
|
---|
644 | if (dfopt) {
|
---|
645 | df = (IPH_OFFSET(iph) & PP_HTONS(IP_DF)) != 0;
|
---|
646 | #if defined(IP_MTU_DISCOVER)
|
---|
647 | df = df ? IP_PMTUDISC_DO : IP_PMTUDISC_DONT;
|
---|
648 | #endif
|
---|
649 | if (df != pxudp->df) {
|
---|
650 | status = setsockopt(pxudp->sock, IPPROTO_IP, dfopt,
|
---|
651 | (char *)&df, sizeof(df));
|
---|
652 | if (RT_LIKELY(status == 0)) {
|
---|
653 | pxudp->df = df;
|
---|
654 | }
|
---|
655 | else {
|
---|
656 | DPRINTF(("%s: %R[sockerr]\n", dfoptname, SOCKERRNO()));
|
---|
657 | }
|
---|
658 | }
|
---|
659 | }
|
---|
660 | }
|
---|
661 | else { /* IPv6 */
|
---|
662 | const struct ip6_hdr *iph = ip6_current_header();
|
---|
663 | int ttl;
|
---|
664 |
|
---|
665 | ttl = IP6H_HOPLIM(iph);
|
---|
666 | if (!pxudp->is_mapped) {
|
---|
667 | LWIP_ASSERT1(ttl > 1);
|
---|
668 | --ttl;
|
---|
669 | }
|
---|
670 |
|
---|
671 | if (ttl != pxudp->ttl) {
|
---|
672 | status = setsockopt(pxudp->sock, IPPROTO_IPV6, IPV6_UNICAST_HOPS,
|
---|
673 | (char *)&ttl, sizeof(ttl));
|
---|
674 | if (RT_LIKELY(status == 0)) {
|
---|
675 | pxudp->ttl = ttl;
|
---|
676 | }
|
---|
677 | else {
|
---|
678 | DPRINTF(("IPV6_UNICAST_HOPS: %R[sockerr]\n", SOCKERRNO()));
|
---|
679 | }
|
---|
680 | }
|
---|
681 | }
|
---|
682 |
|
---|
683 | if (pxudp->pcb->local_port == 53) {
|
---|
684 | ++pxudp->count;
|
---|
685 | }
|
---|
686 |
|
---|
687 | proxy_sendto(pxudp->sock, p, NULL, 0);
|
---|
688 | pbuf_free(p);
|
---|
689 | }
|
---|
690 |
|
---|
691 |
|
---|
692 | /**
|
---|
693 | * Proxy udp_pcbs are expired by timer, which is signaled by passing
|
---|
694 | * NULL pbuf to the udp_recv() callback. At that point the pcb is
|
---|
695 | * removed from the list of proxy udp pcbs so no new datagrams will be
|
---|
696 | * delivered.
|
---|
697 | */
|
---|
698 | static void
|
---|
699 | pxudp_pcb_expired(struct pxudp *pxudp)
|
---|
700 | {
|
---|
701 | struct udp_pcb *pcb;
|
---|
702 |
|
---|
703 | DPRINTF2(("%s: pxudp %p, pcb %p, sock %d: expired\n",
|
---|
704 | __func__, (void *)pxudp, (void *)pxudp->pcb, pxudp->sock));
|
---|
705 |
|
---|
706 | pcb = pxudp_pcb_dissociate(pxudp);
|
---|
707 | if (pcb != NULL) {
|
---|
708 | udp_remove(pcb);
|
---|
709 | }
|
---|
710 |
|
---|
711 | pxudp_chan_send_weak(POLLMGR_CHAN_PXUDP_DEL, pxudp);
|
---|
712 | }
|
---|
713 |
|
---|
714 |
|
---|
715 | /**
|
---|
716 | */
|
---|
717 | static int
|
---|
718 | pxudp_pmgr_pump(struct pollmgr_handler *handler, SOCKET fd, int revents)
|
---|
719 | {
|
---|
720 | struct pxudp *pxudp;
|
---|
721 | struct pbuf *p;
|
---|
722 | ssize_t nread;
|
---|
723 | err_t error;
|
---|
724 |
|
---|
725 | pxudp = (struct pxudp *)handler->data;
|
---|
726 | LWIP_ASSERT1(handler == &pxudp->pmhdl);
|
---|
727 | LWIP_ASSERT1(fd == pxudp->sock);
|
---|
728 | LWIP_UNUSED_ARG(fd);
|
---|
729 |
|
---|
730 |
|
---|
731 | if (revents & ~(POLLIN|POLLERR)) {
|
---|
732 | DPRINTF(("%s: unexpected revents 0x%x\n", __func__, revents));
|
---|
733 | return pxudp_schedule_delete(pxudp);
|
---|
734 | }
|
---|
735 |
|
---|
736 | /*
|
---|
737 | * XXX: AFAICS, there's no way to match the error with the
|
---|
738 | * outgoing datagram that triggered it, since we do non-blocking
|
---|
739 | * sends from lwip thread.
|
---|
740 | */
|
---|
741 | if (revents & POLLERR) {
|
---|
742 | int sockerr = -1;
|
---|
743 | socklen_t optlen = (socklen_t)sizeof(sockerr);
|
---|
744 | int status;
|
---|
745 |
|
---|
746 | status = getsockopt(pxudp->sock, SOL_SOCKET,
|
---|
747 | SO_ERROR, (char *)&sockerr, &optlen);
|
---|
748 | if (status < 0) {
|
---|
749 | DPRINTF(("%s: sock %d: SO_ERROR failed:%R[sockerr]\n",
|
---|
750 | __func__, pxudp->sock, SOCKERRNO()));
|
---|
751 | }
|
---|
752 | else {
|
---|
753 | DPRINTF(("%s: sock %d: %R[sockerr]\n",
|
---|
754 | __func__, pxudp->sock, sockerr));
|
---|
755 | }
|
---|
756 | }
|
---|
757 |
|
---|
758 | if ((revents & POLLIN) == 0) {
|
---|
759 | return POLLIN;
|
---|
760 | }
|
---|
761 |
|
---|
762 | #ifdef RT_OS_WINDOWS
|
---|
763 | nread = recv(pxudp->sock, (char *)pollmgr_udpbuf, sizeof(pollmgr_udpbuf), 0);
|
---|
764 | #else
|
---|
765 | nread = recv(pxudp->sock, pollmgr_udpbuf, sizeof(pollmgr_udpbuf), 0);
|
---|
766 | #endif
|
---|
767 | if (nread == SOCKET_ERROR) {
|
---|
768 | DPRINTF(("%s: %R[sockerr]\n", __func__, SOCKERRNO()));
|
---|
769 | return POLLIN;
|
---|
770 | }
|
---|
771 |
|
---|
772 | p = pbuf_alloc(PBUF_RAW, (u16_t)nread, PBUF_RAM);
|
---|
773 | if (p == NULL) {
|
---|
774 | DPRINTF(("%s: pbuf_alloc(%d) failed\n", __func__, (int)nread));
|
---|
775 | return POLLIN;
|
---|
776 | }
|
---|
777 |
|
---|
778 | error = pbuf_take(p, pollmgr_udpbuf, (u16_t)nread);
|
---|
779 | if (error != ERR_OK) {
|
---|
780 | DPRINTF(("%s: pbuf_take(%d) failed\n", __func__, (int)nread));
|
---|
781 | pbuf_free(p);
|
---|
782 | return POLLIN;
|
---|
783 | }
|
---|
784 |
|
---|
785 | error = sys_mbox_trypost(&pxudp->inmbox, p);
|
---|
786 | if (error != ERR_OK) {
|
---|
787 | pbuf_free(p);
|
---|
788 | return POLLIN;
|
---|
789 | }
|
---|
790 |
|
---|
791 | proxy_lwip_post(&pxudp->msg_inbound);
|
---|
792 |
|
---|
793 | return POLLIN;
|
---|
794 | }
|
---|
795 |
|
---|
796 |
|
---|
797 | /**
|
---|
798 | * Callback from poll manager to trigger sending to guest.
|
---|
799 | */
|
---|
800 | static void
|
---|
801 | pxudp_pcb_write_inbound(void *ctx)
|
---|
802 | {
|
---|
803 | struct pxudp *pxudp = (struct pxudp *)ctx;
|
---|
804 | LWIP_ASSERT1(pxudp != NULL);
|
---|
805 |
|
---|
806 | if (pxudp->pcb == NULL) {
|
---|
807 | return;
|
---|
808 | }
|
---|
809 |
|
---|
810 | pxudp_pcb_forward_inbound(pxudp);
|
---|
811 | }
|
---|
812 |
|
---|
813 |
|
---|
814 | static void
|
---|
815 | pxudp_pcb_forward_inbound(struct pxudp *pxudp)
|
---|
816 | {
|
---|
817 | struct pbuf *p;
|
---|
818 | u32_t timo;
|
---|
819 | err_t error;
|
---|
820 |
|
---|
821 | if (!sys_mbox_valid(&pxudp->inmbox)) {
|
---|
822 | return;
|
---|
823 | }
|
---|
824 |
|
---|
825 | timo = sys_mbox_tryfetch(&pxudp->inmbox, (void **)&p);
|
---|
826 | if (timo == SYS_MBOX_EMPTY) {
|
---|
827 | return;
|
---|
828 | }
|
---|
829 |
|
---|
830 | error = udp_send(pxudp->pcb, p);
|
---|
831 | if (error != ERR_OK) {
|
---|
832 | DPRINTF(("%s: udp_send(pcb %p) err %d\n",
|
---|
833 | __func__, (void *)pxudp, error));
|
---|
834 | }
|
---|
835 |
|
---|
836 | pbuf_free(p);
|
---|
837 |
|
---|
838 | /*
|
---|
839 | * If we enabled counting in pxudp_pcb_forward_outbound() check
|
---|
840 | * that we have (all) the reply(s).
|
---|
841 | */
|
---|
842 | if (pxudp->count > 0) {
|
---|
843 | --pxudp->count;
|
---|
844 | if (pxudp->count == 0) {
|
---|
845 | pxudp_pcb_expired(pxudp);
|
---|
846 | }
|
---|
847 | }
|
---|
848 | }
|
---|