forked from KolibriOS/kolibrios
140 lines
2.3 KiB
C++
140 lines
2.3 KiB
C++
// 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;
|
|
}
|