#include "rbt.h"

// global variables
time_t delay = 0;

void help(char* name)
{
    printf("usage: %s <CMD> [OPTIONS] [ARGS]\n", name);
    printf("\n");
    printf("Available commands:\n");

    printf("  create <height> <width> [start] [end] [step] <TEMPLATE>\n"
           "            create the corresponding rainbow tables\n");
    printf("  info <FILENAME> [LIMIT]\n"
           "            display some information about the table from given file\n");

    printf("  stats <height> <width> [n] [N]\n"
           "            compute some information about rainbow tables without computing them\n"
           "            (N=number of hash to compute to estimate\n"
           "            time of single hash, default 1000000)\n");

    printf("  crack <H> <FILENAMES> ...\n"
           "            crack the given hash with the rainbow tables\n");

    printf("  bruteforce <H>\n"
           "            brute force the given hash\n");

    printf("  test ...\n"
           "            development tests (run \"%s test list\" for available tests)\n",
        name);

    printf("\n");
    printf("Available options:\n");
    printf("  --md5                   use md5 hash function (default)\n");
    printf("  --sha1                  use sha1 hash function\n");
    printf("  --LM                    use LAN manager hash function\n");

    printf("  --alphabet <s>          allowed characters for clear text\n");
    printf("  -A <N> / --abc <N>      choose standard alphabet:\n");
    printf("         N=26               abcdefghijklmnopqrstuvwxyz (default)\n");
    printf("         N=26A              ABCDEFGHIJKLMNOPQRSTUVWXYZ\n");
    printf("         N=36               abcdefghijklmnopqrstuvwxyz0123456789\n");
    printf("         N=40               "
           "abcdefghijklmnopqrstuvwxyz0123456789,;:$\n");
    printf("         N=52               "
           "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz\n");
    printf("         N=62               "
           "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"
           "\n");
    printf("         N=66               "
           "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz,;:"
           "$\n");

    printf("  --min-size <n> / -m <n> minimum size of clear text (default: 5)\n");
    printf("  --max-size <n> / -M <n> maximum size of clear text (default: 5)\n");
    printf("  -s <n> / --size <n>     exact size of clear text (default: 5)\n");

    printf("  -d <n> / --delay <n>    delay (seconds) between log messages (default 5)\n");

    printf("  -help / -h              this message\n");
}

int main(int argc, char* argv[])
{
    srand(time(NULL));
    delay = 1;

    if (argc == 1) {
        help(argv[0]);
        return 1;
    }

    char short_options[] = "hA:s:d:m:M:";
    static struct option long_options[] = { { "help", no_argument, 0, 'h' },
        { "md5", no_argument, 0, 0 }, { "sha1", no_argument, 0, 0 }, { "LM", no_argument, 0, 0 },
        { "alphabet", required_argument, 0, 0 }, { "abc", required_argument, 0, 'A' },
        { "min-size", required_argument, 0, 'm' }, { "max-size", required_argument, 0, 'M' },
        { "size", required_argument, 0, 's' }, { "delay", required_argument, 0, 'd' },
        { 0, 0, 0, 0 } };

    init_CFG();

    int opt;
    int long_index;
    while ((opt = getopt_long(argc, argv, short_options, long_options, &long_index)) != -1) {
        switch (opt) {
        case 0:
            if (0 == strcmp("md5", long_options[long_index].name)) {
                CFG.hash_nb = 0;
            } else if (0 == strcmp("sha1", long_options[long_index].name)) {
                CFG.hash_nb = 1;
            } else if (0 == strcmp("LM", long_options[long_index].name)) {
                CFG.hash_nb = 2;
            } else if (0 == strcmp("alphabet", long_options[long_index].name)) {
                CFG.alphabet_size = strlen(optarg);
                CFG.alphabet = realloc(CFG.alphabet, CFG.alphabet_size + 1);
                strcpy(CFG.alphabet, optarg);
            } else {
                printf("Unknown option...\n");
                return 1;
            }
            break;
        case 's':
            CFG.min_size = atoi(optarg);
            CFG.max_size = atoi(optarg);
            break;
        case 'm':
            CFG.min_size = atoi(optarg);
            break;
        case 'M':
            CFG.max_size = atoi(optarg);
            break;
        case 'A':
            if (0 == strcmp(optarg, "26")) {
                CFG.alphabet_size = 26;
                CFG.alphabet = realloc(CFG.alphabet, 27);
                strcpy(CFG.alphabet, "abcdefghijklmnopqrstuvwxyz");
            } else if (0 == strcmp(optarg, "26A")) {
                CFG.alphabet_size = 26;
                CFG.alphabet = realloc(CFG.alphabet, 27);
                strcpy(CFG.alphabet, "ABCDEFGHIJKLMNOPQRSTUVWXYZ");
            } else if (0 == strcmp(optarg, "36A")) {
                CFG.alphabet_size = 36;
                CFG.alphabet = realloc(CFG.alphabet, 37);
                strcpy(CFG.alphabet, "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ");
            } else if (0 == strcmp(optarg, "36")) {
                CFG.alphabet_size = 36;
                CFG.alphabet = realloc(CFG.alphabet, 37);
                strcpy(CFG.alphabet, "0123456789abcdefghijklmnopqrstuvwxyz");
            } else if (0 == strcmp(optarg, "40")) {
                CFG.alphabet_size = 40;
                CFG.alphabet = realloc(CFG.alphabet, 41);
                strcpy(CFG.alphabet, "0123456789abcdefghijklmnopqrstuvwxyz,;:$");
            } else if (0 == strcmp(optarg, "52")) {
                CFG.alphabet_size = 52;
                CFG.alphabet = realloc(CFG.alphabet, 53);
                strcpy(CFG.alphabet, "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz");
            } else if (0 == strcmp(optarg, "62")) {
                CFG.alphabet_size = 62;
                CFG.alphabet = realloc(CFG.alphabet, 63);
                strcpy(CFG.alphabet,
                    "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrst"
                    "uvwxyz");
            } else if (0 == strcmp(optarg, "66")) {
                CFG.alphabet_size = 66;
                CFG.alphabet = realloc(CFG.alphabet, 67);
                strcpy(CFG.alphabet,
                    "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrst"
                    "uvwxyz,;:$");
            } else {
                printf("unknown alphabet type: %s\n", optarg);
                exit(2);
            }
            break;

        case 'd':
            delay = atoi(optarg);
            break;
        case 'h':
            help(argv[0]);
            return 0;
        default: /* '?' */
            // TODO
            return 1;
        }
    }

    update_CFG();
    if (delay == 0) {
        delay = INT_MAX;
    }

    if (0 == strcmp(argv[optind], "help")) {
        help(argv[0]);
        return 0;
    } else if (0 == strcmp(argv[optind], "bruteforce")) {
        return main_brute_force(argc - optind - 1, argv + optind + 1);
    } else if (0 == strcmp(argv[optind], "info")) {
        return main_info_table(argc - optind - 1, argv + optind + 1);
    } else if (0 == strcmp(argv[optind], "create")) {
        return main_create_table(argc - optind - 1, argv + optind + 1);
    } else if (0 == strcmp(argv[optind], "test")) {
        return main_test(argc - optind - 1, argv + optind + 1);
    } else if (0 == strcmp(argv[optind], "crack")) {
        return main_crack(argc - optind - 1, argv + optind + 1);
    } else if (0 == strcmp(argv[optind], "stats")) {
        return main_stats(argc - optind - 1, argv + optind + 1);
    } else {
        printf("*** unknown command: '%s'\n", argv[optind]);
        return 2;
    }
}

// vim: textwidth=100