]> git.proxmox.com Git - qemu.git/blobdiff - slirp/socket.c
rng-egd: remove redundant free
[qemu.git] / slirp / socket.c
index 0c15132eae901a9d4361844db32c4584de6ea538..37ac5cf2fb260cffbf85b5f4f80b72d0fbcc9053 100644 (file)
@@ -5,10 +5,9 @@
  * terms and conditions of the copyright.
  */
 
-#define WANT_SYS_IOCTL_H
+#include "qemu-common.h"
 #include <slirp.h>
 #include "ip_icmp.h"
-#include "main.h"
 #ifdef __sun__
 #include <sys/filio.h>
 #endif
 static void sofcantrcvmore(struct socket *so);
 static void sofcantsendmore(struct socket *so);
 
-#if 0
-static void
-so_init()
-{
-       /* Nothing yet */
-}
-#endif
-
 struct socket *
-solookup(head, laddr, lport, faddr, fport)
-       struct socket *head;
-       struct in_addr laddr;
-       u_int lport;
-       struct in_addr faddr;
-       u_int fport;
+solookup(struct socket *head, struct in_addr laddr, u_int lport,
+         struct in_addr faddr, u_int fport)
 {
        struct socket *so;
 
@@ -54,7 +41,7 @@ solookup(head, laddr, lport, faddr, fport)
  * insque() it into the correct linked-list
  */
 struct socket *
-socreate()
+socreate(Slirp *slirp)
 {
   struct socket *so;
 
@@ -63,6 +50,8 @@ socreate()
     memset(so, 0, sizeof(struct socket));
     so->so_state = SS_NOFDREF;
     so->s = -1;
+    so->slirp = slirp;
+    so->pollfds_idx = -1;
   }
   return(so);
 }
@@ -71,18 +60,21 @@ socreate()
  * remque and free a socket, clobber cache
  */
 void
-sofree(so)
-       struct socket *so;
+sofree(struct socket *so)
 {
+  Slirp *slirp = so->slirp;
+
   if (so->so_emu==EMU_RSH && so->extra) {
        sofree(so->extra);
        so->extra=NULL;
   }
-  if (so == tcp_last_so)
-    tcp_last_so = &tcb;
-  else if (so == udp_last_so)
-    udp_last_so = &udb;
-
+  if (so == slirp->tcp_last_so) {
+      slirp->tcp_last_so = &slirp->tcb;
+  } else if (so == slirp->udp_last_so) {
+      slirp->udp_last_so = &slirp->udb;
+  } else if (so == slirp->icmp_last_so) {
+      slirp->icmp_last_so = &slirp->icmp;
+  }
   m_free(so->so_m);
 
   if(so->so_next && so->so_prev)
@@ -91,32 +83,22 @@ sofree(so)
   free(so);
 }
 
-/*
- * Read from so's socket into sb_snd, updating all relevant sbuf fields
- * NOTE: This will only be called if it is select()ed for reading, so
- * a read() of 0 (or less) means it's disconnected
- */
-int
-soread(so)
-       struct socket *so;
+size_t sopreprbuf(struct socket *so, struct iovec *iov, int *np)
 {
-       int n, nn, lss, total;
+       int n, lss, total;
        struct sbuf *sb = &so->so_snd;
        int len = sb->sb_datalen - sb->sb_cc;
-       struct iovec iov[2];
        int mss = so->so_tcpcb->t_maxseg;
 
-       DEBUG_CALL("soread");
+       DEBUG_CALL("sopreprbuf");
        DEBUG_ARG("so = %lx", (long )so);
 
-       /*
-        * No need to check if there's enough room to read.
-        * soread wouldn't have been called if there weren't
-        */
-
-       len = sb->sb_datalen - sb->sb_cc;
+       if (len <= 0)
+               return 0;
 
        iov[0].iov_base = sb->sb_wptr;
+        iov[1].iov_base = NULL;
+        iov[1].iov_len = 0;
        if (sb->sb_wptr < sb->sb_rptr) {
                iov[0].iov_len = sb->sb_rptr - sb->sb_wptr;
                /* Should never succeed, but... */
@@ -154,12 +136,38 @@ soread(so)
                        n = 1;
                }
        }
+       if (np)
+               *np = n;
+
+       return iov[0].iov_len + (n - 1) * iov[1].iov_len;
+}
+
+/*
+ * Read from so's socket into sb_snd, updating all relevant sbuf fields
+ * NOTE: This will only be called if it is select()ed for reading, so
+ * a read() of 0 (or less) means it's disconnected
+ */
+int
+soread(struct socket *so)
+{
+       int n, nn;
+       struct sbuf *sb = &so->so_snd;
+       struct iovec iov[2];
+
+       DEBUG_CALL("soread");
+       DEBUG_ARG("so = %lx", (long )so);
+
+       /*
+        * No need to check if there's enough room to read.
+        * soread wouldn't have been called if there weren't
+        */
+       sopreprbuf(so, iov, &n);
 
 #ifdef HAVE_READV
        nn = readv(so->s, (struct iovec *)iov, n);
        DEBUG_MISC((dfd, " ... read nn = %d bytes\n", nn));
 #else
-       nn = recv(so->s, iov[0].iov_base, iov[0].iov_len,0);
+       nn = qemu_recv(so->s, iov[0].iov_base, iov[0].iov_len,0);
 #endif
        if (nn <= 0) {
                if (nn < 0 && (errno == EINTR || errno == EAGAIN))
@@ -184,7 +192,7 @@ soread(so)
         */
        if (n == 2 && nn == iov[0].iov_len) {
             int ret;
-            ret = recv(so->s, iov[1].iov_base, iov[1].iov_len,0);
+            ret = qemu_recv(so->s, iov[1].iov_base, iov[1].iov_len,0);
             if (ret > 0)
                 nn += ret;
         }
@@ -200,6 +208,48 @@ soread(so)
        return nn;
 }
 
+int soreadbuf(struct socket *so, const char *buf, int size)
+{
+    int n, nn, copy = size;
+       struct sbuf *sb = &so->so_snd;
+       struct iovec iov[2];
+
+       DEBUG_CALL("soreadbuf");
+       DEBUG_ARG("so = %lx", (long )so);
+
+       /*
+        * No need to check if there's enough room to read.
+        * soread wouldn't have been called if there weren't
+        */
+       if (sopreprbuf(so, iov, &n) < size)
+        goto err;
+
+    nn = MIN(iov[0].iov_len, copy);
+    memcpy(iov[0].iov_base, buf, nn);
+
+    copy -= nn;
+    buf += nn;
+
+    if (copy == 0)
+        goto done;
+
+    memcpy(iov[1].iov_base, buf, copy);
+
+done:
+    /* Update fields */
+       sb->sb_cc += size;
+       sb->sb_wptr += size;
+       if (sb->sb_wptr >= (sb->sb_data + sb->sb_datalen))
+               sb->sb_wptr -= sb->sb_datalen;
+    return size;
+err:
+
+    sofcantrcvmore(so);
+    tcp_sockclosed(sototcpcb(so));
+    fprintf(stderr, "soreadbuf buffer to small");
+    return -1;
+}
+
 /*
  * Get urgent data
  *
@@ -208,8 +258,7 @@ soread(so)
  * in the send buffer is sent as urgent data
  */
 void
-sorecvoob(so)
-       struct socket *so;
+sorecvoob(struct socket *so)
 {
        struct tcpcb *tp = sototcpcb(so);
 
@@ -236,8 +285,7 @@ sorecvoob(so)
  * There's a lot duplicated code here, but...
  */
 int
-sosendoob(so)
-       struct socket *so;
+sosendoob(struct socket *so)
 {
        struct sbuf *sb = &so->so_rcv;
        char buff[2048]; /* XXX Shouldn't be sending more oob data than this */
@@ -253,7 +301,7 @@ sosendoob(so)
 
        if (sb->sb_rptr < sb->sb_wptr) {
                /* We can send it directly */
-               n = send(so->s, sb->sb_rptr, so->so_urgc, (MSG_OOB)); /* |MSG_DONTWAIT)); */
+               n = slirp_send(so, sb->sb_rptr, so->so_urgc, (MSG_OOB)); /* |MSG_DONTWAIT)); */
                so->so_urgc -= n;
 
                DEBUG_MISC((dfd, " --- sent %d bytes urgent data, %d urgent bytes left\n", n, so->so_urgc));
@@ -274,7 +322,7 @@ sosendoob(so)
                        so->so_urgc -= n;
                        len += n;
                }
-               n = send(so->s, buff, len, (MSG_OOB)); /* |MSG_DONTWAIT)); */
+               n = slirp_send(so, buff, len, (MSG_OOB)); /* |MSG_DONTWAIT)); */
 #ifdef DEBUG
                if (n != len)
                   DEBUG_ERROR((dfd, "Didn't send all data urgently XXXXX\n"));
@@ -295,8 +343,7 @@ sosendoob(so)
  * updating all sbuf field as necessary
  */
 int
-sowrite(so)
-       struct socket *so;
+sowrite(struct socket *so)
 {
        int  n,nn;
        struct sbuf *sb = &so->so_rcv;
@@ -317,9 +364,9 @@ sowrite(so)
         * sowrite wouldn't have been called otherwise
         */
 
-        len = sb->sb_cc;
-
        iov[0].iov_base = sb->sb_rptr;
+        iov[1].iov_base = NULL;
+        iov[1].iov_len = 0;
        if (sb->sb_rptr < sb->sb_wptr) {
                iov[0].iov_len = sb->sb_wptr - sb->sb_rptr;
                /* Should never succeed, but... */
@@ -344,7 +391,7 @@ sowrite(so)
 
        DEBUG_MISC((dfd, "  ... wrote nn = %d bytes\n", nn));
 #else
-       nn = send(so->s, iov[0].iov_base, iov[0].iov_len,0);
+       nn = slirp_send(so, iov[0].iov_base, iov[0].iov_len,0);
 #endif
        /* This should never happen, but people tell me it does *shrug* */
        if (nn < 0 && (errno == EAGAIN || errno == EINTR))
@@ -361,7 +408,7 @@ sowrite(so)
 #ifndef HAVE_READV
        if (n == 2 && nn == iov[0].iov_len) {
             int ret;
-            ret = send(so->s, iov[1].iov_base, iov[1].iov_len,0);
+            ret = slirp_send(so, iov[1].iov_base, iov[1].iov_len,0);
             if (ret > 0)
                 nn += ret;
         }
@@ -388,11 +435,10 @@ sowrite(so)
  * recvfrom() a UDP socket
  */
 void
-sorecvfrom(so)
-       struct socket *so;
+sorecvfrom(struct socket *so)
 {
        struct sockaddr_in addr;
-       int addrlen = sizeof(struct sockaddr_in);
+       socklen_t addrlen = sizeof(struct sockaddr_in);
 
        DEBUG_CALL("sorecvfrom");
        DEBUG_ARG("so = %lx", (long)so);
@@ -416,15 +462,23 @@ sorecvfrom(so)
            icmp_error(so->so_m, ICMP_UNREACH,code, 0,strerror(errno));
          } else {
            icmp_reflect(so->so_m);
-           so->so_m = 0; /* Don't m_free() it again! */
+            so->so_m = NULL; /* Don't m_free() it again! */
          }
          /* No need for this socket anymore, udp_detach it */
          udp_detach(so);
        } else {                                /* A "normal" UDP packet */
          struct mbuf *m;
-         int len, n;
+          int len;
+#ifdef _WIN32
+          unsigned long n;
+#else
+          int n;
+#endif
 
-         if (!(m = m_get())) return;
+         m = m_get(so->slirp);
+         if (!m) {
+             return;
+         }
          m->m_data += IF_MAXLINKHDR;
 
          /*
@@ -469,12 +523,6 @@ sorecvfrom(so)
                so->so_expire = curtime + SO_EXPIRE;
            }
 
-           /*          if (m->m_len == len) {
-            *                  m_inc(m, MINCSIZE);
-            *                  m->m_len = 0;
-            *          }
-            */
-
            /*
             * If this packet was destined for CTL_ADDR,
             * make it look like that's where it came from, done by udp_output
@@ -488,10 +536,9 @@ sorecvfrom(so)
  * sendto() a socket
  */
 int
-sosendto(so, m)
-       struct socket *so;
-       struct mbuf *m;
+sosendto(struct socket *so, struct mbuf *m)
 {
+       Slirp *slirp = so->slirp;
        int ret;
        struct sockaddr_in addr;
 
@@ -500,16 +547,14 @@ sosendto(so, m)
        DEBUG_ARG("m = %lx", (long)m);
 
         addr.sin_family = AF_INET;
-       if ((so->so_faddr.s_addr & htonl(0xffffff00)) == special_addr.s_addr) {
+       if ((so->so_faddr.s_addr & slirp->vnetwork_mask.s_addr) ==
+           slirp->vnetwork_addr.s_addr) {
          /* It's an alias */
-         switch(ntohl(so->so_faddr.s_addr) & 0xff) {
-         case CTL_DNS:
-           addr.sin_addr = dns_addr;
-           break;
-         case CTL_ALIAS:
-         default:
+         if (so->so_faddr.s_addr == slirp->vnameserver_addr.s_addr) {
+           if (get_dns_addr(&addr.sin_addr) < 0)
+             addr.sin_addr = loopback_addr;
+         } else {
            addr.sin_addr = loopback_addr;
-           break;
          }
        } else
          addr.sin_addr = so->so_faddr;
@@ -529,32 +574,33 @@ sosendto(so, m)
         */
        if (so->so_expire)
                so->so_expire = curtime + SO_EXPIRE;
-       so->so_state = SS_ISFCONNECTED; /* So that it gets select()ed */
+       so->so_state &= SS_PERSISTENT_MASK;
+       so->so_state |= SS_ISFCONNECTED; /* So that it gets select()ed */
        return 0;
 }
 
 /*
- * XXX This should really be tcp_listen
+ * Listen for incoming TCP connections
  */
 struct socket *
-solisten(port, laddr, lport, flags)
-       u_int port;
-       u_int32_t laddr;
-       u_int lport;
-       int flags;
+tcp_listen(Slirp *slirp, uint32_t haddr, u_int hport, uint32_t laddr,
+           u_int lport, int flags)
 {
        struct sockaddr_in addr;
        struct socket *so;
-       int s, addrlen = sizeof(addr), opt = 1;
+       int s, opt = 1;
+       socklen_t addrlen = sizeof(addr);
+       memset(&addr, 0, addrlen);
 
-       DEBUG_CALL("solisten");
-       DEBUG_ARG("port = %d", port);
+       DEBUG_CALL("tcp_listen");
+       DEBUG_ARG("haddr = %x", haddr);
+       DEBUG_ARG("hport = %d", hport);
        DEBUG_ARG("laddr = %x", laddr);
        DEBUG_ARG("lport = %d", lport);
        DEBUG_ARG("flags = %x", flags);
 
-       if ((so = socreate()) == NULL) {
-         /* free(so);      Not sofree() ??? free(NULL) == NOP */
+       so = socreate(slirp);
+       if (!so) {
          return NULL;
        }
 
@@ -563,7 +609,7 @@ solisten(port, laddr, lport, flags)
                free(so);
                return NULL;
        }
-       insque(so,&tcb);
+       insque(so, &slirp->tcb);
 
        /*
         * SS_FACCEPTONCE sockets must time out.
@@ -571,16 +617,17 @@ solisten(port, laddr, lport, flags)
        if (flags & SS_FACCEPTONCE)
           so->so_tcpcb->t_timer[TCPT_KEEP] = TCPTV_KEEP_INIT*2;
 
-       so->so_state = (SS_FACCEPTCONN|flags);
+       so->so_state &= SS_PERSISTENT_MASK;
+       so->so_state |= (SS_FACCEPTCONN | flags);
        so->so_lport = lport; /* Kept in network format */
        so->so_laddr.s_addr = laddr; /* Ditto */
 
        addr.sin_family = AF_INET;
-       addr.sin_addr.s_addr = INADDR_ANY;
-       addr.sin_port = port;
+       addr.sin_addr.s_addr = haddr;
+       addr.sin_port = hport;
 
-       if (((s = socket(AF_INET,SOCK_STREAM,0)) < 0) ||
-           (setsockopt(s,SOL_SOCKET,SO_REUSEADDR,(char *)&opt,sizeof(int)) < 0) ||
+       if (((s = qemu_socket(AF_INET,SOCK_STREAM,0)) < 0) ||
+           (socket_set_fast_reuse(s) < 0) ||
            (bind(s,(struct sockaddr *)&addr, sizeof(addr)) < 0) ||
            (listen(s,1) < 0)) {
                int tmperrno = errno; /* Don't clobber the real reason we failed */
@@ -595,12 +642,12 @@ solisten(port, laddr, lport, flags)
 #endif
                return NULL;
        }
-       setsockopt(s,SOL_SOCKET,SO_OOBINLINE,(char *)&opt,sizeof(int));
+       qemu_setsockopt(s, SOL_SOCKET, SO_OOBINLINE, &opt, sizeof(int));
 
        getsockname(s,(struct sockaddr *)&addr,&addrlen);
        so->so_fport = addr.sin_port;
        if (addr.sin_addr.s_addr == 0 || addr.sin_addr.s_addr == loopback_addr.s_addr)
-          so->so_faddr = alias_addr;
+          so->so_faddr = slirp->vhost_addr;
        else
           so->so_faddr = addr.sin_addr;
 
@@ -608,33 +655,6 @@ solisten(port, laddr, lport, flags)
        return so;
 }
 
-#if 0
-/*
- * Data is available in so_rcv
- * Just write() the data to the socket
- * XXX not yet...
- */
-static void
-sorwakeup(so)
-       struct socket *so;
-{
-/*     sowrite(so); */
-/*     FD_CLR(so->s,&writefds); */
-}
-
-/*
- * Data has been freed in so_snd
- * We have room for a read() if we want to
- * For now, don't read, it'll be done in the main loop
- */
-static void
-sowwakeup(so)
-       struct socket *so;
-{
-       /* Nothing, yet */
-}
-#endif
-
 /*
  * Various session state calls
  * XXX Should be #define's
@@ -642,8 +662,7 @@ sowwakeup(so)
  * times each when only 1 was needed
  */
 void
-soisfconnecting(so)
-       register struct socket *so;
+soisfconnecting(struct socket *so)
 {
        so->so_state &= ~(SS_NOFDREF|SS_ISFCONNECTED|SS_FCANTRCVMORE|
                          SS_FCANTSENDMORE|SS_FWDRAIN);
@@ -651,8 +670,7 @@ soisfconnecting(so)
 }
 
 void
-soisfconnected(so)
-        register struct socket *so;
+soisfconnected(struct socket *so)
 {
        so->so_state &= ~(SS_ISFCONNECTING|SS_FWDRAIN|SS_NOFDREF);
        so->so_state |= SS_ISFCONNECTED; /* Clobber other states */
@@ -663,15 +681,14 @@ sofcantrcvmore(struct socket *so)
 {
        if ((so->so_state & SS_NOFDREF) == 0) {
                shutdown(so->s,0);
-               if(global_writefds) {
-                 FD_CLR(so->s,global_writefds);
-               }
        }
        so->so_state &= ~(SS_ISFCONNECTING);
-       if (so->so_state & SS_FCANTSENDMORE)
-          so->so_state = SS_NOFDREF; /* Don't select it */ /* XXX close() here as well? */
-       else
+       if (so->so_state & SS_FCANTSENDMORE) {
+          so->so_state &= SS_PERSISTENT_MASK;
+          so->so_state |= SS_NOFDREF; /* Don't select it */
+       } else {
           so->so_state |= SS_FCANTRCVMORE;
+       }
 }
 
 static void
@@ -679,30 +696,14 @@ sofcantsendmore(struct socket *so)
 {
        if ((so->so_state & SS_NOFDREF) == 0) {
             shutdown(so->s,1);           /* send FIN to fhost */
-            if (global_readfds) {
-                FD_CLR(so->s,global_readfds);
-            }
-            if (global_xfds) {
-                FD_CLR(so->s,global_xfds);
-            }
        }
        so->so_state &= ~(SS_ISFCONNECTING);
-       if (so->so_state & SS_FCANTRCVMORE)
-          so->so_state = SS_NOFDREF; /* as above */
-       else
+       if (so->so_state & SS_FCANTRCVMORE) {
+          so->so_state &= SS_PERSISTENT_MASK;
+          so->so_state |= SS_NOFDREF; /* as above */
+       } else {
           so->so_state |= SS_FCANTSENDMORE;
-}
-
-void
-soisfdisconnected(so)
-       struct socket *so;
-{
-/*     so->so_state &= ~(SS_ISFCONNECTING|SS_ISFCONNECTED); */
-/*     close(so->s); */
-/*     so->so_state = SS_ISFDISCONNECTED; */
-       /*
-        * XXX Do nothing ... ?
-        */
+       }
 }
 
 /*
@@ -710,12 +711,10 @@ soisfdisconnected(so)
  * Set CANTSENDMORE once all data has been write()n
  */
 void
-sofwdrain(so)
-       struct socket *so;
+sofwdrain(struct socket *so)
 {
        if (so->so_rcv.sb_cc)
                so->so_state |= SS_FWDRAIN;
        else
                sofcantsendmore(so);
 }
-