| 5,11 → 5,395 |
|---|
| ;; ;; |
| ;; Synhronization for MenuetOS. ;; |
| ;; Author: Halyavin Andrey, halyavin@land.ru ;; |
| ;; ;; |
| ;; ;; |
| ;; ;; |
| ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; |
| |
| $Revision$ |
| |
| align 4 |
| ;struct futex* __fastcall create_futex(int *ptr) |
| create_futex: |
| push ecx |
| mov ecx, sizeof.FUTEX |
| call create_object |
| pop ecx |
| test eax, eax |
| jz .fail |
| |
| mov [eax+FUTEX.magic], 'FUTX' |
| mov [eax+FUTEX.destroy], 0 |
| mov [eax+FUTEX.pointer], ecx |
| lea ecx, [eax+FUTEX.wait_list] |
| list_init ecx |
| mov [eax+FUTEX.flags], 0 |
| .fail: |
| ret |
| |
| iglobal |
| align 4 |
| f77call: |
| dd f77.futex_init ;0 |
| dd f77.futex_wait ;1 |
| dd f77.futex_wake ;2 |
| .end: |
| endg |
| |
| align 4 |
| sys_synchronization: |
| f77: |
| cmp ebx, (f77call.end-f77call)/4 |
| jae .fail |
| |
| jmp dword [f77call+ebx*4] |
| |
| .fail: |
| mov [esp+SYSCALL_STACK._eax], -1 |
| ret |
| |
| |
| align 4 |
| .futex_init: |
| call create_futex |
| test eax, eax |
| jz @F |
| mov eax, [eax+FUTEX.handle] |
| @@: |
| mov [esp+SYSCALL_STACK._eax], eax |
| ret |
| |
| align 4 |
| ;ecx futex handle |
| ;edx control value |
| .futex_wait: |
| cmp ecx, 3 |
| jb .epicfail |
| cmp ecx, (PROC.pdt_0 - PROC.htab)/4 |
| jae .epicfail |
| |
| mov esi, [current_process] |
| mov edi, [esi+PROC.htab+ecx*4] |
| |
| cmp [edi+FUTEX.magic], 'FUTX' |
| jne .epicfail |
| cmp [edi+FUTEX.handle], ecx |
| jne .epicfail |
| |
| mov ecx, [edi+FUTEX.pointer] |
| |
| mov eax, edx |
| lock cmpxchg [ecx], edx ;wait until old_value == new_value |
| jz .wait_slow |
| |
| mov [esp+SYSCALL_STACK._eax], 0 |
| ret |
| |
| .wait_slow: |
| pushfd |
| cli |
| |
| sub esp, sizeof.MUTEX_WAITER |
| mov ebx, [TASK_BASE] |
| mov [ebx+TASKDATA.state], 1 |
| mov [esp+MUTEX_WAITER.task], ebx |
| lea esi, [edi+FUTEX.wait_list] |
| |
| list_add_tail esp, esi ;esp= new waiter, esi= list head |
| |
| .again: |
| call change_task |
| |
| lock cmpxchg [ecx], edx |
| jz .again |
| |
| list_del esp |
| add esp, sizeof.MUTEX_WAITER |
| |
| popfd |
| mov [esp+SYSCALL_STACK._eax], 0 |
| ret |
| |
| .epicfail: |
| mov [esp+SYSCALL_STACK._eax], -1 |
| ret |
| |
| align 4 |
| ;ecx futex handle |
| ;edx threads count |
| .futex_wake: |
| cmp ecx, 3 |
| jb .epicfail |
| cmp ecx, (PROC.pdt_0 - PROC.htab)/4 |
| jae .epicfail |
| |
| mov esi, [current_process] |
| mov edi, [esi+PROC.htab+ecx*4] |
| |
| cmp [edi+FUTEX.magic], 'FUTX' |
| jne .epicfail |
| cmp [edi+FUTEX.handle], ecx |
| jne .epicfail |
| |
| xor ecx, ecx |
| |
| pushfd |
| cli |
| |
| lea ebx, [edi+FUTEX.wait_list] |
| mov esi, [edi+FUTEX.wait_list.next] |
| @@: |
| cmp esi, ebx |
| je @F |
| |
| mov eax, [esi+MUTEX_WAITER.task] |
| mov [eax+TASKDATA.state], 0 |
| |
| mov esi, [esi+MUTEX_WAITER.list.next] |
| inc ecx |
| cmp ecx, edx |
| jb @B |
| @@: |
| popfd |
| mov [esp+SYSCALL_STACK._eax], ecx |
| ret |
| |
| RWSEM_WAITING_FOR_WRITE equ 0 |
| RWSEM_WAITING_FOR_READ equ 1 |
| |
| ;void __fastcall mutex_init(struct mutex *lock) |
| |
| align 4 |
| mutex_init: |
| mov [ecx+MUTEX.wait_list.next], ecx |
| mov [ecx+MUTEX.wait_list.prev], ecx |
| mov [ecx+MUTEX.count], 1 |
| ret |
| |
| ;void __fastcall mutex_lock(struct mutex *lock) |
| |
| align 4 |
| mutex_lock: |
| |
| dec [ecx+MUTEX.count] |
| jns .done |
| |
| pushfd |
| cli |
| |
| sub esp, sizeof.MUTEX_WAITER |
| |
| list_add_tail esp, ecx ;esp= new waiter, ecx= list head |
| |
| mov edx, [TASK_BASE] |
| mov [esp+MUTEX_WAITER.task], edx |
| |
| .forever: |
| |
| mov eax, -1 |
| xchg eax, [ecx+MUTEX.count] |
| dec eax |
| jz @F |
| |
| mov [edx+TASKDATA.state], 1 |
| call change_task |
| jmp .forever |
| @@: |
| mov eax, ecx |
| list_del esp |
| |
| cmp [eax+MUTEX.wait_list.next], eax |
| jne @F |
| |
| mov [eax+MUTEX.count], 0 |
| @@: |
| add esp, sizeof.MUTEX_WAITER |
| |
| popfd |
| .done: |
| ret |
| |
| ;void __fastcall mutex_unlock(struct mutex *lock) |
| |
| align 4 |
| mutex_unlock: |
| |
| pushfd |
| cli |
| |
| mov eax, [ecx+MUTEX.wait_list.next] |
| cmp eax, ecx |
| mov [ecx+MUTEX.count], 1 |
| je @F |
| |
| mov eax, [eax+MUTEX_WAITER.task] |
| mov [eax+TASKDATA.state], 0 |
| @@: |
| popfd |
| ret |
| |
| |
| ;void __fastcall init_rwsem(struct rw_semaphore *sem) |
| |
| align 4 |
| init_rwsem: |
| mov [ecx+RWSEM.wait_list.next], ecx |
| mov [ecx+RWSEM.wait_list.prev], ecx |
| mov [ecx+RWSEM.count], 0 |
| ret |
| |
| ;void __fastcall down_read(struct rw_semaphore *sem) |
| |
| align 4 |
| down_read: |
| pushfd |
| cli |
| |
| mov eax, [ecx+RWSEM.count] |
| test eax, eax |
| js @F |
| |
| cmp ecx, [ecx+RWSEM.wait_list.next] |
| je .ok |
| @@: |
| sub esp, sizeof.MUTEX_WAITER |
| |
| mov eax, [TASK_BASE] |
| mov [esp+MUTEX_WAITER.task], eax |
| mov [esp+MUTEX_WAITER.type], RWSEM_WAITING_FOR_READ |
| mov [eax+TASKDATA.state], 1 |
| |
| list_add_tail esp, ecx ;esp= new waiter, ecx= list head |
| |
| call change_task |
| |
| add esp, sizeof.MUTEX_WAITER |
| popfd |
| ret |
| .ok: |
| inc eax |
| mov [ecx+RWSEM.count], eax |
| |
| popfd |
| ret |
| |
| ;void __fastcall down_write(struct rw_semaphore *sem) |
| |
| align 4 |
| down_write: |
| pushfd |
| cli |
| sub esp, sizeof.MUTEX_WAITER |
| |
| mov edx, [TASK_BASE] |
| mov [esp+MUTEX_WAITER.task], edx |
| mov [esp+MUTEX_WAITER.type], RWSEM_WAITING_FOR_WRITE |
| mov [edx+TASKDATA.state], 1 |
| |
| list_add_tail esp, ecx ;esp= new waiter, ecx= list head |
| |
| xor eax, eax |
| not eax |
| |
| .forever: |
| test eax, [ecx+RWSEM.count] |
| jz @F |
| |
| mov [edx+TASKDATA.state], 1 |
| call change_task |
| jmp .forever |
| @@: |
| mov [ecx+RWSEM.count], eax |
| list_del esp |
| |
| add esp, sizeof.MUTEX_WAITER |
| popfd |
| ret |
| |
| ;void __fastcall up_read(struct rw_semaphore *sem) |
| |
| align 4 |
| up_read: |
| pushfd |
| cli |
| |
| dec [ecx+RWSEM.count] |
| jnz @F |
| |
| mov eax, [ecx+RWSEM.wait_list.next] |
| cmp eax, ecx |
| je @F |
| |
| mov eax, [eax+MUTEX_WAITER.task] |
| mov [eax+TASKDATA.state], 0 |
| @@: |
| popfd |
| ret |
| |
| ;void __fastcall up_write(struct rw_semaphore *sem) |
| |
| align 4 |
| up_write: |
| |
| pushfd |
| cli |
| |
| mov eax, [ecx+RWSEM.wait_list.next] |
| mov [ecx+RWSEM.count], 0 |
| |
| cmp ecx, eax |
| je .done |
| |
| mov edx, [eax+MUTEX_WAITER.type] |
| test edx, edx |
| jnz .wake |
| |
| mov eax, [eax+MUTEX_WAITER.task] |
| mov [eax+TASKDATA.state], 0 |
| .done: |
| popfd |
| ret |
| |
| .wake: |
| push ebx |
| push esi |
| push edi |
| |
| xor esi, esi |
| mov edi, ecx |
| |
| .wake_list: |
| |
| mov ebx, [eax+MUTEX_WAITER.list.next] |
| list_del eax |
| mov edx, [eax+MUTEX_WAITER.task] |
| mov [edx+TASKDATA.state], 0 |
| inc esi |
| cmp edi, ebx |
| je .wake_done |
| |
| mov ecx, [ebx+MUTEX_WAITER.type] |
| test ecx, ecx |
| jz .wake_done |
| |
| mov eax, ebx |
| jmp .wake_list |
| |
| .wake_done: |
| add [edi+RWSEM.count], esi |
| |
| pop edi |
| pop esi |
| pop ebx |
| popfd |
| ret |
| |
| |
| purge MUTEX_WAITER |
| purge RWSEM_WAITING_FOR_WRITE |
| purge RWSEM_WAITING_FOR_READ |
| |
| |
| if ~defined sync_inc |
| sync_inc_fix: |
| sync_inc fix sync_inc_fix |