Check for zero byte allocations

Avoid attempting to allocate zero byte buffers, which can lead to
inefficiencies in pool memory usage.

Suggested-by: Matthew Sykes <matthew.sykes@citrix.com>
Signed-off-by: Owen Smith <owen.smith@cloud.com>

diff --git a/src/xenvkbd/util.h b/src/xenvkbd/util.h
index f3937095695b45d574e32b9e67cbd2a18752a651..495162e3ea3550ea8a886fd4e694a72f1d2a653d 100644 (file)
--- a/src/xenvkbd/util.h
+++ b/src/xenvkbd/util.h
@@ -153,6 +153,9 @@ __AllocatePoolWithTag(
     __analysis_assume(PoolType == NonPagedPool ||
                       PoolType == PagedPool);
 
+    if (NumberOfBytes == 0)
+        return NULL;
+
 #if (_MSC_VER >= 1928) // VS 16.9 (EWDK 20344 or later)
     Buffer = ExAllocatePoolUninitialized(PoolType, NumberOfBytes, Tag);
 #else