poll_chatserver.c

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <netdb.h>
#include <poll.h>

#define PORT "9034"

/*
 * Конвертация сокета в строку IP-адреса:
 * addr: struct sockaddr_in или struct sockaddr_in6
 */
const char* inet_ntop2(void* addr, char* buf, size_t size)
{
	struct sockaddr_storage* sas = addr;	
	struct sockaddr_in* sa4;
	struct sockaddr_in6* sa6;

	void* src;

	switch(sas->ss_family) {
		case AF_INET:
			sa4 = addr;
			src = &(sa4->sin_addr);
			break;

		case AF_INET6:
			sa6 = addr;
			src = &(sa6->sin6_addr);
			break;
		default:
			return NULL;
	}

	return inet_ntop(sas->ss_family, src, buf, size);
}

int get_listener_socket(void)
{
	struct addrinfo hints;
	memset(&hints, 0, sizeof(hints));
	hints.ai_family = AF_INET;
	hints.ai_socktype = SOCK_STREAM;
	hints.ai_flags = AI_PASSIVE;
	
	struct addrinfo* ai;
	int gai_res = getaddrinfo(NULL, PORT, &hints, &ai);
	if(gai_res != 0) {
		fprintf(stderr, "pollserver: %s\n", gai_strerror(gai_res));
		exit(1);
	}
	
	int listener_sock;
	struct addrinfo* p;
	for(p = ai; p != NULL; p = p->ai_next) {
		listener_sock = socket(p->ai_family, p->ai_socktype, p->ai_protocol);
		if(listener_sock < 0) {
			continue;
		}
		int yes = 1;
		setsockopt(listener_sock, SOL_SOCKET, SO_REUSEADDR, &yes, sizeof(int));
		
		int bind_res = bind(listener_sock, p->ai_addr, p->ai_addrlen);
		if(bind_res < 0) {
			close(listener_sock);
			continue;
		}

		break;
	}

	if(p == NULL) {
		return -1;
	}

	freeaddrinfo(ai);

	int listen_res = listen(listener_sock, 10);
	if(listen_res == -1) {
		return -1;
	}

	return listener_sock;
}

/* функция добавления нового дескриптора во множество struct pollfd */
void add_to_pfds(struct pollfd** pfds, int newfd, int* fd_count, int* fd_size )
{
	/* если не хватает размера (нет комнаты), увеличиваем: */
	if(*fd_count == *fd_size) {
		*fd_size *= 2;
	       	*pfds = realloc(*pfds, sizeof(**pfds) * (*fd_size));
	}

	(*pfds)[*fd_count].fd = newfd;
	(*pfds)[*fd_count].events = POLLIN;
	(*pfds)[*fd_count].revents = 0;

	(*fd_count)++;
}

/* удаление дескриптора по индексу во множестве */
void del_from_pfds(struct pollfd pfds[], int i, int* fd_count )
{
	/* хак, удаление, когда порядок не важен: */
	/* перемещаем на удаленный индекс хвост массива */
	pfds[i] = pfds[*fd_count-1];
	(*fd_count)--;
}

/*
 * Обработка входящих подключений
 */
void handle_new_connection(int listener, int* fd_count, )