2 Unix SMB/CIFS implementation.
4 Copyright (C) Stefan Metzmacher 2009
6 ** NOTE! The following LGPL license applies to the tevent
7 ** library. This does NOT imply that all of Samba is released
10 This library is free software; you can redistribute it and/or
11 modify it under the terms of the GNU Lesser General Public
12 License as published by the Free Software Foundation; either
13 version 3 of the License, or (at your option) any later version.
15 This library is distributed in the hope that it will be useful,
16 but WITHOUT ANY WARRANTY; without even the implied warranty of
17 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
18 Lesser General Public License for more details.
20 You should have received a copy of the GNU Lesser General Public
21 License along with this library; if not, see <http://www.gnu.org/licenses/>.
25 #include "system/filesys.h"
26 #include "system/network.h"
28 #include "tsocket_internal.h"
30 static int tsocket_bsd_error_from_errno(int ret
,
48 if (sys_errno
== EINTR
) {
53 if (sys_errno
== EINPROGRESS
) {
58 if (sys_errno
== EAGAIN
) {
64 if (sys_errno
== EWOULDBLOCK
) {
73 static int tsocket_bsd_common_prepare_fd(int fd
, bool high_fd
)
86 /* first make a fd >= 3 */
96 for (i
=0; i
<num_fds
; i
++) {
105 /* fd should be nonblocking. */
108 #define FLAG_TO_SET O_NONBLOCK
111 #define FLAG_TO_SET O_NDELAY
113 #define FLAG_TO_SET FNDELAY
117 if ((flags
= fcntl(fd
, F_GETFL
)) == -1) {
121 flags
|= FLAG_TO_SET
;
122 if (fcntl(fd
, F_SETFL
, flags
) == -1) {
128 /* fd should be closed on exec() */
130 result
= flags
= fcntl(fd
, F_GETFD
, 0);
133 result
= fcntl(fd
, F_SETFD
, flags
);
150 static ssize_t
tsocket_bsd_pending(int fd
)
156 ret
= ioctl(fd
, FIONREAD
, &value
);
162 /* this should not be reached */
175 * if no data is available check if the socket is in error state. For
176 * dgram sockets it's the way to return ICMP error messages of
177 * connected sockets to the caller.
179 ret
= getsockopt(fd
, SOL_SOCKET
, SO_ERROR
, &error
, &len
);
190 static const struct tsocket_address_ops tsocket_address_bsd_ops
;
192 struct tsocket_address_bsd
{
195 struct sockaddr_in in
;
197 struct sockaddr_in6 in6
;
199 struct sockaddr_un un
;
200 struct sockaddr_storage ss
;
204 static int _tsocket_address_bsd_from_sockaddr(TALLOC_CTX
*mem_ctx
,
206 socklen_t sa_socklen
,
207 struct tsocket_address
**_addr
,
208 const char *location
)
210 struct tsocket_address
*addr
;
211 struct tsocket_address_bsd
*bsda
;
213 switch (sa
->sa_family
) {
215 if (sa_socklen
< sizeof(struct sockaddr_un
)) {
221 if (sa_socklen
< sizeof(struct sockaddr_in
)) {
228 if (sa_socklen
< sizeof(struct sockaddr_in6
)) {
235 errno
= EAFNOSUPPORT
;
239 if (sa_socklen
> sizeof(struct sockaddr_storage
)) {
244 addr
= tsocket_address_create(mem_ctx
,
245 &tsocket_address_bsd_ops
,
247 struct tsocket_address_bsd
,
256 memcpy(&bsda
->u
.ss
, sa
, sa_socklen
);
262 int _tsocket_address_inet_from_strings(TALLOC_CTX
*mem_ctx
,
266 struct tsocket_address
**_addr
,
267 const char *location
)
269 struct addrinfo hints
;
270 struct addrinfo
*result
= NULL
;
276 * we use SOCKET_STREAM here to get just one result
277 * back from getaddrinfo().
279 hints
.ai_socktype
= SOCK_STREAM
;
280 hints
.ai_flags
= AI_NUMERICHOST
| AI_NUMERICSERV
;
282 if (strcasecmp(fam
, "ip") == 0) {
283 hints
.ai_family
= AF_UNSPEC
;
291 } else if (strcasecmp(fam
, "ipv4") == 0) {
292 hints
.ai_family
= AF_INET
;
297 } else if (strcasecmp(fam
, "ipv6") == 0) {
298 hints
.ai_family
= AF_INET6
;
304 errno
= EAFNOSUPPORT
;
308 snprintf(port_str
, sizeof(port_str
) - 1, "%u", port
);
310 ret
= getaddrinfo(addr
, port_str
, &hints
, &result
);
321 if (result
->ai_socktype
!= SOCK_STREAM
) {
327 ret
= _tsocket_address_bsd_from_sockaddr(mem_ctx
,
335 freeaddrinfo(result
);
340 char *tsocket_address_inet_addr_string(const struct tsocket_address
*addr
,
343 struct tsocket_address_bsd
*bsda
= talloc_get_type(addr
->private_data
,
344 struct tsocket_address_bsd
);
345 char addr_str
[INET6_ADDRSTRLEN
+1];
353 switch (bsda
->u
.sa
.sa_family
) {
355 str
= inet_ntop(bsda
->u
.in
.sin_family
,
356 &bsda
->u
.in
.sin_addr
,
357 addr_str
, sizeof(addr_str
));
361 str
= inet_ntop(bsda
->u
.in6
.sin6_family
,
362 &bsda
->u
.in6
.sin6_addr
,
363 addr_str
, sizeof(addr_str
));
375 return talloc_strdup(mem_ctx
, str
);
378 uint16_t tsocket_address_inet_port(const struct tsocket_address
*addr
)
380 struct tsocket_address_bsd
*bsda
= talloc_get_type(addr
->private_data
,
381 struct tsocket_address_bsd
);
389 switch (bsda
->u
.sa
.sa_family
) {
391 port
= ntohs(bsda
->u
.in
.sin_port
);
395 port
= ntohs(bsda
->u
.in6
.sin6_port
);
406 int tsocket_address_inet_set_port(struct tsocket_address
*addr
,
409 struct tsocket_address_bsd
*bsda
= talloc_get_type(addr
->private_data
,
410 struct tsocket_address_bsd
);
417 switch (bsda
->u
.sa
.sa_family
) {
419 bsda
->u
.in
.sin_port
= htons(port
);
423 bsda
->u
.in6
.sin6_port
= htons(port
);
434 int _tsocket_address_unix_from_path(TALLOC_CTX
*mem_ctx
,
436 struct tsocket_address
**_addr
,
437 const char *location
)
439 struct sockaddr_un un
;
448 un
.sun_family
= AF_UNIX
;
449 strncpy(un
.sun_path
, path
, sizeof(un
.sun_path
));
451 ret
= _tsocket_address_bsd_from_sockaddr(mem_ctx
,
452 (struct sockaddr
*)p
,
460 char *tsocket_address_unix_path(const struct tsocket_address
*addr
,
463 struct tsocket_address_bsd
*bsda
= talloc_get_type(addr
->private_data
,
464 struct tsocket_address_bsd
);
472 switch (bsda
->u
.sa
.sa_family
) {
474 str
= bsda
->u
.un
.sun_path
;
481 return talloc_strdup(mem_ctx
, str
);
484 static char *tsocket_address_bsd_string(const struct tsocket_address
*addr
,
487 struct tsocket_address_bsd
*bsda
= talloc_get_type(addr
->private_data
,
488 struct tsocket_address_bsd
);
491 const char *prefix
= NULL
;
494 switch (bsda
->u
.sa
.sa_family
) {
496 return talloc_asprintf(mem_ctx
, "unix:%s",
497 bsda
->u
.un
.sun_path
);
511 addr_str
= tsocket_address_inet_addr_string(addr
, mem_ctx
);
516 port
= tsocket_address_inet_port(addr
);
518 str
= talloc_asprintf(mem_ctx
, "%s:%s:%u",
519 prefix
, addr_str
, port
);
520 talloc_free(addr_str
);
525 static struct tsocket_address
*tsocket_address_bsd_copy(const struct tsocket_address
*addr
,
527 const char *location
)
529 struct tsocket_address_bsd
*bsda
= talloc_get_type(addr
->private_data
,
530 struct tsocket_address_bsd
);
531 struct tsocket_address
*copy
;
534 ret
= _tsocket_address_bsd_from_sockaddr(mem_ctx
,
546 static const struct tsocket_address_ops tsocket_address_bsd_ops
= {
548 .string
= tsocket_address_bsd_string
,
549 .copy
= tsocket_address_bsd_copy
,
556 struct tevent_fd
*fde
;
558 void *readable_private
;
559 void (*readable_handler
)(void *private_data
);
560 void *writeable_private
;
561 void (*writeable_handler
)(void *private_data
);
564 static void tdgram_bsd_fde_handler(struct tevent_context
*ev
,
565 struct tevent_fd
*fde
,
569 struct tdgram_bsd
*bsds
= talloc_get_type_abort(private_data
,
572 if (flags
& TEVENT_FD_WRITE
) {
573 bsds
->writeable_handler(bsds
->writeable_private
);
576 if (flags
& TEVENT_FD_READ
) {
577 if (!bsds
->readable_handler
) {
578 TEVENT_FD_NOT_READABLE(bsds
->fde
);
581 bsds
->readable_handler(bsds
->readable_private
);
586 static int tdgram_bsd_set_readable_handler(struct tdgram_bsd
*bsds
,
587 struct tevent_context
*ev
,
588 void (*handler
)(void *private_data
),
596 if (!bsds
->readable_handler
) {
599 bsds
->readable_handler
= NULL
;
600 bsds
->readable_private
= NULL
;
605 /* read and write must use the same tevent_context */
606 if (bsds
->event_ptr
!= ev
) {
607 if (bsds
->readable_handler
|| bsds
->writeable_handler
) {
611 bsds
->event_ptr
= NULL
;
612 TALLOC_FREE(bsds
->fde
);
615 if (tevent_fd_get_flags(bsds
->fde
) == 0) {
616 TALLOC_FREE(bsds
->fde
);
618 bsds
->fde
= tevent_add_fd(ev
, bsds
,
619 bsds
->fd
, TEVENT_FD_READ
,
620 tdgram_bsd_fde_handler
,
627 /* cache the event context we're running on */
628 bsds
->event_ptr
= ev
;
629 } else if (!bsds
->readable_handler
) {
630 TEVENT_FD_READABLE(bsds
->fde
);
633 bsds
->readable_handler
= handler
;
634 bsds
->readable_private
= private_data
;
639 static int tdgram_bsd_set_writeable_handler(struct tdgram_bsd
*bsds
,
640 struct tevent_context
*ev
,
641 void (*handler
)(void *private_data
),
649 if (!bsds
->writeable_handler
) {
652 bsds
->writeable_handler
= NULL
;
653 bsds
->writeable_private
= NULL
;
654 TEVENT_FD_NOT_WRITEABLE(bsds
->fde
);
659 /* read and write must use the same tevent_context */
660 if (bsds
->event_ptr
!= ev
) {
661 if (bsds
->readable_handler
|| bsds
->writeable_handler
) {
665 bsds
->event_ptr
= NULL
;
666 TALLOC_FREE(bsds
->fde
);
669 if (tevent_fd_get_flags(bsds
->fde
) == 0) {
670 TALLOC_FREE(bsds
->fde
);
672 bsds
->fde
= tevent_add_fd(ev
, bsds
,
673 bsds
->fd
, TEVENT_FD_WRITE
,
674 tdgram_bsd_fde_handler
,
681 /* cache the event context we're running on */
682 bsds
->event_ptr
= ev
;
683 } else if (!bsds
->writeable_handler
) {
684 TEVENT_FD_WRITEABLE(bsds
->fde
);
687 bsds
->writeable_handler
= handler
;
688 bsds
->writeable_private
= private_data
;
693 struct tdgram_bsd_recvfrom_state
{
694 struct tdgram_context
*dgram
;
698 struct tsocket_address
*src
;
701 static int tdgram_bsd_recvfrom_destructor(struct tdgram_bsd_recvfrom_state
*state
)
703 struct tdgram_bsd
*bsds
= tdgram_context_data(state
->dgram
,
706 tdgram_bsd_set_readable_handler(bsds
, NULL
, NULL
, NULL
);
711 static void tdgram_bsd_recvfrom_handler(void *private_data
);
713 static struct tevent_req
*tdgram_bsd_recvfrom_send(TALLOC_CTX
*mem_ctx
,
714 struct tevent_context
*ev
,
715 struct tdgram_context
*dgram
)
717 struct tevent_req
*req
;
718 struct tdgram_bsd_recvfrom_state
*state
;
719 struct tdgram_bsd
*bsds
= tdgram_context_data(dgram
, struct tdgram_bsd
);
722 req
= tevent_req_create(mem_ctx
, &state
,
723 struct tdgram_bsd_recvfrom_state
);
728 state
->dgram
= dgram
;
733 talloc_set_destructor(state
, tdgram_bsd_recvfrom_destructor
);
735 if (bsds
->fd
== -1) {
736 tevent_req_error(req
, ENOTCONN
);
741 * this is a fast path, not waiting for the
742 * socket to become explicit readable gains
743 * about 10%-20% performance in benchmark tests.
745 tdgram_bsd_recvfrom_handler(req
);
746 if (!tevent_req_is_in_progress(req
)) {
750 ret
= tdgram_bsd_set_readable_handler(bsds
, ev
,
751 tdgram_bsd_recvfrom_handler
,
754 tevent_req_error(req
, errno
);
761 tevent_req_post(req
, ev
);
765 static void tdgram_bsd_recvfrom_handler(void *private_data
)
767 struct tevent_req
*req
= talloc_get_type_abort(private_data
,
769 struct tdgram_bsd_recvfrom_state
*state
= tevent_req_data(req
,
770 struct tdgram_bsd_recvfrom_state
);
771 struct tdgram_context
*dgram
= state
->dgram
;
772 struct tdgram_bsd
*bsds
= tdgram_context_data(dgram
, struct tdgram_bsd
);
773 struct tsocket_address_bsd
*bsda
;
775 struct sockaddr
*sa
= NULL
;
776 socklen_t sa_socklen
= 0;
780 ret
= tsocket_bsd_pending(bsds
->fd
);
785 err
= tsocket_bsd_error_from_errno(ret
, errno
, &retry
);
790 if (tevent_req_error(req
, err
)) {
794 state
->buf
= talloc_array(state
, uint8_t, ret
);
795 if (tevent_req_nomem(state
->buf
, req
)) {
800 state
->src
= tsocket_address_create(state
,
801 &tsocket_address_bsd_ops
,
803 struct tsocket_address_bsd
,
804 __location__
"bsd_recvfrom");
805 if (tevent_req_nomem(state
->src
, req
)) {
812 sa_socklen
= sizeof(bsda
->u
.ss
);
814 * for unix sockets we can't use the size of sockaddr_storage
815 * we would get EINVAL
817 if (bsda
->u
.sa
.sa_family
== AF_UNIX
) {
818 sa_socklen
= sizeof(bsda
->u
.un
);
821 ret
= recvfrom(bsds
->fd
, state
->buf
, state
->len
, 0, sa
, &sa_socklen
);
822 err
= tsocket_bsd_error_from_errno(ret
, errno
, &retry
);
827 if (tevent_req_error(req
, err
)) {
831 if (ret
!= state
->len
) {
832 tevent_req_error(req
, EIO
);
836 tevent_req_done(req
);
839 static ssize_t
tdgram_bsd_recvfrom_recv(struct tevent_req
*req
,
843 struct tsocket_address
**src
)
845 struct tdgram_bsd_recvfrom_state
*state
= tevent_req_data(req
,
846 struct tdgram_bsd_recvfrom_state
);
849 ret
= tsocket_simple_int_recv(req
, perrno
);
851 *buf
= talloc_move(mem_ctx
, &state
->buf
);
854 *src
= talloc_move(mem_ctx
, &state
->src
);
858 tevent_req_received(req
);
862 struct tdgram_bsd_sendto_state
{
863 struct tdgram_context
*dgram
;
867 const struct tsocket_address
*dst
;
872 static int tdgram_bsd_sendto_destructor(struct tdgram_bsd_sendto_state
*state
)
874 struct tdgram_bsd
*bsds
= tdgram_context_data(state
->dgram
,
877 tdgram_bsd_set_writeable_handler(bsds
, NULL
, NULL
, NULL
);
882 static void tdgram_bsd_sendto_handler(void *private_data
);
884 static struct tevent_req
*tdgram_bsd_sendto_send(TALLOC_CTX
*mem_ctx
,
885 struct tevent_context
*ev
,
886 struct tdgram_context
*dgram
,
889 const struct tsocket_address
*dst
)
891 struct tevent_req
*req
;
892 struct tdgram_bsd_sendto_state
*state
;
893 struct tdgram_bsd
*bsds
= tdgram_context_data(dgram
, struct tdgram_bsd
);
896 req
= tevent_req_create(mem_ctx
, &state
,
897 struct tdgram_bsd_sendto_state
);
902 state
->dgram
= dgram
;
908 talloc_set_destructor(state
, tdgram_bsd_sendto_destructor
);
910 if (bsds
->fd
== -1) {
911 tevent_req_error(req
, ENOTCONN
);
916 * this is a fast path, not waiting for the
917 * socket to become explicit writeable gains
918 * about 10%-20% performance in benchmark tests.
920 tdgram_bsd_sendto_handler(req
);
921 if (!tevent_req_is_in_progress(req
)) {
925 ret
= tdgram_bsd_set_writeable_handler(bsds
, ev
,
926 tdgram_bsd_sendto_handler
,
929 tevent_req_error(req
, errno
);
936 tevent_req_post(req
, ev
);
940 static void tdgram_bsd_sendto_handler(void *private_data
)
942 struct tevent_req
*req
= talloc_get_type_abort(private_data
,
944 struct tdgram_bsd_sendto_state
*state
= tevent_req_data(req
,
945 struct tdgram_bsd_sendto_state
);
946 struct tdgram_context
*dgram
= state
->dgram
;
947 struct tdgram_bsd
*bsds
= tdgram_context_data(dgram
, struct tdgram_bsd
);
948 struct sockaddr
*sa
= NULL
;
949 socklen_t sa_socklen
= 0;
955 struct tsocket_address_bsd
*bsda
=
956 talloc_get_type(state
->dst
->private_data
,
957 struct tsocket_address_bsd
);
960 sa_socklen
= sizeof(bsda
->u
.ss
);
962 * for unix sockets we can't use the size of sockaddr_storage
963 * we would get EINVAL
965 if (bsda
->u
.sa
.sa_family
== AF_UNIX
) {
966 sa_socklen
= sizeof(bsda
->u
.un
);
970 ret
= sendto(bsds
->fd
, state
->buf
, state
->len
, 0, sa
, sa_socklen
);
971 err
= tsocket_bsd_error_from_errno(ret
, errno
, &retry
);
976 if (tevent_req_error(req
, err
)) {
982 tevent_req_done(req
);
985 static ssize_t
tdgram_bsd_sendto_recv(struct tevent_req
*req
, int *perrno
)
987 struct tdgram_bsd_sendto_state
*state
= tevent_req_data(req
,
988 struct tdgram_bsd_sendto_state
);
991 ret
= tsocket_simple_int_recv(req
, perrno
);
996 tevent_req_received(req
);
1000 struct tdgram_bsd_disconnect_state
{
1004 static struct tevent_req
*tdgram_bsd_disconnect_send(TALLOC_CTX
*mem_ctx
,
1005 struct tevent_context
*ev
,
1006 struct tdgram_context
*dgram
)
1008 struct tdgram_bsd
*bsds
= tdgram_context_data(dgram
, struct tdgram_bsd
);
1009 struct tevent_req
*req
;
1010 struct tdgram_bsd_disconnect_state
*state
;
1015 req
= tevent_req_create(mem_ctx
, &state
,
1016 struct tdgram_bsd_disconnect_state
);
1021 if (bsds
->fd
== -1) {
1022 tevent_req_error(req
, ENOTCONN
);
1026 ret
= close(bsds
->fd
);
1028 err
= tsocket_bsd_error_from_errno(ret
, errno
, &dummy
);
1029 if (tevent_req_error(req
, err
)) {
1033 tevent_req_done(req
);
1035 tevent_req_post(req
, ev
);
1039 static int tdgram_bsd_disconnect_recv(struct tevent_req
*req
,
1044 ret
= tsocket_simple_int_recv(req
, perrno
);
1046 tevent_req_received(req
);
1050 static const struct tdgram_context_ops tdgram_bsd_ops
= {
1053 .recvfrom_send
= tdgram_bsd_recvfrom_send
,
1054 .recvfrom_recv
= tdgram_bsd_recvfrom_recv
,
1056 .sendto_send
= tdgram_bsd_sendto_send
,
1057 .sendto_recv
= tdgram_bsd_sendto_recv
,
1059 .disconnect_send
= tdgram_bsd_disconnect_send
,
1060 .disconnect_recv
= tdgram_bsd_disconnect_recv
,
1063 static int tdgram_bsd_destructor(struct tdgram_bsd
*bsds
)
1065 TALLOC_FREE(bsds
->fde
);
1066 if (bsds
->fd
!= -1) {
1073 static int tdgram_bsd_dgram_socket(const struct tsocket_address
*local
,
1074 const struct tsocket_address
*remote
,
1076 TALLOC_CTX
*mem_ctx
,
1077 struct tdgram_context
**_dgram
,
1078 const char *location
)
1080 struct tsocket_address_bsd
*lbsda
=
1081 talloc_get_type_abort(local
->private_data
,
1082 struct tsocket_address_bsd
);
1083 struct tsocket_address_bsd
*rbsda
= NULL
;
1084 struct tdgram_context
*dgram
;
1085 struct tdgram_bsd
*bsds
;
1088 bool do_bind
= false;
1089 bool do_reuseaddr
= false;
1090 socklen_t sa_socklen
= sizeof(lbsda
->u
.ss
);
1093 rbsda
= talloc_get_type_abort(remote
->private_data
,
1094 struct tsocket_address_bsd
);
1097 switch (lbsda
->u
.sa
.sa_family
) {
1103 if (lbsda
->u
.un
.sun_path
[0] != 0) {
1104 do_reuseaddr
= true;
1108 * for unix sockets we can't use the size of sockaddr_storage
1109 * we would get EINVAL
1111 sa_socklen
= sizeof(lbsda
->u
.un
);
1114 if (lbsda
->u
.in
.sin_port
!= 0) {
1115 do_reuseaddr
= true;
1118 if (lbsda
->u
.in
.sin_addr
.s_addr
== INADDR_ANY
) {
1124 if (lbsda
->u
.in6
.sin6_port
!= 0) {
1125 do_reuseaddr
= true;
1128 if (memcmp(&in6addr_any
,
1129 &lbsda
->u
.in6
.sin6_addr
,
1130 sizeof(in6addr_any
)) != 0) {
1140 fd
= socket(lbsda
->u
.sa
.sa_family
, SOCK_DGRAM
, 0);
1145 fd
= tsocket_bsd_common_prepare_fd(fd
, true);
1150 dgram
= tdgram_context_create(mem_ctx
,
1156 int saved_errno
= errno
;
1158 errno
= saved_errno
;
1163 talloc_set_destructor(bsds
, tdgram_bsd_destructor
);
1168 ret
= setsockopt(fd
, SOL_SOCKET
, SO_BROADCAST
,
1169 (const void *)&val
, sizeof(val
));
1171 int saved_errno
= errno
;
1173 errno
= saved_errno
;
1181 ret
= setsockopt(fd
, SOL_SOCKET
, SO_REUSEADDR
,
1182 (const void *)&val
, sizeof(val
));
1184 int saved_errno
= errno
;
1186 errno
= saved_errno
;
1192 ret
= bind(fd
, &lbsda
->u
.sa
, sa_socklen
);
1194 int saved_errno
= errno
;
1196 errno
= saved_errno
;
1202 ret
= connect(fd
, &rbsda
->u
.sa
, sa_socklen
);
1204 int saved_errno
= errno
;
1206 errno
= saved_errno
;
1215 int _tdgram_inet_udp_socket(const struct tsocket_address
*local
,
1216 const struct tsocket_address
*remote
,
1217 TALLOC_CTX
*mem_ctx
,
1218 struct tdgram_context
**dgram
,
1219 const char *location
)
1221 struct tsocket_address_bsd
*lbsda
=
1222 talloc_get_type_abort(local
->private_data
,
1223 struct tsocket_address_bsd
);
1226 switch (lbsda
->u
.sa
.sa_family
) {
1238 ret
= tdgram_bsd_dgram_socket(local
, remote
, false,
1239 mem_ctx
, dgram
, location
);
1244 int _tdgram_unix_socket(const struct tsocket_address
*local
,
1245 const struct tsocket_address
*remote
,
1246 TALLOC_CTX
*mem_ctx
,
1247 struct tdgram_context
**dgram
,
1248 const char *location
)
1250 struct tsocket_address_bsd
*lbsda
=
1251 talloc_get_type_abort(local
->private_data
,
1252 struct tsocket_address_bsd
);
1255 switch (lbsda
->u
.sa
.sa_family
) {
1263 ret
= tdgram_bsd_dgram_socket(local
, remote
, false,
1264 mem_ctx
, dgram
, location
);
1269 struct tstream_bsd
{
1273 struct tevent_fd
*fde
;
1275 void *readable_private
;
1276 void (*readable_handler
)(void *private_data
);
1277 void *writeable_private
;
1278 void (*writeable_handler
)(void *private_data
);
1281 static void tstream_bsd_fde_handler(struct tevent_context
*ev
,
1282 struct tevent_fd
*fde
,
1286 struct tstream_bsd
*bsds
= talloc_get_type_abort(private_data
,
1287 struct tstream_bsd
);
1289 if (flags
& TEVENT_FD_WRITE
) {
1290 bsds
->writeable_handler(bsds
->writeable_private
);
1293 if (flags
& TEVENT_FD_READ
) {
1294 if (!bsds
->readable_handler
) {
1295 if (bsds
->writeable_handler
) {
1296 bsds
->writeable_handler(bsds
->writeable_private
);
1299 TEVENT_FD_NOT_READABLE(bsds
->fde
);
1302 bsds
->readable_handler(bsds
->readable_private
);
1307 static int tstream_bsd_set_readable_handler(struct tstream_bsd
*bsds
,
1308 struct tevent_context
*ev
,
1309 void (*handler
)(void *private_data
),
1317 if (!bsds
->readable_handler
) {
1320 bsds
->readable_handler
= NULL
;
1321 bsds
->readable_private
= NULL
;
1326 /* read and write must use the same tevent_context */
1327 if (bsds
->event_ptr
!= ev
) {
1328 if (bsds
->readable_handler
|| bsds
->writeable_handler
) {
1332 bsds
->event_ptr
= NULL
;
1333 TALLOC_FREE(bsds
->fde
);
1336 if (tevent_fd_get_flags(bsds
->fde
) == 0) {
1337 TALLOC_FREE(bsds
->fde
);
1339 bsds
->fde
= tevent_add_fd(ev
, bsds
,
1340 bsds
->fd
, TEVENT_FD_READ
,
1341 tstream_bsd_fde_handler
,
1348 /* cache the event context we're running on */
1349 bsds
->event_ptr
= ev
;
1350 } else if (!bsds
->readable_handler
) {
1351 TEVENT_FD_READABLE(bsds
->fde
);
1354 bsds
->readable_handler
= handler
;
1355 bsds
->readable_private
= private_data
;
1360 static int tstream_bsd_set_writeable_handler(struct tstream_bsd
*bsds
,
1361 struct tevent_context
*ev
,
1362 void (*handler
)(void *private_data
),
1370 if (!bsds
->writeable_handler
) {
1373 bsds
->writeable_handler
= NULL
;
1374 bsds
->writeable_private
= NULL
;
1375 TEVENT_FD_NOT_WRITEABLE(bsds
->fde
);
1380 /* read and write must use the same tevent_context */
1381 if (bsds
->event_ptr
!= ev
) {
1382 if (bsds
->readable_handler
|| bsds
->writeable_handler
) {
1386 bsds
->event_ptr
= NULL
;
1387 TALLOC_FREE(bsds
->fde
);
1390 if (tevent_fd_get_flags(bsds
->fde
) == 0) {
1391 TALLOC_FREE(bsds
->fde
);
1393 bsds
->fde
= tevent_add_fd(ev
, bsds
,
1395 TEVENT_FD_READ
| TEVENT_FD_WRITE
,
1396 tstream_bsd_fde_handler
,
1403 /* cache the event context we're running on */
1404 bsds
->event_ptr
= ev
;
1405 } else if (!bsds
->writeable_handler
) {
1406 uint16_t flags
= tevent_fd_get_flags(bsds
->fde
);
1407 flags
|= TEVENT_FD_READ
| TEVENT_FD_WRITE
;
1408 tevent_fd_set_flags(bsds
->fde
, flags
);
1411 bsds
->writeable_handler
= handler
;
1412 bsds
->writeable_private
= private_data
;
1417 static ssize_t
tstream_bsd_pending_bytes(struct tstream_context
*stream
)
1419 struct tstream_bsd
*bsds
= tstream_context_data(stream
,
1420 struct tstream_bsd
);
1423 if (bsds
->fd
== -1) {
1428 ret
= tsocket_bsd_pending(bsds
->fd
);
1433 struct tstream_bsd_readv_state
{
1434 struct tstream_context
*stream
;
1436 struct iovec
*vector
;
1442 static int tstream_bsd_readv_destructor(struct tstream_bsd_readv_state
*state
)
1444 struct tstream_bsd
*bsds
= tstream_context_data(state
->stream
,
1445 struct tstream_bsd
);
1447 tstream_bsd_set_readable_handler(bsds
, NULL
, NULL
, NULL
);
1452 static void tstream_bsd_readv_handler(void *private_data
);
1454 static struct tevent_req
*tstream_bsd_readv_send(TALLOC_CTX
*mem_ctx
,
1455 struct tevent_context
*ev
,
1456 struct tstream_context
*stream
,
1457 struct iovec
*vector
,
1460 struct tevent_req
*req
;
1461 struct tstream_bsd_readv_state
*state
;
1462 struct tstream_bsd
*bsds
= tstream_context_data(stream
, struct tstream_bsd
);
1465 req
= tevent_req_create(mem_ctx
, &state
,
1466 struct tstream_bsd_readv_state
);
1471 state
->stream
= stream
;
1472 /* we make a copy of the vector so that we can modify it */
1473 state
->vector
= talloc_array(state
, struct iovec
, count
);
1474 if (tevent_req_nomem(state
->vector
, req
)) {
1477 memcpy(state
->vector
, vector
, sizeof(struct iovec
)*count
);
1478 state
->count
= count
;
1481 talloc_set_destructor(state
, tstream_bsd_readv_destructor
);
1483 if (bsds
->fd
== -1) {
1484 tevent_req_error(req
, ENOTCONN
);
1489 * this is a fast path, not waiting for the
1490 * socket to become explicit readable gains
1491 * about 10%-20% performance in benchmark tests.
1493 tstream_bsd_readv_handler(req
);
1494 if (!tevent_req_is_in_progress(req
)) {
1498 ret
= tstream_bsd_set_readable_handler(bsds
, ev
,
1499 tstream_bsd_readv_handler
,
1502 tevent_req_error(req
, errno
);
1509 tevent_req_post(req
, ev
);
1513 static void tstream_bsd_readv_handler(void *private_data
)
1515 struct tevent_req
*req
= talloc_get_type_abort(private_data
,
1517 struct tstream_bsd_readv_state
*state
= tevent_req_data(req
,
1518 struct tstream_bsd_readv_state
);
1519 struct tstream_context
*stream
= state
->stream
;
1520 struct tstream_bsd
*bsds
= tstream_context_data(stream
, struct tstream_bsd
);
1525 ret
= readv(bsds
->fd
, state
->vector
, state
->count
);
1527 /* propagate end of file */
1528 tevent_req_error(req
, EPIPE
);
1531 err
= tsocket_bsd_error_from_errno(ret
, errno
, &retry
);
1536 if (tevent_req_error(req
, err
)) {
1543 if (ret
< state
->vector
[0].iov_len
) {
1545 base
= (uint8_t *)state
->vector
[0].iov_base
;
1547 state
->vector
[0].iov_base
= base
;
1548 state
->vector
[0].iov_len
-= ret
;
1551 ret
-= state
->vector
[0].iov_len
;
1557 * there're maybe some empty vectors at the end
1558 * which we need to skip, otherwise we would get
1559 * ret == 0 from the readv() call and return EPIPE
1561 while (state
->count
> 0) {
1562 if (state
->vector
[0].iov_len
> 0) {
1569 if (state
->count
> 0) {
1570 /* we have more to read */
1574 tevent_req_done(req
);
1577 static int tstream_bsd_readv_recv(struct tevent_req
*req
,
1580 struct tstream_bsd_readv_state
*state
= tevent_req_data(req
,
1581 struct tstream_bsd_readv_state
);
1584 ret
= tsocket_simple_int_recv(req
, perrno
);
1589 tevent_req_received(req
);
1593 struct tstream_bsd_writev_state
{
1594 struct tstream_context
*stream
;
1596 struct iovec
*vector
;
1602 static int tstream_bsd_writev_destructor(struct tstream_bsd_writev_state
*state
)
1604 struct tstream_bsd
*bsds
= tstream_context_data(state
->stream
,
1605 struct tstream_bsd
);
1607 tstream_bsd_set_writeable_handler(bsds
, NULL
, NULL
, NULL
);
1612 static void tstream_bsd_writev_handler(void *private_data
);
1614 static struct tevent_req
*tstream_bsd_writev_send(TALLOC_CTX
*mem_ctx
,
1615 struct tevent_context
*ev
,
1616 struct tstream_context
*stream
,
1617 const struct iovec
*vector
,
1620 struct tevent_req
*req
;
1621 struct tstream_bsd_writev_state
*state
;
1622 struct tstream_bsd
*bsds
= tstream_context_data(stream
, struct tstream_bsd
);
1625 req
= tevent_req_create(mem_ctx
, &state
,
1626 struct tstream_bsd_writev_state
);
1631 state
->stream
= stream
;
1632 /* we make a copy of the vector so that we can modify it */
1633 state
->vector
= talloc_array(state
, struct iovec
, count
);
1634 if (tevent_req_nomem(state
->vector
, req
)) {
1637 memcpy(state
->vector
, vector
, sizeof(struct iovec
)*count
);
1638 state
->count
= count
;
1641 talloc_set_destructor(state
, tstream_bsd_writev_destructor
);
1643 if (bsds
->fd
== -1) {
1644 tevent_req_error(req
, ENOTCONN
);
1649 * this is a fast path, not waiting for the
1650 * socket to become explicit writeable gains
1651 * about 10%-20% performance in benchmark tests.
1653 tstream_bsd_writev_handler(req
);
1654 if (!tevent_req_is_in_progress(req
)) {
1658 ret
= tstream_bsd_set_writeable_handler(bsds
, ev
,
1659 tstream_bsd_writev_handler
,
1662 tevent_req_error(req
, errno
);
1669 tevent_req_post(req
, ev
);
1673 static void tstream_bsd_writev_handler(void *private_data
)
1675 struct tevent_req
*req
= talloc_get_type_abort(private_data
,
1677 struct tstream_bsd_writev_state
*state
= tevent_req_data(req
,
1678 struct tstream_bsd_writev_state
);
1679 struct tstream_context
*stream
= state
->stream
;
1680 struct tstream_bsd
*bsds
= tstream_context_data(stream
, struct tstream_bsd
);
1685 ret
= writev(bsds
->fd
, state
->vector
, state
->count
);
1687 /* propagate end of file */
1688 tevent_req_error(req
, EPIPE
);
1691 err
= tsocket_bsd_error_from_errno(ret
, errno
, &retry
);
1696 if (tevent_req_error(req
, err
)) {
1703 if (ret
< state
->vector
[0].iov_len
) {
1705 base
= (uint8_t *)state
->vector
[0].iov_base
;
1707 state
->vector
[0].iov_base
= base
;
1708 state
->vector
[0].iov_len
-= ret
;
1711 ret
-= state
->vector
[0].iov_len
;
1717 * there're maybe some empty vectors at the end
1718 * which we need to skip, otherwise we would get
1719 * ret == 0 from the writev() call and return EPIPE
1721 while (state
->count
> 0) {
1722 if (state
->vector
[0].iov_len
> 0) {
1729 if (state
->count
> 0) {
1730 /* we have more to read */
1734 tevent_req_done(req
);
1737 static int tstream_bsd_writev_recv(struct tevent_req
*req
, int *perrno
)
1739 struct tstream_bsd_writev_state
*state
= tevent_req_data(req
,
1740 struct tstream_bsd_writev_state
);
1743 ret
= tsocket_simple_int_recv(req
, perrno
);
1748 tevent_req_received(req
);
1752 struct tstream_bsd_disconnect_state
{
1756 static struct tevent_req
*tstream_bsd_disconnect_send(TALLOC_CTX
*mem_ctx
,
1757 struct tevent_context
*ev
,
1758 struct tstream_context
*stream
)
1760 struct tstream_bsd
*bsds
= tstream_context_data(stream
, struct tstream_bsd
);
1761 struct tevent_req
*req
;
1762 struct tstream_bsd_disconnect_state
*state
;
1767 req
= tevent_req_create(mem_ctx
, &state
,
1768 struct tstream_bsd_disconnect_state
);
1773 if (bsds
->fd
== -1) {
1774 tevent_req_error(req
, ENOTCONN
);
1778 ret
= close(bsds
->fd
);
1780 err
= tsocket_bsd_error_from_errno(ret
, errno
, &dummy
);
1781 if (tevent_req_error(req
, err
)) {
1785 tevent_req_done(req
);
1787 tevent_req_post(req
, ev
);
1791 static int tstream_bsd_disconnect_recv(struct tevent_req
*req
,
1796 ret
= tsocket_simple_int_recv(req
, perrno
);
1798 tevent_req_received(req
);
1802 static const struct tstream_context_ops tstream_bsd_ops
= {
1805 .pending_bytes
= tstream_bsd_pending_bytes
,
1807 .readv_send
= tstream_bsd_readv_send
,
1808 .readv_recv
= tstream_bsd_readv_recv
,
1810 .writev_send
= tstream_bsd_writev_send
,
1811 .writev_recv
= tstream_bsd_writev_recv
,
1813 .disconnect_send
= tstream_bsd_disconnect_send
,
1814 .disconnect_recv
= tstream_bsd_disconnect_recv
,
1817 static int tstream_bsd_destructor(struct tstream_bsd
*bsds
)
1819 TALLOC_FREE(bsds
->fde
);
1820 if (bsds
->fd
!= -1) {
1827 int _tstream_bsd_existing_socket(TALLOC_CTX
*mem_ctx
,
1829 struct tstream_context
**_stream
,
1830 const char *location
)
1832 struct tstream_context
*stream
;
1833 struct tstream_bsd
*bsds
;
1835 stream
= tstream_context_create(mem_ctx
,
1845 talloc_set_destructor(bsds
, tstream_bsd_destructor
);
1851 struct tstream_bsd_connect_state
{
1853 struct tevent_fd
*fde
;
1854 struct tstream_conext
*stream
;
1857 static int tstream_bsd_connect_destructor(struct tstream_bsd_connect_state
*state
)
1859 TALLOC_FREE(state
->fde
);
1860 if (state
->fd
!= -1) {
1868 static void tstream_bsd_connect_fde_handler(struct tevent_context
*ev
,
1869 struct tevent_fd
*fde
,
1871 void *private_data
);
1873 static struct tevent_req
* tstream_bsd_connect_send(TALLOC_CTX
*mem_ctx
,
1874 struct tevent_context
*ev
,
1876 const struct tsocket_address
*local
,
1877 const struct tsocket_address
*remote
)
1879 struct tevent_req
*req
;
1880 struct tstream_bsd_connect_state
*state
;
1881 struct tsocket_address_bsd
*lbsda
=
1882 talloc_get_type_abort(local
->private_data
,
1883 struct tsocket_address_bsd
);
1884 struct tsocket_address_bsd
*rbsda
=
1885 talloc_get_type_abort(remote
->private_data
,
1886 struct tsocket_address_bsd
);
1890 bool do_bind
= false;
1891 bool do_reuseaddr
= false;
1892 socklen_t sa_socklen
= sizeof(rbsda
->u
.ss
);
1894 req
= tevent_req_create(mem_ctx
, &state
,
1895 struct tstream_bsd_connect_state
);
1902 talloc_set_destructor(state
, tstream_bsd_connect_destructor
);
1904 /* give the wrappers a chance to report an error */
1905 if (sys_errno
!= 0) {
1906 tevent_req_error(req
, sys_errno
);
1910 switch (lbsda
->u
.sa
.sa_family
) {
1912 if (lbsda
->u
.un
.sun_path
[0] != 0) {
1913 do_reuseaddr
= true;
1917 * for unix sockets we can't use the size of sockaddr_storage
1918 * we would get EINVAL
1920 sa_socklen
= sizeof(rbsda
->u
.un
);
1923 if (lbsda
->u
.in
.sin_port
!= 0) {
1924 do_reuseaddr
= true;
1927 if (lbsda
->u
.in
.sin_addr
.s_addr
== INADDR_ANY
) {
1933 if (lbsda
->u
.in6
.sin6_port
!= 0) {
1934 do_reuseaddr
= true;
1937 if (memcmp(&in6addr_any
,
1938 &lbsda
->u
.in6
.sin6_addr
,
1939 sizeof(in6addr_any
)) != 0) {
1945 tevent_req_error(req
, EINVAL
);
1949 state
->fd
= socket(lbsda
->u
.sa
.sa_family
, SOCK_STREAM
, 0);
1950 if (state
->fd
== -1) {
1951 tevent_req_error(req
, errno
);
1955 state
->fd
= tsocket_bsd_common_prepare_fd(state
->fd
, true);
1956 if (state
->fd
== -1) {
1957 tevent_req_error(req
, errno
);
1964 ret
= setsockopt(state
->fd
, SOL_SOCKET
, SO_REUSEADDR
,
1965 (const void *)&val
, sizeof(val
));
1967 tevent_req_error(req
, errno
);
1973 ret
= bind(state
->fd
, &lbsda
->u
.sa
, sizeof(lbsda
->u
.ss
));
1975 tevent_req_error(req
, errno
);
1980 ret
= connect(state
->fd
, &rbsda
->u
.sa
, sa_socklen
);
1981 err
= tsocket_bsd_error_from_errno(ret
, errno
, &retry
);
1986 if (tevent_req_error(req
, err
)) {
1990 tevent_req_done(req
);
1994 state
->fde
= tevent_add_fd(ev
, state
,
1996 TEVENT_FD_READ
| TEVENT_FD_WRITE
,
1997 tstream_bsd_connect_fde_handler
,
1999 if (tevent_req_nomem(state
->fde
, req
)) {
2006 tevent_req_post(req
, ev
);
2010 static void tstream_bsd_connect_fde_handler(struct tevent_context
*ev
,
2011 struct tevent_fd
*fde
,
2015 struct tevent_req
*req
= talloc_get_type_abort(private_data
,
2017 struct tstream_bsd_connect_state
*state
= tevent_req_data(req
,
2018 struct tstream_bsd_connect_state
);
2021 socklen_t len
= sizeof(error
);
2025 ret
= getsockopt(state
->fd
, SOL_SOCKET
, SO_ERROR
, &error
, &len
);
2032 err
= tsocket_bsd_error_from_errno(ret
, errno
, &retry
);
2037 if (tevent_req_error(req
, err
)) {
2041 tevent_req_done(req
);
2044 static int tstream_bsd_connect_recv(struct tevent_req
*req
,
2046 TALLOC_CTX
*mem_ctx
,
2047 struct tstream_context
**stream
,
2048 const char *location
)
2050 struct tstream_bsd_connect_state
*state
= tevent_req_data(req
,
2051 struct tstream_bsd_connect_state
);
2054 ret
= tsocket_simple_int_recv(req
, perrno
);
2056 ret
= _tstream_bsd_existing_socket(mem_ctx
,
2064 TALLOC_FREE(state
->fde
);
2069 tevent_req_received(req
);
2073 struct tevent_req
* tstream_inet_tcp_connect_send(TALLOC_CTX
*mem_ctx
,
2074 struct tevent_context
*ev
,
2075 const struct tsocket_address
*local
,
2076 const struct tsocket_address
*remote
)
2078 struct tsocket_address_bsd
*lbsda
=
2079 talloc_get_type_abort(local
->private_data
,
2080 struct tsocket_address_bsd
);
2081 struct tevent_req
*req
;
2084 switch (lbsda
->u
.sa
.sa_family
) {
2096 req
= tstream_bsd_connect_send(mem_ctx
, ev
, sys_errno
, local
, remote
);
2101 int _tstream_inet_tcp_connect_recv(struct tevent_req
*req
,
2103 TALLOC_CTX
*mem_ctx
,
2104 struct tstream_context
**stream
,
2105 const char *location
)
2107 return tstream_bsd_connect_recv(req
, perrno
, mem_ctx
, stream
, location
);
2110 struct tevent_req
* tstream_unix_connect_send(TALLOC_CTX
*mem_ctx
,
2111 struct tevent_context
*ev
,
2112 const struct tsocket_address
*local
,
2113 const struct tsocket_address
*remote
)
2115 struct tsocket_address_bsd
*lbsda
=
2116 talloc_get_type_abort(local
->private_data
,
2117 struct tsocket_address_bsd
);
2118 struct tevent_req
*req
;
2121 switch (lbsda
->u
.sa
.sa_family
) {
2129 req
= tstream_bsd_connect_send(mem_ctx
, ev
, sys_errno
, local
, remote
);
2134 int _tstream_unix_connect_recv(struct tevent_req
*req
,
2136 TALLOC_CTX
*mem_ctx
,
2137 struct tstream_context
**stream
,
2138 const char *location
)
2140 return tstream_bsd_connect_recv(req
, perrno
, mem_ctx
, stream
, location
);
2143 int _tstream_unix_socketpair(TALLOC_CTX
*mem_ctx1
,
2144 struct tstream_context
**_stream1
,
2145 TALLOC_CTX
*mem_ctx2
,
2146 struct tstream_context
**_stream2
,
2147 const char *location
)
2153 struct tstream_context
*stream1
= NULL
;
2154 struct tstream_context
*stream2
= NULL
;
2156 ret
= socketpair(AF_UNIX
, SOCK_STREAM
, 0, fds
);
2163 fd1
= tsocket_bsd_common_prepare_fd(fd1
, true);
2165 int sys_errno
= errno
;
2171 fd2
= tsocket_bsd_common_prepare_fd(fd2
, true);
2173 int sys_errno
= errno
;
2179 ret
= _tstream_bsd_existing_socket(mem_ctx1
,
2184 int sys_errno
= errno
;
2191 ret
= _tstream_bsd_existing_socket(mem_ctx2
,
2196 int sys_errno
= errno
;
2197 talloc_free(stream1
);
2203 *_stream1
= stream1
;
2204 *_stream2
= stream2
;