From 30853e5c25e0dfe27c6fda6e4fc53748a671de6e Mon Sep 17 00:00:00 2001
From: Changhua <lichanghua0821@gmail.com>
Date: Sat, 13 Aug 2022 21:57:23 +0800
Subject: [PATCH] Catch memory allocate failure

---
 Spy/rpc_server.cpp | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/Spy/rpc_server.cpp b/Spy/rpc_server.cpp
index f92f7c0..78bc09b 100644
--- a/Spy/rpc_server.cpp
+++ b/Spy/rpc_server.cpp
@@ -156,7 +156,12 @@ int server_GetDbTables(const wchar_t *db, int *pNum, PPRpcTables *tbls)
 
     int index = 0;
     for (auto it = tables.begin(); it != tables.end(); it++) {
-        PRpcTables p = (PRpcTables)midl_user_allocate(sizeof(RpcTables_t));
+        PRpcTables p = (PRpcTables)midl_user_allocate(sizeof(RpcTables_t));
+        if (p == NULL) {
+            printf("server_GetDbTables midl_user_allocate Failed for p\n");
+            return -3;
+        }
+
         p->table    = it->table;
         p->sql      = it->sql;
         pp[index++] = p;