120 lines
5.1 KiB
JavaScript
120 lines
5.1 KiB
JavaScript
// To make the binary (DOS_STUB_PROGRAM):
|
|
// $ cd tools/dos-stub
|
|
// $ nasm -f bin -o dos-stub.bin dos-stub.asm
|
|
// $ node -e "console.log([].map.call(fs.readFileSync('tools/dos-stub/dos-stub.bin'), (v)=>`0x${Buffer.from([v]).toString('hex')}`).join(','))"
|
|
//
|
|
// NOTE: the original dos-stub.asm program and the bit code in DOS_STUB_PROGRAM are under the 0-BSD license.
|
|
import ImageDataDirectoryArray from '../format/ImageDataDirectoryArray.js';
|
|
import ImageDosHeader from '../format/ImageDosHeader.js';
|
|
import ImageFileHeader from '../format/ImageFileHeader.js';
|
|
import ImageNtHeaders from '../format/ImageNtHeaders.js';
|
|
import ImageOptionalHeader from '../format/ImageOptionalHeader.js';
|
|
import ImageOptionalHeader64 from '../format/ImageOptionalHeader64.js';
|
|
import { copyBuffer, roundUp } from './functions.js';
|
|
// fill with '0x00' to make 8-bytes alignment
|
|
// prettier-ignore
|
|
var DOS_STUB_PROGRAM = new Uint8Array([
|
|
0x0e, 0x1f, 0xba, 0x0e, 0x00, 0xb4, 0x09, 0xcd, 0x21, 0xb8, 0x01, 0x4c, 0xcd, 0x21, 0x44, 0x4f,
|
|
0x53, 0x20, 0x6d, 0x6f, 0x64, 0x65, 0x20, 0x6e, 0x6f, 0x74, 0x20, 0x73, 0x75, 0x70, 0x70, 0x6f,
|
|
0x72, 0x74, 0x65, 0x64, 0x2e, 0x0d, 0x0d, 0x0a, 0x24, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
|
|
]);
|
|
var DOS_STUB_SIZE = roundUp(ImageDosHeader.size + DOS_STUB_PROGRAM.length, 0x80);
|
|
var DEFAULT_FILE_ALIGNMENT = 512;
|
|
export function getDosStubDataSize() {
|
|
return DOS_STUB_SIZE;
|
|
}
|
|
export function fillDosStubData(bin) {
|
|
var dos = ImageDosHeader.from(bin);
|
|
dos.magic = ImageDosHeader.DEFAULT_MAGIC;
|
|
// last page size
|
|
dos.lastPageSize = DOS_STUB_SIZE % 512;
|
|
// total page count
|
|
dos.pages = Math.ceil(DOS_STUB_SIZE / 512);
|
|
// no relocations
|
|
dos.relocations = 0;
|
|
// header size as paragraph count (1 paragraph = 16 bytes)
|
|
dos.headerSizeInParagraph = Math.ceil(ImageDosHeader.size / 16);
|
|
dos.minAllocParagraphs = 0;
|
|
dos.maxAllocParagraphs = 0xffff;
|
|
dos.initialSS = 0;
|
|
dos.initialSP = 0x80;
|
|
// (no relocations, but set offset after the header)
|
|
dos.relocationTableAddress = ImageDosHeader.size;
|
|
dos.newHeaderAddress = DOS_STUB_SIZE;
|
|
copyBuffer(bin, ImageDosHeader.size, DOS_STUB_PROGRAM, 0, DOS_STUB_PROGRAM.length);
|
|
}
|
|
export function estimateNewHeaderSize(is32Bit) {
|
|
return (
|
|
// magic
|
|
4 +
|
|
ImageFileHeader.size +
|
|
(is32Bit ? ImageOptionalHeader.size : ImageOptionalHeader64.size) +
|
|
ImageDataDirectoryArray.size);
|
|
}
|
|
export function fillPeHeaderEmptyData(bin, offset, totalBinSize, is32Bit, isDLL) {
|
|
var _bin;
|
|
var _offset;
|
|
if ('buffer' in bin) {
|
|
_bin = bin.buffer;
|
|
_offset = bin.byteOffset + offset;
|
|
}
|
|
else {
|
|
_bin = bin;
|
|
_offset = offset;
|
|
}
|
|
new DataView(_bin, _offset).setUint32(0, ImageNtHeaders.DEFAULT_SIGNATURE, true);
|
|
var fh = ImageFileHeader.from(_bin, _offset + 4);
|
|
fh.machine = is32Bit ? 0x14c : 0x8664;
|
|
fh.numberOfSections = 0; // no sections
|
|
fh.timeDateStamp = 0;
|
|
fh.pointerToSymbolTable = 0;
|
|
fh.numberOfSymbols = 0;
|
|
fh.sizeOfOptionalHeader =
|
|
(is32Bit ? ImageOptionalHeader.size : ImageOptionalHeader64.size) +
|
|
ImageDataDirectoryArray.size;
|
|
fh.characteristics = isDLL ? 0x2102 : 0x102;
|
|
var oh = (is32Bit ? ImageOptionalHeader : ImageOptionalHeader64).from(_bin, _offset + 4 + ImageFileHeader.size);
|
|
oh.magic = is32Bit
|
|
? ImageOptionalHeader.DEFAULT_MAGIC
|
|
: ImageOptionalHeader64.DEFAULT_MAGIC;
|
|
// oh.majorLinkerVersion = 0;
|
|
// oh.minorLinkerVersion = 0;
|
|
oh.sizeOfCode = 0;
|
|
oh.sizeOfInitializedData = 0;
|
|
oh.sizeOfUninitializedData = 0;
|
|
oh.addressOfEntryPoint = 0;
|
|
oh.baseOfCode = 0x1000;
|
|
// oh.baseOfData = 0; // for 32bit only
|
|
oh.imageBase = is32Bit ? 0x1000000 : 0x180000000;
|
|
oh.sectionAlignment = 4096;
|
|
oh.fileAlignment = DEFAULT_FILE_ALIGNMENT;
|
|
oh.majorOperatingSystemVersion = 6;
|
|
oh.minorOperatingSystemVersion = 0;
|
|
// oh.majorImageVersion = 0;
|
|
// oh.minorImageVersion = 0;
|
|
oh.majorSubsystemVersion = 6;
|
|
oh.minorSubsystemVersion = 0;
|
|
// oh.win32VersionValue = 0;
|
|
oh.sizeOfHeaders = roundUp(totalBinSize, oh.fileAlignment);
|
|
// oh.checkSum = 0;
|
|
oh.subsystem = 2; // IMAGE_SUBSYSTEM_WINDOWS_GUI
|
|
oh.dllCharacteristics =
|
|
(is32Bit ? 0 : 0x20) + // IMAGE_DLL_CHARACTERISTICS_HIGH_ENTROPY_VA
|
|
0x40 + // IMAGE_DLLCHARACTERISTICS_DYNAMIC_BASE
|
|
0x100; // IMAGE_DLLCHARACTERISTICS_NX_COMPAT
|
|
oh.sizeOfStackReserve = 0x100000;
|
|
oh.sizeOfStackCommit = 0x1000;
|
|
oh.sizeOfHeapReserve = 0x100000;
|
|
oh.sizeOfHeapCommit = 0x1000;
|
|
// oh.loaderFlags = 0;
|
|
oh.numberOfRvaAndSizes =
|
|
ImageDataDirectoryArray.size / ImageDataDirectoryArray.itemSize;
|
|
}
|
|
export function makeEmptyNtExecutableBinary(is32Bit, isDLL) {
|
|
var bufferSize = roundUp(DOS_STUB_SIZE + estimateNewHeaderSize(is32Bit), DEFAULT_FILE_ALIGNMENT);
|
|
var bin = new ArrayBuffer(bufferSize);
|
|
fillDosStubData(bin);
|
|
fillPeHeaderEmptyData(bin, DOS_STUB_SIZE, bufferSize, is32Bit, isDLL);
|
|
return bin;
|
|
}
|