2.3.2/src/slirpvde/misc.c

   1 /*
   2  * Copyright (c) 1995 Danny Gasparovski.
   3  *
   4  * Please read the file COPYRIGHT for the
   5  * terms and conditions of the copyright.
   6  */
   7
   8 #include <slirp.h>
   9 #include <libslirp.h>
  10
  11 #if 0
  12 #include "monitor.h"
  13 #endif
  14
  15 #ifdef DEBUG
  16 int slirp_debug = DBG_CALL|DBG_MISC|DBG_ERROR;
  17 #endif
  18
  19 struct quehead {
  20         struct quehead *qh_link;
  21         struct quehead *qh_rlink;
  22 };
  23
  24 inline void
  25 insque(void *a, void *b)
  26 {
  27         register struct quehead *element = (struct quehead *) a;
  28         register struct quehead *head = (struct quehead *) b;
  29         element->qh_link = head->qh_link;
  30         head->qh_link = (struct quehead *)element;
  31         element->qh_rlink = (struct quehead *)head;
  32         ((struct quehead *)(element->qh_link))->qh_rlink
  33         = (struct quehead *)element;
  34 }
  35
  36 inline void
  37 remque(void *a)
  38 {
  39   register struct quehead *element = (struct quehead *) a;
  40   ((struct quehead *)(element->qh_link))->qh_rlink = element->qh_rlink;
  41   ((struct quehead *)(element->qh_rlink))->qh_link = element->qh_link;
  42   element->qh_rlink = NULL;
  43 }
  44
  45 int add_exec(struct ex_list **ex_ptr, int do_pty, char *exec,
  46              struct in_addr addr, int port)
  47 {
  48         struct ex_list *tmp_ptr;
  49
  50         /* First, check if the port is "bound" */
  51         for (tmp_ptr = *ex_ptr; tmp_ptr; tmp_ptr = tmp_ptr->ex_next) {
  52                 if (port == tmp_ptr->ex_fport &&
  53                     addr.s_addr == tmp_ptr->ex_addr.s_addr)
  54                         return -1;
  55         }
  56
  57         tmp_ptr = *ex_ptr;
  58         *ex_ptr = (struct ex_list *)malloc(sizeof(struct ex_list));
  59         (*ex_ptr)->ex_fport = port;
  60         (*ex_ptr)->ex_addr = addr;
  61         (*ex_ptr)->ex_pty = do_pty;
  62         (*ex_ptr)->ex_exec = (do_pty == 3) ? exec : strdup(exec);
  63         (*ex_ptr)->ex_next = tmp_ptr;
  64         return 0;
  65 }
  66
  67 #ifndef HAVE_STRERROR
  68
  69 /*
  70  * For systems with no strerror
  71  */
  72
  73 extern int sys_nerr;
  74 extern char *sys_errlist[];
  75
  76 char *
  77 strerror(error)
  78         int error;
  79 {
  80         if (error < sys_nerr)
  81            return sys_errlist[error];
  82         else
  83            return "Unknown error.";
  84 }
  85
  86 #endif
  87
  88
  89 #if 0
  90 #ifdef _WIN32
  91
  92 int
  93 fork_exec(struct socket *so, const char *ex, int do_pty)
  94 {
  95     /* not implemented */
  96     return 0;
  97 }
  98
  99 #else
 100
 101 /*
 102  * XXX This is ugly
 103  * We create and bind a socket, then fork off to another
 104  * process, which connects to this socket, after which we
 105  * exec the wanted program.  If something (strange) happens,
 106  * the accept() call could block us forever.
 107  *
 108  * do_pty = 0   Fork/exec inetd style
 109  * do_pty = 1   Fork/exec using slirp.telnetd
 110  * do_ptr = 2   Fork/exec using pty
 111  */
 112 int
 113 fork_exec(struct socket *so, const char *ex, int do_pty)
 114 {
 115         int s;
 116         struct sockaddr_in addr;
 117         socklen_t addrlen = sizeof(addr);
 118         int opt;
 119         int master = -1;
 120         const char *argv[256];
 121         /* don't want to clobber the original */
 122         char *bptr;
 123         const char *curarg;
 124         int c, i, ret;
 125
 126         DEBUG_CALL("fork_exec");
 127         DEBUG_ARG("so = %lx", (long)so);
 128         DEBUG_ARG("ex = %lx", (long)ex);
 129         DEBUG_ARG("do_pty = %lx", (long)do_pty);
 130
 131         if (do_pty == 2) {
 132                 return 0;
 133         } else {
 134                 addr.sin_family = AF_INET;
 135                 addr.sin_port = 0;
 136                 addr.sin_addr.s_addr = INADDR_ANY;
 137
 138                 if ((s = qemu_socket(AF_INET, SOCK_STREAM, 0)) < 0 ||
 139                     bind(s, (struct sockaddr *)&addr, addrlen) < 0 ||
 140                     listen(s, 1) < 0) {
 141                         lprint("Error: inet socket: %s\n", strerror(errno));
 142                         closesocket(s);
 143
 144                         return 0;
 145                 }
 146         }
 147
 148         switch(fork()) {
 149          case -1:
 150                 lprint("Error: fork failed: %s\n", strerror(errno));
 151                 close(s);
 152                 if (do_pty == 2)
 153                    close(master);
 154                 return 0;
 155
 156          case 0:
 157                 /* Set the DISPLAY */
 158                 if (do_pty == 2) {
 159                         (void) close(master);
 160 #ifdef TIOCSCTTY /* XXXXX */
 161                         (void) setsid();
 162                         ioctl(s, TIOCSCTTY, (char *)NULL);
 163 #endif
 164                 } else {
 165                         getsockname(s, (struct sockaddr *)&addr, &addrlen);
 166                         close(s);
 167                         /*
 168                          * Connect to the socket
 169                          * XXX If any of these fail, we're in trouble!
 170                          */
 171                         s = qemu_socket(AF_INET, SOCK_STREAM, 0);
 172                         addr.sin_addr = loopback_addr;
 173                         do {
 174                             ret = connect(s, (struct sockaddr *)&addr, addrlen);
 175                         } while (ret < 0 && errno == EINTR);
 176                 }
 177
 178                 dup2(s, 0);
 179                 dup2(s, 1);
 180                 dup2(s, 2);
 181                 for (s = getdtablesize() - 1; s >= 3; s--)
 182                    close(s);
 183
 184                 i = 0;
 185                 bptr = qemu_strdup(ex); /* No need to free() this */
 186                 if (do_pty == 1) {
 187                         /* Setup "slirp.telnetd -x" */
 188                         argv[i++] = "slirp.telnetd";
 189                         argv[i++] = "-x";
 190                         argv[i++] = bptr;
 191                 } else
 192                    do {
 193                         /* Change the string into argv[] */
 194                         curarg = bptr;
 195                         while (*bptr != ' ' && *bptr != (char)0)
 196                            bptr++;
 197                         c = *bptr;
 198                         *bptr++ = (char)0;
 199                         argv[i++] = strdup(curarg);
 200                    } while (c);
 201
 202                 argv[i] = NULL;
 203                 execvp(argv[0], (char **)argv);
 204
 205                 /* Ooops, failed, let's tell the user why */
 206                   {
 207                           char buff[256];
 208
 209                           snprintf(buff, sizeof(buff),
 210                                    "Error: execvp of %s failed: %s\n",
 211                                    argv[0], strerror(errno));
 212                           write(2, buff, strlen(buff)+1);
 213                   }
 214                 close(0); close(1); close(2); /* XXX */
 215                 exit(1);
 216
 217          default:
 218                 if (do_pty == 2) {
 219                         close(s);
 220                         so->s = master;
 221                 } else {
 222                         /*
 223                          * XXX this could block us...
 224                          * XXX Should set a timer here, and if accept() doesn't
 225                          * return after X seconds, declare it a failure
 226                          * The only reason this will block forever is if socket()
 227                          * of connect() fail in the child process
 228                          */
 229                         do {
 230                             so->s = accept(s, (struct sockaddr *)&addr, &addrlen);
 231                         } while (so->s < 0 && errno == EINTR);
 232                         closesocket(s);
 233                         opt = 1;
 234                         setsockopt(so->s,SOL_SOCKET,SO_REUSEADDR,(char *)&opt,sizeof(int));
 235                         opt = 1;
 236                         setsockopt(so->s,SOL_SOCKET,SO_OOBINLINE,(char *)&opt,sizeof(int));
 237                 }
 238                 fd_nonblock(so->s);
 239
 240                 /* Append the telnet options now */
 241                 if (so->so_m != NULL && do_pty == 1)  {
 242                         sbappend(so, so->so_m);
 243                         so->so_m = NULL;
 244                 }
 245
 246                 return 1;
 247         }
 248 }
 249 #endif
 250 #endif
 251
 252 #ifndef HAVE_STRDUP
 253 char *
 254 strdup(str)
 255         const char *str;
 256 {
 257         char *bptr;
 258
 259         bptr = (char *)malloc(strlen(str)+1);
 260         strcpy(bptr, str);
 261
 262         return bptr;
 263 }
 264 #endif
 265
 266 void lprint(const char *format, ...)
 267 {
 268     va_list args;
 269
 270     va_start(args, format);
 271                 vfprintf(stderr, format, args);
 272     va_end(args);
 273 }
 274
 275 #ifdef BAD_SPRINTF
 276
 277 #undef vsprintf
 278 #undef sprintf
 279
 280 /*
 281  * Some BSD-derived systems have a sprintf which returns char *
 282  */
 283
 284 int
 285 vsprintf_len(string, format, args)
 286         char *string;
 287         const char *format;
 288         va_list args;
 289 {
 290         vsprintf(string, format, args);
 291         return strlen(string);
 292 }
 293
 294 int
 295 #ifdef __STDC__
 296 sprintf_len(char *string, const char *format, ...)
 297 #else
 298 sprintf_len(va_alist) va_dcl
 299 #endif
 300 {
 301         va_list args;
 302 #ifdef __STDC__
 303         va_start(args, format);
 304 #else
 305         char *string;
 306         char *format;
 307         va_start(args);
 308         string = va_arg(args, char *);
 309         format = va_arg(args, char *);
 310 #endif
 311         vsprintf(string, format, args);
 312         return strlen(string);
 313 }
 314
 315 #endif
 316
 317 void
 318 u_sleep(int usec)
 319 {
 320         struct timeval t;
 321         fd_set fdset;
 322
 323         FD_ZERO(&fdset);
 324
 325         t.tv_sec = 0;
 326         t.tv_usec = usec * 1000;
 327
 328         select(0, &fdset, &fdset, &fdset, &t);
 329 }
 330
 331 /*
 332  * Set fd blocking and non-blocking
 333  */
 334
 335 void
 336 fd_nonblock(int fd)
 337 {
 338 #ifdef FIONBIO
 339 #ifdef _WIN32
 340         unsigned long opt = 1;
 341 #else
 342         int opt = 1;
 343 #endif
 344
 345         ioctlsocket(fd, FIONBIO, &opt);
 346 #else
 347         int opt;
 348
 349         opt = fcntl(fd, F_GETFL, 0);
 350         opt |= O_NONBLOCK;
 351         fcntl(fd, F_SETFL, opt);
 352 #endif
 353 }
 354
 355 void
 356 fd_block(int fd)
 357 {
 358 #ifdef FIONBIO
 359 #ifdef _WIN32
 360         unsigned long opt = 0;
 361 #else
 362         int opt = 0;
 363 #endif
 364
 365         ioctlsocket(fd, FIONBIO, &opt);
 366 #else
 367         int opt;
 368
 369         opt = fcntl(fd, F_GETFL, 0);
 370         opt &= ~O_NONBLOCK;
 371         fcntl(fd, F_SETFL, opt);
 372 #endif
 373 }
 374
 375 #if 0
 376 void slirp_connection_info(Slirp *slirp, Monitor *mon)
 377 {
 378     const char * const tcpstates[] = {
 379         [TCPS_CLOSED]       = "CLOSED",
 380         [TCPS_LISTEN]       = "LISTEN",
 381         [TCPS_SYN_SENT]     = "SYN_SENT",
 382         [TCPS_SYN_RECEIVED] = "SYN_RCVD",
 383         [TCPS_ESTABLISHED]  = "ESTABLISHED",
 384         [TCPS_CLOSE_WAIT]   = "CLOSE_WAIT",
 385         [TCPS_FIN_WAIT_1]   = "FIN_WAIT_1",
 386         [TCPS_CLOSING]      = "CLOSING",
 387         [TCPS_LAST_ACK]     = "LAST_ACK",
 388         [TCPS_FIN_WAIT_2]   = "FIN_WAIT_2",
 389         [TCPS_TIME_WAIT]    = "TIME_WAIT",
 390     };
 391     struct in_addr dst_addr;
 392     struct sockaddr_in src;
 393     socklen_t src_len;
 394     uint16_t dst_port;
 395     struct socket *so;
 396     const char *state;
 397     char buf[20];
 398     int n;
 399
 400     monitor_printf(mon, "  Protocol[State]    FD  Source Address  Port   "
 401                         "Dest. Address  Port RecvQ SendQ\n");
 402
 403     for (so = slirp->tcb.so_next; so != &slirp->tcb; so = so->so_next) {
 404         if (so->so_state & SS_HOSTFWD) {
 405             state = "HOST_FORWARD";
 406         } else if (so->so_tcpcb) {
 407             state = tcpstates[so->so_tcpcb->t_state];
 408         } else {
 409             state = "NONE";
 410         }
 411         if (so->so_state & (SS_HOSTFWD | SS_INCOMING)) {
 412             src_len = sizeof(src);
 413             getsockname(so->s, (struct sockaddr *)&src, &src_len);
 414             dst_addr = so->so_laddr;
 415             dst_port = so->so_lport;
 416         } else {
 417             src.sin_addr = so->so_laddr;
 418             src.sin_port = so->so_lport;
 419             dst_addr = so->so_faddr;
 420             dst_port = so->so_fport;
 421         }
 422         n = snprintf(buf, sizeof(buf), "  TCP[%s]", state);
 423         memset(&buf[n], ' ', 19 - n);
 424         buf[19] = 0;
 425         monitor_printf(mon, "%s %3d %15s %5d ", buf, so->s,
 426                        src.sin_addr.s_addr ? inet_ntoa(src.sin_addr) : "*",
 427                        ntohs(src.sin_port));
 428         monitor_printf(mon, "%15s %5d %5d %5d\n",
 429                        inet_ntoa(dst_addr), ntohs(dst_port),
 430                        so->so_rcv.sb_cc, so->so_snd.sb_cc);
 431     }
 432
 433     for (so = slirp->udb.so_next; so != &slirp->udb; so = so->so_next) {
 434         if (so->so_state & SS_HOSTFWD) {
 435             n = snprintf(buf, sizeof(buf), "  UDP[HOST_FORWARD]");
 436             src_len = sizeof(src);
 437             getsockname(so->s, (struct sockaddr *)&src, &src_len);
 438             dst_addr = so->so_laddr;
 439             dst_port = so->so_lport;
 440         } else {
 441             n = snprintf(buf, sizeof(buf), "  UDP[%d sec]",
 442                          (so->so_expire - curtime) / 1000);
 443             src.sin_addr = so->so_laddr;
 444             src.sin_port = so->so_lport;
 445             dst_addr = so->so_faddr;
 446             dst_port = so->so_fport;
 447         }
 448         memset(&buf[n], ' ', 19 - n);
 449         buf[19] = 0;
 450         monitor_printf(mon, "%s %3d %15s %5d ", buf, so->s,
 451                        src.sin_addr.s_addr ? inet_ntoa(src.sin_addr) : "*",
 452                        ntohs(src.sin_port));
 453         monitor_printf(mon, "%15s %5d %5d %5d\n",
 454                        inet_ntoa(dst_addr), ntohs(dst_port),
 455                        so->so_rcv.sb_cc, so->so_snd.sb_cc);
 456     }
 457 }
 458 #endif