lib/vconn-tcp.c

   1 /*
   2  * Copyright (c) 2008, 2009 Nicira Networks.
   3  *
   4  * Permission to use, copy, modify, and/or distribute this software for any
   5  * purpose with or without fee is hereby granted, provided that the above
   6  * copyright notice and this permission notice appear in all copies.
   7  *
   8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
   9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
  10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
  11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
  12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
  13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
  14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
  15  */
  16
  17 #include <config.h>
  18 #include "vconn.h"
  19 #include <errno.h>
  20 #include <inttypes.h>
  21 #include <sys/types.h>
  22 #include <netinet/in.h>
  23 #include <netinet/tcp.h>
  24 #include <stdlib.h>
  25 #include <string.h>
  26 #include <unistd.h>
  27 #include "packets.h"
  28 #include "socket-util.h"
  29 #include "util.h"
  30 #include "openflow/openflow.h"
  31 #include "vconn-provider.h"
  32 #include "vconn-stream.h"
  33
  34 #include "vlog.h"
  35 #define THIS_MODULE VLM_vconn_tcp
  36
  37 /* Active TCP. */
  38
  39 static int
  40 new_tcp_vconn(const char *name, int fd, int connect_status,
  41               const struct sockaddr_in *sin, struct vconn **vconnp)
  42 {
  43     int on = 1;
  44     int retval;
  45
  46     retval = setsockopt(fd, IPPROTO_TCP, TCP_NODELAY, &on, sizeof on);
  47     if (retval) {
  48         VLOG_ERR("%s: setsockopt(TCP_NODELAY): %s", name, strerror(errno));
  49         close(fd);
  50         return errno;
  51     }
  52
  53     return new_stream_vconn(name, fd, connect_status, sin->sin_addr.s_addr,
  54                             true, vconnp);
  55 }
  56
  57 static int
  58 tcp_open(const char *name, char *suffix, struct vconn **vconnp)
  59 {
  60     char *save_ptr;
  61     const char *host_name;
  62     const char *port_string;
  63     struct sockaddr_in sin;
  64     int retval;
  65     int fd;
  66
  67     /* Glibc 2.7 has a bug in strtok_r when compiling with optimization that
  68      * can cause segfaults here:
  69      * http://sources.redhat.com/bugzilla/show_bug.cgi?id=5614.
  70      * Using "::" instead of the obvious ":" works around it. */
  71     host_name = strtok_r(suffix, "::", &save_ptr);
  72     port_string = strtok_r(NULL, "::", &save_ptr);
  73     if (!host_name) {
  74         ovs_error(0, "%s: bad peer name format", name);
  75         return EAFNOSUPPORT;
  76     }
  77
  78     memset(&sin, 0, sizeof sin);
  79     sin.sin_family = AF_INET;
  80     if (lookup_ip(host_name, &sin.sin_addr)) {
  81         return ENOENT;
  82     }
  83     sin.sin_port = htons(port_string ? atoi(port_string) : OFP_TCP_PORT);
  84
  85     fd = socket(AF_INET, SOCK_STREAM, 0);
  86     if (fd < 0) {
  87         VLOG_ERR("%s: socket: %s", name, strerror(errno));
  88         return errno;
  89     }
  90
  91     retval = set_nonblocking(fd);
  92     if (retval) {
  93         close(fd);
  94         return retval;
  95     }
  96
  97     retval = connect(fd, (struct sockaddr *) &sin, sizeof sin);
  98     if (retval < 0) {
  99         if (errno == EINPROGRESS) {
 100             return new_tcp_vconn(name, fd, EAGAIN, &sin, vconnp);
 101         } else {
 102             int error = errno;
 103             VLOG_ERR("%s: connect: %s", name, strerror(error));
 104             close(fd);
 105             return error;
 106         }
 107     } else {
 108         return new_tcp_vconn(name, fd, 0, &sin, vconnp);
 109     }
 110 }
 111
 112 struct vconn_class tcp_vconn_class = {
 113     "tcp",                      /* name */
 114     tcp_open,                   /* open */
 115     NULL,                       /* close */
 116     NULL,                       /* connect */
 117     NULL,                       /* recv */
 118     NULL,                       /* send */
 119     NULL,                       /* wait */
 120 };
 121 \f
 122 /* Passive TCP. */
 123
 124 static int ptcp_accept(int fd, const struct sockaddr *sa, size_t sa_len,
 125                        struct vconn **vconnp);
 126
 127 static int
 128 ptcp_open(const char *name, char *suffix, struct pvconn **pvconnp)
 129 {
 130     struct sockaddr_in sin;
 131     int retval;
 132     int fd;
 133     unsigned int yes  = 1;
 134
 135     fd = socket(AF_INET, SOCK_STREAM, 0);
 136     if (fd < 0) {
 137         VLOG_ERR("%s: socket: %s", name, strerror(errno));
 138         return errno;
 139     }
 140
 141     if (setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof yes) < 0) {
 142         VLOG_ERR("%s: setsockopt(SO_REUSEADDR): %s", name, strerror(errno));
 143         return errno;
 144     }
 145
 146     memset(&sin, 0, sizeof sin);
 147     sin.sin_family = AF_INET;
 148     sin.sin_addr.s_addr = htonl(INADDR_ANY);
 149     sin.sin_port = htons(atoi(suffix) ? atoi(suffix) : OFP_TCP_PORT);
 150     retval = bind(fd, (struct sockaddr *) &sin, sizeof sin);
 151     if (retval < 0) {
 152         int error = errno;
 153         VLOG_ERR("%s: bind: %s", name, strerror(error));
 154         close(fd);
 155         return error;
 156     }
 157
 158     return new_pstream_pvconn("ptcp", fd, ptcp_accept, pvconnp);
 159 }
 160
 161 static int
 162 ptcp_accept(int fd, const struct sockaddr *sa, size_t sa_len,
 163             struct vconn **vconnp)
 164 {
 165     const struct sockaddr_in *sin = (const struct sockaddr_in *) sa;
 166     char name[128];
 167
 168     if (sa_len == sizeof(struct sockaddr_in) && sin->sin_family == AF_INET) {
 169         sprintf(name, "tcp:"IP_FMT, IP_ARGS(&sin->sin_addr));
 170         if (sin->sin_port != htons(OFP_TCP_PORT)) {
 171             sprintf(strchr(name, '\0'), ":%"PRIu16, ntohs(sin->sin_port));
 172         }
 173     } else {
 174         strcpy(name, "tcp");
 175     }
 176     return new_tcp_vconn(name, fd, 0, sin, vconnp);
 177 }
 178
 179 struct pvconn_class ptcp_pvconn_class = {
 180     "ptcp",
 181     ptcp_open,
 182     NULL,
 183     NULL,
 184     NULL
 185 };
 186