src/lib/cmdline_socket.c

   1 /*-
   2  * Copyright (c) <2010>, Intel Corporation
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  *
   9  * - Redistributions of source code must retain the above copyright
  10  *   notice, this list of conditions and the following disclaimer.
  11  *
  12  * - Redistributions in binary form must reproduce the above copyright
  13  *   notice, this list of conditions and the following disclaimer in
  14  *   the documentation and/or other materials provided with the
  15  *   distribution.
  16  *
  17  * - Neither the name of Intel Corporation nor the names of its
  18  *   contributors may be used to endorse or promote products derived
  19  *   from this software without specific prior written permission.
  20  *
  21  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  22  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  23  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
  24  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
  25  * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
  26  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  27  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
  28  * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  29  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
  30  * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  31  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
  32  * OF THE POSSIBILITY OF SUCH DAMAGE.
  33  */
  34
  35 /*
  36  * Copyright (c) 2009, Olivier MATZ <zer0@droids-corp.org>
  37  * All rights reserved.
  38  * Redistribution and use in source and binary forms, with or without
  39  * modification, are permitted provided that the following conditions are met:
  40  *
  41  *     * Redistributions of source code must retain the above copyright
  42  *       notice, this list of conditions and the following disclaimer.
  43  *     * Redistributions in binary form must reproduce the above copyright
  44  *       notice, this list of conditions and the following disclaimer in the
  45  *       documentation and/or other materials provided with the distribution.
  46  *     * Neither the name of the University of California, Berkeley nor the
  47  *       names of its contributors may be used to endorse or promote products
  48  *       derived from this software without specific prior written permission.
  49  *
  50  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND ANY
  51  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  52  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  53  * DISCLAIMED. IN NO EVENT SHALL THE REGENTS AND CONTRIBUTORS BE LIABLE FOR ANY
  54  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  55  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  56  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  57  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  58  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  59  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  60  */
  61
  62 #include <stdio.h>
  63 #include <string.h>
  64 #include <unistd.h>
  65 #include <stdlib.h>
  66 #include <stdarg.h>
  67 #include <inttypes.h>
  68 #include <fcntl.h>
  69
  70 #include <sys/socket.h>
  71 #include <netinet/in.h>
  72 #include <sys/socket.h>
  73 #include <sys/un.h>
  74
  75 #include "cmdline_parse.h"
  76 #include "cmdline_rdline.h"
  77 #include "cmdline.h"
  78
  79 /**********************/
  80
  81 int
  82 cmdline_tcpv4_listen(in_addr_t addr, uint16_t port)
  83 {
  84         int s;
  85         struct sockaddr_in sin_ci;
  86         int optval = 1;
  87
  88         s = socket(PF_INET, SOCK_STREAM, 0);
  89         if (s < 0) {
  90                 dprintf("socket() failed\n");
  91                 return s;
  92         }
  93
  94         if (setsockopt(s, SOL_SOCKET, SO_REUSEADDR,
  95                        &optval, sizeof(optval)) == -1) {
  96                 dprintf("setsockopt() failed\n");
  97                 goto end;
  98         }
  99
 100         memset(&sin_ci, 0, sizeof(sin_ci));
 101         sin_ci.sin_family = AF_INET;
 102         sin_ci.sin_addr.s_addr = addr;
 103         sin_ci.sin_port = htons(port);
 104 #ifndef __linux__
 105         sin_ci.sin_len = sizeof(sin_ci);
 106 #endif
 107         if (bind(s, (struct sockaddr *)&sin_ci, sizeof(sin_ci)) < 0) {
 108                 dprintf("bind() failed\n");
 109                 goto end;
 110         }
 111
 112         if (listen(s, 1) < 0) {
 113                 dprintf("listen() failed\n");
 114                 goto end;
 115         }
 116
 117         return s;
 118 end:
 119         close(s);
 120         return -1;
 121 }
 122
 123 int
 124 cmdline_tcpv6_listen(struct in6_addr addr6, uint16_t port)
 125 {
 126         int s;
 127         struct sockaddr_in6 sin6_ci;
 128
 129         s = socket(PF_INET6, SOCK_STREAM, 0);
 130         if (s < 0) {
 131                 dprintf("socket() failed\n");
 132                 return s;
 133         }
 134
 135         bzero(&sin6_ci, sizeof(sin6_ci));
 136         sin6_ci.sin6_family = AF_INET6;
 137 #ifndef __linux__
 138         sin6_ci.sin6_len = sizeof(sin6_ci);
 139 #endif
 140         memcpy(&sin6_ci.sin6_addr, &addr6, sizeof(sin6_ci.sin6_addr));
 141         sin6_ci.sin6_port = htons(port);
 142         if (bind(s, (struct sockaddr *) &sin6_ci, sizeof(sin6_ci)) < 0) {
 143                 dprintf("bind() failed\n");
 144                 goto end;
 145         }
 146
 147         if (listen(s, 1) < 0) {
 148                 dprintf("listen() failed\n");
 149                 goto end;
 150         }
 151
 152         return s;
 153 end:
 154         close(s);
 155         return -1;
 156 }
 157
 158 int
 159 cmdline_unix_listen(char *filename)
 160 {
 161         int s;
 162         struct sockaddr_un servAddr;
 163
 164         s = socket(AF_UNIX, SOCK_STREAM, 0);
 165         if (s < 0) {
 166                 dprintf("socket() failed\n");
 167                 return s;
 168         }
 169
 170         bzero(&servAddr, sizeof(servAddr));
 171         servAddr.sun_family = AF_UNIX;
 172         memcpy(servAddr.sun_path, filename , strlen(filename));
 173
 174         unlink(filename);
 175         if(bind(s, (struct sockaddr *) &servAddr, sizeof(servAddr)) < 0) {
 176                 dprintf("bind() failed\n");
 177                 goto end;
 178         }
 179
 180         if (listen(s, 1) < 0) {
 181                 dprintf("listen() failed\n");
 182                 goto end;
 183         }
 184
 185         return s;
 186 end:
 187         close(s);
 188         return -1;
 189 }
 190
 191 struct cmdline *
 192 cmdline_stdin_new(cmdline_parse_ctx_t *ctx, const char *prompt)
 193 {
 194         return (cmdline_new(ctx, prompt, 0, 1));
 195 }
 196
 197 struct cmdline *
 198 cmdline_file_new(cmdline_parse_ctx_t *ctx, const char *prompt, const char *path)
 199 {
 200         int fd;
 201         fd = open(path, O_RDONLY, 0);
 202         if (fd < 0) {
 203                 dprintf("open() failed\n");
 204                 return NULL;
 205         }
 206         return (cmdline_new(ctx, prompt, fd, -1));
 207 }
 208
 209 struct cmdline *
 210 cmdline_accept(cmdline_parse_ctx_t *ctx, const char *prompt, int s)
 211 {
 212         int s2;
 213         struct sockaddr sin;
 214         socklen_t sinlen;
 215
 216         sinlen = sizeof(struct sockaddr);
 217
 218         if ((s2 = accept(s, &sin, &sinlen)) < 0) {
 219                 dprintf("accept() failed\n");
 220                 return NULL;
 221         }
 222
 223         return (cmdline_new(ctx, prompt, s2, s2));
 224 }
 225