C++网络聊天室

单线程模式

客户端基于poll监听标准输入和socket， 服务端基于poll监听socket请求和socket上传来的信息。

Client

#include<poll.h>
#include<stdio.h>
#include<stdlib.h>
#include<netinet/in.h>
#include<sys/socket.h>
#include<assert.h>
#include<string.h>
#include<sys/types.h>
#include<unistd.h>
#include<errno.h>
#include<fcntl.h>
#include<arpa/inet.h>
const int BUF_SIZE = 1024;

void set_pfd(pollfd& pfd, int fd, int event, int revent) {
   
	pfd.fd = fd;
	pfd.events = event;
	pfd.revents = revent;
}

int main() {
   
	int port;
	char ip[40];
	printf("Input port num and ip address !\n");
	scanf("%d %s", &port, ip);
	sockaddr_in server_address;
	bzero(&server_address, sizeof(server_address));
	server_address.sin_family = AF_INET;
	server_address.sin_port = htons(port);
	inet_pton(AF_INET, ip, &server_address.sin_addr);

	int sockfd = socket(AF_INET, SOCK_STREAM, 0);
	assert(sockfd >= 0);

	int re = connect(sockfd, (sockaddr*)&server_address, sizeof(server_address));
	if (re < 0) {
   
		printf("Connect failure\n");
		close(sockfd);
		return -1;
	}

	pollfd fds[2];
	set_pfd(fds[0], 0, POLLIN, 0);
	set_pfd(fds[1], sockfd, POLLIN | POLLRDHUP, 0);

	int pipefd[2];
	re = pipe(pipefd);
	assert(re != -1);

	char* read_buf[BUF_SIZE];

	printf("Begin !\n");
	while (true) {
   
		re = poll(fds, 2, -1);
		if (re < 0) {
   
			printf("poll failure\n");
			break;
		}

		if (fds[1].revents & POLLIN) {
   
			memset(read_buf, '\0', BUF_SIZE);
			recv(sockfd, read_buf, BUF_SIZE - 1, 0);
			printf("recv msg : %s\n", read_buf);
		}
		else if (fds[1].revents & POLLRDHUP) {
   
			printf("Server close the connection\n");
			break;
		}

		if (fds[0].revents & POLLIN) {
   
			// 使用零拷贝方法，将标准输入的信息输入到socket上发送到服务端
			splice(0, NULL, pipefd[1], NULL, 32768, SPLICE_F_MORE | SPLICE_F_MOVE);
			splice(pipefd[0], NULL, sockfd, NULL, 32768, SPLICE_F_MORE | SPLICE_F_MOVE);
		}
	}

	close(sockfd);
	return 0;
}

Server

#include<poll.h>
#include<stdio.h>
#include<stdlib.h>
#include<netinet/in.h>
#include<sys/socket.h>
#include<assert.h>
#include<string.h>
#include<sys/types.h>
#include<unistd.h>
#include<errno.h>
#include<fcntl.h>
const int BUF_SIZE = 1024;
const int MAX_USER_NUM = 10;
const int MAX_FDS = 65535;
// 现有游客数计数器
int user_count = 0;
struct user
{
   
	sockaddr_in addr;
	char* write;
	char read[BUF_SIZE];
};

void set_pfd(pollfd& pfd, int fd, int event, int revent);
int setnonblocking(int fd);
int main(int argc, char **argv) {
   
	if (argc <= 1) {
   
		printf("请输入端口号!\n");
		return -1;
	}
	int port = atoi(argv[1]);

	sockaddr_in address;
	bzero(&address, sizeof(address));
	address.sin_family = AF_INET;
	address.sin_port = htons(port);
	address.sin_addr.s_addr = htonl(INADDR_ANY);

	int sockfd = socket(AF_INET, SOCK_STREAM, 0);
	assert(sockfd >= 0);
	int ret = bind(sockfd, (sockaddr*)&address, sizeof(address));
	assert(ret != -1);

	ret = listen(sockfd, MAX_USER_NUM