X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~mdw/git/tripe/blobdiff_plain/e04c2d50fd96f3f31bc96851c55c6efecc10469c..06a174dfe42020da6138e22c82803cbba9c66b49:/pkstream/pkstream.c

diff --git a/pkstream/pkstream.c b/pkstream/pkstream.c
index df839993..4a1ccf50 100644
--- 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);