网络编程——用进程实现可供多客户端访问的服务器

#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <string.h>
#include <unistd.h>

#define  PORT   9998

void handl_client(int client_socket)
{
	char buf[1024] = {"asdsadsadsadsa"};
	
	while (1)
	{
		// 读-1 sizeof() 打印
		int ret = read(client_socket, buf, sizeof(buf)-1);
		
		if (-1 == ret)
		{
			perror ("read error");
		}
		
		if (0 == ret)
		{
			printf ("客户端退出\n");
			
			break;
		}
		
		buf[ret] = '\0';
		printf ("收到客户端数据：%s\n", buf);
		int i;
		
		for (i = 0; i < ret-1; i++)
		{
			buf[i] += 'A' - 'a';
		}
		
		// 将数据（小写转大写）回写到客户端，相当于回复
		write(client_socket, buf, ret);
	}	
}

// 监听套接字
int init()
{
	// 创建 绑定 连接
	int listen_socket = socket(AF_INET, SOCK_STREAM, 0);
	
	if(-1 == listen_socket)
	{
		perror("创建套接字失败");
		
		return -1;
	}
	
	struct sockaddr_in addr;
	memset(&addr, 0, sizeof(struct sockaddr_in));
	
	addr.sin_family      = AF_INET;             /* Internet地址族 */
	addr.sin_port        = htons(PORT);         /* 端口号 */
	addr.sin_addr.s_addr = htonl(INADDR_ANY);   /* IP地址, 绑定本地的所有ip地址*/

	// 2、绑定本地的 ip 地址和端口
	// 第一个参数：需要绑定的套接字
	// 第二个参数：是要绑定的ip和端口
	// 第三个参数：第二个参数所占的字节数
	int ret = bind(listen_socket, (const struct sockaddr *)&addr, sizeof(addr));
	
	if(-1 == ret)
	{
		perror("绑定失败");
		
		return -1;
	}
	
	// 3、监听套接字
	ret = listen(listen_socket, 5);
	
	if(-1 == ret)
	{
		perror("监听失败");
		return -1;
	}
	
	return listen_socket;
	
}

// 通信套接字
int myAccept(int listen_socket)
{
	struct sockaddr_in client_addr;
	socklen_t  len = sizeof(client_addr);
	
	// 阻塞型函数
	int client_socket = accept(listen_socket, (struct sockaddr *)&client_addr, &len);
	
	if (-1 == client_socket)
	{
		perror("accept  失败");
		return -1;
	}
	
	printf ("客户端的 ip = %s, 端口 = %d\n", inet_ntoa(client_addr.sin_addr), ntohs(client_addr.sin_port));
	
	return client_socket;
}

// 主函数
// 用多进程实现可多客户端访问的服务器
int main(int argc, char **argv)
{
	// 创建套接字、绑定本地、监听/处理连接请求/建立连接
	int listen_socket = init();
	
	if(-1 == listen_socket)
	{
		perror("创建套接字失败");
		
		return -1;
	}
	
	while(1)
	{
		// 进行通信，函数调用里的accept为阻塞型函数，无客户端连接时一直等待
		int client_socket = myAccept(listen_socket);
		
		// 创建子进程为客户端服务
		pid_t pid = fork();
		
		switch(pid)
		{
			case -1:
				perror("fork失败");
				break;
			case 0:
				close(listen_socket);
				
				// 把通信套接字传过去
				handl_client(client_socket);
				close(client_socket);
				break;
			default:
				close(client_socket);
				break;
		}
	}
	
	close(listen_socket);
	
	return 0;
}