网络编程DAY4 并发服务器

最新推荐文章于 2024-09-05 16:37:49 发布

qq_49846847

最新推荐文章于 2024-09-05 16:37:49 发布

阅读量29

点赞数

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

本文链接：https://blog.csdn.net/qq_49846847/article/details/129568059

版权

本文展示了两个示例，一个是基于多进程的并发服务器，通过fork创建子进程来处理客户端连接；另一个是多线程并发服务器，使用pthread_create创建线程来服务客户端。每个示例都包括了接收客户端连接、处理消息并发送响应的过程。

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

1、多进程并发服务器

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

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

#define PORT 6666       //端口范围1024~49151
#define IP "192.168.31.106"  //IP地址，本机IP终端输入ifconfig查看

void zombie_callback(int sig)
{
	while(waitpid(-1,NULL,WNOHANG) > 0);
}

int deal_cli_msg(int newfd,struct sockaddr_in cin)
{
	char buf[128] = "";
	ssize_t res = -1;
	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)
		{
			fprintf(stderr,"[%s : %d] newfd=%d 客户端下线\n",\
					inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);
			break;
		}
		printf("[%s : %d] newfd=%d : %s\n",\
				inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd,buf);
		//发送
		strcat(buf,"*_*");
		if(send(newfd,buf,sizeof(buf),0) < 0)
		{
			ERR_MSG("send");
			return -1;
		}
		printf("send success\n");
	}
	return 0;
}
int main(int argc, const char *argv[])
{
	//捕获17号信号
	__sighandler_t s = signal(17,zombie_callback);
	if(s == SIG_ERR)
	{
		ERR_MSG("signal");
		return -1;
	}
	//socket

	int sfd = socket(AF_INET,SOCK_STREAM,0);
	if(sfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}
	//快速复用
	int resue = 1;
	if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&resue,sizeof(resue)) < 0)
	{
		ERR_MSG("setsockopt");
		return -1;
	}
	//bind
	struct sockaddr_in sin;
	sin.sin_family   =AF_INET;    //固定的，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;
	}

	//listen
	if(listen(sfd,128) < 0)
	{
		ERR_MSG("listen");
		return -1;
	}

	//accept
	struct sockaddr_in cin; //存储连接成功的客户端的地址信息
	socklen_t addrlen = sizeof(cin); 
	int newfd = -2;
	pid_t cpid = 0;

	while(1)
	{
		//父进程负责连接
		newfd = accept(sfd,(struct sockaddr*)&cin,&addrlen);
		if(newfd < 0)
		{
			ERR_MSG("accept");
			return -1;
		}

		printf("[%s : %d] newfd=%d 客户端连接成功\n",\
				inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);
		//运行到当前位置，客户端连接成功
		//创建一个子进程，用于与客户端交互
		cpid = fork();
		if(cpid == 0)
		{
			close(sfd); // 对于子进程而言，sfd没用

			deal_cli_msg(newfd,cin);

			close(newfd);
			exit(0);     //退出子进程

		}
		else if(cpid > 0)
		{
			close(newfd); //对于父进程而言，newfd没用
		}
		else
		{
			ERR_MSG("fork");
			return -1;
		}
	}
	close(sfd); //关闭所有套接字文件描述符
 	return 0;
}

2、多线程并发服务器

#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>

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

#define PORT 6666
#define IP "192.168.31.106"
struct cli_msg
{
	int newfd;
	struct sockaddr_in cin;
};
void* deal_cli_msg(void* arg);
int main(int argc, const char *argv[])
{
	//socket
	int sfd = socket(AF_INET,SOCK_STREAM,0);
	if(sfd < 0)
	{
		ERR_MSG("socket");
		return -1;
	}
	//bind
	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)) < 0)
	{
		ERR_MSG("bind");
		return -1;
	}

	//listen
	if(listen(sfd,128) < 0)

	{
		ERR_MSG("listen");
		return -1;
	}
	struct sockaddr_in cin;
	socklen_t addrlen = sizeof(cin);
	int newfd = -1;

	pthread_t tid;
	struct cli_msg info;

	while(1)
	{
		//accept
		newfd = accept(sfd,(struct sockaddr*)&cin,&addrlen);
		if(newfd < 0)
		{
			ERR_MSG("accept");
			return -1;
		}
		printf("[%s : %d] newfd=%d 客户端连接成功\n",\
				inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);

		info.newfd = newfd;
		info.cin = cin;

		if(pthread_create(&tid,NULL,deal_cli_msg,&info) != 0)
		{
			fprintf(stderr,"pthread_create failed __%d__\n",__LINE__);
			return -1;
		}

		pthread_detach(tid);
	}

	close(sfd);
	return 0;
}
void* deal_cli_msg(void* arg)
{
	int newfd = ((struct cli_msg*)arg)->newfd;
	struct sockaddr_in cin = ((struct cli_msg*)arg)->cin;

	char buf[128] = "";
	ssize_t res = -1;
	while(1)
	{
		bzero(buf,sizeof(buf));
		//接收
		res = recv(newfd,buf,sizeof(buf),0);
		if(res < 0)
		{
			ERR_MSG("recv");
			break;
		}
		else if(0 == res)
		{
			fprintf(stderr,"[%s : %d] newfd=%d 客户端下线\n",\
					inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd);
			break;
		}
		printf("[%s : %d] newfd=%d : %s\n",\
				inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd,buf);
		//发送
		strcat(buf,"*_*");
		if(send(newfd,buf,sizeof(buf),0)<0)
		{
			ERR_MSG("send");
			break;
		}
		printf("send success\n");
	}
	close(newfd);

	pthread_exit(NULL);
}