Method load_pe

src/load.cpp:17–135

Load the specified PE from disk.

// Load the specified PE from disk.
cobf_error cobf::load_pe()
{
    // Check if already loaded.
    if (!this->pe_rawf.empty()) return cobf_error::COBF_PE_LOADED;

    // Get an access to the file.
    cobf_error ret_err = cobf_error::COBF_NO_ERROR;
    HANDLE h_file = CreateFileA(
        this->pe_path.c_str(),      // The PE file path.
        GENERIC_READ,               // Access options.
        FILE_SHARE_READ,            // The share mode.
        NULL,                       // Security attributes.
        OPEN_EXISTING,              // Disposition.
        FILE_ATTRIBUTE_NORMAL,      // File attributes.
        NULL                        // The template handle.
        );

    // Check the return.
    if (h_file == INVALID_HANDLE_VALUE)
    {
        // Cannot open the specified file.
        return cobf_error::COBF_CANNOT_OPEN_FILE;
    };

    // Needed local variables.
    PIMAGE_DOS_HEADER dos_hdr;
    PIMAGE_NT_HEADERS nt_hdrs;
    PIMAGE_IMPORT_DESCRIPTOR p_imports;
    size_t imports_size;

    // Getting the size.
    LARGE_INTEGER f_size;
    if (!GetFileSizeEx(h_file, &f_size))
    {
        // Cannot get the file size.
        ret_err = cobf_error::COBF_CANNOT_GET_SIZE;
        goto LOAD_FINISH;
    };

    // Resizing the raw file array.
    this->pe_rawf.resize((size_t)f_size.QuadPart);

    // Reading the file.
    DWORD r_size;
    if (!ReadFile(
        h_file,                     // The PE file handle.
        this->pe_rawf.data(),       // The data of the vector.
        (DWORD)this->pe_rawf.size(),// Size to read.
        &r_size,                    // Read bytes.
        NULL                        // Overlapped struct.
    )) 
    {
        // Cannot read.
        ret_err = cobf_error::COBF_CANNOT_READ_FILE;
        goto LOAD_FINISH;
    };

    // Get the dos header.