cmd.c

#include "cmd.h"

static int cmdParse(const char* str) {
    int i;
    for (i = 0; i < CMD_ENUM_LENGTH; i++) {
        if (mdlStrCmp(str, CMD_STR[i]) || mdlStrCmp(str, CMD_STR_SHORT[i]) ) {
            return i;
        }
    }
    return CMD_NO_CMD;
}

static int cmdDecrypt(const char *pw, const char *path) {
    EncDecryptFileOut out = encDecryptFile(pw, path);

    if( out.r > 0 ) {
        fwrite(out.m, out.m_l, 1, stdout);
        if (CMD_PROMPT_ON && out.m[out.m_l - 1] != '\n') putchar('\n');
    }

    encDecryptFileOutClear(&out);

    return out.r;
}

static int cmdDecryptToFile(const char* in_path, const char* out_path, const char* pw) {
    EncDecryptFileOut out = encDecryptFile(pw, in_path);

    if ( out.r > 0 ) {
        int r = mdlIoToFile(out.m, out.m_l, out_path);
        if ( r != 0 ) return r;
    }

    encDecryptFileOutClear(&out);

    return out.r;
}

static int cmdRun(int cmd, char** arg, size_t arg_l);
static void cmdPrompt() {
    if ( CMD_PROMPT_ON != 1 ) { // to make nesting prompts impossible
        CMD_PROMPT_ON = 1;
        char** sa = 0;
        size_t sa_l = 0;
        do {
            printf("%s", CMD_PROMPT_STR);
            free(sa);
            sa = mdlStrSplitQS(mdlIoGetStr(), &sa_l);
        } while (cmdRun(cmdParse(sa[0]), sa + 1, sa_l - 1) == 0 );
        CMD_PROMPT_ON = 0;
    }
}

static void cmdHelp(const char* arg_cmd) {
    if ( ! arg_cmd ) {
        int i;
        for(i = 0; i < CMD_ENUM_LENGTH; i++) printf("%s\n", CMD_STR_HELP[i]);
        printf("\n%s\n", CMD_HELP_SHELL_HISTORY_WARNING);
    } else {
        printf("%s\n", CMD_STR_HELP[cmdParse(arg_cmd)]);
    }
}

static int error(char* fmt, ...) {
    va_list args;
    va_start(args, fmt);
    vprintf(fmt, args);
    va_end(args);
    return 0;
}

#define error_equal_int(a,b) error("Error: %s command requires exactly %d arguments.\n", CMD_STR[a], b);
#define error_between_int_and_int(a,b,c) error("Error: %s command requires between %d and %d arguments.\n", CMD_STR[a], b, c);

/* returns 0 if everything is successful, 1 if there is a need to exit the program */
static int cmdRun(int cmd, char** arg, size_t arg_l) {
    switch (cmd) {
        case CMD_EXIT:
            return 1;
        case CMD_ENCRYPT_MESSAGE:
            if ( arg_l != 3 ) return error_equal_int(cmd, 3);
            encEncryptMessage(arg[2], arg[1], arg[0]);
            break;
        case CMD_ENCRYPT_FILE:
            if ( arg_l != 3 ) return error_equal_int(cmd, 3);
            encEncryptFile(arg[2], arg[0], arg[1]);
            break;
        case CMD_DECRYPT:
            if ( arg_l != 2 ) return error_equal_int(cmd, 2);
            cmdDecrypt(arg[1], arg[0]);
            break;
        case CMD_DECRYPT_TO_FILE:
            if ( arg_l != 2 ) return error_equal_int(cmd, 2);
            cmdDecryptToFile(arg[0], arg[1], arg[2]);
            break;
        case CMD_PROMPT:
            cmdPrompt();
            break;
        case CMD_HELP:
            if ( arg_l > 1 ) return error_between_int_and_int(cmd, 0, 1);
            cmdHelp(arg[0]);
            break;
    }
    return 0;
}

#define CMD_ARGC 1

int cmdLine(int argc, char **argv) {
    return cmdRun(cmdParse(argv[CMD_ARGC]), argv + 2, argc - 2);
}