[tfaber] 72964: [KMTESTS:IO] - Add a test for the maximum data length for MDLs CORE-12132

Author: tfaber
Date: Thu Oct 13 12:28:01 2016
New Revision: 72964

URL: http://svn.reactos.org/svn/reactos?rev=72964&view=rev
Log:
[KMTESTS:IO]
- Add a test for the maximum data length for MDLs
CORE-12132

Modified:
    trunk/rostests/kmtests/ntos_io/IoMdl.c

Modified: trunk/rostests/kmtests/ntos_io/IoMdl.c
URL: http://svn.reactos.org/svn/reactos/trunk/rostests/kmtests/ntos_io/IoMdl.c?re...
==============================================================================
--- trunk/rostests/kmtests/ntos_io/IoMdl.c	[iso-8859-1] (original)
+++ trunk/rostests/kmtests/ntos_io/IoMdl.c	[iso-8859-1] Thu Oct 13 12:28:01 2016
@@ -16,12 +16,13 @@
     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",
+      "IoAllocateMdl should fail allocation of 2Gb or more, but got Mdl=0x%X\n",
       (UINT_PTR)Mdl);

if (Mdl)
@@ -30,22 +31,40 @@
     // Now create a valid MDL
     VirtualAddress = ExAllocatePool(NonPagedPool, MdlSize);
     Mdl = IoAllocateMdl(VirtualAddress, MdlSize, FALSE, FALSE, NULL);
-    ok(Mdl != NULL, "Mdl allocation failed");
+    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",
+    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",
+        "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");
+    ok(Irp != NULL, "IRP allocation failed\n");
     Mdl = IoAllocateMdl(VirtualAddress, MdlSize, FALSE, FALSE, Irp);
-    ok(Mdl != NULL, "Mdl allocation failed");
-    ok(Irp->MdlAddress == Mdl, "Irp->MdlAddress should be 0x%X, but is 0x%X",
+    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);