038ce3b3b6882c8795e5750e062808eb7d5908e3
[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
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
26
27
28 /*
29   approximate errno mapping
30 */
31 static NTSTATUS unixdom_error(int ernum)
32 {
33         switch (ernum) {
34         case EBADF:
35         case ENOTCONN:
36         case ENOTSOCK:
37         case EFAULT:
38         case EINVAL:
39                 return NT_STATUS_INVALID_PARAMETER;
40         case EAGAIN:
41         case EINTR:
42                 return STATUS_MORE_ENTRIES;
43         case ECONNREFUSED:
44                 return NT_STATUS_CONNECTION_REFUSED;
45         case ENOBUFS:
46         case ENOMEM:
47                 return NT_STATUS_NO_MEMORY;
48         case ENFILE:
49         case EMFILE:
50                 return NT_STATUS_INSUFFICIENT_RESOURCES;
51         case EPIPE:
52                 return NT_STATUS_CONNECTION_DISCONNECTED;
53         case EMSGSIZE:
54                 return NT_STATUS_INVALID_BUFFER_SIZE;
55         }
56         
57         return NT_STATUS_UNSUCCESSFUL;
58 }
59
60 static NTSTATUS unixdom_init(struct socket_context *sock)
61 {
62         sock->fd = socket(PF_UNIX, SOCK_STREAM, 0);
63         if (sock->fd == -1) {
64                 return NT_STATUS_INSUFFICIENT_RESOURCES;
65         }
66         sock->private_data = NULL;
67
68         return NT_STATUS_OK;
69 }
70
71 static void unixdom_close(struct socket_context *sock)
72 {
73         close(sock->fd);
74         /* if we were listening, then don't leave the socket lying
75            around in the filesystem */
76
77 #if 0
78         /* FIXME - this doesn't work after fork(), etc */
79         if (sock->private_data) {
80                 unlink((const char *)sock->private_data);
81         }
82 #endif
83 }
84
85 static NTSTATUS unixdom_connect(struct socket_context *sock,
86                                 const char *my_address, int my_port,
87                                 const char *srv_address, int srv_port,
88                                 uint32_t flags)
89 {
90         struct sockaddr_un srv_addr;
91         int ret;
92
93         if (strlen(srv_address)+1 > sizeof(srv_addr.sun_path)) {
94                 return NT_STATUS_INVALID_PARAMETER;
95         }
96
97         ZERO_STRUCT(srv_addr);
98         srv_addr.sun_family = AF_UNIX;
99         strncpy(srv_addr.sun_path, srv_address, sizeof(srv_addr.sun_path));
100
101         if (!(flags & SOCKET_FLAG_BLOCK)) {
102                 ret = set_blocking(sock->fd, False);
103                 if (ret == -1) {
104                         return NT_STATUS_INVALID_PARAMETER;
105                 }
106         }
107
108         ret = connect(sock->fd, (const struct sockaddr *)&srv_addr, sizeof(srv_addr));
109         if (ret == -1) {
110                 return unixdom_error(errno);
111         }
112
113         sock->state = SOCKET_STATE_CLIENT_CONNECTED;
114
115         return NT_STATUS_OK;
116 }
117
118 static NTSTATUS unixdom_listen(struct socket_context *sock,
119                                const char *my_address, int port,
120                                int queue_size, uint32_t flags)
121 {
122         struct sockaddr_un my_addr;
123         int ret;
124
125         if (strlen(my_address)+1 > sizeof(my_addr.sun_path)) {
126                 return NT_STATUS_INVALID_PARAMETER;
127         }
128
129         /* delete if it already exists */
130         unlink(my_address);
131
132         ZERO_STRUCT(my_addr);
133         my_addr.sun_family = AF_UNIX;
134         strncpy(my_addr.sun_path, my_address, sizeof(my_addr.sun_path));
135
136         ret = bind(sock->fd, (struct sockaddr *)&my_addr, sizeof(my_addr));
137         if (ret == -1) {
138                 return unixdom_error(errno);
139         }
140
141         ret = listen(sock->fd, queue_size);
142         if (ret == -1) {
143                 return unixdom_error(errno);
144         }
145
146         if (!(flags & SOCKET_FLAG_BLOCK)) {
147                 ret = set_blocking(sock->fd, False);
148                 if (ret == -1) {
149                         return unixdom_error(errno);
150                 }
151         }
152
153         sock->state = SOCKET_STATE_SERVER_LISTEN;
154         sock->private_data = (void *)talloc_strdup(sock, my_address);
155
156         return NT_STATUS_OK;
157 }
158
159 static NTSTATUS unixdom_accept(struct socket_context *sock, 
160                                struct socket_context **new_sock, 
161                                uint32_t flags)
162 {
163         struct sockaddr_un cli_addr;
164         socklen_t cli_addr_len = sizeof(cli_addr);
165         int new_fd;
166
167         new_fd = accept(sock->fd, (struct sockaddr *)&cli_addr, &cli_addr_len);
168         if (new_fd == -1) {
169                 return unixdom_error(errno);
170         }
171
172         (*new_sock) = talloc_p(NULL, struct socket_context);
173         if (!(*new_sock)) {
174                 close(new_fd);
175                 return NT_STATUS_NO_MEMORY;
176         }
177
178         /* copy the socket_context */
179         (*new_sock)->type               = sock->type;
180         (*new_sock)->state              = SOCKET_STATE_SERVER_CONNECTED;
181         (*new_sock)->flags              = flags;
182
183         (*new_sock)->fd                 = new_fd;
184
185         (*new_sock)->private_data       = NULL;
186         (*new_sock)->ops                = sock->ops;
187
188         return NT_STATUS_OK;
189 }
190
191 static NTSTATUS unixdom_recv(struct socket_context *sock, TALLOC_CTX *mem_ctx,
192                              DATA_BLOB *blob, size_t wantlen, uint32_t flags)
193 {
194         ssize_t gotlen;
195         void *buf;
196         int flgs = 0;
197
198         buf = talloc(mem_ctx, wantlen);
199         if (!buf) {
200                 return NT_STATUS_NO_MEMORY;
201         }
202
203         /* TODO: we need to map all flags here */
204         if (flags & SOCKET_FLAG_PEEK) {
205                 flgs |= MSG_PEEK;
206         }
207
208         if (!(flags & SOCKET_FLAG_BLOCK)) {
209                 flgs |= MSG_DONTWAIT;
210         }
211
212         if (flags & SOCKET_FLAG_BLOCK) {
213                 flgs |= MSG_WAITALL;
214         }
215
216         gotlen = recv(sock->fd, buf, wantlen, flgs);
217         if (gotlen == 0) {
218                 talloc_free(buf);
219                 return NT_STATUS_END_OF_FILE;
220         } else if (gotlen == -1) {
221                 NTSTATUS status = unixdom_error(errno);
222                 talloc_free(buf);
223                 return status;
224         }
225
226         blob->length = gotlen;
227         blob->data = talloc_realloc(mem_ctx, buf, gotlen);
228         if (!blob->data) {
229                 return NT_STATUS_NO_MEMORY;
230         }
231
232         return NT_STATUS_OK;
233 }
234
235 static NTSTATUS unixdom_send(struct socket_context *sock, TALLOC_CTX *mem_ctx,
236                              const DATA_BLOB *blob, size_t *sendlen, uint32_t flags)
237 {
238         ssize_t len;
239         int flgs = 0;
240
241         *sendlen = 0;
242
243         /* TODO: we need to map all flags here */
244         if (!(flags & SOCKET_FLAG_BLOCK)) {
245                 flgs |= MSG_DONTWAIT;
246         }
247
248         len = send(sock->fd, blob->data, blob->length, flgs);
249         if (len == -1) {
250                 return unixdom_error(errno);
251         }       
252
253         *sendlen = len;
254
255         return NT_STATUS_OK;
256 }
257
258 static NTSTATUS unixdom_set_option(struct socket_context *sock, 
259                                    const char *option, const char *val)
260 {
261         set_socket_options(sock->fd, option);
262         return NT_STATUS_OK;
263 }
264
265 static char *unixdom_get_peer_name(struct socket_context *sock, TALLOC_CTX *mem_ctx)
266 {
267         return talloc_strdup(mem_ctx, "LOCAL/unixdom");
268 }
269
270 static char *unixdom_get_peer_addr(struct socket_context *sock, TALLOC_CTX *mem_ctx)
271 {
272         return talloc_strdup(mem_ctx, "LOCAL/unixdom");
273 }
274
275 static int unixdom_get_peer_port(struct socket_context *sock)
276 {
277         return 0;
278 }
279
280 static char *unixdom_get_my_addr(struct socket_context *sock, TALLOC_CTX *mem_ctx)
281 {
282         return talloc_strdup(mem_ctx, "LOCAL/unixdom");
283 }
284
285 static int unixdom_get_my_port(struct socket_context *sock)
286 {
287         return 0;
288 }
289
290 static int unixdom_get_fd(struct socket_context *sock)
291 {
292         return sock->fd;
293 }
294
295 static const struct socket_ops unixdom_ops = {
296         .name           = "unix",
297         .type           = SOCKET_TYPE_STREAM,
298
299         .init           = unixdom_init,
300         .connect        = unixdom_connect,
301         .listen         = unixdom_listen,
302         .accept         = unixdom_accept,
303         .recv           = unixdom_recv,
304         .send           = unixdom_send,
305         .close          = unixdom_close,
306
307         .set_option     = unixdom_set_option,
308
309         .get_peer_name  = unixdom_get_peer_name,
310         .get_peer_addr  = unixdom_get_peer_addr,
311         .get_peer_port  = unixdom_get_peer_port,
312         .get_my_addr    = unixdom_get_my_addr,
313         .get_my_port    = unixdom_get_my_port,
314
315         .get_fd         = unixdom_get_fd
316 };
317
318 const struct socket_ops *socket_unixdom_ops(void)
319 {
320         return &unixdom_ops;
321 }
322
323 NTSTATUS socket_unixdom_init(void)
324 {
325         return NT_STATUS_OK;
326 }