基于TCP协议select多路复用-1

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

#include <unistd.h>
#include <string.h>
#include <time.h>
#include <sys/types.h>
#include <sys/select.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#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 <ip> <port>\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;
}
 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值