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     if (listen(fd, 10) < 0) {
  92         error = errno;
  93         VLOG_ERR("%s: listen: %s", name, strerror(error));
  94         close(fd);
  95         return error;
  96     }
  97
  98     return new_fd_pstream("punix", fd, punix_accept,
  99                           xstrdup(suffix), pstreamp);
 100 }
 101
 102 static int
 103 punix_accept(int fd, const struct sockaddr *sa, size_t sa_len,
 104              struct stream **streamp)
 105 {
 106     const struct sockaddr_un *sun = (const struct sockaddr_un *) sa;
 107     int name_len = get_unix_name_len(sa_len);
 108     char name[128];
 109
 110     if (name_len > 0) {
 111         snprintf(name, sizeof name, "unix:%.*s", name_len, sun->sun_path);
 112     } else {
 113         strcpy(name, "unix");
 114     }
 115     return new_fd_stream(name, fd, 0, NULL, streamp);
 116 }
 117
 118 struct pstream_class punix_pstream_class = {
 119     "punix",
 120     punix_open,
 121     NULL,
 122     NULL,
 123     NULL
 124 };
 125