Public/winfsp
1
0
Fork 0
mirror of https://github.com/winfsp/winfsp.git synced 2025-04-22 16:33:02 -05:00
Code Issues Packages Projects Releases Wiki Activity

sys: POOL_NX_OPTIN

Browse Source
This commit is contained in:
Bill Zissimopoulos 2016-12-14 15:06:00 -08:00
parent 08c283f2a9
commit 40735c4687
3 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/sys/driver.c
View File

@ -176,6 +176,8 @@ NTSTATUS DriverEntry(
static VOID FspDriverMultiVersionInitialize(VOID) static VOID FspDriverMultiVersionInitialize(VOID)
{ {
ExInitializeDriverRuntime(DrvRtPoolNxOptIn);
if (RtlIsNtDdiVersionAvailable(NTDDI_WIN7)) if (RtlIsNtDdiVersionAvailable(NTDDI_WIN7))
{ {
UNICODE_STRING Name; UNICODE_STRING Name;

2
src/sys/driver.h
View File

@ -19,6 +19,8 @@
#define WINFSP_SYS_DRIVER_H_INCLUDED #define WINFSP_SYS_DRIVER_H_INCLUDED
#define WINFSP_SYS_INTERNAL #define WINFSP_SYS_INTERNAL
#define POOL_NX_OPTIN 1
#include <ntifs.h> #include <ntifs.h>
#include <ntstrsafe.h> #include <ntstrsafe.h>
#include <wdmsec.h> #include <wdmsec.h>

12
src/sys/util.c
View File

@ -851,8 +851,13 @@ NTSTATUS FspSafeMdlCreate(PMDL UserMdl, LOCK_OPERATION Operation, FSP_SAFE_MDL *
{ {
PAGED_CODE(); PAGED_CODE();
NTSTATUS Result; *PSafeMdl = 0;
PVOID VirtualAddress = MmGetSystemAddressForMdlSafe(UserMdl, NormalPagePriority); PVOID VirtualAddress = MmGetSystemAddressForMdlSafe(UserMdl, NormalPagePriority);
if (0 == VirtualAddress)
return STATUS_INSUFFICIENT_RESOURCES;
NTSTATUS Result;
ULONG ByteCount = MmGetMdlByteCount(UserMdl); ULONG ByteCount = MmGetMdlByteCount(UserMdl);
ULONG PageCount = ADDRESS_AND_SIZE_TO_SPAN_PAGES(VirtualAddress, ByteCount); ULONG PageCount = ADDRESS_AND_SIZE_TO_SPAN_PAGES(VirtualAddress, ByteCount);
FSP_SAFE_MDL *SafeMdl; FSP_SAFE_MDL *SafeMdl;
@ -865,8 +870,6 @@ NTSTATUS FspSafeMdlCreate(PMDL UserMdl, LOCK_OPERATION Operation, FSP_SAFE_MDL *
ASSERT(0 != PageCount); ASSERT(0 != PageCount);
ASSERT(FlagOn(UserMdl->MdlFlags, MDL_SOURCE_IS_NONPAGED_POOL | MDL_PAGES_LOCKED)); ASSERT(FlagOn(UserMdl->MdlFlags, MDL_SOURCE_IS_NONPAGED_POOL | MDL_PAGES_LOCKED));
*PSafeMdl = 0;
SafeMdl = FspAllocNonPaged(sizeof *SafeMdl); SafeMdl = FspAllocNonPaged(sizeof *SafeMdl);
if (0 == SafeMdl) if (0 == SafeMdl)
{ {
@ -996,6 +999,9 @@ VOID FspSafeMdlCopyBack(FSP_SAFE_MDL *SafeMdl)
return; return;
PVOID VirtualAddress = MmGetSystemAddressForMdlSafe(SafeMdl->UserMdl, NormalPagePriority); PVOID VirtualAddress = MmGetSystemAddressForMdlSafe(SafeMdl->UserMdl, NormalPagePriority);
if (0 == VirtualAddress)
return; /* should never happen, already checked in FspSafeMdlCreate */
ULONG ByteCount = MmGetMdlByteCount(SafeMdl->UserMdl); ULONG ByteCount = MmGetMdlByteCount(SafeMdl->UserMdl);
ULONG PageCount = ADDRESS_AND_SIZE_TO_SPAN_PAGES(VirtualAddress, ByteCount); ULONG PageCount = ADDRESS_AND_SIZE_TO_SPAN_PAGES(VirtualAddress, ByteCount);
ULONG ByteOffsetBgn0, ByteOffsetEnd0, ByteOffsetEnd1; ULONG ByteOffsetBgn0, ByteOffsetEnd0, ByteOffsetEnd1;