并发服务器代码

多进程并发服务器代码

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

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

int deal_cli_msg(int newfd,struct sockaddr_in cin);

//回收僵尸进程
void handler(int sig)
{
	while(waitpid(-1,NULL,WNOHANG) > 0);
}
int main(int argc, const char *argv[])
{
	//捕捉17号信号，回收僵尸进程
	__sighandler_t s = signal(SIGCHLD,handler);
	if(SIG_ERR == s)
	{
		ERR_MSG("signal");
		return -1;
	}

	//创建套接字：socket
	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("允许端口快速重用\n");
	
	//填充服务器的地址信息结构体
	//该真实结构体根据地址制定
	struct sockaddr_in sin;
	sin.sin_family    =  AF_INET;
	sin.sin_port      =  htons(PORT);
	sin.sin_addr.s_addr = inet_addr(IP);

	//绑定服务器的IP和端口：必须绑定:bind
	if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin)) < 0)
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("bind success __%d__\n",__LINE__);

	//将套接字设置为被动监听状态:lieten
	if (listen(sfd, 128) < 0)
	{
		ERR_MSG("listen");
		return -1;
	}
	printf("listen success __%d__\n",__LINE__);

	struct sockaddr_in cin; //储存连接成功的客户端信息
	socklen_t addrlen = sizeof(cin);

	int newfd =-1;
	pid_t cpid = -1;

	//从已完成连接的队列头中获取一个客户端信息，生成一个新的文件描述符
	while(1)
	{
		//父进程只负责连接
		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__);

		 //能运行到当前位置，则代表有客户端连接成功，且生成一个新的newfd
		 //此时需要创建一个子进程，只负责与客户端进行交互
		 cpid = fork();
		 if(cpid > 0)
		 {
			 close(newfd);
		 }
		 else if(cpid == 0)  //子进程，与客户端交互
		 {
			 close(sfd);
			 deal_cli_msg(newfd,cin);
			 close(newfd);

			 exit(0);
			 //子进程只能负责交互，运行到当前位置则代表客户端已经退出，需要退出子进程
		 }
		 else
		 {
			 ERR_MSG("fork");
			 return -1;
		 }
	}
	//关闭文件描述符
	close(sfd);
	return 0;
}

int deal_cli_msg(int newfd,struct sockaddr_in cin)
{
	ssize_t res = 0;
	char buf[128] = "";

	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("[%s : %d] newfd = %d 客户端下线__%d__\n",\
				 inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd,__LINE__);
			 break;
		}
		 printf("[%s : %d] newfd = %d %s__%d__\n",\
				 inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd,buf,__LINE__);
			

		//发送给客户端:send
		strcat(buf,"*_*");
		if(send(newfd,buf,sizeof(buf),0) < 0)
		{
			ERR_MSG("send");
			return -1;
		}
		printf("发送成功\n");
	}


	return 0;
}

多线程并发服务器代码

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

#define IP "192.168.0.116"
#define PORT 8888

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

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;
	}

	//允许端口快速被重复使用,当检测到该端口号没有被进程正常占用时候
	//允许端口快速被新的进程占用覆盖
	int reuse = 1;
	if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse)) < 0)
	{
		ERR_MSG("setsockopt");
		return -1;
	}
	printf("允许端口快速重用\n");
	
	//填充服务器的地址信息结构体
	//该真实结构体根据地址制定
	struct sockaddr_in sin;
	sin.sin_family    =  AF_INET;
	sin.sin_port      =  htons(PORT);
	sin.sin_addr.s_addr = inet_addr(IP);

	//绑定服务器的IP和端口：必须绑定:bind
	if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin)) < 0)
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("bind success __%d__\n",__LINE__);

	//将套接字设置为被动监听状态:lieten
	if (listen(sfd, 128) < 0)
	{
		ERR_MSG("listen");
		return -1;
	}
	printf("listen success __%d__\n",__LINE__);

	struct sockaddr_in cin; //储存连接成功的客户端信息
	socklen_t addrlen = sizeof(cin);

	int newfd =-1;
	pthread_t tid;
	struct cli_msg info;

	//从已完成连接的队列头中获取一个客户端信息，生成一个新的文件描述符
	while(1)
	{
		//主线程只负责连接
		//accept函数阻塞之前会先预选一个没有被使用过的文件描述符作为newfd，
		//当解除阻塞的时候，如果预选的newfd没有被使用过，则直接返回预选的文件描述符newfd
		//如果预选的newfd已经被使用过了，则需要重新遍历文件描述符表，拿到一个没有被使用过的文件描述符作为newfd
		 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__);

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

		 //能运行到当前位置，则代表有客户端连接成功
		 //此时需要创建一个分支线程，用于与客户端进行交互使用
		 if(pthread_create(&tid,NULL,deal_cli_msg,(void*)&info) != 0)
		 {
			 fprintf(stderr,"line:%d pthread_create failed\n0",__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;

	ssize_t res = 0;
	char buf[128] = "";

	while(1)
	{
		//对字符串的操作，操作之前或者操作之后要进行清空
		//防止之前的数据对这次操作有干扰
		bzero(buf,sizeof(buf));

		//接收
		res = recv(newfd,buf,sizeof(buf),0);
		if(res < 0)
		{
			ERR_MSG("recv");
			break;
		}
		else if(0 == res)
		{
			 printf("[%s : %d] newfd = %d 客户端下线__%d__\n",\
				 inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd,__LINE__);
			 break;
		}
		 printf("[%s : %d] newfd = %d %s__%d__\n",\
				 inet_ntoa(cin.sin_addr),ntohs(cin.sin_port),newfd,buf,__LINE__);
			

		//发送给客户端:send
		strcat(buf,"*_*");
		if(send(newfd,buf,sizeof(buf),0) < 0)
		{
			ERR_MSG("send");
			break;
		}
		printf("发送成功\n");
	}

	close(newfd);
	pthread_exit(NULL);

	return 0;
}