test: added timer for client

[p2p-kernel-protocol.git] / test / client / client.c

diff --git a/test/client/client.c b/test/client/client.c
index 1e70583..e0cf58e 100644 (file)
--- 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;
 }