网编day5

最新推荐文章于 2024-08-26 22:15:39 发布

weixin_58102424

最新推荐文章于 2024-08-26 22:15:39 发布

阅读量43

点赞数

文章标签：网络

本文链接：https://blog.csdn.net/weixin_58102424/article/details/131177826

版权

该代码示例展示了如何使用select系统调用创建一个TCP服务器，监听并处理来自客户端的连接和交互。服务器端接收连接请求，将新连接的套接字加入到文件描述符集合中，等待数据收发。客户端则通过connect连接服务器，并进行数据发送和接收。

摘要由CSDN通过智能技术生成

select的tcp服务器

#include<myhead.h>
#include <sys/types.h>         
#include <sys/socket.h>
#include<arpa/inet.h>
#include<netinet/in.h>
#include <sys/wait.h>


#define ERR_MSG(msg)do{\
	fprintf(stderr,"line:%d: %s %s",__LINE__,__FILE__,__func__);\
	perror(msg);\
}while(0)

#define IP "192.168.2.205"
#define PORT 7777


int main(int argc, const char *argv[])
{
	//创建流试套接字
	int sfd=socket(AF_INET,SOCK_STREAM,0);
	if(sfd<0)
	{ 		ERR_MSG("socket");
		return -1;
	}

	//填充服务器的地址信息结构体，给bind函数绑定
	//真实的地址信息结构体根据地址族制定：AF_INET:man 7 ip
	struct sockaddr_in sin;
	sin.sin_family =AF_INET;   //必须填AF_INET;
	sin.sin_port  =htons(PORT);//1024~49151 网络字节序
	sin.sin_addr.s_addr=inet_addr(IP);//ifconfig出来本机IP

	//绑定服务器的IP和端口
	if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("bind success __%d__\n",__LINE__);

	//将套接字设置为被动监听状态
	if(listen(sfd,128)<0)
	{
		ERR_MSG("listen");
		return -1;
	}
	printf("listen success __%d__\n",__LINE__);

	//创建要检测的集合
	fd_set readfds,tempfds;


	//清空集合
	FD_ZERO(&readfds);
	FD_ZERO(&tempfds);

	
	//将所需要的文件描述符设置到集合中
	FD_SET(0,&readfds);
	FD_SET(sfd,&readfds);




	int maxfd=sfd;
	//存储客户端的地址信息结构体
	struct sockaddr_in cin;
	socklen_t addrlen=sizeof(cin);
	int newfd=-1;
	//pid_t cpid=0;
	char buf[128]="";
	int s_res=0;


	//获取连接成功后新的套接字文件描述符

	while(1)
	{
		tempfds=readfds;
		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))
			continue;
		//可以运行到当前位置，说明i代表的文件描述符在集合中


		if(i==0)
		{
		printf("触发键盘事件\t");
	//	fflush(stdout);
		bzero(buf,sizeof(buf));
		fgets(buf,sizeof(buf),stdin);
		buf[strlen(buf)-1]=0;

		printf("%s\n",buf);
		}

		else if(i==sfd)
		{
		fprintf(stderr,"触发客户端连接事件");
		newfd=accept(sfd,(struct sockaddr*)&cin,&addrlen);
		if(newfd<0)
		{
		ERR_MSG("accept");
		return -1;
		}
		printf("客户端连接成功\n");
		//将newfd添加到集合中，让内核检测
		FD_SET(newfd,&readfds);
		//更新maxfd
		maxfd=maxfd>newfd?maxfd:newfd;

		}




else
{
		if(FD_ISSET(i,&tempfds))
		{
		fprintf(stderr,"触发客户端交互事件\n");
		bzero(buf,sizeof(buf));
		int res=recv(i,buf,sizeof(buf),0);
		if(res<0)
		{
		ERR_MSG("recv");
		return -1;
		}

		else if(0==res)
		{
		printf("[%s %d] newfd=%d 客户端下线 __%d__\n",\
		inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),i,__LINE__);

		//关闭文件描述符
		close(i);

		//从集合中剔除关闭文件描述符
		FD_CLR(i,&readfds);

		//更新最大文件描述符值
		int j=maxfd;
		while(!FD_ISSET(j,&readfds)&&j-->0);
		maxfd=j;
		continue;
		}


		printf("[%s %d] newfd=%d message=%s __%d__\n",\
		inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),i,buf, __LINE__);


		strcat(buf,"*_*");
		if(send(i,buf,sizeof(buf),0)<0)
		{
		ERR_MSG("send");
		return -1;
		}
		printf("send success __%d__\n",__LINE__);
		}
	
	}
}
	}
	close(sfd);
	return 0;
}

select的tcp客户端

#include<myhead.h>
#include <sys/types.h>          /* See NOTES */
#include <sys/socket.h>
#include <arpa/inet.h>
#include <sys/select.h>

#include <sys/time.h>
#include <unistd.h>
#include <sys/select.h>


#define IP "192.168.2.205"
#define PORT 7777

int main(int argc, const char *argv[])
{
	//创建套接字
	int sfd;
	sfd=socket(AF_INET,SOCK_STREAM,0);

	//填充服务器的地址信息结构体
	struct sockaddr_in sin;
	sin.sin_family=AF_INET;
	sin.sin_port=htons(PORT);
	sin.sin_addr.s_addr=inet_addr(IP);
	socklen_t addrlen=sizeof(sin);

	//用connect与服务器连接
	if(connect(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		perror("connect");
		return -1;
	}
//	printf("触发客户端连接事件\n");

	//创建要检测的集合
	fd_set readfds;
	fd_set tempfds;

	//清空集合
	FD_ZERO(&readfds);
	FD_ZERO(&tempfds);

	//将所需要的文件描述符放入集合中
	FD_SET(0,&readfds);
	FD_SET(sfd,&readfds);


	char buf[128];

	int s_res=0;
	int res=0;
	while(1)
	{
		tempfds=readfds;
		s_res=select(sfd+1,&tempfds,NULL,NULL,NULL);

			if(s_res<0)
			{
				perror("select");
				return -1;
			}
			else if(0==s_res)
			{
				printf("time out\n");
				return -1;
			}


		if(FD_ISSET(0,&tempfds))
		{
			printf("触发键盘交互事件");
			bzero(buf,sizeof(buf));
			scanf("%s",buf);
			//	while(getchar()!=10);
			buf[strlen(buf)]=0;

			if(send(sfd,buf,sizeof(buf),0)<0)
			{
				perror("send");
				return -1;
				printf("send success-->  %s\n ",buf);
			}

		}

		if(FD_ISSET(sfd,&tempfds))
		{
	printf("触发客户端连接事件");
			bzero(buf,sizeof(buf));
			//	while(getchar()!=10);

			res=recv(sfd,buf,sizeof(buf),0);
				if(res<0)
				{
					perror("send");
					return -1;
				}
					else if(0==res)
					{
						printf("客户端下线\n");
						break;
					}
					printf("recv success-->  %s \n",buf);

		}

	}
		return 0;
	}