X-Git-Url: https://hackdaworld.org/gitweb/?a=blobdiff_plain;f=receive.c;h=9fe86e256bb98ff132608ffa01f173d64c984cd1;hb=d333f09513d243b67eb0ab5d5272461184f1e5a8;hp=ba61d5d816322fb4f19bdd878a21c72f1c32e941;hpb=afeae6873ce38f8e98c0b2f28233ca02b61f05a0;p=my-code%2Fivac.git

diff --git a/receive.c b/receive.c
index ba61d5d..9fe86e2 100644
--- a/receive.c
+++ b/receive.c
@@ -26,14 +26,16 @@
 /* read, close */
 #include <unistd.h>
 
+/* defines ... */
+#define MAX_SIZE 1024
+
 int main(int argc, char *argv[]) {
   int receive_fd;
   struct sockaddr_in target_addr;
-  socklen_t target_addr_len;
   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);
   }
 
@@ -53,13 +55,11 @@ int main(int argc, char *argv[]) {
   exit(1);
   }
 
-  printf("connected.\n");
-
   receive_bytes=1;
   while(receive_bytes>0) {
-	unsigned char buf[1000];
+	unsigned char buf[MAX_SIZE];
 	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);