网络编程----用select实现服务器客户端收发

最新推荐文章于 2024-07-27 14:51:42 发布

许胖胖的梦想

最新推荐文章于 2024-07-27 14:51:42 发布

阅读量83

点赞数

文章标签：网络服务器运维

本文链接：https://blog.csdn.net/m0_68697184/article/details/130114991

版权

//服务器
#include <stdio.h>
#include <stdlib.h>
#include <string.h>	
#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>


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

#define IP "192.168.8.178"   //本机IP  ifconfig
#define PORT 6666       //1024-49151

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

	//允许端口快速重用
	int reuse = 1;
	if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse,sizeof(reuse)) <0)
	{
		ERR_MSG("setsockopt");
		return -1;
	}
	printf("允许端口快速重用成功\n");
	
	//填充地址信息结构体
	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);  //本机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, tmpfds;     //vi -t fd_set,集合中只有一个成员是长整型数组

	//将readfds清空
	FD_ZERO(&readfds);
	FD_ZERO(&tmpfds);

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

	int maxfd = sfd;    //最大文件描述符

	int s_res = 0;
	char buf[128] = "";
	struct sockaddr_in cin;  
	socklen_t addrlen = sizeof(cin);

	struct sockaddr_in saveCin[1024-4];

	int newfd = -1;
	ssize_t res = 0;

	while(1)
	{
		tmpfds = readfds;
		s_res = select (sfd+1, &tmpfds,NULL,NULL,NULL);
		if(s_res <0)
		{
			ERR_MSG("select");
			return -1;
		}
		else if(0 ==s_res)
		{
			printf("time out...\n");
			return -1;
		}

		for(int i =0; i<=maxfd;i++)
		{
			if(!FD_ISSET(i, &tmpfds))
			{
				continue;
			}

			if(0 == i)
			{
				printf("触发键盘输入事件>>>");
				int sndfd = -1;
				res = scanf("%d %s",&sndfd,buf);
				while(getchar()!=10);
				if(res != 2)
				{
					fprintf(stderr,"请输入正确格式: int string\n");
					continue;
				}

				//能运行到当前位置，则代表输入的格式整数
				if(sendfd<=sfd ||sndfd>=1024 || !FD_iSSET(sndfd,&readfds))
				{
					fprintf(stderr,"sndfd=%d 文件描述符错误\n",sndfd);
					continue;
				}

				if(send(sndfd,buf,sizeof(buf),0) <0)
				{
					ERR_MSG("send");
					return -1;
				}

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

				saveCin[newfd-4] = cin;

				//将newfd添加到读集合中
				FD_SET(newfd, &readfds);
				
				//更新maxfd
				maxfd = maxfd>newfd?maxfd:newfd;

			}
			else
			{
				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 客户端下线__%d__\n",\
						inet_ntoa(cin.sin_addr),ntohs(cin.sin_port), i ,__LINE__);
					//关闭文件描述符
					close(i);
					//从集合中删除文件描述符
					FD_CLR(i,&readfds);
					
					//更新maxfd
					//从目前最大文件描述符往小判断，直到判断在集合中
					//该文件描述符就是最大的
					int j =0;
					for(j=maxfd;j>=0;j--)
					{
						if(FD_ISSET(j,&readfds))
						{
							maxfd = j;
							break;
						}
					}

					if(j<0)   
						maxfd =-1;
					
					continue;
				}
				printf("[%s:%d] newfd=%d :%s__%d__\n",\
					inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd,buf,__LINE__);
			}
		}
	}

	

	close(sfd);
	return 0;
}

客户端
#include <stdio.h>
#include <stdlib.h>
#include <string.h>	
#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>


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

#define IP "192.168.8.178"   //本机IP  ifconfig
#define PORT 6666       //1024-49151

int main(int argc, const char *argv[])
{
	//创建流式套接字
	int cfd = socket(AF_INET,SOCK_STREAM,0);
	if(cfd <0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("cfd = %d\n", cfd);

	//允许端口快速重用
	int reuse = 1;
	if(setsockopt(cfd, SOL_SOCKET, SO_REUSEADDR, &reuse,sizeof(reuse)) <0)
	{
		ERR_MSG("setsockopt");
		return -1;
	}
	printf("允许端口快速重用成功\n");

	//绑定客户端的地址信息，非必须绑定的
	//如果不绑定，则由操作系统自动选择一个端口号，以及本机可用IP绑定到套接字上
	
	//填充服务器的地址信息结构体，供给connect函数使用
	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);  //本机IP


	//连接服:务器
	if(connect(cfd, (struct sockaddr*)&sin ,sizeof(sin)) <0)
	{
		ERR_MSG("connect");
		return -1;
	}
	printf("connect success__%d__",__LINE__);


	//设置一个集合
	fd_set readfds,tmpfds;

	//清空
	FD_ZERO(&readdfds);
	FD_ZERO(&tmpfds);

	//将需要的文件描述符添加到集合中
	FD_SET(0, &readfds);
	FD_SET(cfd, &readfds);

	//标记最大的文件描述符
	int maxfd=cfd;
	char buf[128] ="";
	ssize_t res=0;
	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;
		}
		if(0 ==s_res)
		{
			printf("time out...\n");
			break;
		}
		//能运行到这里，则有文件描述符准备就绪
		if(FD_ISSET(0,&tempfds))
		{
		//发信息给服务器
			bzero(buf,sizeof(buf));
			fgets(buf,sizeof(buf),stdin);
			buf[strlen(buf)-1]='\0';
			if(send(cfd,buf,sizeof(buf),0) <0)
			{
				ERR_MSG("send");
			}
			printf("send success\n");
		}

		if(FD_ISSET(cfd,&tempfds))
		{
			//接收服务器的信息
		res = recv(cfd , buf ,sizeof(buf) , 0);
		if(-1 ==res)
		{
			ERR_MSG("recv");
		}
		
		else if(0 ==res)
		{
			printf(stderr,"cfd=%d\n 客户端下线\n",cfd);
			break;
		}
		printf("cfd=%d: %s\n",cfd,buf);
		}
	}

	close(cfd);

	return 0;
}