X-Git-Url: https://code.delx.au/refind/blobdiff_plain/fe53e20674a0f3dff46aa42d5676a5304dac92bf..480ba418c97ece5557ac0efc5dc189ff19fb8b8f:/refind/driver_support.c

diff --git a/refind/driver_support.c b/refind/driver_support.c
index 4780dae..a547d65 100644
--- a/refind/driver_support.c
+++ b/refind/driver_support.c
@@ -19,6 +19,7 @@
  */
 
 #include "driver_support.h"
+#include "lib.h"
 #include "../include/refit_call_wrapper.h"
 
 #ifdef __MAKEWITH_GNUEFI
@@ -228,24 +229,19 @@ LibScanHandleDatabase (
             }
           }
 
-          FreePool (OpenInfo);
+          MyFreePool (OpenInfo);
         }
       }
 
-      FreePool (ProtocolGuidArray);
+      MyFreePool (ProtocolGuidArray);
     }
   }
 
   return EFI_SUCCESS;
 
 Error:
-  if (*HandleType != NULL) {
-    FreePool (*HandleType);
-  }
-
-  if (*HandleBuffer != NULL) {
-    FreePool (*HandleBuffer);
-  }
+  MyFreePool (*HandleType);
+  MyFreePool (*HandleBuffer);
 
   *HandleCount  = 0;
   *HandleBuffer = NULL;