嵌入式学习-网络编程-Day5

嵌入式学习-网络编程-Day5

一、思维导图

二、作业

1.使用poll实现TCP服务器的并发

include <myhead.h>
//tcp服务器端
int main(int argc, const char *argv[])
{
	//1 创建通信套接字
	int sfd=-1;
	if((sfd=socket(AF_INET,SOCK_STREAM,0))==-1)
	{
		perror("socket error");
		return -1;
	}
	//快速重用端口号
	int reuse = 1;
	if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))==-1)
	{
		perror("setsockopt error");
		return -1;
	}

	//2 绑定IP地址和端口号
	struct sockaddr_in sin;  //地址信息结构体  man 7 ip
	sin.sin_family = AF_INET;//通信域 ipv4
	sin.sin_port = htons(8888);//端口号 转为网络字节序 2字节
	sin.sin_addr.s_addr = inet_addr("192.168.122.39");//ip地址
	if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))==-1)//绑定
	{
		perror("bind error");
		return -1;
	}

	//3 监听
	if(listen(sfd,128)==-1)
	{
		perror("listen error");
		return -1;
	}

	//4 接受客户端链接请求
	struct sockaddr_in cin;          //用于接收地址信息
	socklen_t socklen = sizeof(cin); //用于接收地址信息的大小
	
	//定义地址信息结构体数组,存储链接的客户端信息
	struct sockaddr_in cin_arr[1024];

	//定义一个等待文件描述符结构体数组
	int count_fd=4;  //文件描述符的个数
	struct pollfd pfd[count_fd];
	pfd[0].fd=0;
	pfd[3].fd=sfd;

	pfd[0].events=POLLIN;
	pfd[3].events=POLLIN;
	
	int maxfd=sfd;  //存储最大的文件描述符
	int newfd=-1;
	
	char rbuf[128]="";
	char wbuf[128]="";
	while(1)
	{
		int flag=poll(pfd,count_fd,-1);
		
		if(flag<0)
		{
			perror("poll error");
			return -1;
		}
		
		//1 新客户端链接 
		if(pfd[sfd].revents=POLLIN)
		{
			newfd=accept(sfd,(struct sockaddr*)&cin,&socklen);
			if(newfd==-1)
			{
				perror("accept error");
				return -1;
			}
			cin_arr[newfd]=cin;
			printf("[%s:%d]发来连接请求，newfd=%d\n",inet_ntoa(cin_arr[newfd].sin_addr),ntohs(cin_arr[newfd].sin_port),newfd);
			pfd[newfd].events=POLLIN;
			pfd[newfd].fd=newfd;
			//更新描述符的长度
			if(newfd>maxfd)
			{
				maxfd=newfd;
				count_fd++;
			}
		}

		//向客户端发送消息
		if(pfd[0].revents==POLLIN)
		{
			memset(wbuf,0,sizeof(wbuf));
			//scanf(" %s",wbuf);
			fgets(wbuf,sizeof(wbuf),stdin);
			wbuf[strlen(wbuf)-1]=0;
			
			for(int i=4;((i<=count_fd)&&(pfd[i].fd!=-1));i++)
			{
				//sendto(i,wbuf,sizeof(wbuf),0,(struct sockaddr*)&cin_arr[i],sizeof(cin_arr[i]));
				send(i,wbuf,sizeof(wbuf),0);
			}
			printf("发送成功\n");
		}

		//接收客户端的消息
		for(int i=4;i<=maxfd&&pfd[i].fd!=-1;i++)
		{
			if(pfd[i].revents==POLLIN)
			{
				//接收客户端消息
				memset(rbuf,0,sizeof(rbuf));
				//int res=read(i,rbuf,sizeof(rbuf));
				int res=recv(i,rbuf,sizeof(rbuf),0);
				if(res==0)
				{
					printf("客户端已经下线\n");
					close(i);
					//更新pfd 从集合中删除
					pfd[i].fd=-1;
				}
				printf("[%s:%d]:%s\n",inet_ntoa(cin_arr[i].sin_addr),\
						ntohs(cin_arr[i].sin_port),rbuf);
				//给客户端发消息
				strcat(rbuf,"^_^");
				send(i,rbuf,sizeof(rbuf),0);
				printf("发送成功\n");
			}
		}
	}
	return 0;
}

使用select实现TCP客户端的并发

#include <myhead.h>
//TCP 客户端
#define SER_IP "192.168.122.39"  //服务器ip地址
#define SER_PORT 8888          //服务器端口号

int main(int argc, const char *argv[])
{
	//1 创建套接字
	int cfd=-1;
	if((cfd=socket(AF_INET,SOCK_STREAM,0))==-1)
	{
		perror("socket error\n");
		return -1;
	}
	//2 可以不绑定
	//2.1填写地址信息结构体
	struct sockaddr_in cin;
	cin.sin_family=AF_INET;
	cin.sin_port=htons(6666);
	cin.sin_addr.s_addr=inet_addr("192.168.122.39");
	//2.2绑定
	if(bind(cfd,(struct sockaddr*)&cin,sizeof(cin))==-1)
	{
		perror("bind error");
		return -1;
	}
	//3 连接服务器
	struct sockaddr_in sin;
	sin.sin_family=AF_INET;
	sin.sin_port=htons(8888);
	sin.sin_addr.s_addr=inet_addr("192.168.122.39");

	if(connect(cfd,(struct sockaddr*)&sin,sizeof(sin))==-1)
	{
		perror("connect error");
		return -1;
	}
	printf("connect success\n");
	
	//准备文件描述符容器
	fd_set readfds,tempfds;
	FD_ZERO(&readfds);
	FD_SET(0,&readfds);
	FD_SET(cfd,&readfds);
	int maxfd=cfd;
	
	//4 收发数据
	char buf[128]="";
	while(1)
	{
		tempfds=readfds;
		int res=select(maxfd+1,&tempfds,NULL,NULL,NULL);
		if(res==-1)
		{
			perror("select error");
			return -1;
		}else if(res==0)
		{
			printf("timeout\n");
			return -1;
		}
		if(FD_ISSET(0,&tempfds))
		{
			//发送数据
			memset(buf,0,sizeof(buf));
			read(0,buf,sizeof(buf));
			buf[strlen(buf)-1]=0;
			if(strcmp(buf,"quit")==0)
			{
				printf("已退出\n");
				close(cfd);
				break;
			}
			write(cfd,buf,sizeof(buf));
			printf("发送成功\n");
		}
		if(FD_ISSET(cfd,&tempfds))
		{
			//接收数据
			recv(cfd,buf,sizeof(buf),0);
			printf("[%s:%d]:%s\n",SER_IP,SER_PORT,buf);
		}
	}
	return 0;
}