如何判斷虛擬memory頁面是否被鎖定？

Question

比如說，如果在某個時候我進程中的虛擬 memory 范圍被這樣鎖定：

//Memory was reserved & committed as such
void* pMem = ::VirtualAlloc(NULL, 4096, MEM_COMMIT | MEM_RESERVE, PAGE_READWRITE);

//...

//And then
::VirtualLock(pMem, 4096);

所以在我的進程中有一個虛擬 memory 中的任意頁面地址，我可以判斷它是否被locked了嗎？

Answer 1

通過使用win32 api這是不可能的。 但如果將ZwQueryVirtualMemory與未記錄的 MEMORY_INFORMATION_CLASS值一起使用 - 這是可能的。 對於數據結構定義 - 請參閱ntmmapi.h

我們需要在MemoryWorkingSetExInformation使用MEMORY_WORKING_SET_EX_BLOCK並使用ULONG_PTR Locked : 1; 會員

演示測試：

NTSTATUS IsPageLocked(PVOID BaseAddress, BOOLEAN& Locked)
{
    MEMORY_WORKING_SET_EX_INFORMATION mwsei = { BaseAddress };

    NTSTATUS status = ZwQueryVirtualMemory(NtCurrentProcess(), 0, MemoryWorkingSetExInformation, &mwsei, sizeof(mwsei), 0);

    if (0 <= status)
    {
        if (mwsei.VirtualAttributes.Valid)
        {
            Locked = mwsei.VirtualAttributes.Locked;
        }
        else
        {
            status = STATUS_INVALID_ADDRESS;
        }
    }

    return status;
}

NTSTATUS IsPageLockedEx(PVOID BaseAddress, BOOLEAN& Locked)
{
    NTSTATUS status = IsPageLocked(BaseAddress, Locked);
    if (0 > status)
    {
        DbgPrint("IsPageLocked - error %x\n", status);
    }
    else
    {
        DbgPrint("IsPageLocked = %x\n", Locked);
    }
    return status;
}

void CheckVA()
{
    if (PVOID pv = VirtualAlloc(NULL, PAGE_SIZE, MEM_COMMIT | MEM_RESERVE, PAGE_READWRITE))
    {
        BOOLEAN Locked;
        NTSTATUS status = IsPageLockedEx(pv, Locked);

        if (status == STATUS_INVALID_ADDRESS)
        {
            // at this point  physical memory for pv not yet committed. reference memory for commit physical page
            if (*(PBYTE)pv) __nop();

            status = IsPageLockedEx(pv, Locked);
        }

        if (VirtualLock(pv, PAGE_SIZE))
        {
            IsPageLockedEx(pv, Locked);

            if (VirtualUnlock(pv, PAGE_SIZE))
            {
                IsPageLockedEx(pv, Locked);
            }
            else
            {
                __debugbreak();
            }
        }

        VirtualFree(pv, 0, MEM_RELEASE);
    }
}

和DbgPrint輸出

IsPageLocked - error c0000141
IsPageLocked = 0
IsPageLocked = 1
IsPageLocked = 0

Answer 2

對於將來關注此問題的任何人，XP 和更高版本現在包括執行此操作的 API： QueryWorkingSet和QueryWorkingSetEx 。

在PSAPI_WORKING_SET_EX_BLOCK結構中尋找Locked位，類似於依賴 ZwQueryVirtualMemory 的答案。