网络编程——TCP协议(多进程 &多线程)

一、TCP多进程并发服务器

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<sys/types.h>
#include<unistd.h>
#include<arpa/inet.h>
#include<sys/socket.h>
#include<sys/wait.h>
#include<arpa/inet.h>
#include<signal.h>
#include<netinet/in.h>
#define ERR_MSG(msg) do{\
	printf("line:%d\n",__LINE__);\
	perror(msg);\
}while(0)
#define PORT 6666//端口号
#define IP "192.168.10.112"
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;
	struct sockaddr_in sin;
	sin.sin_family   =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;
	}
	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)
	{
		//父进程只负责连接
		int 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__);
		//能运行到当前位置，表示客户端连接成功，则需要出创建一个子进程用于与客户端交互
		if(fork()==0)
		{
			close(sfd);
			deal_cli_msg(newfd,cin);
			exit(0);
		}
		close(newfd);
	}
	close(sfd);
	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(res==0)
		{
			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");

	}
	//关闭套接字
	close(newfd);
	return 0;
}

二、TCP多线程并发服务器

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<sys/types.h>
#include<unistd.h>
#include<arpa/inet.h>
#include<sys/socket.h>
#define ERR_MSG(msg) do{\
	printf("line:%d\n",__LINE__);\
	perror(msg);\
}while(0)
#define PORT 6666//端口号
#define IP "192.168.9.124"
int main(int argc, const char *argv[])
{

	//创建套接字
	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;
	struct sockaddr_in sin;
	sin.sin_family   =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;
	}
	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=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__);
	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(res==0)
		{
			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");

	}
	//关闭套接字
	close(sfd);
	close(newfd);
	return 0;
}