2014-05-30 08:55:32 +00:00
|
|
|
#include "KeMemory32.h"
|
|
|
|
#include "KeDef.h"
|
2014-06-30 01:53:16 +00:00
|
|
|
#include "KeKernelStruct32.h"
|
2014-05-30 08:55:32 +00:00
|
|
|
#include "KeGraph32.h"
|
|
|
|
|
2014-06-03 07:05:45 +00:00
|
|
|
ULONG32 HYPKERNELAPI SegmentToPhyscicalAddress(ULONG16 Segment)
|
2014-05-30 08:55:32 +00:00
|
|
|
{
|
2014-06-30 01:53:16 +00:00
|
|
|
PIAGDT32 PDescriptor = &KeGDT[(Segment >> 3)];
|
|
|
|
return (PDescriptor->BaseHigh << 24 | PDescriptor->BaseMid << 16 | PDescriptor->BaseLow);
|
2014-05-30 08:55:32 +00:00
|
|
|
};
|
|
|
|
|
2014-06-03 07:05:45 +00:00
|
|
|
PVOID HYPKERNELAPI KeMemorySet(PVOID Source, BYTE Value, ULONG32 Length)
|
2014-05-30 08:55:32 +00:00
|
|
|
{
|
|
|
|
PBYTE src = (PBYTE)Source;
|
2014-06-30 01:53:16 +00:00
|
|
|
while (Length > 0)
|
2014-05-30 08:55:32 +00:00
|
|
|
{
|
|
|
|
_asm_KePrintHex32(Length);
|
|
|
|
_asm_KePrintStr32("\n");
|
|
|
|
*src = Value;
|
|
|
|
src++;
|
|
|
|
Length--;
|
|
|
|
};
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
2014-06-03 07:05:45 +00:00
|
|
|
PVOID HYPKERNELAPI KeMemoryCopy(PVOID src, PVOID dst, ULONG32 size)
|
2014-05-30 08:55:32 +00:00
|
|
|
{
|
|
|
|
ULONG32 i = 0;
|
2014-06-30 01:53:16 +00:00
|
|
|
for (i = 0; i < size; i++)
|
2014-05-30 08:55:32 +00:00
|
|
|
{
|
2014-06-30 01:53:16 +00:00
|
|
|
*((PCHAR)dst + i) = *((PCHAR)src + i);
|
2014-05-30 08:55:32 +00:00
|
|
|
}
|
|
|
|
return dst;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2014-06-30 01:53:16 +00:00
|
|
|
VOID HYPKERNELAPI KeInitializeWritePDE4KDescriptor32(PKePDE4KDescriptor32 Desc, ULONG32 P, ULONG32 RW, ULONG32 US, ULONG32 PWT, ULONG32 PCD, ULONG32 A, ULONG32 PS, ULONG32 Address)
|
2014-05-30 08:55:32 +00:00
|
|
|
{
|
|
|
|
Desc->P = P;
|
|
|
|
Desc->Address = Address;
|
|
|
|
Desc->A = A;
|
|
|
|
Desc->PS = PS;
|
|
|
|
Desc->US = US;
|
|
|
|
Desc->RW = RW;
|
|
|
|
Desc->PWT = PWT;
|
|
|
|
Desc->PCD = PCD;
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2014-06-03 07:05:45 +00:00
|
|
|
VOID HYPKERNELAPI KeWritePDE4K32(PKePDE4KDescriptor32 Desc, PVOID Dst)
|
2014-05-30 08:55:32 +00:00
|
|
|
{
|
|
|
|
ULONG32 temp = 0;
|
|
|
|
temp = Desc->Address;
|
|
|
|
temp = (temp & 0xfffff000);
|
|
|
|
temp = temp + Desc->P;
|
|
|
|
temp = temp + Desc->RW;
|
|
|
|
temp = temp + Desc->US;
|
|
|
|
temp = temp + Desc->PWT;
|
|
|
|
temp = temp + Desc->PCD;
|
|
|
|
temp = temp + Desc->A;
|
|
|
|
temp = temp + Desc->PS;
|
|
|
|
*(PULONG32)Dst = temp;
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2014-06-30 01:53:16 +00:00
|
|
|
VOID HYPKERNELAPI KeInitializeWritePTE4K32Descriptor(PKePTE4KDescriptor32 Desc, ULONG32 P, ULONG32 RW, ULONG32 US, ULONG32 PWT, ULONG32 PCD, ULONG32 A, ULONG32 D, ULONG32 PS, ULONG32 G, ULONG32 Address)
|
2014-05-30 08:55:32 +00:00
|
|
|
{
|
|
|
|
Desc->P = P;
|
|
|
|
Desc->Address = Address;
|
|
|
|
Desc->A = A;
|
|
|
|
Desc->US = US;
|
|
|
|
Desc->RW = RW;
|
|
|
|
Desc->PWT = PWT;
|
|
|
|
Desc->PCD = PCD;
|
|
|
|
Desc->PS = PS;
|
|
|
|
Desc->G = G;
|
|
|
|
Desc->D = D;
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2014-06-30 01:53:16 +00:00
|
|
|
VOID HYPKERNELAPI KeGetPDEInfo(PKePDE4KDescriptor32 pDesc, PVOID PDEPtr)
|
2014-05-30 08:55:32 +00:00
|
|
|
{
|
2014-06-30 01:53:16 +00:00
|
|
|
ULONG32 temp = *(PULONG32)PDEPtr;
|
2014-05-30 08:55:32 +00:00
|
|
|
pDesc->Address = (temp >> 12) << 12;
|
|
|
|
pDesc->A = temp & KeWritePDE4K32_A_1;
|
|
|
|
pDesc->P = temp & KeWritePDE4K32_P_1;
|
|
|
|
pDesc->PCD = temp & KeWritePDE4K32_PCD_1;
|
|
|
|
pDesc->PS = temp & KeWritePDE4K32_PS_1;
|
|
|
|
pDesc->US = temp & KeWritePDE4K32_US_1;
|
|
|
|
pDesc->RW = temp & KeWritePDE4K32_RW_1;
|
|
|
|
pDesc->PWT = temp & KeWritePDE4K32_PWT_1;
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2014-06-30 01:53:16 +00:00
|
|
|
VOID HYPKERNELAPI KeGetPTEInfo(PKePTE4KDescriptor32 pDesc, PVOID PTEPtr)
|
2014-05-30 08:55:32 +00:00
|
|
|
{
|
2014-06-30 01:53:16 +00:00
|
|
|
ULONG32 temp = *(PULONG32)PTEPtr;
|
|
|
|
pDesc->Address = (temp >> 12) << 12;
|
2014-05-30 08:55:32 +00:00
|
|
|
pDesc->P = temp & KeWritePTE4K32_P_1;
|
|
|
|
pDesc->D = temp & KeWritePTE4K32_D_1;
|
|
|
|
pDesc->US = temp & KeWritePTE4K32_US_1;
|
|
|
|
pDesc->PCD = temp & KeWritePTE4K32_PCD_1;
|
|
|
|
pDesc->PWT = temp & KeWritePTE4K32_PWT_1;
|
|
|
|
pDesc->PS = temp & KeWritePTE4K32_PS_1;
|
|
|
|
pDesc->RW = temp & KeWritePTE4K32_RW_1;
|
|
|
|
pDesc->G = temp & KeWritePTE4K32_G_1;
|
|
|
|
return;
|
|
|
|
}
|
|
|
|
|
2014-06-30 01:53:16 +00:00
|
|
|
VOID HYPKERNELAPI KeWritePTE4K32(PKePTE4KDescriptor32 Desc, PVOID Dst)
|
2014-05-30 08:55:32 +00:00
|
|
|
{
|
|
|
|
ULONG32 temp = 0;
|
|
|
|
temp = Desc->Address;
|
|
|
|
temp = (temp & 0xfffff000);
|
|
|
|
temp += Desc->P;
|
|
|
|
temp += Desc->RW;
|
|
|
|
temp += Desc->US;
|
|
|
|
temp += Desc->PWT;
|
|
|
|
temp += Desc->PCD;
|
|
|
|
temp += Desc->A;
|
|
|
|
temp += Desc->D;
|
|
|
|
temp += Desc->PS;
|
|
|
|
temp += Desc->G;
|
|
|
|
*(PULONG32)Dst = temp;
|
|
|
|
}
|
|
|
|
|
2014-06-30 01:53:16 +00:00
|
|
|
ULONG32 HYPKERNELAPI KeGetPDEIndex4K32(ULONG32 VirtualAddress)
|
2014-05-30 08:55:32 +00:00
|
|
|
{
|
2014-06-30 01:53:16 +00:00
|
|
|
return VirtualAddress >> 22;
|
2014-05-30 08:55:32 +00:00
|
|
|
}
|
|
|
|
|
2014-06-30 01:53:16 +00:00
|
|
|
ULONG32 HYPKERNELAPI KeGetPTEIndex4K32(ULONG32 VirtualAddress)
|
2014-05-30 08:55:32 +00:00
|
|
|
{
|
2014-06-30 01:53:16 +00:00
|
|
|
return (VirtualAddress << 10) >> 22;
|
2014-05-30 08:55:32 +00:00
|
|
|
}
|
|
|
|
|
2014-06-03 07:05:45 +00:00
|
|
|
|
|
|
|
|
2014-06-28 01:35:23 +00:00
|
|
|
ULONG32 HYPKERNELAPI MmMapVirtualAddress4K32(PVOID PDEPtr, ULONG32 PhysicalAddress, ULONG32 VirtualAddress, PKePTE4KDescriptor32 PTEDesc)
|
2014-05-30 08:55:32 +00:00
|
|
|
{
|
2014-06-30 01:53:16 +00:00
|
|
|
//Physical Address and Virtual Address will be trimmed to be 4K-aligned
|
|
|
|
KePDE4KDescriptor32 PDE;
|
|
|
|
KeGetPDEInfo(&PDE, PDEPtr);
|
|
|
|
if (PDE.P == KeWritePDE4K32_P_0)
|
|
|
|
{
|
|
|
|
//create PDE
|
|
|
|
}
|
2014-06-03 07:05:45 +00:00
|
|
|
ULONG32 PDEIndex = KeGetPDEIndex4K32(VirtualAddress);
|
2014-06-30 01:53:16 +00:00
|
|
|
ULONG32 PTEIndex = KeGetPTEIndex4K32(VirtualAddress);
|
2014-06-28 01:35:23 +00:00
|
|
|
ULONG32 PTEBase = (*(PULONG32)((ULONG32)PDEPtr + PDEIndex * 4) >> 12) << 12;
|
2014-05-30 08:55:32 +00:00
|
|
|
PVOID Target = (PVOID)((*(PULONG32)(PTEBase + PTEIndex * 4) >> 12) << 12);
|
2014-06-30 01:53:16 +00:00
|
|
|
KePTE4KDescriptor32 PTE;
|
|
|
|
KeGetPTEInfo(&PTE, Target);
|
|
|
|
if (PTE.P == KeWritePTE4K32_P_0)
|
|
|
|
{
|
|
|
|
//create PTE
|
|
|
|
}
|
|
|
|
PTEDesc->Address = VirtualAddress;
|
|
|
|
KeWritePTE4K32(&PTEDesc, Target);
|
|
|
|
return 0;
|
|
|
|
}
|
2014-06-28 01:35:23 +00:00
|
|
|
|
|
|
|
|
2014-06-30 01:53:16 +00:00
|
|
|
ULONG32 HYPKERNELAPI KeGetBit(PVOID Source, ULONG32 Position)
|
|
|
|
{
|
|
|
|
if (!Source)
|
|
|
|
return 0;
|
|
|
|
ULONG32 Quotient = Position / 8;
|
|
|
|
ULONG32 Remainder = Position % 8;
|
|
|
|
PULONG8 CurrentPos = (PULONG8)Source;
|
|
|
|
ULONG8 Value;
|
|
|
|
while (Quotient > 0)
|
|
|
|
{
|
|
|
|
CurrentPos++;
|
|
|
|
Quotient--;
|
|
|
|
}
|
|
|
|
Value = *CurrentPos;
|
|
|
|
Value = Value << (7 - Remainder);
|
|
|
|
Value = Value >> 7;
|
|
|
|
return Value;
|
2014-06-28 01:35:23 +00:00
|
|
|
}
|
|
|
|
|
2014-06-30 01:53:16 +00:00
|
|
|
ULONG32 HYPKERNELAPI KeSetBit(PVOID Source, ULONG32 Position, ULONG32 Value)
|
2014-06-28 01:35:23 +00:00
|
|
|
{
|
2014-06-30 01:53:16 +00:00
|
|
|
if (!Source || Position < 0 || (Value != 0 && Value != 1))
|
2014-06-28 01:35:23 +00:00
|
|
|
return 1;
|
2014-06-30 01:53:16 +00:00
|
|
|
ULONG32 Quotient = Position / 8;
|
|
|
|
ULONG32 Remainder = Position % 8;
|
|
|
|
PULONG8 CurrentPos = (PULONG8)Source;
|
|
|
|
while (Quotient > 0)
|
|
|
|
{
|
|
|
|
CurrentPos++;
|
|
|
|
Quotient--;
|
|
|
|
}
|
|
|
|
if (Value == 1)
|
|
|
|
{
|
|
|
|
ULONG8 temp = 1;
|
|
|
|
temp = temp << Remainder;
|
|
|
|
*CurrentPos = *CurrentPos | temp;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
ULONG8 temp = 0xfe;
|
|
|
|
while (Remainder > 0)
|
|
|
|
{
|
|
|
|
temp = temp << 1;
|
|
|
|
temp++;
|
|
|
|
Remainder--;
|
|
|
|
}
|
|
|
|
*CurrentPos = *CurrentPos & temp;
|
|
|
|
}
|
|
|
|
return 0;
|
2014-05-30 08:55:32 +00:00
|
|
|
}
|