X-Git-Url: https://hackdaworld.org/gitweb/?p=my-code%2Fivac.git;a=blobdiff_plain;f=receive.c;h=65b9b95fde7352eec4661bd1cdf4e74e821fcfdd;hp=36f8ffd770de4a8f33a9ac85b57ceddceb1fa34f;hb=fa6a332e18a1ed735e380d1d2081f5778009d6ab;hpb=3cc99313fc75043c07720bf64e5ac76496057864

diff --git a/receive.c b/receive.c
index 36f8ffd..65b9b95 100644
--- a/receive.c
+++ b/receive.c
@@ -32,7 +32,7 @@ int main(int argc, char *argv[]) {
   int receive_bytes, write_bytes;
 
   if(argc!=3) {
-    printf("usage: %s <port>\n",argv[0]);
+    printf("usage: %s <target-ip> <port>\n",argv[0]);
     exit(1);
   }
 
@@ -58,7 +58,7 @@ int main(int argc, char *argv[]) {
   while(receive_bytes>0) {
 	unsigned char buf[1000];
 	receive_bytes=recv(receive_fd,buf,sizeof(buf),0);
-        write_bytes=write(1,buf,sizeof(buf));
+        write_bytes=write(1,buf,receive_bytes);
   }
 
   close(receive_fd);