Function emitPE

src/e9patch/e9pe.cpp:283–519

* Emit the (modified) PE executable. */

 * Emit the (modified) PE executable.
 */
size_t emitPE(Binary *B, const MappingSet &mappings, size_t mapping_size)
{
    uint8_t *data = B->patched.bytes;
    size_t size = B->patched.size;
    uint32_t mapping_align = (uint32_t)WINDOWS_VIRTUAL_ALLOC_SIZE;
    stat_input_file_size = size;
    size = ALIGN(size, mapping_align);
     
    // Emit all mappings:
    PIMAGE_OPTIONAL_HEADER64 opt_hdr = B->pe.opt_hdr;
    uint32_t size_of_image = opt_hdr->SizeOfImage;
    for (auto mapping: mappings)
    {
        uint8_t *base = data + size;
        mapping->offset = (off_t)size;
        flattenMapping(B, base, mapping, /*int3=*/0xcc);
        size += mapping->size;
    }

    // Emit the loader:
    uint32_t section_align = opt_hdr->SectionAlignment;
    uint32_t file_align = B->pe.opt_hdr->FileAlignment;
    size = ALIGN(size, file_align);
    off_t loader_offset = (off_t)size;
    struct e9_config_s *config = (struct e9_config_s *)(data + size);
    size_t config_offset = size;
    size += sizeof(struct e9_config_s) + sizeof(struct e9_config_pe_s);
    const char magic[]   = "E9PATCH";
    memcpy(config->magic, magic, sizeof(magic));
    const char version[] = STRING(VERSION);
    memcpy(config->version, version, sizeof(version));
    config->base  = (intptr_t)size_of_image;

    std::vector<Bounds> bounds;
    config->maps[1] = (uint32_t)(size - config_offset);
    for (auto mapping: mappings)
    {
        stat_num_physical_bytes += mapping->size;
        off_t offset_0 = mapping->offset;
        for (; mapping != nullptr; mapping = mapping->merged)
        {
            bounds.clear();
            getVirtualBounds(mapping, WINDOWS_VIRTUAL_ALLOC_SIZE, bounds);
            bool r = ((mapping->prot & PROT_READ) != 0);
            bool w = ((mapping->prot & PROT_WRITE) != 0);
            bool x = ((mapping->prot & PROT_EXEC) != 0);
            for (const auto b: bounds)
            {
                intptr_t base = mapping->base + b.lb;
                size_t len    = b.ub - b.lb;
                off_t offset  = offset_0 + b.lb;

                debug("load trampoline: MapViewOfFileEx(addr=" ADDRESS_FORMAT
                    ",size=%zu,offset=+%zu,prot=%c%c%c)",
                    ADDRESS(base), len, offset_0, (r? 'r': '-'),
                    (w? 'w': '-'), (x? 'x': '-'));
                stat_num_virtual_bytes += len;