2 Unix SMB/CIFS implementation.
4 Copyright (C) Stefan Metzmacher 2004
6 This program is free software; you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation; either version 2 of the License, or
9 (at your option) any later version.
11 This program is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 GNU General Public License for more details.
16 You should have received a copy of the GNU General Public License
17 along with this program; if not, write to the Free Software
18 Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
23 static NTSTATUS ipv4_tcp_init(struct socket_context *sock)
25 sock->fd = socket(PF_INET, SOCK_STREAM, 0);
27 /* TODO: we need to map from errno to NTSTATUS here! */
28 return NT_STATUS_FOOBAR;
34 static void ipv4_tcp_close(struct socket_context *sock)
39 static NTSTATUS ipv4_tcp_connect(struct socket_context *sock,
40 const char *my_address, int my_port,
41 const char *srv_address, int srv_port,
44 struct sockaddr_in my_addr;
45 struct sockaddr_in srv_addr;
47 struct in_addr srv_ip;
50 ret = inet_aton(my_address, &my_ip);
52 /* not a valid ipv4 address */
53 return NT_STATUS_FOOBAR;
57 #ifdef HAVE_SOCK_SIN_LEN
58 my_addr.sin_len = sizeof(my_addr);
60 my_addr.sin_addr = my_ip;
61 my_addr.sin_port = htons(my_port);
62 my_addr.sin_family = PF_INET;
64 ret = inet_aton(srv_address, &srv_ip);
66 /* not a valid ipv4 address */
67 return NT_STATUS_FOOBAR;
70 ret = bind(sock->fd, (struct sockaddr *)&my_addr, sizeof(my_addr));
72 /* TODO: we need to map from errno to NTSTATUS here! */
73 return NT_STATUS_FOOBAR;
76 ZERO_STRUCT(srv_addr);
77 #ifdef HAVE_SOCK_SIN_LEN
78 srv_addr.sin_len = sizeof(srv_addr);
80 srv_addr.sin_addr = srv_ip;
81 srv_addr.sin_port = htons(srv_port);
82 srv_addr.sin_family = PF_INET;
84 if (!(flags & SOCKET_FLAG_BLOCK)) {
85 ret = set_blocking(sock->fd, False);
87 /* TODO: we need to map from errno to NTSTATUS here! */
88 return NT_STATUS_FOOBAR;
92 ret = connect(sock->fd, (const struct sockaddr *)&srv_addr, sizeof(srv_addr));
94 /* TODO: we need to map from errno to NTSTATUS here! */
95 return NT_STATUS_FOOBAR;
98 sock->state = SOCKET_STATE_CLIENT_CONNECTED;
103 static NTSTATUS ipv4_tcp_listen(struct socket_context *sock,
104 const char *my_address, int port,
105 int queue_size, uint32_t flags)
107 struct sockaddr_in my_addr;
108 struct in_addr ip_addr;
111 ret = inet_aton(my_address, &ip_addr);
113 /* not a valid ipv4 address */
114 return NT_STATUS_FOOBAR;
117 ZERO_STRUCT(my_addr);
118 #ifdef HAVE_SOCK_SIN_LEN
119 my_addr.sin_len = sizeof(my_addr);
121 my_addr.sin_addr = ip_addr;
122 my_addr.sin_port = htons(port);
123 my_addr.sin_family = PF_INET;
125 ret = bind(sock->fd, (struct sockaddr *)&my_addr, sizeof(my_addr));
127 /* TODO: we need to map from errno to NTSTATUS here! */
128 return NT_STATUS_FOOBAR;
131 ret = listen(sock->fd, queue_size);
133 /* TODO: we need to map from errno to NTSTATUS here! */
134 return NT_STATUS_FOOBAR;
137 if (!(flags & SOCKET_FLAG_BLOCK)) {
138 ret = set_blocking(sock->fd, False);
140 /* TODO: we need to map from errno to NTSTATUS here! */
141 return NT_STATUS_FOOBAR;
145 sock->state= SOCKET_STATE_SERVER_LISTEN;
150 static NTSTATUS ipv4_tcp_accept(struct socket_context *sock, struct socket_context **new_sock, uint32_t flags)
152 struct sockaddr_in cli_addr;
153 socklen_t cli_addr_len = sizeof(cli_addr);
156 new_fd = accept(sock->fd, (struct sockaddr *)&cli_addr, &cli_addr_len);
158 /* TODO: we need to map from errno to NTSTATUS here! */
159 return NT_STATUS_FOOBAR;
162 /* TODO: we could add a 'accept_check' hook here
163 * which get the black/white lists via socket_set_accept_filter()
164 * or something like that
168 (*new_sock) = talloc_p(NULL, struct socket_context);
171 return NT_STATUS_NO_MEMORY;
174 /* copy the socket_context */
175 (*new_sock)->type = sock->type;
176 (*new_sock)->state = SOCKET_STATE_SERVER_CONNECTED;
177 (*new_sock)->flags = flags;
179 (*new_sock)->fd = new_fd;
181 (*new_sock)->private_data = NULL;
182 (*new_sock)->ops = sock->ops;
187 static NTSTATUS ipv4_tcp_recv(struct socket_context *sock, TALLOC_CTX *mem_ctx,
188 DATA_BLOB *blob, size_t wantlen, uint32_t flags)
194 buf = talloc(mem_ctx, wantlen);
196 return NT_STATUS_NO_MEMORY;
199 /* TODO: we need to map all flags here */
200 if (flags & SOCKET_FLAG_PEEK) {
204 if (!(flags & SOCKET_FLAG_BLOCK)) {
205 flgs |= MSG_DONTWAIT;
208 if (flags & SOCKET_FLAG_BLOCK) {
212 gotlen = recv(sock->fd, buf, wantlen, flgs);
215 return NT_STATUS_END_OF_FILE;
216 } else if (gotlen == -1) {
217 NTSTATUS status = NT_STATUS_UNSUCCESSFUL;
224 status = NT_STATUS_INVALID_PARAMETER;
228 status = STATUS_MORE_ENTRIES;
231 status = NT_STATUS_CONNECTION_REFUSED;
238 blob->length = gotlen;
239 blob->data = talloc_realloc(mem_ctx, buf, gotlen);
241 return NT_STATUS_NO_MEMORY;
247 static NTSTATUS ipv4_tcp_send(struct socket_context *sock, TALLOC_CTX *mem_ctx,
248 const DATA_BLOB *blob, size_t *sendlen, uint32_t flags)
253 /* TODO: we need to map all flags here */
254 if (!(flags & SOCKET_FLAG_BLOCK)) {
255 flgs |= MSG_DONTWAIT;
258 len = send(sock->fd, blob->data, blob->length, flgs);
260 NTSTATUS status = NT_STATUS_UNSUCCESSFUL;
266 status = NT_STATUS_INVALID_PARAMETER;
269 status = NT_STATUS_INVALID_BUFFER_SIZE;
272 /*case EWOULDBLOCK: this is an alis of EAGAIN --metze */
275 status = STATUS_MORE_ENTRIES;
278 status = NT_STATUS_FOOBAR;
281 status = NT_STATUS_NO_MEMORY;
284 status = NT_STATUS_CONNECTION_DISCONNECTED;
295 static NTSTATUS ipv4_tcp_set_option(struct socket_context *sock, const char *option, const char *val)
297 set_socket_options(sock->fd, option);
301 static char *ipv4_tcp_get_peer_name(struct socket_context *sock, TALLOC_CTX *mem_ctx)
303 struct sockaddr_in peer_addr;
304 socklen_t len = sizeof(peer_addr);
308 ret = getpeername(sock->fd, (struct sockaddr *)&peer_addr, &len);
313 he = gethostbyaddr((char *)&peer_addr.sin_addr, sizeof(peer_addr.sin_addr), AF_INET);
318 return talloc_strdup(mem_ctx, he->h_name);
321 static char *ipv4_tcp_get_peer_addr(struct socket_context *sock, TALLOC_CTX *mem_ctx)
323 struct sockaddr_in peer_addr;
324 socklen_t len = sizeof(peer_addr);
327 ret = getpeername(sock->fd, (struct sockaddr *)&peer_addr, &len);
332 return talloc_strdup(mem_ctx, inet_ntoa(peer_addr.sin_addr));
335 static int ipv4_tcp_get_peer_port(struct socket_context *sock)
337 struct sockaddr_in peer_addr;
338 socklen_t len = sizeof(peer_addr);
341 ret = getpeername(sock->fd, (struct sockaddr *)&peer_addr, &len);
346 return ntohs(peer_addr.sin_port);
349 static char *ipv4_tcp_get_my_addr(struct socket_context *sock, TALLOC_CTX *mem_ctx)
351 struct sockaddr_in my_addr;
352 socklen_t len = sizeof(my_addr);
355 ret = getsockname(sock->fd, (struct sockaddr *)&my_addr, &len);
360 return talloc_strdup(mem_ctx, inet_ntoa(my_addr.sin_addr));
363 static int ipv4_tcp_get_my_port(struct socket_context *sock)
365 struct sockaddr_in my_addr;
366 socklen_t len = sizeof(my_addr);
369 ret = getsockname(sock->fd, (struct sockaddr *)&my_addr, &len);
374 return ntohs(my_addr.sin_port);
377 static int ipv4_tcp_get_fd(struct socket_context *sock)
382 static const struct socket_ops ipv4_tcp_ops = {
384 .type = SOCKET_TYPE_STREAM,
386 .init = ipv4_tcp_init,
387 .connect = ipv4_tcp_connect,
388 .listen = ipv4_tcp_listen,
389 .accept = ipv4_tcp_accept,
390 .recv = ipv4_tcp_recv,
391 .send = ipv4_tcp_send,
392 .close = ipv4_tcp_close,
394 .set_option = ipv4_tcp_set_option,
396 .get_peer_name = ipv4_tcp_get_peer_name,
397 .get_peer_addr = ipv4_tcp_get_peer_addr,
398 .get_peer_port = ipv4_tcp_get_peer_port,
399 .get_my_addr = ipv4_tcp_get_my_addr,
400 .get_my_port = ipv4_tcp_get_my_port,
402 .get_fd = ipv4_tcp_get_fd
405 const struct socket_ops *socket_ipv4_ops(void)
407 return &ipv4_tcp_ops;
410 NTSTATUS socket_ipv4_init(void)