forked from KolibriOS/kolibrios
233 lines
5.4 KiB
NASM
233 lines
5.4 KiB
NASM
; Kolibri kernel packer
|
|
; (C) copyright diamond 2006, 2007
|
|
;
|
|
; This program is free software; you can redistribute it and/or modify
|
|
; it under the terms of the GNU General Public License as published by
|
|
; the Free Software Foundation; either version 2 of the License, or
|
|
; (at your option) any later version.
|
|
;
|
|
; This program is distributed in the hope that it will be useful,
|
|
; but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
; GNU General Public License for more details.
|
|
|
|
; Uses LZMA compression library by Igor Pavlov
|
|
; (for more information on LZMA and 7-Zip visit http://www.7-zip.org)
|
|
; (plain-C packer is ported by diamond)
|
|
|
|
format MS COFF
|
|
|
|
extrn '_lzma_compress@16' as lzma_compress
|
|
extrn '_lzma_set_dict_size@4' as lzma_set_dict_size
|
|
|
|
section '.text' code executable readable
|
|
|
|
die_with_err:
|
|
pop esi
|
|
@@:
|
|
lodsb
|
|
test al, al
|
|
jz @f
|
|
mov cl, al
|
|
push 63
|
|
pop eax
|
|
push 1
|
|
pop ebx
|
|
int 40h
|
|
jmp @b
|
|
@@:
|
|
mov al, 63
|
|
mov cl, 13
|
|
int 40h
|
|
mov cl, 10
|
|
int 40h
|
|
or eax, -1
|
|
int 40h
|
|
|
|
public _start
|
|
_start:
|
|
push 70
|
|
pop eax
|
|
mov ebx, fn70_read
|
|
int 40h
|
|
cmp eax, 6
|
|
jz read_ok
|
|
read_err:
|
|
call die_with_err
|
|
db 'KerPack: cannot load kernel.mnt',0
|
|
read_ok:
|
|
push 18
|
|
call lzma_set_dict_size
|
|
; find jump to 32-bit code
|
|
mov edi, infile - 1
|
|
@@:
|
|
inc edi
|
|
cmp dword [edi], 0E88EE08Eh ; mov fs,ax/mov gs,ax
|
|
jnz @b
|
|
cmp dword [edi+4], 00BCD08Eh ; mov ss,ax/mov esp,00xxxxxx
|
|
jnz @b
|
|
add edi, 11
|
|
mov [inptr], edi
|
|
sub edi, infile
|
|
mov [indelta], edi
|
|
lea eax, [ebx+0x10000]
|
|
mov [..loader_patch3+2], eax
|
|
sub ebx, edi
|
|
mov [insize], ebx
|
|
call preprocess_calltrick2
|
|
mov al, [cti]
|
|
mov [loader_patch5-1], al
|
|
mov eax, [ctn]
|
|
mov [loader_patch4+1], eax
|
|
mov eax, [inptr]
|
|
add eax, outfile - infile + loader_size - 5
|
|
push workmem
|
|
push [insize]
|
|
push eax
|
|
push [inptr]
|
|
call lzma_compress
|
|
add eax, loader_size-5
|
|
mov [loader_patch1+6], eax
|
|
add eax, [indelta]
|
|
mov [outsize], eax
|
|
mov eax, [indelta]
|
|
mov ecx, dword [eax + outfile + loader_size - 4]
|
|
bswap ecx
|
|
mov [loader_patch2+4], ecx
|
|
add eax, 0x10000
|
|
mov [loader_patch1+1], eax
|
|
mov esi, infile
|
|
mov edi, outfile
|
|
mov ecx, [indelta]
|
|
rep movsb
|
|
mov esi, loader_start
|
|
mov ecx, loader_size
|
|
rep movsb
|
|
push 70
|
|
pop eax
|
|
mov ebx, fn70_write
|
|
int 40h
|
|
test eax, eax
|
|
jz @f
|
|
call die_with_err
|
|
db 'KerPack: cannot save kernel.mnt',0
|
|
@@:
|
|
call die_with_err
|
|
db 'KerPack: all is OK',0
|
|
|
|
preprocess_calltrick2:
|
|
; input preprocessing
|
|
mov edi, ct1
|
|
xor eax, eax
|
|
push edi
|
|
mov ecx, 256/4
|
|
rep stosd
|
|
pop edi
|
|
mov ecx, ebx
|
|
mov esi, [inptr]
|
|
mov ebx, inbuftmp
|
|
xchg eax, edx
|
|
input_pre2:
|
|
lodsb
|
|
@@:
|
|
cmp al, 0Fh
|
|
jnz ip1
|
|
dec ecx
|
|
jz input_pre_done2
|
|
lodsb
|
|
cmp al, 80h
|
|
jb @b
|
|
cmp al, 90h
|
|
jb @f
|
|
ip1:
|
|
sub al, 0E8h
|
|
cmp al, 1
|
|
ja input_pre_cont2
|
|
@@:
|
|
cmp ecx, 5
|
|
jb input_pre_done2
|
|
lodsd
|
|
add eax, esi
|
|
sub eax, [inptr]
|
|
cmp eax, [insize]
|
|
jae xxx2
|
|
cmp eax, 1000000h
|
|
jae xxx2
|
|
sub ecx, 4
|
|
xchg al, ah
|
|
rol eax, 16
|
|
xchg al, ah
|
|
mov [esi-4], eax
|
|
inc edx
|
|
mov [ebx], esi
|
|
add ebx, 4
|
|
jmp input_pre_cont2
|
|
xxx2: sub esi, 4
|
|
movzx eax, byte [esi]
|
|
mov byte [eax+edi], 1
|
|
input_pre_cont2:
|
|
loop input_pre2
|
|
input_pre_done2:
|
|
mov [ctn], edx
|
|
xor eax, eax
|
|
mov ecx, 256
|
|
repnz scasb
|
|
jnz pack_calltrick_done
|
|
not cl
|
|
mov [cti], cl
|
|
@@:
|
|
cmp ebx, inbuftmp
|
|
jz pack_calltrick_done
|
|
sub ebx, 4
|
|
mov eax, [ebx]
|
|
mov [eax-4], cl
|
|
jmp @b
|
|
pack_calltrick_done:
|
|
ret
|
|
|
|
include 'loader_lzma.asm'
|
|
|
|
section '.data' data readable writeable
|
|
db 'MENUET01'
|
|
dd 1
|
|
dd _start
|
|
dd bss_start ; i_end
|
|
dd bss_end ; memory
|
|
dd mtstack_end ; esp
|
|
dd 0 ; params
|
|
dd 0 ; icon
|
|
fn70_read:
|
|
dd 0
|
|
dd 0
|
|
dd 0
|
|
dd 200*1024
|
|
dd infile
|
|
filename db '/rd/1/kernel.mnt',0
|
|
fn70_write:
|
|
dd 2
|
|
dd 0
|
|
dd 0
|
|
outsize dd ?
|
|
dd outfile
|
|
db 0
|
|
dd filename
|
|
section '.bss' readable writeable
|
|
bss_start:
|
|
align 4
|
|
inptr dd ?
|
|
indelta dd ?
|
|
insize dd ?
|
|
ct1 rb 256
|
|
ctn dd ?
|
|
cti db ?
|
|
|
|
align 4
|
|
mtstack rb 1000h
|
|
mtstack_end:
|
|
|
|
infile rb 200*1024
|
|
inbuftmp rb 200*1024
|
|
outfile rb 200*1024
|
|
workmem rb 6A8000h
|
|
bss_end:
|