SwitchLayout.c File Reference

Functions for switching memory layouts. More...

#include "pch.h"

Functions
_Use_decl_annotations_ CR3_TYPE	SwitchToProcessMemoryLayout (UINT32 ProcessId)
	Switch to another process's cr3.
CR3_TYPE	SwitchToCurrentProcessMemoryLayout ()
	Switch to guest's running process's cr3.
_Use_decl_annotations_ CR3_TYPE	SwitchToProcessMemoryLayoutByCr3 (CR3_TYPE TargetCr3)
	Switch to another process's cr3.
_Use_decl_annotations_ VOID	SwitchToPreviousProcess (CR3_TYPE PreviousProcess)
	Switch to previous process's cr3.

Detailed Description

Functions for switching memory layouts.

Author

Sina Karvandi (sina@.nosp@m.hype.nosp@m.rdbg..nosp@m.org)

Version

0.2

Date

2023-04-27

Copyright

This project is released under the GNU Public License v3.

Function Documentation

SwitchToCurrentProcessMemoryLayout()

CR3_TYPE SwitchToCurrentProcessMemoryLayout

(

)

Switch to guest's running process's cr3.

this function can be called from vmx-root mode

Returns

CR3_TYPE The cr3 of current process which can be used by SwitchToPreviousProcess function

{
    CR3_TYPE GuestCr3;
    CR3_TYPE CurrentProcessCr3 = {0};
 
    GuestCr3.Flags = LayoutGetCurrentProcessCr3().Flags;
 
    //
    // Read the current cr3
    //
    CurrentProcessCr3.Flags = CpuReadCr3();
 
    //
    // Change to a new cr3 (of target process)
    //
    CpuWriteCr3(GuestCr3.Flags);
 
    return CurrentProcessCr3;
}

SwitchToPreviousProcess()

_Use_decl_annotations_ VOID SwitchToPreviousProcess

(

CR3_TYPE

PreviousProcess

)

Switch to previous process's cr3.

Parameters

PreviousProcess

Cr3 of previous process which is returned by SwitchToProcessMemoryLayout

Returns

VOID

{
    //
    // Restore the original cr3
    //
    CpuWriteCr3(PreviousProcess.Flags);
}

SwitchToProcessMemoryLayout()

_Use_decl_annotations_ CR3_TYPE SwitchToProcessMemoryLayout

(

UINT32

ProcessId

)

Switch to another process's cr3.

this function should NOT be called from vmx-root mode

Parameters

ProcessId

ProcessId to switch

Returns

CR3_TYPE The cr3 of current process which can be used by SwitchToPreviousProcess function

{
    UINT64    GuestCr3;
    PEPROCESS TargetEprocess;
    CR3_TYPE  CurrentProcessCr3 = {0};
 
    if (PsLookupProcessByProcessId((HANDLE)ProcessId, &TargetEprocess) != STATUS_SUCCESS)
    {
        //
        // There was an error, probably the process id was not found
        //
        return CurrentProcessCr3;
    }
 
    //
    // Due to KVA Shadowing, we need to switch to a different directory table base
    // if the PCID indicates this is a user mode directory table base.
    //
    NT_KPROCESS * CurrentProcess = (NT_KPROCESS *)(TargetEprocess);
    GuestCr3                     = CurrentProcess->DirectoryTableBase;
 
    //
    // Read the current cr3
    //
    CurrentProcessCr3.Flags = CpuReadCr3();
 
    //
    // Change to a new cr3 (of target process)
    //
    CpuWriteCr3(GuestCr3);
 
    ObDereferenceObject(TargetEprocess);
 
    return CurrentProcessCr3;
}

SwitchToProcessMemoryLayoutByCr3()

_Use_decl_annotations_ CR3_TYPE SwitchToProcessMemoryLayoutByCr3

(

CR3_TYPE

TargetCr3

)

Switch to another process's cr3.

Parameters

TargetCr3

cr3 to switch

Returns

CR3_TYPE The cr3 of current process which can be used by SwitchToPreviousProcess function

{
    CR3_TYPE CurrentProcessCr3 = {0};
 
    //
    // Read the current cr3
    //
    CurrentProcessCr3.Flags = CpuReadCr3();
 
    //
    // Change to a new cr3 (of target process)
    //
    CpuWriteCr3(TargetCr3.Flags);
 
    return CurrentProcessCr3;
}