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 3 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, see <http://www.gnu.org/licenses/>.
20 #ifndef _SAMBA_SOCKET_H
21 #define _SAMBA_SOCKET_H
23 #include "lib/events/events.h"
25 struct socket_context;
32 struct socket_address {
36 struct sockaddr *sockaddr;
43 NTSTATUS (*fn_init)(struct socket_context *sock);
46 NTSTATUS (*fn_connect)(struct socket_context *sock,
47 const struct socket_address *my_address,
48 const struct socket_address *server_address,
51 /* complete a non-blocking connect */
52 NTSTATUS (*fn_connect_complete)(struct socket_context *sock,
56 NTSTATUS (*fn_listen)(struct socket_context *sock,
57 const struct socket_address *my_address,
58 int queue_size, uint32_t flags);
59 NTSTATUS (*fn_accept)(struct socket_context *sock,
60 struct socket_context **new_sock);
63 NTSTATUS (*fn_recv)(struct socket_context *sock, void *buf,
64 size_t wantlen, size_t *nread);
65 NTSTATUS (*fn_send)(struct socket_context *sock,
66 const DATA_BLOB *blob, size_t *sendlen);
68 NTSTATUS (*fn_sendto)(struct socket_context *sock,
69 const DATA_BLOB *blob, size_t *sendlen,
70 const struct socket_address *dest_addr);
71 NTSTATUS (*fn_recvfrom)(struct socket_context *sock,
72 void *buf, size_t wantlen, size_t *nread,
73 TALLOC_CTX *addr_ctx, struct socket_address **src_addr);
74 NTSTATUS (*fn_pending)(struct socket_context *sock, size_t *npending);
76 void (*fn_close)(struct socket_context *sock);
78 NTSTATUS (*fn_set_option)(struct socket_context *sock, const char *option, const char *val);
80 char *(*fn_get_peer_name)(struct socket_context *sock, TALLOC_CTX *mem_ctx);
81 struct socket_address *(*fn_get_peer_addr)(struct socket_context *sock, TALLOC_CTX *mem_ctx);
82 struct socket_address *(*fn_get_my_addr)(struct socket_context *sock, TALLOC_CTX *mem_ctx);
84 int (*fn_get_fd)(struct socket_context *sock);
88 SOCKET_STATE_UNDEFINED,
90 SOCKET_STATE_CLIENT_START,
91 SOCKET_STATE_CLIENT_CONNECTED,
92 SOCKET_STATE_CLIENT_STARTTLS,
93 SOCKET_STATE_CLIENT_ERROR,
95 SOCKET_STATE_SERVER_LISTEN,
96 SOCKET_STATE_SERVER_CONNECTED,
97 SOCKET_STATE_SERVER_STARTTLS,
98 SOCKET_STATE_SERVER_ERROR
101 #define SOCKET_FLAG_BLOCK 0x00000001
102 #define SOCKET_FLAG_PEEK 0x00000002
103 #define SOCKET_FLAG_TESTNONBLOCK 0x00000004
104 #define SOCKET_FLAG_ENCRYPT 0x00000008 /* This socket
105 * implementation requires
107 * consistant, because it
108 * is encrypting data.
110 * TESTNONBLOCK case */
111 #define SOCKET_FLAG_NOCLOSE 0x00000010 /* don't auto-close on free */
114 struct socket_context {
115 enum socket_type type;
116 enum socket_state state;
122 const struct socket_ops *ops;
123 const char *backend_name;
125 /* specific to the ip backend */
129 struct resolve_context;
132 NTSTATUS socket_create_with_ops(TALLOC_CTX *mem_ctx, const struct socket_ops *ops,
133 struct socket_context **new_sock,
134 enum socket_type type, uint32_t flags);
135 NTSTATUS socket_create(const char *name, enum socket_type type,
136 struct socket_context **new_sock, uint32_t flags);
137 NTSTATUS socket_connect(struct socket_context *sock,
138 const struct socket_address *my_address,
139 const struct socket_address *server_address,
141 NTSTATUS socket_connect_complete(struct socket_context *sock, uint32_t flags);
142 NTSTATUS socket_listen(struct socket_context *sock,
143 const struct socket_address *my_address,
144 int queue_size, uint32_t flags);
145 NTSTATUS socket_accept(struct socket_context *sock, struct socket_context **new_sock);
146 NTSTATUS socket_recv(struct socket_context *sock, void *buf,
147 size_t wantlen, size_t *nread);
148 NTSTATUS socket_recvfrom(struct socket_context *sock, void *buf,
149 size_t wantlen, size_t *nread,
150 TALLOC_CTX *addr_ctx, struct socket_address **src_addr);
151 NTSTATUS socket_send(struct socket_context *sock,
152 const DATA_BLOB *blob, size_t *sendlen);
153 NTSTATUS socket_sendto(struct socket_context *sock,
154 const DATA_BLOB *blob, size_t *sendlen,
155 const struct socket_address *dest_addr);
156 NTSTATUS socket_pending(struct socket_context *sock, size_t *npending);
157 NTSTATUS socket_set_option(struct socket_context *sock, const char *option, const char *val);
158 char *socket_get_peer_name(struct socket_context *sock, TALLOC_CTX *mem_ctx);
159 struct socket_address *socket_get_peer_addr(struct socket_context *sock, TALLOC_CTX *mem_ctx);
160 struct socket_address *socket_get_my_addr(struct socket_context *sock, TALLOC_CTX *mem_ctx);
161 int socket_get_fd(struct socket_context *sock);
162 NTSTATUS socket_dup(struct socket_context *sock);
163 struct socket_address *socket_address_from_strings(TALLOC_CTX *mem_ctx,
167 struct socket_address *socket_address_from_sockaddr(TALLOC_CTX *mem_ctx,
168 struct sockaddr *sockaddr,
170 const struct socket_ops *socket_getops_byname(const char *name, enum socket_type type);
171 bool allow_access(TALLOC_CTX *mem_ctx,
172 const char **deny_list, const char **allow_list,
173 const char *cname, const char *caddr);
174 bool socket_check_access(struct socket_context *sock,
175 const char *service_name,
176 const char **allow_list, const char **deny_list);
178 struct composite_context *socket_connect_send(struct socket_context *sock,
179 struct socket_address *my_address,
180 struct socket_address *server_address,
182 struct resolve_context *resolve_ctx,
183 struct event_context *event_ctx);
184 NTSTATUS socket_connect_recv(struct composite_context *ctx);
185 NTSTATUS socket_connect_ev(struct socket_context *sock,
186 struct socket_address *my_address,
187 struct socket_address *server_address,
189 struct resolve_context *resolve_ctx,
190 struct event_context *ev);
192 struct composite_context *socket_connect_multi_send(TALLOC_CTX *mem_ctx,
193 const char *server_address,
194 int num_server_ports,
195 uint16_t *server_ports,
196 struct resolve_context *resolve_ctx,
197 struct event_context *event_ctx);
198 NTSTATUS socket_connect_multi_recv(struct composite_context *ctx,
200 struct socket_context **result,
202 NTSTATUS socket_connect_multi(TALLOC_CTX *mem_ctx, const char *server_address,
203 int num_server_ports, uint16_t *server_ports,
204 struct resolve_context *resolve_ctx,
205 struct event_context *event_ctx,
206 struct socket_context **result,
208 void set_socket_options(int fd, const char *options);
209 void socket_set_flags(struct socket_context *socket, unsigned flags);
211 #endif /* _SAMBA_SOCKET_H */