Re-try AuxKlibQueryModuleInformation() if buffer was too small

Signed-off-by: Paul Durrant <paul.durrant@citrix.com>

diff --git a/src/xen/module.c b/src/xen/module.c
index 2a708257f806a14119402f1e5785e50e332a37ea..775ea5f094320de8986c1b1d30ead67511b98205 100644 (file)
--- a/src/xen/module.c
+++ b/src/xen/module.c
@@ -467,6 +467,7 @@ ModuleInitialize(
     if (BufferSize == 0)
         goto fail3;
 
+again:
     Count = BufferSize / sizeof (AUX_MODULE_EXTENDED_INFO);
     QueryInfo = __ModuleAllocate(sizeof (AUX_MODULE_EXTENDED_INFO) * Count);
 
@@ -477,8 +478,13 @@ ModuleInitialize(
     status = AuxKlibQueryModuleInformation(&BufferSize,
                                            sizeof (AUX_MODULE_EXTENDED_INFO),
                                            QueryInfo);
-    if (!NT_SUCCESS(status))
-        goto fail5;
+    if (!NT_SUCCESS(status)) {
+        if (status != STATUS_BUFFER_TOO_SMALL)
+            goto fail5;
+
+        __ModuleFree(QueryInfo);
+        goto again;
+    }
 
     InitializeListHead(&Context->List);
     Context->Cursor = &Context->List;