Add counters for MPTP buffers and syscalls.

[swifty.git] / src / raw / swift_raw_simple_test.c

diff --git a/src/raw/swift_raw_simple_test.c b/src/raw/swift_raw_simple_test.c
index a3dae44..dfd1a11 100644 (file)
--- a/src/raw/swift_raw_simple_test.c
+++ b/src/raw/swift_raw_simple_test.c
@@ -8,12 +8,18 @@
 #include <stdlib.h>
 #include <string.h>
 #include <sys/types.h>
+#include <arpa/inet.h>
 
 #include "swift_types.h"
 #include "swift_raw.h"
 #include "debug.h"
 #include "util.h"
 
+static void usage(const char *argv0)
+{
+       fprintf(stderr, "Usage: %s local_address swift_hash [remote_address]\n", argv0);
+}
+
 /*
  * Create a socket, bind it and send data.
  */
@@ -26,27 +32,24 @@ int main(int argc, char *argv[])
        ssize_t bytes_sent;
        int rc;
 
-       if (argc < 3)
-       {
-               fprintf(stderr,"Usage \"./client ip_local hash ip_dest?\" .");
+       if (argc != 3 && argc != 4) {
+               usage(argv[0]);
+               exit(EXIT_FAILURE);
        }
 
        sockfd = sw_socket(PF_INET, SOCK_DGRAM, IPPROTO_SWIFT);
        DIE(sockfd < 0, "sw_socket");
 
-       /* TODO: init_addr */
-       local_addr.sin_addr.s_addr = htonl(argv[1]);
+       memset(&local_addr, 0, sizeof(local_addr));
+       inet_pton(AF_INET, argv[1], &local_addr.sin_addr.s_addr);
        memcpy(&local_addr.sw_hash, argv[2], sizeof(struct sw_hash));
        rc = sw_bind(sockfd, (struct sockaddr *) &local_addr, sizeof(local_addr));
        DIE(rc < 0, "sw_bind");
 
-       /* TODO: init remote_addr */
-       if (argv > 3)
-       {
-               remote_addr.sin_addr.s_addr = htonl(argv[1]);
-               memcpy(&remote_addr.sw_hash, argv[3], sizeof(struct sw_hash));
-               bytes_sent = sw_sendto(sockfd, buffer, BUFSIZ, 0,
-                                       (struct sockaddr *) &remote_addr, sizeof(remote_addr));
+       if (argc > 3) {
+               inet_pton(AF_INET, argv[3], &remote_addr.sin_addr.s_addr);
+               memcpy(&remote_addr.sw_hash, argv[2], sizeof(struct sw_hash));
+               bytes_sent = sw_sendto(sockfd, buffer, BUFSIZ, 0, (struct sockaddr *) &remote_addr, sizeof(remote_addr));
                DIE(bytes_sent < 0, "sw_sendto");
        }