3feb1bf234f6e331fb100c98a52dbb3b3ae470f2
[bbaumbach/samba-autobuild/.git] / source4 / lib / socket / socket_unix.c
1 /* 
2    Unix SMB/CIFS implementation.
3
4    unix domain socket functions
5
6    Copyright (C) Stefan Metzmacher 2004
7    Copyright (C) Andrew Tridgell 2004-2005
8    
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 2 of the License, or
12    (at your option) any later version.
13    
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.
18    
19    You should have received a copy of the GNU General Public License
20    along with this program; if not, write to the Free Software
21    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
22 */
23
24 #include "includes.h"
25 #include "lib/socket/socket.h"
26 #include "system/network.h"
27 #include "system/filesys.h"
28
29
30
31 /*
32   approximate errno mapping
33 */
34 static NTSTATUS unixdom_error(int ernum)
35 {
36         return map_nt_error_from_unix(ernum);
37 }
38
39 static NTSTATUS unixdom_init(struct socket_context *sock)
40 {
41         int type;
42
43         switch (sock->type) {
44         case SOCKET_TYPE_STREAM:
45                 type = SOCK_STREAM;
46                 break;
47         case SOCKET_TYPE_DGRAM:
48                 type = SOCK_DGRAM;
49                 break;
50         default:
51                 return NT_STATUS_INVALID_PARAMETER;
52         }
53
54         sock->fd = socket(PF_UNIX, type, 0);
55         if (sock->fd == -1) {
56                 return map_nt_error_from_unix(errno);
57         }
58         sock->private_data = NULL;
59
60         sock->backend_name = "unix";
61
62         return NT_STATUS_OK;
63 }
64
65 static void unixdom_close(struct socket_context *sock)
66 {
67         close(sock->fd);
68 }
69
70 static NTSTATUS unixdom_connect_complete(struct socket_context *sock, uint32_t flags)
71 {
72         int error=0, ret;
73         socklen_t len = sizeof(error);
74
75         /* check for any errors that may have occurred - this is needed
76            for non-blocking connect */
77         ret = getsockopt(sock->fd, SOL_SOCKET, SO_ERROR, &error, &len);
78         if (ret == -1) {
79                 return map_nt_error_from_unix(errno);
80         }
81         if (error != 0) {
82                 return map_nt_error_from_unix(error);
83         }
84
85         if (!(flags & SOCKET_FLAG_BLOCK)) {
86                 ret = set_blocking(sock->fd, False);
87                 if (ret == -1) {
88                         return map_nt_error_from_unix(errno);
89                 }
90         }
91
92         sock->state = SOCKET_STATE_CLIENT_CONNECTED;
93
94         return NT_STATUS_OK;
95 }
96
97 static NTSTATUS unixdom_connect(struct socket_context *sock,
98                                 const struct socket_address *my_address, 
99                                 const struct socket_address *srv_address, 
100                                 uint32_t flags)
101 {
102         int ret;
103
104         if (srv_address->sockaddr) {
105                 ret = connect(sock->fd, srv_address->sockaddr, srv_address->sockaddrlen);
106         } else {
107                 struct sockaddr_un srv_addr;
108                 if (strlen(srv_address->addr)+1 > sizeof(srv_addr.sun_path)) {
109                         return NT_STATUS_OBJECT_PATH_INVALID;
110                 }
111                 
112                 ZERO_STRUCT(srv_addr);
113                 srv_addr.sun_family = AF_UNIX;
114                 strncpy(srv_addr.sun_path, srv_address->addr, sizeof(srv_addr.sun_path));
115
116                 ret = connect(sock->fd, (const struct sockaddr *)&srv_addr, sizeof(srv_addr));
117         }
118         if (ret == -1) {
119                 return unixdom_error(errno);
120         }
121
122         return unixdom_connect_complete(sock, flags);
123 }
124
125 static NTSTATUS unixdom_listen(struct socket_context *sock,
126                                const struct socket_address *my_address, 
127                                int queue_size, uint32_t flags)
128 {
129         struct sockaddr_un my_addr;
130         int ret;
131
132         /* delete if it already exists */
133         if (my_address->addr) {
134                 unlink(my_address->addr);
135         }
136
137         if (my_address->sockaddr) {
138                 ret = bind(sock->fd, (struct sockaddr *)&my_addr, sizeof(my_addr));
139         } else if (my_address->addr == NULL) {
140                 return NT_STATUS_INVALID_PARAMETER;
141         } else {
142                 if (strlen(my_address->addr)+1 > sizeof(my_addr.sun_path)) {
143                         return NT_STATUS_OBJECT_PATH_INVALID;
144                 }
145                 
146                 
147                 ZERO_STRUCT(my_addr);
148                 my_addr.sun_family = AF_UNIX;
149                 strncpy(my_addr.sun_path, my_address->addr, sizeof(my_addr.sun_path));
150                 
151                 ret = bind(sock->fd, (struct sockaddr *)&my_addr, sizeof(my_addr));
152         }
153         if (ret == -1) {
154                 return unixdom_error(errno);
155         }
156
157         if (sock->type == SOCKET_TYPE_STREAM) {
158                 ret = listen(sock->fd, queue_size);
159                 if (ret == -1) {
160                         return unixdom_error(errno);
161                 }
162         }
163
164         if (!(flags & SOCKET_FLAG_BLOCK)) {
165                 ret = set_blocking(sock->fd, False);
166                 if (ret == -1) {
167                         return unixdom_error(errno);
168                 }
169         }
170
171         sock->state = SOCKET_STATE_SERVER_LISTEN;
172         sock->private_data = (void *)talloc_strdup(sock, my_address->addr);
173
174         return NT_STATUS_OK;
175 }
176
177 static NTSTATUS unixdom_accept(struct socket_context *sock, 
178                                struct socket_context **new_sock)
179 {
180         struct sockaddr_un cli_addr;
181         socklen_t cli_addr_len = sizeof(cli_addr);
182         int new_fd;
183
184         if (sock->type != SOCKET_TYPE_STREAM) {
185                 return NT_STATUS_INVALID_PARAMETER;
186         }
187
188         new_fd = accept(sock->fd, (struct sockaddr *)&cli_addr, &cli_addr_len);
189         if (new_fd == -1) {
190                 return unixdom_error(errno);
191         }
192
193         if (!(sock->flags & SOCKET_FLAG_BLOCK)) {
194                 int ret = set_blocking(new_fd, False);
195                 if (ret == -1) {
196                         close(new_fd);
197                         return map_nt_error_from_unix(errno);
198                 }
199         }
200
201         (*new_sock) = talloc(NULL, struct socket_context);
202         if (!(*new_sock)) {
203                 close(new_fd);
204                 return NT_STATUS_NO_MEMORY;
205         }
206
207         /* copy the socket_context */
208         (*new_sock)->type               = sock->type;
209         (*new_sock)->state              = SOCKET_STATE_SERVER_CONNECTED;
210         (*new_sock)->flags              = sock->flags;
211
212         (*new_sock)->fd                 = new_fd;
213
214         (*new_sock)->private_data       = NULL;
215         (*new_sock)->ops                = sock->ops;
216         (*new_sock)->backend_name       = sock->backend_name;
217
218         return NT_STATUS_OK;
219 }
220
221 static NTSTATUS unixdom_recv(struct socket_context *sock, void *buf, 
222                              size_t wantlen, size_t *nread, uint32_t flags)
223 {
224         ssize_t gotlen;
225         int flgs = 0;
226
227         /* TODO: we need to map all flags here */
228         if (flags & SOCKET_FLAG_PEEK) {
229                 flgs |= MSG_PEEK;
230         }
231
232         if (flags & SOCKET_FLAG_BLOCK) {
233                 flgs |= MSG_WAITALL;
234         }
235
236         *nread = 0;
237
238         gotlen = recv(sock->fd, buf, wantlen, flgs);
239         if (gotlen == 0) {
240                 return NT_STATUS_END_OF_FILE;
241         } else if (gotlen == -1) {
242                 return unixdom_error(errno);
243         }
244
245         *nread = gotlen;
246
247         return NT_STATUS_OK;
248 }
249
250 static NTSTATUS unixdom_send(struct socket_context *sock,
251                              const DATA_BLOB *blob, size_t *sendlen, uint32_t flags)
252 {
253         ssize_t len;
254         int flgs = 0;
255
256         *sendlen = 0;
257
258         len = send(sock->fd, blob->data, blob->length, flgs);
259         if (len == -1) {
260                 return unixdom_error(errno);
261         }       
262
263         *sendlen = len;
264
265         return NT_STATUS_OK;
266 }
267
268
269 static NTSTATUS unixdom_sendto(struct socket_context *sock, 
270                                const DATA_BLOB *blob, size_t *sendlen, uint32_t flags,
271                                const struct socket_address *dest)
272 {
273         ssize_t len;
274         int flgs = 0;
275         *sendlen = 0;
276                 
277         if (dest->sockaddr) {
278                 len = sendto(sock->fd, blob->data, blob->length, flgs, 
279                              dest->sockaddr, dest->sockaddrlen);
280         } else {
281                 struct sockaddr_un srv_addr;
282                 
283                 if (strlen(dest->addr)+1 > sizeof(srv_addr.sun_path)) {
284                         return NT_STATUS_OBJECT_PATH_INVALID;
285                 }
286                 
287                 ZERO_STRUCT(srv_addr);
288                 srv_addr.sun_family = AF_UNIX;
289                 strncpy(srv_addr.sun_path, dest->addr, sizeof(srv_addr.sun_path));
290                 
291                 len = sendto(sock->fd, blob->data, blob->length, flgs, 
292                              (struct sockaddr *)&srv_addr, sizeof(srv_addr));
293         }
294         if (len == -1) {
295                 return map_nt_error_from_unix(errno);
296         }       
297
298         *sendlen = len;
299
300         return NT_STATUS_OK;
301 }
302
303
304 static NTSTATUS unixdom_set_option(struct socket_context *sock, 
305                                    const char *option, const char *val)
306 {
307         return NT_STATUS_OK;
308 }
309
310 static char *unixdom_get_peer_name(struct socket_context *sock, TALLOC_CTX *mem_ctx)
311 {
312         return talloc_strdup(mem_ctx, "LOCAL/unixdom");
313 }
314
315 static struct socket_address *unixdom_get_peer_addr(struct socket_context *sock, TALLOC_CTX *mem_ctx)
316 {
317         struct sockaddr_in *peer_addr;
318         socklen_t len = sizeof(*peer_addr);
319         struct socket_address *peer;
320         int ret;
321
322         peer = talloc(mem_ctx, struct socket_address);
323         if (!peer) {
324                 return NULL;
325         }
326         
327         peer->family = sock->backend_name;
328         peer_addr = talloc(peer, struct sockaddr_in);
329         if (!peer_addr) {
330                 talloc_free(peer);
331                 return NULL;
332         }
333
334         peer->sockaddr = (struct sockaddr *)peer_addr;
335
336         ret = getpeername(sock->fd, peer->sockaddr, &len);
337         if (ret == -1) {
338                 talloc_free(peer);
339                 return NULL;
340         }
341
342         peer->sockaddrlen = len;
343
344         peer->port = 0;
345         peer->addr = talloc_strdup(peer, "LOCAL/unixdom");
346         if (!peer->addr) {
347                 talloc_free(peer);
348                 return NULL;
349         }
350
351         return peer;
352 }
353
354 static struct socket_address *unixdom_get_my_addr(struct socket_context *sock, TALLOC_CTX *mem_ctx)
355 {
356         struct sockaddr_in *local_addr;
357         socklen_t len = sizeof(*local_addr);
358         struct socket_address *local;
359         int ret;
360         
361         local = talloc(mem_ctx, struct socket_address);
362         if (!local) {
363                 return NULL;
364         }
365         
366         local->family = sock->backend_name;
367         local_addr = talloc(local, struct sockaddr_in);
368         if (!local_addr) {
369                 talloc_free(local);
370                 return NULL;
371         }
372
373         local->sockaddr = (struct sockaddr *)local_addr;
374
375         ret = getsockname(sock->fd, local->sockaddr, &len);
376         if (ret == -1) {
377                 talloc_free(local);
378                 return NULL;
379         }
380
381         local->sockaddrlen = len;
382
383         local->port = 0;
384         local->addr = talloc_strdup(local, "LOCAL/unixdom");
385         if (!local->addr) {
386                 talloc_free(local);
387                 return NULL;
388         }
389
390         return local;
391 }
392
393 static int unixdom_get_fd(struct socket_context *sock)
394 {
395         return sock->fd;
396 }
397
398 static NTSTATUS unixdom_pending(struct socket_context *sock, size_t *npending)
399 {
400         int value = 0;
401         if (ioctl(sock->fd, FIONREAD, &value) == 0) {
402                 *npending = value;
403                 return NT_STATUS_OK;
404         }
405         return map_nt_error_from_unix(errno);
406 }
407
408 static const struct socket_ops unixdom_ops = {
409         .name                   = "unix",
410         .fn_init                = unixdom_init,
411         .fn_connect             = unixdom_connect,
412         .fn_connect_complete    = unixdom_connect_complete,
413         .fn_listen              = unixdom_listen,
414         .fn_accept              = unixdom_accept,
415         .fn_recv                = unixdom_recv,
416         .fn_send                = unixdom_send,
417         .fn_sendto              = unixdom_sendto,
418         .fn_close               = unixdom_close,
419         .fn_pending             = unixdom_pending,
420
421         .fn_set_option          = unixdom_set_option,
422
423         .fn_get_peer_name       = unixdom_get_peer_name,
424         .fn_get_peer_addr       = unixdom_get_peer_addr,
425         .fn_get_my_addr         = unixdom_get_my_addr,
426
427         .fn_get_fd              = unixdom_get_fd
428 };
429
430 const struct socket_ops *socket_unixdom_ops(enum socket_type type)
431 {
432         return &unixdom_ops;
433 }