2 Unix SMB/CIFS implementation.
4 Copyright (C) Stefan Metzmacher 2009
6 ** NOTE! The following LGPL license applies to the tsocket
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 int _tsocket_address_bsd_from_sockaddr(TALLOC_CTX
*mem_ctx
,
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 ssize_t
tsocket_address_bsd_sockaddr(const struct tsocket_address
*addr
,
266 struct tsocket_address_bsd
*bsda
= talloc_get_type(addr
->private_data
,
267 struct tsocket_address_bsd
);
275 switch (bsda
->u
.sa
.sa_family
) {
277 rlen
= sizeof(struct sockaddr_un
);
280 rlen
= sizeof(struct sockaddr_in
);
284 rlen
= sizeof(struct sockaddr_in6
);
288 errno
= EAFNOSUPPORT
;
292 if (sa_socklen
< rlen
) {
297 if (sa_socklen
> sizeof(struct sockaddr_storage
)) {
298 memset(sa
, 0, sa_socklen
);
299 sa_socklen
= sizeof(struct sockaddr_storage
);
302 memcpy(sa
, &bsda
->u
.ss
, sa_socklen
);
306 int _tsocket_address_inet_from_strings(TALLOC_CTX
*mem_ctx
,
310 struct tsocket_address
**_addr
,
311 const char *location
)
313 struct addrinfo hints
;
314 struct addrinfo
*result
= NULL
;
320 * we use SOCKET_STREAM here to get just one result
321 * back from getaddrinfo().
323 hints
.ai_socktype
= SOCK_STREAM
;
324 hints
.ai_flags
= AI_NUMERICHOST
| AI_NUMERICSERV
;
326 if (strcasecmp(fam
, "ip") == 0) {
327 hints
.ai_family
= AF_UNSPEC
;
335 } else if (strcasecmp(fam
, "ipv4") == 0) {
336 hints
.ai_family
= AF_INET
;
341 } else if (strcasecmp(fam
, "ipv6") == 0) {
342 hints
.ai_family
= AF_INET6
;
348 errno
= EAFNOSUPPORT
;
352 snprintf(port_str
, sizeof(port_str
) - 1, "%u", port
);
354 ret
= getaddrinfo(addr
, port_str
, &hints
, &result
);
365 if (result
->ai_socktype
!= SOCK_STREAM
) {
371 ret
= _tsocket_address_bsd_from_sockaddr(mem_ctx
,
379 freeaddrinfo(result
);
384 char *tsocket_address_inet_addr_string(const struct tsocket_address
*addr
,
387 struct tsocket_address_bsd
*bsda
= talloc_get_type(addr
->private_data
,
388 struct tsocket_address_bsd
);
389 char addr_str
[INET6_ADDRSTRLEN
+1];
397 switch (bsda
->u
.sa
.sa_family
) {
399 str
= inet_ntop(bsda
->u
.in
.sin_family
,
400 &bsda
->u
.in
.sin_addr
,
401 addr_str
, sizeof(addr_str
));
405 str
= inet_ntop(bsda
->u
.in6
.sin6_family
,
406 &bsda
->u
.in6
.sin6_addr
,
407 addr_str
, sizeof(addr_str
));
419 return talloc_strdup(mem_ctx
, str
);
422 uint16_t tsocket_address_inet_port(const struct tsocket_address
*addr
)
424 struct tsocket_address_bsd
*bsda
= talloc_get_type(addr
->private_data
,
425 struct tsocket_address_bsd
);
433 switch (bsda
->u
.sa
.sa_family
) {
435 port
= ntohs(bsda
->u
.in
.sin_port
);
439 port
= ntohs(bsda
->u
.in6
.sin6_port
);
450 int tsocket_address_inet_set_port(struct tsocket_address
*addr
,
453 struct tsocket_address_bsd
*bsda
= talloc_get_type(addr
->private_data
,
454 struct tsocket_address_bsd
);
461 switch (bsda
->u
.sa
.sa_family
) {
463 bsda
->u
.in
.sin_port
= htons(port
);
467 bsda
->u
.in6
.sin6_port
= htons(port
);
478 int _tsocket_address_unix_from_path(TALLOC_CTX
*mem_ctx
,
480 struct tsocket_address
**_addr
,
481 const char *location
)
483 struct sockaddr_un un
;
491 if (strlen(path
) > sizeof(un
.sun_path
)-1) {
492 errno
= ENAMETOOLONG
;
497 un
.sun_family
= AF_UNIX
;
498 strncpy(un
.sun_path
, path
, sizeof(un
.sun_path
)-1);
500 ret
= _tsocket_address_bsd_from_sockaddr(mem_ctx
,
501 (struct sockaddr
*)p
,
509 char *tsocket_address_unix_path(const struct tsocket_address
*addr
,
512 struct tsocket_address_bsd
*bsda
= talloc_get_type(addr
->private_data
,
513 struct tsocket_address_bsd
);
521 switch (bsda
->u
.sa
.sa_family
) {
523 str
= bsda
->u
.un
.sun_path
;
530 return talloc_strdup(mem_ctx
, str
);
533 static char *tsocket_address_bsd_string(const struct tsocket_address
*addr
,
536 struct tsocket_address_bsd
*bsda
= talloc_get_type(addr
->private_data
,
537 struct tsocket_address_bsd
);
540 const char *prefix
= NULL
;
543 switch (bsda
->u
.sa
.sa_family
) {
545 return talloc_asprintf(mem_ctx
, "unix:%s",
546 bsda
->u
.un
.sun_path
);
560 addr_str
= tsocket_address_inet_addr_string(addr
, mem_ctx
);
565 port
= tsocket_address_inet_port(addr
);
567 str
= talloc_asprintf(mem_ctx
, "%s:%s:%u",
568 prefix
, addr_str
, port
);
569 talloc_free(addr_str
);
574 static struct tsocket_address
*tsocket_address_bsd_copy(const struct tsocket_address
*addr
,
576 const char *location
)
578 struct tsocket_address_bsd
*bsda
= talloc_get_type(addr
->private_data
,
579 struct tsocket_address_bsd
);
580 struct tsocket_address
*copy
;
583 ret
= _tsocket_address_bsd_from_sockaddr(mem_ctx
,
595 static const struct tsocket_address_ops tsocket_address_bsd_ops
= {
597 .string
= tsocket_address_bsd_string
,
598 .copy
= tsocket_address_bsd_copy
,
605 struct tevent_fd
*fde
;
607 void *readable_private
;
608 void (*readable_handler
)(void *private_data
);
609 void *writeable_private
;
610 void (*writeable_handler
)(void *private_data
);
613 static void tdgram_bsd_fde_handler(struct tevent_context
*ev
,
614 struct tevent_fd
*fde
,
618 struct tdgram_bsd
*bsds
= talloc_get_type_abort(private_data
,
621 if (flags
& TEVENT_FD_WRITE
) {
622 bsds
->writeable_handler(bsds
->writeable_private
);
625 if (flags
& TEVENT_FD_READ
) {
626 if (!bsds
->readable_handler
) {
627 TEVENT_FD_NOT_READABLE(bsds
->fde
);
630 bsds
->readable_handler(bsds
->readable_private
);
635 static int tdgram_bsd_set_readable_handler(struct tdgram_bsd
*bsds
,
636 struct tevent_context
*ev
,
637 void (*handler
)(void *private_data
),
645 if (!bsds
->readable_handler
) {
648 bsds
->readable_handler
= NULL
;
649 bsds
->readable_private
= NULL
;
654 /* read and write must use the same tevent_context */
655 if (bsds
->event_ptr
!= ev
) {
656 if (bsds
->readable_handler
|| bsds
->writeable_handler
) {
660 bsds
->event_ptr
= NULL
;
661 TALLOC_FREE(bsds
->fde
);
664 if (tevent_fd_get_flags(bsds
->fde
) == 0) {
665 TALLOC_FREE(bsds
->fde
);
667 bsds
->fde
= tevent_add_fd(ev
, bsds
,
668 bsds
->fd
, TEVENT_FD_READ
,
669 tdgram_bsd_fde_handler
,
676 /* cache the event context we're running on */
677 bsds
->event_ptr
= ev
;
678 } else if (!bsds
->readable_handler
) {
679 TEVENT_FD_READABLE(bsds
->fde
);
682 bsds
->readable_handler
= handler
;
683 bsds
->readable_private
= private_data
;
688 static int tdgram_bsd_set_writeable_handler(struct tdgram_bsd
*bsds
,
689 struct tevent_context
*ev
,
690 void (*handler
)(void *private_data
),
698 if (!bsds
->writeable_handler
) {
701 bsds
->writeable_handler
= NULL
;
702 bsds
->writeable_private
= NULL
;
703 TEVENT_FD_NOT_WRITEABLE(bsds
->fde
);
708 /* read and write must use the same tevent_context */
709 if (bsds
->event_ptr
!= ev
) {
710 if (bsds
->readable_handler
|| bsds
->writeable_handler
) {
714 bsds
->event_ptr
= NULL
;
715 TALLOC_FREE(bsds
->fde
);
718 if (tevent_fd_get_flags(bsds
->fde
) == 0) {
719 TALLOC_FREE(bsds
->fde
);
721 bsds
->fde
= tevent_add_fd(ev
, bsds
,
722 bsds
->fd
, TEVENT_FD_WRITE
,
723 tdgram_bsd_fde_handler
,
730 /* cache the event context we're running on */
731 bsds
->event_ptr
= ev
;
732 } else if (!bsds
->writeable_handler
) {
733 TEVENT_FD_WRITEABLE(bsds
->fde
);
736 bsds
->writeable_handler
= handler
;
737 bsds
->writeable_private
= private_data
;
742 struct tdgram_bsd_recvfrom_state
{
743 struct tdgram_context
*dgram
;
747 struct tsocket_address
*src
;
750 static int tdgram_bsd_recvfrom_destructor(struct tdgram_bsd_recvfrom_state
*state
)
752 struct tdgram_bsd
*bsds
= tdgram_context_data(state
->dgram
,
755 tdgram_bsd_set_readable_handler(bsds
, NULL
, NULL
, NULL
);
760 static void tdgram_bsd_recvfrom_handler(void *private_data
);
762 static struct tevent_req
*tdgram_bsd_recvfrom_send(TALLOC_CTX
*mem_ctx
,
763 struct tevent_context
*ev
,
764 struct tdgram_context
*dgram
)
766 struct tevent_req
*req
;
767 struct tdgram_bsd_recvfrom_state
*state
;
768 struct tdgram_bsd
*bsds
= tdgram_context_data(dgram
, struct tdgram_bsd
);
771 req
= tevent_req_create(mem_ctx
, &state
,
772 struct tdgram_bsd_recvfrom_state
);
777 state
->dgram
= dgram
;
782 talloc_set_destructor(state
, tdgram_bsd_recvfrom_destructor
);
784 if (bsds
->fd
== -1) {
785 tevent_req_error(req
, ENOTCONN
);
790 * this is a fast path, not waiting for the
791 * socket to become explicit readable gains
792 * about 10%-20% performance in benchmark tests.
794 tdgram_bsd_recvfrom_handler(req
);
795 if (!tevent_req_is_in_progress(req
)) {
799 ret
= tdgram_bsd_set_readable_handler(bsds
, ev
,
800 tdgram_bsd_recvfrom_handler
,
803 tevent_req_error(req
, errno
);
810 tevent_req_post(req
, ev
);
814 static void tdgram_bsd_recvfrom_handler(void *private_data
)
816 struct tevent_req
*req
= talloc_get_type_abort(private_data
,
818 struct tdgram_bsd_recvfrom_state
*state
= tevent_req_data(req
,
819 struct tdgram_bsd_recvfrom_state
);
820 struct tdgram_context
*dgram
= state
->dgram
;
821 struct tdgram_bsd
*bsds
= tdgram_context_data(dgram
, struct tdgram_bsd
);
822 struct tsocket_address_bsd
*bsda
;
824 struct sockaddr
*sa
= NULL
;
825 socklen_t sa_socklen
= 0;
829 ret
= tsocket_bsd_pending(bsds
->fd
);
834 err
= tsocket_bsd_error_from_errno(ret
, errno
, &retry
);
839 if (tevent_req_error(req
, err
)) {
843 state
->buf
= talloc_array(state
, uint8_t, ret
);
844 if (tevent_req_nomem(state
->buf
, req
)) {
849 state
->src
= tsocket_address_create(state
,
850 &tsocket_address_bsd_ops
,
852 struct tsocket_address_bsd
,
853 __location__
"bsd_recvfrom");
854 if (tevent_req_nomem(state
->src
, req
)) {
861 sa_socklen
= sizeof(bsda
->u
.ss
);
863 * for unix sockets we can't use the size of sockaddr_storage
864 * we would get EINVAL
866 if (bsda
->u
.sa
.sa_family
== AF_UNIX
) {
867 sa_socklen
= sizeof(bsda
->u
.un
);
870 ret
= recvfrom(bsds
->fd
, state
->buf
, state
->len
, 0, sa
, &sa_socklen
);
871 err
= tsocket_bsd_error_from_errno(ret
, errno
, &retry
);
876 if (tevent_req_error(req
, err
)) {
880 if (ret
!= state
->len
) {
881 tevent_req_error(req
, EIO
);
885 tevent_req_done(req
);
888 static ssize_t
tdgram_bsd_recvfrom_recv(struct tevent_req
*req
,
892 struct tsocket_address
**src
)
894 struct tdgram_bsd_recvfrom_state
*state
= tevent_req_data(req
,
895 struct tdgram_bsd_recvfrom_state
);
898 ret
= tsocket_simple_int_recv(req
, perrno
);
900 *buf
= talloc_move(mem_ctx
, &state
->buf
);
903 *src
= talloc_move(mem_ctx
, &state
->src
);
907 tevent_req_received(req
);
911 struct tdgram_bsd_sendto_state
{
912 struct tdgram_context
*dgram
;
916 const struct tsocket_address
*dst
;
921 static int tdgram_bsd_sendto_destructor(struct tdgram_bsd_sendto_state
*state
)
923 struct tdgram_bsd
*bsds
= tdgram_context_data(state
->dgram
,
926 tdgram_bsd_set_writeable_handler(bsds
, NULL
, NULL
, NULL
);
931 static void tdgram_bsd_sendto_handler(void *private_data
);
933 static struct tevent_req
*tdgram_bsd_sendto_send(TALLOC_CTX
*mem_ctx
,
934 struct tevent_context
*ev
,
935 struct tdgram_context
*dgram
,
938 const struct tsocket_address
*dst
)
940 struct tevent_req
*req
;
941 struct tdgram_bsd_sendto_state
*state
;
942 struct tdgram_bsd
*bsds
= tdgram_context_data(dgram
, struct tdgram_bsd
);
945 req
= tevent_req_create(mem_ctx
, &state
,
946 struct tdgram_bsd_sendto_state
);
951 state
->dgram
= dgram
;
957 talloc_set_destructor(state
, tdgram_bsd_sendto_destructor
);
959 if (bsds
->fd
== -1) {
960 tevent_req_error(req
, ENOTCONN
);
965 * this is a fast path, not waiting for the
966 * socket to become explicit writeable gains
967 * about 10%-20% performance in benchmark tests.
969 tdgram_bsd_sendto_handler(req
);
970 if (!tevent_req_is_in_progress(req
)) {
974 ret
= tdgram_bsd_set_writeable_handler(bsds
, ev
,
975 tdgram_bsd_sendto_handler
,
978 tevent_req_error(req
, errno
);
985 tevent_req_post(req
, ev
);
989 static void tdgram_bsd_sendto_handler(void *private_data
)
991 struct tevent_req
*req
= talloc_get_type_abort(private_data
,
993 struct tdgram_bsd_sendto_state
*state
= tevent_req_data(req
,
994 struct tdgram_bsd_sendto_state
);
995 struct tdgram_context
*dgram
= state
->dgram
;
996 struct tdgram_bsd
*bsds
= tdgram_context_data(dgram
, struct tdgram_bsd
);
997 struct sockaddr
*sa
= NULL
;
998 socklen_t sa_socklen
= 0;
1004 struct tsocket_address_bsd
*bsda
=
1005 talloc_get_type(state
->dst
->private_data
,
1006 struct tsocket_address_bsd
);
1009 sa_socklen
= sizeof(bsda
->u
.ss
);
1011 * for unix sockets we can't use the size of sockaddr_storage
1012 * we would get EINVAL
1014 if (bsda
->u
.sa
.sa_family
== AF_UNIX
) {
1015 sa_socklen
= sizeof(bsda
->u
.un
);
1019 ret
= sendto(bsds
->fd
, state
->buf
, state
->len
, 0, sa
, sa_socklen
);
1020 err
= tsocket_bsd_error_from_errno(ret
, errno
, &retry
);
1025 if (tevent_req_error(req
, err
)) {
1031 tevent_req_done(req
);
1034 static ssize_t
tdgram_bsd_sendto_recv(struct tevent_req
*req
, int *perrno
)
1036 struct tdgram_bsd_sendto_state
*state
= tevent_req_data(req
,
1037 struct tdgram_bsd_sendto_state
);
1040 ret
= tsocket_simple_int_recv(req
, perrno
);
1045 tevent_req_received(req
);
1049 struct tdgram_bsd_disconnect_state
{
1053 static struct tevent_req
*tdgram_bsd_disconnect_send(TALLOC_CTX
*mem_ctx
,
1054 struct tevent_context
*ev
,
1055 struct tdgram_context
*dgram
)
1057 struct tdgram_bsd
*bsds
= tdgram_context_data(dgram
, struct tdgram_bsd
);
1058 struct tevent_req
*req
;
1059 struct tdgram_bsd_disconnect_state
*state
;
1064 req
= tevent_req_create(mem_ctx
, &state
,
1065 struct tdgram_bsd_disconnect_state
);
1070 if (bsds
->fd
== -1) {
1071 tevent_req_error(req
, ENOTCONN
);
1075 ret
= close(bsds
->fd
);
1077 err
= tsocket_bsd_error_from_errno(ret
, errno
, &dummy
);
1078 if (tevent_req_error(req
, err
)) {
1082 tevent_req_done(req
);
1084 tevent_req_post(req
, ev
);
1088 static int tdgram_bsd_disconnect_recv(struct tevent_req
*req
,
1093 ret
= tsocket_simple_int_recv(req
, perrno
);
1095 tevent_req_received(req
);
1099 static const struct tdgram_context_ops tdgram_bsd_ops
= {
1102 .recvfrom_send
= tdgram_bsd_recvfrom_send
,
1103 .recvfrom_recv
= tdgram_bsd_recvfrom_recv
,
1105 .sendto_send
= tdgram_bsd_sendto_send
,
1106 .sendto_recv
= tdgram_bsd_sendto_recv
,
1108 .disconnect_send
= tdgram_bsd_disconnect_send
,
1109 .disconnect_recv
= tdgram_bsd_disconnect_recv
,
1112 static int tdgram_bsd_destructor(struct tdgram_bsd
*bsds
)
1114 TALLOC_FREE(bsds
->fde
);
1115 if (bsds
->fd
!= -1) {
1122 static int tdgram_bsd_dgram_socket(const struct tsocket_address
*local
,
1123 const struct tsocket_address
*remote
,
1125 TALLOC_CTX
*mem_ctx
,
1126 struct tdgram_context
**_dgram
,
1127 const char *location
)
1129 struct tsocket_address_bsd
*lbsda
=
1130 talloc_get_type_abort(local
->private_data
,
1131 struct tsocket_address_bsd
);
1132 struct tsocket_address_bsd
*rbsda
= NULL
;
1133 struct tdgram_context
*dgram
;
1134 struct tdgram_bsd
*bsds
;
1137 bool do_bind
= false;
1138 bool do_reuseaddr
= false;
1139 bool is_inet
= false;
1140 int sa_fam
= lbsda
->u
.sa
.sa_family
;
1141 socklen_t sa_socklen
= sizeof(lbsda
->u
.ss
);
1144 rbsda
= talloc_get_type_abort(remote
->private_data
,
1145 struct tsocket_address_bsd
);
1148 switch (lbsda
->u
.sa
.sa_family
) {
1154 if (lbsda
->u
.un
.sun_path
[0] != 0) {
1155 do_reuseaddr
= true;
1159 * for unix sockets we can't use the size of sockaddr_storage
1160 * we would get EINVAL
1162 sa_socklen
= sizeof(lbsda
->u
.un
);
1165 if (lbsda
->u
.in
.sin_port
!= 0) {
1166 do_reuseaddr
= true;
1169 if (lbsda
->u
.in
.sin_addr
.s_addr
!= INADDR_ANY
) {
1173 sa_socklen
= sizeof(rbsda
->u
.in
);
1177 if (lbsda
->u
.in6
.sin6_port
!= 0) {
1178 do_reuseaddr
= true;
1181 if (memcmp(&in6addr_any
,
1182 &lbsda
->u
.in6
.sin6_addr
,
1183 sizeof(in6addr_any
)) != 0) {
1187 sa_socklen
= sizeof(rbsda
->u
.in6
);
1195 if (!do_bind
&& is_inet
&& rbsda
) {
1196 sa_fam
= rbsda
->u
.sa
.sa_family
;
1199 sa_socklen
= sizeof(rbsda
->u
.in
);
1203 sa_socklen
= sizeof(rbsda
->u
.in6
);
1209 fd
= socket(sa_fam
, SOCK_DGRAM
, 0);
1214 fd
= tsocket_bsd_common_prepare_fd(fd
, true);
1219 dgram
= tdgram_context_create(mem_ctx
,
1225 int saved_errno
= errno
;
1227 errno
= saved_errno
;
1232 talloc_set_destructor(bsds
, tdgram_bsd_destructor
);
1237 ret
= setsockopt(fd
, SOL_SOCKET
, SO_BROADCAST
,
1238 (const void *)&val
, sizeof(val
));
1240 int saved_errno
= errno
;
1242 errno
= saved_errno
;
1250 ret
= setsockopt(fd
, SOL_SOCKET
, SO_REUSEADDR
,
1251 (const void *)&val
, sizeof(val
));
1253 int saved_errno
= errno
;
1255 errno
= saved_errno
;
1261 ret
= bind(fd
, &lbsda
->u
.sa
, sa_socklen
);
1263 int saved_errno
= errno
;
1265 errno
= saved_errno
;
1271 if (rbsda
->u
.sa
.sa_family
!= sa_fam
) {
1277 ret
= connect(fd
, &rbsda
->u
.sa
, sa_socklen
);
1279 int saved_errno
= errno
;
1281 errno
= saved_errno
;
1290 int _tdgram_inet_udp_socket(const struct tsocket_address
*local
,
1291 const struct tsocket_address
*remote
,
1292 TALLOC_CTX
*mem_ctx
,
1293 struct tdgram_context
**dgram
,
1294 const char *location
)
1296 struct tsocket_address_bsd
*lbsda
=
1297 talloc_get_type_abort(local
->private_data
,
1298 struct tsocket_address_bsd
);
1301 switch (lbsda
->u
.sa
.sa_family
) {
1313 ret
= tdgram_bsd_dgram_socket(local
, remote
, false,
1314 mem_ctx
, dgram
, location
);
1319 int _tdgram_unix_socket(const struct tsocket_address
*local
,
1320 const struct tsocket_address
*remote
,
1321 TALLOC_CTX
*mem_ctx
,
1322 struct tdgram_context
**dgram
,
1323 const char *location
)
1325 struct tsocket_address_bsd
*lbsda
=
1326 talloc_get_type_abort(local
->private_data
,
1327 struct tsocket_address_bsd
);
1330 switch (lbsda
->u
.sa
.sa_family
) {
1338 ret
= tdgram_bsd_dgram_socket(local
, remote
, false,
1339 mem_ctx
, dgram
, location
);
1344 struct tstream_bsd
{
1348 struct tevent_fd
*fde
;
1350 void *readable_private
;
1351 void (*readable_handler
)(void *private_data
);
1352 void *writeable_private
;
1353 void (*writeable_handler
)(void *private_data
);
1356 static void tstream_bsd_fde_handler(struct tevent_context
*ev
,
1357 struct tevent_fd
*fde
,
1361 struct tstream_bsd
*bsds
= talloc_get_type_abort(private_data
,
1362 struct tstream_bsd
);
1364 if (flags
& TEVENT_FD_WRITE
) {
1365 bsds
->writeable_handler(bsds
->writeable_private
);
1368 if (flags
& TEVENT_FD_READ
) {
1369 if (!bsds
->readable_handler
) {
1370 if (bsds
->writeable_handler
) {
1371 bsds
->writeable_handler(bsds
->writeable_private
);
1374 TEVENT_FD_NOT_READABLE(bsds
->fde
);
1377 bsds
->readable_handler(bsds
->readable_private
);
1382 static int tstream_bsd_set_readable_handler(struct tstream_bsd
*bsds
,
1383 struct tevent_context
*ev
,
1384 void (*handler
)(void *private_data
),
1392 if (!bsds
->readable_handler
) {
1395 bsds
->readable_handler
= NULL
;
1396 bsds
->readable_private
= NULL
;
1401 /* read and write must use the same tevent_context */
1402 if (bsds
->event_ptr
!= ev
) {
1403 if (bsds
->readable_handler
|| bsds
->writeable_handler
) {
1407 bsds
->event_ptr
= NULL
;
1408 TALLOC_FREE(bsds
->fde
);
1411 if (tevent_fd_get_flags(bsds
->fde
) == 0) {
1412 TALLOC_FREE(bsds
->fde
);
1414 bsds
->fde
= tevent_add_fd(ev
, bsds
,
1415 bsds
->fd
, TEVENT_FD_READ
,
1416 tstream_bsd_fde_handler
,
1423 /* cache the event context we're running on */
1424 bsds
->event_ptr
= ev
;
1425 } else if (!bsds
->readable_handler
) {
1426 TEVENT_FD_READABLE(bsds
->fde
);
1429 bsds
->readable_handler
= handler
;
1430 bsds
->readable_private
= private_data
;
1435 static int tstream_bsd_set_writeable_handler(struct tstream_bsd
*bsds
,
1436 struct tevent_context
*ev
,
1437 void (*handler
)(void *private_data
),
1445 if (!bsds
->writeable_handler
) {
1448 bsds
->writeable_handler
= NULL
;
1449 bsds
->writeable_private
= NULL
;
1450 TEVENT_FD_NOT_WRITEABLE(bsds
->fde
);
1455 /* read and write must use the same tevent_context */
1456 if (bsds
->event_ptr
!= ev
) {
1457 if (bsds
->readable_handler
|| bsds
->writeable_handler
) {
1461 bsds
->event_ptr
= NULL
;
1462 TALLOC_FREE(bsds
->fde
);
1465 if (tevent_fd_get_flags(bsds
->fde
) == 0) {
1466 TALLOC_FREE(bsds
->fde
);
1468 bsds
->fde
= tevent_add_fd(ev
, bsds
,
1470 TEVENT_FD_READ
| TEVENT_FD_WRITE
,
1471 tstream_bsd_fde_handler
,
1478 /* cache the event context we're running on */
1479 bsds
->event_ptr
= ev
;
1480 } else if (!bsds
->writeable_handler
) {
1481 uint16_t flags
= tevent_fd_get_flags(bsds
->fde
);
1482 flags
|= TEVENT_FD_READ
| TEVENT_FD_WRITE
;
1483 tevent_fd_set_flags(bsds
->fde
, flags
);
1486 bsds
->writeable_handler
= handler
;
1487 bsds
->writeable_private
= private_data
;
1492 static ssize_t
tstream_bsd_pending_bytes(struct tstream_context
*stream
)
1494 struct tstream_bsd
*bsds
= tstream_context_data(stream
,
1495 struct tstream_bsd
);
1498 if (bsds
->fd
== -1) {
1503 ret
= tsocket_bsd_pending(bsds
->fd
);
1508 struct tstream_bsd_readv_state
{
1509 struct tstream_context
*stream
;
1511 struct iovec
*vector
;
1517 static int tstream_bsd_readv_destructor(struct tstream_bsd_readv_state
*state
)
1519 struct tstream_bsd
*bsds
= tstream_context_data(state
->stream
,
1520 struct tstream_bsd
);
1522 tstream_bsd_set_readable_handler(bsds
, NULL
, NULL
, NULL
);
1527 static void tstream_bsd_readv_handler(void *private_data
);
1529 static struct tevent_req
*tstream_bsd_readv_send(TALLOC_CTX
*mem_ctx
,
1530 struct tevent_context
*ev
,
1531 struct tstream_context
*stream
,
1532 struct iovec
*vector
,
1535 struct tevent_req
*req
;
1536 struct tstream_bsd_readv_state
*state
;
1537 struct tstream_bsd
*bsds
= tstream_context_data(stream
, struct tstream_bsd
);
1540 req
= tevent_req_create(mem_ctx
, &state
,
1541 struct tstream_bsd_readv_state
);
1546 state
->stream
= stream
;
1547 /* we make a copy of the vector so that we can modify it */
1548 state
->vector
= talloc_array(state
, struct iovec
, count
);
1549 if (tevent_req_nomem(state
->vector
, req
)) {
1552 memcpy(state
->vector
, vector
, sizeof(struct iovec
)*count
);
1553 state
->count
= count
;
1556 talloc_set_destructor(state
, tstream_bsd_readv_destructor
);
1558 if (bsds
->fd
== -1) {
1559 tevent_req_error(req
, ENOTCONN
);
1564 * this is a fast path, not waiting for the
1565 * socket to become explicit readable gains
1566 * about 10%-20% performance in benchmark tests.
1568 tstream_bsd_readv_handler(req
);
1569 if (!tevent_req_is_in_progress(req
)) {
1573 ret
= tstream_bsd_set_readable_handler(bsds
, ev
,
1574 tstream_bsd_readv_handler
,
1577 tevent_req_error(req
, errno
);
1584 tevent_req_post(req
, ev
);
1588 static void tstream_bsd_readv_handler(void *private_data
)
1590 struct tevent_req
*req
= talloc_get_type_abort(private_data
,
1592 struct tstream_bsd_readv_state
*state
= tevent_req_data(req
,
1593 struct tstream_bsd_readv_state
);
1594 struct tstream_context
*stream
= state
->stream
;
1595 struct tstream_bsd
*bsds
= tstream_context_data(stream
, struct tstream_bsd
);
1600 ret
= readv(bsds
->fd
, state
->vector
, state
->count
);
1602 /* propagate end of file */
1603 tevent_req_error(req
, EPIPE
);
1606 err
= tsocket_bsd_error_from_errno(ret
, errno
, &retry
);
1611 if (tevent_req_error(req
, err
)) {
1618 if (ret
< state
->vector
[0].iov_len
) {
1620 base
= (uint8_t *)state
->vector
[0].iov_base
;
1622 state
->vector
[0].iov_base
= base
;
1623 state
->vector
[0].iov_len
-= ret
;
1626 ret
-= state
->vector
[0].iov_len
;
1632 * there're maybe some empty vectors at the end
1633 * which we need to skip, otherwise we would get
1634 * ret == 0 from the readv() call and return EPIPE
1636 while (state
->count
> 0) {
1637 if (state
->vector
[0].iov_len
> 0) {
1644 if (state
->count
> 0) {
1645 /* we have more to read */
1649 tevent_req_done(req
);
1652 static int tstream_bsd_readv_recv(struct tevent_req
*req
,
1655 struct tstream_bsd_readv_state
*state
= tevent_req_data(req
,
1656 struct tstream_bsd_readv_state
);
1659 ret
= tsocket_simple_int_recv(req
, perrno
);
1664 tevent_req_received(req
);
1668 struct tstream_bsd_writev_state
{
1669 struct tstream_context
*stream
;
1671 struct iovec
*vector
;
1677 static int tstream_bsd_writev_destructor(struct tstream_bsd_writev_state
*state
)
1679 struct tstream_bsd
*bsds
= tstream_context_data(state
->stream
,
1680 struct tstream_bsd
);
1682 tstream_bsd_set_writeable_handler(bsds
, NULL
, NULL
, NULL
);
1687 static void tstream_bsd_writev_handler(void *private_data
);
1689 static struct tevent_req
*tstream_bsd_writev_send(TALLOC_CTX
*mem_ctx
,
1690 struct tevent_context
*ev
,
1691 struct tstream_context
*stream
,
1692 const struct iovec
*vector
,
1695 struct tevent_req
*req
;
1696 struct tstream_bsd_writev_state
*state
;
1697 struct tstream_bsd
*bsds
= tstream_context_data(stream
, struct tstream_bsd
);
1700 req
= tevent_req_create(mem_ctx
, &state
,
1701 struct tstream_bsd_writev_state
);
1706 state
->stream
= stream
;
1707 /* we make a copy of the vector so that we can modify it */
1708 state
->vector
= talloc_array(state
, struct iovec
, count
);
1709 if (tevent_req_nomem(state
->vector
, req
)) {
1712 memcpy(state
->vector
, vector
, sizeof(struct iovec
)*count
);
1713 state
->count
= count
;
1716 talloc_set_destructor(state
, tstream_bsd_writev_destructor
);
1718 if (bsds
->fd
== -1) {
1719 tevent_req_error(req
, ENOTCONN
);
1724 * this is a fast path, not waiting for the
1725 * socket to become explicit writeable gains
1726 * about 10%-20% performance in benchmark tests.
1728 tstream_bsd_writev_handler(req
);
1729 if (!tevent_req_is_in_progress(req
)) {
1733 ret
= tstream_bsd_set_writeable_handler(bsds
, ev
,
1734 tstream_bsd_writev_handler
,
1737 tevent_req_error(req
, errno
);
1744 tevent_req_post(req
, ev
);
1748 static void tstream_bsd_writev_handler(void *private_data
)
1750 struct tevent_req
*req
= talloc_get_type_abort(private_data
,
1752 struct tstream_bsd_writev_state
*state
= tevent_req_data(req
,
1753 struct tstream_bsd_writev_state
);
1754 struct tstream_context
*stream
= state
->stream
;
1755 struct tstream_bsd
*bsds
= tstream_context_data(stream
, struct tstream_bsd
);
1760 ret
= writev(bsds
->fd
, state
->vector
, state
->count
);
1762 /* propagate end of file */
1763 tevent_req_error(req
, EPIPE
);
1766 err
= tsocket_bsd_error_from_errno(ret
, errno
, &retry
);
1771 if (tevent_req_error(req
, err
)) {
1778 if (ret
< state
->vector
[0].iov_len
) {
1780 base
= (uint8_t *)state
->vector
[0].iov_base
;
1782 state
->vector
[0].iov_base
= base
;
1783 state
->vector
[0].iov_len
-= ret
;
1786 ret
-= state
->vector
[0].iov_len
;
1792 * there're maybe some empty vectors at the end
1793 * which we need to skip, otherwise we would get
1794 * ret == 0 from the writev() call and return EPIPE
1796 while (state
->count
> 0) {
1797 if (state
->vector
[0].iov_len
> 0) {
1804 if (state
->count
> 0) {
1805 /* we have more to read */
1809 tevent_req_done(req
);
1812 static int tstream_bsd_writev_recv(struct tevent_req
*req
, int *perrno
)
1814 struct tstream_bsd_writev_state
*state
= tevent_req_data(req
,
1815 struct tstream_bsd_writev_state
);
1818 ret
= tsocket_simple_int_recv(req
, perrno
);
1823 tevent_req_received(req
);
1827 struct tstream_bsd_disconnect_state
{
1831 static struct tevent_req
*tstream_bsd_disconnect_send(TALLOC_CTX
*mem_ctx
,
1832 struct tevent_context
*ev
,
1833 struct tstream_context
*stream
)
1835 struct tstream_bsd
*bsds
= tstream_context_data(stream
, struct tstream_bsd
);
1836 struct tevent_req
*req
;
1837 struct tstream_bsd_disconnect_state
*state
;
1842 req
= tevent_req_create(mem_ctx
, &state
,
1843 struct tstream_bsd_disconnect_state
);
1848 if (bsds
->fd
== -1) {
1849 tevent_req_error(req
, ENOTCONN
);
1853 ret
= close(bsds
->fd
);
1855 err
= tsocket_bsd_error_from_errno(ret
, errno
, &dummy
);
1856 if (tevent_req_error(req
, err
)) {
1860 tevent_req_done(req
);
1862 tevent_req_post(req
, ev
);
1866 static int tstream_bsd_disconnect_recv(struct tevent_req
*req
,
1871 ret
= tsocket_simple_int_recv(req
, perrno
);
1873 tevent_req_received(req
);
1877 static const struct tstream_context_ops tstream_bsd_ops
= {
1880 .pending_bytes
= tstream_bsd_pending_bytes
,
1882 .readv_send
= tstream_bsd_readv_send
,
1883 .readv_recv
= tstream_bsd_readv_recv
,
1885 .writev_send
= tstream_bsd_writev_send
,
1886 .writev_recv
= tstream_bsd_writev_recv
,
1888 .disconnect_send
= tstream_bsd_disconnect_send
,
1889 .disconnect_recv
= tstream_bsd_disconnect_recv
,
1892 static int tstream_bsd_destructor(struct tstream_bsd
*bsds
)
1894 TALLOC_FREE(bsds
->fde
);
1895 if (bsds
->fd
!= -1) {
1902 int _tstream_bsd_existing_socket(TALLOC_CTX
*mem_ctx
,
1904 struct tstream_context
**_stream
,
1905 const char *location
)
1907 struct tstream_context
*stream
;
1908 struct tstream_bsd
*bsds
;
1910 stream
= tstream_context_create(mem_ctx
,
1920 talloc_set_destructor(bsds
, tstream_bsd_destructor
);
1926 struct tstream_bsd_connect_state
{
1928 struct tevent_fd
*fde
;
1929 struct tstream_conext
*stream
;
1932 static int tstream_bsd_connect_destructor(struct tstream_bsd_connect_state
*state
)
1934 TALLOC_FREE(state
->fde
);
1935 if (state
->fd
!= -1) {
1943 static void tstream_bsd_connect_fde_handler(struct tevent_context
*ev
,
1944 struct tevent_fd
*fde
,
1946 void *private_data
);
1948 static struct tevent_req
* tstream_bsd_connect_send(TALLOC_CTX
*mem_ctx
,
1949 struct tevent_context
*ev
,
1951 const struct tsocket_address
*local
,
1952 const struct tsocket_address
*remote
)
1954 struct tevent_req
*req
;
1955 struct tstream_bsd_connect_state
*state
;
1956 struct tsocket_address_bsd
*lbsda
=
1957 talloc_get_type_abort(local
->private_data
,
1958 struct tsocket_address_bsd
);
1959 struct tsocket_address_bsd
*rbsda
=
1960 talloc_get_type_abort(remote
->private_data
,
1961 struct tsocket_address_bsd
);
1965 bool do_bind
= false;
1966 bool do_reuseaddr
= false;
1967 bool is_inet
= false;
1968 int sa_fam
= lbsda
->u
.sa
.sa_family
;
1969 socklen_t sa_socklen
= sizeof(rbsda
->u
.ss
);
1971 req
= tevent_req_create(mem_ctx
, &state
,
1972 struct tstream_bsd_connect_state
);
1979 talloc_set_destructor(state
, tstream_bsd_connect_destructor
);
1981 /* give the wrappers a chance to report an error */
1982 if (sys_errno
!= 0) {
1983 tevent_req_error(req
, sys_errno
);
1987 switch (lbsda
->u
.sa
.sa_family
) {
1989 if (lbsda
->u
.un
.sun_path
[0] != 0) {
1990 do_reuseaddr
= true;
1994 * for unix sockets we can't use the size of sockaddr_storage
1995 * we would get EINVAL
1997 sa_socklen
= sizeof(rbsda
->u
.un
);
2000 if (lbsda
->u
.in
.sin_port
!= 0) {
2001 do_reuseaddr
= true;
2004 if (lbsda
->u
.in
.sin_addr
.s_addr
!= INADDR_ANY
) {
2008 sa_socklen
= sizeof(rbsda
->u
.in
);
2012 if (lbsda
->u
.in6
.sin6_port
!= 0) {
2013 do_reuseaddr
= true;
2016 if (memcmp(&in6addr_any
,
2017 &lbsda
->u
.in6
.sin6_addr
,
2018 sizeof(in6addr_any
)) != 0) {
2022 sa_socklen
= sizeof(rbsda
->u
.in6
);
2026 tevent_req_error(req
, EINVAL
);
2030 if (!do_bind
&& is_inet
) {
2031 sa_fam
= rbsda
->u
.sa
.sa_family
;
2034 sa_socklen
= sizeof(rbsda
->u
.in
);
2038 sa_socklen
= sizeof(rbsda
->u
.in6
);
2044 state
->fd
= socket(sa_fam
, SOCK_STREAM
, 0);
2045 if (state
->fd
== -1) {
2046 tevent_req_error(req
, errno
);
2050 state
->fd
= tsocket_bsd_common_prepare_fd(state
->fd
, true);
2051 if (state
->fd
== -1) {
2052 tevent_req_error(req
, errno
);
2059 ret
= setsockopt(state
->fd
, SOL_SOCKET
, SO_REUSEADDR
,
2060 (const void *)&val
, sizeof(val
));
2062 tevent_req_error(req
, errno
);
2068 ret
= bind(state
->fd
, &lbsda
->u
.sa
, sa_socklen
);
2070 tevent_req_error(req
, errno
);
2075 if (rbsda
->u
.sa
.sa_family
!= sa_fam
) {
2076 tevent_req_error(req
, EINVAL
);
2080 ret
= connect(state
->fd
, &rbsda
->u
.sa
, sa_socklen
);
2081 err
= tsocket_bsd_error_from_errno(ret
, errno
, &retry
);
2086 if (tevent_req_error(req
, err
)) {
2090 tevent_req_done(req
);
2094 state
->fde
= tevent_add_fd(ev
, state
,
2096 TEVENT_FD_READ
| TEVENT_FD_WRITE
,
2097 tstream_bsd_connect_fde_handler
,
2099 if (tevent_req_nomem(state
->fde
, req
)) {
2106 tevent_req_post(req
, ev
);
2110 static void tstream_bsd_connect_fde_handler(struct tevent_context
*ev
,
2111 struct tevent_fd
*fde
,
2115 struct tevent_req
*req
= talloc_get_type_abort(private_data
,
2117 struct tstream_bsd_connect_state
*state
= tevent_req_data(req
,
2118 struct tstream_bsd_connect_state
);
2121 socklen_t len
= sizeof(error
);
2125 ret
= getsockopt(state
->fd
, SOL_SOCKET
, SO_ERROR
, &error
, &len
);
2132 err
= tsocket_bsd_error_from_errno(ret
, errno
, &retry
);
2137 if (tevent_req_error(req
, err
)) {
2141 tevent_req_done(req
);
2144 static int tstream_bsd_connect_recv(struct tevent_req
*req
,
2146 TALLOC_CTX
*mem_ctx
,
2147 struct tstream_context
**stream
,
2148 const char *location
)
2150 struct tstream_bsd_connect_state
*state
= tevent_req_data(req
,
2151 struct tstream_bsd_connect_state
);
2154 ret
= tsocket_simple_int_recv(req
, perrno
);
2156 ret
= _tstream_bsd_existing_socket(mem_ctx
,
2164 TALLOC_FREE(state
->fde
);
2169 tevent_req_received(req
);
2173 struct tevent_req
* tstream_inet_tcp_connect_send(TALLOC_CTX
*mem_ctx
,
2174 struct tevent_context
*ev
,
2175 const struct tsocket_address
*local
,
2176 const struct tsocket_address
*remote
)
2178 struct tsocket_address_bsd
*lbsda
=
2179 talloc_get_type_abort(local
->private_data
,
2180 struct tsocket_address_bsd
);
2181 struct tevent_req
*req
;
2184 switch (lbsda
->u
.sa
.sa_family
) {
2196 req
= tstream_bsd_connect_send(mem_ctx
, ev
, sys_errno
, local
, remote
);
2201 int _tstream_inet_tcp_connect_recv(struct tevent_req
*req
,
2203 TALLOC_CTX
*mem_ctx
,
2204 struct tstream_context
**stream
,
2205 const char *location
)
2207 return tstream_bsd_connect_recv(req
, perrno
, mem_ctx
, stream
, location
);
2210 struct tevent_req
* tstream_unix_connect_send(TALLOC_CTX
*mem_ctx
,
2211 struct tevent_context
*ev
,
2212 const struct tsocket_address
*local
,
2213 const struct tsocket_address
*remote
)
2215 struct tsocket_address_bsd
*lbsda
=
2216 talloc_get_type_abort(local
->private_data
,
2217 struct tsocket_address_bsd
);
2218 struct tevent_req
*req
;
2221 switch (lbsda
->u
.sa
.sa_family
) {
2229 req
= tstream_bsd_connect_send(mem_ctx
, ev
, sys_errno
, local
, remote
);
2234 int _tstream_unix_connect_recv(struct tevent_req
*req
,
2236 TALLOC_CTX
*mem_ctx
,
2237 struct tstream_context
**stream
,
2238 const char *location
)
2240 return tstream_bsd_connect_recv(req
, perrno
, mem_ctx
, stream
, location
);
2243 int _tstream_unix_socketpair(TALLOC_CTX
*mem_ctx1
,
2244 struct tstream_context
**_stream1
,
2245 TALLOC_CTX
*mem_ctx2
,
2246 struct tstream_context
**_stream2
,
2247 const char *location
)
2253 struct tstream_context
*stream1
= NULL
;
2254 struct tstream_context
*stream2
= NULL
;
2256 ret
= socketpair(AF_UNIX
, SOCK_STREAM
, 0, fds
);
2263 fd1
= tsocket_bsd_common_prepare_fd(fd1
, true);
2265 int sys_errno
= errno
;
2271 fd2
= tsocket_bsd_common_prepare_fd(fd2
, true);
2273 int sys_errno
= errno
;
2279 ret
= _tstream_bsd_existing_socket(mem_ctx1
,
2284 int sys_errno
= errno
;
2291 ret
= _tstream_bsd_existing_socket(mem_ctx2
,
2296 int sys_errno
= errno
;
2297 talloc_free(stream1
);
2303 *_stream1
= stream1
;
2304 *_stream2
= stream2
;