ReceiveTempSMS
/
Sweden-Number
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

ntoskrnl.exe: Implement ExAcquireResourceSharedLite(). 

Signed-off-by: Zebediah Figura <z.figura12@gmail.com>
Signed-off-by: Alexandre Julliard <julliard@winehq.org>
This commit is contained in:
Zebediah Figura 2019-04-08 22:30:12 -05:00 committed by Alexandre Julliard
parent 257c56f5f3
commit dcfc7ab165
3 changed files with 84 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
dlls/ntoskrnl.exe/ntoskrnl.exe.spec
View File

@ -121,7 +121,7 @@
@ stdcall DbgQueryDebugFilterState(long long)
@ stub DbgSetDebugFilterState
@ stdcall ExAcquireResourceExclusiveLite(ptr long)
@ stub ExAcquireResourceSharedLite
@ stdcall ExAcquireResourceSharedLite(ptr long)
@ stub ExAcquireSharedStarveExclusive
@ stub ExAcquireSharedWaitForExclusive
@ stub ExAllocateFromPagedLookasideList

82
dlls/ntoskrnl.exe/sync.c
View File

@ -19,6 +19,7 @@
*/
#include "config.h"
#include <limits.h>
#include <stdarg.h>
#include "ntstatus.h"
@ -742,6 +743,25 @@ NTSTATUS WINAPI ExInitializeResourceLite( ERESOURCE *resource )
return STATUS_SUCCESS;
}
/* Find an existing entry in the shared owner list, or create a new one. */
static OWNER_ENTRY *resource_get_shared_entry( ERESOURCE *resource, ERESOURCE_THREAD thread )
{
ULONG i, count;
for (i = 0; i < resource->OwnerEntry.TableSize; ++i)
{
if (resource->OwnerTable[i].OwnerThread == thread)
return &resource->OwnerTable[i];
}
count = ++resource->OwnerEntry.TableSize;
resource->OwnerTable = heap_realloc(resource->OwnerTable, count * sizeof(*resource->OwnerTable));
resource->OwnerTable[count - 1].OwnerThread = thread;
resource->OwnerTable[count - 1].OwnerCount = 0;
return &resource->OwnerTable[count - 1];
}
/***********************************************************************
* ExAcquireResourceExclusiveLite (NTOSKRNL.EXE.@)
*/
@ -798,3 +818,65 @@ BOOLEAN WINAPI ExAcquireResourceExclusiveLite( ERESOURCE *resource, BOOLEAN wait
return TRUE;
}
/***********************************************************************
* ExAcquireResourceSharedLite (NTOSKRNL.EXE.@)
*/
BOOLEAN WINAPI ExAcquireResourceSharedLite( ERESOURCE *resource, BOOLEAN wait )
{
OWNER_ENTRY *entry;
KIRQL irql;
TRACE("resource %p, wait %u.\n", resource, wait);
KeAcquireSpinLock( &resource->SpinLock, &irql );
entry = resource_get_shared_entry( resource, (ERESOURCE_THREAD)KeGetCurrentThread() );
if (resource->Flag & ResourceOwnedExclusive)
{
if (resource->OwnerEntry.OwnerThread == (ERESOURCE_THREAD)KeGetCurrentThread())
{
/* We own the resource exclusively, so increase recursion. */
resource->ActiveEntries++;
KeReleaseSpinLock( &resource->SpinLock, irql );
return TRUE;
}
}
else if (entry->OwnerCount || !resource->NumberOfExclusiveWaiters)
{
/* Either we already own the resource shared, or there are no exclusive
* owners or waiters, so we can grab it shared. */
entry->OwnerCount++;
resource->ActiveEntries++;
KeReleaseSpinLock( &resource->SpinLock, irql );
return TRUE;
}
if (!wait)
{
KeReleaseSpinLock( &resource->SpinLock, irql );
return FALSE;
}
if (!resource->SharedWaiters)
{
resource->SharedWaiters = heap_alloc( sizeof(*resource->SharedWaiters) );
KeInitializeSemaphore( resource->SharedWaiters, 0, INT_MAX );
}
resource->NumberOfSharedWaiters++;
KeReleaseSpinLock( &resource->SpinLock, irql );
KeWaitForSingleObject( resource->SharedWaiters, Executive, KernelMode, FALSE, NULL );
KeAcquireSpinLock( &resource->SpinLock, &irql );
entry->OwnerCount++;
resource->ActiveEntries++;
resource->NumberOfSharedWaiters--;
KeReleaseSpinLock( &resource->SpinLock, irql );
return TRUE;
}

1
include/ddk/wdm.h
View File

@ -1513,6 +1513,7 @@ static inline void IoSetCompletionRoutine(IRP *irp, PIO_COMPLETION_ROUTINE routi
NTSTATUS WINAPI DbgQueryDebugFilterState(ULONG, ULONG);
void WINAPI ExAcquireFastMutexUnsafe(PFAST_MUTEX);
BOOLEAN WINAPI ExAcquireResourceExclusiveLite(ERESOURCE*,BOOLEAN);
BOOLEAN WINAPI ExAcquireResourceSharedLite(ERESOURCE*,BOOLEAN);
PVOID WINAPI ExAllocatePool(POOL_TYPE,SIZE_T);
PVOID WINAPI ExAllocatePoolWithQuota(POOL_TYPE,SIZE_T);
PVOID WINAPI ExAllocatePoolWithTag(POOL_TYPE,SIZE_T,ULONG);