laboratorioMalware.cpp

#include "funciones.hpp"
#define _io     ios_base::sync_with_stdio(false); cin.tie(NULL);


int main(int argc, char const *argv[]){_io

    vector<string> malwares; //vector de nombre de archivos de malware
    vector<string> ensambladores; //vector de nombre de archivos ensamblador
    string herramienta; //herramienta a usar

    //Obtenemos los archivos dentro de la carpeta
    obtenerArchivos(malwares);

    //?obtencion de hashes
    system("printf \"ID          SHA1\\n\" > hashes.txt");
    for (int i=0; i<malwares.size(); i++)
    {
        //imprime el id: mi
        const string imprime=" printf \"m%-4d\" "+to_string(i)+string(" >> hashes.txt");
        system(imprime.c_str()); //ejecuta el comando
        SHA1(malwares[i]); //obtiene su hash SHA1
    }
    //por si no existe la carpeta
    if (mkdir("Ensambladores", 0777)== -1)
        cout << "Ya existe la carpeta Ensambladores o hubo un error"<<endl;
    if (mkdir("Ensambladores/Objdump", 0777)== -1)
        cout << "Ya existe la carpeta Objdump o hubo un error"<<endl;
    if (mkdir("Ensambladores/Radare2", 0777)== -1)
        cout << "Ya existe la carpeta Radare2 o hubo un error"<<endl;
    
    //?Obtencion de ensambladores
    herramienta="Objdump";
    for (int m=0; m<malwares.size(); m++)
    {
        //obtiene el ensamblador
        obtenerEnsamblador(m,malwares[m],herramienta,ensambladores);
    }

    herramienta="Radare2";
    for (int m=0; m<malwares.size(); m++)
    {
        //obtiene el ensamblador
        obtenerEnsamblador(m,malwares[m],herramienta,ensambladores);
    }
    
/*     for (auto &ensamblador : ensambladores)
    {
        cout << ensamblador<<endl;
    } */
    
    //obtencion de n-grams
    if (mkdir("Resultados", 0777)== -1)
        cout << "Ya existe la carpeta Resultados o hubo un error"<<endl;
    if (mkdir("Resultados/Objdump", 0777)== -1)
        cout << "Ya existe la carpeta Objdump o hubo un error"<<endl;
    if (mkdir("Resultados/Radare2", 0777)== -1)
        cout << "Ya existe la carpeta Radare2 o hubo un error"<<endl;

    herramienta="Objdump";
    for (int dm=0; dm<ensambladores.size()/2; dm++)
    {
        //obtiene los ngrams
        getNGrams(dm,ensambladores[dm],herramienta);
    }

    herramienta="Radare2";
    for (int dm=ensambladores.size()/2; dm<ensambladores.size(); dm++)
    {
        //obtiene los ngrams
        getNGrams(dm-(ensambladores.size()/2),ensambladores[dm],herramienta);
    }

    return 0;
}