source: nanoscale/trunk/configparser.c

/* -*- mode: c; c-basic-offset: 4; indent-tabs-mode: nil -*- */
/* ======================================================================
 *  Copyright (c) 2016  HUBzero Foundation, LLC
 * ----------------------------------------------------------------------
 *  See the file "license.terms" for information on usage and
 *  redistribution of this file, and for a DISCLAIMER OF ALL WARRANTIES.
 * ======================================================================
 */
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "server.h" 
static char argBuf[4096];

static int
ParseSwitches(RenderServer *serverPtr)
{
    while (1) {
        const char *flag;
        const char *token;
        char c;

        flag = strtok(NULL, " ");
        if (flag == NULL || flag[0] != '-') {
            TRACE("End parsing switches");
            break;
        }
        TRACE("switch: \"%s\"", flag);
        c = flag[1];
        if ((c == 'i') && (strcmp(flag, "-input") == 0)) {
            int f;

            token = strtok(NULL, " ");
            if (token == NULL || sscanf(token, "%d", &f) != 1) {
                ERROR("Bad -input value");
                return FALSE;
            }
            TRACE("%s: %d", flag, f);
            serverPtr->inputFd = f;
        } else if ((c == 'o') && (strcmp(flag, "-output") == 0)) {
            int f;

            token = strtok(NULL, " ");
            if (token == NULL || sscanf(token, "%d", &f) != 1) {
                ERROR("Bad -output value");
                return FALSE;
            }
            TRACE("%s: %d", flag, f);
            serverPtr->outputFd = f;
        } else if ((c == 'l') && (strcmp(flag, "-logstdout") == 0)) {
            int state;

            token = strtok(NULL, " ");
            if (token == NULL || sscanf(token, "%d", &state) != 1) {
                ERROR("Bad -logstdout value");
                return FALSE;
            }
            TRACE("%s: %d", flag, state);
            serverPtr->logStdout = state;
        } else if ((c == 'l') && (strcmp(flag, "-logstderr") == 0)) {
            int state;

            token = strtok(NULL, " ");
            if (token == NULL || sscanf(token, "%d", &state) != 1) {
                ERROR("Bad -logstderr value");
                return FALSE;
            }
            TRACE("%s: %d", flag, state);
            serverPtr->logStderr = state;
        } else if ((c == 'c') && (strcmp(flag, "-combinelogs") == 0)) {
            int state;

            token = strtok(NULL, " ");
            if (token == NULL || sscanf(token, "%d", &state) != 1) {
                ERROR("Bad -combinelogs value");
                return FALSE;
            }
            TRACE("%s: %d", flag, state);
            serverPtr->combineLogs = state;
        } else {
            ERROR("unknown switch \"%s\"", flag);
            return FALSE;
        }
    }

    return TRUE;
}

char *MergeArgs(int numCmdArgs, char *const *cmdArgs)
{
    int i;
    int pos = 0;
    memset(argBuf, 0, sizeof(argBuf));
    for (i = 0; i < numCmdArgs; i++) {
        strcpy(argBuf + pos, cmdArgs[i]);
        pos += strlen(cmdArgs[i]) + 1;
        if (i < numCmdArgs - 1) {
            argBuf[pos-1] = ' ';
        }
    }
    return argBuf;
}

void FreeArgs(char *cmd)
{
    memset(argBuf, 0, sizeof(argBuf));
    return;
}

int
ParseServersFile(const char *fileName, int *numServers, RenderServer **serverTable)
{
    char line[4096];
    char *token;
    FILE *filep;
    int ret;
    RenderServer *entry;

    filep = fopen(fileName, "r");
    if (filep == NULL)
        return FALSE;

    entry = NULL;
    while (fgets(line, sizeof(line), filep) != NULL) {
        if (line[0] == '#' || line[0] == '\n')
            continue;
        TRACE("line: \"%s\"\n", line);
        token = strtok(line, " ");
        if (strncmp(token, "server", 6) == 0) {
            char name[256];
            int port;

            token = strtok(NULL, " ");
            ret = sscanf(token, "%s", name);
            if (ret != 1) {
                ERROR("Bad server line: \"%s\"", line);
                fclose(filep);
                return FALSE;
            }
            token = strtok(NULL, " ");
            ret = sscanf(token, "%d", &port);
            if (ret != 1) {
                ERROR("Bad server line: \"%s\"", line);
                fclose(filep);
                return FALSE;
            }
            name[sizeof(name)-1] = '\0';
            if (*numServers+1 >= MAX_SERVERS) {
                ERROR("Too many servers in configuration");
                fclose(filep);
                return FALSE;
            }
            (*numServers)++;
            entry = NewServer(name);
            serverTable[*numServers-1] = entry;
            entry->port = port;
            INFO("Registering server: \"%s\" on port %d", name, port);
            if (!ParseSwitches(entry)) {
                ERROR("Failed to parse switches");
                fclose(filep);
                return FALSE;
            }
        } else if (entry != NULL && strncmp(token, "cmd", 3) == 0) {
            char *cmdArg;

             while ((cmdArg = strtok(NULL, " \n")) != NULL) {
                TRACE("Arg %d: \"%s\"", entry->numCmdArgs, cmdArg);
                entry->numCmdArgs++;
                entry->cmdArgs = (char *const *)realloc((char **)entry->cmdArgs, sizeof(char *)*(entry->numCmdArgs+1));
                ((char **)(entry->cmdArgs))[entry->numCmdArgs-1] = strdup(cmdArg);
                ((char **)(entry->cmdArgs))[entry->numCmdArgs] = NULL;
            }
        } else if (entry != NULL && strncmp(token, "env", 3) == 0) {
            char *varName;

            while ((varName = strtok(NULL, " \n")) != NULL) {
                entry->numEnvArgs += 2;
                entry->envArgs = (char *const *)realloc((char **)entry->envArgs, sizeof(char *)*(entry->numEnvArgs));
                ((char **)(entry->envArgs))[entry->numEnvArgs-2] = strdup(varName);
                ((char **)(entry->envArgs))[entry->numEnvArgs-1] = NULL;
                char *value = strtok(NULL, " \n");
                if (value == NULL) {
                    ERROR("Missing value for env var \"%s\"", varName);
                    return FALSE;
                }
                TRACE("Env %s=\"%s\"", varName, value);
                ((char **)(entry->envArgs))[entry->numEnvArgs-1] = strdup(value);
            }
        } else {
            ERROR("Unknown entry: \"%s\"", line);
            fclose(filep);
            return FALSE;
        }
    }
    fclose(filep);
    return TRUE;
}