; 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
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 'kolibrios/res/system/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:
;-------------------------------------------------------------------------------