Function main

cli/cli.cpp:45–120

The main entry.

// The main entry.
int main(int argc, char** argv)
{
    // Expecting the input file, out file and the config file.
    if (argc < 3 || argc > 4)
    {
        // Print help.
        cout << *argv << " <input file> <out file> [config file]" << endl;
        return 1;
    };

    // Getting the data.
    char* input_file = argv[1];
    char* out_file = argv[2];
    char* config_file = argc == 4 ? argv[3] : (char*)"config.ini";

    // Needed locals.
    cobf_error ret_err;
    char error[MAX_PATH];
    try
    {
        // Loading the input file.
        cobf obf_file = cobf(input_file);
        if ((ret_err = obf_file.load_pe()) != cobf_error::COBF_NO_ERROR)
        {
            // Error at loading.
            cobf_format_message(ret_err, error, sizeof(error));
            cout << "[-] Error at loading the file: " << error << endl;
            return 1;
        };

        // Log.
        cout << "[+] File loaded successfully." << endl;

        // Process the config file.
        if (!parse_ini_file(config_file, ini_line_handler, &obf_file))
        {
            // Error at processing the config.
            cout << "[-] Error occurred while processing the config file." << endl;
            return 1;
        };

        // Obfuscating it.
        if ((ret_err = obf_file.generate(out_file)) != cobf_error::COBF_NO_ERROR)
        {
            // Error at obfuscating.
            cobf_format_message(ret_err, error, sizeof(error));
            cout << "[-] Error at obfuscating the file: " << error << endl;
            return 1;
        };

        // Log.
        cout << "[+] File obfuscated successfully." << endl;

        // Unloading it.
        if ((ret_err = obf_file.unload_pe()) != cobf_error::COBF_NO_ERROR)
        {
            // Error at unloading.
            cobf_format_message(ret_err, error, sizeof(error));