API unit test script

2016-08-29 14:47:28 -07:00
parent 15199791b4
commit 0ea3df14f8
17 changed files with 685 additions and 13 deletions
--- a/tests/api_test/clients.sh
+++ b/tests/api_test/clients.sh
@@ -0,0 +1,46 @@
+#!/bin/bash
+
+echo "\n\n\n\nStarting client(s)"
+
+# test.sh [udp|tcp|all] [nwid]
+TEST_EXECUTABLE_PATH="build/tests"
+protocol=$1
+NWID=$2
+HOME_DIR=$(pwd)/$TEST_EXECUTABLE_PATH
+ZT_HOME_PATH=$(pwd)/$TEST_EXECUTABLE_PATH/zerotier
+
+BUILD_PATH=$(pwd)/build
+localAddr="127.0.0.1"
+
+export ZT_NC_NETWORK=$ZT_HOME_PATH/nc_$NWID
+export DYLD_LIBRARY_PATH=.$HOME_DIR/libztintercept.so
+
+echo "network = " $NWID
+echo "protocol = " $protocol
+echo "ZT_NC_NETWORK = " $ZT_NC_NETWORK
+echo "ZT_NC_NETWORK = " $ZT_NC_NETWORK
+
+echo "ZT_HOME_PATH = " $ZT_HOME_PATH
+echo "DYLD_LIBRARY_PATH = " $DYLD_LIBRARY_PATH
+
+# Start ZeroTier service
+echo "Starting ZeroTier background service..."
+mkdir -p $ZT_HOME_PATH/networks.d
+touch $ZT_HOME_PATH/networks.d/$NWID.conf
+$BUILD_PATH/zerotier-sdk-service -U -p$RANDOM $ZT_HOME_PATH &
+zt_service_pid=$!
+
+if [ $protocol="tcp" ]; then
+    echo "Starting TCP test..."
+    random_tcp_server_port=$RANDOM
+    ./$TEST_EXECUTABLE_PATH/Darwin.tcp_server.out $random_tcp_server_port &
+    tcp_server_pid=$!    
+    echo "TCP SERVER AT = " $localAddr ":" $random_tcp_server_port
+    sleep 3
+    ./$TEST_EXECUTABLE_PATH/Darwin.tcp_client.out $localAddr $random_tcp_server_port &
+    tcp_client_pid=$!
+fi
+
+echo "Waiting for test to conclude..."
+sleep 5
+kill -9 $zt_service_PID $tcp_server_pid $tcp_client_pid
--- a/tests/api_test/servers.sh
+++ b/tests/api_test/servers.sh
@@ -0,0 +1,51 @@
+#!/bin/bash
+
+echo "\n\n\n\nStarting server(s)"
+
+# test.sh [udp|tcp|all] [nwid]
+TEST_EXECUTABLE_PATH="build/tests"
+protocol=$1
+NWID=$2
+HOME_DIR=$(pwd)/$TEST_EXECUTABLE_PATH
+ZT_HOME_PATH=$(pwd)/$TEST_EXECUTABLE_PATH/zerotier
+
+BUILD_PATH=$(pwd)/build
+TEST_PATH=$(pwd)/build/tests
+localAddr="127.0.0.1"
+
+export ZT_NC_NETWORK=$ZT_HOME_PATH/nc_$NWID
+export DYLD_LIBRARY_PATH=.$HOME_DIR/libztintercept.so
+
+echo "network = " $NWID
+echo "protocol = " $protocol
+echo "ZT_NC_NETWORK = " $ZT_NC_NETWORK
+echo "ZT_NC_NETWORK = " $ZT_NC_NETWORK
+echo "ZT_HOME_PATH = " $ZT_HOME_PATH
+echo "DYLD_LIBRARY_PATH = " $DYLD_LIBRARY_PATH
+
+# Start ZeroTier service
+echo "Starting ZeroTier background service..."
+mkdir -p $ZT_HOME_PATH/networks.d
+touch $ZT_HOME_PATH/networks.d/$NWID.conf
+$BUILD_PATH/zerotier-sdk-service -U -p$RANDOM $ZT_HOME_PATH &
+zt_service_pid=$!
+
+if [ $protocol="tcp" ]; then
+    echo "Starting TCP test..."
+    random_tcp_server_port=$RANDOM
+    echo $random_tcp_server_port > $TEST_EXECUTABLE_PATH/tcp_server.port
+    ./$TEST_EXECUTABLE_PATH/Darwin.tcp_server.out $random_tcp_server_port &
+    tcp_server_pid=$!    
+
+    # echo "TCP SERVER AT = " $localAddr ":" $random_tcp_server_port
+    # sleep 3
+    # ./$TEST_EXECUTABLE_PATH/Darwin.tcp_client.out $localAddr $random_tcp_server_port &
+    # tcp_client_pid=$!
+fi
+
+echo "Waiting for test to conclude..."
+echo $random_tcp_server_port
+# ,$random_udp_server_port
+sleep 10
+echo "Cleaning up"
+kill -9 $zt_service_PID $tcp_server_pid $tcp_client_pid
--- a/tests/api_test/tcp_client.c
+++ b/tests/api_test/tcp_client.c
@@ -0,0 +1,49 @@
+// TCP Client test program
+
+#include <stdio.h>
+#include <string.h>
+#include <sys/socket.h>
+#include <arpa/inet.h>
+ 
+int atoi(const char *str);
+int close(int filedes);
+
+int main(int argc , char *argv[])
+{
+    if(argc < 3) {
+        printf("usage: client <addr> <port>\n");
+        return 1;
+    }
+   
+    int sock, port = atoi(argv[2]);
+    struct sockaddr_in server;
+    char message[1000] , server_reply[2000];
+     
+    sock = socket(AF_INET , SOCK_STREAM , 0);
+    if (sock == -1) {
+        printf("could not create socket");
+    }     
+    server.sin_addr.s_addr = inet_addr(argv[1]);
+    server.sin_family = AF_INET;
+    server.sin_port = htons( port );
+ 
+    printf("connecting...\n");
+    if (connect(sock , (struct sockaddr *)&server , sizeof(server)) < 0) {
+        perror("connect failed. Error");
+        return 1;
+    }
+    printf("connected\n");
+
+    char *msg = "welcome to the machine!";
+    // TX         
+    if(send(sock, msg, strlen(msg), 0) < 0) {
+        printf("send failed");
+        return 1;
+    }
+    else {
+        printf("sent message: %s\n", msg);
+        printf("len = %ld\n", strlen(msg));
+    }
+    close(sock);
+    return 0;
+}
--- a/tests/api_test/tcp_server.c
+++ b/tests/api_test/tcp_server.c
@@ -0,0 +1,67 @@
+// TCP Server test program
+
+#include <stdio.h>
+#include <string.h>
+#include <sys/socket.h>
+#include <arpa/inet.h> 
+#include <unistd.h>  
+ 
+int atoi(const char *str);
+
+int main(int argc , char *argv[])
+{
+    if(argc < 2) {
+	   printf("usage: tcp_server <port>\n");
+       return 0;
+    }
+   
+    int socket_desc, client_sock, c, read_size, port = atoi(argv[1]);
+    struct sockaddr_in server , client;
+    char client_message[2000];
+     
+    socket_desc = socket(AF_INET , SOCK_STREAM , 0);
+    if (socket_desc == -1) {
+        printf("could not create socket");
+        return 0;
+    }
+     
+    server.sin_family = AF_INET;
+    server.sin_addr.s_addr = INADDR_ANY;
+    server.sin_port = htons(port);
+     
+    printf("binding on port %d\n", port);
+    if( bind(socket_desc,(struct sockaddr *)&server , sizeof(server)) < 0) {
+        perror("bind failed. Error");
+        return 0;
+    }
+    printf("listening\n");
+    listen(socket_desc , 3); 
+    printf("waiting to accept\n");
+    c = sizeof(struct sockaddr_in);
+    
+    client_sock = accept(socket_desc, (struct sockaddr *)&client, (socklen_t*)&c);
+    if (client_sock < 0) {
+        perror("accept failed");
+        return 0;
+    }
+    printf("connection accepted\n reading...\n");
+     
+    // RX
+
+    int msglen = 1024;
+    unsigned long count = 0;
+    while(1)
+    {
+        count++;
+       int bytes_read = read(client_sock, client_message, msglen);
+       printf("[%lu] RX = (%d): ", count, bytes_read);
+       for(int i=0; i<bytes_read; i++) {
+          printf("%c", client_message[i]);
+       }
+
+       // TX
+       int bytes_written = write(client_sock, "Server here!", 12); 
+       printf("\t\nTX = %d\n", bytes_written);
+    }
+    return 0;
+}
--- a/tests/api_test/test.sh
+++ b/tests/api_test/test.sh
@@ -0,0 +1,10 @@
+#!/bin/bash
+
+echo "Performing unit tests..."
+
+chmod 755 build/tests/servers.sh
+chmod 755 build/tests/clients.sh
+
+./build/tests/servers.sh $1 $2 &
+sleep 3
+./build/tests/clients.sh $1 $2 &
--- a/tests/api_test/udp_client.c
+++ b/tests/api_test/udp_client.c
@@ -0,0 +1,92 @@
+/* 
+ * udpclient.c - A simple UDP client
+ * usage: udpclient <host> <port>
+ */
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include <netdb.h> 
+#include <fcntl.h>
+
+#define BUFSIZE 1024
+
+/* 
+ * error - wrapper for perror
+ */
+void error(char *msg) {
+    perror(msg);
+    exit(0);
+}
+
+int main(int argc, char **argv) {
+    int sockfd, portno, n;
+    int serverlen;
+    struct sockaddr_in serveraddr;
+    struct hostent *server;
+    char *hostname;
+    char buf[BUFSIZE];
+
+    /* check command line arguments */
+    if (argc != 3) {
+       fprintf(stderr,"usage: %s <hostname> <port>\n", argv[0]);
+       exit(0);
+    }
+    hostname = argv[1];
+    portno = atoi(argv[2]);
+
+    /* socket: create the socket */
+    sockfd = socket(AF_INET, SOCK_DGRAM, 0);
+    if (sockfd < 0) 
+        error("ERROR opening socket");
+
+    /* gethostbyname: get the server's DNS entry */
+    server = gethostbyname(hostname);
+    if (server == NULL) {
+        fprintf(stderr,"ERROR, no such host as %s\n", hostname);
+        exit(0);
+    }
+
+    /* build the server's Internet address */
+    bzero((char *) &serveraddr, sizeof(serveraddr));
+    serveraddr.sin_family = AF_INET;
+    bcopy((char *)server->h_addr, 
+      (char *)&serveraddr.sin_addr.s_addr, server->h_length);
+    serveraddr.sin_port = htons(portno);
+
+    /* get a message from the user */
+    char *msg = "A message to the server!\0";
+    fcntl(sockfd, F_SETFL, O_NONBLOCK); 
+    long count = 0;
+    while(1)
+    {
+        count++;
+        printf("\nTX(%lu)...\n", count);
+        usleep(10000);
+        //bzero(buf, BUFSIZE);
+        //printf("\nPlease enter msg: ");
+        //fgets(buf, BUFSIZE, stdin);
+
+        /* send the message to the server */
+        serverlen = sizeof(serveraddr);
+        printf("A\n");
+        n = sendto(sockfd, msg, strlen(msg), 0, (struct sockaddr *)&serveraddr, serverlen);
+        printf("B\n");
+        //if (n < 0) 
+        //    error("ERROR in sendto");
+        
+        /* print the server's reply */
+        printf("C\n");
+        memset(buf, 0, sizeof(buf));
+        printf("D\n");
+        n = recvfrom(sockfd, buf, BUFSIZE, 0, (struct sockaddr *)&serveraddr, (socklen_t *)&serverlen);
+        printf("E\n");
+        //if (n < 0) 
+        //    printf("ERROR in recvfrom: %d", n);
+        printf("Echo from server: %s", buf);
+    }
+    return 0;
+}
--- a/tests/api_test/udp_server.c
+++ b/tests/api_test/udp_server.c
@@ -0,0 +1,83 @@
+// UDP Server test program
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <netinet/in.h> 
+#include <arpa/inet.h>
+#include <netdb.h>
+
+#define MAXBUF 1024*1024
+
+void echo( int sd ) {
+  char bufin[MAXBUF];
+  struct sockaddr_in remote;
+  int n;
+  socklen_t len = sizeof(remote);
+  long count = 0;
+
+  while (1) {
+    usleep(50);
+    count++;
+    // read a datagram from the socket (put result in bufin)
+    n=recvfrom(sd,bufin,MAXBUF,0,(struct sockaddr *)&remote,&len);
+    // print out the address of the sender
+    printf("DGRAM from %s:%d\n", inet_ntoa(remote.sin_addr), ntohs(remote.sin_port));
+
+    if (n<0) {
+      perror("Error receiving data");
+    } else {
+      printf("GOT %d BYTES (count = %ld)\n", n, count);
+      // Got something, just send it back
+      // sendto(sd,bufin,n,0,(struct sockaddr *)&remote,len);
+    }
+  }
+}
+
+int main(int argc, char *argv[]) {
+
+  if(argc < 2) {
+    printf("usage: udp_server <port>\n");
+    return 0;
+  }
+  int ld, port = atoi(argv[1]);
+  socklen_t len;
+  struct sockaddr_in skaddr;
+  struct sockaddr_in skaddr2;
+
+  // Create socket
+  if ((ld = socket( PF_INET, SOCK_DGRAM, 0)) < 0) {
+    printf("error creating socket\n");
+    return 0;
+  }
+  // Create address
+  skaddr.sin_family = AF_INET;
+  skaddr.sin_addr.s_addr = htonl(INADDR_ANY);
+  skaddr.sin_port = htons(port);
+  // Bind to address
+  if (bind(ld, (struct sockaddr *) &skaddr, sizeof(skaddr))<0) {
+    printf("error binding\n");
+    return 0;
+  }
+  // find out what port we were assigned
+  len = sizeof( skaddr2 );
+  if (getsockname(ld, (struct sockaddr *) &skaddr2, &len)<0) {
+    printf("error getsockname\n");
+    return 0;
+  }
+  // Display address:port to verify it was sent over RPC correctly
+  port = ntohs(skaddr2.sin_port);
+  int ip = skaddr2.sin_addr.s_addr;
+  unsigned char d[4];
+  d[0] = ip & 0xFF;
+  d[1] = (ip >>  8) & 0xFF;
+  d[2] = (ip >> 16) & 0xFF;
+  d[3] = (ip >> 24) & 0xFF;
+  printf("bound to address: %d.%d.%d.%d : %d\n", d[0],d[1],d[2],d[3], port);
+          
+  // RX
+  echo(ld);
+  return(0);
+}