标准MDL方法修改Page、NonPage内存的属性

时间：2016-05-24 18:48:40 阅读：365 评论：0 收藏：0 [点我收藏+]

typedef struct _REPROTECT_CONTEXT
{
PMDL Mdl;
PUCHAR LockedVa;
} REPROTECT_CONTEXT, * PREPROTECT_CONTEXT;
NTSTATUS
MmLockVaForWrite(
__in PVOID Va,
__in ULONG Length,
__out PREPROTECT_CONTEXT ReprotectContext
)
{
NTSTATUS Status;
Status = STATUS_SUCCESS;
ReprotectContext->Mdl = 0;
ReprotectContext->LockedVa = 0;
ReprotectContext->Mdl = IoAllocateMdl(
Va,
Length,
FALSE,
FALSE,
0
);
if (!ReprotectContext->Mdl)
{
return STATUS_INSUFFICIENT_RESOURCES;
}
//
// Retrieve a locked VA mapping.
//
__try
{
MmProbeAndLockPages(
ReprotectContext->Mdl,
KernelMode,
IoModifyAccess
);
}
__except (EXCEPTION_EXECUTE_HANDLER)
{
return GetExceptionCode();
}
ReprotectContext->LockedVa = (PUCHAR)MmMapLockedPagesSpecifyCache(
ReprotectContext->Mdl,
KernelMode,
MmCached,
0,
FALSE,
NormalPagePriority
);
if (!ReprotectContext->LockedVa)
{
IoFreeMdl(
ReprotectContext->Mdl
);
ReprotectContext->Mdl = 0;
return STATUS_ACCESS_VIOLATION;
}
//
// Reprotect.
//
Status = MmProtectMdlSystemAddress(
ReprotectContext->Mdl,
PAGE_EXECUTE_READWRITE
);
if (!NT_SUCCESS(Status))
{
MmUnmapLockedPages(
ReprotectContext->LockedVa,
ReprotectContext->Mdl
);
MmUnlockPages(
ReprotectContext->Mdl
);
IoFreeMdl(
ReprotectContext->Mdl
);
ReprotectContext->LockedVa = 0;
ReprotectContext->Mdl = 0;
}
return Status;
}
NTSTATUS
MmUnlockVaForWrite(
__in PREPROTECT_CONTEXT ReprotectContext
)
{
if (ReprotectContext->LockedVa)
{
MmUnmapLockedPages(
ReprotectContext->LockedVa,
ReprotectContext->Mdl
);
MmUnlockPages(
ReprotectContext->Mdl
);
IoFreeMdl(
ReprotectContext->Mdl
);
ReprotectContext->LockedVa = 0;
ReprotectContext->Mdl = 0;
}
return STATUS_SUCCESS;
}

原文：http://www.cnblogs.com/kuangke/p/5524443.html

踩

(0)

评论一句话评论（0）

分享档案

更多>

2021年09月23日 (328)
2021年09月24日 (313)
2021年09月17日 (191)
2021年09月15日 (369)
2021年09月16日 (411)
2021年09月13日 (439)
2021年09月11日 (398)
2021年09月12日 (393)
2021年09月10日 (160)
2021年09月08日 (222)