fixed send(from) recv(from) bug, added dgramrcv.c

[my-code/ivac.git] / receive.c

diff --git a/receive.c b/receive.c
index ba61d5d..65b9b95 100644 (file)
--- a/receive.c
+++ b/receive.c
@@ -29,11 +29,10 @@
 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);
   }
 
@@ -59,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);