forked from KolibriOS/kolibrios
476 lines
13 KiB
NASM
476 lines
13 KiB
NASM
; 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 '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
|
|
.params_done:
|
|
;--------------------------------------
|
|
;mcall 5,[delay] ;first mount attempt without delay
|
|
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
|
|
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
|
|
|
|
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
|
|
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
|
|
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
|
|
mov [compare_flag],byte 0
|
|
pop esi ecx
|
|
;--------------------------------------
|
|
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:
|
|
;------------------------------------------------------------------------------- |