以fork方式實現高性能服務器

前期以socket編程實現TCP Server與Client通信:
https://blog.csdn.net/catkin_ws/article/details/107140544
但是會存在問題:
當存在多個客戶端與服務器端通信,則只有一個客戶端能與服務器端通信。
因爲我們的處理程序是放在了循環函數中,只有客戶端退出循環函數,其餘客戶端纔會搶佔與服務器端通信的資源。

以fork方式實現高性能服務器,則每次客戶端尋求與服務器通信時,都建立進程。每個子進程都有客戶端與服務器端的通信

測試:

在這裏插入圖片描述

代碼實現Server:

#include <iostream>
#include <sys/socket.h>
#include <netinet/in.h>//struct
#include <unistd.h>//close head
#include <string.h>//bzero
#define PORT 8111
#define MESSAGE_SIZE 1024
int main(int argc, char* argv[])
{
	int socket_fd;
	int accept_fd;
	int backlog = 10;
	int ret = -1;
	int flag = 1;
	pid_t pid;
	struct sockaddr_in local_addr, remote_addr;
	char in_buf[MESSAGE_SIZE] = {0,};
	
	//create socket
	socket_fd = socket(AF_INET, SOCK_STREAM, 0);
	if (socket_fd == -1)
	{
		std::cout << "file to create socket!" << std::endl;
		exit(-1);
	}
	//set socket options
	ret = setsockopt(socket_fd, 
					SOL_SOCKET, 
					SO_REUSEADDR, 
					&flag, 
					sizeof(flag));
	if (ret == -1)
	{
		std::cout << "failed to set socket options!" << std::endl;
	}
	//set localaddr
	local_addr.sin_family = AF_INET;
	local_addr.sin_port = PORT;
	local_addr.sin_addr.s_addr = INADDR_ANY;//0 RENHE IP DOU LISTEN
	bzero(&(local_addr.sin_zero),8);
	//bind socket
	ret = bind(socket_fd, (struct sockaddr *)&local_addr, sizeof(struct sockaddr));
	if (ret == -1)
	{
		std::cout << "failed to bind addr!" << std::endl;
		exit(-1);
	}
	//listen
	ret = listen(socket_fd, backlog);
	if (ret == -1)
	{
		std::cout << "failed to listen socket!" << std::endl;
		exit(-1);
	}
	for(;;)
	{
		socklen_t addr_len = sizeof(struct sockaddr);
		accept_fd  = accept(socket_fd, 
							(struct sockaddr *) &remote_addr, 
							&addr_len);						
		pid = fork();
		if (pid == 0)
		{
			for (;;)//loop daozhi zhiyou yige client
			{
				ret = recv(accept_fd, (void *)in_buf, MESSAGE_SIZE, 0);
				if (ret == 0)
					break;
				std::cout << "receive: " << in_buf << std::endl;
		
				send(accept_fd, (void*)in_buf, MESSAGE_SIZE,0);
			}
			close(accept_fd);
		}		
	}
	if (pid!=0)//fu jincheng guanbi
	{
		close(socket_fd);
	}
	return 0;
}

代碼實現Client:

#include <iostream>
#include <sys/socket.h>//1connect
#include <sys/types.h>//2connect
#include <netinet/in.h> //struct
#include <string.h>//memset
#include <stdio.h>//gets
#include <unistd.h>//close
#include <arpa/inet.h>//inet
#define PORT 8111
#define MESSAGE_LEN 1024
using namespace std;
int main(int argc, char* argv[])
{
	int socket_fd;
	int ret = -1;
	char sendbuf[MESSAGE_LEN] = {0,};
	char recvbuf[MESSAGE_LEN] = {0,};
	struct sockaddr_in serverAddr;
	socket_fd = socket(AF_INET, SOCK_STREAM, 0);
	if (socket_fd < 0)
	{
		cout << "failed to create socket" << endl;
		exit(-1);
	}
	
	serverAddr.sin_family = AF_INET;
	serverAddr.sin_port = PORT;
	serverAddr.sin_addr.s_addr = inet_addr("127.0.0.1");
	ret = connect(socket_fd, 
				(struct sockaddr *)&serverAddr, 
				sizeof(struct sockaddr));
	if (ret < 0)
	{
		cout << "failed to connect!" << endl;
		exit(-1);
	}
	while(1)
	{
		memset(sendbuf, 0, MESSAGE_LEN);
		gets(sendbuf);
		ret = send(socket_fd, sendbuf, strlen(sendbuf), 0);
		if (ret <= 0)
		{
			cout << "failed to send data!" << endl;
			break;		
		}
		//guanbi 1 kill -9 1111      2 duibi
		if (strcmp(sendbuf, "quit") == 0)
		{
			break;
		}
		ret = recv(socket_fd, recvbuf, MESSAGE_LEN, 0);
		recvbuf[ret] = '\0';
		cout << "recv:" << recvbuf << endl;
	}
	close(socket_fd);
	
	
	return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章