; 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 <organization> 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 <copyright holder> 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, START, IM_END, I_END, 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 'kolibrios/res/system/kolibri.lbl',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]
	cmp	[esi], byte '/'
	jne @f
	mov esi, [params] ;user gave us the path so lets mount it
	inc	esi
	mov	edi,f30_3_work_area+64
	call	proc_copy_path
	mcall	30,3,f30_3_work_area
	mcall   -1
@@:	
	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_path:
	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_path
	imul	esi,ebp,304
	add	esi,[read_folder_1.return]
	add	esi,32+40
	mov	[edi-1],byte '/'
	call	proc_copy_path
	inc	[retrieved_devices_table_counter]
	inc	ebp
	cmp	ebp,[right_folder_block]
	jb	.start_copy_device_path
.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_path
	dec	edi
	mov	esi,real_additional_dir
	call	proc_copy_path
; 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_path:
	cld
@@:
	lodsb
	stosb
	test	al,al
	jnz	@r
	ret
;---------------------------------------------------------------------
copy_folder_name_1:
	mov	edi,read_folder_1_name+1
	jmp	proc_copy_path
;---------------------------------------------------------------------
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:
	db 'kolibrios',0 
	rb 118
;-------------------------------------------------------------------------------
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:
;-------------------------------------------------------------------------------