[Anet-checkins] CVS: ANet/ANet_Daemon/Linux/IP_Sample syncserver.c,NONE,1.1 tcpclient.c,NONE,1.1 udpclient.c,NONE,1.1

[Benoit N. <be...@us...>]

Update of /cvsroot/anet/ANet/ANet_Daemon/Linux/IP_Sample
In directory usw-pr-cvs1:/tmp/cvs-serv27272

Added Files:
	syncserver.c tcpclient.c udpclient.c 
Log Message:
""

--- NEW FILE: syncserver.c ---
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <netdb.h>
#include <stdio.h>
#include <unistd.h> /* close */


#define SUCCESS 0
#define ERROR   1

#define END_LINE 0x0A
#define SERVER_PORT 1500
#define MAX_MSG 100

/* function readline */
int read_line();

int main (int argc, char *argv[]) {

  int n, sd, newSd, cliLen;

  struct sockaddr_in cliAddr, servAddr;
  char line[MAX_MSG];

  if(argc < 2) {
    printf("usage: %s <TCP | UDP>\n",argv[0]);
    exit(1);
  }
  /* create socket */
  if (strcmp(argv[1],"TCP") == 0) {
     sd = socket(AF_INET, SOCK_STREAM, 0);
     if(sd<0) {
       perror("cannot open socket ");
       return ERROR;
     }
  } else if (strcmp(argv[1],"UDP") == 0) {
    sd = socket(AF_INET, SOCK_DGRAM, 0);
     if(sd<0) {
       perror("cannot open socket ");
       return ERROR;
     }
  }

  /* bind server port */
  servAddr.sin_family = AF_INET;
  servAddr.sin_addr.s_addr = htonl(INADDR_ANY);
  servAddr.sin_port = htons(SERVER_PORT);
  
  if(bind(sd, (struct sockaddr *) &servAddr, sizeof(servAddr))<0) {
    perror("cannot bind port ");
    return ERROR;
  }
  if (strcmp(argv[1],"TCP") == 0) {
    listen(sd,5);
  }

  while(1) {

    /* init line */
    memset(line,0x0,MAX_MSG);
    cliLen = sizeof(cliAddr);

    if (strcmp(argv[1],"UDP") == 0) {
       
        printf("%s: waiting for data on port UDP %u\n",argv[0],SERVER_PORT);
       n = recvfrom(sd,line,MAX_MSG,0,(struct sockaddr *) &cliAddr,&cliLen);
       if (n < 0) {
          printf("%s : cannot receive data\n",argv[1]);
          continue;
       }
       printf("%s: from %s : UDP %u :%s \n",
               argv[0],inet_ntoa(cliAddr.sin_addr),ntohs(cliAddr.sin_port),
               line);
    } else {

      printf("%s: waiting for data on port TCP %u\n",argv[0],SERVER_PORT);

      newSd = accept(sd, (struct sockaddr *) &cliAddr, &cliLen);
      if(newSd<0) {
        perror("cannot accept connection ");
        return ERROR;
      }

      /* receive segments */
      while(read_line(newSd,line)!=ERROR) {
      
        printf("%s: received from %s:TCP%d : %s\n", argv[0],
	     inet_ntoa(cliAddr.sin_addr),
	     ntohs(cliAddr.sin_port), line);

      } /* while(read_line) */
   } /* if else */

  } /* while (1) */

}

int read_line(int newSd, char *line_to_return) {

  static int rcv_ptr=0;
  static char rcv_msg[MAX_MSG];
  static int n;
  int offset;

  offset=0;

  while(1) {
    if(rcv_ptr==0) {
      /* read data from socket */
      memset(rcv_msg,0x0,MAX_MSG); /* init buffer */
      n = recv(newSd, rcv_msg, MAX_MSG, 0); /* wait for data */
      if (n<0) {
	perror(" cannot receive data ");
	return ERROR;
      } else if (n==0) {
	printf(" connection closed by client\n");
	close(newSd);
	return ERROR;
      }
    }

    /* copy line into 'line_to_return' */
    while(*(rcv_msg+rcv_ptr)!=END_LINE && rcv_ptr<n) {
      memcpy(line_to_return+offset,rcv_msg+rcv_ptr,1);
      offset++;
      rcv_ptr++;
    }
    
    /* end of line + end of buffer => return line */
    if(rcv_ptr==n-1) { 
      /* set last byte to END_LINE */
      *(line_to_return+offset)=END_LINE;
      rcv_ptr=0;
      return ++offset;
    }
    
    /* end of line but still some data in buffer => return line */
    if(rcv_ptr <n-1) {
      /* set last byte to END_LINE */
      *(line_to_return+offset)=END_LINE;
      rcv_ptr++;
      return ++offset;
    }
    printf("Recived Message -> %s\n",rcv_msg);

    /* end of buffer but line is not ended => */
    /*  wait for more data to arrive on socket */
    if(rcv_ptr == n) {
      rcv_ptr = 0;
    }

  } /* while */
}

--- NEW FILE: tcpclient.c ---
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <netdb.h>
#include <stdio.h>
#include <unistd.h> /* close */

#define SERVER_PORT 1600
#define MAX_MSG 512

int main (int argc, char *argv[]) {

  int sd, rc, i;
  struct sockaddr_in localAddr, servAddr;
  struct hostent *h;

  if(argc < 3) {
    printf("usage: %s <server> <data1> <data2> ... <dataN>\n",argv[0]);
    exit(1);
  }

  h = gethostbyname(argv[1]);
  if(h==NULL) {
    printf("%s: unknown host '%s'\n",argv[0],argv[1]);
    exit(1);
  }

  servAddr.sin_family = h->h_addrtype;
  memcpy((char *) &servAddr.sin_addr.s_addr, h->h_addr_list[0], h->h_length);
  servAddr.sin_port = htons(SERVER_PORT);

  /* create socket */
  sd = socket(AF_INET, SOCK_STREAM, 0);
  if(sd<0) {
    perror("cannot open socket ");
    exit(1);
  }

  /* bind any port number */
  localAddr.sin_family = AF_INET;
  localAddr.sin_addr.s_addr = htonl(INADDR_ANY);
  localAddr.sin_port = htons(0);

  rc = bind(sd, (struct sockaddr *) &localAddr, sizeof(localAddr));
  if(rc<0) {
    printf("%s: cannot bind port TCP %u\n",argv[0],SERVER_PORT);
    perror("error ");
    exit(1);
  }
				
  /* connect to server */
  rc = connect(sd, (struct sockaddr *) &servAddr, sizeof(servAddr));
  if(rc<0) {
    perror("cannot connect ");
    exit(1);
  }

  for(i=2;i<argc;i++) {

    rc = send(sd, argv[i], strlen(argv[i]) + 1, 0);

    if(rc<0) {
      perror("cannot send data ");
      close(sd);
      exit(1);
    
    }

    printf("%s: data%u sent (%s)\n",argv[0],i-1,argv[i]);


  }

return 0;

}


--- NEW FILE: udpclient.c ---
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <netdb.h>
#include <stdio.h>
#include <unistd.h>
#include <string.h> /* memset() */
#include <sys/time.h> /* select() */

#define REMOTE_SERVER_PORT 1600
#define MAX_MSG	512


int main(int argc, char *argv[]) {

  int sd, rc, i;
  struct sockaddr_in cliAddr, remoteServAddr;
  struct hostent *h;

  /* check command line args */
  if(argc<3) {
    printf("usage : %s <server> <data1> ... <dataN> \n", argv[0]);
    exit(1);
  }

  /* get server IP address (no check if input is IP address or DNS name */
  h = gethostbyname(argv[1]);
  if(h==NULL) {
    printf("%s: unknown host '%s' \n", argv[0], argv[1]);
    exit(1);
  }

  printf("%s: sending data to '%s' (IP : %s) \n", argv[0], h->h_name,
	 inet_ntoa(*(struct in_addr *)h->h_addr_list[0]));

  remoteServAddr.sin_family = h->h_addrtype;
  memcpy((char *) &remoteServAddr.sin_addr.s_addr,
	 h->h_addr_list[0], h->h_length);
  remoteServAddr.sin_port = htons(REMOTE_SERVER_PORT);

  /* socket creation */
  sd = socket(AF_INET,SOCK_DGRAM,0);
  if(sd<0) {
    printf("%s: cannot open socket \n",argv[0]);
    exit(1);
  }

  /* bind any port */
  cliAddr.sin_family = AF_INET;
  cliAddr.sin_addr.s_addr = htonl(INADDR_ANY);
  cliAddr.sin_port = htons(0);
  
  rc = bind(sd, (struct sockaddr *) &cliAddr, sizeof(cliAddr));
  if(rc<0) {
    printf("%s: cannot bind port\n", argv[0]);
    exit(1);
  }


  /* send data */
  for(i=2;i<argc;i++) {
    rc = sendto(sd, argv[i], strlen(argv[i])+1, 0, 
		(struct sockaddr *) &remoteServAddr,
		sizeof(remoteServAddr));

    if(rc<0) {
      printf("%s: cannot send data %d \n",argv[0],i-1);
      close(sd);
      exit(1);
    }

  }

  return 1;

}