IoMdl.c File Reference

#include <kmt_test.h>
#include <debug.h>

Include dependency graph for IoMdl.c:

Go to the source code of this file.

Macros
#define	NDEBUG

Functions
	START_TEST (IoMdl)

Macro Definition Documentation

NDEBUG

#define NDEBUG

Definition at line 10 of file IoMdl.c.

Function Documentation

START_TEST()

START_TEST

(

IoMdl

)

Definition at line 13 of file IoMdl.c.

{
    PMDL Mdl;
    PIRP Irp;
    PVOID VirtualAddress;
    ULONG MdlSize = 2*4096+184; // 2 pages and some random value
    ULONG TooLargeMdlSize = (0x10000 - sizeof(MDL)) / sizeof(ULONG_PTR) * PAGE_SIZE;
 
    // Try to alloc 2Gb MDL
    Mdl = IoAllocateMdl(NULL, 2048UL*0x100000, FALSE, FALSE, NULL);
 
    ok(Mdl == NULL,
      "IoAllocateMdl should fail allocation of 2Gb or more, but got Mdl=0x%X\n",
      (UINT_PTR)Mdl);
 
    if (Mdl)
        IoFreeMdl(Mdl);
 
    // Now create a valid MDL
    VirtualAddress = ExAllocatePool(NonPagedPool, MdlSize);
    Mdl = IoAllocateMdl(VirtualAddress, MdlSize, FALSE, FALSE, NULL);
    ok(Mdl != NULL, "Mdl allocation failed\n");
    // Check fields of the allocated struct
    ok(Mdl->Next == NULL, "Mdl->Next should be NULL, but is 0x%X\n",
        (UINT_PTR)Mdl->Next);
    ok(Mdl->ByteCount == MdlSize,
        "Mdl->ByteCount should be equal to MdlSize, but is 0x%X\n",
        (UINT_PTR)Mdl->ByteCount);
    // TODO: Check other fields of MDL struct
 
    IoFreeMdl(Mdl);
 
    // Test maximum size for an MDL
    Mdl = IoAllocateMdl(NULL, TooLargeMdlSize, FALSE, FALSE, NULL);
    ok(Mdl == NULL, "Mdl allocation for %lu bytes succeeded\n", TooLargeMdlSize);
    if (Mdl) IoFreeMdl(Mdl);
 
    Mdl = IoAllocateMdl(NULL, TooLargeMdlSize - PAGE_SIZE + 1, FALSE, FALSE, NULL);
    ok(Mdl == NULL, "Mdl allocation for %lu bytes succeeded\n", TooLargeMdlSize - PAGE_SIZE + 1);
    if (Mdl) IoFreeMdl(Mdl);
 
    Mdl = IoAllocateMdl(NULL, TooLargeMdlSize - PAGE_SIZE, FALSE, FALSE, NULL);
    ok(Mdl != NULL, "Mdl allocation for %lu bytes failed\n", TooLargeMdlSize - PAGE_SIZE);
    if (Mdl) IoFreeMdl(Mdl);
 
    Mdl = IoAllocateMdl(NULL, TooLargeMdlSize / 2, FALSE, FALSE, NULL);
    ok(Mdl != NULL, "Mdl allocation for %lu bytes failed\n", TooLargeMdlSize / 2);
    if (Mdl) IoFreeMdl(Mdl);
 
    // Allocate now with pointer to an Irp
    Irp = IoAllocateIrp(1, FALSE);
    ok(Irp != NULL, "IRP allocation failed\n");
    Mdl = IoAllocateMdl(VirtualAddress, MdlSize, FALSE, FALSE, Irp);
    ok(Mdl != NULL, "Mdl allocation failed\n");
    ok(Irp->MdlAddress == Mdl, "Irp->MdlAddress should be 0x%X, but is 0x%X\n",
        (UINT_PTR)Mdl, (UINT_PTR)Irp->MdlAddress);
 
    IoFreeMdl(Mdl);
 
    // TODO: Check a case when SecondaryBuffer == TRUE
 
    IoFreeIrp(Irp);
    ExFreePool(VirtualAddress);
}