;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; ;; ;; ;; Copyright (C) KolibriOS team 2004-2007. All rights reserved. ;; ;; Distributed under terms of the GNU General Public License ;; ;; ;; ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; $Revision$ ;************************************************************* ;* 13.02.2010 Find all partition and check supported FS ;* 12.07.2007 Check all 4 entry of MBR and EMBR ;* 29.04.2006 Elimination of hangup after the ;* expiration hd_wait_timeout - Mario79 ;* 28.01.2006 find all Fat16/32 partition in all input point ;* to MBR - Mario79 ;************************************************************* uglobal align 4 ;****************************************************** ; Please do not change this place - variables in text ; Mario79 ; START place ;****************************************************** PARTITION_START dd 0x3f PARTITION_END dd 0 fs_type db 0 ; 1=NTFS, 2=EXT2/3, 16=FAT16, 32=FAT32 align 4 fs_dependent_data_start: ; FATxx data SECTORS_PER_FAT dd 0x1f3a NUMBER_OF_FATS dd 0x2 SECTORS_PER_CLUSTER dd 0x8 BYTES_PER_SECTOR dd 0x200 ; Note: if BPS <> 512 need lots of changes ROOT_CLUSTER dd 2 ; first rootdir cluster FAT_START dd 0 ; start of fat table ROOT_START dd 0 ; start of rootdir (only fat16) ROOT_SECTORS dd 0 ; count of rootdir sectors (only fat16) DATA_START dd 0 ; start of data area (=first cluster 2) LAST_CLUSTER dd 0 ; last availabe cluster ADR_FSINFO dd 0 ; used only by fat32 fatRESERVED dd 0x0FFFFFF6 fatBAD dd 0x0FFFFFF7 fatEND dd 0x0FFFFFF8 fatMASK dd 0x0FFFFFFF fatStartScan dd 2 fs_dependent_data_end: file_system_data_size = $ - PARTITION_START if file_system_data_size > 96 ERROR: sizeof(file system data) too big! end if virtual at fs_dependent_data_start ; NTFS data ntfs_data: .sectors_per_cluster dd ? .mft_cluster dd ? .mftmirr_cluster dd ? .frs_size dd ? ; FRS size in bytes .iab_size dd ? ; IndexAllocationBuffer size in bytes .frs_buffer dd ? .iab_buffer dd ? .mft_retrieval dd ? .mft_retrieval_size dd ? .mft_retrieval_alloc dd ? .mft_retrieval_end dd ? .cur_index_size dd ? .cur_index_buf dd ? if $ > fs_dependent_data_end ERROR: increase sizeof(fs_dependent_data)! end if end virtual virtual at fs_dependent_data_start ; EXT2 data ext2_data: .log_block_size dd ? .block_size dd ? .count_block_in_block dd ? .blocks_per_group dd ? .inodes_per_group dd ? .global_desc_table dd ? .root_inode dd ? ; pointer to root inode in memory .inode_size dd ? .count_pointer_in_block dd ? ; block_size / 4 .count_pointer_in_block_square dd ? ; (block_size / 4)**2 .ext2_save_block dd ? ; блок на глобальную 1 процедуру .ext2_temp_block dd ? ; блок для мелких процедур .ext2_save_inode dd ? ; inode на глобальную процедуру .ext2_temp_inode dd ? ; inode для мелких процедур .sb dd ? ; superblock .groups_count dd ? if $ > fs_dependent_data_end ERROR: increase sizeof(fs_dependent_data)! end if end virtual ;*************************************************************************** ; End place ; Mario79 ;*************************************************************************** endg iglobal partition_types: ; list of fat16/32 partitions db 0x04 ; DOS: fat16 <32M db 0x06 ; DOS: fat16 >32M db 0x0b ; WIN95: fat32 db 0x0c ; WIN95: fat32, LBA-mapped db 0x0e ; WIN95: fat16, LBA-mapped db 0x14 ; Hidden DOS: fat16 <32M db 0x16 ; Hidden DOS: fat16 >32M db 0x1b ; Hidden WIN95: fat32 db 0x1c ; Hidden WIN95: fat32, LBA-mapped db 0x1e ; Hidden WIN95: fat16, LBA-mapped db 0xc4 ; DRDOS/secured: fat16 <32M db 0xc6 ; DRDOS/secured: fat16 >32M db 0xcb ; DRDOS/secured: fat32 db 0xcc ; DRDOS/secured: fat32, LBA-mapped db 0xce ; DRDOS/secured: fat16, LBA-mapped db 0xd4 ; Old Multiuser DOS secured: fat16 <32M db 0xd6 ; Old Multiuser DOS secured: fat16 >32M db 0x07 ; NTFS db 0x27 ; NTFS, hidden db 0x83 ; Linux native file system (ext2fs) partition_types_end: extended_types: ; list of extended partitions db 0x05 ; DOS: extended partition db 0x0f ; WIN95: extended partition, LBA-mapped db 0xc5 ; DRDOS/secured: extended partition db 0xd5 ; Old Multiuser DOS secured: extended partition extended_types_end: endg ; Partition chain used: ; MBR <--------------------- ; | | ; |-> PARTITION1 | ; |-> EXTENDED PARTITION - ;not need be second partition ; |-> PARTITION3 ; |-> PARTITION4 set_PARTITION_variables: set_FAT32_variables: ;deprecated and [problem_partition], 0 call reserve_hd1 call reserve_hd_channel pushad cmp dword [hdpos],0 je problem_hd xor ecx,ecx ; partition count ;or edx,-1 ; flag for partition xor eax,eax ; address MBR xor ebp,ebp ; extended partition start new_mbr: test ebp,ebp ; is there extended partition? (MBR or EMBR) jnz extended_already_set ; yes xchg ebp,eax ; no. set it now extended_already_set: add eax,ebp ; mbr=mbr+0, ext_part=ext_start+relat_start mov ebx,buffer call hd_read cmp [hd_error],0 jne problem_hd cmp word [ebx+0x1fe],0xaa55 ; is it valid boot sector? jnz end_partition_chain push eax ; push only one time cmp dword [ebx+0x1be+0xc],0 ; skip over empty partition jnz test_primary_partition_0 cmp dword [ebx+0x1be+0xc+16],0 jnz test_primary_partition_1 cmp dword [ebx+0x1be+0xc+16+16],0 jnz test_primary_partition_2 cmp dword [ebx+0x1be+0xc+16+16+16],0 jnz test_primary_partition_3 pop eax jmp end_partition_chain test_primary_partition_0: mov al,[ebx+0x1be+4] ; get primary partition type call scan_partition_types jnz test_primary_partition_1 ; no. skip over inc ecx cmp ecx,[known_part] ; is it wanted partition? jnz test_primary_partition_1 ; no pop eax ;mov edx, eax ; start sector add eax, [ebx+0x1be+8] ; add relative start ;mov [PARTITON_START],edx ;push edx mov edx, [ebx+0x1be+12] ; length ;add edx, eax ; add length ;dec edx ; PARTITION_END is inclusive ;mov [PARTITION_END], edx ; note that this can be changed ; when file system data will be available mov cl, [ebx+0x1be+4] ; fs_type ;mov [fs_type], dl ; save for FS recognizer (separate FAT vs NTFS) ;pop edx jmp hd_and_partition_ok test_primary_partition_1: mov al,[ebx+0x1be+4+16] ; get primary partition type call scan_partition_types jnz test_primary_partition_2 ; no. skip over inc ecx cmp ecx,[known_part] ; is it wanted partition? jnz test_primary_partition_2 ; no pop eax add eax, [ebx+0x1be+8+16] mov edx, [ebx+0x1be+12+16] mov cl, [ebx+0x1be+4+16] jmp hd_and_partition_ok ;mov edx, eax ;add edx, [ebx+0x1be+8+16] ;push edx ;add edx, [ebx+0x1be+12+16] ;dec edx ;mov [PARTITION_END], edx ;mov al, [ebx+0x1be+4+16] ;mov [fs_type], dl ;pop edx test_primary_partition_2: mov al,[ebx+0x1be+4+16+16] ; get primary partition type call scan_partition_types jnz test_primary_partition_3 ; no. skip over inc ecx cmp ecx,[known_part] ; is it wanted partition? jnz test_primary_partition_3 ; no pop eax add eax, [ebx+0x1be+8+16+16] mov edx, [ebx+0x1be+12+16+16] mov cl, [ebx+0x1be+4+16+16] jmp hd_and_partition_ok ;mov edx, eax ;add edx, [ebx+0x1be+8+16+16] ;push edx ;add edx, [ebx+0x1be+12+16+16] ;dec edx ;mov [PARTITION_END], edx ;mov al, [ebx+0x1be+4+16+16] ;mov [fs_type], dl ;pop edx test_primary_partition_3: mov al,[ebx+0x1be+4+16+16+16] ; get primary partition type call scan_partition_types jnz test_ext_partition_0 ; no. skip over inc ecx cmp ecx,[known_part] ; is it wanted partition? jnz test_ext_partition_0 ; no pop eax add eax, [ebx+0x1be+8+16+16+16] mov edx, [ebx+0x1be+12+16+16+16] mov cl, [ebx+0x1be+4+16+16+16] jmp hd_and_partition_ok ;mov edx, eax ;add edx, [ebx+0x1be+8+16+16+16] ;push edx ;add edx, [ebx+0x1be+12+16+16+16] ;dec edx ;mov [PARTITION_END], edx ;mov al, [ebx+0x1be+4+16+16+16] ;mov [fs_type], dl ;pop edx test_ext_partition_0: pop eax ; просто выкидываем из стека mov al,[ebx+0x1be+4] ; get extended partition type call scan_extended_types jnz test_ext_partition_1 mov eax,[ebx+0x1be+8] ; add relative start test eax,eax ; is there extended partition? jnz new_mbr ; yes. read it test_ext_partition_1: mov al,[ebx+0x1be+4+16] ; get extended partition type call scan_extended_types jnz test_ext_partition_2 mov eax,[ebx+0x1be+8+16] ; add relative start test eax,eax ; is there extended partition? jnz new_mbr ; yes. read it test_ext_partition_2: mov al,[ebx+0x1be+4+16+16] ; get extended partition type call scan_extended_types jnz test_ext_partition_3 mov eax,[ebx+0x1be+8+16+16] ; add relative start test eax,eax ; is there extended partition? jnz new_mbr ; yes. read it test_ext_partition_3: mov al,[ebx+0x1be+4+16+16+16] ; get extended partition type call scan_extended_types jnz end_partition_chain ; no. end chain mov eax,[ebx+0x1be+8+16+16+16] ; get start of extended partition test eax,eax ; is there extended partition? jnz new_mbr ; yes. read it end_partition_chain: ;mov [partition_count],ecx ;cmp edx,-1 ; found wanted partition? ;jnz hd_and_partition_ok ; yes. install it ;jmp problem_partition_or_fat problem_hd: or [problem_partition], 2 jmp return_from_part_set scan_partition_types: push ecx mov edi,partition_types mov ecx,partition_types_end-partition_types cld repne scasb ; is partition type ok? pop ecx ret scan_extended_types: push ecx mov edi,extended_types mov ecx,extended_types_end-extended_types cld repne scasb ; is it extended partition? pop ecx ret problem_fat_dec_count: ; bootsector is missing or another problem ; dec [partition_count] ; remove it from partition_count problem_partition_or_fat: or [problem_partition],1 return_from_part_set: popad ;mov [fs_type],0 call free_hd_channel mov [hd1_status],0 ; free ret hd_and_partition_ok: ;eax = PARTITION_START edx=PARTITION_LENGTH cl=fs_type mov [fs_type], cl ;mov eax,edx mov [PARTITION_START],eax add edx, eax dec edx mov [PARTITION_END], edx ; mov edx, [PARTITION_END] ; sub edx, eax ; inc edx ; edx = length of partition зачем оно нам?? ; mov [hd_setup],1 mov ebx,buffer call hd_read ; read boot sector of partition cmp [hd_error], 0 jz boot_read_ok cmp [fs_type], 7 jnz problem_fat_dec_count ; NTFS duplicates bootsector: ; NT4/2k/XP+ saves bootsector copy in the end of disk ; NT 3.51 saves bootsector copy in the middle of disk and [hd_error], 0 mov eax, [PARTITION_END] call hd_read cmp [hd_error], 0 jnz @f call ntfs_test_bootsec jnc boot_read_ok @@: and [hd_error], 0 mov eax, edx shr eax, 1 add eax, [PARTITION_START] call hd_read cmp [hd_error], 0 jnz problem_fat_dec_count ; no chance... boot_read_ok: ; if we are running on NTFS, check bootsector call ntfs_test_bootsec ; test ntfs jnc ntfs_setup call ext2_test_superblock ; test ext2fs jnc ext2_setup mov eax, [PARTITION_START] ;ext2 test changes [buffer] call hd_read cmp [hd_error], 0 jnz problem_fat_dec_count cmp word [ebx+0x1fe],0xaa55 ; is it valid boot sector? jnz problem_fat_dec_count movzx eax,word [ebx+0xe] ; sectors reserved add eax,[PARTITION_START] mov [FAT_START],eax ; fat_start = partition_start + reserved movzx eax,byte [ebx+0xd] ; sectors per cluster test eax,eax jz problem_fat_dec_count mov [SECTORS_PER_CLUSTER],eax movzx ecx,word [ebx+0xb] ; bytes per sector cmp ecx,0x200 jnz problem_fat_dec_count mov [BYTES_PER_SECTOR],ecx movzx eax,word [ebx+0x11] ; count of rootdir entries (=0 fat32) mov edx,32 mul edx dec ecx add eax,ecx ; round up if not equal count inc ecx ; bytes per sector div ecx mov [ROOT_SECTORS],eax ; count of rootdir sectors movzx eax,word [ebx+0x16] ; sectors per fat <65536 test eax,eax jnz fat16_fatsize mov eax,[ebx+0x24] ; sectors per fat fat16_fatsize: mov [SECTORS_PER_FAT],eax movzx eax,byte [ebx+0x10] ; number of fats test eax,eax ; if 0 it's not fat partition jz problem_fat_dec_count mov [NUMBER_OF_FATS],eax imul eax,[SECTORS_PER_FAT] add eax,[FAT_START] mov [ROOT_START],eax ; rootdir = fat_start + fat_size * fat_count add eax,[ROOT_SECTORS] ; rootdir sectors should be 0 on fat32 mov [DATA_START],eax ; data area = rootdir + rootdir_size movzx eax,word [ebx+0x13] ; total sector count <65536 test eax,eax jnz fat16_total mov eax,[ebx+0x20] ; total sector count fat16_total: add eax,[PARTITION_START] dec eax mov [PARTITION_END],eax inc eax sub eax,[DATA_START] ; eax = count of data sectors xor edx,edx div dword [SECTORS_PER_CLUSTER] inc eax mov [LAST_CLUSTER],eax dec eax ; cluster count mov [fatStartScan],2 ; limits by Microsoft Hardware White Paper v1.03 cmp eax,4085 ; 0xff5 jb problem_fat_dec_count ; fat12 not supported cmp eax,65525 ; 0xfff5 jb fat16_partition fat32_partition: mov eax,[ebx+0x2c] ; rootdir cluster mov [ROOT_CLUSTER],eax movzx eax,word [ebx+0x30] ; fs info sector add eax,[PARTITION_START] mov [ADR_FSINFO],eax call hd_read mov eax,[ebx+0x1ec] cmp eax,-1 jz @f mov [fatStartScan],eax @@: popad mov [fatRESERVED],0x0FFFFFF6 mov [fatBAD],0x0FFFFFF7 mov [fatEND],0x0FFFFFF8 mov [fatMASK],0x0FFFFFFF mov [fs_type],32 ; Fat32 call free_hd_channel mov [hd1_status],0 ; free ret fat16_partition: xor eax,eax mov [ROOT_CLUSTER],eax popad mov [fatRESERVED],0x0000FFF6 mov [fatBAD],0x0000FFF7 mov [fatEND],0x0000FFF8 mov [fatMASK],0x0000FFFF mov [fs_type],16 ; Fat16 call free_hd_channel mov [hd1_status],0 ; free ret