/*
 * Command line executable file that outputs
 * some information about the exectuable passed to.
 */

#include <sys/stat.h>

#include <iostream>
#include <cstdlib>

#include "binaryninjacore.h"
#include "binaryninjaapi.h"

using namespace BinaryNinja;
using namespace std;

#ifndef __WIN32__
#include <libgen.h>
#include <dlfcn.h>
string get_plugins_directory()
{
    Dl_info info;
    if (!dladdr((void *)BNGetBundledPluginDirectory, &info))
        return NULL;

    stringstream ss;
    ss << dirname((char *)info.dli_fname) << "/plugins/";
    return ss.str();
}
#else
string get_plugins_directory()
{
    return "C:\\Program Files\\Vector35\\Binary Ninja\\plugins\\";
}
#endif

bool is_file(char *fname)
{
    struct stat buf;
    if (stat(fname, &buf) == 0 && (buf.st_mode & S_IFREG) == S_IFREG)
        return true;

    return false;
}

int main(int argc, char *argv[])
{
    if (argc != 2) {
        cerr << "USAGE: " << argv[0] << " <file_name>" << endl;
        exit(-1);
    }

    char *fname = argv[1];
    if (!is_file(fname)) {
        cerr << "Error: " << fname << " is not a regular file" << endl;
        exit(-1);
    }

    /* In order to initiate the bundled plugins properly, the location 
     * of where bundled plugins directory is must be set. Since
     * libbinaryninjacore is in the path get the path to it and use it to
     * determine the plugins directory */
    SetBundledPluginDirectory(get_plugins_directory());
    InitCorePlugins();
    InitUserPlugins();

    auto bd = BinaryData(new FileMetadata(), fname);
    BinaryView *bv;

    for (auto type : BinaryViewType::GetViewTypes()) {
        if (type->IsTypeValidForData(&bd) && type->GetName() != "Raw") {
            bv = type->Create(&bd);
            break;
        }
    }

    if (bv->GetTypeName() == "Raw"){
        cerr << "Error: Unable to get any other view type besides Raw";
        exit(-1);
    }

    bv->UpdateAnalysis();
    while (bv->GetAnalysisProgress().state != IdleState);

    cout << "Target:   " << fname << endl << endl;
    cout << "TYPE:     " << bv->GetTypeName() << endl;
    cout << "START:    0x" << hex << bv->GetStart() << endl;
    cout << "ENTRY:    0x" << hex << bv->GetEntryPoint() << endl;
    cout << "PLATFORM: " << bv->GetDefaultPlatform()->GetName() << endl;
    cout << endl;

    cout << "---------- 10 Functions ----------" << endl;
    int x = 0;
    for (auto func : bv->GetAnalysisFunctionList()) {
        cout << hex << func->GetStart() << " " << func->GetSymbol()->GetFullName() << endl;
        if (++x >= 10)
            break;
    }
    cout << endl;

    cout << "---------- 10 Strings ----------" << endl;
    x = 0;
    for (auto str_ref : bv->GetStrings()) {
        char *str = (char *)malloc(str_ref.length+1);
        bv->Read(str, str_ref.start, str_ref.length);
        str[str_ref.length] = 0;

        cout << hex << str_ref.start << " ("
             << dec << str_ref.length << ") "
             << str << endl;
        free(str);

        if (++x >= 10)
            break;
    }

    return 0;
}