TCP并发服务器

多进程并发

#include<myhead.h>

int Client_interaction(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()");
			exit(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);

		//發送
		strcat(buf, "*_*");
		if(send(newfd, buf, sizeof(buf), 0) < 0)
		{
			ERR_MSG("send");
			exit(1);
		}
		printf("send success\n");
	}

	//關閉文件描述符
	close(newfd);
	return 0;
}

void handler(int signum)
{
	if(signum == SIGCHLD)
	{
		while(waitpid(-1, NULL, WNOHANG) > 0);
	}
}

int main(int argc, const char *argv[])
{
	//創建流式套接字socket
	int sfd = socket(AF_INET, SOCK_STREAM, 0);
	if(sfd < 0)
	{
		ERR_MSG("socket()");
		exit(1);
	}
	printf("socket create success sfd = %d\n", sfd);

	if(signal(SIGCHLD, handler) == SIG_ERR)
	{
		ERR_MSG("signal()");
		exit(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 = {\
		AF_INET, \
			htons(8989), \
			.sin_addr.s_addr = inet_addr("192.168.114.29")\
	};

	if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("bind()");
		exit(1);
	}

	//將套接字設置爲被動監聽狀態listen
	if(listen(sfd, 128) < 0)
	{
		ERR_MSG("listen()");
		exit(1);
	}
	printf("Listen success\n");

	//獲取一個已經完成的客戶端信息，生成一個新的文件描述符
	int newfd = -1;
	struct sockaddr_in cin;
	socklen_t addrlen = sizeof(cin);
	int pid;

	while(1)
	{
		newfd = accept(sfd, (struct sockaddr*)&cin, &addrlen);
		if(newfd < 0)
		{
			ERR_MSG("newfd");
			exit(1);
		}
		printf("[%s:%d]客戶端連接成功 newfd=%d\n",\
				inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), newfd);

		pid = fork();
		if(0 == pid)
		{
			close(sfd);
			Client_interaction(newfd, cin);
			exit(0);
		}
		close(newfd);
	}
	if(close(sfd) < 0)
	{
		ERR_MSG("close()");
		exit(1);
	}

	return 0;
}

多线程并发

#include<myhead.h>

struct climsg
{
	int newfd;
	struct sockaddr_in cin;
};

void* Client_interaction(void* arg)
{
	struct sockaddr_in cin = ((struct climsg*)arg)->cin;
	int newfd = ((struct climsg*)arg)->newfd;

	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()");
			exit(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);

		//發送
		strcat(buf, "*_*");
		if(send(newfd, buf, sizeof(buf), 0) < 0)
		{
			ERR_MSG("send");
			exit(1);
		}
		printf("send success\n");
	}

	//關閉文件描述符
	close(newfd);
	pthread_exit(NULL);
}

int main(int argc, const char *argv[])
{
	//創建流式套接字socket
	int sfd = socket(AF_INET, SOCK_STREAM, 0);
	if(sfd < 0)
	{
		ERR_MSG("socket()");
		exit(1);
	}
	printf("socket create success 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");

	//綁定服務器的地址信息
	struct sockaddr_in sin = {\
		AF_INET, \
			htons(1314), \
			.sin_addr.s_addr = inet_addr("192.168.114.29")\
	};

	if(bind(sfd, (struct sockaddr*)&sin, sizeof(sin)) < 0)
	{
		ERR_MSG("bind()");
		exit(1);
	}

	//將套接字設置爲被動監聽狀態listen
	if(listen(sfd, 128) < 0)
	{
		ERR_MSG("listen()");
		exit(1);
	}
	printf("Listen success\n");

	//獲取一個已經完成的客戶端信息，生成一個新的文件描述符
	int newfd = -1;
	struct sockaddr_in cin;
	socklen_t addrlen = sizeof(cin);
	pthread_t tid;
	struct climsg info;

	while(1)
	{
		newfd = accept(sfd, (struct sockaddr*)&cin, &addrlen);
		if(newfd < 0)
		{
			ERR_MSG("newfd");
			exit(1);
		}
		printf("[%s:%d]客戶端連接成功 newfd=%d\n",\
				inet_ntoa(cin.sin_addr), ntohs(cin.sin_port), newfd);

		info.newfd = newfd;
		info.cin = cin;
		//创建线程
		if((tid = pthread_create(&tid, NULL, Client_interaction, (void*)&info)) != 0)
		{
			fprintf(stderr, "pthread_create failed __%d__", __LINE__);
			exit(1);
		}
		pthread_detach(tid);
	}
	if(close(sfd) < 0)
	{
		ERR_MSG("close()");
		exit(1);
	}
	return 0;
}