Subversion Repositories Kolibri OS

; Search Additional Partition for KolibriOS applications

;

; Copyright (c) 2013-2014, Marat Zakiyanov aka Mario79, aka Mario

; All rights reserved.

;

; Redistribution and use in source and binary forms, with or without

; modification, are permitted provided that the following conditions are met:

;	 * Redistributions of source code must retain the above copyright

;	   notice, this list of conditions and the following disclaimer.

;	 * Redistributions in binary form must reproduce the above copyright

;	   notice, this list of conditions and the following disclaimer in the

;	   documentation and/or other materials provided with the distribution.

;	 * Neither the name of the  nor the

;	   names of its contributors may be used to endorse or promote products

;	   derived from this software without specific prior written permission.

;

; THIS SOFTWARE IS PROVIDED BY Marat Zakiyanov ''AS IS'' AND ANY

; EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED

; WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE

; DISCLAIMED. IN NO EVENT SHALL  BE LIABLE FOR ANY

; DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES

; (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;

; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND

; ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

; (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS

; SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

;*****************************************************************************

	use32

	org	0x0

	db 'MENUET01'

	dd 0x01

	dd START

	dd IM_END

	dd I_END

	dd stacktop

params dd PARAMS

	dd 0x0

;---------------------------------------------------------------------

delay dd 500

mount_attempt dd 0

;---------------------------------------------------------------------

fileinfo:

.subfunction	dd 5

.Offset		dd 0

.Offset_1	dd 0

.size		dd 0

.return		dd folder_data

		db 0

.name:		dd basic_file_path

;---------------------------------------------------------------------

read_folder:

.subfunction	dd 1

.start		dd 0

.flags		dd 0

.size		dd 32

.return		dd folder_data

		db 0

.name:		dd read_folder_name

;---------------------------------------------------------------------

read_folder_1:

.subfunction	dd 1

.start		dd 0

.flags		dd 0

.size		dd 32

.return		dd folder_data_1

		db 0

.name:		dd read_folder_1_name

;---------------------------------------------------------------------

start_dir:

	db '/',0

;-------------------------------------------------------------------------------

basic_file_path:

	db '/sys/settings/'

basic_file_name:

	db 'kolibri.lbl',0

additional_dir_name:

    db 'kolibrios',0

real_additional_dir:

	db '/kolibrios',0

;-------------------------------------------------------------------------------

include	'../../macros.inc'

define __DEBUG__ 1

define __DEBUG_LEVEL__ 2	; 1 = verbose, 2 = main only

include "../../debug-fdo.inc"

;-------------------------------------------------------------------------------

START:

; process cmdline params

	mov	esi, [params]

	test	[esi], byte 0xFF

	jz	.params_done

	cmp	word[esi], '-d' ; delay

	jne	.params_done

	add	esi, 2

; str2uint(delay)

	xor	eax, eax

	xor	ecx, ecx

.convert:

	lodsb

	test	al, al

	jz	.converted

	lea	ecx, [ecx + ecx * 4]

	lea	ecx, [eax + ecx * 2 - '0']

	jmp	.convert

.converted:

	mov	[delay], ecx

;--------------------------------------

	DEBUGF	1, "Searchap: get basic file\n"

	call	load_file	; download the master file

	xor	eax,eax

	cmp	[fs_error],eax

	jne	exit

	mov	eax,[fileinfo.size]

	mov	[basic_file_size],eax

.params_done:

;--------------------------------------

	mov	ebx,start_dir

	mov	ax,[ebx]

	mov	ebx,read_folder_name

	mov	[ebx],ax

	mov	ebx,read_folder_1_name

	mov	[ebx],ax

	call	device_detect_f70

;--------------------------------------

	call	print_retrieved_devices_table

	call	search_and_load_pointer_file_label

;---------------------------------------------------------------------

exit:

;--------------------------------------

	DEBUGF	1, "Searchap: just exit\n"

;--------------------------------------

	cmp [mount_dir],1

	je @f

	cmp [mount_attempt], 1

	je @f

	mov [mount_attempt], 1 ;second mount attempt with delay

	DEBUGF	2, "Searchap: second attempt after 5 seconds!\n"

	mcall	5,[delay]

	jmp START.params_done

@@:

	mcall	-1

;---------------------------------------------------------------------

device_detect_f70:

;--------------------------------------

	DEBUGF	1, "Searchap: read_folder_name: %s\n",read_folder_name

;--------------------------------------

	mcall	70,read_folder

	test	eax,eax

	jz	@f

	cmp	eax,6

	je	@f

;--------------------------------------

	DEBUGF	1, "Searchap: read_folder_error\n"

;--------------------------------------

	jmp	exit

@@:

;--------------------------------------

	call	print_root_dir

;--------------------------------------

	mov	[left_folder_block],ebx

	xor	eax,eax

	mov	[temp_counter_1],eax

	mov	[retrieved_devices_table_counter],eax

.start_temp_counter_1:

	imul	esi,[temp_counter_1],304

	add	esi,[read_folder.return]

	add	esi,32+40

	call	copy_folder_name_1

;--------------------------------------

	DEBUGF	1, "Searchap: read_folder_1_name: %s\n",read_folder_1_name

;--------------------------------------

	mcall	70,read_folder_1

	test	eax,eax

	jz	@f

	cmp	eax,6

	je	@f

;--------------------------------------

	DEBUGF	1, "Searchap: read_folder_error_1\n"

;--------------------------------------

	jmp	exit

@@:

	mov	eax,[read_folder_1.return]

	cmp	[eax+4],dword 0

	je	.continue

	mov	[right_folder_block],ebx

	xor	ebp,ebp

.start_copy_device_patch:

	imul	edi,[retrieved_devices_table_counter],10

	add	edi,retrieved_devices_table

	mov	[edi],byte '/'

	inc	edi

	imul	esi,[temp_counter_1],304

	add	esi,[read_folder.return]

	add	esi,32+40

	call	proc_copy_patch

	imul	esi,ebp,304

	add	esi,[read_folder_1.return]

	add	esi,32+40

	mov	[edi-1],byte '/'

	call	proc_copy_patch

	inc	[retrieved_devices_table_counter]

	inc	ebp

	cmp	ebp,[right_folder_block]

	jb	.start_copy_device_patch

.continue:

	inc	[temp_counter_1]

	mov	eax,[temp_counter_1]

	cmp	eax,[left_folder_block]

	jb	.start_temp_counter_1

	mov	esi,retrieved_devices_table+1

	call	copy_folder_name

	mov	esi,retrieved_devices_table+3

	xor	ecx,ecx

@@:

	add	esi,8

	cld

	lodsw

	inc	ecx

	cmp	ecx,[retrieved_devices_table_counter]

	ja	@f

	cmp	ax,'hd'

	jne	@r

	sub	esi,2

	call	copy_folder_name_1

	ret

@@:

	mov	esi,retrieved_devices_table+1

	call	copy_folder_name_1

	ret

;---------------------------------------------------------------------

load_file:

	mov	[fileinfo.subfunction],dword 5

	xor	eax,eax

	mov	[fileinfo.size],eax

	mov	[fs_error],eax

;--------------------------------------

	DEBUGF	1, "Searchap: get file info\n"

;--------------------------------------

	mcall	68,1

	mcall	70,fileinfo

	mov	[fs_error],eax

	test	eax,eax

	jnz	.file_error

;--------------------------------------

	DEBUGF	1, "Searchap: file info ok\n"

;--------------------------------------

	xor	eax,eax

	mov	[fileinfo.subfunction],eax	;dword 0

	mov	eax,[fileinfo.return]

	mov	ecx,[eax+32]

;--------------------------------------

	DEBUGF	1, "Searchap: real file size: %d\n",ecx

;--------------------------------------

	test	ecx,ecx

	jz	.file_error

	mov	eax,304*32+32 ; 9 Kb

	cmp	ecx,eax

	jbe	@f

	mov	ecx,eax

;-----------------------------------

@@:

	mov	[fileinfo.size],ecx

;--------------------------------------

	DEBUGF	1, "Searchap: get file\n"

;--------------------------------------

	mcall	68,1

	mcall	70,fileinfo

	mov	[fs_error],eax

	test	eax,eax

	jz	@f

;	cmp	eax,6

;	jne	.file_error

;	xor	eax,eax

;	mov	[fs_error],eax

;	jmp	@f

;-----------------------------------

.file_error:

;--------------------------------------

	DEBUGF	1, "Searchap: read file - error!\n"

;--------------------------------------

	ret

;-----------------------------------

@@:

;--------------------------------------

	DEBUGF	1, "Searchap: read file corrected size: %d\n",[fileinfo.size]

;--------------------------------------

	ret

;---------------------------------------------------------------------

search_and_load_pointer_file_label:

	mov	[fileinfo.return],dword folder_data_1

	mov	ecx,[retrieved_devices_table_counter]

	dec	ecx	; /rd/1/ no need to check

	mov	[fileinfo.name],dword read_folder_name

	mov	esi,retrieved_devices_table

;	sub	esi,10	; deleted because /rd/1/ no need to check

.next_entry:

;--------------------------------------

	DEBUGF	1, "\nSearchap: copy next entry\n"

;--------------------------------------

	add	esi,10

	push	esi

	add	esi,1

	call	copy_folder_name

	mov	esi,basic_file_name-1

	dec	edi

	call	copy_folder_name.1

	pop	esi

;--------------------------------------

	DEBUGF	1, "Searchap: %s\n",dword[fileinfo.name]

;--------------------------------------

;	mcall	5,10

	push	ecx

	call	load_file

	pop	ecx

	xor	eax,eax

	cmp	[fs_error],eax

	jne	@f

	mov	eax,[fileinfo.size]

	cmp	eax,[basic_file_size]

	jae	.sucess

@@:

	dec	ecx

	jnz	.next_entry

;--------------------------------------

	DEBUGF	2, "Searchap: additional partition is not found!\n"

;--------------------------------------

	ret

.sucess:

	;call	compare_files_and_mount

	call	compare_files_and_mount.mount_now ;no need to compare files content

	cmp	[compare_flag],byte 0

	jne	@b

	cmp	[mount_dir],1

	je	@f

	DEBUGF	2, "Searchap: sorry, but the additional partition is not found!\n"

@@:

	ret

;---------------------------------------------------------------------

compare_files_and_mount:

	push	ecx esi

	mov	ecx,[basic_file_size]

	mov	esi,folder_data

	mov	edi,folder_data_1

.next_char:

	cld

	lodsb

	mov	ah,[edi]

	inc	edi

	cmp	al,ah

	jne	.not_match

	dec	ecx

	jnz	.next_char

	pop	esi ecx

.mount_now:

	mov	[compare_flag],byte 0

;--------------------------------------

	DEBUGF	2, "Searchap: compare files - success!\n"

	DEBUGF	2, "Searchap: mount directory: %s\n",esi

	mov	[mount_dir],1

;--------------------------------------

; prepare real directory path for mounting

	inc	esi

	mov	edi,f30_3_work_area+64

	call	proc_copy_patch

	dec	edi

	mov	esi,real_additional_dir

	call	proc_copy_patch

; prepare fake directory name

	mov	esi,additional_dir_name

	mov	edi,f30_3_work_area

	call	proc_copy_patch

; here is call kernel function to mount the found partition

; as "/kolibrios" directory to root directory "/"

	mcall	30,3,f30_3_work_area

	ret

;--------------------------------------

.not_match:

	mov	[compare_flag],byte 1

	pop	esi ecx

;--------------------------------------

	DEBUGF	1, "Searchap: compared files doesn't match!\n"

;--------------------------------------

	ret

;---------------------------------------------------------------------

copy_folder_name:

	mov	edi,read_folder_name+1

.1:

proc_copy_patch:

	cld

@@:

	lodsb

	stosb

	test	al,al

	jnz	@r

	ret

;---------------------------------------------------------------------

copy_folder_name_1:

	mov	edi,read_folder_1_name+1

	jmp	proc_copy_patch

;---------------------------------------------------------------------

print_retrieved_devices_table:

	mov	ecx,[retrieved_devices_table_counter]

	mov	edx,retrieved_devices_table

	DEBUGF	1, "Searchap: retrieved_devices_table:\n"

	DEBUGF	1, "Searchap: ----------\n"

@@:

	DEBUGF	1, "Searchap: %s\n",edx

	add	edx,10

	dec	ecx

	jnz	@b

	DEBUGF	1, "\nSearchap: ----------\n"

	ret

;---------------------------------------------------------------------

print_root_dir:

	DEBUGF	1, "Searchap: ----------\n"

	DEBUGF	1, "Searchap: root dir:\n"

	DEBUGF	1, "Searchap: ----------\n"

	pusha

	mov	ecx,ebx

	mov	edx,folder_data+32+40

@@:

	DEBUGF	1, "Searchap: %s\n",edx

	add	edx,304

	dec	ecx

	jnz	@b

	popa

	DEBUGF	1, "\nSearchap: ----------\n"

	ret

;-------------------------------------------------------------------------------

include_debug_strings

;-------------------------------------------------------------------------------

IM_END:

;-------------------------------------------------------------------------------

align 4

PARAMS: rb 256

align 4

left_folder_block	rd 1

right_folder_block	rd 1

temp_counter_1		rd 1

retrieved_devices_table_counter	rd 1

basic_file_size		rd 1

fs_error		rd 1

compare_flag		rb 1

mount_dir		rb 1

;-------------------------------------------------------------------------------

align 4

f30_3_work_area:

	rb 128

;-------------------------------------------------------------------------------

align 4

retrieved_devices_table:

	rb 10*100

;-------------------------------------------------------------------------------

align 4

read_folder_name:

	rb 256

;-------------------------------------------------------------------------------

align 4

read_folder_1_name:

	rb 256

;-------------------------------------------------------------------------------

align 4

folder_data:

	rb 304*32+32 ; 9 Kb

;-------------------------------------------------------------------------------

align 4

folder_data_1:

	rb 304*32+32 ; 9 Kb

;-------------------------------------------------------------------------------

align 4

	rb 512

stacktop:

;-------------------------------------------------------------------------------

I_END:

;-------------------------------------------------------------------------------