11.22 作业 select实现TCP并发客户端/poll实现TCP并发服务器

#include <myhead.h>
#define SERIP "192.168.115.162"
#define SERPORT 8888
#define CLIIP "192.168.115.162"
#define CLIPORT 6666
int main(int argc, const char *argv[])
{
	//创建客户端用于通信的套接字
	int cfd = socket(AF_INET,SOCK_STREAM,0);
	if(cfd == -1){
		perror("socket error");
		return -1;
	}
	printf("cfd = %d\n",cfd);
	//绑定
	struct sockaddr_in cin;
	cin.sin_family = AF_INET;
	cin.sin_port   = htons(CLIPORT);
	cin.sin_addr.s_addr = inet_addr(CLIIP);
	if(bind(cfd,(struct sockaddr*)&cin,sizeof(cin)) == -1){
		perror("bind error");
		return -1;
	}
	printf("bind success\n");

	//连接服务器
	struct sockaddr_in sin;
	sin.sin_family = AF_INET;
	sin.sin_port   = htons(SERPORT);
	sin.sin_addr.s_addr   = inet_addr(SERIP);
	//连接服务器
	if(connect(cfd,(struct sockaddr*)&sin,sizeof(sin)) == -1){
		perror("connect error");
		return -1;
	}
	//收发数据
	char buf[128] = "";
	char rbuf[128] = "";

	///定义一个用于检测的文件描述符的集合
	fd_set readfds,tempfds;
	///清空容器中的内容
	FD_ZERO(&readfds);
	///将要检测的文件描述付放入集合中
	FD_SET(cfd,&readfds);  //将cfd文件描述符放入
	FD_SET(0,&readfds);    //将0号文件描述符放入

	int res = 0;   //接收select的返回值
	while(1){
		tempfds = readfds;
		///使用select阻塞等待集合重点文件描述符有事件产生
		res = select(cfd+1,&tempfds,NULL,NULL,NULL);
		if(res == -1){
			perror("select error");
			return -1;
		}else if(res == 0){
			printf("time out\n");
			return -1;
		}

		bzero(buf,sizeof(buf));
		bzero(rbuf,sizeof(rbuf));

		///判断0是否还在集合中
		if(FD_ISSET(0,&tempfds)){
			printf("请输入>>>");
			fgets(buf,sizeof(buf),stdin); //从终端输入中读取数据
			buf[strlen(buf)-1] = '\0';

			//将数据发送给服务器
			send(cfd,buf,sizeof(buf),0);
			//如果输入是quit退出
			if(strcmp(buf,"quit") == 0){   
				break;
			}
		}
		///判断cfd是否还在集合中
		if(FD_ISSET(cfd,&tempfds)){
			//接收服务器发来的消息
			int res = recv(cfd,rbuf,sizeof(rbuf),0);
			if(res == 0){
				printf("客户端已经关闭\n");
				break;
			}
			printf("rbuf = %s\n",rbuf);
		}
	}
	close(cfd);
	return 0;
}

#include<myhead.h>
#define PORT 8888              //端口号
#define IP "192.168.115.162"       //IP地址

int main(int argc, const char *argv[])
{
    //1、创建用于接受连接的套接字
    int sfd = socket(AF_INET, SOCK_STREAM, 0);
    if(sfd == -1)
    {
        perror("socket error");
        return -1;
    }

    printf("socket success sfd = %d\n", sfd);    //4
    //设置端口号快速重用
    int reuse = 1;
    if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) == -1)
    {
        perror("setsockopt error");
        return -1;
    }
    printf("设置端口快速重用成功 _%d_ %s_ %s_\n", __LINE__, __FILE__, __func__);
    //2、绑定IP地址和端口号
    //2.1、填充要绑定的地址信息结构体
    struct sockaddr_in sin;
    sin.sin_family     = AF_INET;         //表明是ipv4
    sin.sin_port     = htons(PORT);        //端口号
    sin.sin_addr.s_addr = inet_addr(IP);     //IP地址

    //2.2、绑定
    if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin))==-1)
    {
        perror("bind error");
        return -1;
    }
    printf("bind success _%d_ %s_ %s_\n", __LINE__, __FILE__, __func__);

    //3、将套接字设置成被动监听状态
    if(listen(sfd, 128) == -1)
    {
        perror("listen error");
        return -1;
    }

    printf("listen success _%d_ %s_ %s_\n", __LINE__, __FILE__, __func__);

    //4、阻塞等待客户端连接请求，如果有新的客户端连接，则创建一个新的用于通信的套接字
    //4.1、定义客户端地址信息结构体
    struct sockaddr_in cin;             //客户端地址信息结构体
    cin.sin_family     = AF_INET;
    socklen_t socklen = sizeof(cin);          //客户端地址信息的大小

	//定义一个容器
	char buf[128] = "";
	int maxfd = sfd;
	///定义一个集合管理sfd和newfd
	struct pollfd fds[maxfd-2];
	///将sfd文件描述符放入
	fds[0].fd = sfd;
	fds[0].events = POLLIN;  //表明要进行读事件
	
	int rct = 0; //接收poll返回的结果


	while(1){
		rct = poll(fds,maxfd+1,-1);    //第三个参数如果是负数，表明一直等待
		if(rct == -1){
			perror("poll error");
			return -1;
		}else if(rct == 0){
			printf("time out\n");
			return -1;
		}
		if(fds[0].revents == POLLIN){
			//4.2、阻塞接收客户端的链接请求，并且获取客户端的地址信息
			int newfd = accept(sfd, (struct sockaddr*)&cin, &socklen);
			if(newfd == -1)
			{
				perror("accept error");
				return -1;
			}
			printf("accept success _%d_ %s_ %s_\n", __LINE__, __FILE__, __func__);
			fds[newfd].fd = newfd;
			fds[newfd].events = POLLIN;
			if(newfd>maxfd){
				maxfd = newfd;
			}
		}
		for(int i = 1;i<=maxfd;i++){
			if(fds[i].revents == POLLIN){
				//清空字符串
				bzero(buf, sizeof(buf));
				int res = recv(i, buf, sizeof(buf), 0);        //从套接字中读取客户端发来的消息

				//判断收到的结果
				if(res == 0)
				{
					printf("客户端已经下线\n");
					close(i);
					break;
				}else if(res < 0)
				{
					perror("recv error");
					return -1;
				}

				printf("[%s:%d]：%s\n", inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), buf);

				//将读取的信息，加上一些字符发送回去
				strcat(buf, "*_*");
				send(i, buf, sizeof(buf), 0);

			}
		}

	}
	close(sfd);
}