【0814作业】多线程并发服务器

1) 代码

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

#define PORT 6666               //1024-49151
#define IP "192.168.1.102"    //ifconfig查看本机ip

//需要传入到分支线程的数据类型
struct CliMsg
{
	int newfd;
	struct sockaddr_in cin;
};

void *deal_cli_msg(void *arg);
void handlr(int sig)
{
	while(waitpid(-1,NULL,WNOHANG) > 0);
}

int main(int argc, const char *argv[])
{
	if(signal(17,handlr) == SIG_ERR)
	{
		ERR_MSG("signal");
		return -1;
	}

	//创建流式套接字
	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(网络端口号的字节序)(端口号存储在2个字节的无符号整数中)
	sin.sin_addr.s_addr    = inet_addr(IP); //本机IP inconfig查看(本机IP地址的字节序)

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

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

	struct sockaddr_in cin;   //存储客户端的地址信息
	socklen_t addrlen = sizeof(cin);
	int newfd = -1;
	pthread_t tid;
	struct CliMsg info;
	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);
		info.newfd = newfd;
		info.cin=cin;
		if(pthread_create(&tid,NULL,deal_cli_msg,&info) != 0)
		{
			ERR_MSG("pthread_create");
			return -1;
		}
		pthread_detach(tid);
	}
	//关闭所有文件描述符
	close(sfd);
	return 0;
}
//线程执行体
void* deal_cli_msg(void *arg)
{
	int newfd = ((struct CliMsg *)arg)->newfd;
	struct sockaddr_in cin = ((struct CliMsg *)arg)->cin;

	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");
			break;
		}
		else if(0 == res)
		{
			printf("[%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 NULL;
		}
		printf("send succuss\n");
	}
	close(newfd);
	pthread_exit(NULL);
}