X-Git-Url: http://www.chiark.greenend.org.uk/ucgi/~mdw/git/tripe/blobdiff_plain/53a941d3f97a50964587c6e9533b1e43e74a57a8..add23883a6378567be6b7866bc2ff2c32400a346:/pkstream/pkstream.c

diff --git a/pkstream/pkstream.c b/pkstream/pkstream.c
index b84c98b2..4a1ccf50 100644
--- a/pkstream/pkstream.c
+++ b/pkstream/pkstream.c
@@ -54,6 +54,8 @@
 #include <mLib/sel.h>
 #include <mLib/selpk.h>
 
+#include "util.h"
+
 /*----- Data structures ---------------------------------------------------*/
 
 typedef struct pk {
@@ -127,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);
@@ -392,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);