2 Unix SMB/CIFS implementation.
6 Copyright (C) Stefan Metzmacher 2004
7 Copyright (C) Andrew Tridgell 2004-2005
9 This program is free software; you can redistribute it and/or modify
10 it under the terms of the GNU General Public License as published by
11 the Free Software Foundation; either version 3 of the License, or
12 (at your option) any later version.
14 This program is distributed in the hope that it will be useful,
15 but WITHOUT ANY WARRANTY; without even the implied warranty of
16 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 GNU General Public License for more details.
19 You should have received a copy of the GNU General Public License
20 along with this program. If not, see <http://www.gnu.org/licenses/>.
24 #include "system/filesys.h"
25 #include "lib/socket/socket.h"
26 #include "system/network.h"
28 static NTSTATUS ipv4_init(struct socket_context *sock)
33 case SOCKET_TYPE_STREAM:
36 case SOCKET_TYPE_DGRAM:
40 return NT_STATUS_INVALID_PARAMETER;
43 sock->fd = socket(PF_INET, type, 0);
45 return map_nt_error_from_unix(errno);
48 sock->backend_name = "ipv4";
53 static void ipv4_close(struct socket_context *sock)
58 static NTSTATUS ipv4_connect_complete(struct socket_context *sock, uint32_t flags)
61 socklen_t len = sizeof(error);
63 /* check for any errors that may have occurred - this is needed
64 for non-blocking connect */
65 ret = getsockopt(sock->fd, SOL_SOCKET, SO_ERROR, &error, &len);
67 return map_nt_error_from_unix(errno);
70 return map_nt_error_from_unix(error);
73 if (!(flags & SOCKET_FLAG_BLOCK)) {
74 ret = set_blocking(sock->fd, False);
76 return map_nt_error_from_unix(errno);
80 sock->state = SOCKET_STATE_CLIENT_CONNECTED;
86 static NTSTATUS ipv4_connect(struct socket_context *sock,
87 const struct socket_address *my_address,
88 const struct socket_address *srv_address,
91 struct sockaddr_in srv_addr;
92 struct ipv4_addr my_ip;
93 struct ipv4_addr srv_ip;
96 if (my_address && my_address->sockaddr) {
97 ret = bind(sock->fd, my_address->sockaddr, my_address->sockaddrlen);
99 return map_nt_error_from_unix(errno);
101 } else if (my_address) {
102 my_ip = interpret_addr2(my_address->addr);
104 if (my_ip.addr != 0 || my_address->port != 0) {
105 struct sockaddr_in my_addr;
106 ZERO_STRUCT(my_addr);
107 #ifdef HAVE_SOCK_SIN_LEN
108 my_addr.sin_len = sizeof(my_addr);
110 my_addr.sin_addr.s_addr = my_ip.addr;
111 my_addr.sin_port = htons(my_address->port);
112 my_addr.sin_family = PF_INET;
114 ret = bind(sock->fd, (struct sockaddr *)&my_addr, sizeof(my_addr));
116 return map_nt_error_from_unix(errno);
121 if (srv_address->sockaddr) {
122 ret = connect(sock->fd, srv_address->sockaddr, srv_address->sockaddrlen);
124 return map_nt_error_from_unix(errno);
127 srv_ip = interpret_addr2(srv_address->addr);
129 return NT_STATUS_BAD_NETWORK_NAME;
132 ZERO_STRUCT(srv_addr);
133 #ifdef HAVE_SOCK_SIN_LEN
134 srv_addr.sin_len = sizeof(srv_addr);
136 srv_addr.sin_addr.s_addr= srv_ip.addr;
137 srv_addr.sin_port = htons(srv_address->port);
138 srv_addr.sin_family = PF_INET;
140 ret = connect(sock->fd, (const struct sockaddr *)&srv_addr, sizeof(srv_addr));
142 return map_nt_error_from_unix(errno);
146 return ipv4_connect_complete(sock, flags);
151 note that for simplicity of the API, socket_listen() is also
152 use for DGRAM sockets, but in reality only a bind() is done
154 static NTSTATUS ipv4_listen(struct socket_context *sock,
155 const struct socket_address *my_address,
156 int queue_size, uint32_t flags)
158 struct sockaddr_in my_addr;
159 struct ipv4_addr ip_addr;
162 socket_set_option(sock, "SO_REUSEADDR=1", NULL);
164 if (my_address->sockaddr) {
165 ret = bind(sock->fd, my_address->sockaddr, my_address->sockaddrlen);
167 ip_addr = interpret_addr2(my_address->addr);
169 ZERO_STRUCT(my_addr);
170 #ifdef HAVE_SOCK_SIN_LEN
171 my_addr.sin_len = sizeof(my_addr);
173 my_addr.sin_addr.s_addr = ip_addr.addr;
174 my_addr.sin_port = htons(my_address->port);
175 my_addr.sin_family = PF_INET;
177 ret = bind(sock->fd, (struct sockaddr *)&my_addr, sizeof(my_addr));
181 return map_nt_error_from_unix(errno);
184 if (sock->type == SOCKET_TYPE_STREAM) {
185 ret = listen(sock->fd, queue_size);
187 return map_nt_error_from_unix(errno);
191 if (!(flags & SOCKET_FLAG_BLOCK)) {
192 ret = set_blocking(sock->fd, False);
194 return map_nt_error_from_unix(errno);
198 sock->state= SOCKET_STATE_SERVER_LISTEN;
203 static NTSTATUS ipv4_accept(struct socket_context *sock, struct socket_context **new_sock)
205 struct sockaddr_in cli_addr;
206 socklen_t cli_addr_len = sizeof(cli_addr);
209 if (sock->type != SOCKET_TYPE_STREAM) {
210 return NT_STATUS_INVALID_PARAMETER;
213 new_fd = accept(sock->fd, (struct sockaddr *)&cli_addr, &cli_addr_len);
215 return map_nt_error_from_unix(errno);
218 if (!(sock->flags & SOCKET_FLAG_BLOCK)) {
219 int ret = set_blocking(new_fd, False);
222 return map_nt_error_from_unix(errno);
226 /* TODO: we could add a 'accept_check' hook here
227 * which get the black/white lists via socket_set_accept_filter()
228 * or something like that
232 (*new_sock) = talloc(NULL, struct socket_context);
235 return NT_STATUS_NO_MEMORY;
238 /* copy the socket_context */
239 (*new_sock)->type = sock->type;
240 (*new_sock)->state = SOCKET_STATE_SERVER_CONNECTED;
241 (*new_sock)->flags = sock->flags;
243 (*new_sock)->fd = new_fd;
245 (*new_sock)->private_data = NULL;
246 (*new_sock)->ops = sock->ops;
247 (*new_sock)->backend_name = sock->backend_name;
252 static NTSTATUS ipv4_recv(struct socket_context *sock, void *buf,
253 size_t wantlen, size_t *nread)
259 gotlen = recv(sock->fd, buf, wantlen, 0);
261 return NT_STATUS_END_OF_FILE;
262 } else if (gotlen == -1) {
263 return map_nt_error_from_unix(errno);
272 static NTSTATUS ipv4_recvfrom(struct socket_context *sock, void *buf,
273 size_t wantlen, size_t *nread,
274 TALLOC_CTX *addr_ctx, struct socket_address **_src)
277 struct sockaddr_in *from_addr;
278 socklen_t from_len = sizeof(*from_addr);
279 struct socket_address *src;
282 src = talloc(addr_ctx, struct socket_address);
284 return NT_STATUS_NO_MEMORY;
287 src->family = sock->backend_name;
289 from_addr = talloc(src, struct sockaddr_in);
292 return NT_STATUS_NO_MEMORY;
295 src->sockaddr = (struct sockaddr *)from_addr;
299 gotlen = recvfrom(sock->fd, buf, wantlen, 0,
300 src->sockaddr, &from_len);
303 return NT_STATUS_END_OF_FILE;
304 } else if (gotlen == -1) {
306 return map_nt_error_from_unix(errno);
309 src->sockaddrlen = from_len;
311 addr = inet_ntoa(from_addr->sin_addr);
314 return NT_STATUS_INTERNAL_ERROR;
316 src->addr = talloc_strdup(src, addr);
317 if (src->addr == NULL) {
319 return NT_STATUS_NO_MEMORY;
321 src->port = ntohs(from_addr->sin_port);
328 static NTSTATUS ipv4_send(struct socket_context *sock,
329 const DATA_BLOB *blob, size_t *sendlen)
335 len = send(sock->fd, blob->data, blob->length, 0);
337 return map_nt_error_from_unix(errno);
345 static NTSTATUS ipv4_sendto(struct socket_context *sock,
346 const DATA_BLOB *blob, size_t *sendlen,
347 const struct socket_address *dest_addr)
351 if (dest_addr->sockaddr) {
352 len = sendto(sock->fd, blob->data, blob->length, 0,
353 dest_addr->sockaddr, dest_addr->sockaddrlen);
355 struct sockaddr_in srv_addr;
356 struct ipv4_addr addr;
358 ZERO_STRUCT(srv_addr);
359 #ifdef HAVE_SOCK_SIN_LEN
360 srv_addr.sin_len = sizeof(srv_addr);
362 addr = interpret_addr2(dest_addr->addr);
363 if (addr.addr == 0) {
364 return NT_STATUS_HOST_UNREACHABLE;
366 srv_addr.sin_addr.s_addr = addr.addr;
367 srv_addr.sin_port = htons(dest_addr->port);
368 srv_addr.sin_family = PF_INET;
372 len = sendto(sock->fd, blob->data, blob->length, 0,
373 (struct sockaddr *)&srv_addr, sizeof(srv_addr));
376 return map_nt_error_from_unix(errno);
384 static NTSTATUS ipv4_set_option(struct socket_context *sock, const char *option, const char *val)
386 set_socket_options(sock->fd, option);
390 static char *ipv4_get_peer_name(struct socket_context *sock, TALLOC_CTX *mem_ctx)
392 struct sockaddr_in peer_addr;
393 socklen_t len = sizeof(peer_addr);
397 ret = getpeername(sock->fd, (struct sockaddr *)&peer_addr, &len);
402 he = gethostbyaddr((char *)&peer_addr.sin_addr, sizeof(peer_addr.sin_addr), AF_INET);
407 return talloc_strdup(mem_ctx, he->h_name);
410 static struct socket_address *ipv4_get_peer_addr(struct socket_context *sock, TALLOC_CTX *mem_ctx)
412 struct sockaddr_in *peer_addr;
413 socklen_t len = sizeof(*peer_addr);
415 struct socket_address *peer;
418 peer = talloc(mem_ctx, struct socket_address);
423 peer->family = sock->backend_name;
424 peer_addr = talloc(peer, struct sockaddr_in);
430 peer->sockaddr = (struct sockaddr *)peer_addr;
432 ret = getpeername(sock->fd, peer->sockaddr, &len);
438 peer->sockaddrlen = len;
440 addr = inet_ntoa(peer_addr->sin_addr);
445 peer->addr = talloc_strdup(peer, addr);
450 peer->port = ntohs(peer_addr->sin_port);
455 static struct socket_address *ipv4_get_my_addr(struct socket_context *sock, TALLOC_CTX *mem_ctx)
457 struct sockaddr_in *local_addr;
458 socklen_t len = sizeof(*local_addr);
460 struct socket_address *local;
463 local = talloc(mem_ctx, struct socket_address);
468 local->family = sock->backend_name;
469 local_addr = talloc(local, struct sockaddr_in);
475 local->sockaddr = (struct sockaddr *)local_addr;
477 ret = getsockname(sock->fd, local->sockaddr, &len);
483 local->sockaddrlen = len;
485 addr = inet_ntoa(local_addr->sin_addr);
490 local->addr = talloc_strdup(local, addr);
495 local->port = ntohs(local_addr->sin_port);
499 static int ipv4_get_fd(struct socket_context *sock)
504 static NTSTATUS ipv4_pending(struct socket_context *sock, size_t *npending)
507 if (ioctl(sock->fd, FIONREAD, &value) == 0) {
511 return map_nt_error_from_unix(errno);
514 static const struct socket_ops ipv4_ops = {
516 .fn_init = ipv4_init,
517 .fn_connect = ipv4_connect,
518 .fn_connect_complete = ipv4_connect_complete,
519 .fn_listen = ipv4_listen,
520 .fn_accept = ipv4_accept,
521 .fn_recv = ipv4_recv,
522 .fn_recvfrom = ipv4_recvfrom,
523 .fn_send = ipv4_send,
524 .fn_sendto = ipv4_sendto,
525 .fn_pending = ipv4_pending,
526 .fn_close = ipv4_close,
528 .fn_set_option = ipv4_set_option,
530 .fn_get_peer_name = ipv4_get_peer_name,
531 .fn_get_peer_addr = ipv4_get_peer_addr,
532 .fn_get_my_addr = ipv4_get_my_addr,
533 .fn_get_fd = ipv4_get_fd
536 const struct socket_ops *socket_ipv4_ops(enum socket_type type)