2 Unix SMB/CIFS implementation.
4 Socket IPv4/IPv6 functions
6 Copyright (C) Stefan Metzmacher 2004
7 Copyright (C) Andrew Tridgell 2004-2005
8 Copyright (C) Jelmer Vernooij 2004
10 This program is free software; you can redistribute it and/or modify
11 it under the terms of the GNU General Public License as published by
12 the Free Software Foundation; either version 3 of the License, or
13 (at your option) any later version.
15 This program 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
18 GNU General Public License for more details.
20 You should have received a copy of the GNU General Public License
21 along with this program. If not, see <http://www.gnu.org/licenses/>.
25 #include "system/filesys.h"
26 #include "lib/socket/socket.h"
27 #include "system/network.h"
29 static NTSTATUS ipv4_init(struct socket_context *sock)
34 case SOCKET_TYPE_STREAM:
37 case SOCKET_TYPE_DGRAM:
41 return NT_STATUS_INVALID_PARAMETER;
44 sock->fd = socket(PF_INET, type, 0);
46 return map_nt_error_from_unix(errno);
49 sock->backend_name = "ipv4";
50 sock->family = AF_INET;
55 static void ip_close(struct socket_context *sock)
60 static NTSTATUS ip_connect_complete(struct socket_context *sock, uint32_t flags)
63 socklen_t len = sizeof(error);
65 /* check for any errors that may have occurred - this is needed
66 for non-blocking connect */
67 ret = getsockopt(sock->fd, SOL_SOCKET, SO_ERROR, &error, &len);
69 return map_nt_error_from_unix(errno);
72 return map_nt_error_from_unix(error);
75 if (!(flags & SOCKET_FLAG_BLOCK)) {
76 ret = set_blocking(sock->fd, false);
78 return map_nt_error_from_unix(errno);
82 sock->state = SOCKET_STATE_CLIENT_CONNECTED;
88 static NTSTATUS ipv4_connect(struct socket_context *sock,
89 const struct socket_address *my_address,
90 const struct socket_address *srv_address,
93 struct sockaddr_in srv_addr;
95 struct in_addr srv_ip;
98 if (my_address && my_address->sockaddr) {
99 ret = bind(sock->fd, my_address->sockaddr, my_address->sockaddrlen);
101 return map_nt_error_from_unix(errno);
103 } else if (my_address) {
104 my_ip = interpret_addr2(my_address->addr);
106 if (my_ip.s_addr != 0 || my_address->port != 0) {
107 struct sockaddr_in my_addr;
108 ZERO_STRUCT(my_addr);
109 #ifdef HAVE_SOCK_SIN_LEN
110 my_addr.sin_len = sizeof(my_addr);
112 my_addr.sin_addr.s_addr = my_ip.s_addr;
113 my_addr.sin_port = htons(my_address->port);
114 my_addr.sin_family = PF_INET;
116 ret = bind(sock->fd, (struct sockaddr *)&my_addr, sizeof(my_addr));
118 return map_nt_error_from_unix(errno);
123 if (srv_address->sockaddr) {
124 ret = connect(sock->fd, srv_address->sockaddr, srv_address->sockaddrlen);
126 return map_nt_error_from_unix(errno);
129 srv_ip = interpret_addr2(srv_address->addr);
130 if (!srv_ip.s_addr) {
131 return NT_STATUS_BAD_NETWORK_NAME;
134 ZERO_STRUCT(srv_addr);
135 #ifdef HAVE_SOCK_SIN_LEN
136 srv_addr.sin_len = sizeof(srv_addr);
138 srv_addr.sin_addr.s_addr= srv_ip.s_addr;
139 srv_addr.sin_port = htons(srv_address->port);
140 srv_addr.sin_family = PF_INET;
142 ret = connect(sock->fd, (const struct sockaddr *)&srv_addr, sizeof(srv_addr));
144 return map_nt_error_from_unix(errno);
148 return ip_connect_complete(sock, flags);
153 note that for simplicity of the API, socket_listen() is also
154 use for DGRAM sockets, but in reality only a bind() is done
156 static NTSTATUS ipv4_listen(struct socket_context *sock,
157 const struct socket_address *my_address,
158 int queue_size, uint32_t flags)
160 struct sockaddr_in my_addr;
161 struct in_addr ip_addr;
164 socket_set_option(sock, "SO_REUSEADDR=1", NULL);
166 if (my_address->sockaddr) {
167 ret = bind(sock->fd, my_address->sockaddr, my_address->sockaddrlen);
169 ip_addr = interpret_addr2(my_address->addr);
171 ZERO_STRUCT(my_addr);
172 #ifdef HAVE_SOCK_SIN_LEN
173 my_addr.sin_len = sizeof(my_addr);
175 my_addr.sin_addr.s_addr = ip_addr.s_addr;
176 my_addr.sin_port = htons(my_address->port);
177 my_addr.sin_family = PF_INET;
179 ret = bind(sock->fd, (struct sockaddr *)&my_addr, sizeof(my_addr));
183 return map_nt_error_from_unix(errno);
186 if (sock->type == SOCKET_TYPE_STREAM) {
187 ret = listen(sock->fd, queue_size);
189 return map_nt_error_from_unix(errno);
193 if (!(flags & SOCKET_FLAG_BLOCK)) {
194 ret = set_blocking(sock->fd, false);
196 return map_nt_error_from_unix(errno);
200 sock->state= SOCKET_STATE_SERVER_LISTEN;
205 static NTSTATUS ipv4_accept(struct socket_context *sock, struct socket_context **new_sock)
207 struct sockaddr_in cli_addr;
208 socklen_t cli_addr_len = sizeof(cli_addr);
211 if (sock->type != SOCKET_TYPE_STREAM) {
212 return NT_STATUS_INVALID_PARAMETER;
215 new_fd = accept(sock->fd, (struct sockaddr *)&cli_addr, &cli_addr_len);
217 return map_nt_error_from_unix(errno);
220 if (!(sock->flags & SOCKET_FLAG_BLOCK)) {
221 int ret = set_blocking(new_fd, false);
224 return map_nt_error_from_unix(errno);
228 /* TODO: we could add a 'accept_check' hook here
229 * which get the black/white lists via socket_set_accept_filter()
230 * or something like that
234 (*new_sock) = talloc(NULL, struct socket_context);
237 return NT_STATUS_NO_MEMORY;
240 /* copy the socket_context */
241 (*new_sock)->type = sock->type;
242 (*new_sock)->state = SOCKET_STATE_SERVER_CONNECTED;
243 (*new_sock)->flags = sock->flags;
245 (*new_sock)->fd = new_fd;
247 (*new_sock)->private_data = NULL;
248 (*new_sock)->ops = sock->ops;
249 (*new_sock)->backend_name = sock->backend_name;
254 static NTSTATUS ip_recv(struct socket_context *sock, void *buf,
255 size_t wantlen, size_t *nread)
261 gotlen = recv(sock->fd, buf, wantlen, 0);
263 return NT_STATUS_END_OF_FILE;
264 } else if (gotlen == -1) {
265 return map_nt_error_from_unix(errno);
274 static NTSTATUS ipv4_recvfrom(struct socket_context *sock, void *buf,
275 size_t wantlen, size_t *nread,
276 TALLOC_CTX *addr_ctx, struct socket_address **_src)
279 struct sockaddr_in *from_addr;
280 socklen_t from_len = sizeof(*from_addr);
281 struct socket_address *src;
282 char addrstring[INET_ADDRSTRLEN];
284 src = talloc(addr_ctx, struct socket_address);
286 return NT_STATUS_NO_MEMORY;
289 src->family = sock->backend_name;
291 from_addr = talloc(src, struct sockaddr_in);
294 return NT_STATUS_NO_MEMORY;
297 src->sockaddr = (struct sockaddr *)from_addr;
301 gotlen = recvfrom(sock->fd, buf, wantlen, 0,
302 src->sockaddr, &from_len);
305 return NT_STATUS_END_OF_FILE;
306 } else if (gotlen == -1) {
308 return map_nt_error_from_unix(errno);
311 src->sockaddrlen = from_len;
313 if (inet_ntop(AF_INET, &from_addr->sin_addr, addrstring,
314 sizeof(addrstring)) == NULL) {
316 return NT_STATUS_INTERNAL_ERROR;
318 src->addr = talloc_strdup(src, addrstring);
319 if (src->addr == NULL) {
321 return NT_STATUS_NO_MEMORY;
323 src->port = ntohs(from_addr->sin_port);
330 static NTSTATUS ip_send(struct socket_context *sock,
331 const DATA_BLOB *blob, size_t *sendlen)
337 len = send(sock->fd, blob->data, blob->length, 0);
339 return map_nt_error_from_unix(errno);
347 static NTSTATUS ipv4_sendto(struct socket_context *sock,
348 const DATA_BLOB *blob, size_t *sendlen,
349 const struct socket_address *dest_addr)
353 if (dest_addr->sockaddr) {
354 len = sendto(sock->fd, blob->data, blob->length, 0,
355 dest_addr->sockaddr, dest_addr->sockaddrlen);
357 struct sockaddr_in srv_addr;
360 ZERO_STRUCT(srv_addr);
361 #ifdef HAVE_SOCK_SIN_LEN
362 srv_addr.sin_len = sizeof(srv_addr);
364 addr = interpret_addr2(dest_addr->addr);
365 if (addr.s_addr == 0) {
366 return NT_STATUS_HOST_UNREACHABLE;
368 srv_addr.sin_addr.s_addr = addr.s_addr;
369 srv_addr.sin_port = htons(dest_addr->port);
370 srv_addr.sin_family = PF_INET;
374 len = sendto(sock->fd, blob->data, blob->length, 0,
375 (struct sockaddr *)&srv_addr, sizeof(srv_addr));
378 return map_nt_error_from_unix(errno);
386 static NTSTATUS ipv4_set_option(struct socket_context *sock, const char *option, const char *val)
388 set_socket_options(sock->fd, option);
392 static char *ipv4_get_peer_name(struct socket_context *sock, TALLOC_CTX *mem_ctx)
394 struct sockaddr_in peer_addr;
395 socklen_t len = sizeof(peer_addr);
399 ret = getpeername(sock->fd, (struct sockaddr *)&peer_addr, &len);
404 he = gethostbyaddr((char *)&peer_addr.sin_addr, sizeof(peer_addr.sin_addr), AF_INET);
409 return talloc_strdup(mem_ctx, he->h_name);
412 static struct socket_address *ipv4_get_peer_addr(struct socket_context *sock, TALLOC_CTX *mem_ctx)
414 struct sockaddr_in *peer_addr;
415 socklen_t len = sizeof(*peer_addr);
416 struct socket_address *peer;
417 char addrstring[INET_ADDRSTRLEN];
420 peer = talloc(mem_ctx, struct socket_address);
425 peer->family = sock->backend_name;
426 peer_addr = talloc(peer, struct sockaddr_in);
432 peer->sockaddr = (struct sockaddr *)peer_addr;
434 ret = getpeername(sock->fd, peer->sockaddr, &len);
440 peer->sockaddrlen = len;
442 if (inet_ntop(AF_INET, &peer_addr->sin_addr, addrstring,
443 sizeof(addrstring)) == NULL) {
447 peer->addr = talloc_strdup(peer, addrstring);
452 peer->port = ntohs(peer_addr->sin_port);
457 static struct socket_address *ipv4_get_my_addr(struct socket_context *sock, TALLOC_CTX *mem_ctx)
459 struct sockaddr_in *local_addr;
460 socklen_t len = sizeof(*local_addr);
461 struct socket_address *local;
462 char addrstring[INET_ADDRSTRLEN];
465 local = talloc(mem_ctx, struct socket_address);
470 local->family = sock->backend_name;
471 local_addr = talloc(local, struct sockaddr_in);
477 local->sockaddr = (struct sockaddr *)local_addr;
479 ret = getsockname(sock->fd, local->sockaddr, &len);
485 local->sockaddrlen = len;
487 if (inet_ntop(AF_INET, &local_addr->sin_addr, addrstring,
488 sizeof(addrstring)) == NULL) {
492 local->addr = talloc_strdup(local, addrstring);
497 local->port = ntohs(local_addr->sin_port);
501 static int ip_get_fd(struct socket_context *sock)
506 static NTSTATUS ip_pending(struct socket_context *sock, size_t *npending)
509 if (ioctl(sock->fd, FIONREAD, &value) == 0) {
513 return map_nt_error_from_unix(errno);
516 static const struct socket_ops ipv4_ops = {
518 .fn_init = ipv4_init,
519 .fn_connect = ipv4_connect,
520 .fn_connect_complete = ip_connect_complete,
521 .fn_listen = ipv4_listen,
522 .fn_accept = ipv4_accept,
524 .fn_recvfrom = ipv4_recvfrom,
526 .fn_sendto = ipv4_sendto,
527 .fn_pending = ip_pending,
528 .fn_close = ip_close,
530 .fn_set_option = ipv4_set_option,
532 .fn_get_peer_name = ipv4_get_peer_name,
533 .fn_get_peer_addr = ipv4_get_peer_addr,
534 .fn_get_my_addr = ipv4_get_my_addr,
536 .fn_get_fd = ip_get_fd
539 const struct socket_ops *socket_ipv4_ops(enum socket_type type)
546 static struct in6_addr interpret_addr6(const char *name)
550 if (name == NULL) return in6addr_any;
552 if (strcasecmp(name, "localhost") == 0) {
556 he = gethostbyname2(name, PF_INET6);
558 if (he == NULL) return in6addr_any;
560 return *((struct in6_addr *)he->h_addr);
563 static NTSTATUS ipv6_init(struct socket_context *sock)
567 switch (sock->type) {
568 case SOCKET_TYPE_STREAM:
571 case SOCKET_TYPE_DGRAM:
575 return NT_STATUS_INVALID_PARAMETER;
578 sock->fd = socket(PF_INET6, type, 0);
579 if (sock->fd == -1) {
580 return map_nt_error_from_unix(errno);
583 sock->backend_name = "ipv6";
584 sock->family = AF_INET6;
589 static NTSTATUS ipv6_tcp_connect(struct socket_context *sock,
590 const struct socket_address *my_address,
591 const struct socket_address *srv_address,
596 if (my_address && my_address->sockaddr) {
597 ret = bind(sock->fd, my_address->sockaddr, my_address->sockaddrlen);
599 return map_nt_error_from_unix(errno);
601 } else if (my_address) {
602 struct in6_addr my_ip;
603 my_ip = interpret_addr6(my_address->addr);
605 if (memcmp(&my_ip, &in6addr_any, sizeof(my_ip)) || my_address->port != 0) {
606 struct sockaddr_in6 my_addr;
607 ZERO_STRUCT(my_addr);
608 my_addr.sin6_addr = my_ip;
609 my_addr.sin6_port = htons(my_address->port);
610 my_addr.sin6_family = PF_INET6;
612 ret = bind(sock->fd, (struct sockaddr *)&my_addr, sizeof(my_addr));
614 return map_nt_error_from_unix(errno);
619 if (srv_address->sockaddr) {
620 ret = connect(sock->fd, srv_address->sockaddr, srv_address->sockaddrlen);
622 struct in6_addr srv_ip;
623 struct sockaddr_in6 srv_addr;
624 srv_ip = interpret_addr6(srv_address->addr);
625 if (memcmp(&srv_ip, &in6addr_any, sizeof(srv_ip)) == 0) {
626 return NT_STATUS_BAD_NETWORK_NAME;
629 ZERO_STRUCT(srv_addr);
630 srv_addr.sin6_addr = srv_ip;
631 srv_addr.sin6_port = htons(srv_address->port);
632 srv_addr.sin6_family = PF_INET6;
634 ret = connect(sock->fd, (const struct sockaddr *)&srv_addr, sizeof(srv_addr));
637 return map_nt_error_from_unix(errno);
640 return ip_connect_complete(sock, flags);
643 static NTSTATUS ipv6_listen(struct socket_context *sock,
644 const struct socket_address *my_address,
645 int queue_size, uint32_t flags)
647 struct sockaddr_in6 my_addr;
648 struct in6_addr ip_addr;
651 socket_set_option(sock, "SO_REUSEADDR=1", NULL);
653 if (my_address->sockaddr) {
654 ret = bind(sock->fd, my_address->sockaddr, my_address->sockaddrlen);
656 ip_addr = interpret_addr6(my_address->addr);
658 ZERO_STRUCT(my_addr);
659 my_addr.sin6_addr = ip_addr;
660 my_addr.sin6_port = htons(my_address->port);
661 my_addr.sin6_family = PF_INET6;
663 ret = bind(sock->fd, (struct sockaddr *)&my_addr, sizeof(my_addr));
667 return map_nt_error_from_unix(errno);
670 if (sock->type == SOCKET_TYPE_STREAM) {
671 ret = listen(sock->fd, queue_size);
673 return map_nt_error_from_unix(errno);
677 if (!(flags & SOCKET_FLAG_BLOCK)) {
678 ret = set_blocking(sock->fd, false);
680 return map_nt_error_from_unix(errno);
684 sock->state= SOCKET_STATE_SERVER_LISTEN;
689 static NTSTATUS ipv6_tcp_accept(struct socket_context *sock, struct socket_context **new_sock)
691 struct sockaddr_in cli_addr;
692 socklen_t cli_addr_len = sizeof(cli_addr);
695 if (sock->type != SOCKET_TYPE_STREAM) {
696 return NT_STATUS_INVALID_PARAMETER;
699 new_fd = accept(sock->fd, (struct sockaddr *)&cli_addr, &cli_addr_len);
701 return map_nt_error_from_unix(errno);
704 if (!(sock->flags & SOCKET_FLAG_BLOCK)) {
705 int ret = set_blocking(new_fd, false);
708 return map_nt_error_from_unix(errno);
712 /* TODO: we could add a 'accept_check' hook here
713 * which get the black/white lists via socket_set_accept_filter()
714 * or something like that
718 (*new_sock) = talloc(NULL, struct socket_context);
721 return NT_STATUS_NO_MEMORY;
724 /* copy the socket_context */
725 (*new_sock)->type = sock->type;
726 (*new_sock)->state = SOCKET_STATE_SERVER_CONNECTED;
727 (*new_sock)->flags = sock->flags;
729 (*new_sock)->fd = new_fd;
731 (*new_sock)->private_data = NULL;
732 (*new_sock)->ops = sock->ops;
733 (*new_sock)->backend_name = sock->backend_name;
738 static NTSTATUS ipv6_recvfrom(struct socket_context *sock, void *buf,
739 size_t wantlen, size_t *nread,
740 TALLOC_CTX *addr_ctx, struct socket_address **_src)
743 struct sockaddr_in6 *from_addr;
744 socklen_t from_len = sizeof(*from_addr);
745 struct socket_address *src;
746 char addrstring[INET6_ADDRSTRLEN];
748 src = talloc(addr_ctx, struct socket_address);
750 return NT_STATUS_NO_MEMORY;
753 src->family = sock->backend_name;
755 from_addr = talloc(src, struct sockaddr_in6);
758 return NT_STATUS_NO_MEMORY;
761 src->sockaddr = (struct sockaddr *)from_addr;
765 gotlen = recvfrom(sock->fd, buf, wantlen, 0,
766 src->sockaddr, &from_len);
769 return NT_STATUS_END_OF_FILE;
770 } else if (gotlen == -1) {
772 return map_nt_error_from_unix(errno);
775 src->sockaddrlen = from_len;
777 if (inet_ntop(AF_INET6, &from_addr->sin6_addr, addrstring, sizeof(addrstring)) == NULL) {
778 DEBUG(0, ("Unable to convert address to string: %s\n", strerror(errno)));
780 return NT_STATUS_INTERNAL_ERROR;
783 src->addr = talloc_strdup(src, addrstring);
784 if (src->addr == NULL) {
786 return NT_STATUS_NO_MEMORY;
788 src->port = ntohs(from_addr->sin6_port);
795 static NTSTATUS ipv6_sendto(struct socket_context *sock,
796 const DATA_BLOB *blob, size_t *sendlen,
797 const struct socket_address *dest_addr)
801 if (dest_addr->sockaddr) {
802 len = sendto(sock->fd, blob->data, blob->length, 0,
803 dest_addr->sockaddr, dest_addr->sockaddrlen);
805 struct sockaddr_in6 srv_addr;
806 struct in6_addr addr;
808 ZERO_STRUCT(srv_addr);
809 addr = interpret_addr6(dest_addr->addr);
810 if (addr.s6_addr == 0) {
811 return NT_STATUS_HOST_UNREACHABLE;
813 srv_addr.sin6_addr = addr;
814 srv_addr.sin6_port = htons(dest_addr->port);
815 srv_addr.sin6_family = PF_INET6;
819 len = sendto(sock->fd, blob->data, blob->length, 0,
820 (struct sockaddr *)&srv_addr, sizeof(srv_addr));
823 return map_nt_error_from_unix(errno);
831 static NTSTATUS ipv6_set_option(struct socket_context *sock, const char *option, const char *val)
833 set_socket_options(sock->fd, option);
837 static char *ipv6_tcp_get_peer_name(struct socket_context *sock, TALLOC_CTX *mem_ctx)
839 struct sockaddr_in6 peer_addr;
840 socklen_t len = sizeof(peer_addr);
844 ret = getpeername(sock->fd, (struct sockaddr *)&peer_addr, &len);
849 he = gethostbyaddr((char *)&peer_addr.sin6_addr, sizeof(peer_addr.sin6_addr), AF_INET6);
854 return talloc_strdup(mem_ctx, he->h_name);
857 static struct socket_address *ipv6_tcp_get_peer_addr(struct socket_context *sock, TALLOC_CTX *mem_ctx)
859 struct sockaddr_in6 *peer_addr;
860 socklen_t len = sizeof(*peer_addr);
861 struct socket_address *peer;
864 const char *addr_ret;
866 peer = talloc(mem_ctx, struct socket_address);
871 peer->family = sock->backend_name;
872 peer_addr = talloc(peer, struct sockaddr_in6);
878 peer->sockaddr = (struct sockaddr *)peer_addr;
880 ret = getpeername(sock->fd, peer->sockaddr, &len);
886 peer->sockaddrlen = len;
888 addr_ret = inet_ntop(AF_INET6, &peer_addr->sin6_addr, addr, sizeof(addr));
889 if (addr_ret == NULL) {
894 peer->addr = talloc_strdup(peer, addr_ret);
895 if (peer->addr == NULL) {
900 peer->port = ntohs(peer_addr->sin6_port);
905 static struct socket_address *ipv6_tcp_get_my_addr(struct socket_context *sock, TALLOC_CTX *mem_ctx)
907 struct sockaddr_in6 *local_addr;
908 socklen_t len = sizeof(*local_addr);
909 struct socket_address *local;
911 char addrstring[INET6_ADDRSTRLEN];
913 local = talloc(mem_ctx, struct socket_address);
918 local->family = sock->backend_name;
919 local_addr = talloc(local, struct sockaddr_in6);
925 local->sockaddr = (struct sockaddr *)local_addr;
927 ret = getsockname(sock->fd, local->sockaddr, &len);
933 local->sockaddrlen = len;
935 if (inet_ntop(AF_INET6, &local_addr->sin6_addr, addrstring,
936 sizeof(addrstring)) == NULL) {
937 DEBUG(0, ("Unable to convert address to string: %s\n",
943 local->addr = talloc_strdup(mem_ctx, addrstring);
948 local->port = ntohs(local_addr->sin6_port);
953 static const struct socket_ops ipv6_tcp_ops = {
955 .fn_init = ipv6_init,
956 .fn_connect = ipv6_tcp_connect,
957 .fn_connect_complete = ip_connect_complete,
958 .fn_listen = ipv6_listen,
959 .fn_accept = ipv6_tcp_accept,
961 .fn_recvfrom = ipv6_recvfrom,
963 .fn_sendto = ipv6_sendto,
964 .fn_pending = ip_pending,
965 .fn_close = ip_close,
967 .fn_set_option = ipv6_set_option,
969 .fn_get_peer_name = ipv6_tcp_get_peer_name,
970 .fn_get_peer_addr = ipv6_tcp_get_peer_addr,
971 .fn_get_my_addr = ipv6_tcp_get_my_addr,
973 .fn_get_fd = ip_get_fd
976 const struct socket_ops *socket_ipv6_ops(enum socket_type type)
978 return &ipv6_tcp_ops;