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 #include <termios.h>
  70
  71 #ifndef CMDLINE_NO_SOCKET
  72 #include <sys/socket.h>
  73 #include <netinet/in.h>
  74 #include <sys/socket.h>
  75 #include <sys/un.h>
  76 #endif
  77
  78 #include "cmdline_parse.h"
  79 #include "cmdline_rdline.h"
  80 #include "cmdline_socket.h"
  81 #include "cmdline.h"
  82
  83 /**********************/
  84
  85 #ifndef CMDLINE_NO_SOCKET
  86 int
  87 cmdline_tcpv4_listen(in_addr_t addr, uint16_t port)
  88 {
  89         int s;
  90         struct sockaddr_in sin_ci;
  91         int optval = 1;
  92
  93         s = socket(PF_INET, SOCK_STREAM, 0);
  94         if (s < 0) {
  95                 dprintf("socket() failed\n");
  96                 return s;
  97         }
  98
  99         if (setsockopt(s, SOL_SOCKET, SO_REUSEADDR,
 100                        &optval, sizeof(optval)) == -1) {
 101                 dprintf("setsockopt() failed\n");
 102                 goto end;
 103         }
 104
 105         memset(&sin_ci, 0, sizeof(sin_ci));
 106         sin_ci.sin_family = AF_INET;
 107         sin_ci.sin_addr.s_addr = addr;
 108         sin_ci.sin_port = htons(port);
 109 #ifndef __linux__
 110         sin_ci.sin_len = sizeof(sin_ci);
 111 #endif
 112         if (bind(s, (struct sockaddr *)&sin_ci, sizeof(sin_ci)) < 0) {
 113                 dprintf("bind() failed\n");
 114                 goto end;
 115         }
 116
 117         if (listen(s, 1) < 0) {
 118                 dprintf("listen() failed\n");
 119                 goto end;
 120         }
 121
 122         return s;
 123 end:
 124         close(s);
 125         return -1;
 126 }
 127
 128 int
 129 cmdline_tcpv6_listen(struct in6_addr addr6, uint16_t port)
 130 {
 131         int s;
 132         struct sockaddr_in6 sin6_ci;
 133
 134         s = socket(PF_INET6, SOCK_STREAM, 0);
 135         if (s < 0) {
 136                 dprintf("socket() failed\n");
 137                 return s;
 138         }
 139
 140         bzero(&sin6_ci, sizeof(sin6_ci));
 141         sin6_ci.sin6_family = AF_INET6;
 142 #ifndef __linux__
 143         sin6_ci.sin6_len = sizeof(sin6_ci);
 144 #endif
 145         memcpy(&sin6_ci.sin6_addr, &addr6, sizeof(sin6_ci.sin6_addr));
 146         sin6_ci.sin6_port = htons(port);
 147         if (bind(s, (struct sockaddr *) &sin6_ci, sizeof(sin6_ci)) < 0) {
 148                 dprintf("bind() failed\n");
 149                 goto end;
 150         }
 151
 152         if (listen(s, 1) < 0) {
 153                 dprintf("listen() failed\n");
 154                 goto end;
 155         }
 156
 157         return s;
 158 end:
 159         close(s);
 160         return -1;
 161 }
 162
 163 int
 164 cmdline_unix_listen(char *filename)
 165 {
 166         int s;
 167         struct sockaddr_un servAddr;
 168
 169         s = socket(AF_UNIX, SOCK_STREAM, 0);
 170         if (s < 0) {
 171                 dprintf("socket() failed\n");
 172                 return s;
 173         }
 174
 175         bzero(&servAddr, sizeof(servAddr));
 176         servAddr.sun_family = AF_UNIX;
 177         memcpy(servAddr.sun_path, filename , strlen(filename));
 178
 179         unlink(filename);
 180         if(bind(s, (struct sockaddr *) &servAddr, sizeof(servAddr)) < 0) {
 181                 dprintf("bind() failed\n");
 182                 goto end;
 183         }
 184
 185         if (listen(s, 1) < 0) {
 186                 dprintf("listen() failed\n");
 187                 goto end;
 188         }
 189
 190         return s;
 191 end:
 192         close(s);
 193         return -1;
 194 }
 195
 196 struct cmdline *
 197 cmdline_accept(cmdline_parse_ctx_t *ctx, const char *prompt, int s)
 198 {
 199         int s2;
 200         struct sockaddr sin;
 201         socklen_t sinlen;
 202
 203         sinlen = sizeof(struct sockaddr);
 204
 205         if ((s2 = accept(s, &sin, &sinlen)) < 0) {
 206                 dprintf("accept() failed\n");
 207                 return NULL;
 208         }
 209
 210         return (cmdline_new(ctx, prompt, s2, s2));
 211 }
 212 #endif
 213
 214 struct cmdline *
 215 cmdline_file_new(cmdline_parse_ctx_t *ctx, const char *prompt, const char *path)
 216 {
 217         int fd;
 218         fd = open(path, O_RDONLY, 0);
 219         if (fd < 0) {
 220                 dprintf("open() failed\n");
 221                 return NULL;
 222         }
 223         return (cmdline_new(ctx, prompt, fd, -1));
 224 }
 225
 226 struct cmdline *
 227 cmdline_stdin_new(cmdline_parse_ctx_t *ctx, const char *prompt)
 228 {
 229         struct cmdline *cl;
 230 #ifdef CMDLINE_TERMIOS
 231         struct termios oldterm, term;
 232
 233         tcgetattr(0, &oldterm);
 234         memcpy(&term, &oldterm, sizeof(term));
 235         term.c_lflag &= ~(ICANON | ECHO | ISIG);
 236         tcsetattr(0, TCSANOW, &term);
 237         setbuf(stdin, NULL);
 238 #endif
 239
 240         cl = cmdline_new(ctx, prompt, 0, 1);
 241
 242 #ifdef CMDLINE_TERMIOS
 243         memcpy(&cl->oldterm, &oldterm, sizeof(term));
 244 #endif
 245         return cl;
 246 }
 247
 248 void
 249 cmdline_stdin_exit(struct cmdline *cl)
 250 {
 251 #ifdef CMDLINE_TERMIOS
 252         tcsetattr(fileno(stdin), TCSANOW, &cl->oldterm);
 253 #else
 254         /* silent the compiler */
 255         cl = cl;
 256 #endif
 257 }