Mercurial > hg > terminal-chess / file revision

/*
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.
 *
 * Copyright 2014 Mike Becker. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are met:
 *
 *   1. Redistributions of source code must retain the above copyright
 *      notice, this list of conditions and the following disclaimer.
 *
 *   2. Redistributions in binary form must reproduce the above copyright
 *      notice, this list of conditions and the following disclaimer in the
 *      documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 *
 */

#include "terminal-chess.h"
#include "game.h"
#include <ncurses.h>
#include <errno.h>
#include <string.h>

static int server_open(Server *server, char *port) {
    printw("\nListening for client...\n");
    refresh();
    if (net_create(server, port)) {
        addstr("Server creation failed");
        return 1;
    }

    if (net_listen(server)) {
        addstr("Listening for client failed");
        return 1;
    }
    
    return 0;
}

static int server_handshake(Client *client) {
    net_send_code(client->fd, NETCODE_VERSION);
    if (net_recieve_code(client->fd) != NETCODE_VERSION) {
        addstr("Client uses an incompatible software version.");
        return 1;
    }

    addstr("Client connected - transmitting gameinfo...");
    refresh();
    
    return 0;
}

int server_run(Settings *settings) {
    Server server;
    
    dump_gameinfo(&(settings->gameinfo));
    GameState continuegame;
    gamestate_init(&continuegame);
    if (settings->continuepgn) {
        // preload PGN data before handshake
        FILE *pgnfile = fopen(settings->continuepgn, "r");
        if (pgnfile) {
            int result = read_pgn(pgnfile, &continuegame,
                &(settings->gameinfo));
            fclose(pgnfile);
            if (result != EXIT_SUCCESS) {
                addstr("Invalid PGN file content.\n");
                return EXIT_FAILURE;
            }
            if (!is_game_running(&continuegame)) {
                addstr("Game has ended. Use -S to analyze it.\n");
                return EXIT_FAILURE;
            }
            addch('\n');
            dump_moveinfo(&continuegame);
            addch('\n');
        } else {
            printw("Can't read PGN file (%s)\n", strerror(errno));
            return EXIT_FAILURE;
        }
    }
    
    if (server_open(&server, settings->port)) {
        net_destroy(&server);
        return EXIT_FAILURE;
    }
    
    if (server_handshake(server.client)) {
        net_destroy(&server);
        return EXIT_FAILURE;
    }

    int fd = server.client->fd;
    if (settings->continuepgn) {
        // Continue game, send PGN data
        uint16_t mc = 0;
        MoveList *movelist = continuegame.movelist;
        while (movelist) {
            mc++;
            movelist = movelist->next;
        }
        
        Move* moves = calloc(mc, sizeof(Move));
        
        movelist = continuegame.movelist;
        mc = 0;
        while (movelist) {
            moves[mc] = movelist->move;
            mc++;
            movelist = movelist->next;
        }
        
        size_t pgndata_size = sizeof(GameInfo)+sizeof(mc)+mc*sizeof(Move);
        char *pgndata = malloc(pgndata_size);
        memcpy(pgndata, &(settings->gameinfo), sizeof(GameInfo));
        memcpy(pgndata+sizeof(GameInfo), &mc, sizeof(mc));
        memcpy(pgndata+sizeof(GameInfo)+sizeof(mc), moves, mc*sizeof(Move));
        free(moves);
        net_send_data(fd, NETCODE_PGNDATA, pgndata, pgndata_size);
        free(pgndata);
    } else {
        // Start new game
        net_send_data(fd, NETCODE_GAMEINFO,
            &(settings->gameinfo), sizeof(GameInfo));
    }
    addstr("\rClient connected - awaiting challenge acceptance...");
    refresh();
    int code = net_recieve_code(fd);
    if (code == NETCODE_ACCEPT) {
        addstr("\rClient connected - challenge accepted.");
        clrtoeol();
        if (settings->continuepgn) {
            game_continue(settings, fd, &continuegame);
        } else {
            game_start(settings, fd);
        }
    } else if (code == NETCODE_DECLINE) {
        addstr("\rClient connected - challenge declined.");
        clrtoeol();
    } else if (code == NETCODE_CONNLOST) {
        addstr("\rClient connected - but gave no response.");
        clrtoeol();
    } else {
        addstr("\rInvalid client response");
        clrtoeol();
        
        net_destroy(&server);
        return EXIT_FAILURE;
    }
    
    net_destroy(&server);
    return EXIT_SUCCESS;
}