mini-utils/dos_header.cpp at master · khedoros/mini-utils · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
#include<iostream>
#include<fstream>
#include<stdint.h>
#include<assert.h>
#include<vector>
using namespace std;

typedef struct {
    char magic[2];
    uint16_t last_page_bytes;
    uint16_t total_pages;
    uint16_t reloc_entries;
    uint16_t header_para_size;
    uint16_t min_paras_after_code;
    uint16_t max_paras_after_code;
    uint16_t stack_seg_offset;
    uint16_t initial_sp;
    uint16_t chksum;
    uint16_t init_ip;
    uint16_t init_cs;
    uint16_t reloc_offset;
    uint16_t overlay_num;
    uint8_t poss_id[4];
} mz_header;

int main(int argc, char *argv[]) {
    assert(sizeof(mz_header) == 32);
    ifstream in;
    in.open(argv[1]);
    in.seekg(0,ios::end);
    size_t size = in.tellg();
    in.seekg(0,ios::beg);
    cout<<"Filename: "<<argv[1]<<endl<<"Filesize: "<<size<<endl;
    mz_header h;
    in.read(reinterpret_cast<char *>(&h), sizeof(mz_header));
    vector<uint16_t> reloc_table;
    reloc_table.resize(h.reloc_entries*2);
    in.seekg(h.reloc_offset,ios::beg);
    in.read(reinterpret_cast<char *>(&(reloc_table[0])), h.reloc_entries*4);
    printf("Bytes on last page: %04x\nTotal pages: %04x\nRelocation entries: %d\nHeader size in paragraphs: %04x\nMin mem needed after exe: %04x\nMax mem needed after exe: %04x\nSS offset: %04x\nInitial SP: %04x\nChecksum: %04x\nEntry point: %04x:%04x\nOffset of relocation table: %04x\nOverlay number (expect 0): %04x\nCould be ID: %02x %02x %02x %02x\n",h.last_page_bytes,h.total_pages,h.reloc_entries,h.header_para_size,h.min_paras_after_code,h.max_paras_after_code,h.stack_seg_offset,h.initial_sp,h.chksum,h.init_cs,h.init_ip,h.reloc_offset,h.overlay_num,h.poss_id[0],h.poss_id[1],h.poss_id[2],h.poss_id[3]);
    if(h.poss_id[0] == 0x01 && h.poss_id[1] == 0x00 && h.poss_id[2] == 0xfb) {
        std::cout<<"Probably Borland TLink, version "<<h.poss_id[3]/16<<endl;
    }
    cout<<"Relocation table: "<<h.reloc_entries<<" entries"<<endl;
    bool odd=false;
    uint16_t offset=0;
    for(uint16_t off:reloc_table) {
        if(odd)
            printf("%04x:%04x\n",off,offset);
        else
            offset = off;
        odd=!odd;
    }
    return 0;
}