TCP服务器的搭建练习

最新推荐文章于 2024-09-23 16:43:34 发布

meiyuefish

最新推荐文章于 2024-09-23 16:43:34 发布

阅读量156

点赞数

文章标签：服务器数据库

本文链接：https://blog.csdn.net/meiyuefish/article/details/130349058

版权

本文提供了两个示例，一个是使用select模型实现的TCP服务器，它能处理多个客户端连接并进行通信；另一个是使用poll模型的TCP客户端，它可以向服务器发送和接收数据。代码详细展示了socket编程中的连接、监听、接受、发送和接收等关键步骤。

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

1. 将TCP的select服务器重新写一遍（上交）

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <netinet/in.h>
#include <string.h>
#include <unistd.h>
#include <pthread.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 1150 
#define IP "192.168.0.131"   //本机IP,使用ifconfig

/*
   void* deal_cli_msg(void* arg);

   struct cli_msg
   {
   int newfd;
   struct sockaddr_in cin;
   };
   */


int main(int argc, const char *argv[])
{

	//创建流式套接字
	int  sfd = socket(AF_INET,SOCK_STREAM,0);
	if(sfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}

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

	//填充服务器地址信息
	struct sockaddr_in sin;
	sin.sin_family   =   AF_INET;
	sin.sin_port     =   htons(PORT);//端口号
	sin.sin_addr.s_addr   =   inet_addr(IP);//IP地址

	//绑定
	if(bind(sfd,(struct sockaddr *)&sin,sizeof(sin))<0)
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("bind  success sfd = __%d__\n",__LINE__);

	//监听
	if(listen(sfd,128) < 0)
	{
		ERR_MSG("listen");
		return -1;
	}
	printf("listern success\n");

	//创建读集合 fd_set 存放的是long int 类型数组成员
	//readfds中需存储要检测的文件描述符
	//readfds数组中的数组为随机数，需要赋值为无意义的整形数据

	fd_set readfds,tempfds;

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

	FD_SET(0,&readfds);
	FD_SET(sfd,&readfds);

	int maxfd = sfd;    //存储集合中的最大文件描述符

	struct sockaddr_in saveCin[1024-4];
	struct sockaddr_in cin;             //存储连接成功的客户端地址信息
	socklen_t addrlen = sizeof(cin);
	int newfd = -1;
	char buf[1024] = "";
	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;
		}
		else if(0 == s_res)
		{
			printf("time out...\n");
			break;
		}
		printf("__%d__\n",__LINE__);

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

			if(0==i)
			{
				printf("keybord time>>>");
				fflush(stdout);
				
				int sndfd;
				res = scanf("%d %s",&sndfd,buf);
				while(getchar() != '\n');
				if(res != 2)
				{
					fprintf(stderr,"input correct format:fd string\n");
					continue;
				}

				//输入格式非法
				if(sndfd<=sfd ||sndfd>=1024 || !FD_ISSET(sndfd,&readfds))
				{
					fprintf(stderr,"sndfd =%d is illegal\n",sndfd);
					connect;
				}
				if(send(sndfd,buf,sizeof(buf),0) < 0)
				{
					ERR_MSG("send");
					return -1;
				}
				printf("send to %d success\n",sndfd);
			}


			else if(sfd == i)
			{
				printf("username time>>>");
				fflush(stdout);

				newfd = accept(sfd,(struct sockaddr*)&cin,&addrlen);
				if(newfd < 0)
				{
					ERR_MSG("accept");
					return -1;
				}
				printf("[ %s : %d ]newfd = %d connect success __%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
			{
				fprintf(stderr,"Cli time>>>>>");
				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 unconnect __%d__\n",\
							inet_ntoa(saveCin[i-4].sin_addr),ntohs(saveCin[1-4].sin_port),i,__LINE__);

					close(i);

					FD_CLR(i,&readfds);

					int j = maxfd;
					for( ; j>=0;j--)
					{
						if(FD_ISSET(j,&readfds))
							break;
					}
					maxfd = j;
					continue;
				}
				printf("[ %s : %d ]newfd = %d : %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\n");

			}
		}
	}
	close(sfd);
	return 0;
}

2. 将TCP的poll客户端写一遍（上交）

#include <stdio.h>
#include <sys/types.h>    
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>
#include <unistd.h>
#include <poll.h>

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

#define PORT 1150
#define IP "192.168.0.131"


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

	//绑定，非必须绑定
	//不绑定责随机分配端口号
	
	//2.填充服务器地址信息
	struct sockaddr_in sin;
	sin.sin_family     = AF_INET;
	sin.sin_port       = htons(PORT);//端口号的网络字节序1024--49151
	sin.sin_addr.s_addr= inet_addr(IP);//IP 本机IP ifconfig


	
	//连接服务器
	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[1024] = "";
	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");
			continue;
		}
		//读
		if(fds[0].revents & POLLIN)
		{
			printf("keyborad event\n");
			bzero(buf,sizeof(buf));
			fgets(buf,sizeof(buf),stdin);
			buf[strlen(buf)-1] ='\0';

			//send
			if(send(cfd,buf,sizeof(buf),0) < 0)
			{
				ERR_MSG("send");
				return -1;
			}
			printf("send success\n");
		}
		if(fds[1].revents & POLLIN)
		{
			printf("Server Event\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("Server Disconnect\n");
				break;
			}
			printf("%s %d\n",buf,__LINE__);
		}
	
	}
	//关闭文件描述符
	close(cfd);
	return 0;
}

3. 将TCP的select客户端写一遍（上交）

#include <stdio.h>
#include <sys/types.h>    
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <string.h>
#include <unistd.h>
#include <poll.h>

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

#define PORT 1150
#define IP "192.168.0.131"


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

	//绑定，非必须绑定
	//不绑定责随机分配端口号
	
	//2.填充服务器地址信息
	struct sockaddr_in sin;
	sin.sin_family     = AF_INET;
	sin.sin_port       = htons(PORT);//端口号的网络字节序1024--49151
	sin.sin_addr.s_addr= inet_addr(IP);//IP 本机IP ifconfig


	
	//连接服务器
	if(connect(cfd,(struct sockaddr*)&sin,sizeof(sin)) < 0);
	{
		ERR_MSG("connect");
		return -1;
	}
	printf("connect success\n");

	fd_set fds;

	FD_SET(0,&fds);

	FD_SET(cfd,&fds);

	int maxfd = cfd;

	char buf[1024] = "";
	ssize_t res = 0;
	int p_res = -1;
	while(1)
	{
		p_res = select(maxfd+1,&fds,NULL,NULL,NULL);
		if(p_res < 0)
		{
			ERR_MSG("poll");
			return -1;
		}
		else if(0==p_res)
		{
			printf("time out...\n");
			continue;
		}
		//读
		if(FD_ISSET(0,&fds))
		{
			printf("keyborad event\n");
			bzero(buf,sizeof(buf));
			fgets(buf,sizeof(buf),stdin);
			buf[strlen(buf)-1] ='\0';

			//send
			if(send(cfd,buf,sizeof(buf),0) < 0)
			{
				ERR_MSG("send");
				return -1;
			}
			printf("send success\n");
		}

		if(FD_ISSET(cfd,&fds))
		{
			printf("Server Event\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("Server Disconnect\n");
				break;
			}
			printf("%s %d\n",buf,__LINE__);
		}
	
	}
	//关闭文件描述符
	close(cfd);
	return 0;
}