0,0 → 1,139 |
// pe2kos.cpp : Defines the entry point for the console application. |
// |
|
//#define TARGET_USES_PATH |
|
#include <string.h> |
#include <stdio.h> |
#include <io.h> |
|
typedef unsigned __int32 DWORD; |
typedef unsigned __int16 WORD; |
typedef unsigned __int8 BYTE; |
|
#pragma pack(1) |
|
struct kos_Header |
{ |
char kosSign[8]; |
DWORD res1; |
DWORD start; |
DWORD end; |
DWORD memory; |
DWORD stack; |
DWORD res2; |
DWORD res3; |
// |
kos_Header() |
{ |
memcpy( kosSign, "MENUET01", 8 ); |
res1 = 1; |
res2 = 0; |
res3 = 0; |
} |
}; |
|
|
void load_section( BYTE *pePtr, BYTE *secDescPtr, FILE *fp ) |
{ |
DWORD *dwPtr = (DWORD *)secDescPtr; |
DWORD virtualSize; |
|
// |
virtualSize = ( dwPtr[2] >= dwPtr[4] ) ? dwPtr[2] : dwPtr[4]; |
// |
fseek( fp, dwPtr[3], SEEK_SET ); |
fwrite( pePtr + dwPtr[5], virtualSize, 1, fp ); |
} |
|
|
int main(int argc, char* argv[]) |
{ |
FILE *pe, *kos; |
kos_Header hdr; |
BYTE *peBuff; |
BYTE *scanPtr; |
long peSize; |
DWORD baseNdx; |
WORD sCount, sLim; |
|
// |
if ( argc != 3 ) return 0; |
|
// |
pe = fopen( argv[1], "rb" ); |
if ( pe == NULL ) return 0; |
// |
kos = fopen( argv[2], "w+b" ); |
if ( kos == NULL ) |
{ |
fclose( pe ); |
return 0; |
} |
// |
fseek( pe, 0, SEEK_END ); |
peSize = ftell( pe ); |
fseek( pe, 0, SEEK_SET ); |
// |
peBuff = new BYTE[peSize]; |
fread( peBuff, peSize, 1, pe ); |
fclose( pe ); |
// |
// |
scanPtr = peBuff + ((DWORD *)(peBuff + 0x3C))[0]; |
// |
baseNdx = 0xF8; |
// |
sLim = ((WORD *)(scanPtr + 6))[0]; |
// |
for ( sCount = 0; sCount < sLim; sCount++ ) |
{ |
// |
switch ( scanPtr[baseNdx + 0x24] ) |
{ |
// |
case 0x20: |
case 0x40: |
load_section( peBuff, scanPtr + baseNdx, kos ); |
break; |
// |
default: |
break; |
} |
// |
baseNdx += 0x28; |
|
} |
// |
fseek( kos, 0, SEEK_END ); |
fflush(kos); |
peSize = ftell( kos ); |
for (;;) |
{ |
fseek(kos, -1, SEEK_CUR); |
if (fgetc(kos)) |
break; |
fseek(kos, -1, SEEK_CUR); |
} |
_chsize(_fileno(kos), ftell(kos)); |
// |
hdr.start = ((DWORD *)(scanPtr + 0x28))[0]; |
hdr.end = ftell(kos); |
hdr.stack = peSize + 4096; |
#ifdef TARGET_USES_PATH |
hdr.res3 = hdr.stack; |
hdr.memory = hdr.res3 + 2048; |
#else |
hdr.memory = hdr.stack; |
#endif |
// |
fseek( kos, 0, SEEK_SET ); |
fwrite( &hdr, sizeof(hdr), 1, kos ); |
|
// |
delete [] peBuff; |
// |
fclose( kos ); |
|
return 0; |
} |