source4/lib/socket/socket_ipv6.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Socket IPv6 functions
   4    Copyright (C) Stefan Metzmacher 2004
   5    Copyright (C) Jelmer Vernooij 2004
   6
   7    This program is free software; you can redistribute it and/or modify
   8    it under the terms of the GNU General Public License as published by
   9    the Free Software Foundation; either version 2 of the License, or
  10    (at your option) any later version.
  11
  12    This program is distributed in the hope that it will be useful,
  13    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15    GNU General Public License for more details.
  16
  17    You should have received a copy of the GNU General Public License
  18    along with this program; if not, write to the Free Software
  19    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  20 */
  21
  22 #include "includes.h"
  23 #include "lib/socket/socket.h"
  24 #include "system/network.h"
  25
  26 static struct in6_addr interpret_addr6(const char *name)
  27 {
  28         struct hostent *he;
  29
  30         if (name == NULL) return in6addr_any;
  31
  32         he = gethostbyname2(name, PF_INET6);
  33
  34         if (he == NULL) return in6addr_any;
  35
  36         return *((struct in6_addr *)he->h_addr);
  37 }
  38
  39 static NTSTATUS ipv6_tcp_init(struct socket_context *sock)
  40 {
  41         sock->fd = socket(PF_INET6, SOCK_STREAM, 0);
  42         if (sock->fd == -1) {
  43                 return map_nt_error_from_unix(errno);
  44         }
  45
  46         sock->backend_name = "ipv6";
  47
  48         return NT_STATUS_OK;
  49 }
  50
  51 static void ipv6_tcp_close(struct socket_context *sock)
  52 {
  53         close(sock->fd);
  54 }
  55
  56 static NTSTATUS ipv6_tcp_connect_complete(struct socket_context *sock, uint32_t flags)
  57 {
  58         int error=0, ret;
  59         socklen_t len = sizeof(error);
  60
  61         /* check for any errors that may have occurred - this is needed
  62            for non-blocking connect */
  63         ret = getsockopt(sock->fd, SOL_SOCKET, SO_ERROR, &error, &len);
  64         if (ret == -1) {
  65                 return map_nt_error_from_unix(errno);
  66         }
  67         if (error != 0) {
  68                 return map_nt_error_from_unix(error);
  69         }
  70
  71         if (!(flags & SOCKET_FLAG_BLOCK)) {
  72                 ret = set_blocking(sock->fd, False);
  73                 if (ret == -1) {
  74                         return map_nt_error_from_unix(errno);
  75                 }
  76         }
  77
  78         sock->state = SOCKET_STATE_CLIENT_CONNECTED;
  79
  80         return NT_STATUS_OK;
  81 }
  82
  83 static NTSTATUS ipv6_tcp_connect(struct socket_context *sock,
  84                                  const struct socket_address *my_address,
  85                                  const struct socket_address *srv_address,
  86                                  uint32_t flags)
  87 {
  88         int ret;
  89
  90         if (my_address && my_address->sockaddr) {
  91                 ret = bind(sock->fd, my_address->sockaddr, my_address->sockaddrlen);
  92                 if (ret == -1) {
  93                         return map_nt_error_from_unix(errno);
  94                 }
  95         } else if (my_address) {
  96                 struct in6_addr my_ip;
  97                 my_ip = interpret_addr6(my_address->addr);
  98
  99                 if (memcmp(&my_ip, &in6addr_any, sizeof(my_ip)) || my_address->port != 0) {
 100                         struct sockaddr_in6 my_addr;
 101                         ZERO_STRUCT(my_addr);
 102                         my_addr.sin6_addr       = my_ip;
 103                         my_addr.sin6_port       = htons(my_address->port);
 104                         my_addr.sin6_family     = PF_INET6;
 105
 106                         ret = bind(sock->fd, (struct sockaddr *)&my_addr, sizeof(my_addr));
 107                         if (ret == -1) {
 108                                 return map_nt_error_from_unix(errno);
 109                         }
 110                 }
 111         }
 112
 113         if (srv_address->sockaddr) {
 114                 ret = connect(sock->fd, srv_address->sockaddr, srv_address->sockaddrlen);
 115         } else {
 116                 struct in6_addr srv_ip;
 117                 struct sockaddr_in6 srv_addr;
 118                 srv_ip = interpret_addr6(srv_address->addr);
 119                 if (memcmp(&srv_ip, &in6addr_any, sizeof(srv_ip)) == 0) {
 120                         return NT_STATUS_BAD_NETWORK_NAME;
 121                 }
 122
 123                 ZERO_STRUCT(srv_addr);
 124                 srv_addr.sin6_addr      = srv_ip;
 125                 srv_addr.sin6_port      = htons(srv_address->port);
 126                 srv_addr.sin6_family    = PF_INET6;
 127
 128                 ret = connect(sock->fd, (const struct sockaddr *)&srv_addr, sizeof(srv_addr));
 129         }
 130         if (ret == -1) {
 131                 return map_nt_error_from_unix(errno);
 132         }
 133
 134         return ipv6_tcp_connect_complete(sock, flags);
 135 }
 136
 137 static NTSTATUS ipv6_tcp_listen(struct socket_context *sock,
 138                                 const struct socket_address *my_address,
 139                                 int queue_size, uint32_t flags)
 140 {
 141         struct sockaddr_in6 my_addr;
 142         struct in6_addr ip_addr;
 143         int ret;
 144
 145         socket_set_option(sock, "SO_REUSEADDR=1", NULL);
 146
 147         if (my_address->sockaddr) {
 148                 ret = bind(sock->fd, my_address->sockaddr, my_address->sockaddrlen);
 149         } else {
 150                 ip_addr = interpret_addr6(my_address->addr);
 151
 152                 ZERO_STRUCT(my_addr);
 153                 my_addr.sin6_addr       = ip_addr;
 154                 my_addr.sin6_port       = htons(my_address->port);
 155                 my_addr.sin6_family     = PF_INET6;
 156
 157                 ret = bind(sock->fd, (struct sockaddr *)&my_addr, sizeof(my_addr));
 158         }
 159
 160         if (ret == -1) {
 161                 return map_nt_error_from_unix(errno);
 162         }
 163
 164         ret = listen(sock->fd, queue_size);
 165         if (ret == -1) {
 166                 return map_nt_error_from_unix(errno);
 167         }
 168
 169         if (!(flags & SOCKET_FLAG_BLOCK)) {
 170                 ret = set_blocking(sock->fd, False);
 171                 if (ret == -1) {
 172                         return map_nt_error_from_unix(errno);
 173                 }
 174         }
 175
 176         sock->state= SOCKET_STATE_SERVER_LISTEN;
 177
 178         return NT_STATUS_OK;
 179 }
 180
 181 static NTSTATUS ipv6_tcp_accept(struct socket_context *sock, struct socket_context **new_sock)
 182 {
 183         struct sockaddr_in cli_addr;
 184         socklen_t cli_addr_len = sizeof(cli_addr);
 185         int new_fd;
 186
 187         new_fd = accept(sock->fd, (struct sockaddr *)&cli_addr, &cli_addr_len);
 188         if (new_fd == -1) {
 189                 return map_nt_error_from_unix(errno);
 190         }
 191
 192         if (!(sock->flags & SOCKET_FLAG_BLOCK)) {
 193                 int ret = set_blocking(new_fd, False);
 194                 if (ret == -1) {
 195                         close(new_fd);
 196                         return map_nt_error_from_unix(errno);
 197                 }
 198         }
 199
 200         /* TODO: we could add a 'accept_check' hook here
 201          *       which get the black/white lists via socket_set_accept_filter()
 202          *       or something like that
 203          *       --metze
 204          */
 205
 206         (*new_sock) = talloc(NULL, struct socket_context);
 207         if (!(*new_sock)) {
 208                 close(new_fd);
 209                 return NT_STATUS_NO_MEMORY;
 210         }
 211
 212         /* copy the socket_context */
 213         (*new_sock)->type               = sock->type;
 214         (*new_sock)->state              = SOCKET_STATE_SERVER_CONNECTED;
 215         (*new_sock)->flags              = sock->flags;
 216
 217         (*new_sock)->fd                 = new_fd;
 218
 219         (*new_sock)->private_data       = NULL;
 220         (*new_sock)->ops                = sock->ops;
 221         (*new_sock)->backend_name       = sock->backend_name;
 222
 223         return NT_STATUS_OK;
 224 }
 225
 226 static NTSTATUS ipv6_tcp_recv(struct socket_context *sock, void *buf,
 227                               size_t wantlen, size_t *nread, uint32_t flags)
 228 {
 229         ssize_t gotlen;
 230         int flgs = 0;
 231
 232         /* TODO: we need to map all flags here */
 233         if (flags & SOCKET_FLAG_PEEK) {
 234                 flgs |= MSG_PEEK;
 235         }
 236
 237         if (flags & SOCKET_FLAG_BLOCK) {
 238                 flgs |= MSG_WAITALL;
 239         }
 240
 241         *nread = 0;
 242
 243         gotlen = recv(sock->fd, buf, wantlen, flgs);
 244         if (gotlen == 0) {
 245                 return NT_STATUS_END_OF_FILE;
 246         } else if (gotlen == -1) {
 247                 return map_nt_error_from_unix(errno);
 248         }
 249
 250         *nread = gotlen;
 251
 252         return NT_STATUS_OK;
 253 }
 254
 255 static NTSTATUS ipv6_tcp_send(struct socket_context *sock,
 256                               const DATA_BLOB *blob, size_t *sendlen, uint32_t flags)
 257 {
 258         ssize_t len;
 259         int flgs = 0;
 260
 261         *sendlen = 0;
 262
 263         len = send(sock->fd, blob->data, blob->length, flgs);
 264         if (len == -1) {
 265                 return map_nt_error_from_unix(errno);
 266         }
 267
 268         *sendlen = len;
 269
 270         return NT_STATUS_OK;
 271 }
 272
 273 static NTSTATUS ipv6_tcp_set_option(struct socket_context *sock, const char *option, const char *val)
 274 {
 275         set_socket_options(sock->fd, option);
 276         return NT_STATUS_OK;
 277 }
 278
 279 static char *ipv6_tcp_get_peer_name(struct socket_context *sock, TALLOC_CTX *mem_ctx)
 280 {
 281         struct sockaddr_in6 peer_addr;
 282         socklen_t len = sizeof(peer_addr);
 283         struct hostent *he;
 284         int ret;
 285
 286         ret = getpeername(sock->fd, (struct sockaddr *)&peer_addr, &len);
 287         if (ret == -1) {
 288                 return NULL;
 289         }
 290
 291         he = gethostbyaddr((char *)&peer_addr.sin6_addr, sizeof(peer_addr.sin6_addr), AF_INET6);
 292         if (he == NULL) {
 293                 return NULL;
 294         }
 295
 296         return talloc_strdup(mem_ctx, he->h_name);
 297 }
 298
 299 static struct socket_address *ipv6_tcp_get_peer_addr(struct socket_context *sock, TALLOC_CTX *mem_ctx)
 300 {
 301         struct sockaddr_in6 *peer_addr;
 302         socklen_t len = sizeof(*peer_addr);
 303         struct socket_address *peer;
 304         int ret;
 305         struct hostent *he;
 306
 307         peer = talloc(mem_ctx, struct socket_address);
 308         if (!peer) {
 309                 return NULL;
 310         }
 311
 312         peer->family = sock->backend_name;
 313         peer_addr = talloc(peer, struct sockaddr_in6);
 314         if (!peer_addr) {
 315                 talloc_free(peer);
 316                 return NULL;
 317         }
 318
 319         peer->sockaddr = (struct sockaddr *)peer_addr;
 320
 321         ret = getpeername(sock->fd, peer->sockaddr, &len);
 322         if (ret == -1) {
 323                 talloc_free(peer);
 324                 return NULL;
 325         }
 326
 327         peer->sockaddrlen = len;
 328
 329         he = gethostbyaddr((char *)&peer_addr->sin6_addr, len, AF_INET6);
 330
 331         if (!he || !he->h_name) {
 332                 talloc_free(peer);
 333                 return NULL;
 334         }
 335
 336         peer->addr = talloc_strdup(mem_ctx, he->h_name);
 337         if (!peer->addr) {
 338                 talloc_free(peer);
 339                 return NULL;
 340         }
 341         peer->port = ntohs(peer_addr->sin6_port);
 342
 343         return peer;
 344 }
 345
 346 static struct socket_address *ipv6_tcp_get_my_addr(struct socket_context *sock, TALLOC_CTX *mem_ctx)
 347 {
 348         struct sockaddr_in6 *local_addr;
 349         socklen_t len = sizeof(*local_addr);
 350         struct socket_address *local;
 351         int ret;
 352         struct hostent *he;
 353
 354         local = talloc(mem_ctx, struct socket_address);
 355         if (!local) {
 356                 return NULL;
 357         }
 358
 359         local->family = sock->backend_name;
 360         local_addr = talloc(local, struct sockaddr_in6);
 361         if (!local_addr) {
 362                 talloc_free(local);
 363                 return NULL;
 364         }
 365
 366         local->sockaddr = (struct sockaddr *)local_addr;
 367
 368         ret = getsockname(sock->fd, local->sockaddr, &len);
 369         if (ret == -1) {
 370                 talloc_free(local);
 371                 return NULL;
 372         }
 373
 374         local->sockaddrlen = len;
 375
 376         he = gethostbyaddr((char *)&local_addr->sin6_addr, len, AF_INET6);
 377
 378         if (!he || !he->h_name) {
 379                 talloc_free(local);
 380                 return NULL;
 381         }
 382
 383         local->addr = talloc_strdup(mem_ctx, he->h_name);
 384         if (!local->addr) {
 385                 talloc_free(local);
 386                 return NULL;
 387         }
 388         local->port = ntohs(local_addr->sin6_port);
 389
 390         return local;
 391 }
 392
 393 static int ipv6_tcp_get_fd(struct socket_context *sock)
 394 {
 395         return sock->fd;
 396 }
 397
 398 static const struct socket_ops ipv6_tcp_ops = {
 399         .name                   = "ipv6",
 400         .fn_init                = ipv6_tcp_init,
 401         .fn_connect             = ipv6_tcp_connect,
 402         .fn_connect_complete    = ipv6_tcp_connect_complete,
 403         .fn_listen              = ipv6_tcp_listen,
 404         .fn_accept              = ipv6_tcp_accept,
 405         .fn_recv                = ipv6_tcp_recv,
 406         .fn_send                = ipv6_tcp_send,
 407         .fn_close               = ipv6_tcp_close,
 408
 409         .fn_set_option          = ipv6_tcp_set_option,
 410
 411         .fn_get_peer_name       = ipv6_tcp_get_peer_name,
 412         .fn_get_peer_addr       = ipv6_tcp_get_peer_addr,
 413         .fn_get_my_addr         = ipv6_tcp_get_my_addr,
 414
 415         .fn_get_fd              = ipv6_tcp_get_fd
 416 };
 417
 418 const struct socket_ops *socket_ipv6_ops(enum socket_type type)
 419 {
 420         if (type != SOCKET_TYPE_STREAM) {
 421                 return NULL;
 422         }
 423         return &ipv6_tcp_ops;
 424 }