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

2012年(3)

2011年(43)

2010年(110)

分类: LINUX

2010-10-04 00:16:36

实现功能如下:
同时监听终端的输入和TCP的监听套接字
 
实现代码如下:

#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, n, i;
 char buf[N];
 struct sockaddr_in myaddr, cliaddr;
 socklen_t clilen;
 fd_set rdfs;
 struct timeval mytime;
 if (argc < 3)
 {
  printf("Usage : %s \n", argv[0]);
  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);
 maxfd = listenfd;

 while ( 1 )
 {
  FD_ZERO(&rdfs);
  FD_SET(0, &rdfs);
  FD_SET(listenfd, &rdfs);
  
  mytime.tv_sec = 3;
  mytime.tv_usec = 0;
  if ((n = select(maxfd+1, &rdfs, NULL, NULL, &mytime)) < 0)
  {
   perror("fail to select");
   exit(-1);
  }
  else if (n > 0)
  {
   for (i=0; i<=maxfd; i++)
   {
    if ( FD_ISSET(i, &rdfs) )
    {
     if (STDIN_FILENO == i)
     {
      fgets(buf, N, stdin);
      printf("You input %d characters\n", strlen(buf));
     }
     if (i == listenfd)
     {
      clilen = sizeof(cliaddr);
      if ((connfd = accept(listenfd, (SA *)&cliaddr, &clilen)) < 0)
      {
       perror("fail to accept");
       exit(-1);
      }
      printf("connection from %s:%d\n", inet_ntoa(cliaddr.sin_addr),
                                     ntohs(cliaddr.sin_port));
      close(connfd);
     }
    }
   }  // end for
  }
  else   // timeout
  {
   printf("timeout\n");
  } 
 } // end while
 return 0;
}
 
阅读(1381) | 评论(0) | 转发(2) |
给主人留下些什么吧!~~