server.c

来自「QT 做的俄罗斯方快 基于LINUX下的」· C语言 代码 · 共 71 行

#include <stdio.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <fcntl.h>
#include <errno.h>

#include "server.h"
#include "block.h"

int init_net_server(char * ip_addr)
{
    int listenfd;
    struct sockaddr_in servaddr;

    if ((listenfd = socket(AF_INET, SOCK_STREAM, 0)) < 0) {
        perror("socket");
        return -1;
    }

    servaddr.sin_family = AF_INET;
    servaddr.sin_addr.s_addr = inet_addr(ip_addr);
    servaddr.sin_port = htons(9877);
    if (bind(listenfd, (struct sockaddr *)&servaddr, sizeof(servaddr)) < 0) {
        perror("bind");
        goto close_fd;
    }
    //setsockopt();

    if (listen(listenfd, NET_MAX_CONNECT) < 0) {
        perror("listen");
        goto close_fd;
    };

    return listenfd;

close_fd:
    close(listenfd);
    return -1;
}

int select_accept(int listenfd)
{
    int connfd;
    struct timeval timeout;
    struct sockaddr_in cliaddr;
    socklen_t clilen;
    fd_set  fdset;
    int status;

    FD_ZERO(&fdset);
    FD_SET(listenfd, &fdset);
    timeout.tv_sec = 0;
    timeout.tv_usec = 0;
    if (select(listenfd + 1, &fdset, NULL, NULL, &timeout) == -1)
        return -1;

    if (!FD_ISSET(listenfd, &fdset))
        return -1;

    if ((connfd = accept(listenfd, (struct sockaddr *)&cliaddr, &clilen)) < 0)
        return -1;

    do {
        read(connfd, (char *)&status, sizeof(int));
    } while (status != PLAYER_JOIN);

    fcntl(connfd, F_SETFL, O_NONBLOCK);

    return connfd;
}