src/main.c

/*
 * 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 <string.h>

int get_settings(int argc, char **argv, Settings *settings) {
    char *valid;
    
    for (char opt ; (opt = getopt(argc, argv, "hp:")) != -1 ;) {
        switch (opt) {
            case 'p':
                if (strtol(optarg, &valid, 10) < 1025 || *valid != '\0') {
                    fprintf(stderr,
                        "Invalid port number (%s) - choose a number > 1024\n",
                        optarg);
                    return 1;
                } else {
                    settings->port = optarg;
                }
                break;
            case 'h':
            case '?':
                settings->printhelp = 1;
                break;
        }
    }
    
    if (optind == argc - 1) {
        settings->serverhost = argv[optind];
    } else if (optind < argc - 1) {
        fprintf(stderr, "Too many arguments\n");
        return 1;
    }
    
    return 0;
}

Settings default_settings() {
    Settings settings;
    memset(&settings, 0, sizeof(Settings));
    settings.port = "27015";
    return settings;
}

int cleanup(Settings *settings, int exitcode) {
    if (settings->server) {
        if (net_destroy(settings->server)) {
            perror("Server shutdown failed");
        }
    }
    
    return exitcode;
}

int main(int argc, char **argv) {
    
    Settings settings = default_settings();
    get_settings(argc, argv, &settings);
    
    if (settings.printhelp) {
        printf(
            "Usage: %s [OPTION]... [HOST]\n"
            "Starts/joins a network chess game\n\n"
            "  -h            This help page\n"
            "  -p            TCP port to use (default: 27015)\n",
            argv[0]);
        return EXIT_SUCCESS;
    }
    
    Server server;
    settings.server = &server;
    
    if (is_server(&settings)) {
        if (net_find(&server, settings.serverhost, settings.port)) {
            perror("Can't find server");
            return cleanup(&settings, EXIT_FAILURE);
        }
        
        if (net_connect(&server)) {
            perror("Can't connect to server");
            return cleanup(&settings, EXIT_FAILURE);
        }
    } else {
        if (net_create(&server, settings.port)) {
            perror("Server creation failed");
            return cleanup(&settings, EXIT_FAILURE);
        }
        
        if (net_listen(&server)) {
            perror("Listening for client failed");
            return cleanup(&settings, EXIT_FAILURE);
        }
    }
    
    return cleanup(&settings, EXIT_SUCCESS);
}