From 0fb3c4300ce49ea01232a7b38e59feaab92a626f Mon Sep 17 00:00:00 2001
From: CleverMouse <theclevermouse@yandex.ru>
Date: Fri, 20 Dec 2013 17:38:01 +0000
Subject: [PATCH] KernelAlloc/KernelFree are called from IRQ handlers in
 network drivers, so make them spinlock-protected instead of mutex-protected

git-svn-id: svn://kolibrios.org@4391 a494cfbc-eb01-0410-851d-a64ba20cac60
---
 kernel/trunk/core/heap.inc | 34 +++++++++++++---------------------
 1 file changed, 13 insertions(+), 21 deletions(-)

diff --git a/kernel/trunk/core/heap.inc b/kernel/trunk/core/heap.inc
index a32294e131..abf1c36709 100644
--- a/kernel/trunk/core/heap.inc
+++ b/kernel/trunk/core/heap.inc
@@ -284,8 +284,7 @@ proc alloc_kernel_space stdcall, size:dword
         cmp     eax, [heap_free]
         ja      .error
 
-        mov     ecx, heap_mutex
-        call    mutex_lock
+        spin_lock_irqsave heap_mutex
 
         mov     eax, [size]
 
@@ -326,11 +325,11 @@ proc alloc_kernel_space stdcall, size:dword
         sub     [edi+block_size], edx
 
         mov     eax, [edi+block_size]
-           calc_index eax
+        calc_index eax
         cmp     eax, [block_ind]
         je      .add_used
 
-           list_del edi
+        list_del edi
 
         mov     ecx, [block_ind]
         lea     edx, [mem_block_list+ecx*8]
@@ -340,13 +339,12 @@ proc alloc_kernel_space stdcall, size:dword
 @@:
         bts     [mem_block_mask], eax
         lea     edx, [mem_block_list+eax*8]  ;edx= list head
-           list_add edi, edx
+        list_add edi, edx
 .add_used:
 
         call    md.add_to_used
 
-        mov     ecx, heap_mutex
-        call    mutex_unlock
+        spin_unlock_irqrestore heap_mutex
         mov     eax, [esi+block_base]
         pop     edi
         pop     esi
@@ -354,7 +352,7 @@ proc alloc_kernel_space stdcall, size:dword
         ret
 
 .m_eq_size:
-           list_del edi
+        list_del edi
         lea     edx, [mem_block_list+ebx*8]
         cmp     edx, [edx]
         jnz     @f
@@ -364,8 +362,7 @@ proc alloc_kernel_space stdcall, size:dword
         jmp     .add_used
 
 .error_unlock:
-        mov     ecx, heap_mutex
-        call    mutex_unlock
+        spin_unlock_irqrestore heap_mutex
 .error:
         xor     eax, eax
         pop     edi
@@ -377,8 +374,7 @@ endp
 align 4
 proc free_kernel_space stdcall uses ebx ecx edx esi edi, base:dword
 
-        mov     ecx, heap_mutex
-        call    mutex_lock
+        spin_lock_irqsave heap_mutex
 
         mov     eax, [base]
 
@@ -446,8 +442,7 @@ proc free_kernel_space stdcall uses ebx ecx edx esi edi, base:dword
         lea     edx, [mem_block_list+eax*8]
         list_add edi, edx
 .m_eq:
-        mov     ecx, heap_mutex
-        call    mutex_unlock
+        spin_unlock_irqrestore heap_mutex
         xor     eax, eax
         not     eax
         ret
@@ -459,8 +454,7 @@ proc free_kernel_space stdcall uses ebx ecx edx esi edi, base:dword
         jmp     .add_block
 
 .fail:
-        mov     ecx, heap_mutex
-        call    mutex_unlock
+        spin_unlock_irqrestore heap_mutex
         xor     eax, eax
         ret
 endp
@@ -544,17 +538,15 @@ proc kernel_free stdcall, base:dword
 
         push    ebx esi
 
-        mov     ecx, heap_mutex
-        call    mutex_lock
+        spin_lock_irqsave heap_mutex
 
         mov     eax, [base]
         call    md.find_used
 
-        mov     ecx, heap_mutex
         cmp     [esi+block_flags], USED_BLOCK
         jne     .fail
 
-        call    mutex_unlock
+        spin_unlock_irqrestore heap_mutex
 
         mov     eax, [esi+block_base]
         mov     ecx, [esi+block_size]
@@ -564,7 +556,7 @@ proc kernel_free stdcall, base:dword
         pop     esi ebx
         ret
 .fail:
-        call    mutex_unlock
+        spin_unlock_irqrestore heap_mutex
         xor     eax, eax
         pop     esi ebx
         ret