IO多路复用

搭建select的TCP客户端

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include<head.h>
#define PORT 6666
#define IP "192.168.124.81"
int main(int argc, const char *argv[])
{
	//创建流试套接字
	int cfd = socket(AF_INET,SOCK_STREAM,0);
	if(cfd<0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("%d\n",cfd);


	//绑定服务器的IP和端口--->非必须绑定
	//填充服务器的地址信息结构体，真实的地址信息结构体根据地质族制定
	//AF_INET
	//要连接哪个服务器，就填对应服务器的IP和端口
	struct sockaddr_in sin;
	sin.sin_family=AF_INET;
	sin.sin_port=htons(PORT);
	sin.sin_addr.s_addr=inet_addr(IP);
	//接连服务器
	if(connect(cfd,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("connect");
		return -1;
	}
	printf("connect uccess\n");

	fd_set readfds,tempfds;
	FD_ZERO(&readfds);
	FD_ZERO(&tempfds);

	FD_SET(0,&readfds);
	FD_SET(cfd,&readfds);


	char buf[128]="";
	ssize_t res=0;
	ssize_t s_res;
	while(1)
	{
		tempfds=readfds;
		s_res=select(cfd+1,&tempfds,NULL,NULL,NULL);
		if(s_res<0)
		{
			ERR_MSG("select");
			return -1;
		}
		else if(0==s_res)
		{
			printf("time out...\n");
			break;
		}

		//发送数据
		if(FD_ISSET(0,&tempfds))
		{		bzero(buf,sizeof(buf));
			printf("请输入--->");
			fgets(buf,sizeof(buf),stdin);
			buf[strlen(buf)-1]=0;
			if(send(cfd,buf,sizeof(buf),0)<0)
			{
				ERR_MSG("send");
				return -1;
			}
			printf("send success\n");
		}
		if(FD_ISSET(cfd,&readfds))
		{
			//接受数据
			res = recv(cfd,buf,sizeof(buf),0);
			if(res<0)
			{
				ERR_MSG("recv");
				return -1;
			}
			else if(0==res)
			{	printf("[%s : %d]cfd=%d 	服务器掉线\n",IP,PORT,cfd);

				break;
			}	
			printf("[%s : %d]cfd=%d :%s\n",IP,PORT,cfd,buf);
		}
	}
	close(cfd);

	return 0;
}

将select的TCP服务器重新搭建

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include<head.h>
#define PORT 6666
#define IP "192.168.124.81"
int main(int argc, const char *argv[])
{
	//创建流试套接字
	int sfd = socket(AF_INET,SOCK_STREAM,0);
	if(sfd<0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("%d\n",sfd);
	//设置允许端口被快速复用
	int reuse = 1;
    if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)
    {
        ERR_MSG("setsockopt");
        return -1;
    }
    printf("允许端口快速重用成功\n");

	//绑定服务器的IP和端口
	struct sockaddr_in sin;
	sin.sin_family=AF_INET;
	sin.sin_port=htons(PORT);
	sin.sin_addr.s_addr=inet_addr(IP);
	if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("bind success\n");
	//将套接字设置为被动监听状态
	if(listen(sfd,128)<0)
	{
		ERR_MSG("listen");
		return -1;
	}
	printf("listen success\n");
	//创建一个读集合
	fd_set readfds,tempfds;
	//初始化成无意义的文件描述符
	FD_ZERO(&readfds);
	FD_ZERO(&tempfds);
	//将需要的文件描述符添加到读集合中
	FD_SET(0,&readfds);
	FD_SET(sfd,&readfds);
	//从已完成连接的队列中获取一个客户端消息，生成一个新的人间
	//该文件描述符才是与客户端通信的文件描述符
	struct sockaddr_in cin;	
	int maxfd = sfd;
	char buf[128]="";
	ssize_t res=0;
	int s_res=-1;
	int newfd=-1;
	struct sockaddr_in saveCin[1024-3];
	socklen_t addrlen = sizeof(cin);
	while(1)
	{
		tempfds=readfds;
		//调用IO多路复用函数
		s_res=select(maxfd+1,&tempfds,NULL,NULL,NULL);
		if(s_res<0)
		{
			ERR_MSG("select");
			return -1;
		}
		else if(0==s_res)
		{
			printf("time out....\n");
			break;
		}

		for(int i=0;i<=maxfd;i++)
		{
			if(FD_ISSET(i,&tempfds)==0)
				continue;
			if(0==i)
			{
				printf("触发键盘输入事件\n");
				fgets(buf,sizeof(buf),stdin);
				buf[strlen(buf)-1]=0;
				printf(":%s\n",buf);

			}
			else if(sfd==i)
			{
				printf("触发客户端连接事件\n");
				newfd=accept(sfd,(struct sockaddr*)&cin,&addrlen);
				if(newfd<0)
				{
					ERR_MSG("accept");
					return -1;
				}
				printf("[%s : %d]newfd=%d  客户端连接成功\n",\
						inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);

				saveCin[newfd-3]=cin;
				//将newfd添加到集合中
				FD_SET(newfd,&readfds);
				//更新maxfd
				maxfd=maxfd>newfd?maxfd:newfd;

			}
			else
			{
				printf("触发客户端交互事件\n");
				bzero(buf,sizeof(buf));
				//接受数据
				res = recv(i,buf,sizeof(buf),0);
				if(res<0)
				{
					ERR_MSG("recv");
					return -1;
				}
				else if(0==res)
				{	printf("[%s : %d]newfd=%d  客户端下线\n",\
						inet_ntoa(saveCin[i-3].sin_addr),ntohs(saveCin[i-3].sin_port),i);
				//关闭下线客户端的文件描述符
				close(i);
				//将下线客户端的文件描述符从集合中删除
				FD_CLR(i,&readfds);
				//更新mxfd
				//从大往小判断
				

				while(FD_ISSET(maxfd,&readfds)==0&&maxfd-->=0);
/*			
				int j = maxfd;
				for(;j>0;j--)
				{
					if(FD_ISSET(j,&readfds))
						break;
				}
				maxfd=j;
*/

				break;
				}	
				printf("[%s : %d]newfd=%d :%s\n ",\
						inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),i,buf);	
				//发送数据
				strcat(buf,"*_*");
				if(send(newfd,buf,sizeof(buf),0)<0)
				{
					ERR_MSG("send");
					return -1;
				}
				printf("send success\n");

			}
		}
	}

	close(sfd);

	return 0;
}

tftp项目上传代码

int do_upload(int sfd,struct sockaddr_in sin)
{
	char buf[516]="";
	char c[20]="";
	printf("请输入要上传的文件名:");
	scanf("%s",c);
	while(getchar()!=10);	
	int fd=open(c,O_RDONLY);
	if(fd<0)
	{
		ERR_MSG("open");
		printf("__%d__\n",__LINE__);
		return -1;
	}

	unsigned short *p1=(unsigned short*)buf;
	*p1=htons(2);
	char* p2=buf+2;
	strcpy(p2,c);
	char* p3=p2+strlen(p2)+1;
	strcpy(p3,"octet");
	int len=2+strlen(c)+1+strlen("octet")+1;
	printf("%d\n",len);
	if(sendto(sfd,buf,len,0,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("sendto");
		printf("__%d__\n",__LINE__);
		return -1;
	}
	socklen_t addrlen = sizeof(sin);
	ssize_t res = 0;
	unsigned short num =0;
	while(1)
	{
		bzero(buf,sizeof(buf));
		res=recvfrom(sfd,buf,sizeof(buf),0,(struct sockaddr*)&sin,&addrlen);	
		if(res<0)
		{
			ERR_MSG("recvfrom");
			printf("%d\n",buf[1]);
			break;
		}
		if(4==buf[1])
		{
			if(*(unsigned short*)(buf+2)==htons((num)))
			{
				num++;
				*(unsigned short*)buf=htons(3);
				*(unsigned short*)(buf+2)=htons(num);

				res= read(fd,buf+4,512);
				if(res<0)
				{
					ERR_MSG("read");
					break;
				}
				else if(0==res)
				{
					printf("文件上传完毕\n");
					break;
				}	
				if(sendto(sfd,buf,res+4,0,(struct sockaddr*)&sin,sizeof(sin))<0)
				{
					ERR_MSG("sendto");
					printf("__%d__\n",__LINE__);
					return -1;
				}

			}
		}
		else if(5==buf[1])
		{
			printf("错误: %d %s\n",ntohs(*(unsigned short*)(buf+2)),buf+4);
			close(fd);
			return -1;
		}

	}
	close(sfd);
	return 0;

搭建poll客户端

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include<head.h>
#define PORT 6666
#define IP "192.168.124.81"
int main(int argc, const char *argv[])
{
	//创建流试套接字
	int cfd = socket(AF_INET,SOCK_STREAM,0);
	if(cfd<0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("%d\n",cfd);


	//绑定服务器的IP和端口--->非必须绑定
	//填充服务器的地址信息结构体，真实的地址信息结构体根据地质族制定
	//AF_INET
	//要连接哪个服务器，就填对应服务器的IP和端口
	struct sockaddr_in sin;
	sin.sin_family=AF_INET;
	sin.sin_port=htons(PORT);
	sin.sin_addr.s_addr=inet_addr(IP);
	//接连服务器
	if(connect(cfd,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("connect");
		return -1;
	}
	printf("connect uccess\n");

	struct pollfd fds[2];
	fds[0].fd=0;
	fds[0].events=POLLIN;
	fds[1].fd=cfd;
	fds[1].events=POLLIN;
	char buf[128]="";
	ssize_t res=0;
	int p_res=-1;
	while(1)
	{
		p_res = poll(fds,2,-1);
		if(p_res<0)
		{
			ERR_MSG("poll");
			return -1;
		}
		else if(0==p_res)
		{
			printf("time out...\n");
			break;
		}
		if((fds[0].revents&POLLIN)!=0)
		{
			printf("触发键盘输入事件\n");
			bzero(buf,sizeof(buf));
			printf("请输入--->");
			fgets(buf,sizeof(buf),stdin);
			buf[strlen(buf)-1]=0;
			//发送数据

			if(send(cfd,buf,sizeof(buf),0)<0)
			{
				ERR_MSG("send");
				return -1;
			}
			printf("send success\n");
		}
		//接受数据
		if(fds[1].revents&POLLIN)
		{
			printf("触发服务器交互事件\n");
			bzero(buf,sizeof(buf));
			res = recv(cfd,buf,sizeof(buf),0);
			if(res<0)
			{
				ERR_MSG("recv");
				return -1;
			}
			else if(0==res)
			{	printf("[%s : %d]cfd=%d 	服务器掉线\n",IP,PORT,cfd);

				break;
			}	
			printf("[%s : %d]cfd=%d :%s\n",IP,PORT,cfd,buf);
		}

	}
	close(cfd);

	return 0;
}