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,                       /* wait */
  73 };
  74 \f
  75 /* Passive UNIX socket. */
  76
  77 static int punix_accept(int fd, const struct sockaddr *sa, size_t sa_len,
  78                         struct stream **streamp);
  79
  80 static int
  81 punix_open(const char *name UNUSED, char *suffix, struct pstream **pstreamp)
  82 {
  83     int fd, error;
  84
  85     fd = make_unix_socket(SOCK_STREAM, true, true, suffix, NULL);
  86     if (fd < 0) {
  87         VLOG_ERR("%s: binding failed: %s", suffix, strerror(errno));
  88         return errno;
  89     }
  90
  91     error = set_nonblocking(fd);
  92     if (error) {
  93         close(fd);
  94         return error;
  95     }
  96
  97     if (listen(fd, 10) < 0) {
  98         error = errno;
  99         VLOG_ERR("%s: listen: %s", name, strerror(error));
 100         close(fd);
 101         return error;
 102     }
 103
 104     return new_fd_pstream("punix", fd, punix_accept,
 105                           xstrdup(suffix), pstreamp);
 106 }
 107
 108 static int
 109 punix_accept(int fd, const struct sockaddr *sa, size_t sa_len,
 110              struct stream **streamp)
 111 {
 112     const struct sockaddr_un *sun = (const struct sockaddr_un *) sa;
 113     int name_len = get_unix_name_len(sa_len);
 114     char name[128];
 115
 116     if (name_len > 0) {
 117         snprintf(name, sizeof name, "unix:%.*s", name_len, sun->sun_path);
 118     } else {
 119         strcpy(name, "unix");
 120     }
 121     return new_fd_stream(name, fd, 0, NULL, streamp);
 122 }
 123
 124 struct pstream_class punix_pstream_class = {
 125     "punix",
 126     punix_open,
 127     NULL,
 128     NULL,
 129     NULL
 130 };
 131