Source code of Windows XP (NT5)
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

168 lines
3.3 KiB

/*++
Copyright (c) 1994 Microsoft Corporation
Module Name:
lockcode.c
Abstract:
Author:
Chuck Lenzmeier (chuckl) 30-Jan-1994
Manny Weiser (mannyw) 17-May-1994
Revision History:
--*/
#include "Procs.h"
#ifndef QFE_BUILD
#ifdef ALLOC_PRAGMA
#pragma alloc_text( PAGE, NwReferenceUnlockableCodeSection )
#pragma alloc_text( PAGE, NwDereferenceUnlockableCodeSection )
#endif
extern BOOLEAN TimerStop; // From Timer.c
//
// The debug trace level
//
#define Dbg (DEBUG_TRACE_CREATE)
VOID
NwReferenceUnlockableCodeSection (
VOID
)
{
ULONG oldCount;
//
// Lock the lockable code database.
//
ExAcquireResourceExclusiveLite( &NwUnlockableCodeResource, TRUE );
//
// Increment the reference count for the section.
//
oldCount = NwSectionDescriptor.ReferenceCount++;
if ( oldCount == 0 && NwSectionDescriptor.Handle == NULL ) {
//
// This is the first reference to the section. Start the timer.
// Lock our code.
//
NwSectionDescriptor.Handle = MmLockPagableCodeSection( NwSectionDescriptor.Base );
StartTimer( );
} else {
//
// This is not the first reference to the section. The section
// had better be locked!
//
ASSERT( NwSectionDescriptor.Handle != NULL );
//
// Restart the timer if the rdr was stopped but didn't unload.
//
if (TimerStop == TRUE) {
StartTimer();
}
}
DebugTrace(+0, Dbg, "NwReferenceCodeSection %d\n", NwSectionDescriptor.ReferenceCount );
ExReleaseResourceLite( &NwUnlockableCodeResource );
return;
} // NwReferenceUnlockableCodeSection
VOID
NwDereferenceUnlockableCodeSection (
VOID
)
{
ULONG newCount;
//
// Lock the lockable code database.
//
ExAcquireResourceExclusiveLite( &NwUnlockableCodeResource, TRUE );
ASSERT( NwSectionDescriptor.Handle != NULL );
ASSERT( NwSectionDescriptor.ReferenceCount > 0 &&
NwSectionDescriptor.ReferenceCount < 0x7FFF );
//
// Decrement the reference count for the section.
//
newCount = --NwSectionDescriptor.ReferenceCount;
DebugTrace(+0, Dbg, "NwDereferenceCodeSection %d\n", NwSectionDescriptor.ReferenceCount );
ExReleaseResourceLite( &NwUnlockableCodeResource );
return;
} // NwDereferenceUnlockableCodeSection
BOOLEAN
NwUnlockCodeSections(
IN BOOLEAN BlockIndefinitely
)
{
//
// Lock the lockable code database.
//
if (!ExAcquireResourceExclusiveLite( &NwUnlockableCodeResource, BlockIndefinitely )) {
return FALSE; // Avoid potential deadlock in timer.c
}
DebugTrace(+0, Dbg, "NwUnlockCodeSections %d\n", NwSectionDescriptor.ReferenceCount );
if ( NwSectionDescriptor.ReferenceCount == 0 ) {
if ( NwSectionDescriptor.Handle != NULL ) {
//
// This is the last reference to the section. Stop the timer and
// unlock the code.
//
StopTimer();
MmUnlockPagableImageSection( NwSectionDescriptor.Handle );
NwSectionDescriptor.Handle = NULL;
}
ExReleaseResourceLite( &NwUnlockableCodeResource );
return TRUE;
}
ExReleaseResourceLite( &NwUnlockableCodeResource );
return FALSE;
}
#endif