11/22

最新推荐文章于 2024-05-12 08:52:10 发布

指尖流烟

最新推荐文章于 2024-05-12 08:52:10 发布

阅读量18

点赞数

文章标签： linux

本文链接：https://blog.csdn.net/2201_75550624/article/details/134562903

版权

客户端 tcp

#include <myhand.h>

#define SERIP "192.168.114.118"
#define SERPORT 8888
#define CLIIP "192.168.246.135"
#define CLIPORT 7777
int main(int argc, const char *argv[])
{
	//1.建立套接字
	int cfd =socket(AF_INET,SOCK_STREAM,0);
	if(cfd ==-1)
	{
		perror("socket error");
		return -1;
	}
	printf("socket success cfd=%d\n",cfd);

	//设置端口快速重用
	int reuse =1;
	if(setsockopt(cfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))==-1)
	{
		perror("setsockopt error");
		return -1;
	}
	printf("设置端口快速重用成功_%d_%s_%s\n",__LINE__,__FILE__,__func__);
	//2.绑定
	//定义地址结构体
	struct sockaddr_in cin;
	cin.sin_family =AF_INET;
	cin.sin_port   =htons(CLIPORT);
	cin.sin_addr.s_addr =inet_addr(CLIIP);
	//bind
	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(SERPORT);
	sin.sin_addr.s_addr =inet_addr(SERIP);
    //connect
	if(connect(cfd,(struct sockaddr*)&sin,sizeof(sin))==-1)
	{
	 	perror("connect error ");
		return -1;
	}
	printf("connect success \n");
	
	//4.收发数据(send recv read write)
	char buf[128]="";
	char rbuf[128]="";

	定义一个用于检测文件描述符的结合
	fd_set readfds,tempfds;   //在栈区定义的
	
	清空容器
	FD_ZERO(&readfds);
	将要检测的文件描述符放进去
	FD_SET(cfd,&readfds);
	FD_SET(0,&readfds);

	//定义一个res接收select的返回值
	int res=0;
	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    ; 
		}



		判断cfd是否还在集合中
		if(FD_ISSET(cfd,&tempfds))
		{    
			fgets(buf,sizeof(buf),stdin);
			buf[strlen(buf)-1];
			//将消息发给服务端
			send(cfd,buf,sizeof(buf),0);
            //如果输入quit则退出
			if(strcmp(buf,"quit")==0)
			{
				break;
			}
			
		}
		if(FD_ISSET(0,&tempfds))
		{
			//接收服务端发送过来的消息
			int res=recv(cfd,rbuf,sizeof(rbuf),0);
			if(res ==0)
			{
				printf("服务器已经关闭\n");
				break;
			}
			printf("rbuf=%s\n",rbuf);
		}
	}
        
    //5.关闭套接字
	close(cfd);
	return 0;
}

服务器端

#include <myhand.h>

#define PORT 8888
#define IP "192.168.246.135"

int main(int argc, const char *argv[])
{
	//创建套接字
	int sfd =socket(AF_INET,SOCK_STREAM,0);
	if(sfd==-1)
	{
		perror("socket error");
		return -1;
	}
	printf("socket success sfd=%d\n",sfd);
	//设置端口快速重用
	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__);



	//绑定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))==-1)
	{
		perror("bind error");
		return -1;
	}
	printf("bind success _%d_%s_%s_\n",__LINE__,__FILE__,__func__);


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

	定义一个用于检测文件描述符的集合
	fd_set readfds,tempfds; //在栈区定义

	清空容器内容
	FD_ZERO(&readfds);
   
	将要检测的文件描述符放入集合中
	FD_SET(sfd,&readfds);      //将sfd文件描述符放入
	FD_SET(0,&readfds);        //将0号文件描述符放入




	//定义一个容器
	char buf[128]="";  	
	int res =0;        //接收select的返回值
	int newfd =-1;     //存放用于最新连接客户端的套接字
	int maxfd =sfd;    //定义控制select函数中的最大文件描述符

	struct sockaddr_in saveCin[1024];    //用于存放客户端地址信息结构体
    

	while(1)
	{
		将集合内容复制一份
		tempfds =readfds;

		使用select阻塞等待集合中的文件描述符有事件产生
	    

		res =select(maxfd+1,&tempfds,NULL,NULL,NULL);
		if(res ==-1)
		{
			perror("select error");
			return -1;
		}
		else if(res==0)
		{
			printf("time out\n");
			return -1;
		}

		//遍历所以集合中文件描述符
		for(int i=0; i<=maxfd ;i++ )
		{
			//判断当前i是否在集合中，如果不在，直接判断下一个
			if(!FD_ISSET(i,&tempfds))
			{
				continue;
			}

			判断sfd是否在集合中
			if(i==sfd)
			{
				//阻塞接收客户端的连接请求，并获取客户端地址信息
				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__);
				将newfd放入readfds中
				FD_SET(newfd ,&readfds);
				//更新maxfd
				if(newfd>maxfd)
				{
					maxfd=newfd;
				}
				//将最新的客户端套接字放入数组的下标为new的位置
				saveCin[newfd] =cin;
				printf("newfd=%d\n",newfd);

			}
			else if(i==0)     //判断是否是终端输入
			{
           		char buf1[128]="";
				bzero(buf,sizeof(buf));
				//从终端获取数据
				fgets(buf,sizeof(buf),stdin);
				buf[strlen(buf)-1]='\0';
				printf("触发终端输入事件:%s\n",buf);

				sprintf(buf1,"%s%s","系统消息:",buf);

				//将数据发送给所有用客户端
				for(int j=4;j<=maxfd;j++)
				{
					send(j,buf1,sizeof(buf1),0);
				}

			}
			else
			{
				//收发数据使用newfd完成通信
				char buf[128]="";
				//清空字符串
				bzero(buf,sizeof(buf));
				int ret =recv(i,buf,sizeof(buf),0);
				if(ret==0)
				{
					printf("客户端已经下线\n");
					将当前的文件描述符从集合中删除
					FD_CLR(i,&readfds);

					更新maxfd
					for(int j=maxfd;j>=0;j--)
					{
						//判断当前的j是否在集合中，如果在，则为maxfd
						if(FD_ISSET(j,&readfds))
						{
							maxfd=j;
							break;
						}
					}
					continue; //继续判断下一个
				}
				else if(ret<0)
				{
					perror("recv error");
					return -1;
				}
				printf("[%s:%d]:%s\n",inet_ntoa(saveCin[i].sin_addr),ntohs(saveCin[i].sin_port),buf);
				//将读取的信息，加上一下字符发送回去
				strcat(buf,"\(*_*)/");
				send(i,buf,sizeof(buf),0);
			}
		}
	}


   		close(sfd);


	return 0;
}