; 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 dd 0x0 dd 0x0 ;--------------------------------------------------------------------- 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: mcall 5,500 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" ;-------------------------------------- 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 parttition 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 sucess!\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: compare files is not 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 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: ;-------------------------------------------------------------------------------