From d3eb673c05bd403985aa5bfd72ebbb3b07e0a923 Mon Sep 17 00:00:00 2001
From: pathoswithin <pathoswithin@kolibrios.org>
Date: Sun, 4 Dec 2016 10:56:24 +0000
Subject: [PATCH] NTFS fix

git-svn-id: svn://kolibrios.org@6781 a494cfbc-eb01-0410-851d-a64ba20cac60
---
 kernel/trunk/fs/ntfs.inc | 72 +++++++++++++++++++---------------------
 1 file changed, 35 insertions(+), 37 deletions(-)

diff --git a/kernel/trunk/fs/ntfs.inc b/kernel/trunk/fs/ntfs.inc
index 8bfb8c4e93..5dc3b0b419 100644
--- a/kernel/trunk/fs/ntfs.inc
+++ b/kernel/trunk/fs/ntfs.inc
@@ -3192,16 +3192,18 @@ ntfsSpaceAlloc:
 ;   out:
 ; [ebp+NTFS.fileDataStart] = allocated block starting cluster
 ; CF=1 -> disk full
-        push    eax
         mov     ecx, [ebp+NTFS.BitmapBuffer]
         add     edi, ecx
         add     ecx, [ebp+NTFS.BitmapSize]
         sub     ecx, edi
-        jnc     @f
+        ja      @f
+        push    eax
         call    bitmapBuffering
+        pop     eax
         shl     ecx, 2
 @@:
         shr     ecx, 2
+        push    ecx
         mov     eax, [ebp+NTFS.fileDataSize]
         shr     eax, 5
         jz      .small
@@ -3254,59 +3256,56 @@ ntfsSpaceAlloc:
         add     esp, 8
         jmp     .start
 
-.small:     ; less than 32 clusters
-        mov     eax, -1
-        repz scasd          ; search for zero bits
-        test    ecx, ecx
-        jnz     @f
-        call    bitmapBuffering
-        jmp     .small
 @@:
         sub     edi, 4
-        mov     eax, [edi]
+        call    bitmapBuffering
+        push    ecx
+.small:     ; less than 32 clusters
+        pop     ecx
+        or      eax, -1
+        repz scasd
+        jecxz   @b
+        push    ecx
+        mov     eax, [edi-4]
         not     eax
 @@:
-        bsf     ebx, eax    ; first 0
-        jz      .again
+        bsf     ecx, eax    ; first 0
+        jz      .small
         not     eax
         shr     eax, cl
         shl     eax, cl
         bsf     edx, eax    ; next 1
         jz      @f
-        sub     edx, ebx
+        sub     edx, ecx
         cmp     edx, [ebp+NTFS.fileDataSize]
         jnc     .got        ; fits inside
-        bsf     ebx, eax
+        bsf     ecx, eax
         not     eax
         shr     eax, cl
         shl     eax, cl
         jmp     @b
+
 @@:         ; next dword
-        mov     eax, [edi+4]
+        mov     eax, [edi]
         bsf     edx, eax
         jz      .got        ; empty
         add     edx, 32
-        sub     edx, ebx
+        sub     edx, ecx
         cmp     edx, [ebp+NTFS.fileDataSize]
-        jnc     .got        ; share between dwords
-.again:
-        add     edi, 4
-        jmp     .small
-
+        jc      .small
 .got:
-        push    ebx         ; starting bit
+        sub     edi, 4
+        push    ecx         ; starting bit
         push    edi         ; starting dword
 .done:      ; mark space
-        mov     ecx, [esp+4]
+        pop     edi ecx
         cmp     ecx, 32
         jc      @f
         xor     ecx, ecx
-        add     dword [esp], 4
-        mov     [esp+4], ecx
+        add     edi, 4
 @@:
-        mov     edi, [esp]
-        xor     eax, eax
-        dec     eax
+        push    ecx edi
+        or      eax, -1
         shr     eax, cl
         shl     eax, cl
         neg     ecx
@@ -3324,8 +3323,7 @@ ntfsSpaceAlloc:
         push    ecx
         shr     ecx, 5
         add     edi, 4
-        xor     eax, eax
-        dec     eax
+        or      eax, -1
         rep stosd
         pop     ecx
         and     ecx, 31
@@ -3362,7 +3360,6 @@ ntfsSpaceFree:
         mov     eax, edi
         add     eax, ebx
         shr     eax, 3
-        inc     eax
         cmp     eax, [ebp+NTFS.BitmapSize]
         jc      @f
         add     eax, [ebp+NTFS.BitmapBuffer]
@@ -3447,17 +3444,18 @@ bitmapBuffering:
         call    fs_read64_app
         test    eax, eax
         jnz     .err
-        add     [ebp+NTFS.BitmapSize], 8000h
-        mov     eax, [ebp+NTFS.BitmapTotalSize]
-        cmp     eax, [ebp+NTFS.BitmapSize]
+        mov     eax, [ebp+NTFS.BitmapSize]
+        add     eax, 8000h
+        cmp     [ebp+NTFS.BitmapTotalSize], eax
         jnc     @f
-        mov     [ebp+NTFS.BitmapSize], eax
+        mov     eax, [ebp+NTFS.BitmapTotalSize]
 @@:
+        mov     [ebp+NTFS.BitmapSize], eax
         pop     ebx
         mov     ecx, [ebp+NTFS.BitmapBuffer]
-        add     ecx, [ebp+NTFS.BitmapSize]
+        add     ecx, eax
         sub     ecx, edi
-        jc      bitmapBuffering
+        jbe     bitmapBuffering
         shr     ecx, 2
         ret