f5ee84a7cc1cdc304732976bdfa98219942ba414
[bbaumbach/samba-autobuild/.git] / source4 / lib / socket / socket.c
1 /* 
2    Unix SMB/CIFS implementation.
3    Socket functions
4    Copyright (C) Stefan Metzmacher 2004
5    
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.
10    
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.
15    
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.
19 */
20
21 #include "includes.h"
22
23 /*
24   auto-close sockets on free
25 */
26 static int socket_destructor(void *ptr)
27 {
28         struct socket_context *sock = ptr;
29         if (sock->ops->close) {
30                 sock->ops->close(sock);
31         }
32         return 0;
33 }
34
35 NTSTATUS socket_create(const char *name, enum socket_type type, struct socket_context **new_sock, uint32_t flags)
36 {
37         NTSTATUS status;
38
39         (*new_sock) = talloc_p(NULL, struct socket_context);
40         if (!(*new_sock)) {
41                 return NT_STATUS_NO_MEMORY;
42         }
43
44         (*new_sock)->type = type;
45         (*new_sock)->state = SOCKET_STATE_UNDEFINED;
46         (*new_sock)->flags = flags;
47
48         (*new_sock)->fd = -1;
49
50         (*new_sock)->private_data = NULL;
51         (*new_sock)->ops = socket_getops_byname(name, type);
52         if (!(*new_sock)->ops) {
53                 talloc_free(*new_sock);
54                 return NT_STATUS_INVALID_PARAMETER;
55         }
56
57         status = (*new_sock)->ops->init((*new_sock));
58         if (!NT_STATUS_IS_OK(status)) {
59                 talloc_free(*new_sock);
60                 return status;
61         }
62
63         talloc_set_destructor(*new_sock, socket_destructor);
64
65         return NT_STATUS_OK;
66 }
67
68 void socket_destroy(struct socket_context *sock)
69 {
70         /* the close is handled by the destructor */
71         talloc_free(sock);
72 }
73
74 NTSTATUS socket_connect(struct socket_context *sock,
75                         const char *my_address, int my_port,
76                         const char *server_address, int server_port,
77                         uint32_t flags)
78 {
79         if (sock->type != SOCKET_TYPE_STREAM) {
80                 return NT_STATUS_INVALID_PARAMETER;
81         }
82
83         if (sock->state != SOCKET_STATE_UNDEFINED) {
84                 return NT_STATUS_INVALID_PARAMETER;
85         }
86
87         if (!sock->ops->connect) {
88                 return NT_STATUS_NOT_IMPLEMENTED;
89         }
90
91         return sock->ops->connect(sock, my_address, my_port, server_address, server_port, flags);
92 }
93
94 NTSTATUS socket_listen(struct socket_context *sock, const char *my_address, int port, int queue_size, uint32_t flags)
95 {
96         if (sock->type != SOCKET_TYPE_STREAM) {
97                 return NT_STATUS_INVALID_PARAMETER;
98         }
99
100         if (sock->state != SOCKET_STATE_UNDEFINED) {
101                 return NT_STATUS_INVALID_PARAMETER;
102         }
103
104         if (!sock->ops->listen) {
105                 return NT_STATUS_NOT_IMPLEMENTED;
106         }
107
108         return sock->ops->listen(sock, my_address, port, queue_size, flags);
109 }
110
111 NTSTATUS socket_accept(struct socket_context *sock, struct socket_context **new_sock, uint32_t flags)
112 {
113         NTSTATUS status;
114
115         if (sock->type != SOCKET_TYPE_STREAM) {
116                 return NT_STATUS_INVALID_PARAMETER;
117         }
118
119         if (sock->state != SOCKET_STATE_SERVER_LISTEN) {
120                 return NT_STATUS_INVALID_PARAMETER;
121         }
122
123         if (!sock->ops->accept) {
124                 return NT_STATUS_NOT_IMPLEMENTED;
125         }
126
127         status = sock->ops->accept(sock, new_sock, flags);
128
129         if (NT_STATUS_IS_OK(status)) {
130                 talloc_set_destructor(*new_sock, socket_destructor);
131         }
132
133         return status;
134 }
135
136 NTSTATUS socket_recv(struct socket_context *sock, TALLOC_CTX *mem_ctx,
137                      DATA_BLOB *blob, size_t wantlen, uint32_t flags)
138 {
139         if (sock->type != SOCKET_TYPE_STREAM) {
140                 return NT_STATUS_INVALID_PARAMETER;
141         }
142
143         if (sock->state != SOCKET_STATE_CLIENT_CONNECTED &&
144             sock->state != SOCKET_STATE_SERVER_CONNECTED) {
145                 return NT_STATUS_INVALID_PARAMETER;
146         }
147
148         if (!sock->ops->recv) {
149                 return NT_STATUS_NOT_IMPLEMENTED;
150         }
151
152         return sock->ops->recv(sock, mem_ctx, blob, wantlen, flags);
153 }
154
155 NTSTATUS socket_send(struct socket_context *sock, TALLOC_CTX *mem_ctx,
156                         const DATA_BLOB *blob, size_t *sendlen, uint32_t flags)
157 {
158         if (sock->type != SOCKET_TYPE_STREAM) {
159                 return NT_STATUS_INVALID_PARAMETER;
160         }
161
162         if (sock->state != SOCKET_STATE_CLIENT_CONNECTED &&
163             sock->state != SOCKET_STATE_SERVER_CONNECTED) {
164                 return NT_STATUS_INVALID_PARAMETER;
165         }
166
167         if (!sock->ops->send) {
168                 return NT_STATUS_NOT_IMPLEMENTED;
169         }
170
171         return sock->ops->send(sock, mem_ctx, blob, sendlen, flags);
172 }
173
174 NTSTATUS socket_set_option(struct socket_context *sock, const char *option, const char *val)
175 {
176         if (!sock->ops->set_option) {
177                 return NT_STATUS_NOT_IMPLEMENTED;
178         }
179
180         return sock->ops->set_option(sock, option, val);
181 }
182
183 char *socket_get_peer_name(struct socket_context *sock, TALLOC_CTX *mem_ctx)
184 {
185         if (!sock->ops->get_peer_name) {
186                 return NULL;
187         }
188
189         return sock->ops->get_peer_name(sock, mem_ctx);
190 }
191
192 char *socket_get_peer_addr(struct socket_context *sock, TALLOC_CTX *mem_ctx)
193 {
194         if (!sock->ops->get_peer_addr) {
195                 return NULL;
196         }
197
198         return sock->ops->get_peer_addr(sock, mem_ctx);
199 }
200
201 int socket_get_peer_port(struct socket_context *sock)
202 {
203         if (!sock->ops->get_peer_port) {
204                 return -1;
205         }
206
207         return sock->ops->get_peer_port(sock);
208 }
209
210 char *socket_get_my_addr(struct socket_context *sock, TALLOC_CTX *mem_ctx)
211 {
212         if (!sock->ops->get_my_addr) {
213                 return NULL;
214         }
215
216         return sock->ops->get_my_addr(sock, mem_ctx);
217 }
218
219 int socket_get_my_port(struct socket_context *sock)
220 {
221         if (!sock->ops->get_my_port) {
222                 return -1;
223         }
224
225         return sock->ops->get_my_port(sock);
226 }
227
228 int socket_get_fd(struct socket_context *sock)
229 {
230         if (!sock->ops->get_fd) {
231                 return -1;
232         }
233
234         return sock->ops->get_fd(sock);
235 }
236
237 const struct socket_ops *socket_getops_byname(const char *name, enum socket_type type)
238 {
239         if (strcmp("ip", name) == 0 || 
240             strcmp("ipv4", name) == 0) {
241                 return socket_ipv4_ops();
242         }
243
244         if (strcmp("unix", name) == 0) {
245                 return socket_unixdom_ops();
246         }
247
248         return NULL;
249 }