8.30网络编程作业

多进程并发服务器

#include<myhead.h>


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

#define PORT 6666            //端口号，范围1024~29151
#define IP   "192.168.114.109" //本机IP,ifconfig


//接收发送函数声明
int task(int newfd, struct sockaddr_in cin);

//回收僵尸进程
void handler(int sig)
{
    while(waitpid(-1, NULL, WNOHANG) > 0);
}

int main(int argc, const char *argv[])
{
	//捕获17）SIGCHLD信号，为信号注册新的处理函数
	if(signal(17, 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("创建套接字成功 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");

	//填充地址信息结构体给bind函数绑定，
	//真实的地址信息结构体根据地址族指定 AF_INET man 7 IP
	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


	//绑定服务器的地址信息--->必须绑定 bind
	if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
	    ERR_MSG("bind");
	    return -1;
	}
    printf("服务器地址信息绑定成功\n");


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


	int newfd = -1;
	struct sockaddr_in cin;         //储存客户端地址信息
	socklen_t addrlen = sizeof(cin);    //真实的信息结构体大小

	pid_t pid = -1;     //创建进程号

	//获取一个已经完成的客户端信息，生成一个新的文件描述符。accept
	while(1)
	{
	    //父进程只负责连接
		newfd = accept(sfd, (struct sockaddr*)&cin, &addrlen);
		if(newfd < 0)
		{
		    ERR_MSG("newfd");
			return -1;
		}
		printf("[%s : %d]客户端连接成功 newfd=%d\n",\
				inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), newfd);

		//运行到此代表客户端连接成功，子进程只负责交互
		pid = fork();
		if(0 == pid)            //子进程
		{
		    close(sfd);         //对于子进程而言，sfd无用处
			task(newfd, cin);
			exit(0);            //子进程完成交互任务，退出子进程
		}
		close(newfd);           //此时对于父进程而言，newfd无用处
	}

	close(newfd);
    
	//关闭文件描述符
	if(close(sfd) < 0)
	{
	    ERR_MSG("close");
		return -1;
	}

	return 0;
}


//子进程交互函数，接收发送
int task(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\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);

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

	close(newfd);
    return 0;
}

线程并发服务器

#include<myhead.h>


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

#define PORT 6666            //端口号，范围1024~29151
#define IP   "192.168.114.109" //本机IP,ifconfig


//需要传入到分支线程的参数
struct Node
{
	int newfd;
	struct sockaddr_in cin;
};


void *task1(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;
	}
	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");

	//填充地址信息结构体给bind函数绑定，
	//真实的地址信息结构体根据地址族指定 AF_INET man 7 IP
	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


	//绑定服务器的地址信息--->必须绑定 bind
	if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("bind");
		return -1;
	}
	printf("服务器地址信息绑定成功\n");


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


	int newfd = -1;
	struct sockaddr_in cin;         //储存客户端地址信息
	socklen_t addrlen = sizeof(cin);    //真实的信息结构体大小

	pthread_t tid;        //创建线程号
	struct Node L;

	//获取一个已经完成的客户端信息，生成一个新的文件描述符。accept
	while(1)
    {
		//主进程只负责连接
		newfd = accept(sfd, (struct sockaddr*)&cin, &addrlen);
		if(newfd < 0)
		{
			ERR_MSG("newfd");
			return -1;
		}
		printf("[%s : %d]客户端连接成功 newfd=%d\n",\
				inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), newfd);

		//运行到此代表客户端连接成功，子线程只负责交互
		L.newfd = newfd;
		L.cin = cin;
		if(pthread_create(&tid, NULL, task1, (void *)&L) != 0)
		{
			fprintf(stderr, "pthread_create failed __%d__\n", __LINE__);
			return -1;
		}
		//分离线程，子线程退出后，由内核自动回收
		pthread_detach(tid);

    }

	//关闭文件描述符
	if(close(sfd) < 0)
	{
		ERR_MSG("close");
		return -1;
	}

	return 0;
}


//线程体执行体函数，接收发送
void *task1(void *arg)
{
	int newfd = ((struct Node*)arg)->newfd;
	struct sockaddr_in cin = ((struct Node*)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);

		//发送
		if(send(newfd, buf, sizeof(buf), 0) < 0)
		{
			ERR_MSG("send");
			break;
		}
		printf("发送成功\n");
	}

	close(newfd);

	pthread_exit(NULL);
	return 0;
}