test/select/test07_cli.c

   1 /*
   2  * Test name: test07_cli.c
   3  *
   4  * Objective: Test an impatient TCP client with timeout which waits input both
   5  * from the terminal and from the network connection.
   6  *
   7  * Description: Implements a echo client using the TCP protocol with a timeout
   8  * value. It waites for data on both the terminal and the network connection and
   9  * prints an angry message asking for more data if there is a timeout.
  10  *
  11  */
  12
  13 #include <sys/types.h>
  14 #include <sys/ioctl.h>
  15 #include <sys/select.h>
  16 #include <sys/asynchio.h>
  17 #include <stdio.h>
  18 #include <string.h>
  19 #include <stdlib.h>
  20 #include <stdarg.h>
  21 #include <fcntl.h>
  22 #include <errno.h>
  23 #include <unistd.h>
  24 #include <net/netlib.h>
  25 #include <net/gen/netdb.h>
  26 #include <net/gen/in.h>
  27 #include <net/gen/tcp.h>
  28 #include <net/gen/tcp_io.h>
  29 #include <net/hton.h>
  30
  31 #define PORT 6060L
  32
  33 int tcp_connect(char *host, long port)
  34 {
  35   /* creates a tcp connection with specified host and port */
  36   char *tcp_device;
  37   struct hostent *hp;
  38   int netfd;
  39   nwio_tcpcl_t tcpcopt;
  40   nwio_tcpconf_t tcpconf;
  41   ipaddr_t dirhost;
  42   int tries;
  43   int result;
  44
  45   /* get host address */
  46   if ((hp = gethostbyname(host)) == (struct hostent*) NULL)
  47   {
  48     fprintf(stderr,"Unknown host\n");
  49     return(-1);
  50   }
  51   memcpy((char *)&dirhost, (char *)hp->h_addr, hp->h_length);
  52
  53   /* Get default TCP device */
  54   if (( tcp_device = getenv("TCP_DEVICE") ) == NULL)
  55     tcp_device = TCP_DEVICE;
  56
  57   /* Establish TCP connection */
  58   if ((netfd = open(tcp_device, O_RDWR)) < 0)
  59   {
  60     fprintf(stderr,"Error opening TCP device\n");
  61     return -1;
  62   }
  63
  64   /* Configure TCP connection */
  65   tcpconf.nwtc_flags=NWTC_LP_SEL | NWTC_SET_RA | NWTC_SET_RP;
  66   tcpconf.nwtc_remaddr = dirhost;
  67   tcpconf.nwtc_remport = (tcpport_t) htons(port);
  68
  69   if ((result = ioctl(netfd, NWIOSTCPCONF, &tcpconf) ) <0)
  70   {
  71     fprintf(stderr, "Error establishing communication\n");
  72     printf("Error:  %d\n",result);
  73     close(netfd);
  74     return -1;
  75   }
  76
  77   /* Get configuration for TCP comm */
  78   if ((result = ioctl(netfd, NWIOGTCPCONF, &tcpconf) ) < 0)
  79   {
  80     fprintf(stderr,"Error getting configuration\n");
  81     printf("Error: %d\n", result);
  82     close(netfd);
  83     return -1;
  84   }
  85
  86   /* Establish connection */
  87   tcpcopt.nwtcl_flags = 0;
  88   tries = 0;
  89   while (tries < 10) {
  90         if ( (result = ioctl(netfd, NWIOTCPCONN, &tcpcopt)) < 0 ) {
  91                 if (errno != EAGAIN)
  92                 {
  93                         fprintf(stderr, "Server is not listening\n");
  94                         close(netfd);
  95                         return(-1);
  96                 }
  97                 fprintf(stderr, "Unable to connect\n");
  98                 sleep(1);
  99                 tries++;
 100         }
 101         else
 102                 break;  /* Connection */
 103  }
 104  /* Check result value */
 105  if (result < 0) {
 106         fprintf(stderr, "Error connecting\n");
 107         fprintf(stderr, "Error: %d\n", result);
 108         printf("Number of tries: %d\n", tries);
 109         printf("Error: %d\n", errno);
 110         close(netfd);
 111         return -1;
 112  }
 113  return netfd;
 114 }
 115
 116 int main(int argc,char *argv[]) {
 117   int fd;
 118   ssize_t data_read;
 119   char send_buf[1024];
 120   char recv_buf[1024];
 121   fd_set fds_read;
 122   int ret;
 123   struct timeval timeout;
 124
 125   /* Check parameters */
 126   if (argc !=2) {
 127     fprintf(stderr,"Usage: %s host\n", argv[0]);
 128     exit(-1);
 129   }
 130
 131   if ((fd = tcp_connect(argv[1], PORT) ) < 0)
 132     exit(-1);
 133   printf("Connected to server\n");
 134
 135   while (1)
 136   {
 137
 138     /* init fd_set */
 139     FD_ZERO(&fds_read);
 140     FD_SET(0, &fds_read); /* stdin */
 141     FD_SET(fd, &fds_read);
 142     /* set timeout */
 143     timeout.tv_sec = 3;
 144     timeout.tv_usec = 0;
 145
 146     printf("Send data: ");
 147     fflush(stdout);
 148     /* Wait until it is possible to read with select */
 149     ret = select(4, &fds_read, NULL, NULL, &timeout);
 150     if (ret < 0) {
 151         fprintf(stderr, "Error on select waiting for write: %d\n", errno);
 152         exit(-1);
 153     }
 154     /* timeout */
 155     if (ret == 0) {
 156         printf("\nClient says: Hey! I want to send some data!!\n");
 157         continue;
 158     }
 159     /* handle data from network */
 160     if (FD_ISSET(fd, &fds_read)) {
 161         data_read = read(fd, &recv_buf, 1024);
 162         printf("Server says: %s\n\n", recv_buf);
 163     }
 164     /* handle data from terminal */
 165     if (FD_ISSET(0, &fds_read)) {
 166         /* Get a string and send it */
 167         gets(send_buf);
 168         write(fd, &send_buf, strlen(send_buf)+1);
 169
 170         /* If data sent is exit then break */
 171         if (!strcmp(send_buf,"exit"))
 172                 break;
 173
 174         /* Get server response */
 175         data_read = read(fd, &recv_buf, 1024);
 176         printf("Received: %s\n\n", recv_buf);
 177     }
 178   }
 179
 180   /* Close TCP communication */
 181   close(fd);
 182 }