Subversion Repositories Kolibri OS

Rev

Rev 983 | Rev 1379 | Go to most recent revision | Details | Compare with Previous | Last modification | View Log | RSS feed

Rev Author Line No. Line
431 serge 1
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
2
;;                                                              ;;
983 diamond 3
;; Copyright (C) KolibriOS team 2004-2008. All rights reserved. ;;
431 serge 4
;; Distributed under terms of the GNU General Public License    ;;
5
;;                                                              ;;
6
;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
7
 
593 mikedld 8
$Revision: 1378 $
9
 
10
 
1378 turbanoff 11
ntfs_test_bootsec:
256 diamond 12
; in: ebx->buffer, edx=size of partition
13
; out: CF set <=> invalid
14
; 1. Name=='NTFS    '
15
        cmp     dword [ebx+3], 'NTFS'
16
        jnz     .no
17
        cmp     dword [ebx+7], '    '
18
        jnz     .no
19
; 2. Number of bytes per sector is the same as for physical device
20
; (that is, 0x200 for hard disk)
21
        cmp     word [ebx+11], 0x200
22
        jnz     .no
23
; 3. Number of sectors per cluster must be power of 2
24
        movzx   eax, byte [ebx+13]
25
        dec     eax
26
        js      .no
27
        test    al, [ebx+13]
28
        jnz     .no
29
; 4. FAT parameters must be zero
30
        cmp     word [ebx+14], 0
31
        jnz     .no
32
        cmp     dword [ebx+16], 0
33
        jnz     .no
34
        cmp     byte [ebx+20], 0
35
        jnz     .no
36
        cmp     word [ebx+22], 0
37
        jnz     .no
38
        cmp     dword [ebx+32], 0
39
        jnz     .no
40
; 5. Number of sectors <= partition size
41
        cmp     dword [ebx+0x2C], 0
42
        ja      .no
43
        cmp     [ebx+0x28], edx
44
        ja      .no
45
; 6. $MFT and $MFTMirr clusters must be within partition
46
        cmp     dword [ebx+0x34], 0
47
        ja      .no
48
        push    edx
49
        movzx   eax, byte [ebx+13]
50
        mul     dword [ebx+0x30]
51
        test    edx, edx
52
        pop     edx
53
        jnz     .no
54
        cmp     eax, edx
55
        ja      .no
56
        cmp     dword [ebx+0x3C], 0
57
        ja      .no
58
        push    edx
59
        movzx   eax, byte [ebx+13]
60
        mul     dword [ebx+0x38]
61
        test    edx, edx
62
        pop     edx
63
        jnz     .no
64
        cmp     eax, edx
65
        ja      .no
66
; 7. Clusters per FRS must be either negative and in [-31,-9] or positive and power of 2
67
        movsx   eax, byte [ebx+0x40]
68
        cmp     al, -31
69
        jl      .no
70
        cmp     al, -9
71
        jle     @f
72
        dec     eax
73
        js      .no
74
        test    [ebx+0x40], al
75
        jnz     .no
76
@@:
77
; 8. Same for clusters per IndexAllocationBuffer
78
        movsx   eax, byte [ebx+0x44]
79
        cmp     al, -31
80
        jl      .no
81
        cmp     al, -9
82
        jle     @f
83
        dec     eax
84
        js      .no
85
        test    [ebx+0x44], al
86
        jnz     .no
87
@@:
88
; OK, this is correct NTFS bootsector
89
        clc
90
        ret
91
.no:
92
; No, this bootsector isn't NTFS
93
        stc
94
        ret
95
 
96
ntfs_setup:             ; CODE XREF: part_set.inc
97
; By given bootsector, initialize some NTFS variables
1378 turbanoff 98
;        call    ntfs_test_bootsec
99
;        jc      problem_fat_dec_count
256 diamond 100
        movzx   eax, byte [ebx+13]
101
        mov     [ntfs_data.sectors_per_cluster], eax
102
        mov     eax, [ebx+0x28]
103
        add     eax, [PARTITION_START]
104
        dec     eax
105
        mov     [PARTITION_END], eax
106
        mov     [fs_type], 1
107
        mov     eax, [ebx+0x30]
108
        mov     [ntfs_data.mft_cluster], eax
109
        mov     eax, [ebx+0x38]
110
        mov     [ntfs_data.mftmirr_cluster], eax
111
        movsx   eax, byte [ebx+0x40]
112
        test    eax, eax
113
        js      .1
114
        mul     [ntfs_data.sectors_per_cluster]
115
        shl     eax, 9
116
        jmp     .2
117
.1:
118
        neg     eax
119
        mov     ecx, eax
120
        mov     eax, 1
121
        shl     eax, cl
122
.2:
123
        mov     [ntfs_data.frs_size], eax
124
        movsx   eax, byte [ebx+0x44]
125
        test    eax, eax
126
        js      .3
127
        mul     [ntfs_data.sectors_per_cluster]
128
        shl     eax, 9
129
        jmp     .4
130
.3:
131
        neg     eax
132
        mov     ecx, eax
133
        mov     eax, 1
134
        shl     eax, cl
135
.4:
136
        mov     [ntfs_data.iab_size], eax
137
; allocate space for buffers
138
        add     eax, [ntfs_data.frs_size]
139
        push    eax
140
        call    kernel_alloc
141
        test    eax, eax
142
        jz      problem_fat_dec_count
143
        mov     [ntfs_data.frs_buffer], eax
144
        add     eax, [ntfs_data.frs_size]
145
        mov     [ntfs_data.iab_buffer], eax
146
; read $MFT disposition
147
        mov     eax, [ntfs_data.mft_cluster]
148
        mul     [ntfs_data.sectors_per_cluster]
149
        call    ntfs_read_frs_sector
150
        cmp     [hd_error], 0
151
        jnz     .usemirr
152
        cmp     dword [ebx], 'FILE'
153
        jnz     .usemirr
154
        call    ntfs_restore_usa_frs
155
        jnc     .mftok
156
.usemirr:
157
        and     [hd_error], 0
158
        mov     eax, [ntfs_data.mftmirr_cluster]
159
        mul     [ntfs_data.sectors_per_cluster]
160
        call    ntfs_read_frs_sector
161
        cmp     [hd_error], 0
162
        jnz     @f
163
        cmp     dword [ebx], 'FILE'
164
        jnz     @f
165
        call    ntfs_restore_usa_frs
166
        jnc     .mftok
167
@@:
168
; $MFT and $MFTMirr invalid!
169
.fail_free_frs:
170
        push    [ntfs_data.frs_buffer]
171
        call    kernel_free
172
        jmp     problem_fat_dec_count
173
.fail_free_mft:
174
        push    [ntfs_data.mft_retrieval]
175
        call    kernel_free
176
        jmp     .fail_free_frs
177
.mftok:
178
; read $MFT table retrieval information
179
; start with one page, increase if not enough (when MFT too fragmented)
180
        push    ebx
181
        push    0x1000
182
        call    kernel_alloc
183
        pop     ebx
184
        test    eax, eax
185
        jz      .fail_free_frs
186
        mov     [ntfs_data.mft_retrieval], eax
187
        and     [ntfs_data.mft_retrieval_size], 0
188
        mov     [ntfs_data.mft_retrieval_alloc], 0x1000/8
189
; $MFT base record must contain unnamed non-resident $DATA attribute
190
        movzx   eax, word [ebx+14h]
191
        add     eax, ebx
192
.scandata:
193
        cmp     dword [eax], -1
194
        jz      .fail_free_mft
195
        cmp     dword [eax], 0x80
196
        jnz     @f
197
        cmp     byte [eax+9], 0
198
        jz      .founddata
199
@@:
200
        add     eax, [eax+4]
201
        jmp     .scandata
202
.founddata:
203
        cmp     byte [eax+8], 0
204
        jz      .fail_free_mft
205
; load first portion of $DATA attribute retrieval information
206
        mov     edx, [eax+0x18]
207
        mov     [ntfs_data.mft_retrieval_end], edx
208
        mov     esi, eax
209
        movzx   eax, word [eax+0x20]
210
        add     esi, eax
211
        sub     esp, 10h
212
.scanmcb:
213
        call    ntfs_decode_mcb_entry
214
        jnc     .scanmcbend
215
        call    .get_mft_retrieval_ptr
216
        mov     edx, [esp]      ; block length
217
        mov     [eax], edx
218
        mov     edx, [esp+8]    ; block addr (relative)
219
        mov     [eax+4], edx
220
        inc     [ntfs_data.mft_retrieval_size]
221
        jmp     .scanmcb
222
.scanmcbend:
223
        add     esp, 10h
224
; there may be other portions of $DATA attribute in auxiliary records;
225
; if they will be needed, they will be loaded later
226
 
227
        mov     [ntfs_data.cur_index_size], 0x1000/0x200
228
        push    0x1000
229
        call    kernel_alloc
230
        test    eax, eax
231
        jz      .fail_free_mft
232
        mov     [ntfs_data.cur_index_buf], eax
233
 
234
        popad
321 diamond 235
        call    free_hd_channel
256 diamond 236
        and     [hd1_status], 0
237
        ret
238
 
239
.get_mft_retrieval_ptr:
240
        pushad
241
        mov     eax, [ntfs_data.mft_retrieval_size]
242
        cmp     eax, [ntfs_data.mft_retrieval_alloc]
243
        jnz     .ok
244
        add     eax, 0x1000/8
245
        mov     [ntfs_data.mft_retrieval_alloc], eax
246
        shl     eax, 3
247
        push    eax
248
        call    kernel_alloc
249
        test    eax, eax
250
        jnz     @f
251
        popad
252
        add     esp, 14h
253
        jmp     .fail_free_mft
254
@@:
255
        mov     esi, [ntfs_data.mft_retrieval]
256
        mov     edi, eax
257
        mov     ecx, [ntfs_data.mft_retrieval_size]
258
        add     ecx, ecx
259
        rep     movsd
260
        push    [ntfs_data.mft_retrieval]
261
        mov     [ntfs_data.mft_retrieval], eax
262
        call    kernel_free
263
        mov     eax, [ntfs_data.mft_retrieval_size]
264
.ok:
265
        shl     eax, 3
266
        add     eax, [ntfs_data.mft_retrieval]
267
        mov     [esp+28], eax
268
        popad
269
        ret
270
 
271
ntfs_read_frs_sector:
272
        push    eax ecx
273
        add     eax, [PARTITION_START]
274
        mov     ecx, [ntfs_data.frs_size]
275
        shr     ecx, 9
276
        mov     ebx, [ntfs_data.frs_buffer]
277
        push    ebx
278
@@:
279
        call    hd_read
280
        cmp     [hd_error], 0
281
        jnz     .fail
282
        add     ebx, 0x200
283
        inc     eax
284
        loop    @b
285
.fail:
286
        pop     ebx
287
        pop     ecx eax
288
        ret
289
 
290
uglobal
291
align 4
292
ntfs_cur_attr   dd      ?
293
ntfs_cur_iRecord dd     ?
294
ntfs_cur_offs   dd      ?       ; in sectors
295
ntfs_cur_size   dd      ?       ; in sectors
296
ntfs_cur_buf    dd      ?
297
ntfs_cur_read   dd      ?       ; [output]
265 diamond 298
ntfs_bCanContinue db    ?
299
                rb      3
256 diamond 300
 
301
ntfs_attrlist_buf       rb      0x400
273 diamond 302
ntfs_attrlist_mft_buf   rb      0x400
256 diamond 303
ntfs_bitmap_buf         rb      0x400
304
 
305
ntfs_attr_iRecord       dd      ?
306
ntfs_attr_iBaseRecord   dd      ?
307
ntfs_attr_offs          dd      ?
308
ntfs_attr_list          dd      ?
257 diamond 309
ntfs_attr_size          dq      ?
262 diamond 310
ntfs_cur_tail           dd      ?
256 diamond 311
endg
312
 
313
ntfs_read_attr:
314
; in: global variables
315
; out: [ntfs_cur_read]
316
        pushad
317
        and     [ntfs_cur_read], 0
318
        cmp     [ntfs_cur_iRecord], 0
319
        jnz     .nomft
320
        cmp     [ntfs_cur_attr], 0x80
321
        jnz     .nomft
322
        mov     eax, [ntfs_data.mft_retrieval_end]
323
        inc     eax
324
        mul     [ntfs_data.sectors_per_cluster]
325
        cmp     eax, [ntfs_cur_offs]
326
        jbe     .nomft
327
; precalculated part of $Mft $DATA
328
        mov     esi, [ntfs_data.mft_retrieval]
329
        mov     eax, [ntfs_cur_offs]
330
        xor     edx, edx
331
        div     [ntfs_data.sectors_per_cluster]
332
; eax = VCN, edx = offset in sectors from beginning of cluster
333
        xor     ecx, ecx        ; ecx will contain LCN
334
.mftscan:
335
        add     ecx, [esi+4]
336
        sub     eax, [esi]
337
        jb      @f
338
        add     esi, 8
339
        push    eax
340
        mov     eax, [ntfs_data.mft_retrieval_end]
341
        shl     eax, 3
342
        add     eax, [ntfs_data.mft_retrieval]
343
        cmp     eax, esi
344
        pop     eax
345
        jnz     .mftscan
346
        jmp     .nomft
347
@@:
348
        push    ecx
349
        add     ecx, eax
350
        add     ecx, [esi]
351
        push    eax
352
        push    edx
353
        mov     eax, [ntfs_data.sectors_per_cluster]
354
        mul     ecx
355
; eax = sector on partition
356
        add     eax, [PARTITION_START]
357
        pop     edx
358
        add     eax, edx
359
        mov     ebx, [ntfs_cur_buf]
360
        pop     ecx
361
        neg     ecx
362
        imul    ecx, [ntfs_data.sectors_per_cluster]
363
        sub     ecx, edx
364
        cmp     ecx, [ntfs_cur_size]
365
        jb      @f
366
        mov     ecx, [ntfs_cur_size]
367
@@:
368
; ecx = number of sequential sectors to read
369
        call    hd_read
370
        cmp     [hd_error], 0
371
        jnz     .errread
372
        add     [ntfs_cur_read], 0x200
373
        dec     [ntfs_cur_size]
374
        inc     [ntfs_cur_offs]
375
        add     ebx, 0x200
376
        mov     [ntfs_cur_buf], ebx
377
        inc     eax
378
        loop    @b
379
        pop     ecx
380
        xor     eax, eax
381
        xor     edx, edx
382
        cmp     [ntfs_cur_size], eax
383
        jz      @f
384
        add     esi, 8
385
        push    eax
386
        mov     eax, [ntfs_data.mft_retrieval_end]
387
        shl     eax, 3
388
        add     eax, [ntfs_data.mft_retrieval]
389
        cmp     eax, esi
390
        pop     eax
391
        jz      .nomft
392
        jmp     .mftscan
393
@@:
394
        popad
395
        ret
396
.errread:
397
        pop     ecx
398
.errret:
399
        stc
400
        popad
401
        ret
402
.nomft:
403
; 1. Read file record.
404
; N.B. This will do recursive call of read_attr for $MFT::$Data.
405
        mov     eax, [ntfs_cur_iRecord]
406
        mov     [ntfs_attr_iRecord], eax
407
        and     [ntfs_attr_list], 0
257 diamond 408
        or      dword [ntfs_attr_size], -1
409
        or      dword [ntfs_attr_size+4], -1
256 diamond 410
        or      [ntfs_attr_iBaseRecord], -1
411
        call    ntfs_read_file_record
412
        test    eax, eax
413
        jz      .errret
414
; 2. Find required attribute.
415
        mov     eax, [ntfs_data.frs_buffer]
416
; a) For auxiliary records, read base record
417
; N.B. If base record is present,
418
;      base iRecord may be 0 (for $Mft), but SequenceNumber is nonzero
419
        cmp     dword [eax+24h], 0
420
        jz      @f
421
        mov     eax, [eax+20h]
422
;        test    eax, eax
423
;        jz      @f
424
.beginfindattr:
425
        mov     [ntfs_attr_iRecord], eax
426
        call    ntfs_read_file_record
427
        test    eax, eax
428
        jz      .errret
429
@@:
430
; b) Scan for required attribute and for $ATTR_LIST
431
        mov     eax, [ntfs_data.frs_buffer]
432
        movzx   ecx, word [eax+14h]
433
        add     eax, ecx
434
        mov     ecx, [ntfs_cur_attr]
435
        and     [ntfs_attr_offs], 0
436
.scanattr:
437
        cmp     dword [eax], -1
438
        jz      .scandone
439
        cmp     dword [eax], ecx
440
        jz      .okattr
441
        cmp     [ntfs_attr_iBaseRecord], -1
442
        jnz     .scancont
443
        cmp     dword [eax], 0x20       ; $ATTR_LIST
444
        jnz     .scancont
445
        mov     [ntfs_attr_list], eax
446
        jmp     .scancont
447
.okattr:
273 diamond 448
; ignore named $DATA attributes (aka NTFS streams)
449
        cmp     ecx, 0x80
450
        jnz     @f
451
        cmp     byte [eax+9], 0
452
        jnz     .scancont
453
@@:
256 diamond 454
        mov     [ntfs_attr_offs], eax
455
.scancont:
456
        add     eax, [eax+4]
457
        jmp     .scanattr
262 diamond 458
.continue:
459
        pushad
460
        and     [ntfs_cur_read], 0
256 diamond 461
.scandone:
462
; c) Check for required offset and length
463
        mov     ecx, [ntfs_attr_offs]
464
        jecxz   .noattr
265 diamond 465
        push    [ntfs_cur_size]
466
        push    [ntfs_cur_read]
256 diamond 467
        call    .doreadattr
265 diamond 468
        pop     edx
469
        pop     eax
256 diamond 470
        jc      @f
265 diamond 471
        cmp     [ntfs_bCanContinue], 0
472
        jz      @f
473
        sub     edx, [ntfs_cur_read]
474
        neg     edx
475
        shr     edx, 9
476
        sub     eax, edx
477
        mov     [ntfs_cur_size], eax
256 diamond 478
        jnz     .not_in_cur
479
@@:
480
        popad
481
        ret
482
.noattr:
483
.not_in_cur:
484
        cmp     [ntfs_cur_attr], 0x20
485
        jz      @f
486
        mov     ecx, [ntfs_attr_list]
487
        test    ecx, ecx
488
        jnz     .lookattr
489
.ret_is_attr:
490
        cmp     [ntfs_attr_offs], 1     ; CF set <=> ntfs_attr_offs == 0
491
        popad
492
        ret
493
.lookattr:
494
; required attribute or required offset was not found in base record;
495
; it may be present in auxiliary records;
496
; scan $ATTR_LIST
497
        mov     eax, [ntfs_attr_iBaseRecord]
498
        cmp     eax, -1
499
        jz      @f
500
        call    ntfs_read_file_record
501
        test    eax, eax
502
        jz      .errret
503
        or      [ntfs_attr_iBaseRecord], -1
504
@@:
505
        push    [ntfs_cur_offs]
506
        push    [ntfs_cur_size]
507
        push    [ntfs_cur_read]
508
        push    [ntfs_cur_buf]
257 diamond 509
        push    dword [ntfs_attr_size]
510
        push    dword [ntfs_attr_size+4]
511
        or      dword [ntfs_attr_size], -1
512
        or      dword [ntfs_attr_size+4], -1
273 diamond 513
        and     [ntfs_cur_offs], 0
256 diamond 514
        mov     [ntfs_cur_size], 2
515
        and     [ntfs_cur_read], 0
273 diamond 516
        mov     eax, ntfs_attrlist_buf
517
        cmp     [ntfs_cur_iRecord], 0
518
        jnz     @f
519
        mov     eax, ntfs_attrlist_mft_buf
520
@@:
521
        mov     [ntfs_cur_buf], eax
522
        push    eax
256 diamond 523
        call    .doreadattr
273 diamond 524
        pop     esi
525
        mov     edx, 1
257 diamond 526
        pop     dword [ntfs_attr_size+4]
527
        pop     dword [ntfs_attr_size]
256 diamond 528
        mov     ebp, [ntfs_cur_read]
529
        pop     [ntfs_cur_buf]
530
        pop     [ntfs_cur_read]
531
        pop     [ntfs_cur_size]
532
        pop     [ntfs_cur_offs]
533
        jc      .errret
534
        or      edi, -1
273 diamond 535
        lea     ebp, [ebp+esi-1Ah]
256 diamond 536
.scanliststart:
537
        mov     eax, [ntfs_cur_attr]
538
.scanlist:
539
        cmp     esi, ebp
540
        jae     .scanlistdone
541
        cmp     eax, [esi]
542
        jz      @f
543
.scanlistcont:
544
        movzx   ecx, word [esi+4]
545
        add     esi, ecx
546
        jmp     .scanlist
547
@@:
273 diamond 548
; ignore named $DATA attributes (aka NTFS streams)
549
        cmp     eax, 0x80
550
        jnz     @f
551
        cmp     byte [esi+6], 0
552
        jnz     .scanlistcont
553
@@:
256 diamond 554
        push    eax
555
        mov     eax, [esi+8]
257 diamond 556
        test    eax, eax
557
        jnz     .testf
558
        mov     eax, dword [ntfs_attr_size]
559
        and     eax, dword [ntfs_attr_size+4]
560
        cmp     eax, -1
561
        jnz     .testfz
562
; if attribute is in auxiliary records, its size is defined only in first
563
        mov     eax, [esi+10h]
564
        call    ntfs_read_file_record
565
        test    eax, eax
566
        jnz     @f
567
.errret_pop:
568
        pop     eax
569
        jmp     .errret
570
@@:
571
        mov     eax, [ntfs_data.frs_buffer]
572
        movzx   ecx, word [eax+14h]
573
        add     eax, ecx
574
        mov     ecx, [ntfs_cur_attr]
575
@@:
576
        cmp     dword [eax], -1
577
        jz      .errret_pop
578
        cmp     dword [eax], ecx
579
        jz      @f
273 diamond 580
.l1:
257 diamond 581
        add     eax, [eax+4]
582
        jmp     @b
583
@@:
273 diamond 584
        cmp     eax, 0x80
585
        jnz     @f
586
        cmp     byte [eax+9], 0
587
        jnz     .l1
588
@@:
257 diamond 589
        cmp     byte [eax+8], 0
590
        jnz     .sdnores
591
        mov     eax, [eax+10h]
592
        mov     dword [ntfs_attr_size], eax
593
        and     dword [ntfs_attr_size+4], 0
594
        jmp     .testfz
595
.sdnores:
596
        mov     ecx, [eax+30h]
597
        mov     dword [ntfs_attr_size], ecx
598
        mov     ecx, [eax+34h]
599
        mov     dword [ntfs_attr_size+4], ecx
600
.testfz:
601
        xor     eax, eax
602
.testf:
256 diamond 603
        imul    eax, [ntfs_data.sectors_per_cluster]
604
        cmp     eax, [ntfs_cur_offs]
605
        pop     eax
606
        ja      @f
607
        mov     edi, [esi+10h]  ; keep previous iRecord
608
        jmp     .scanlistcont
609
@@:
257 diamond 610
.scanlistfound:
256 diamond 611
        cmp     edi, -1
612
        jnz     @f
613
        popad
614
        ret
615
@@:
616
        mov     eax, [ntfs_cur_iRecord]
617
        mov     [ntfs_attr_iBaseRecord], eax
618
        mov     eax, edi
619
        jmp     .beginfindattr
257 diamond 620
.sde:
621
        popad
622
        stc
623
        ret
256 diamond 624
.scanlistdone:
625
        sub     ebp, ntfs_attrlist_buf-1Ah
273 diamond 626
        cmp     [ntfs_cur_iRecord], 0
256 diamond 627
        jnz     @f
273 diamond 628
        sub     ebp, ntfs_attrlist_mft_buf-ntfs_attrlist_buf
629
@@:
256 diamond 630
        cmp     ebp, 0x400
257 diamond 631
        jnz     .scanlistfound
256 diamond 632
        inc     edx
633
        push    esi edi
634
        mov     esi, ntfs_attrlist_buf+0x200
635
        mov     edi, ntfs_attrlist_buf
273 diamond 636
        cmp     [ntfs_cur_iRecord], 0
637
        jnz     @f
638
        mov     esi, ntfs_attrlist_mft_buf+0x200
639
        mov     edi, ntfs_attrlist_mft_buf
640
@@:
256 diamond 641
        mov     ecx, 0x200/4
642
        rep     movsd
273 diamond 643
        mov     eax, edi
256 diamond 644
        pop     edi esi
645
        sub     esi, 0x200
646
        push    [ntfs_cur_offs]
647
        push    [ntfs_cur_size]
648
        push    [ntfs_cur_read]
649
        push    [ntfs_cur_buf]
273 diamond 650
        push    dword [ntfs_attr_size]
651
        push    dword [ntfs_attr_size+4]
652
        or      dword [ntfs_attr_size], -1
653
        or      dword [ntfs_attr_size+4], -1
256 diamond 654
        mov     [ntfs_cur_offs], edx
655
        mov     [ntfs_cur_size], 1
656
        and     [ntfs_cur_read], 0
273 diamond 657
        mov     [ntfs_cur_buf], eax
658
        mov     ecx, [ntfs_attr_list]
256 diamond 659
        push    esi edx
660
        call    .doreadattr
661
        pop     edx esi
662
        mov     ebp, [ntfs_cur_read]
273 diamond 663
        pop     dword [ntfs_attr_size+4]
664
        pop     dword [ntfs_attr_size]
256 diamond 665
        pop     [ntfs_cur_buf]
666
        pop     [ntfs_cur_read]
667
        pop     [ntfs_cur_size]
668
        pop     [ntfs_cur_offs]
669
        jc      .errret
273 diamond 670
        add     ebp, ntfs_attrlist_buf+0x200-0x1A
671
        cmp     [ntfs_cur_iRecord], 0
672
        jnz     .scanliststart
673
        add     ebp, ntfs_attrlist_mft_buf-ntfs_attrlist_buf
256 diamond 674
        jmp     .scanliststart
675
 
676
.doreadattr:
265 diamond 677
        mov     [ntfs_bCanContinue], 0
256 diamond 678
        cmp     byte [ecx+8], 0
679
        jnz     .nonresident
680
        mov     eax, [ecx+10h]  ; length
681
        mov     esi, eax
682
        mov     edx, [ntfs_cur_offs]
683
        shr     eax, 9
684
        cmp     eax, edx
685
        jb      .okret
686
        shl     edx, 9
687
        sub     esi, edx
688
        movzx   eax, word [ecx+14h]
689
        add     edx, eax
690
        add     edx, ecx        ; edx -> data
691
        mov     eax, [ntfs_cur_size]
692
        cmp     eax, (0xFFFFFFFF shr 9)+1
693
        jbe     @f
694
        mov     eax, (0xFFFFFFFF shr 9)+1
695
@@:
696
        shl     eax, 9
697
        cmp     eax, esi
698
        jbe     @f
699
        mov     eax, esi
700
@@:
701
; eax = length, edx -> data
702
        mov     [ntfs_cur_read], eax
703
        mov     ecx, eax
704
        mov     eax, edx
705
        mov     ebx, [ntfs_cur_buf]
706
        call    memmove
257 diamond 707
        and     [ntfs_cur_size], 0      ; CF=0
256 diamond 708
        ret
709
.nonresident:
257 diamond 710
; Not all auxiliary records contain correct FileSize info
711
        mov     eax, dword [ntfs_attr_size]
712
        mov     edx, dword [ntfs_attr_size+4]
713
        push    eax
714
        and     eax, edx
715
        cmp     eax, -1
716
        pop     eax
717
        jnz     @f
256 diamond 718
        mov     eax, [ecx+30h]  ; FileSize
719
        mov     edx, [ecx+34h]
257 diamond 720
        mov     dword [ntfs_attr_size], eax
721
        mov     dword [ntfs_attr_size+4], edx
722
@@:
256 diamond 723
        add     eax, 0x1FF
724
        adc     edx, 0
725
        shrd    eax, edx, 9
726
        sub     eax, [ntfs_cur_offs]
727
        ja      @f
728
; return with nothing read
257 diamond 729
        and     [ntfs_cur_size], 0
256 diamond 730
.okret:
731
        clc
732
        ret
733
@@:
734
; reduce read length
262 diamond 735
        and     [ntfs_cur_tail], 0
256 diamond 736
        cmp     [ntfs_cur_size], eax
737
        jb      @f
738
        mov     [ntfs_cur_size], eax
262 diamond 739
        mov     eax, dword [ntfs_attr_size]
740
        and     eax, 0x1FF
741
        mov     [ntfs_cur_tail], eax
256 diamond 742
@@:
743
        cmp     [ntfs_cur_size], 0
744
        jz      .okret
745
        mov     eax, [ntfs_cur_offs]
746
        xor     edx, edx
747
        div     [ntfs_data.sectors_per_cluster]
748
        sub     eax, [ecx+10h]  ; first_vbo
749
        jb      .okret
750
; eax = cluster, edx = starting sector
751
        sub     esp, 10h
752
        movzx   esi, word [ecx+20h]     ; mcb_info_ofs
753
        add     esi, ecx
754
        xor     ebp, ebp
755
.readloop:
756
        call    ntfs_decode_mcb_entry
757
        jnc     .break
758
        add     ebp, [esp+8]
759
        sub     eax, [esp]
760
        jae     .readloop
761
        push    ecx
762
        push    eax
763
        add     eax, [esp+8]
764
        add     eax, ebp
765
        imul    eax, [ntfs_data.sectors_per_cluster]
766
        add     eax, edx
767
        add     eax, [PARTITION_START]
768
        pop     ecx
769
        neg     ecx
770
        imul    ecx, [ntfs_data.sectors_per_cluster]
771
        sub     ecx, edx
772
        cmp     ecx, [ntfs_cur_size]
773
        jb      @f
774
        mov     ecx, [ntfs_cur_size]
775
@@:
776
        mov     ebx, [ntfs_cur_buf]
777
@@:
778
        call    hd_read
779
        cmp     [hd_error], 0
780
        jnz     .errread2
781
        add     ebx, 0x200
782
        mov     [ntfs_cur_buf], ebx
783
        inc     eax
784
        add     [ntfs_cur_read], 0x200
785
        dec     [ntfs_cur_size]
786
        inc     [ntfs_cur_offs]
787
        loop    @b
788
        pop     ecx
265 diamond 789
        xor     eax, eax
790
        xor     edx, edx
256 diamond 791
        cmp     [ntfs_cur_size], 0
792
        jnz     .readloop
793
        add     esp, 10h
262 diamond 794
        mov     eax, [ntfs_cur_tail]
795
        test    eax, eax
820 diamond 796
        jz      @f
262 diamond 797
        sub     eax, 0x200
798
        add     [ntfs_cur_read], eax
820 diamond 799
@@:
800
        clc
801
        ret
256 diamond 802
.errread2:
803
        pop     ecx
804
        add     esp, 10h
820 diamond 805
        stc
806
        ret
256 diamond 807
.break:
808
        add     esp, 10h        ; CF=0
265 diamond 809
        mov     [ntfs_bCanContinue], 1
256 diamond 810
        ret
811
 
812
ntfs_read_file_record:
813
; in: eax=iRecord
814
; out: [ntfs_data.frs_buffer] contains information
815
;      eax=0 - failed, eax=1 - success
816
; Read attr $DATA of $Mft, starting from eax*[ntfs_data.frs_size]
817
        push    ecx edx
818
        mov     ecx, [ntfs_data.frs_size]
819
        mul     ecx
820
        shrd    eax, edx, 9
821
        shr     edx, 9
822
        jnz     .err
265 diamond 823
        push    [ntfs_attr_iRecord]
824
        push    [ntfs_attr_iBaseRecord]
825
        push    [ntfs_attr_offs]
826
        push    [ntfs_attr_list]
827
        push    dword [ntfs_attr_size+4]
828
        push    dword [ntfs_attr_size]
256 diamond 829
        push    [ntfs_cur_iRecord]
830
        push    [ntfs_cur_attr]
831
        push    [ntfs_cur_offs]
832
        push    [ntfs_cur_size]
833
        push    [ntfs_cur_buf]
834
        push    [ntfs_cur_read]
835
        mov     [ntfs_cur_attr], 0x80   ; $DATA
836
        and     [ntfs_cur_iRecord], 0   ; $Mft
837
        mov     [ntfs_cur_offs], eax
838
        shr     ecx, 9
839
        mov     [ntfs_cur_size], ecx
840
        mov     eax, [ntfs_data.frs_buffer]
841
        mov     [ntfs_cur_buf], eax
842
        call    ntfs_read_attr
843
        mov     eax, [ntfs_cur_read]
844
        pop     [ntfs_cur_read]
845
        pop     [ntfs_cur_buf]
846
        pop     [ntfs_cur_size]
847
        pop     [ntfs_cur_offs]
848
        pop     [ntfs_cur_attr]
849
        pop     [ntfs_cur_iRecord]
265 diamond 850
        pop     dword [ntfs_attr_size]
851
        pop     dword [ntfs_attr_size+4]
852
        pop     [ntfs_attr_list]
853
        pop     [ntfs_attr_offs]
854
        pop     [ntfs_attr_iBaseRecord]
855
        pop     [ntfs_attr_iRecord]
256 diamond 856
        pop     edx ecx
857
        jc      .errret
858
        cmp     eax, [ntfs_data.frs_size]
859
        jnz     .errret
860
        mov     eax, [ntfs_data.frs_buffer]
861
        cmp     dword [eax], 'FILE'
862
        jnz     .errret
863
        push    ebx
864
        mov     ebx, eax
865
        call    ntfs_restore_usa_frs
866
        pop     ebx
867
        setnc   al
868
        movzx   eax, al
869
.ret:
870
        ret
871
.err:
872
        pop     edx ecx
873
.errret:
874
        xor     eax, eax
875
        ret
876
 
877
ntfs_restore_usa_frs:
878
        mov     eax, [ntfs_data.frs_size]
879
ntfs_restore_usa:
880
        pushad
881
        shr     eax, 9
882
        mov     ecx, eax
883
        inc     eax
884
        cmp     [ebx+6], ax
885
        jnz     .err
886
        movzx   eax, word [ebx+4]
887
        lea     esi, [eax+ebx]
888
        lodsw
889
        mov     edx, eax
890
        lea     edi, [ebx+0x1FE]
891
@@:
892
        cmp     [edi], dx
893
        jnz     .err
894
        lodsw
895
        stosw
896
        add     edi, 0x1FE
897
        loop    @b
898
        popad
899
        clc
900
        ret
901
.err:
902
        popad
903
        stc
904
        ret
905
 
906
ntfs_decode_mcb_entry:
907
        push    eax ecx edi
908
        lea     edi, [esp+16]
909
        xor     eax, eax
910
        lodsb
911
        test    al, al
912
        jz      .end
913
        mov     ecx, eax
914
        and     ecx, 0xF
915
        cmp     ecx, 8
916
        ja      .end
917
        push    ecx
918
        rep     movsb
919
        pop     ecx
920
        sub     ecx, 8
921
        neg     ecx
922
        cmp     byte [esi-1], 80h
923
        jae     .end
924
        push    eax
925
        xor     eax, eax
926
        rep     stosb
927
        pop     ecx
928
        shr     ecx, 4
929
        cmp     ecx, 8
930
        ja      .end
931
        push    ecx
932
        rep     movsb
933
        pop     ecx
934
        sub     ecx, 8
935
        neg     ecx
936
        cmp     byte [esi-1], 80h
257 diamond 937
        cmc
256 diamond 938
        sbb     eax, eax
939
        rep     stosb
940
        stc
941
.end:
942
        pop     edi ecx eax
943
        ret
944
 
257 diamond 945
ntfs_find_lfn:
521 diamond 946
; in: esi+ebp -> name
257 diamond 947
; out: CF=1 - file not found
262 diamond 948
;      else CF=0, [ntfs_cur_iRecord] valid, eax->record in parent directory
257 diamond 949
        mov     [ntfs_cur_iRecord], 5   ; start parse from root cluster
950
.doit2:
951
        mov     [ntfs_cur_attr], 0x90   ; $INDEX_ROOT
952
        and     [ntfs_cur_offs], 0
953
        mov     eax, [ntfs_data.cur_index_size]
954
        mov     [ntfs_cur_size], eax
955
        mov     eax, [ntfs_data.cur_index_buf]
956
        mov     [ntfs_cur_buf], eax
957
        call    ntfs_read_attr
958
        jnc     @f
959
.ret:
960
        ret
961
@@:
962
        cmp     [ntfs_cur_read], 0x20
963
        jc      .ret
964
        pushad
965
        mov     esi, [ntfs_data.cur_index_buf]
966
        mov     eax, [esi+14h]
967
        add     eax, 10h
968
        cmp     [ntfs_cur_read], eax
969
        jae     .readok1
970
        add     eax, 1FFh
971
        shr     eax, 9
972
        cmp     eax, [ntfs_data.cur_index_size]
973
        ja      @f
974
.stc_ret:
975
        popad
976
        stc
977
        ret
978
@@:
979
; reallocate
980
        push    eax
981
        push    [ntfs_data.cur_index_buf]
982
        call    kernel_free
983
        pop     eax
984
        mov     [ntfs_data.cur_index_size], eax
985
        push    eax
986
        call    kernel_alloc
987
        test    eax, eax
988
        jnz     @f
989
        and     [ntfs_data.cur_index_size], 0
990
        and     [ntfs_data.cur_index_buf], 0
991
        jmp     .stc_ret
992
@@:
993
        mov     [ntfs_data.cur_index_buf], eax
994
        popad
995
        jmp     .doit2
996
.readok1:
997
        mov     ebp, [esi+8]    ; subnode_size
998
        shr     ebp, 9
999
        cmp     ebp, [ntfs_data.cur_index_size]
1000
        jbe     .ok2
1001
        push    esi ebp
1002
        push    ebp
1003
        call    kernel_alloc
1004
        pop     ebp esi
1005
        test    eax, eax
1006
        jz      .stc_ret
1007
        mov     edi, eax
1008
        mov     ecx, [ntfs_data.cur_index_size]
1009
        shl     ecx, 9-2
1010
        rep     movsd
1011
        mov     esi, eax
1012
        mov     [ntfs_data.cur_index_size], ebp
1013
        push    esi ebp
1014
        push    [ntfs_data.cur_index_buf]
1015
        call    kernel_free
1016
        pop     ebp esi
1017
        mov     [ntfs_data.cur_index_buf], esi
1018
.ok2:
1019
        add     esi, 10h
1020
        mov     edi, [esp+4]
1021
; edi -> name, esi -> current index data, ebp = subnode size
1022
.scanloop:
1023
        add     esi, [esi]
1024
.scanloopint:
1025
        test    byte [esi+0Ch], 2
1026
        jnz     .subnode
1027
        push    esi
1028
        add     esi, 0x52
1029
        movzx   ecx, byte [esi-2]
1030
        push    edi
1031
@@:
1032
        lodsw
1033
        call    uni2ansi_char
1034
        call    char_toupper
1035
        push    eax
1036
        mov     al, [edi]
1037
        inc     edi
820 diamond 1038
        cmp     al, '/'
1039
        jz      .slash
257 diamond 1040
        call    char_toupper
1041
        cmp     al, [esp]
1042
        pop     eax
1043
        loopz   @b
1044
        jz      .found
1045
        pop     edi
1046
        pop     esi
1047
        jb      .subnode
1048
.scanloopcont:
1049
        movzx   eax, word [esi+8]
1050
        add     esi, eax
1051
        jmp     .scanloopint
820 diamond 1052
.slash:
1053
        pop     eax
1054
        pop     edi
1055
        pop     esi
257 diamond 1056
.subnode:
1057
        test    byte [esi+0Ch], 1
1058
        jz      .notfound
1059
        movzx   eax, word [esi+8]
1060
        mov     eax, [esi+eax-8]
1061
        mul     [ntfs_data.sectors_per_cluster]
1062
        mov     [ntfs_cur_offs], eax
1063
        mov     [ntfs_cur_attr], 0xA0   ; $INDEX_ALLOCATION
1064
        mov     [ntfs_cur_size], ebp
1065
        mov     eax, [ntfs_data.cur_index_buf]
1066
        mov     esi, eax
1067
        mov     [ntfs_cur_buf], eax
1068
        call    ntfs_read_attr
1069
        mov     eax, ebp
1070
        shl     eax, 9
1071
        cmp     [ntfs_cur_read], eax
1072
        jnz     .notfound
1073
        cmp     dword [esi], 'INDX'
1074
        jnz     .notfound
1075
        mov     ebx, esi
1076
        call    ntfs_restore_usa
1077
        jc      .notfound
1078
        add     esi, 0x18
1079
        jmp     .scanloop
1080
.notfound:
1081
        popad
1082
        stc
1083
        ret
1084
.found:
1085
        cmp     byte [edi], 0
1086
        jz      .done
1087
        cmp     byte [edi], '/'
1088
        jz      .next
1089
        pop     edi
1090
        pop     esi
1091
        jmp     .scanloopcont
1092
.done:
1093
.next:
1094
        pop     esi
1095
        pop     esi
1096
        mov     eax, [esi]
1097
        mov     [ntfs_cur_iRecord], eax
262 diamond 1098
        mov     [esp+1Ch], esi
257 diamond 1099
        mov     [esp+4], edi
1100
        popad
1101
        inc     esi
1102
        cmp     byte [esi-1], 0
1103
        jnz     .doit2
521 diamond 1104
        test    ebp, ebp
1105
        jz      @f
1106
        mov     esi, ebp
1107
        xor     ebp, ebp
1108
        jmp     .doit2
1109
@@:
257 diamond 1110
        ret
1111
 
256 diamond 1112
;----------------------------------------------------------------
1113
;
1114
;  ntfs_HdRead - read NTFS hard disk
1115
;
1116
;  esi  points to filename
1117
;  ebx  pointer to 64-bit number = first wanted byte, 0+
1118
;       may be ebx=0 - start from first byte
1119
;  ecx  number of bytes to read, 0+
1120
;  edx  mem location to return data
1121
;
1122
;  ret ebx = bytes read or 0xffffffff file not found
1123
;      eax = 0 ok read or other = errormsg
1124
;
1125
;--------------------------------------------------------------
1126
ntfs_HdRead:
262 diamond 1127
        cmp     byte [esi], 0
1128
        jnz     @f
256 diamond 1129
        or      ebx, -1
262 diamond 1130
        push    ERROR_ACCESS_DENIED
1131
        pop     eax
256 diamond 1132
        ret
262 diamond 1133
@@:
1134
        call    ntfs_find_lfn
1135
        jnc     .found
1136
        or      ebx, -1
1137
        push    ERROR_FILE_NOT_FOUND
1138
        pop     eax
1139
        ret
1140
.found:
1141
        mov     [ntfs_cur_attr], 0x80   ; $DATA
1142
        and     [ntfs_cur_offs], 0
1143
        and     [ntfs_cur_size], 0
1144
        call    ntfs_read_attr
1145
        jnc     @f
1146
        or      ebx, -1
1147
        push    ERROR_ACCESS_DENIED
1148
        pop     eax
1149
        ret
1150
@@:
1151
        pushad
1152
        and     dword [esp+10h], 0
1153
        xor     eax, eax
1154
        test    ebx, ebx
1155
        jz      .zero1
1156
        cmp     dword [ebx+4], 0x200
1157
        jb      @f
1158
.eof0:
1159
        popad
1160
        xor     ebx, ebx
1161
.eof:
1162
        push    ERROR_END_OF_FILE
1163
        pop     eax
1164
        ret
1165
@@:
1166
        mov     eax, [ebx]
1167
        test    eax, 0x1FF
1168
        jz      .alignedstart
1169
        push    edx
1170
        mov     edx, [ebx+4]
1171
        shrd    eax, edx, 9
1172
        pop     edx
1173
        mov     [ntfs_cur_offs], eax
1174
        mov     [ntfs_cur_size], 1
1175
        mov     [ntfs_cur_buf], ntfs_bitmap_buf
1176
        call    ntfs_read_attr.continue
1177
        mov     eax, [ebx]
1178
        and     eax, 0x1FF
1179
        lea     esi, [ntfs_bitmap_buf+eax]
1180
        sub     eax, [ntfs_cur_read]
1181
        jae     .eof0
1182
        neg     eax
1183
        push    ecx
1184
        cmp     ecx, eax
1185
        jb      @f
1186
        mov     ecx, eax
1187
@@:
1188
        mov     [esp+10h+4], ecx
1189
        mov     edi, edx
1190
        rep     movsb
1191
        mov     edx, edi
1192
        pop     ecx
1193
        sub     ecx, [esp+10h]
1194
        jnz     @f
1195
.retok:
1196
        popad
1197
        xor     eax, eax
1198
        ret
1199
@@:
1200
        cmp     [ntfs_cur_read], 0x200
1201
        jz      .alignedstart
1202
.eof_ebx:
1203
        popad
1204
        jmp     .eof
1205
.alignedstart:
1206
        mov     eax, [ebx]
1207
        push    edx
1208
        mov     edx, [ebx+4]
1209
        add     eax, 511
1210
        adc     edx, 0
1211
        shrd    eax, edx, 9
1212
        pop     edx
1213
.zero1:
1214
        mov     [ntfs_cur_offs], eax
1215
        mov     [ntfs_cur_buf], edx
1216
        mov     eax, ecx
1217
        shr     eax, 9
1218
        mov     [ntfs_cur_size], eax
1219
        add     eax, [ntfs_cur_offs]
1220
        push    eax
1221
        call    ntfs_read_attr.continue
1222
        pop     [ntfs_cur_offs]
1223
        mov     eax, [ntfs_cur_read]
1224
        add     [esp+10h], eax
1225
        mov     eax, ecx
1226
        and     eax, not 0x1FF
1227
        cmp     [ntfs_cur_read], eax
1228
        jnz     .eof_ebx
1229
        and     ecx, 0x1FF
1230
        jz      .retok
1231
        add     edx, [ntfs_cur_read]
1232
        mov     [ntfs_cur_size], 1
1233
        mov     [ntfs_cur_buf], ntfs_bitmap_buf
1234
        call    ntfs_read_attr.continue
1235
        cmp     [ntfs_cur_read], ecx
1236
        jb      @f
1237
        mov     [ntfs_cur_read], ecx
1238
@@:
1239
        xchg    ecx, [ntfs_cur_read]
1240
        push    ecx
1241
        mov     edi, edx
1242
        mov     esi, ntfs_bitmap_buf
1243
        add     [esp+10h+4], ecx
1244
        rep     movsb
1245
        pop     ecx
1246
        xor     eax, eax
1247
        cmp     ecx, [ntfs_cur_read]
1248
        jz      @f
1249
        mov     al, ERROR_END_OF_FILE
1250
@@:
1251
        mov     [esp+1Ch], eax
1252
        popad
1253
        ret
256 diamond 1254
 
1255
;----------------------------------------------------------------
1256
;
1257
;  ntfs_HdReadFolder - read NTFS hard disk folder
1258
;
1259
;  esi  points to filename
1260
;  ebx  pointer to structure 32-bit number = first wanted block, 0+
1261
;                          & flags (bitfields)
1262
; flags: bit 0: 0=ANSI names, 1=UNICODE names
1263
;  ecx  number of blocks to read, 0+
1264
;  edx  mem location to return data
1265
;
1266
;  ret ebx = blocks read or 0xffffffff folder not found
1267
;      eax = 0 ok read or other = errormsg
1268
;
1269
;--------------------------------------------------------------
1378 turbanoff 1270
ntfs_HdReadFolder:
1271
	xchg	bx, bx
256 diamond 1272
        mov     eax, 5          ; root cluster
1273
        cmp     byte [esi], 0
1274
        jz      .doit
257 diamond 1275
        call    ntfs_find_lfn
1276
        jnc     .doit2
256 diamond 1277
.notfound:
1278
        or      ebx, -1
1279
        push    ERROR_FILE_NOT_FOUND
1280
.pop_ret:
1281
        pop     eax
1282
        ret
1283
.doit:
1284
        mov     [ntfs_cur_iRecord], eax
1285
.doit2:
257 diamond 1286
        mov     [ntfs_cur_attr], 0x10   ; $STANDARD_INFORMATION
1287
        and     [ntfs_cur_offs], 0
1288
        mov     [ntfs_cur_size], 1
1289
        mov     [ntfs_cur_buf], ntfs_bitmap_buf
1290
        call    ntfs_read_attr
1291
        jc      .notfound
256 diamond 1292
        mov     [ntfs_cur_attr], 0x90   ; $INDEX_ROOT
1293
        and     [ntfs_cur_offs], 0
1294
        mov     eax, [ntfs_data.cur_index_size]
1295
        mov     [ntfs_cur_size], eax
1296
        mov     eax, [ntfs_data.cur_index_buf]
1297
        mov     [ntfs_cur_buf], eax
1298
        call    ntfs_read_attr
1299
        jnc     .ok
1300
        cmp     [hd_error], 0
1301
        jz      .notfound
1302
        or      ebx, -1
1303
        push    11
1304
        jmp     .pop_ret
1305
.ok:
1306
        cmp     [ntfs_cur_read], 0x20
1307
        jae     @f
1308
        or      ebx, -1
1309
.fserr:
1310
        push    ERROR_FAT_TABLE
1311
        jmp     .pop_ret
1312
@@:
1313
        pushad
1314
        mov     esi, [ntfs_data.cur_index_buf]
1315
        mov     eax, [esi+14h]
1316
        add     eax, 10h
1317
        cmp     [ntfs_cur_read], eax
1318
        jae     .readok1
1319
        add     eax, 1FFh
1320
        shr     eax, 9
1321
        cmp     eax, [ntfs_data.cur_index_size]
1322
        ja      @f
1323
        popad
1324
        jmp     .fserr
1325
@@:
1326
; reallocate
1327
        push    eax
1328
        push    [ntfs_data.cur_index_buf]
1329
        call    kernel_free
1330
        pop     eax
1331
        mov     [ntfs_data.cur_index_size], eax
1332
        push    eax
1333
        call    kernel_alloc
1334
        test    eax, eax
1335
        jnz     @f
1336
        and     [ntfs_data.cur_index_size], 0
1337
        and     [ntfs_data.cur_index_buf], 0
1338
.nomem:
1339
        popad
1340
        or      ebx, -1
1341
        push    12
1342
        pop     eax
1343
        ret
1344
@@:
1345
        mov     [ntfs_data.cur_index_buf], eax
1346
        popad
1347
        jmp     .doit2
1348
.readok1:
1349
        mov     ebp, [esi+8]    ; subnode_size
1350
        shr     ebp, 9
1351
        cmp     ebp, [ntfs_data.cur_index_size]
1352
        jbe     .ok2
1353
        push    esi ebp
1354
        push    ebp
1355
        call    kernel_alloc
1356
        pop     ebp esi
1357
        test    eax, eax
1358
        jz      .nomem
1359
        mov     edi, eax
1360
        mov     ecx, [ntfs_data.cur_index_size]
1361
        shl     ecx, 9-2
1362
        rep     movsd
1363
        mov     esi, eax
1364
        mov     [ntfs_data.cur_index_size], ebp
1365
        push    esi ebp
1366
        push    [ntfs_data.cur_index_buf]
1367
        call    kernel_free
1368
        pop     ebp esi
1369
        mov     [ntfs_data.cur_index_buf], esi
1370
.ok2:
1371
        add     esi, 10h
1372
        mov     ebx, [esp+10h]
1373
        mov     edx, [esp+14h]
1374
        push    dword [ebx+4]   ; read ANSI/UNICODE name
1375
        mov     ebx, [ebx]
1376
; init header
1377
        mov     edi, edx
1378
        mov     ecx, 32/4
1379
        xor     eax, eax
1380
        rep     stosd
1381
        mov     byte [edx], 1   ; version
1382
        mov     ecx, [esp+4+18h]
1383
        push    edx
1384
        mov     edx, esp
1385
; edi -> BDFE, esi -> current index data, ebp = subnode size, ebx = first wanted block,
1386
; ecx = number of blocks to read
1387
; edx -> parameters block: dd , dd 
257 diamond 1388
        cmp     [ntfs_cur_iRecord], 5
1389
        jz      .skip_specials
1390
; dot and dotdot entries
1391
        push    esi
1392
        xor     esi, esi
1393
        call    .add_special_entry
1394
        inc     esi
1395
        call    .add_special_entry
1396
        pop     esi
1397
.skip_specials:
256 diamond 1398
; at first, dump index root
1399
        add     esi, [esi]
1400
.dump_root:
1401
        test    byte [esi+0Ch], 2
1402
        jnz     .dump_root_done
1403
        call    .add_entry
1404
        movzx   eax, word [esi+8]
1405
        add     esi, eax
1406
        jmp     .dump_root
1407
.dump_root_done:
1408
; now dump all subnodes
1409
        push    ecx edi
1410
        mov     edi, ntfs_bitmap_buf
1411
        mov     [ntfs_cur_buf], edi
1412
        mov     ecx, 0x400/4
1413
        xor     eax, eax
1414
        rep     stosd
1415
        mov     [ntfs_cur_attr], 0xB0   ; $BITMAP
1416
        and     [ntfs_cur_offs], 0
1417
        mov     [ntfs_cur_size], 2
1418
        call    ntfs_read_attr
1419
        pop     edi ecx
1420
        push    0       ; save offset in $BITMAP attribute
1421
        and     [ntfs_cur_offs], 0
1422
.dumploop:
1423
        mov     [ntfs_cur_attr], 0xA0
1424
        mov     [ntfs_cur_size], ebp
1425
        mov     eax, [ntfs_data.cur_index_buf]
1426
        mov     esi, eax
1427
        mov     [ntfs_cur_buf], eax
1428
        push    [ntfs_cur_offs]
257 diamond 1429
        mov     eax, [ntfs_cur_offs]
1430
        imul    eax, ebp
1431
        mov     [ntfs_cur_offs], eax
256 diamond 1432
        call    ntfs_read_attr
1433
        pop     [ntfs_cur_offs]
1434
        mov     eax, ebp
1435
        shl     eax, 9
1436
        cmp     [ntfs_cur_read], eax
1437
        jnz     .done
1438
        push    eax
1439
        mov     eax, [ntfs_cur_offs]
1440
        and     eax, 0x400*8-1
1441
        bt      dword [ntfs_bitmap_buf], eax
1442
        pop     eax
1443
        jnc     .dump_subnode_done
1444
        cmp     dword [esi], 'INDX'
1445
        jnz     .dump_subnode_done
1446
        push    ebx
1447
        mov     ebx, esi
1448
        call    ntfs_restore_usa
1449
        pop     ebx
1450
        jc      .dump_subnode_done
1451
        add     esi, 0x18
1452
        add     esi, [esi]
1453
.dump_subnode:
1454
        test    byte [esi+0Ch], 2
1455
        jnz     .dump_subnode_done
1456
        call    .add_entry
1457
        movzx   eax, word [esi+8]
1458
        add     esi, eax
1459
        jmp     .dump_subnode
1460
.dump_subnode_done:
1461
        inc     [ntfs_cur_offs]
1462
        test    [ntfs_cur_offs], 0x400*8-1
1463
        jnz     .dumploop
1464
        mov     [ntfs_cur_attr], 0xB0
1465
        push    ecx edi
1466
        mov     edi, ntfs_bitmap_buf
1467
        mov     [ntfs_cur_buf], edi
1468
        mov     ecx, 0x400/4
1469
        xor     eax, eax
1470
        rep     stosd
1471
        pop     edi ecx
1472
        pop     eax
1473
        push    [ntfs_cur_offs]
1474
        inc     eax
1475
        mov     [ntfs_cur_offs], eax
1476
        mov     [ntfs_cur_size], 2
1477
        push    eax
1478
        call    ntfs_read_attr
1479
        pop     eax
1480
        pop     [ntfs_cur_offs]
1481
        push    eax
1482
        jmp     .dumploop
1483
.done:
1484
        pop     eax
1485
        pop     edx
1486
        mov     ebx, [edx+4]
1487
        pop     edx
1488
        xor     eax, eax
1489
        dec     ecx
1490
        js      @f
1491
        mov     al, ERROR_END_OF_FILE
1492
@@:
1493
        mov     [esp+1Ch], eax
1494
        mov     [esp+10h], ebx
1495
        popad
1496
        ret
1497
 
257 diamond 1498
.add_special_entry:
1499
        mov     eax, [edx]
1500
        inc     dword [eax+8]   ; new file found
1501
        dec     ebx
1502
        jns     .ret
1503
        dec     ecx
1504
        js      .ret
1505
        inc     dword [eax+4]   ; new file block copied
1506
        mov     eax, [edx+4]
1507
        mov     [edi+4], eax
1508
;        mov     eax, dword [ntfs_bitmap_buf+0x20]
1509
;        or      al, 0x10
1510
        mov     eax, 0x10
1511
        stosd
1512
        scasd
1513
        push    edx
1514
        mov     eax, dword [ntfs_bitmap_buf]
1515
        mov     edx, dword [ntfs_bitmap_buf+4]
1516
        call    ntfs_datetime_to_bdfe
1517
        mov     eax, dword [ntfs_bitmap_buf+0x18]
1518
        mov     edx, dword [ntfs_bitmap_buf+0x1C]
1519
        call    ntfs_datetime_to_bdfe
1520
        mov     eax, dword [ntfs_bitmap_buf+8]
1521
        mov     edx, dword [ntfs_bitmap_buf+0xC]
1522
        call    ntfs_datetime_to_bdfe
1523
        pop     edx
1524
        xor     eax, eax
1525
        stosd
1526
        stosd
1527
        mov     al, '.'
1528
        push    edi ecx
1529
        lea     ecx, [esi+1]
1530
        test    byte [edi-0x24], 1
1531
        jz      @f
1532
        rep     stosw
1533
        pop     ecx
1534
        xor     eax, eax
1535
        stosw
1536
        pop     edi
1537
        add     edi, 520
1538
        ret
1539
@@:
1540
        rep     stosb
1541
        pop     ecx
1542
        xor     eax, eax
1543
        stosb
1544
        pop     edi
1545
        add     edi, 264
1546
.ret:
1547
        ret
1548
 
256 diamond 1549
.add_entry:
1550
; do not return DOS 8.3 names
1551
        cmp     byte [esi+0x51], 2
1552
        jz      .ret
1553
; do not return system files
1554
; ... note that there will be no bad effects if system files also were reported ...
1555
        cmp     dword [esi], 0x10
1556
        jb      .ret
1557
        mov     eax, [edx]
1558
        inc     dword [eax+8]   ; new file found
1559
        dec     ebx
1560
        jns     .ret
1561
        dec     ecx
1562
        js      .ret
1563
        inc     dword [eax+4]   ; new file block copied
1564
        mov     eax, [edx+4]    ; flags
262 diamond 1565
        call    ntfs_direntry_to_bdfe
256 diamond 1566
        push    ecx esi edi
1567
        movzx   ecx, byte [esi+0x50]
1568
        add     esi, 0x52
1569
        test    byte [edi-0x24], 1
1570
        jz      .ansi
1571
        shr     ecx, 1
1572
        rep     movsd
1573
        adc     ecx, ecx
1574
        rep     movsw
1575
        and     word [edi], 0
1576
        pop     edi
1577
        add     edi, 520
1578
        pop     esi ecx
1579
        ret
1580
.ansi:
1581
        jecxz   .skip
1582
@@:
1583
        lodsw
1584
        call    uni2ansi_char
1585
        stosb
1586
        loop    @b
1587
.skip:
1588
        xor     al, al
1589
        stosb
1590
        pop     edi
1591
        add     edi, 264
1592
        pop     esi ecx
1593
        ret
1594
 
262 diamond 1595
ntfs_direntry_to_bdfe:
1596
        mov     [edi+4], eax    ; ANSI/UNICODE name
1597
        mov     eax, [esi+48h]
1598
        test    eax, 0x10000000
1599
        jz      @f
1600
        and     eax, not 0x10000000
1601
        or      al, 0x10
1602
@@:
1603
        stosd
1604
        scasd
1605
        push    edx
1606
        mov     eax, [esi+0x18]
1607
        mov     edx, [esi+0x1C]
1608
        call    ntfs_datetime_to_bdfe
1609
        mov     eax, [esi+0x30]
1610
        mov     edx, [esi+0x34]
1611
        call    ntfs_datetime_to_bdfe
1612
        mov     eax, [esi+0x20]
1613
        mov     edx, [esi+0x24]
1614
        call    ntfs_datetime_to_bdfe
1615
        pop     edx
1616
        mov     eax, [esi+0x40]
1617
        stosd
1618
        mov     eax, [esi+0x44]
1619
        stosd
1620
        ret
1621
 
256 diamond 1622
iglobal
1623
_24             dd      24
1624
_60             dd      60
1625
_10000000       dd      10000000
1626
days400year     dd      365*400+100-4+1
1627
days100year     dd      365*100+25-1
1628
days4year       dd      365*4+1
1629
days1year       dd      365
1630
months  dd      31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31
1631
months2 dd      31, 29, 31, 30, 31, 30, 31, 31, 30, 31, 30, 31
1632
_400            dd      400
1633
_100            dd      100
1634
endg
1635
 
1636
ntfs_datetime_to_bdfe:
1637
; edx:eax = number of 100-nanosecond intervals since January 1, 1601, in UTC
1638
        push    eax
1639
        mov     eax, edx
1640
        xor     edx, edx
1641
        div     [_10000000]
1642
        xchg    eax, [esp]
1643
        div     [_10000000]
1644
        pop     edx
1645
; edx:eax = number of seconds since January 1, 1601
1646
        push    eax
1647
        mov     eax, edx
1648
        xor     edx, edx
1649
        div     [_60]
1650
        xchg    eax, [esp]
1651
        div     [_60]
1652
        mov     [edi], dl
1653
        pop     edx
1654
; edx:eax = number of minutes
1655
        div     [_60]
1656
        mov     [edi+1], dl
1657
; eax = number of hours (note that 2^64/(10^7*60*60) < 2^32)
1658
        xor     edx, edx
1659
        div     [_24]
1660
        mov     [edi+2], dl
1661
        mov     [edi+3], byte 0
1662
; eax = number of days since January 1, 1601
1663
        xor     edx, edx
1664
        div     [days400year]
1665
        imul    eax, 400
1666
        add     eax, 1601
1667
        mov     [edi+6], ax
1668
        mov     eax, edx
1669
        xor     edx, edx
1670
        div     [days100year]
1671
        cmp     al, 4
1672
        jnz     @f
1673
        dec     eax
1674
        add     edx, [days100year]
1675
@@:
1676
        imul    eax, 100
1677
        add     [edi+6], ax
1678
        mov     eax, edx
1679
        xor     edx, edx
1680
        div     [days4year]
1681
        shl     eax, 2
1682
        add     [edi+6], ax
1683
        mov     eax, edx
1684
        xor     edx, edx
1685
        div     [days1year]
1686
        cmp     al, 4
1687
        jnz     @f
1688
        dec     eax
1689
        add     edx, [days1year]
1690
@@:
1691
        add     [edi+6], ax
1692
        push    esi edx
1693
        mov     esi, months
1694
        movzx   eax, word [edi+6]
1695
        test    al, 3
1696
        jnz     .noleap
1697
        xor     edx, edx
1698
        push    eax
1699
        div     [_400]
1700
        pop     eax
1701
        test    edx, edx
1702
        jz      .leap
1703
        xor     edx, edx
1704
        div     [_100]
1705
        test    edx, edx
1706
        jz      .noleap
1707
.leap:
1708
        mov     esi, months2
1709
.noleap:
1710
        pop     edx
1711
        xor     eax, eax
1712
        inc     eax
1713
@@:
1714
        sub     edx, [esi]
1715
        jb      @f
1716
        add     esi, 4
1717
        inc     eax
1718
        jmp     @b
1719
@@:
1720
        add     edx, [esi]
1721
        pop     esi
1722
        inc     edx
1723
        mov     [edi+4], dl
1724
        mov     [edi+5], al
1725
        add     edi, 8
1726
        ret
1727
 
1728
;----------------------------------------------------------------
1729
;
1730
;  ntfs_HdRewrite - write to NTFS hard disk
1731
;
1732
;  esi  points to filename
1733
;  ebx  ignored (reserved)
1734
;  ecx  number of bytes to write, 0+
1735
;  edx  mem location to data
1736
;
1737
;  ret ebx = number of written bytes
1738
;      eax = 0 ok read or other = errormsg
1739
;
1740
;--------------------------------------------------------------
1741
ntfs_HdRewrite:
1742
        xor     ebx, ebx
1743
        mov     eax, ERROR_UNSUPPORTED_FS
1744
        ret
1745
 
1746
;----------------------------------------------------------------
1747
;
1748
;  ntfs_HdWrite - write to NTFS hard disk
1749
;
1750
;  esi  points to filename
1751
;  ebx  pointer to 64-bit number = first wanted byte, 0+
1752
;       may be ebx=0 - start from first byte
1753
;  ecx  number of bytes to write, 0+
1754
;  edx  mem location to data
1755
;
1756
;  ret ebx = bytes written (maybe 0)
1757
;      eax = 0 ok write or other = errormsg
1758
;
1759
;--------------------------------------------------------------
1760
ntfs_HdWrite:
1761
        xor     ebx, ebx
1762
        mov     eax, ERROR_UNSUPPORTED_FS
1763
        ret
1764
 
1765
;----------------------------------------------------------------
1766
;
1767
;  ntfs_HdSetFileEnd - set end of file on NTFS hard disk
1768
;
1769
;  esi  points to filename
1770
;  ebx  points to 64-bit number = new file size
1771
;  ecx  ignored (reserved)
1772
;  edx  ignored (reserved)
1773
;
1774
;  ret eax = 0 ok or other = errormsg
1775
;
1776
;--------------------------------------------------------------
1777
ntfs_HdSetFileEnd:
1778
ntfs_HdSetFileInfo:
1779
;----------------------------------------------------------------
1780
;
1781
;  ntfs_HdDelete - delete file or empty folder from NTFS hard disk
1782
;
1783
;  esi  points to filename
1784
;
1785
;  ret  eax = 0 ok or other = errormsg
1786
;
1787
;--------------------------------------------------------------
1788
ntfs_HdDelete:
1789
        mov     eax, ERROR_UNSUPPORTED_FS
1790
        ret
1791
 
1792
ntfs_HdGetFileInfo:
262 diamond 1793
        cmp     byte [esi], 0
1794
        jnz     @f
1795
        push    2
1796
        pop     eax
256 diamond 1797
        ret
262 diamond 1798
@@:
1799
        call    ntfs_find_lfn
1800
        jnc     .doit
1801
        push    ERROR_FILE_NOT_FOUND
1802
        pop     eax
1803
        cmp     [hd_error], 0
1804
        jz      @f
1805
        mov     al, 11
1806
@@:
1807
        ret
1808
.doit:
1809
        push    esi edi
1810
        mov     esi, eax
1811
        mov     edi, edx
1812
        xor     eax, eax
1813
        call    ntfs_direntry_to_bdfe
1814
        pop     edi esi
1815
        xor     eax, eax
1378 turbanoff 1816
        ret
1817