src/network/test_socket.c

   1 #include <stdio.h>
   2 #include <unistd.h>
   3
   4 #include "socket.h"
   5 #include "socket_io.h"
   6
   7 void usage(const char *appname)
   8 {
   9         fprintf(stderr, "Usage:\n%s -h <host> -p <port>\n", appname);
  10 }
  11
  12 int main(int argc, char **argv)
  13 {
  14         int opt;
  15         int rc;
  16
  17         int sockfd;
  18
  19         char *hostname = NULL;
  20         char *port = NULL;
  21
  22         while ((opt = getopt(argc, argv, "h:p:")) != -1) {
  23                 switch (opt) {
  24                 case 'h' :
  25                         hostname = optarg;
  26                         break;
  27                 case 'p':
  28                         port = optarg;
  29                         break;
  30                 default:
  31                         fprintf(stderr, "Unknown parameter.\n");
  32                         usage(argv[0]);
  33                         return -1;
  34                 }
  35         }
  36
  37         if (!hostname || !port) {
  38                 usage(argv[0]);
  39                 return -1;
  40         }
  41
  42         fprintf(stderr, "Connecting to %s port %s...\n", hostname, port);
  43         sockfd = connect_to_tcp_server(hostname, port);
  44         if (sockfd < 0)
  45                 return -1;
  46
  47         fprintf(stderr, "Connected!\n");
  48
  49         rc = socket_close(sockfd);
  50         return rc;
  51 }