File:  [Qemu by Fabrice Bellard] / qemu / slirp / misc.c
Revision 1.1.1.5 (vendor branch): download - view: text, annotated - select for diffs
Tue Apr 24 17:25:52 2018 UTC (3 years ago) by root
Branches: qemu, MAIN
CVS tags: qemu0111, qemu0110, HEAD
qemu 0.11.0

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

unix.superglobalmegacorp.com