server/tests.at (AWAIT_KXDONE): More warning suppressions.

[tripe] / pkstream / pkstream.c

diff --git a/pkstream/pkstream.c b/pkstream/pkstream.c
index df8399930c697f7915908b3f5dc875ff6d65f037..4a1ccf5004150783d227f91732d5fb4c19798557 100644 (file)
--- a/pkstream/pkstream.c
+++ b/pkstream/pkstream.c
@@ -1,6 +1,4 @@
 /* -*-c-*-
- *
- * $Id: pkstream.c,v 1.3 2004/04/08 01:36:17 mdw Exp $
  *
  * Forwarding UDP packets over a stream
  *
@@ -56,6 +54,8 @@
 #include <mLib/sel.h>
 #include <mLib/selpk.h>
 
+#include "util.h"
+
 /*----- Data structures ---------------------------------------------------*/
 
 typedef struct pk {
@@ -129,7 +129,7 @@ static void rdtcp(octet *b, size_t sz, pkbuf *pk, size_t *k, void *vp)
   }
   pksz = LOAD16(b);
   if (pksz + 2 == sz) {
-    write(fd_udp, b + 2, pksz);
+    DISCARD(write(fd_udp, b + 2, pksz));
     selpk_want(&p->p, 2);
   } else {
     selpk_want(&p->p, pksz + 2);
@@ -394,6 +394,7 @@ int main(int argc, char *argv[])
        break;
       case 'b':
        parseaddr(optarg, &bindaddr.sin_addr, 0);
+       cw.me.sin_addr = bindaddr.sin_addr;
        break;
       case 'c':
        parseaddr(optarg, &connaddr.sin_addr, &pt);