WebSVN – Kolibri OS – Blame – /kernel/branches/Kolibri-acpi/core/peload.inc

Rev	Author	Line No.	Line
750	victor	1	;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
		2	;; ;;
5565	serge	3	;; Copyright (C) KolibriOS team 2004-2015. All rights reserved. ;;
750	victor	4	;; Distributed under terms of the GNU General Public License ;;
		5	;; ;;
		6	;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
		7
		8	$Revision: 5565 $
757	serge	9
		10	include 'export.inc'
		11
		12	align 4
		13
		14	proc load_PE stdcall, file_name:dword
		15	locals
		16	image dd ?
		17	entry dd ?
		18	base dd ?
		19	endl
		20
2434	Serge	21	stdcall load_file, [file_name]
		22	test eax, eax
		23	jz .fail
757	serge	24
2434	Serge	25	mov [image], eax
757	serge	26
5201	serge	27	mov edx, [eax+STRIPPED_PE_HEADER.SizeOfImage]
		28	; mov cl, [eax+STRIPPED_PE_HEADER.Subsystem]
		29	cmp word [eax], STRIPPED_PE_SIGNATURE
		30	jz @f
		31
2434	Serge	32	mov edx, [eax+60]
5201	serge	33	; mov cl, [eax+5Ch+edx]
		34	mov edx, [eax+80+edx]
757	serge	35
5201	serge	36	@@:
		37	mov [entry], 0
		38	; cmp cl, 1
		39	; jnz .cleanup
		40	stdcall kernel_alloc, edx
2434	Serge	41	test eax, eax
		42	jz .cleanup
757	serge	43
2434	Serge	44	mov [base], eax
5565	serge	45	DEBUGF 1,'K : driver %s mapped to %x\n',[file_name],[base]
757	serge	46
5201	serge	47	push ebx ebp
		48	mov ebx, [image]
		49	mov ebp, eax
		50	call map_PE
		51	pop ebp ebx
757	serge	52
2434	Serge	53	mov [entry], eax
		54	test eax, eax
		55	jnz .cleanup
757	serge	56
2434	Serge	57	stdcall kernel_free, [base]
757	serge	58	.cleanup:
2434	Serge	59	stdcall kernel_free, [image]
		60	mov eax, [entry]
		61	ret
757	serge	62	.fail:
2434	Serge	63	xor eax, eax
		64	ret
757	serge	65	endp
		66
5201	serge	67	map_PE: ;ebp=base:dword, ebx=image:dword
2434	Serge	68	push edi
		69	push esi
5201	serge	70	sub esp, .locals_size
		71	virtual at esp
		72	.numsections dd ?
		73	.import_names dd ?
		74	.import_targets dd ?
		75	.peheader dd ?
		76	.bad_import dd ?
		77	.import_idx dd ?
		78	.import_descr dd ?
		79	.relocs_rva dd ?
		80	.relocs_size dd ?
		81	.section_header_size dd ?
		82	.AddressOfEntryPoint dd ?
		83	.ImageBase dd ?
		84	.locals_size = $ - esp
		85	end virtual
		86	cmp word [ebx], STRIPPED_PE_SIGNATURE
		87	jz .stripped
		88
2434	Serge	89	mov edx, ebx
5201	serge	90	add edx, [ebx+60]
		91	movzx eax, word [edx+6]
		92	mov [.numsections], eax
		93	mov eax, [edx+40]
		94	mov [.AddressOfEntryPoint], eax
		95	mov eax, [edx+52]
		96	mov [.ImageBase], eax
		97	mov ecx, [edx+84]
		98	mov [.section_header_size], 40
		99	mov eax, [edx+128]
		100	mov [.import_descr], eax
		101	mov eax, [edx+160]
		102	mov [.relocs_rva], eax
		103	mov eax, [edx+164]
		104	mov [.relocs_size], eax
		105	add edx, 256
		106
		107	jmp .common
		108	.stripped:
		109	mov eax, [ebx+STRIPPED_PE_HEADER.AddressOfEntryPoint]
		110	mov [.AddressOfEntryPoint], eax
		111	mov eax, [ebx+STRIPPED_PE_HEADER.ImageBase]
		112	mov [.ImageBase], eax
		113	movzx eax, [ebx+STRIPPED_PE_HEADER.NumberOfSections]
		114	mov [.numsections], eax
		115	movzx ecx, [ebx+STRIPPED_PE_HEADER.NumberOfRvaAndSizes]
		116	xor eax, eax
		117	mov [.relocs_rva], eax
		118	mov [.relocs_size], eax
		119	test ecx, ecx
		120	jz @f
		121	mov eax, [ebx+sizeof.STRIPPED_PE_HEADER+SPE_DIRECTORY_IMPORT*8]
		122	@@:
		123	mov [.import_descr], eax
		124	cmp ecx, SPE_DIRECTORY_BASERELOC
		125	jbe @f
		126	mov eax, [ebx+sizeof.STRIPPED_PE_HEADER+SPE_DIRECTORY_BASERELOC*8]
		127	mov [.relocs_rva], eax
		128	mov eax, [ebx+sizeof.STRIPPED_PE_HEADER+SPE_DIRECTORY_BASERELOC*8+4]
		129	mov [.relocs_size], eax
		130	@@:
		131	mov [.section_header_size], 28
		132	lea edx, [ebx+ecx*8+sizeof.STRIPPED_PE_HEADER+8]
		133	mov ecx, [ebx+STRIPPED_PE_HEADER.SizeOfHeaders]
		134
		135	.common:
2434	Serge	136	mov esi, ebx
		137	mov edi, ebp
		138	shr ecx, 2
		139	rep movsd
757	serge	140
5201	serge	141	cmp [.numsections], 0
		142	jz .nosections
		143	.copy_sections:
		144	mov eax, [edx+8]
2434	Serge	145	test eax, eax
5201	serge	146	je .no_section_data
2434	Serge	147	mov esi, ebx
		148	mov edi, ebp
5201	serge	149	add esi, [edx+12]
2434	Serge	150	mov ecx, eax
5201	serge	151	add edi, [edx+4]
757	serge	152
4423	Serge	153	add ecx, 3
2434	Serge	154	shr ecx, 2
		155	rep movsd
757	serge	156
5201	serge	157	.no_section_data:
		158	mov ecx, [edx]
2434	Serge	159	cmp ecx, eax
5201	serge	160	jbe .no_section_fill
2434	Serge	161	sub ecx, eax
5201	serge	162	add eax, [edx+4]
2434	Serge	163	lea edi, [eax+ebp]
757	serge	164
2434	Serge	165	xor eax, eax
		166	rep stosb
757	serge	167
5201	serge	168	.no_section_fill:
		169	add edx, [.section_header_size]
		170	dec [.numsections]
		171	jnz .copy_sections
		172	.nosections:
		173	cmp [.relocs_size], 0
		174	je .no_relocations
2434	Serge	175	mov esi, ebp
		176	mov ecx, ebp
5201	serge	177	sub esi, [.ImageBase]
		178	add ecx, [.relocs_rva]
		179	.relocs_block:
		180	mov edi, [ecx]
		181	add edi, ebp
		182	mov ebx, [ecx+4]
		183	add ecx, 8
		184	sub [.relocs_size], ebx
		185	sub ebx, 8
2434	Serge	186	shr ebx, 1
5201	serge	187	jz .relocs_next_block
		188	.one_reloc:
		189	movzx eax, word [ecx]
		190	add ecx, 2
2434	Serge	191	mov edx, eax
		192	shr eax, 12
		193	and edx, 4095
5201	serge	194	cmp eax, 3
		195	jne @f
		196	add [edx+edi], esi
		197	@@:
		198	dec ebx
		199	jnz .one_reloc
		200	.relocs_next_block:
		201	cmp [.relocs_size], 0
		202	jg .relocs_block
		203	.no_relocations:
		204	cmp [.import_descr], 0
		205	je .no_imports
		206	add [.import_descr], ebp
		207	mov [.bad_import], 0
		208	.import_block:
		209	mov ecx, [.import_descr]
		210	cmp dword [ecx+4], 0
		211	jne @f
		212	cmp dword [ecx+12], 0
		213	je .done_imports
		214	@@:
		215	mov edx, dword [ecx]
		216	mov ecx, dword [ecx+16]
4423	Serge	217	test edx, edx
		218	jnz @f
		219	mov edx, ecx
		220	@@:
5201	serge	221	mov [.import_idx], 0
2434	Serge	222	add ecx, ebp
		223	add edx, ebp
5201	serge	224	mov [.import_names], edx
		225	mov [.import_targets], ecx
		226	.import_func:
		227	mov esi, [.import_idx]
		228	mov edi, [.import_names]
		229	mov eax, [edi+esi*4]
2434	Serge	230	test eax, eax
5201	serge	231	je .next_import_block
		232	js .next_import_block
2434	Serge	233	lea edi, [ebp+eax]
5201	serge	234	mov eax, [.import_targets]
		235	mov dword [eax+esi*4], 0
2434	Serge	236	lea esi, [edi+2]
5201	serge	237	movzx ebx, word [edi]
2434	Serge	238	push 32
5201	serge	239	mov ecx, [__exports+32]
		240	mov eax, [ecx+OS_BASE+ebx*4]
884	serge	241	add eax, OS_BASE
2434	Serge	242	push eax
		243	push esi
		244	call strncmp
5201	serge	245	test eax, eax
		246	jz .import_func_found
2434	Serge	247	xor ebx, ebx
5201	serge	248	.import_func_candidate:
2434	Serge	249	push 32
5201	serge	250	mov ecx, [__exports+32]
		251	mov eax, [ecx+OS_BASE+ebx*4]
884	serge	252	add eax, OS_BASE
2434	Serge	253	push eax
		254	push esi
		255	call strncmp
		256	test eax, eax
5201	serge	257	je .import_func_found
2434	Serge	258	inc ebx
5201	serge	259	cmp ebx, [__exports+24]
		260	jb .import_func_candidate
757	serge	261
		262	mov esi, msg_unresolved
		263	call sys_msg_board_str
2434	Serge	264	lea esi, [edi+2]
757	serge	265	call sys_msg_board_str
		266	mov esi, msg_CR
		267	call sys_msg_board_str
		268
5201	serge	269	mov [.bad_import], 1
		270	jmp .next_import_func
		271	.import_func_found:
		272	mov esi, [__exports+28]
		273	mov edx, [.import_idx]
		274	mov ecx, [.import_targets]
		275	mov eax, [esi+OS_BASE+ebx*4]
884	serge	276	add eax, OS_BASE
5201	serge	277	mov [ecx+edx*4], eax
		278	.next_import_func:
		279	inc [.import_idx]
		280	jmp .import_func
		281	.next_import_block:
		282	add [.import_descr], 20
		283	jmp .import_block
		284	.done_imports:
2434	Serge	285	xor eax, eax
5201	serge	286	cmp [.bad_import], 0
		287	jne @f
		288	.no_imports:
2434	Serge	289	mov eax, ebp
5201	serge	290	add eax, [.AddressOfEntryPoint]
		291	@@:
		292	add esp, .locals_size
2434	Serge	293	pop esi
		294	pop edi
5201	serge	295	ret

Subversion Repositories Kolibri OS

(root)/kernel/branches/Kolibri-acpi/core/peload.inc @ 3232 – Rev 5565