X-Git-Url: http://p2p-next.cs.pub.ro/gitweb/?a=blobdiff_plain;f=test%2Fclient%2Fclient.c;h=e0cf58ec3810bac8eee0e3ee732c53d075166a84;hb=71246fb0c8dead16cc33caf83020961c33f8efda;hp=1e705830f65e329f92f823d91fc11f5c2e59b2d4;hpb=2d6f09bf6e183a2a889d52af3c0acc48c5998043;p=p2p-kernel-protocol.git

diff --git a/test/client/client.c b/test/client/client.c
index 1e70583..e0cf58e 100644
--- a/test/client/client.c
+++ b/test/client/client.c
@@ -9,17 +9,19 @@
 
 #include "p2pkp.h"
 
-#define TEST_FILE "/root/Dropbox/school/p2pkp/test/client/test_file"
-
 int main(int argc, char **argv)
 {
 	int sock, err;
 	struct sockaddr_in psin;
 	struct hostent *host;
+	char *filename;
+	int filename_len;
+	struct timeval start;
+	struct timeval stop;
 
-	if (argc % 2 == 0 || argc < 2) {
-		fprintf(stderr, "Usage: %s <peer_ip1> <peer_port1> "
-				"<pper_ip2> <peer_port2> ...\n", argv[0]);
+	if (argc % 2 || argc < 3) {
+		fprintf(stderr, "Usage: %s <file> <peer_ip1> <peer_port1> "
+				"[<pper_ip2> <peer_port2> ...]\n", argv[0]);
 		return -1;
 	}
 
@@ -34,8 +36,10 @@ int main(int argc, char **argv)
 		return -1;
 	}
 	printf("Successfully created AF_P2PKP socket: %d\n", sock);
+	filename = argv[1];
+	filename_len = strlen(filename);
 
-	argv++;
+	argv+=2;
 	while (*argv) {
 		printf("Connecting to: %s:%s\n", argv[0], argv[1]);
 
@@ -56,14 +60,17 @@ int main(int argc, char **argv)
 
 	printf("Starting to write on the socket\n");
 
-	err = write(sock, TEST_FILE, sizeof(TEST_FILE) - 1);
+	gettimeofday(&start, NULL);
+	err = write(sock, filename, filename_len - 1);
 	if (err < 0) {
-		printf("Error while sending file %s: %s(%d)\n", TEST_FILE,
+		printf("Error while sending file %s: %s(%d)\n", filename,
 				strerror(errno), errno);
 		return err;
 	}
+	gettimeofday(&stop, NULL);
 
-	printf("Successfully written file\n");
+	printf("Successfully written %s in %ld seconds\n", filename, 
+			stop.tv_sec - start.tv_sec);
 
 	return 0;
 }