基于tpc协议的服务器框架——简单的服务器收发

老K殿下

于 2022-08-10 00:45:01 发布

阅读量290

点赞数

分类专栏：每日练习文章标签：服务器网络运维

本文链接：https://blog.csdn.net/laoKdx/article/details/126257928

版权

每日练习专栏收录该内容

56 篇文章 2 订阅

订阅专栏

在开始之前，先做几个准备工作

定义答应错误信息的宏函数，包括错误原因以及行号

//打印错误信息的宏函数
#define ERR_MSG(msg) do{\
	fprintf(stderr, "__%d__", __LINE__);\
	perror(msg);\
}while(0)

宏定义网络端口以及IP地址

#define PORT 8888      //自行制定，从1024~49151中选择一个
#define IP "192.168.x.x"          //本机ip，用ifconfig查看，此处隐去

服务器原理图如下

服务器流式套接字类似于“掮客”（bushi），它可以对接客户端的套接字，然后同时生成一个新的套接字与之对应。我们首先就是要创建套接字

/*************************************************************************/
/******************************以下摘自man手册*****************************/

socket函数
功能：指定传输层网络层的协议，并创建一块缓冲区，获取这块缓冲区对应的文件描述符;
头文件：
       #include <sys/types.h>          /* See NOTES */
       #include <sys/socket.h>

       int socket(int domain, int type, int protocol);
参数：
    int domain：Linux支持的协议族（地址族）;
       Name                Purpose                          Man page
       AF_UNIX, AF_LOCAL   Local communication              unix(7)
       AF_INET             IPv4 Internet protocols          ip(7)
       AF_INET6            IPv6 Internet protocols          ipv6(7);

	int type：指定套接字类型
       SOCK_STREAM：字节流式套接字-->TCP ： 默认是IPPROTO_TCP协议
       SOCK_DGRAM：数据报式套接字--->UDP 	默认是IPPROTO_UDP协议
	   SOCK_RAW：原始套接字，需要在第三个参数中手动指定协议，

    int protocol：0，默认协议;

返回值：
    成功，返回套接字对应的文件描述符;
	失败，返回-1，更新errno;

流式套接字创建

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

填充地址信息结构体，然后将地址信息结构体绑定到套接字上

    //填充地址信息结构体，真实的地址信息结构体与协议族相关
	//AF_INET,所以请翻看man 7 ip
	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\n");

其中用到了一些系统自带的函数

/*********************************************************************/
htons  htonl   主机字节序---->网络字节序
原型：
       uint32_t htonl(uint32_t hostlong);
       uint16_t htons(uint16_t hostshort);
参数：
    uint32_t hostlong：32位主机字节序整型;
	uint16_t hostshort：16位主机字节序整型;

以及点分十进制与网络字节序的转换

原型：
       in_addr_t inet_addr(const char *cp);
参数：
     char *cp：源IP地址的点分十进制字符串，例如 “192.168.1.10”;
返回值：
    成功，返回转换后的网络字节序IP地址;
					typedef uint32_t in_addr_t;

	失败，返回INADDR_NONE (usually -1);

只能转换IPv4;

把套接字转化为被动监听状态

//将套接字设置为被动监听状态，让内核取监听是否有客户端链接
	if(listen(sfd, 10) < 0)
	{
		ERR_MSG("listen");
		return -1;
	}
	printf("listen success\n");

    struct  sockaddr_in  cin;
	socklen_t  addrlen = sizeof(cin);
	//从已完成链接的队列头中，取出一个客户端的信息
	//创建生成一个信道套接字文件描述符
	//该文件描述符才是与客户端通信的文件描述符！！！
	int newfd = accept(sfd, (struct sockaddr*)&cin, &addrlen);
	if(newfd < 0)
	{
		perror("accept");
		return -1;
	}

//把网络字节序的IP-->点分十进制  网络字节顺序-->主机字节序
printf("[%s : %d] newfd = %d\n", inet_ntoa( cin.sin_addr ),\
         ntohs( cin.sin_port ), newfd);

要完成服务器的收发，需要多线程或者多进程，不然，会导致卡死在读或者写上

创建线程

//创建线程
	pthread_t tid;
	if(pthread_create(&tid, NULL, mysend, &newfd) != 0)
	{
		perror("pthread_create");
		return -1;
	}

主线程用来接收信息

char buf[128] = "";
ssize_t res = 0;	
while(1)
{
	bzero(buf, sizeof(buf));
	//循环接收
	res = recv(newfd, buf, sizeof(buf), 0);
	if(res < 0)
	{
		ERR_MSG("recv");
		return -1;
	}else if(0 == res)
	{
		printf("\n[%s : %d] newfd = %d客户端退出\n", \
              inet_ntoa( cin.sin_addr ), ntohs( cin.sin_port ), newfd);
		break;
	}
    //输出接收到的信息
	printf("\n[%s : %d] newfd = %d : %s\n",\
           inet_ntoa( cin.sin_addr ), ntohs( cin.sin_port ), newfd, buf);
    //为了代码运行的美观性加一句
	fprintf(stderr,"请输入消息内容>>>");
}

子线程用来处理发送消息

//发送线程
void *mysend(void *arg)
{   //接收newfd
	int *p = (int*)arg; 
	int newfd = *p;
	char buf[128] = "";
	ssize_t res;
	while(1)
	{
		//清零
		bzero(buf, sizeof(buf));
		//循环发送
		fprintf(stderr,"请输入消息内容>>>");
		fgets(buf, sizeof(buf)-1, stdin);   //从终端获取字符
		res = send(newfd, buf, sizeof(buf), 0);
		if(res < 0)
		{
			ERR_MSG("send");
			return NULL;
		}else if(0 == res)
		{
			break;
		}
	}
}

附上完整代码

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

//打印错误信息的宏函数
#define ERR_MSG(msg) do{\
	fprintf(stderr, "__%d__", __LINE__);\
	perror(msg);\
}while(0)

#define PORT 8888      //自行制定，从1024~49151中选择一个
#define IP "192.168.1.106"          //本机ip，用ifconfig查看

//发送线程
void *mysend(void *arg)
{
	int *p = (int*)arg;
	int newfd = *p;
	char buf[128] = "";
	ssize_t res;
	while(1)
	{
		//清零
		bzero(buf, sizeof(buf));
		//循环发送
		fprintf(stderr,"请输入消息内容>>>");
		fgets(buf, sizeof(buf)-1, stdin);
		res = send(newfd, buf, sizeof(buf), 0);
		if(res < 0)
		{
			ERR_MSG("send");
			return NULL;
		}else if(0 == res)
		{
			break;
		}
	}
}

int main(int argc, const char *argv[])
{
	//创建流式套接字
	int sfd = socket(AF_INET, SOCK_STREAM, 0);
	if(sfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("create socket success\n");
	
	//填充地址信息结构体，真实的地址信息结构体与协议族相关
	//AF_INET,所以请翻看man 7 ip
	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\n");

	//将套接字设置为被动监听状态，让内核取监听是否有客户端链接
	if(listen(sfd, 10) < 0)
	{
		ERR_MSG("listen");
		return -1;
	}
	printf("listen success\n");
	
	struct  sockaddr_in  cin;
	socklen_t  addrlen = sizeof(cin);
	//从已完成链接的队列头中，取出一个客户端的信息
	//创建生成一个信道套接字文件描述符
	//该文件描述符才是与客户端通信的文件描述符！！！
	int newfd = accept(sfd, (struct sockaddr*)&cin, &addrlen);
	if(newfd < 0)
	{
		perror("accept");
		return -1;
	}
	
	//把网络字节序的IP-->点分十进制  网络字节顺序-->主机字节序
	printf("[%s : %d] newfd = %d\n", inet_ntoa( cin.sin_addr ), ntohs( cin.sin_port ), newfd);
	
	//创建线程
	pthread_t tid;
	if(pthread_create(&tid, NULL, mysend, &newfd) != 0)
	{
		perror("pthread_create");
		return -1;
	}

	char buf[128] = "";
	ssize_t res = 0;
	while(1)
	{
		bzero(buf, sizeof(buf));
		//循环接收
		res = recv(newfd, buf, sizeof(buf), 0);
		if(res < 0)
		{
			ERR_MSG("recv");
			return -1;
		}else if(0 == res)
		{
			printf("\n[%s : %d] newfd = %d客户端退出\n", inet_ntoa( cin.sin_addr ), ntohs( cin.sin_port ), newfd);
			break;
		}
		printf("\n[%s : %d] newfd = %d : %s\n", inet_ntoa( cin.sin_addr ), ntohs( cin.sin_port ), newfd, buf);
		fprintf(stderr,"请输入消息内容>>>");
	}
	
    
	close(sfd);
	close(newfd);

	return 0;
}