lib/stream-unix.c

   1 /*
   2  * Copyright (c) 2008, 2009 Nicira Networks.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at:
   7  *
   8  *     http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 #include <config.h>
  18 #include "stream.h"
  19 #include <assert.h>
  20 #include <errno.h>
  21 #include <inttypes.h>
  22 #include <netdb.h>
  23 #include <poll.h>
  24 #include <sys/types.h>
  25 #include <sys/un.h>
  26 #include <stdlib.h>
  27 #include <string.h>
  28 #include <unistd.h>
  29 #include "packets.h"
  30 #include "poll-loop.h"
  31 #include "socket-util.h"
  32 #include "util.h"
  33 #include "stream-provider.h"
  34 #include "stream-fd.h"
  35
  36 #include "vlog.h"
  37 #define THIS_MODULE VLM_stream_unix
  38
  39 /* Active UNIX socket. */
  40
  41 /* Number of unix sockets created so far, to ensure binding path uniqueness. */
  42 static int n_unix_sockets;
  43
  44 static int
  45 unix_open(const char *name, char *suffix, struct stream **streamp)
  46 {
  47     const char *connect_path = suffix;
  48     char *bind_path;
  49     int fd;
  50
  51     bind_path = xasprintf("/tmp/stream-unix.%ld.%d",
  52                           (long int) getpid(), n_unix_sockets++);
  53     fd = make_unix_socket(SOCK_STREAM, true, false, bind_path, connect_path);
  54     if (fd < 0) {
  55         VLOG_ERR("%s: connection to %s failed: %s",
  56                  bind_path, connect_path, strerror(-fd));
  57         free(bind_path);
  58         return -fd;
  59     }
  60
  61     return new_fd_stream(name, fd, check_connection_completion(fd),
  62                          bind_path, streamp);
  63 }
  64
  65 struct stream_class unix_stream_class = {
  66     "unix",                     /* name */
  67     unix_open,                  /* open */
  68     NULL,                       /* close */
  69     NULL,                       /* connect */
  70     NULL,                       /* recv */
  71     NULL,                       /* send */
  72     NULL,                       /* run */
  73     NULL,                       /* run_wait */
  74     NULL,                       /* wait */
  75 };
  76 \f
  77 /* Passive UNIX socket. */
  78
  79 static int punix_accept(int fd, const struct sockaddr *sa, size_t sa_len,
  80                         struct stream **streamp);
  81
  82 static int
  83 punix_open(const char *name UNUSED, char *suffix, struct pstream **pstreamp)
  84 {
  85     int fd, error;
  86
  87     fd = make_unix_socket(SOCK_STREAM, true, true, suffix, NULL);
  88     if (fd < 0) {
  89         VLOG_ERR("%s: binding failed: %s", suffix, strerror(errno));
  90         return errno;
  91     }
  92
  93     if (listen(fd, 10) < 0) {
  94         error = errno;
  95         VLOG_ERR("%s: listen: %s", name, strerror(error));
  96         close(fd);
  97         return error;
  98     }
  99
 100     return new_fd_pstream("punix", fd, punix_accept,
 101                           xstrdup(suffix), pstreamp);
 102 }
 103
 104 static int
 105 punix_accept(int fd, const struct sockaddr *sa, size_t sa_len,
 106              struct stream **streamp)
 107 {
 108     const struct sockaddr_un *sun = (const struct sockaddr_un *) sa;
 109     int name_len = get_unix_name_len(sa_len);
 110     char name[128];
 111
 112     if (name_len > 0) {
 113         snprintf(name, sizeof name, "unix:%.*s", name_len, sun->sun_path);
 114     } else {
 115         strcpy(name, "unix");
 116     }
 117     return new_fd_stream(name, fd, 0, NULL, streamp);
 118 }
 119
 120 struct pstream_class punix_pstream_class = {
 121     "punix",
 122     punix_open,
 123     NULL,
 124     NULL,
 125     NULL
 126 };
 127