Chinaunix首页 | 论坛 | 博客
  • 博客访问: 862845
  • 博文数量: 156
  • 博客积分: 6553
  • 博客等级: 准将
  • 技术积分: 3965
  • 用 户 组: 普通用户
  • 注册时间: 2010-06-22 18:36
文章存档

2012年(3)

2011年(43)

2010年(110)

分类: LINUX

2010-10-04 00:19:50

实现功能如下:
TCP服务器在处理客服端链接请求的同时,同时对已连接的客户实现echo服务
 
服务器端实现代码:

#include
#include
#include
#include
#include
#include
#include
#include
#include
#define N 64
typedef struct sockaddr SA;
int main(int argc, char *argv[])
{
 int listenfd, connfd, maxfd, i, nbyte;
 char buf[N];
 struct sockaddr_in myaddr;
 fd_set global_rdfs, current_rdfs;
 if (argc < 3)
 {
  exit(-1);
 }
 if ((listenfd = socket(PF_INET, SOCK_STREAM, 0)) < 0)
 {
  perror("fail to socket");
  exit(-1);
 }
 bzero(&myaddr, sizeof(myaddr));
 myaddr.sin_family = PF_INET;  
 myaddr.sin_port = htons(atoi(argv[2]));
 myaddr.sin_addr.s_addr = inet_addr(argv[1]);
 if (bind(listenfd, (SA *)&myaddr, sizeof(myaddr)) < 0)
 {
  perror("fail to bind");
  exit(-1);
 }
 listen(listenfd, 5);
 
 FD_ZERO(&global_rdfs);
 FD_SET(listenfd, &global_rdfs);
 maxfd = listenfd;
 while ( 1 )
 {
  current_rdfs = global_rdfs;
  if (select(maxfd+1, ¤t_rdfs, NULL, NULL, NULL) < 0)
  {
   perror("fail to select");
   exit(-1);
  }
  for (i=0; i<=maxfd; i++)
  {
   if ( FD_ISSET(i, ¤t_rdfs) )
   {
    if (i == listenfd)
    {
     printf("new connection\n");
     if ((connfd = accept(i, NULL, NULL)) < 0)
     {
      perror("fail to accept");
      exit(-1);
     }
     FD_SET(connfd, &global_rdfs);
     maxfd = maxfd > connfd ? maxfd : connfd;
    }
    else
    {
     if ((nbyte = recv(i, buf, N, 0)) == 0)
     {
      printf("peer closed\n");
      FD_CLR(i, &global_rdfs);
     }
     else
     {
      send(i, buf, nbyte, 0);
     }
    }
   }
  }  // end for
 }
 return 0;
}
 
客户端实现代码:

#include
#include
#include
#include
#include
#include
#include
#define N 64
typedef struct sockaddr SA;
int main(int argc, char *argv[])
{
 int sockfd;
 char buf[N];
 struct sockaddr_in servaddr;
 if (argc < 3)
 {
  printf("Usage : %s \n", argv[0]);
  exit(-1);
 }
 if ((sockfd = socket(PF_INET, SOCK_STREAM, 0)) < 0)
 {
  perror("fail to socket");
  exit(-1);
 }
 bzero(&servaddr, sizeof(servaddr));
 servaddr.sin_family = PF_INET;
 servaddr.sin_port = htons(atoi(argv[2]));
 servaddr.sin_addr.s_addr = inet_addr(argv[1]);
 if (connect(sockfd, (SA *)&servaddr, sizeof(servaddr)) < 0)
 {
  perror("fail to connect");
  exit(-1);
 }
 while ( 1 )
 {
  printf("client > ");
  fgets(buf, N, stdin);
  if (strncmp(buf, "quit", 4) == 0)
  {
   close(sockfd);
   exit(0);
  }
  send(sockfd, buf, N, 0);
  bzero(buf, N);
  recv(sockfd, buf, N, 0);
  printf("recv from server : %s", buf);
 }
 return 0;
}
阅读(4301) | 评论(0) | 转发(2) |
给主人留下些什么吧!~~