source: heimdal/trunk/lib/roken/socket_wrapper.c

Last change on this file was 1, checked in by Paul Smedley, 10 years ago

Initial commit of Heimdal 1.5.3
File size: 41.0 KB
Line 
1/*
2 * Copyright (C) Jelmer Vernooij 2005 <jelmer@samba.org>
3 * Copyright (C) Stefan Metzmacher 2006 <metze@samba.org>
4 *
5 * All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 *
11 * 1. Redistributions of source code must retain the above copyright
12 * notice, this list of conditions and the following disclaimer.
13 *
14 * 2. Redistributions in binary form must reproduce the above copyright
15 * notice, this list of conditions and the following disclaimer in the
16 * documentation and/or other materials provided with the distribution.
17 *
18 * 3. Neither the name of the author nor the names of its contributors
19 * may be used to endorse or promote products derived from this software
20 * without specific prior written permission.
21 *
22 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
23 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
24 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
25 * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
26 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
27 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
28 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
29 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
30 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
31 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
32 * SUCH DAMAGE.
33 *
34 */
35
36/*
37 Socket wrapper library. Passes all socket communication over
38 unix domain sockets if the environment variable SOCKET_WRAPPER_DIR
39 is set.
40*/
41
42#define SOCKET_WRAPPER_NOT_REPLACE
43
44#ifdef _SAMBA_BUILD_
45
46#include "includes.h"
47#include "system/network.h"
48#include "system/filesys.h"
49
50#ifdef malloc
51#undef malloc
52#endif
53#ifdef calloc
54#undef calloc
55#endif
56#ifdef strdup
57#undef strdup
58#endif
59
60#else /* _SAMBA_BUILD_ */
61
62#include <config.h>
63#undef SOCKET_WRAPPER_REPLACE
64
65#include <sys/types.h>
66#ifdef TIME_WITH_SYS_TIME
67#include <sys/time.h>
68#include <time.h>
69#elif defined(HAVE_SYS_TIME_H)
70#include <sys/time.h>
71#else
72#include <time.h>
73#endif
74#include <sys/stat.h>
75#include <sys/socket.h>
76#include <sys/ioctl.h>
77#ifdef HAVE_SYS_FILIO_H
78#include <sys/filio.h>
79#endif
80#include <errno.h>
81#include <sys/un.h>
82#include <netinet/in.h>
83#include <netinet/tcp.h>
84#include <fcntl.h>
85#include <stdlib.h>
86#include <unistd.h>
87#include <string.h>
88#include <stdio.h>
89#include "roken.h"
90
91#include "socket_wrapper.h"
92
93#define HAVE_GETTIMEOFDAY_TZ 1
94
95#define _PUBLIC_
96
97#endif
98
99#define SWRAP_DLIST_ADD(list,item) do { \
100 if (!(list)) { \
101 (item)->prev = NULL; \
102 (item)->next = NULL; \
103 (list) = (item); \
104 } else { \
105 (item)->prev = NULL; \
106 (item)->next = (list); \
107 (list)->prev = (item); \
108 (list) = (item); \
109 } \
110} while (0)
111
112#define SWRAP_DLIST_REMOVE(list,item) do { \
113 if ((list) == (item)) { \
114 (list) = (item)->next; \
115 if (list) { \
116 (list)->prev = NULL; \
117 } \
118 } else { \
119 if ((item)->prev) { \
120 (item)->prev->next = (item)->next; \
121 } \
122 if ((item)->next) { \
123 (item)->next->prev = (item)->prev; \
124 } \
125 } \
126 (item)->prev = NULL; \
127 (item)->next = NULL; \
128} while (0)
129
130/* LD_PRELOAD doesn't work yet, so REWRITE_CALLS is all we support
131 * for now */
132#define REWRITE_CALLS
133
134#ifdef REWRITE_CALLS
135#define real_accept accept
136#define real_connect connect
137#define real_bind bind
138#define real_listen listen
139#define real_getpeername getpeername
140#define real_getsockname getsockname
141#define real_getsockopt getsockopt
142#define real_setsockopt setsockopt
143#define real_recvfrom recvfrom
144#define real_sendto sendto
145#define real_ioctl ioctl
146#define real_recv recv
147#define real_send send
148#define real_socket socket
149#define real_close close
150#define real_dup dup
151#define real_dup2 dup2
152#endif
153
154#ifdef HAVE_GETTIMEOFDAY_TZ
155#define swrapGetTimeOfDay(tval) gettimeofday(tval,NULL)
156#else
157#define swrapGetTimeOfDay(tval) gettimeofday(tval)
158#endif
159
160/* we need to use a very terse format here as IRIX 6.4 silently
161 truncates names to 16 chars, so if we use a longer name then we
162 can't tell which port a packet came from with recvfrom()
163
164 with this format we have 8 chars left for the directory name
165*/
166#define SOCKET_FORMAT "%c%02X%04X"
167#define SOCKET_TYPE_CHAR_TCP 'T'
168#define SOCKET_TYPE_CHAR_UDP 'U'
169#define SOCKET_TYPE_CHAR_TCP_V6 'X'
170#define SOCKET_TYPE_CHAR_UDP_V6 'Y'
171
172#define MAX_WRAPPED_INTERFACES 16
173
174#define SW_IPV6_ADDRESS 1
175
176static struct sockaddr *sockaddr_dup(const void *data, socklen_t len)
177{
178 struct sockaddr *ret = (struct sockaddr *)malloc(len);
179 memcpy(ret, data, len);
180 return ret;
181}
182
183static void set_port(int family, int prt, struct sockaddr *addr)
184{
185 switch (family) {
186 case AF_INET:
187 ((struct sockaddr_in *)addr)->sin_port = htons(prt);
188 break;
189#ifdef HAVE_IPV6
190 case AF_INET6:
191 ((struct sockaddr_in6 *)addr)->sin6_port = htons(prt);
192 break;
193#endif
194 }
195}
196
197static int socket_length(int family)
198{
199 switch (family) {
200 case AF_INET:
201 return sizeof(struct sockaddr_in);
202#ifdef HAVE_IPV6
203 case AF_INET6:
204 return sizeof(struct sockaddr_in6);
205#endif
206 }
207 return -1;
208}
209
210
211
212struct socket_info
213{
214 int fd;
215
216 int family;
217 int type;
218 int protocol;
219 int bound;
220 int bcast;
221 int is_server;
222
223 char *path;
224 char *tmp_path;
225
226 struct sockaddr *myname;
227 socklen_t myname_len;
228
229 struct sockaddr *peername;
230 socklen_t peername_len;
231
232 struct {
233 unsigned long pck_snd;
234 unsigned long pck_rcv;
235 } io;
236
237 struct socket_info *prev, *next;
238};
239
240static struct socket_info *sockets;
241
242
243static const char *socket_wrapper_dir(void)
244{
245 const char *s = getenv("SOCKET_WRAPPER_DIR");
246 if (s == NULL) {
247 return NULL;
248 }
249 if (strncmp(s, "./", 2) == 0) {
250 s += 2;
251 }
252 return s;
253}
254
255static unsigned int socket_wrapper_default_iface(void)
256{
257 const char *s = getenv("SOCKET_WRAPPER_DEFAULT_IFACE");
258 if (s) {
259 unsigned int iface;
260 if (sscanf(s, "%u", &iface) == 1) {
261 if (iface >= 1 && iface <= MAX_WRAPPED_INTERFACES) {
262 return iface;
263 }
264 }
265 }
266
267 return 1;/* 127.0.0.1 */
268}
269
270static int convert_un_in(const struct sockaddr_un *un, struct sockaddr *in, socklen_t *len)
271{
272 unsigned int iface;
273 unsigned int prt;
274 const char *p;
275 char type;
276
277 p = strrchr(un->sun_path, '/');
278 if (p) p++; else p = un->sun_path;
279
280 if (sscanf(p, SOCKET_FORMAT, &type, &iface, &prt) != 3) {
281 errno = EINVAL;
282 return -1;
283 }
284
285 if (iface == 0 || iface > MAX_WRAPPED_INTERFACES) {
286 errno = EINVAL;
287 return -1;
288 }
289
290 if (prt > 0xFFFF) {
291 errno = EINVAL;
292 return -1;
293 }
294
295 switch(type) {
296 case SOCKET_TYPE_CHAR_TCP:
297 case SOCKET_TYPE_CHAR_UDP: {
298 struct sockaddr_in *in2 = (struct sockaddr_in *)in;
299
300 if ((*len) < sizeof(*in2)) {
301 errno = EINVAL;
302 return -1;
303 }
304
305 memset(in2, 0, sizeof(*in2));
306 in2->sin_family = AF_INET;
307 in2->sin_addr.s_addr = htonl((127<<24) | iface);
308 in2->sin_port = htons(prt);
309
310 *len = sizeof(*in2);
311 break;
312 }
313#ifdef HAVE_IPV6
314 case SOCKET_TYPE_CHAR_TCP_V6:
315 case SOCKET_TYPE_CHAR_UDP_V6: {
316 struct sockaddr_in6 *in2 = (struct sockaddr_in6 *)in;
317
318 if ((*len) < sizeof(*in2)) {
319 errno = EINVAL;
320 return -1;
321 }
322
323 memset(in2, 0, sizeof(*in2));
324 in2->sin6_family = AF_INET6;
325 in2->sin6_addr.s6_addr[0] = SW_IPV6_ADDRESS;
326 in2->sin6_port = htons(prt);
327
328 *len = sizeof(*in2);
329 break;
330 }
331#endif
332 default:
333 errno = EINVAL;
334 return -1;
335 }
336
337 return 0;
338}
339
340static int convert_in_un_remote(struct socket_info *si, const struct sockaddr *inaddr, struct sockaddr_un *un,
341 int *bcast)
342{
343 char type = '\0';
344 unsigned int prt;
345 unsigned int iface;
346 int is_bcast = 0;
347
348 if (bcast) *bcast = 0;
349
350 switch (si->family) {
351 case AF_INET: {
352 const struct sockaddr_in *in =
353 (const struct sockaddr_in *)inaddr;
354 unsigned int addr = ntohl(in->sin_addr.s_addr);
355 char u_type = '\0';
356 char b_type = '\0';
357 char a_type = '\0';
358
359 switch (si->type) {
360 case SOCK_STREAM:
361 u_type = SOCKET_TYPE_CHAR_TCP;
362 break;
363 case SOCK_DGRAM:
364 u_type = SOCKET_TYPE_CHAR_UDP;
365 a_type = SOCKET_TYPE_CHAR_UDP;
366 b_type = SOCKET_TYPE_CHAR_UDP;
367 break;
368 }
369
370 prt = ntohs(in->sin_port);
371 if (a_type && addr == 0xFFFFFFFF) {
372 /* 255.255.255.255 only udp */
373 is_bcast = 2;
374 type = a_type;
375 iface = socket_wrapper_default_iface();
376 } else if (b_type && addr == 0x7FFFFFFF) {
377 /* 127.255.255.255 only udp */
378 is_bcast = 1;
379 type = b_type;
380 iface = socket_wrapper_default_iface();
381 } else if ((addr & 0xFFFFFF00) == 0x7F000000) {
382 /* 127.0.0.X */
383 is_bcast = 0;
384 type = u_type;
385 iface = (addr & 0x000000FF);
386 } else {
387 errno = ENETUNREACH;
388 return -1;
389 }
390 if (bcast) *bcast = is_bcast;
391 break;
392 }
393#ifdef HAVE_IPV6
394 case AF_INET6: {
395 const struct sockaddr_in6 *in =
396 (const struct sockaddr_in6 *)inaddr;
397
398 switch (si->type) {
399 case SOCK_STREAM:
400 type = SOCKET_TYPE_CHAR_TCP_V6;
401 break;
402 case SOCK_DGRAM:
403 type = SOCKET_TYPE_CHAR_UDP_V6;
404 break;
405 }
406
407 /* XXX no multicast/broadcast */
408
409 prt = ntohs(in->sin6_port);
410 iface = SW_IPV6_ADDRESS;
411
412 break;
413 }
414#endif
415 default:
416 errno = ENETUNREACH;
417 return -1;
418 }
419
420 if (prt == 0) {
421 errno = EINVAL;
422 return -1;
423 }
424
425 if (is_bcast) {
426 snprintf(un->sun_path, sizeof(un->sun_path), "%s/EINVAL",
427 socket_wrapper_dir());
428 /* the caller need to do more processing */
429 return 0;
430 }
431
432 snprintf(un->sun_path, sizeof(un->sun_path), "%s/"SOCKET_FORMAT,
433 socket_wrapper_dir(), type, iface, prt);
434
435 return 0;
436}
437
438static int convert_in_un_alloc(struct socket_info *si, const struct sockaddr *inaddr, struct sockaddr_un *un,
439 int *bcast)
440{
441 char type = '\0';
442 unsigned int prt;
443 unsigned int iface;
444 struct stat st;
445 int is_bcast = 0;
446
447 if (bcast) *bcast = 0;
448
449 switch (si->family) {
450 case AF_INET: {
451 const struct sockaddr_in *in =
452 (const struct sockaddr_in *)inaddr;
453 unsigned int addr = ntohl(in->sin_addr.s_addr);
454 char u_type = '\0';
455 char d_type = '\0';
456 char b_type = '\0';
457 char a_type = '\0';
458
459 prt = ntohs(in->sin_port);
460
461 switch (si->type) {
462 case SOCK_STREAM:
463 u_type = SOCKET_TYPE_CHAR_TCP;
464 d_type = SOCKET_TYPE_CHAR_TCP;
465 break;
466 case SOCK_DGRAM:
467 u_type = SOCKET_TYPE_CHAR_UDP;
468 d_type = SOCKET_TYPE_CHAR_UDP;
469 a_type = SOCKET_TYPE_CHAR_UDP;
470 b_type = SOCKET_TYPE_CHAR_UDP;
471 break;
472 }
473
474 if (addr == 0) {
475 /* 0.0.0.0 */
476 is_bcast = 0;
477 type = d_type;
478 iface = socket_wrapper_default_iface();
479 } else if (a_type && addr == 0xFFFFFFFF) {
480 /* 255.255.255.255 only udp */
481 is_bcast = 2;
482 type = a_type;
483 iface = socket_wrapper_default_iface();
484 } else if (b_type && addr == 0x7FFFFFFF) {
485 /* 127.255.255.255 only udp */
486 is_bcast = 1;
487 type = b_type;
488 iface = socket_wrapper_default_iface();
489 } else if ((addr & 0xFFFFFF00) == 0x7F000000) {
490 /* 127.0.0.X */
491 is_bcast = 0;
492 type = u_type;
493 iface = (addr & 0x000000FF);
494 } else {
495 errno = EADDRNOTAVAIL;
496 return -1;
497 }
498 break;
499 }
500#ifdef HAVE_IPV6
501 case AF_INET6: {
502 const struct sockaddr_in6 *in =
503 (const struct sockaddr_in6 *)inaddr;
504
505 switch (si->type) {
506 case SOCK_STREAM:
507 type = SOCKET_TYPE_CHAR_TCP_V6;
508 break;
509 case SOCK_DGRAM:
510 type = SOCKET_TYPE_CHAR_UDP_V6;
511 break;
512 }
513
514 /* XXX no multicast/broadcast */
515
516 prt = ntohs(in->sin6_port);
517 iface = SW_IPV6_ADDRESS;
518
519 break;
520 }
521#endif
522 default:
523 errno = ENETUNREACH;
524 return -1;
525 }
526
527
528 if (bcast) *bcast = is_bcast;
529
530 if (prt == 0) {
531 /* handle auto-allocation of ephemeral ports */
532 for (prt = 5001; prt < 10000; prt++) {
533 snprintf(un->sun_path, sizeof(un->sun_path), "%s/"SOCKET_FORMAT,
534 socket_wrapper_dir(), type, iface, prt);
535 if (stat(un->sun_path, &st) == 0) continue;
536
537 set_port(si->family, prt, si->myname);
538 }
539 }
540
541 snprintf(un->sun_path, sizeof(un->sun_path), "%s/"SOCKET_FORMAT,
542 socket_wrapper_dir(), type, iface, prt);
543 return 0;
544}
545
546static struct socket_info *find_socket_info(int fd)
547{
548 struct socket_info *i;
549 for (i = sockets; i; i = i->next) {
550 if (i->fd == fd)
551 return i;
552 }
553
554 return NULL;
555}
556
557static int sockaddr_convert_to_un(struct socket_info *si, const struct sockaddr *in_addr, socklen_t in_len,
558 struct sockaddr_un *out_addr, int alloc_sock, int *bcast)
559{
560 if (!out_addr)
561 return 0;
562
563 out_addr->sun_family = AF_UNIX;
564
565 switch (in_addr->sa_family) {
566 case AF_INET:
567#ifdef HAVE_IPV6
568 case AF_INET6:
569#endif
570 switch (si->type) {
571 case SOCK_STREAM:
572 case SOCK_DGRAM:
573 break;
574 default:
575 errno = ESOCKTNOSUPPORT;
576 return -1;
577 }
578 if (alloc_sock) {
579 return convert_in_un_alloc(si, in_addr, out_addr, bcast);
580 } else {
581 return convert_in_un_remote(si, in_addr, out_addr, bcast);
582 }
583 default:
584 break;
585 }
586
587 errno = EAFNOSUPPORT;
588 return -1;
589}
590
591static int sockaddr_convert_from_un(const struct socket_info *si,
592 const struct sockaddr_un *in_addr,
593 socklen_t un_addrlen,
594 int family,
595 struct sockaddr *out_addr,
596 socklen_t *out_addrlen)
597{
598 if (out_addr == NULL || out_addrlen == NULL)
599 return 0;
600
601 if (un_addrlen == 0) {
602 *out_addrlen = 0;
603 return 0;
604 }
605
606 switch (family) {
607 case AF_INET:
608#ifdef HAVE_IPV6
609 case AF_INET6:
610#endif
611 switch (si->type) {
612 case SOCK_STREAM:
613 case SOCK_DGRAM:
614 break;
615 default:
616 errno = ESOCKTNOSUPPORT;
617 return -1;
618 }
619 return convert_un_in(in_addr, out_addr, out_addrlen);
620 default:
621 break;
622 }
623
624 errno = EAFNOSUPPORT;
625 return -1;
626}
627
628enum swrap_packet_type {
629 SWRAP_CONNECT_SEND,
630 SWRAP_CONNECT_UNREACH,
631 SWRAP_CONNECT_RECV,
632 SWRAP_CONNECT_ACK,
633 SWRAP_ACCEPT_SEND,
634 SWRAP_ACCEPT_RECV,
635 SWRAP_ACCEPT_ACK,
636 SWRAP_RECVFROM,
637 SWRAP_SENDTO,
638 SWRAP_SENDTO_UNREACH,
639 SWRAP_PENDING_RST,
640 SWRAP_RECV,
641 SWRAP_RECV_RST,
642 SWRAP_SEND,
643 SWRAP_SEND_RST,
644 SWRAP_CLOSE_SEND,
645 SWRAP_CLOSE_RECV,
646 SWRAP_CLOSE_ACK
647};
648
649struct swrap_file_hdr {
650 unsigned long magic;
651 unsigned short version_major;
652 unsigned short version_minor;
653 long timezone;
654 unsigned long sigfigs;
655 unsigned long frame_max_len;
656#define SWRAP_FRAME_LENGTH_MAX 0xFFFF
657 unsigned long link_type;
658};
659#define SWRAP_FILE_HDR_SIZE 24
660
661struct swrap_packet {
662 struct {
663 unsigned long seconds;
664 unsigned long micro_seconds;
665 unsigned long recorded_length;
666 unsigned long full_length;
667 } frame;
668#define SWRAP_PACKET__FRAME_SIZE 16
669
670 struct {
671 struct {
672 unsigned char ver_hdrlen;
673 unsigned char tos;
674 unsigned short packet_length;
675 unsigned short identification;
676 unsigned char flags;
677 unsigned char fragment;
678 unsigned char ttl;
679 unsigned char protocol;
680 unsigned short hdr_checksum;
681 unsigned long src_addr;
682 unsigned long dest_addr;
683 } hdr;
684#define SWRAP_PACKET__IP_HDR_SIZE 20
685
686 union {
687 struct {
688 unsigned short source_port;
689 unsigned short dest_port;
690 unsigned long seq_num;
691 unsigned long ack_num;
692 unsigned char hdr_length;
693 unsigned char control;
694 unsigned short window;
695 unsigned short checksum;
696 unsigned short urg;
697 } tcp;
698#define SWRAP_PACKET__IP_P_TCP_SIZE 20
699 struct {
700 unsigned short source_port;
701 unsigned short dest_port;
702 unsigned short length;
703 unsigned short checksum;
704 } udp;
705#define SWRAP_PACKET__IP_P_UDP_SIZE 8
706 struct {
707 unsigned char type;
708 unsigned char code;
709 unsigned short checksum;
710 unsigned long unused;
711 } icmp;
712#define SWRAP_PACKET__IP_P_ICMP_SIZE 8
713 } p;
714 } ip;
715};
716#define SWRAP_PACKET_SIZE 56
717
718static const char *socket_wrapper_pcap_file(void)
719{
720 static int initialized = 0;
721 static const char *s = NULL;
722 static const struct swrap_file_hdr h;
723 static const struct swrap_packet p;
724
725 if (initialized == 1) {
726 return s;
727 }
728 initialized = 1;
729
730 /*
731 * TODO: don't use the structs use plain buffer offsets
732 * and PUSH_U8(), PUSH_U16() and PUSH_U32()
733 *
734 * for now make sure we disable PCAP support
735 * if the struct has alignment!
736 */
737 if (sizeof(h) != SWRAP_FILE_HDR_SIZE) {
738 return NULL;
739 }
740 if (sizeof(p) != SWRAP_PACKET_SIZE) {
741 return NULL;
742 }
743 if (sizeof(p.frame) != SWRAP_PACKET__FRAME_SIZE) {
744 return NULL;
745 }
746 if (sizeof(p.ip.hdr) != SWRAP_PACKET__IP_HDR_SIZE) {
747 return NULL;
748 }
749 if (sizeof(p.ip.p.tcp) != SWRAP_PACKET__IP_P_TCP_SIZE) {
750 return NULL;
751 }
752 if (sizeof(p.ip.p.udp) != SWRAP_PACKET__IP_P_UDP_SIZE) {
753 return NULL;
754 }
755 if (sizeof(p.ip.p.icmp) != SWRAP_PACKET__IP_P_ICMP_SIZE) {
756 return NULL;
757 }
758
759 s = getenv("SOCKET_WRAPPER_PCAP_FILE");
760 if (s == NULL) {
761 return NULL;
762 }
763 if (strncmp(s, "./", 2) == 0) {
764 s += 2;
765 }
766 return s;
767}
768
769static struct swrap_packet *swrap_packet_init(struct timeval *tval,
770 const struct sockaddr_in *src_addr,
771 const struct sockaddr_in *dest_addr,
772 int socket_type,
773 const unsigned char *payload,
774 size_t payload_len,
775 unsigned long tcp_seq,
776 unsigned long tcp_ack,
777 unsigned char tcp_ctl,
778 int unreachable,
779 size_t *_packet_len)
780{
781 struct swrap_packet *ret;
782 struct swrap_packet *packet;
783 size_t packet_len;
784 size_t alloc_len;
785 size_t nonwire_len = sizeof(packet->frame);
786 size_t wire_hdr_len = 0;
787 size_t wire_len = 0;
788 size_t icmp_hdr_len = 0;
789 size_t icmp_truncate_len = 0;
790 unsigned char protocol = 0, icmp_protocol = 0;
791 unsigned short src_port = src_addr->sin_port;
792 unsigned short dest_port = dest_addr->sin_port;
793
794 switch (socket_type) {
795 case SOCK_STREAM:
796 protocol = 0x06; /* TCP */
797 wire_hdr_len = sizeof(packet->ip.hdr) + sizeof(packet->ip.p.tcp);
798 wire_len = wire_hdr_len + payload_len;
799 break;
800
801 case SOCK_DGRAM:
802 protocol = 0x11; /* UDP */
803 wire_hdr_len = sizeof(packet->ip.hdr) + sizeof(packet->ip.p.udp);
804 wire_len = wire_hdr_len + payload_len;
805 break;
806 }
807
808 if (unreachable) {
809 icmp_protocol = protocol;
810 protocol = 0x01; /* ICMP */
811 if (wire_len > 64 ) {
812 icmp_truncate_len = wire_len - 64;
813 }
814 icmp_hdr_len = sizeof(packet->ip.hdr) + sizeof(packet->ip.p.icmp);
815 wire_hdr_len += icmp_hdr_len;
816 wire_len += icmp_hdr_len;
817 }
818
819 packet_len = nonwire_len + wire_len;
820 alloc_len = packet_len;
821 if (alloc_len < sizeof(struct swrap_packet)) {
822 alloc_len = sizeof(struct swrap_packet);
823 }
824 ret = (struct swrap_packet *)malloc(alloc_len);
825 if (!ret) return NULL;
826
827 packet = ret;
828
829 packet->frame.seconds = tval->tv_sec;
830 packet->frame.micro_seconds = tval->tv_usec;
831 packet->frame.recorded_length = wire_len - icmp_truncate_len;
832 packet->frame.full_length = wire_len - icmp_truncate_len;
833
834 packet->ip.hdr.ver_hdrlen = 0x45; /* version 4 and 5 * 32 bit words */
835 packet->ip.hdr.tos = 0x00;
836 packet->ip.hdr.packet_length = htons(wire_len - icmp_truncate_len);
837 packet->ip.hdr.identification = htons(0xFFFF);
838 packet->ip.hdr.flags = 0x40; /* BIT 1 set - means don't fraqment */
839 packet->ip.hdr.fragment = htons(0x0000);
840 packet->ip.hdr.ttl = 0xFF;
841 packet->ip.hdr.protocol = protocol;
842 packet->ip.hdr.hdr_checksum = htons(0x0000);
843 packet->ip.hdr.src_addr = src_addr->sin_addr.s_addr;
844 packet->ip.hdr.dest_addr = dest_addr->sin_addr.s_addr;
845
846 if (unreachable) {
847 packet->ip.p.icmp.type = 0x03; /* destination unreachable */
848 packet->ip.p.icmp.code = 0x01; /* host unreachable */
849 packet->ip.p.icmp.checksum = htons(0x0000);
850 packet->ip.p.icmp.unused = htonl(0x00000000);
851
852 /* set the ip header in the ICMP payload */
853 packet = (struct swrap_packet *)(((unsigned char *)ret) + icmp_hdr_len);
854 packet->ip.hdr.ver_hdrlen = 0x45; /* version 4 and 5 * 32 bit words */
855 packet->ip.hdr.tos = 0x00;
856 packet->ip.hdr.packet_length = htons(wire_len - icmp_hdr_len);
857 packet->ip.hdr.identification = htons(0xFFFF);
858 packet->ip.hdr.flags = 0x40; /* BIT 1 set - means don't fraqment */
859 packet->ip.hdr.fragment = htons(0x0000);
860 packet->ip.hdr.ttl = 0xFF;
861 packet->ip.hdr.protocol = icmp_protocol;
862 packet->ip.hdr.hdr_checksum = htons(0x0000);
863 packet->ip.hdr.src_addr = dest_addr->sin_addr.s_addr;
864 packet->ip.hdr.dest_addr = src_addr->sin_addr.s_addr;
865
866 src_port = dest_addr->sin_port;
867 dest_port = src_addr->sin_port;
868 }
869
870 switch (socket_type) {
871 case SOCK_STREAM:
872 packet->ip.p.tcp.source_port = src_port;
873 packet->ip.p.tcp.dest_port = dest_port;
874 packet->ip.p.tcp.seq_num = htonl(tcp_seq);
875 packet->ip.p.tcp.ack_num = htonl(tcp_ack);
876 packet->ip.p.tcp.hdr_length = 0x50; /* 5 * 32 bit words */
877 packet->ip.p.tcp.control = tcp_ctl;
878 packet->ip.p.tcp.window = htons(0x7FFF);
879 packet->ip.p.tcp.checksum = htons(0x0000);
880 packet->ip.p.tcp.urg = htons(0x0000);
881
882 break;
883
884 case SOCK_DGRAM:
885 packet->ip.p.udp.source_port = src_addr->sin_port;
886 packet->ip.p.udp.dest_port = dest_addr->sin_port;
887 packet->ip.p.udp.length = htons(8 + payload_len);
888 packet->ip.p.udp.checksum = htons(0x0000);
889
890 break;
891 }
892
893 if (payload && payload_len > 0) {
894 unsigned char *p = (unsigned char *)ret;
895 p += nonwire_len;
896 p += wire_hdr_len;
897 memcpy(p, payload, payload_len);
898 }
899
900 *_packet_len = packet_len - icmp_truncate_len;
901 return ret;
902}
903
904static int swrap_get_pcap_fd(const char *fname)
905{
906 static int fd = -1;
907
908 if (fd != -1) return fd;
909
910 fd = open(fname, O_WRONLY|O_CREAT|O_EXCL|O_APPEND, 0644);
911 if (fd != -1) {
912 struct swrap_file_hdr file_hdr;
913 file_hdr.magic = 0xA1B2C3D4;
914 file_hdr.version_major = 0x0002;
915 file_hdr.version_minor = 0x0004;
916 file_hdr.timezone = 0x00000000;
917 file_hdr.sigfigs = 0x00000000;
918 file_hdr.frame_max_len = SWRAP_FRAME_LENGTH_MAX;
919 file_hdr.link_type = 0x0065; /* 101 RAW IP */
920
921 write(fd, &file_hdr, sizeof(file_hdr));
922 return fd;
923 }
924
925 fd = open(fname, O_WRONLY|O_APPEND, 0644);
926
927 return fd;
928}
929
930static void swrap_dump_packet(struct socket_info *si, const struct sockaddr *addr,
931 enum swrap_packet_type type,
932 const void *buf, size_t len)
933{
934 const struct sockaddr_in *src_addr;
935 const struct sockaddr_in *dest_addr;
936 const char *file_name;
937 unsigned long tcp_seq = 0;
938 unsigned long tcp_ack = 0;
939 unsigned char tcp_ctl = 0;
940 int unreachable = 0;
941 struct timeval tv;
942 struct swrap_packet *packet;
943 size_t packet_len = 0;
944 int fd;
945
946 file_name = socket_wrapper_pcap_file();
947 if (!file_name) {
948 return;
949 }
950
951 switch (si->family) {
952 case AF_INET:
953#ifdef HAVE_IPV6
954 case AF_INET6:
955#endif
956 break;
957 default:
958 return;
959 }
960
961 switch (type) {
962 case SWRAP_CONNECT_SEND:
963 if (si->type != SOCK_STREAM) return;
964
965 src_addr = (const struct sockaddr_in *)si->myname;
966 dest_addr = (const struct sockaddr_in *)addr;
967
968 tcp_seq = si->io.pck_snd;
969 tcp_ack = si->io.pck_rcv;
970 tcp_ctl = 0x02; /* SYN */
971
972 si->io.pck_snd += 1;
973
974 break;
975
976 case SWRAP_CONNECT_RECV:
977 if (si->type != SOCK_STREAM) return;
978
979 dest_addr = (const struct sockaddr_in *)si->myname;
980 src_addr = (const struct sockaddr_in *)addr;
981
982 tcp_seq = si->io.pck_rcv;
983 tcp_ack = si->io.pck_snd;
984 tcp_ctl = 0x12; /** SYN,ACK */
985
986 si->io.pck_rcv += 1;
987
988 break;
989
990 case SWRAP_CONNECT_UNREACH:
991 if (si->type != SOCK_STREAM) return;
992
993 dest_addr = (const struct sockaddr_in *)si->myname;
994 src_addr = (const struct sockaddr_in *)addr;
995
996 /* Unreachable: resend the data of SWRAP_CONNECT_SEND */
997 tcp_seq = si->io.pck_snd - 1;
998 tcp_ack = si->io.pck_rcv;
999 tcp_ctl = 0x02; /* SYN */
1000 unreachable = 1;
1001
1002 break;
1003
1004 case SWRAP_CONNECT_ACK:
1005 if (si->type != SOCK_STREAM) return;
1006
1007 src_addr = (const struct sockaddr_in *)si->myname;
1008 dest_addr = (const struct sockaddr_in *)addr;
1009
1010 tcp_seq = si->io.pck_snd;
1011 tcp_ack = si->io.pck_rcv;
1012 tcp_ctl = 0x10; /* ACK */
1013
1014 break;
1015
1016 case SWRAP_ACCEPT_SEND:
1017 if (si->type != SOCK_STREAM) return;
1018
1019 dest_addr = (const struct sockaddr_in *)si->myname;
1020 src_addr = (const struct sockaddr_in *)addr;
1021
1022 tcp_seq = si->io.pck_rcv;
1023 tcp_ack = si->io.pck_snd;
1024 tcp_ctl = 0x02; /* SYN */
1025
1026 si->io.pck_rcv += 1;
1027
1028 break;
1029
1030 case SWRAP_ACCEPT_RECV:
1031 if (si->type != SOCK_STREAM) return;
1032
1033 src_addr = (const struct sockaddr_in *)si->myname;
1034 dest_addr = (const struct sockaddr_in *)addr;
1035
1036 tcp_seq = si->io.pck_snd;
1037 tcp_ack = si->io.pck_rcv;
1038 tcp_ctl = 0x12; /* SYN,ACK */
1039
1040 si->io.pck_snd += 1;
1041
1042 break;
1043
1044 case SWRAP_ACCEPT_ACK:
1045 if (si->type != SOCK_STREAM) return;
1046
1047 dest_addr = (const struct sockaddr_in *)si->myname;
1048 src_addr = (const struct sockaddr_in *)addr;
1049
1050 tcp_seq = si->io.pck_rcv;
1051 tcp_ack = si->io.pck_snd;
1052 tcp_ctl = 0x10; /* ACK */
1053
1054 break;
1055
1056 case SWRAP_SEND:
1057 src_addr = (const struct sockaddr_in *)si->myname;
1058 dest_addr = (const struct sockaddr_in *)si->peername;
1059
1060 tcp_seq = si->io.pck_snd;
1061 tcp_ack = si->io.pck_rcv;
1062 tcp_ctl = 0x18; /* PSH,ACK */
1063
1064 si->io.pck_snd += len;
1065
1066 break;
1067
1068 case SWRAP_SEND_RST:
1069 dest_addr = (const struct sockaddr_in *)si->myname;
1070 src_addr = (const struct sockaddr_in *)si->peername;
1071
1072 if (si->type == SOCK_DGRAM) {
1073 swrap_dump_packet(si, si->peername,
1074 SWRAP_SENDTO_UNREACH,
1075 buf, len);
1076 return;
1077 }
1078
1079 tcp_seq = si->io.pck_rcv;
1080 tcp_ack = si->io.pck_snd;
1081 tcp_ctl = 0x14; /** RST,ACK */
1082
1083 break;
1084
1085 case SWRAP_PENDING_RST:
1086 dest_addr = (const struct sockaddr_in *)si->myname;
1087 src_addr = (const struct sockaddr_in *)si->peername;
1088
1089 if (si->type == SOCK_DGRAM) {
1090 return;
1091 }
1092
1093 tcp_seq = si->io.pck_rcv;
1094 tcp_ack = si->io.pck_snd;
1095 tcp_ctl = 0x14; /* RST,ACK */
1096
1097 break;
1098
1099 case SWRAP_RECV:
1100 dest_addr = (const struct sockaddr_in *)si->myname;
1101 src_addr = (const struct sockaddr_in *)si->peername;
1102
1103 tcp_seq = si->io.pck_rcv;
1104 tcp_ack = si->io.pck_snd;
1105 tcp_ctl = 0x18; /* PSH,ACK */
1106
1107 si->io.pck_rcv += len;
1108
1109 break;
1110
1111 case SWRAP_RECV_RST:
1112 dest_addr = (const struct sockaddr_in *)si->myname;
1113 src_addr = (const struct sockaddr_in *)si->peername;
1114
1115 if (si->type == SOCK_DGRAM) {
1116 return;
1117 }
1118
1119 tcp_seq = si->io.pck_rcv;
1120 tcp_ack = si->io.pck_snd;
1121 tcp_ctl = 0x14; /* RST,ACK */
1122
1123 break;
1124
1125 case SWRAP_SENDTO:
1126 src_addr = (const struct sockaddr_in *)si->myname;
1127 dest_addr = (const struct sockaddr_in *)addr;
1128
1129 si->io.pck_snd += len;
1130
1131 break;
1132
1133 case SWRAP_SENDTO_UNREACH:
1134 dest_addr = (const struct sockaddr_in *)si->myname;
1135 src_addr = (const struct sockaddr_in *)addr;
1136
1137 unreachable = 1;
1138
1139 break;
1140
1141 case SWRAP_RECVFROM:
1142 dest_addr = (const struct sockaddr_in *)si->myname;
1143 src_addr = (const struct sockaddr_in *)addr;
1144
1145 si->io.pck_rcv += len;
1146
1147 break;
1148
1149 case SWRAP_CLOSE_SEND:
1150 if (si->type != SOCK_STREAM) return;
1151
1152 src_addr = (const struct sockaddr_in *)si->myname;
1153 dest_addr = (const struct sockaddr_in *)si->peername;
1154
1155 tcp_seq = si->io.pck_snd;
1156 tcp_ack = si->io.pck_rcv;
1157 tcp_ctl = 0x11; /* FIN, ACK */
1158
1159 si->io.pck_snd += 1;
1160
1161 break;
1162
1163 case SWRAP_CLOSE_RECV:
1164 if (si->type != SOCK_STREAM) return;
1165
1166 dest_addr = (const struct sockaddr_in *)si->myname;
1167 src_addr = (const struct sockaddr_in *)si->peername;
1168
1169 tcp_seq = si->io.pck_rcv;
1170 tcp_ack = si->io.pck_snd;
1171 tcp_ctl = 0x11; /* FIN,ACK */
1172
1173 si->io.pck_rcv += 1;
1174
1175 break;
1176
1177 case SWRAP_CLOSE_ACK:
1178 if (si->type != SOCK_STREAM) return;
1179
1180 src_addr = (const struct sockaddr_in *)si->myname;
1181 dest_addr = (const struct sockaddr_in *)si->peername;
1182
1183 tcp_seq = si->io.pck_snd;
1184 tcp_ack = si->io.pck_rcv;
1185 tcp_ctl = 0x10; /* ACK */
1186
1187 break;
1188 default:
1189 return;
1190 }
1191
1192 swrapGetTimeOfDay(&tv);
1193
1194 packet = swrap_packet_init(&tv, src_addr, dest_addr, si->type,
1195 (const unsigned char *)buf, len,
1196 tcp_seq, tcp_ack, tcp_ctl, unreachable,
1197 &packet_len);
1198 if (!packet) {
1199 return;
1200 }
1201
1202 fd = swrap_get_pcap_fd(file_name);
1203 if (fd != -1) {
1204 write(fd, packet, packet_len);
1205 }
1206
1207 free(packet);
1208}
1209
1210_PUBLIC_ int swrap_socket(int family, int type, int protocol)
1211{
1212 struct socket_info *si;
1213 int fd;
1214
1215 if (!socket_wrapper_dir()) {
1216 return real_socket(family, type, protocol);
1217 }
1218
1219 switch (family) {
1220 case AF_INET:
1221#ifdef HAVE_IPV6
1222 case AF_INET6:
1223#endif
1224 break;
1225 case AF_UNIX:
1226 return real_socket(family, type, protocol);
1227 default:
1228 errno = EAFNOSUPPORT;
1229 return -1;
1230 }
1231
1232 switch (type) {
1233 case SOCK_STREAM:
1234 break;
1235 case SOCK_DGRAM:
1236 break;
1237 default:
1238 errno = EPROTONOSUPPORT;
1239 return -1;
1240 }
1241
1242#if 0
1243 switch (protocol) {
1244 case 0:
1245 break;
1246 default:
1247 errno = EPROTONOSUPPORT;
1248 return -1;
1249 }
1250#endif
1251
1252 fd = real_socket(AF_UNIX, type, 0);
1253
1254 if (fd == -1) return -1;
1255
1256 si = (struct socket_info *)calloc(1, sizeof(struct socket_info));
1257
1258 si->family = family;
1259 si->type = type;
1260 si->protocol = protocol;
1261 si->fd = fd;
1262
1263 SWRAP_DLIST_ADD(sockets, si);
1264
1265 return si->fd;
1266}
1267
1268_PUBLIC_ int swrap_accept(int s, struct sockaddr *addr, socklen_t *addrlen)
1269{
1270 struct socket_info *parent_si, *child_si;
1271 int fd;
1272 struct sockaddr_un un_addr;
1273 socklen_t un_addrlen = sizeof(un_addr);
1274 struct sockaddr_un un_my_addr;
1275 socklen_t un_my_addrlen = sizeof(un_my_addr);
1276 struct sockaddr *my_addr;
1277 socklen_t my_addrlen, len;
1278 int ret;
1279
1280 parent_si = find_socket_info(s);
1281 if (!parent_si) {
1282 return real_accept(s, addr, addrlen);
1283 }
1284
1285 /*
1286 * assume out sockaddr have the same size as the in parent
1287 * socket family
1288 */
1289 my_addrlen = socket_length(parent_si->family);
1290 if (my_addrlen < 0) {
1291 errno = EINVAL;
1292 return -1;
1293 }
1294
1295 my_addr = malloc(my_addrlen);
1296 if (my_addr == NULL) {
1297 return -1;
1298 }
1299
1300 memset(&un_addr, 0, sizeof(un_addr));
1301 memset(&un_my_addr, 0, sizeof(un_my_addr));
1302
1303 ret = real_accept(s, (struct sockaddr *)&un_addr, &un_addrlen);
1304 if (ret == -1) {
1305 free(my_addr);
1306 return ret;
1307 }
1308
1309 fd = ret;
1310
1311 len = my_addrlen;
1312 ret = sockaddr_convert_from_un(parent_si, &un_addr, un_addrlen,
1313 parent_si->family, my_addr, &len);
1314 if (ret == -1) {
1315 free(my_addr);
1316 close(fd);
1317 return ret;
1318 }
1319
1320 child_si = (struct socket_info *)malloc(sizeof(struct socket_info));
1321 memset(child_si, 0, sizeof(*child_si));
1322
1323 child_si->fd = fd;
1324 child_si->family = parent_si->family;
1325 child_si->type = parent_si->type;
1326 child_si->protocol = parent_si->protocol;
1327 child_si->bound = 1;
1328 child_si->is_server = 1;
1329
1330 child_si->peername_len = len;
1331 child_si->peername = sockaddr_dup(my_addr, len);
1332
1333 if (addr != NULL && addrlen != NULL) {
1334 *addrlen = len;
1335 if (*addrlen >= len)
1336 memcpy(addr, my_addr, len);
1337 *addrlen = 0;
1338 }
1339
1340 ret = real_getsockname(fd, (struct sockaddr *)&un_my_addr, &un_my_addrlen);
1341 if (ret == -1) {
1342 free(child_si);
1343 close(fd);
1344 return ret;
1345 }
1346
1347 len = my_addrlen;
1348 ret = sockaddr_convert_from_un(child_si, &un_my_addr, un_my_addrlen,
1349 child_si->family, my_addr, &len);
1350 if (ret == -1) {
1351 free(child_si);
1352 free(my_addr);
1353 close(fd);
1354 return ret;
1355 }
1356
1357 child_si->myname_len = len;
1358 child_si->myname = sockaddr_dup(my_addr, len);
1359 free(my_addr);
1360
1361 SWRAP_DLIST_ADD(sockets, child_si);
1362
1363 swrap_dump_packet(child_si, addr, SWRAP_ACCEPT_SEND, NULL, 0);
1364 swrap_dump_packet(child_si, addr, SWRAP_ACCEPT_RECV, NULL, 0);
1365 swrap_dump_packet(child_si, addr, SWRAP_ACCEPT_ACK, NULL, 0);
1366
1367 return fd;
1368}
1369
1370static int autobind_start_init;
1371static int autobind_start;
1372
1373/* using sendto() or connect() on an unbound socket would give the
1374 recipient no way to reply, as unlike UDP and TCP, a unix domain
1375 socket can't auto-assign emphemeral port numbers, so we need to
1376 assign it here */
1377static int swrap_auto_bind(struct socket_info *si)
1378{
1379 struct sockaddr_un un_addr;
1380 int i;
1381 char type;
1382 int ret;
1383 int port;
1384 struct stat st;
1385
1386 if (autobind_start_init != 1) {
1387 autobind_start_init = 1;
1388 autobind_start = getpid();
1389 autobind_start %= 50000;
1390 autobind_start += 10000;
1391 }
1392
1393 un_addr.sun_family = AF_UNIX;
1394
1395 switch (si->family) {
1396 case AF_INET: {
1397 struct sockaddr_in in;
1398
1399 switch (si->type) {
1400 case SOCK_STREAM:
1401 type = SOCKET_TYPE_CHAR_TCP;
1402 break;
1403 case SOCK_DGRAM:
1404 type = SOCKET_TYPE_CHAR_UDP;
1405 break;
1406 default:
1407 errno = ESOCKTNOSUPPORT;
1408 return -1;
1409 }
1410
1411 memset(&in, 0, sizeof(in));
1412 in.sin_family = AF_INET;
1413 in.sin_addr.s_addr = htonl(127<<24 |
1414 socket_wrapper_default_iface());
1415
1416 si->myname_len = sizeof(in);
1417 si->myname = sockaddr_dup(&in, si->myname_len);
1418 break;
1419 }
1420#ifdef HAVE_IPV6
1421 case AF_INET6: {
1422 struct sockaddr_in6 in6;
1423
1424 switch (si->type) {
1425 case SOCK_STREAM:
1426 type = SOCKET_TYPE_CHAR_TCP_V6;
1427 break;
1428 case SOCK_DGRAM:
1429 type = SOCKET_TYPE_CHAR_UDP_V6;
1430 break;
1431 default:
1432 errno = ESOCKTNOSUPPORT;
1433 return -1;
1434 }
1435
1436 memset(&in6, 0, sizeof(in6));
1437 in6.sin6_family = AF_INET6;
1438 in6.sin6_addr.s6_addr[0] = SW_IPV6_ADDRESS;
1439 si->myname_len = sizeof(in6);
1440 si->myname = sockaddr_dup(&in6, si->myname_len);
1441 break;
1442 }
1443#endif
1444 default:
1445 errno = ESOCKTNOSUPPORT;
1446 return -1;
1447 }
1448
1449 if (autobind_start > 60000) {
1450 autobind_start = 10000;
1451 }
1452
1453 for (i=0;i<1000;i++) {
1454 port = autobind_start + i;
1455 snprintf(un_addr.sun_path, sizeof(un_addr.sun_path),
1456 "%s/"SOCKET_FORMAT, socket_wrapper_dir(),
1457 type, socket_wrapper_default_iface(), port);
1458 if (stat(un_addr.sun_path, &st) == 0) continue;
1459
1460 ret = real_bind(si->fd, (struct sockaddr *)&un_addr, sizeof(un_addr));
1461 if (ret == -1) return ret;
1462
1463 si->tmp_path = strdup(un_addr.sun_path);
1464 si->bound = 1;
1465 autobind_start = port + 1;
1466 break;
1467 }
1468 if (i == 1000) {
1469 errno = ENFILE;
1470 return -1;
1471 }
1472
1473 set_port(si->family, port, si->myname);
1474
1475 return 0;
1476}
1477
1478
1479_PUBLIC_ int swrap_connect(int s, const struct sockaddr *serv_addr, socklen_t addrlen)
1480{
1481 int ret;
1482 struct sockaddr_un un_addr;
1483 struct socket_info *si = find_socket_info(s);
1484
1485 if (!si) {
1486 return real_connect(s, serv_addr, addrlen);
1487 }
1488
1489 if (si->bound == 0) {
1490 ret = swrap_auto_bind(si);
1491 if (ret == -1) return -1;
1492 }
1493
1494 if (si->family != serv_addr->sa_family) {
1495 errno = EINVAL;
1496 return -1;
1497 }
1498
1499 ret = sockaddr_convert_to_un(si, (const struct sockaddr *)serv_addr, addrlen, &un_addr, 0, NULL);
1500 if (ret == -1) return -1;
1501
1502 swrap_dump_packet(si, serv_addr, SWRAP_CONNECT_SEND, NULL, 0);
1503
1504 ret = real_connect(s, (struct sockaddr *)&un_addr,
1505 sizeof(struct sockaddr_un));
1506
1507 /* to give better errors */
1508 if (ret == -1 && errno == ENOENT) {
1509 errno = EHOSTUNREACH;
1510 }
1511
1512 if (ret == 0) {
1513 si->peername_len = addrlen;
1514 si->peername = sockaddr_dup(serv_addr, addrlen);
1515
1516 swrap_dump_packet(si, serv_addr, SWRAP_CONNECT_RECV, NULL, 0);
1517 swrap_dump_packet(si, serv_addr, SWRAP_CONNECT_ACK, NULL, 0);
1518 } else {
1519 swrap_dump_packet(si, serv_addr, SWRAP_CONNECT_UNREACH, NULL, 0);
1520 }
1521
1522 return ret;
1523}
1524
1525_PUBLIC_ int swrap_bind(int s, const struct sockaddr *myaddr, socklen_t addrlen)
1526{
1527 int ret;
1528 struct sockaddr_un un_addr;
1529 struct socket_info *si = find_socket_info(s);
1530
1531 if (!si) {
1532 return real_bind(s, myaddr, addrlen);
1533 }
1534
1535 si->myname_len = addrlen;
1536 si->myname = sockaddr_dup(myaddr, addrlen);
1537
1538 ret = sockaddr_convert_to_un(si, (const struct sockaddr *)myaddr, addrlen, &un_addr, 1, &si->bcast);
1539 if (ret == -1) return -1;
1540
1541 unlink(un_addr.sun_path);
1542
1543 ret = real_bind(s, (struct sockaddr *)&un_addr,
1544 sizeof(struct sockaddr_un));
1545
1546 if (ret == 0) {
1547 si->bound = 1;
1548 }
1549
1550 return ret;
1551}
1552
1553_PUBLIC_ int swrap_listen(int s, int backlog)
1554{
1555 int ret;
1556 struct socket_info *si = find_socket_info(s);
1557
1558 if (!si) {
1559 return real_listen(s, backlog);
1560 }
1561
1562 ret = real_listen(s, backlog);
1563
1564 return ret;
1565}
1566
1567_PUBLIC_ int swrap_getpeername(int s, struct sockaddr *name, socklen_t *addrlen)
1568{
1569 struct socket_info *si = find_socket_info(s);
1570
1571 if (!si) {
1572 return real_getpeername(s, name, addrlen);
1573 }
1574
1575 if (!si->peername)
1576 {
1577 errno = ENOTCONN;
1578 return -1;
1579 }
1580
1581 memcpy(name, si->peername, si->peername_len);
1582 *addrlen = si->peername_len;
1583
1584 return 0;
1585}
1586
1587_PUBLIC_ int swrap_getsockname(int s, struct sockaddr *name, socklen_t *addrlen)
1588{
1589 struct socket_info *si = find_socket_info(s);
1590
1591 if (!si) {
1592 return real_getsockname(s, name, addrlen);
1593 }
1594
1595 memcpy(name, si->myname, si->myname_len);
1596 *addrlen = si->myname_len;
1597
1598 return 0;
1599}
1600
1601_PUBLIC_ int swrap_getsockopt(int s, int level, int optname, void *optval, socklen_t *optlen)
1602{
1603 struct socket_info *si = find_socket_info(s);
1604
1605 if (!si) {
1606 return real_getsockopt(s, level, optname, optval, optlen);
1607 }
1608
1609 if (level == SOL_SOCKET) {
1610 return real_getsockopt(s, level, optname, optval, optlen);
1611 }
1612
1613 errno = ENOPROTOOPT;
1614 return -1;
1615}
1616
1617_PUBLIC_ int swrap_setsockopt(int s, int level, int optname, const void *optval, socklen_t optlen)
1618{
1619 struct socket_info *si = find_socket_info(s);
1620
1621 if (!si) {
1622 return real_setsockopt(s, level, optname, optval, optlen);
1623 }
1624
1625 if (level == SOL_SOCKET) {
1626 return real_setsockopt(s, level, optname, optval, optlen);
1627 }
1628
1629 switch (si->family) {
1630 case AF_INET:
1631 return 0;
1632 default:
1633 errno = ENOPROTOOPT;
1634 return -1;
1635 }
1636}
1637
1638_PUBLIC_ ssize_t swrap_recvfrom(int s, void *buf, size_t len, int flags, struct sockaddr *from, socklen_t *fromlen)
1639{
1640 struct sockaddr_un un_addr;
1641 socklen_t un_addrlen = sizeof(un_addr);
1642 int ret;
1643 struct socket_info *si = find_socket_info(s);
1644
1645 if (!si) {
1646 return real_recvfrom(s, buf, len, flags, from, fromlen);
1647 }
1648
1649 /* irix 6.4 forgets to null terminate the sun_path string :-( */
1650 memset(&un_addr, 0, sizeof(un_addr));
1651 ret = real_recvfrom(s, buf, len, flags, (struct sockaddr *)&un_addr, &un_addrlen);
1652 if (ret == -1)
1653 return ret;
1654
1655 if (sockaddr_convert_from_un(si, &un_addr, un_addrlen,
1656 si->family, from, fromlen) == -1) {
1657 return -1;
1658 }
1659
1660 swrap_dump_packet(si, from, SWRAP_RECVFROM, buf, ret);
1661
1662 return ret;
1663}
1664
1665
1666_PUBLIC_ ssize_t swrap_sendto(int s, const void *buf, size_t len, int flags, const struct sockaddr *to, socklen_t tolen)
1667{
1668 struct sockaddr_un un_addr;
1669 int ret;
1670 struct socket_info *si = find_socket_info(s);
1671 int bcast = 0;
1672
1673 if (!si) {
1674 return real_sendto(s, buf, len, flags, to, tolen);
1675 }
1676
1677 switch (si->type) {
1678 case SOCK_STREAM:
1679 ret = real_send(s, buf, len, flags);
1680 break;
1681 case SOCK_DGRAM:
1682 if (si->bound == 0) {
1683 ret = swrap_auto_bind(si);
1684 if (ret == -1) return -1;
1685 }
1686
1687 ret = sockaddr_convert_to_un(si, to, tolen, &un_addr, 0, &bcast);
1688 if (ret == -1) return -1;
1689
1690 if (bcast) {
1691 struct stat st;
1692 unsigned int iface;
1693 unsigned int prt = ntohs(((const struct sockaddr_in *)to)->sin_port);
1694 char type;
1695
1696 type = SOCKET_TYPE_CHAR_UDP;
1697
1698 for(iface=0; iface <= MAX_WRAPPED_INTERFACES; iface++) {
1699 snprintf(un_addr.sun_path, sizeof(un_addr.sun_path), "%s/"SOCKET_FORMAT,
1700 socket_wrapper_dir(), type, iface, prt);
1701 if (stat(un_addr.sun_path, &st) != 0) continue;
1702
1703 /* ignore the any errors in broadcast sends */
1704 real_sendto(s, buf, len, flags, (struct sockaddr *)&un_addr, sizeof(un_addr));
1705 }
1706
1707 swrap_dump_packet(si, to, SWRAP_SENDTO, buf, len);
1708
1709 return len;
1710 }
1711
1712 ret = real_sendto(s, buf, len, flags, (struct sockaddr *)&un_addr, sizeof(un_addr));
1713 break;
1714 default:
1715 ret = -1;
1716 errno = EHOSTUNREACH;
1717 break;
1718 }
1719
1720 /* to give better errors */
1721 if (ret == -1 && errno == ENOENT) {
1722 errno = EHOSTUNREACH;
1723 }
1724
1725 if (ret == -1) {
1726 swrap_dump_packet(si, to, SWRAP_SENDTO, buf, len);
1727 swrap_dump_packet(si, to, SWRAP_SENDTO_UNREACH, buf, len);
1728 } else {
1729 swrap_dump_packet(si, to, SWRAP_SENDTO, buf, ret);
1730 }
1731
1732 return ret;
1733}
1734
1735_PUBLIC_ int swrap_ioctl(int s, int r, void *p)
1736{
1737 int ret;
1738 struct socket_info *si = find_socket_info(s);
1739 int value;
1740
1741 if (!si) {
1742 return real_ioctl(s, r, p);
1743 }
1744
1745 ret = real_ioctl(s, r, p);
1746
1747 switch (r) {
1748 case FIONREAD:
1749 value = *((int *)p);
1750 if (ret == -1 && errno != EAGAIN && errno != ENOBUFS) {
1751 swrap_dump_packet(si, NULL, SWRAP_PENDING_RST, NULL, 0);
1752 } else if (value == 0) { /* END OF FILE */
1753 swrap_dump_packet(si, NULL, SWRAP_PENDING_RST, NULL, 0);
1754 }
1755 break;
1756 }
1757
1758 return ret;
1759}
1760
1761_PUBLIC_ ssize_t swrap_recv(int s, void *buf, size_t len, int flags)
1762{
1763 int ret;
1764 struct socket_info *si = find_socket_info(s);
1765
1766 if (!si) {
1767 return real_recv(s, buf, len, flags);
1768 }
1769
1770 ret = real_recv(s, buf, len, flags);
1771 if (ret == -1 && errno != EAGAIN && errno != ENOBUFS) {
1772 swrap_dump_packet(si, NULL, SWRAP_RECV_RST, NULL, 0);
1773 } else if (ret == 0) { /* END OF FILE */
1774 swrap_dump_packet(si, NULL, SWRAP_RECV_RST, NULL, 0);
1775 } else {
1776 swrap_dump_packet(si, NULL, SWRAP_RECV, buf, ret);
1777 }
1778
1779 return ret;
1780}
1781
1782
1783_PUBLIC_ ssize_t swrap_send(int s, const void *buf, size_t len, int flags)
1784{
1785 int ret;
1786 struct socket_info *si = find_socket_info(s);
1787
1788 if (!si) {
1789 return real_send(s, buf, len, flags);
1790 }
1791
1792 ret = real_send(s, buf, len, flags);
1793
1794 if (ret == -1) {
1795 swrap_dump_packet(si, NULL, SWRAP_SEND, buf, len);
1796 swrap_dump_packet(si, NULL, SWRAP_SEND_RST, NULL, 0);
1797 } else {
1798 swrap_dump_packet(si, NULL, SWRAP_SEND, buf, ret);
1799 }
1800
1801 return ret;
1802}
1803
1804_PUBLIC_ int swrap_close(int fd)
1805{
1806 struct socket_info *si = find_socket_info(fd);
1807 int ret;
1808
1809 if (!si) {
1810 return real_close(fd);
1811 }
1812
1813 SWRAP_DLIST_REMOVE(sockets, si);
1814
1815 if (si->myname && si->peername) {
1816 swrap_dump_packet(si, NULL, SWRAP_CLOSE_SEND, NULL, 0);
1817 }
1818
1819 ret = real_close(fd);
1820
1821 if (si->myname && si->peername) {
1822 swrap_dump_packet(si, NULL, SWRAP_CLOSE_RECV, NULL, 0);
1823 swrap_dump_packet(si, NULL, SWRAP_CLOSE_ACK, NULL, 0);
1824 }
1825
1826 if (si->path) free(si->path);
1827 if (si->myname) free(si->myname);
1828 if (si->peername) free(si->peername);
1829 if (si->tmp_path) {
1830 unlink(si->tmp_path);
1831 free(si->tmp_path);
1832 }
1833 free(si);
1834
1835 return ret;
1836}
1837
1838static int
1839dup_internal(const struct socket_info *si_oldd, int fd)
1840{
1841 struct socket_info *si_newd;
1842
1843 si_newd = (struct socket_info *)calloc(1, sizeof(struct socket_info));
1844
1845 si_newd->fd = fd;
1846
1847 si_newd->family = si_oldd->family;
1848 si_newd->type = si_oldd->type;
1849 si_newd->protocol = si_oldd->protocol;
1850 si_newd->bound = si_oldd->bound;
1851 si_newd->bcast = si_oldd->bcast;
1852 if (si_oldd->path)
1853 si_newd->path = strdup(si_oldd->path);
1854 if (si_oldd->tmp_path)
1855 si_newd->tmp_path = strdup(si_oldd->tmp_path);
1856 si_newd->myname =
1857 sockaddr_dup(si_oldd->myname, si_oldd->myname_len);
1858 si_newd->myname_len = si_oldd->myname_len;
1859 si_newd->peername =
1860 sockaddr_dup(si_oldd->peername, si_oldd->peername_len);
1861 si_newd->peername_len = si_oldd->peername_len;
1862
1863 si_newd->io = si_oldd->io;
1864
1865 SWRAP_DLIST_ADD(sockets, si_newd);
1866
1867 return fd;
1868}
1869
1870
1871_PUBLIC_ int swrap_dup(int oldd)
1872{
1873 struct socket_info *si;
1874 int fd;
1875
1876 si = find_socket_info(oldd);
1877 if (si == NULL)
1878 return real_dup(oldd);
1879
1880 fd = real_dup(si->fd);
1881 if (fd < 0)
1882 return fd;
1883
1884 return dup_internal(si, fd);
1885}
1886
1887
1888_PUBLIC_ int swrap_dup2(int oldd, int newd)
1889{
1890 struct socket_info *si_newd, *si_oldd;
1891 int fd;
1892
1893 if (newd == oldd)
1894 return newd;
1895
1896 si_oldd = find_socket_info(oldd);
1897 si_newd = find_socket_info(newd);
1898
1899 if (si_oldd == NULL && si_newd == NULL)
1900 return real_dup2(oldd, newd);
1901
1902 fd = real_dup2(si_oldd->fd, newd);
1903 if (fd < 0)
1904 return fd;
1905
1906 /* close new socket first */
1907 if (si_newd)
1908 swrap_close(newd);
1909
1910 return dup_internal(si_oldd, fd);
1911}
Note: See TracBrowser for help on using the repository browser.