TCP多路复用示例代码

接口声明：int select(int nfds, fd_set *readfds, fd_set *writefds,

                  fd_set *exceptfds, struct timeval *timeout);

参数：

nfds：所有正在监测的套接字的最大值加1

readfds：读就绪文件描述符集合

writefds：写就绪文件描述符集合

exceptfds：异常就绪文件描述符集合

timeout：超时控制

返回值：

成功：就绪文件描述符总数（当超时返回时为0）

失败：-1

 

备注：

文件描述符集合操作函数：

void FD_CLR(int fd, fd_set *set);

int  FD_ISSET(int fd, fd_set *set); //判断文件描述符状态

void FD_SET(int fd, fd_set *set);//加载文件描述符

void FD_ZERO(fd_set *set);//清空文件描述符

/* 多路复用服务端 */
#include <stdio.h>
#include <sys/types.h>          /* See NOTES */
#include <sys/socket.h>
#include <sys/select.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <pthread.h>

int network_init(unsigned short port);

int main(int argc,const char **argv)
{
    int sockfd = network_init(9000);
    struct sockaddr_in useraddr = {};
    socklen_t len = sizeof(useraddr);    
    printf("等待客户连接>>>>>>>>\n");

    int connfd = accept(sockfd,(struct sockaddr*)&useraddr,&len);
    if(connfd == -1)
    {
        perror("accept failed!\n");
        return -1;
    }
    printf("user_IP:%s\n",inet_ntoa(useraddr.sin_addr));

    int maxfd = connfd;
    fd_set readfds;
    while(1)
    {
        FD_ZERO(&readfds);//清空文件描述符集合
        FD_SET(0,&readfds);//加载输入文件描述符
        FD_SET(connfd,&readfds);//加载连接文件描述符

        int ret = select(maxfd+1,&readfds,NULL,NULL,NULL);
        if(FD_ISSET(0,&readfds))
        {
            char buf[128] = {0};
            fgets(buf,128,stdin);
            int len = write(connfd,buf,sizeof(buf));
            if(len < 0)
            {
                perror("write failed!\n");
                break;
            }
        }
        else if(FD_ISSET(connfd,&readfds))
        {
            char readbuf[1024] = {0};
            int len = read(connfd,readbuf,sizeof(readbuf));
            printf("接收到数据:%s\n",readbuf);
            if(len < 0)
            {
                perror("read failed!\n");
                break;
            }
        }
    }
    return 0;
}

int network_init(unsigned short port)
{
    int fd = socket(AF_INET,SOCK_STREAM,0);
    if(fd == -1)
    {
        perror("socket failed!\n");
        return -1;
    }
    struct sockaddr_in server_addr;
    server_addr.sin_addr.s_addr = htonl(INADDR_ANY);
    server_addr.sin_family = AF_INET;
    server_addr.sin_port = htons(port);

    if(bind(fd,(struct sockaddr*)&server_addr,sizeof(server_addr)) == -1)
    {
        perror("bind failed!\n");
        return -1;
    }
    int flag = listen(fd,5);
    if(flag == -1)
    {
        return -1;
    }
    return fd;
}

/* 多路复用客户端 */
#include <stdio.h>
#include <sys/socket.h>
#include <sys/types.h>
#include <sys/select.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>
#include <unistd.h>
#include <stdlib.h>
#include <pthread.h>

int main(int argc, char **argv)
{
	if(argc < 3)
	{
		printf("use:client ip port");
		return -1;
	}
	//1.创建套接字
	int sockfd =  socket(AF_INET, SOCK_STREAM, 0);
	//2.连接服务器
	struct sockaddr_in seraddr;
	memset(&seraddr, 0, sizeof(seraddr));
	seraddr.sin_family = AF_INET;
	seraddr.sin_port = htons( atoi(argv[2]));//端口号
	seraddr.sin_addr.s_addr = inet_addr(argv[1]);
	
	int ret = connect(sockfd, (struct sockaddr*)&seraddr, sizeof(seraddr));
	if(ret < 0)
	{
		perror("connect error");
		return -1;
	}
	int maxfd = sockfd;
	fd_set readfds;
	while(1)
	{
		FD_ZERO(&readfds);
		FD_SET(0,&readfds); //加载描述符到集合
		FD_SET(sockfd,&readfds);
		int ret = select(maxfd+1,&readfds,NULL,NULL,NULL);
		if(FD_ISSET(0,&readfds))
		{
			char buffer[128]={0};
			scanf("%s", buffer);
			ssize_t size = send(sockfd, buffer, strlen(buffer), 0);
			if(size < 0)
			{
				perror("send error");
				break;
			}
		}
		else if(FD_ISSET(sockfd,&readfds))
		{
			char recvbuffer[128]={0};
			memset(recvbuffer, 0, 128);
			int size = read(sockfd, recvbuffer, 128);
			if(size <= 0)
			{
				perror("服务器中断\n");
				break;
			}
			printf("%s\n", recvbuffer);
		}
	}
	//4.关闭套接字
	close(sockfd);
	return 0;
}