source4/lib/socket/socket.h

   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 #ifndef _SAMBA_SOCKET_H
  22 #define _SAMBA_SOCKET_H
  23
  24 #include "lib/events/events.h"
  25
  26 struct socket_context;
  27
  28 enum socket_type {
  29         SOCKET_TYPE_STREAM,
  30         SOCKET_TYPE_DGRAM
  31 };
  32
  33 struct socket_address {
  34         const char *family;
  35         char *addr;
  36         int port;
  37         struct sockaddr *sockaddr;
  38         size_t sockaddrlen;
  39 };
  40
  41 struct socket_ops {
  42         const char *name;
  43
  44         NTSTATUS (*fn_init)(struct socket_context *sock);
  45
  46         /* client ops */
  47         NTSTATUS (*fn_connect)(struct socket_context *sock,
  48                                const struct socket_address *my_address,
  49                                const struct socket_address *server_address,
  50                                uint32_t flags);
  51
  52         /* complete a non-blocking connect */
  53         NTSTATUS (*fn_connect_complete)(struct socket_context *sock,
  54                                         uint32_t flags);
  55
  56         /* server ops */
  57         NTSTATUS (*fn_listen)(struct socket_context *sock,
  58                               const struct socket_address *my_address,
  59                               int queue_size, uint32_t flags);
  60         NTSTATUS (*fn_accept)(struct socket_context *sock,
  61                               struct socket_context **new_sock);
  62
  63         /* general ops */
  64         NTSTATUS (*fn_recv)(struct socket_context *sock, void *buf,
  65                             size_t wantlen, size_t *nread, uint32_t flags);
  66         NTSTATUS (*fn_send)(struct socket_context *sock,
  67                             const DATA_BLOB *blob, size_t *sendlen, uint32_t flags);
  68
  69         NTSTATUS (*fn_sendto)(struct socket_context *sock,
  70                               const DATA_BLOB *blob, size_t *sendlen, uint32_t flags,
  71                               const struct socket_address *dest_addr);
  72         NTSTATUS (*fn_recvfrom)(struct socket_context *sock,
  73                                 void *buf, size_t wantlen, size_t *nread, uint32_t flags,
  74                                 TALLOC_CTX *addr_ctx, struct socket_address **src_addr);
  75         NTSTATUS (*fn_pending)(struct socket_context *sock, size_t *npending);
  76
  77         void (*fn_close)(struct socket_context *sock);
  78
  79         NTSTATUS (*fn_set_option)(struct socket_context *sock, const char *option, const char *val);
  80
  81         char *(*fn_get_peer_name)(struct socket_context *sock, TALLOC_CTX *mem_ctx);
  82         struct socket_address *(*fn_get_peer_addr)(struct socket_context *sock, TALLOC_CTX *mem_ctx);
  83         struct socket_address *(*fn_get_my_addr)(struct socket_context *sock, TALLOC_CTX *mem_ctx);
  84
  85         int (*fn_get_fd)(struct socket_context *sock);
  86 };
  87
  88 enum socket_state {
  89         SOCKET_STATE_UNDEFINED,
  90
  91         SOCKET_STATE_CLIENT_START,
  92         SOCKET_STATE_CLIENT_CONNECTED,
  93         SOCKET_STATE_CLIENT_STARTTLS,
  94         SOCKET_STATE_CLIENT_ERROR,
  95
  96         SOCKET_STATE_SERVER_LISTEN,
  97         SOCKET_STATE_SERVER_CONNECTED,
  98         SOCKET_STATE_SERVER_STARTTLS,
  99         SOCKET_STATE_SERVER_ERROR
 100 };
 101
 102 #define SOCKET_FLAG_BLOCK        0x00000001
 103 #define SOCKET_FLAG_PEEK         0x00000002
 104 #define SOCKET_FLAG_TESTNONBLOCK 0x00000004
 105
 106 struct socket_context {
 107         enum socket_type type;
 108         enum socket_state state;
 109         uint32_t flags;
 110
 111         int fd;
 112
 113         void *private_data;
 114         const struct socket_ops *ops;
 115         const char *backend_name;
 116 };
 117
 118
 119 /* prototypes */
 120 NTSTATUS socket_create(const char *name, enum socket_type type,
 121                        struct socket_context **new_sock, uint32_t flags);
 122 NTSTATUS socket_connect(struct socket_context *sock,
 123                         const struct socket_address *my_address,
 124                         const struct socket_address *server_address,
 125                         uint32_t flags);
 126 NTSTATUS socket_connect_complete(struct socket_context *sock, uint32_t flags);
 127 NTSTATUS socket_listen(struct socket_context *sock,
 128                        const struct socket_address *my_address,
 129                        int queue_size, uint32_t flags);
 130 NTSTATUS socket_accept(struct socket_context *sock, struct socket_context **new_sock);
 131 NTSTATUS socket_recv(struct socket_context *sock, void *buf,
 132                      size_t wantlen, size_t *nread, uint32_t flags);
 133 NTSTATUS socket_recvfrom(struct socket_context *sock, void *buf,
 134                          size_t wantlen, size_t *nread, uint32_t flags,
 135                          TALLOC_CTX *addr_ctx, struct socket_address **src_addr);
 136 NTSTATUS socket_send(struct socket_context *sock,
 137                      const DATA_BLOB *blob, size_t *sendlen, uint32_t flags);
 138 NTSTATUS socket_sendto(struct socket_context *sock,
 139                        const DATA_BLOB *blob, size_t *sendlen, uint32_t flags,
 140                        const struct socket_address *dest_addr);
 141 NTSTATUS socket_pending(struct socket_context *sock, size_t *npending);
 142 NTSTATUS socket_set_option(struct socket_context *sock, const char *option, const char *val);
 143 char *socket_get_peer_name(struct socket_context *sock, TALLOC_CTX *mem_ctx);
 144 struct socket_address *socket_get_peer_addr(struct socket_context *sock, TALLOC_CTX *mem_ctx);
 145 struct socket_address *socket_get_my_addr(struct socket_context *sock, TALLOC_CTX *mem_ctx);
 146 int socket_get_fd(struct socket_context *sock);
 147 NTSTATUS socket_dup(struct socket_context *sock);
 148 struct socket_address *socket_address_from_strings(TALLOC_CTX *mem_ctx,
 149                                                    const char *type,
 150                                                    const char *host,
 151                                                    int port);
 152 struct socket_address *socket_address_from_sockaddr(TALLOC_CTX *mem_ctx,
 153                                                     struct sockaddr *sockaddr,
 154                                                     size_t addrlen);
 155 const struct socket_ops *socket_getops_byname(const char *name, enum socket_type type);
 156 BOOL allow_access(TALLOC_CTX *mem_ctx,
 157                   const char **deny_list, const char **allow_list,
 158                   const char *cname, const char *caddr);
 159 BOOL socket_check_access(struct socket_context *sock,
 160                          const char *service_name,
 161                          const char **allow_list, const char **deny_list);
 162
 163 struct composite_context *socket_connect_send(struct socket_context *sock,
 164                                               struct socket_address *my_address,
 165                                               struct socket_address *server_address,
 166                                               uint32_t flags,
 167                                               struct event_context *event_ctx);
 168 NTSTATUS socket_connect_recv(struct composite_context *ctx);
 169 NTSTATUS socket_connect_ev(struct socket_context *sock,
 170                            struct socket_address *my_address,
 171                            struct socket_address *server_address,
 172                            uint32_t flags, struct event_context *ev);
 173
 174 struct composite_context *socket_connect_multi_send(TALLOC_CTX *mem_ctx,
 175                                                     const char *server_address,
 176                                                     int num_server_ports,
 177                                                     uint16_t *server_ports,
 178                                                     struct event_context *event_ctx);
 179 NTSTATUS socket_connect_multi_recv(struct composite_context *ctx,
 180                                    TALLOC_CTX *mem_ctx,
 181                                    struct socket_context **result,
 182                                    uint16_t *port);
 183 NTSTATUS socket_connect_multi(TALLOC_CTX *mem_ctx, const char *server_address,
 184                               int num_server_ports, uint16_t *server_ports,
 185                               struct event_context *event_ctx,
 186                               struct socket_context **result,
 187                               uint16_t *port);
 188
 189 #endif /* _SAMBA_SOCKET_H */