X-Git-Url: https://hackdaworld.org/gitweb/?a=blobdiff_plain;f=stream.c;h=b97a670e77a10f78c5750c0c6ee9afd88b8f02b7;hb=00aca7d3f73b593d2b08b0849ca7eb51574de507;hp=da6cc47a1f0eb92ad5e110ca576c6156b7f0a8c4;hpb=afeae6873ce38f8e98c0b2f28233ca02b61f05a0;p=my-code%2Fivac.git

diff --git a/stream.c b/stream.c
index da6cc47..b97a670 100644
--- a/stream.c
+++ b/stream.c
@@ -26,6 +26,9 @@
 /* read, close */
 #include <unistd.h>
 
+/* defines ... */
+#define MAX_SIZE 1000
+
 int main(int argc, char *argv[]) {
   int listen_fd, send_fd;
   struct sockaddr_in local_addr, remote_addr;
@@ -68,12 +71,13 @@ int main(int argc, char *argv[]) {
       /* send stuff .... */
       read_bytes=1;
       while(read_bytes>0) {
-	unsigned char buf[1000];
+	unsigned char buf[MAX_SIZE];
 	read_bytes=read(0,buf,sizeof(buf));
-        send_bytes=send(send_fd,buf,sizeof(buf),0);
+        send_bytes=send(send_fd,buf,read_bytes,0);
       }
 
       close(send_fd);
+      close(listen_fd);
       printf("connection closed ...\n");
       printf("%d from %d total bytes sent.\n",send_bytes,read_bytes);
   }