137 lines
3.5 KiB
C
137 lines
3.5 KiB
C
|
/*++
|
|||
|
|
|
|||
|
|
Copyright (c) 1989 Microsoft Corporation
|
|||
|
|
|
|||
|
|
Module Name:
|
|||
|
|
|
|||
|
|
tsecomm.c
|
|||
|
|
|
|||
|
|
Abstract:
|
|||
|
|
|
|||
|
|
Common security definitions and routines.
|
|||
|
|
|
|||
|
|
This module defines macros that provide a mode-independent
|
|||
|
|
interface for security test procedures.
|
|||
|
|
|
|||
|
|
The mode must be specified by defining one, but not both,
|
|||
|
|
of:
|
|||
|
|
_TST_USER_ (for user mode tests)
|
|||
|
|
_TST_KERNEL_ (for kernel mode tests)
|
|||
|
|
|
|||
|
|
Author:
|
|||
|
|
|
|||
|
|
Jim Kelly (JimK) 23-Mar-1990
|
|||
|
|
|
|||
|
|
Environment:
|
|||
|
|
|
|||
|
|
Test of security.
|
|||
|
|
|
|||
|
|
Revision History:
|
|||
|
|
|
|||
|
|
--*/
|
|||
|
|
|
|||
|
|
#ifndef _TSECOMM_
|
|||
|
|
#define _TSECOMM_
|
|||
|
|
|
|||
|
|
|
|||
|
|
////////////////////////////////////////////////////////////////
|
|||
|
|
// //
|
|||
|
|
// Common Definitions //
|
|||
|
|
// //
|
|||
|
|
////////////////////////////////////////////////////////////////
|
|||
|
|
|
|||
|
|
#define SEASSERT_SUCCESS(s) { \
|
|||
|
|
if (!NT_SUCCESS((s))) { \
|
|||
|
|
DbgPrint("** ! Failed ! **\n"); \
|
|||
|
|
DbgPrint("Status is: 0x%lx \n", (s)); \
|
|||
|
|
} \
|
|||
|
|
ASSERT(NT_SUCCESS(s)); }
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
////////////////////////////////////////////////////////////////
|
|||
|
|
// //
|
|||
|
|
// Kernel Mode Definitions //
|
|||
|
|
// //
|
|||
|
|
////////////////////////////////////////////////////////////////
|
|||
|
|
|
|||
|
|
#ifdef _TST_KERNEL_
|
|||
|
|
|
|||
|
|
#define TstAllocatePool(PoolType,NumberOfBytes) \
|
|||
|
|
(ExAllocatePool( (PoolType), (NumberOfBytes) ))
|
|||
|
|
|
|||
|
|
#define TstDeallocatePool(Pointer, NumberOfBytes) \
|
|||
|
|
(ExFreePool( (Pointer) ))
|
|||
|
|
|
|||
|
|
#endif // _TST_KERNEL_
|
|||
|
|
|
|||
|
|
|
|||
|
|
////////////////////////////////////////////////////////////////
|
|||
|
|
// //
|
|||
|
|
// User Mode Definitions //
|
|||
|
|
// //
|
|||
|
|
////////////////////////////////////////////////////////////////
|
|||
|
|
|
|||
|
|
|
|||
|
|
#ifdef _TST_USER_
|
|||
|
|
|
|||
|
|
#include <nt.h>
|
|||
|
|
#include <ntrtl.h>
|
|||
|
|
#include <nturtl.h>
|
|||
|
|
|
|||
|
|
|
|||
|
|
#include "sep.h"
|
|||
|
|
|
|||
|
|
#define TstAllocatePool(IgnoredPoolType,NumberOfBytes) \
|
|||
|
|
(ITstAllocatePool( (NumberOfBytes) ))
|
|||
|
|
|
|||
|
|
#define TstDeallocatePool(Pointer, NumberOfBytes) \
|
|||
|
|
(ITstDeallocatePool((Pointer),(NumberOfBytes) ))
|
|||
|
|
|
|||
|
|
PVOID
|
|||
|
|
ITstAllocatePool(
|
|||
|
|
IN ULONG NumberOfBytes
|
|||
|
|
)
|
|||
|
|
{
|
|||
|
|
NTSTATUS Status;
|
|||
|
|
PVOID PoolAddress = NULL;
|
|||
|
|
ULONG RegionSize;
|
|||
|
|
|
|||
|
|
RegionSize = NumberOfBytes;
|
|||
|
|
|
|||
|
|
Status = NtAllocateVirtualMemory( NtCurrentProcess(),
|
|||
|
|
&PoolAddress,
|
|||
|
|
0,
|
|||
|
|
&RegionSize,
|
|||
|
|
MEM_COMMIT,
|
|||
|
|
PAGE_READWRITE
|
|||
|
|
);
|
|||
|
|
|
|||
|
|
return PoolAddress;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
VOID
|
|||
|
|
ITstDeallocatePool(
|
|||
|
|
IN PVOID Pointer,
|
|||
|
|
IN ULONG NumberOfBytes
|
|||
|
|
)
|
|||
|
|
{
|
|||
|
|
NTSTATUS Status;
|
|||
|
|
PVOID PoolAddress;
|
|||
|
|
ULONG RegionSize;
|
|||
|
|
|
|||
|
|
RegionSize = NumberOfBytes;
|
|||
|
|
PoolAddress = Pointer;
|
|||
|
|
|
|||
|
|
Status = NtFreeVirtualMemory( NtCurrentProcess(),
|
|||
|
|
&PoolAddress,
|
|||
|
|
&RegionSize,
|
|||
|
|
MEM_DECOMMIT
|
|||
|
|
);
|
|||
|
|
|
|||
|
|
return;
|
|||
|
|
}
|
|||
|
|
#endif // _TST_USER_
|
|||
|
|
|
|||
|
|
#endif //_TSECOMM_
|
|||
|
|
|