5.18作业

//多进程并发服务器
#include"../headers/header.h"
#define ERR_MSG(msg) do{\
	printf("line:%d\n",__LINE__);\
	perror(msg);\
}while(0)

#define PORT 8888
#define IP "192.168.10.218"
int deal_cli_msg(int newfd, struct sockaddr_in cin);

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

int main(int argc, const char *argv[])
{
    //捕获17号信号
    if(signal(SIGCHLD, zombie_handler) == SIG_ERR)
    {
        ERR_MSG("signal");
        return -1;
    }

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

    //功能：允许端口快速被重用，快速被复用
    int reuse =1;
    if(setsockopt(sfd, SOL_SOCKET, SO_REUSEADDR, &reuse, sizeof(reuse)) < 0)
    {
        ERR_MSG("setsockopt");
        return -1;
    }

    //填充服务器自身的地址信息结构体，
    //真实的地址信息结构体根据地址族制定 AF_INET: man 7 ip
    struct sockaddr_in sin;
    sin.sin_family      = AF_INET;      //必须填充AF_INET;
    sin.sin_port        = htons(PORT);  //端口号的网络字节序 1024~49151
    sin.sin_addr.s_addr = inet_addr(IP);    //ifconfig查看本机IP

    //将IP地址和端口号绑定到指定的套接字文件描述符上
    if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
    {
        ERR_MSG("bind");
        return -1;
    }
    printf("bind success __%d__\n", __LINE__);

    //功能：将通讯套接字设置为被动监听状态。用于监听是否有客户端连接;
    //listen函数会让内核维护两个队列：未完成连接的队列， 已完成连接的队列
    if(listen(sfd, 128) < 0)
    {
        ERR_MSG("listen");
        return -1;
    }
    printf("listen success __%d__\n", __LINE__);

    //从已完成连接的队列里面，获取一个客户端信息，生成一个新的文件描述符，
    //该文件描述符才是与客户端通信的文件描述符。
    struct sockaddr_in cin;     //存储客户端信息
    socklen_t len = sizeof(cin);
    int newfd = -1;
    pid_t cpid = 0;

    while(1)
    {
        //父进程只负责连接
        newfd = accept(sfd, (struct sockaddr*)&cin, &len);
        if(newfd < 0)
        {
            perror("accept");
            return -1;
        }
        printf("[%s:%d] 客户端连接成功 newfd=%d __%d__\n",\
                inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), newfd, __LINE__);


        //能运行到当前位置，则代表有客户端连接成功，则需要创建一个子进程用于与客户端交互
        if(fork() == 0)     //子进程只负责与客户端交互
        {
            close(sfd);     //监听用的文件描述符，在子进程中没有作用
            deal_cli_msg(newfd, cin);

            //退出子进程，子进程只负责交互，不负责连接
            exit(0);
        }

        close(newfd);       //在父进程中，newfd没有用，所以关闭
    }


    if(close(sfd) < 0)
    {
        ERR_MSG("close");
        return -1;
    }

    return 0;
}


int deal_cli_msg(int newfd, struct sockaddr_in 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");
            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__);

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


    if(close(newfd) < 0)
    {
        ERR_MSG("close");
        return -1;
    }


    return 0;
}

//多线程并发服务器
#include"../headers/header.h"
#define ERR_MSG(msg) do{\
	printf("line:%d\n",__LINE__);\
	perror(msg);\
}while(0)

#define PORT 8888
#define IP "192.168.10.218"
struct climsg{
	int newfd;
	struct sockaddr_in cin;
};

void* deal_cli_msg(void* arg);

void zombie_handler(int sig)
{
	while(waitpid(-1,NULL,WNOHANG)>0)
		return;
}
int main(int argc, const char *argv[])
{
	//捕获17号信号
	if(signal(17,zombie_handler)==SIG_ERR)
	{
		ERR_MSG("signal");
		return -1;
	}
	//socket
	//创建流式套接字
	int sfd=socket(AF_INET,SOCK_STREAM,0);
	if(sfd<0)
	{
		ERR_MSG("socket");
		return -1;
	}
	printf("socket create success sfd=%d __%d__\n",sfd,__LINE__);

	//功能：允许端口快速被重用
	//setsockopt
	int reuse=1;
	if(setsockopt(sfd,SOL_SOCKET,SO_REUSEADDR,&reuse,sizeof(reuse))<0)
	{
		ERR_MSG("setsockopt");
		return -1;
	}

	//填充服务器自身的地址信息结构体
	//真实的地址信息结构体根据地址族制定 AF_INET:man 7 ip
	struct sockaddr_in sin;//server-info
	sin.sin_family=AF_INET;
	sin.sin_port=htons(PORT);
	sin.sin_addr.s_addr=inet_addr(IP);

	//bind
	//将IP地址和端口号绑定到指定的套接字文件描述符上
	if(bind(sfd,(struct sockaddr*)&sin,sizeof(sin))<0)
	{
		ERR_MSG("bind");
		return -1;
	}	
	printf("bind success __%d__\n",__LINE__);

	//listen
	//功能：将通讯套接字设置为被动监听状态。用于监听是否有客户端连接
	if(listen(sfd,128)<0)
	{
		ERR_MSG("listen");
		return -1;
	}
	printf("listen success __%d__\n",__LINE__);

	//accept
	//从已完成连接的队列里面，获取一个客户端信息，生成一个新的文件描述符，
	//该文件描述符才是与客户端通信的文件描述符。
	struct sockaddr_in cin;
	socklen_t len = sizeof(cin);
	int newfd=-1;
	pthread_t tid;
	struct climsg info;
	while(1)
	{
		newfd=accept(sfd,(struct sockaddr*)&cin,&len);
		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,&info)<0)
		{
			fprintf(stderr,"pthread create failed\n");
			return -1;
		}
		pthread_detach(tid);
	}
	if(close(sfd<0)<0)
	{
		ERR_MSG("close");
		return -1;
	}
	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 __%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__);

        //发送数据
        strcat(buf, "*_*");
        if(send(newfd, buf, sizeof(buf), 0) < 0)
        {
            ERR_MSG("send");
            break;
        }
        printf("发送成功\n");
    }
	if(close(newfd)<0)
	{
		ERR_MSG("close");
	}
	return 0;
}