[PATCH v2 1/1] winspool: Check dmSize in IsValidDevmodeW

[Tingzhong Luo]

From: Tingzhong Luo

---
 dlls/winspool.drv/info.c       | 1 +
 dlls/winspool.drv/tests/info.c | 7 +++++++
 2 files changed, 8 insertions(+)

diff --git a/dlls/winspool.drv/info.c b/dlls/winspool.drv/info.c
index 54ecb81d3be..26e662d7ffb 100644
--- a/dlls/winspool.drv/info.c
+++ b/dlls/winspool.drv/info.c
@@ -1929,6 +1929,7 @@ BOOL WINAPI IsValidDevmodeW(PDEVMODEW dm, SIZE_T size)
     int i;

if (!dm) return FALSE;
+    if (!dm->dmSize || size < dm->dmSize) return FALSE;
     if (size < FIELD_OFFSET(DEVMODEW, dmFields) + sizeof(dm->dmFields)) return FALSE;

for (i = 0; i < ARRAY_SIZE(map); i++)
diff --git a/dlls/winspool.drv/tests/info.c b/dlls/winspool.drv/tests/info.c
index 92b042e354a..6fb279dca1e 100644
--- a/dlls/winspool.drv/tests/info.c
+++ b/dlls/winspool.drv/tests/info.c
@@ -3065,6 +3065,13 @@ static void test_IsValidDevmodeW(void)
         ret = IsValidDevmodeW(&dm, dm.dmSize);
         ok(ret == test[i].ret, "%d: got %d\n", i, ret);
     }
+
+    dm.dmSize = 0;
+    ret = IsValidDevmodeW(&dm, FIELD_OFFSET(DEVMODEW, u2.dmNup) + 4);
+    ok(!ret, "%d: got %d\n", i, ret); /* failed if dm.dmSize is zero */
+    dm.dmSize = FIELD_OFFSET(DEVMODEW, u2.dmNup) + 6;
+    ret = IsValidDevmodeW(&dm, FIELD_OFFSET(DEVMODEW, u2.dmNup) + 4);
+    ok(!ret, "%d: got %d\n", i, ret); /* failed if dm.dmSize greater than bufSize */
 }

START_TEST(info)