网络编程day 5

是你

已于 2023-04-24 23:06:22 修改

阅读量114

点赞数

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

于 2023-04-24 22:51:42 首次发布

本文链接：https://blog.csdn.net/2302_76485417/article/details/130354088

版权

文章展示了如何使用select系统调用实现TCP服务器和客户端。服务器端创建套接字，设置端口重用，监听指定端口，接收客户端连接，处理键盘输入和客户端交互。客户端创建套接字，连接服务器，接收用户输入并发送到服务器，同时监听来自服务器的消息。此外，还提供了使用poll的客户端示例，实现类似的功能。

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

将TCP的select服务器重新写一遍

将TCP的select客户端写一遍

#include <stdio.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>
#include <sys/select.h>
#include <sys/time.h>
 
#define ERR_MSG(msg) do{\
    fprintf(stderr, "line:%d ", __LINE__);\
    perror(msg);\
}while(0)
 
#define IP "192.168.31.45"   //本机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");
 
 
 
    //填充地址信息结构体，
    //真实的地址信息结构体根据地址族指定 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);    //本机IP， ifconfig
 
    //将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中需要防止要检测的文件描述符，所以不能让他是随机值
    //所以需要将readfds清空
    FD_ZERO(&readfds);
    //将需要的文件描述符添加到集合中
    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]; //另存客户端地址信息，0,1,2，sfd不可能有对应的客户端
 
    int newfd = -1; 
    ssize_t res = 0;
 
	while(1)
	{   
		tmpfds = readfds;
		s_res = select(maxfd+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; 
		}
 
		//能运行到当前位置，则代表有文件描述符准备就绪
		//走触发事件的文件描述符对应的处理函数
 
		//当集合中有文件描述符准备就绪了，集合中会只保留产生事件的文件描述符
		//当0准备就绪，集合中会只保留0
		//当sfd准备就绪，集合中会只保留sfd
		//当0和sfd都准备就绪，集合中会保留0和sfd
		//所以只需要判断集合中剩下哪个文件描述符，走对应函数即可
		//循环遍历所有新生成的newfd是否有对应的事件产生
		//即所有newfd对应的客户端是否有发送数据给服务器
		for(int i=0;i<=maxfd;i++)
		{
			if(FD_ISSET(i,&tmpfds)==0)
			{ 
				continue;
			}
			if(0 == i)
			{ 
				printf("触发键盘输入事件>>> ");
				fflush(stdout);
				int sndfd = -1; 
				res = scanf("%d %s", &sndfd, buf);
				while(getchar()!=10);
				if(res != 2)
				{
					fprintf(stderr, "请输入正确格式：int string\n");
					continue;
				}
 
				//能运行到当前位置，则代表输入的格式整数
				if(sndfd<=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; 
				}
				buf[strlen(buf)-1]=0;
				printf("%s\n",buf);
			}
			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[i-4]=cin;
				//将newfd添加到readfds中
				FD_SET(newfd,&readfds);
				//更新maxfd
				maxfd=maxfd>newfd?maxfd:newfd;
			}
			else
			{
				if(FD_ISSET(i,&tmpfds))
				{ 
					printf("触发客户端交互事件>>>");
					fflush(stdout);
					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(saveCin[i-4].sin_addr), ntohs(saveCin[i-4].sin_port), i, __LINE__);
						//关闭文件描述符
						close(i);
						//剔除该文件描述符
						FD_CLR(i,&readfds);//剔除tmpfds，会覆盖，所以剔除readfds
						//更新maxfd,从大到小遍历
						int j=0;
						for(j=maxfd;j>=0;j--)
						{ 
							if(FD_ISSET(j,&readfds))
							{ 
								//	maxfd=j;
								break;
							}
						}
						maxfd=j;
						/*	if(j<0)
							maxfd=-1;*/
						continue;//当前客户端下线要继续遍历
					}
					printf("[%s:%d] newfd=%d : %s __%d__\n", \
							inet_ntoa(saveCin[i-4].sin_addr), ntohs(saveCin[i-4].sin_port), i, buf, __LINE__);
 
					//发送 将数据拼接字符串后发送回去
					strcat(buf, "*_*");
					if(send(i, buf, sizeof(buf), 0) < 0)
					{
						ERR_MSG("send");
						return -1;
					}
					printf("发送成功\n");
				}
			}
		}
	}
	close(sfd);
 
return 0;
}

#include <stdio.h>
#include <stdlib.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <unistd.h>
#include <string.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <sys/wait.h>
#include <sys/select.h>
#include <sys/time.h>
#define ERR_MSG(msg) do{\
	fprintf(stderr,"line:__%d__",__LINE__);\
	perror(msg);\
}while(0)
#define PORT 6666
#define IP "192.168.31.45"
 
int main(int argc, const char *argv[])
{
	//创建流试套接字
	int cfd = socket(AF_INET,SOCK_STREAM,0);
	if(cfd<0)
	{
		ERR_MSG("socket");
		return -1;
	} 
	//允许端口快速被重用
	int reuse = 1;
	if(setsockopt(cfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))<0)
	{
		ERR_MSG("setsockopt");
		return -1;
	}
 
	//填充服务器地址信息的结构体
	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\n");
 
	//创建一个读合集
	fd_set readfds, tempfds;
	FD_ZERO(&readfds);
	FD_ZERO(&tempfds);
 
	//将文件描述符存入读合集
	FD_SET(0,&readfds);
	FD_SET(cfd,&readfds);
 
	//读合集中存储最大的文件描述符
	int maxfd = cfd;
 
	char buf[128] = "";
	ssize_t res = 0;
	int pres = 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(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");
				return -1;
			}
			printf("send success\n");
		}
		if(FD_ISSET(cfd,&tempfds))
		{
			//接收服务器消息事件
			printf("触发服务器交互\n");
			bzero(buf,sizeof(buf));
			res = recv(cfd,buf,sizeof(buf),0);
			if(res<0)
			{
				ERR_MSG("recv");
				return -1;
			}
			else if(res==0)
			{
				printf("服务器断开连接\n");
				break;
			}
			printf("[%s:%d]:%s\n",inet_ntoa(sin.sin_addr),\
					ntohs(sin.sin_port),buf);
		}
	}
	close(cfd);
	return 0;
}

将TCP的poll客户端写一遍

#include<string.h>
#include<stdio.h>
#include<stdlib.h>
#include<arpa/inet.h>
#include<unistd.h>
#include<poll.h>
#include<netinet/in.h>
#include<sys/types.h>
#include<sys/socket.h>
#define ERR_MSG(msg) do{\
	fprintf(stderr,"line=__%d__",__LINE__);\
	perror(msg);\
}while(0)
#define IP "192.168.31.45"
#define PORT 6666
int main(int argc, const char *argv[])
{
	//创建
	int cfd=socket(AF_INET,SOCK_STREAM,0);
	if(cfd<0)
	{ 
		ERR_MSG("socket");
		return -1;
	}
	//存储服务器IP和端口
	struct sockaddr_in sin;
	sin.sin_family=AF_INET;
	sin.sin_port=htons(6666);
	sin.sin_addr.s_addr=inet_addr(IP);
	//连接服务器
	if(connect(cfd,(struct sockaddr*)&sin,sizeof(sin))<0)
	{ 
		ERR_MSG("connect");
		return -1;
	}
	printf("connect success\n");
	struct pollfd fds[2];
	fds[0].fd=0;//指定要监测的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;
		}
		//运行到此，有文件描述符触发，判断集合中文件描述的revents中有没有POLLIN事件
		//需要通过提取revents中的代表POLLIN的那一个bit是否为1
		//因为revents中可能有多个事件，POLLIN只是其中的一个事件
		if(fds[0].revents&POLLIN)
		{ 
			printf("触发键盘输入事件\n");
			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");
				return -1;
			}
			printf("发送成功\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("服务器下线\n");
				break;
			}
			printf("%s__%d__",buf,__LINE__);
		}
	}
	close(cfd);
	return 0;
}