X-Git-Url: https://code.delx.au/refind/blobdiff_plain/0d5ee58bd6562186ea980fcca99484d8272369bf..e0f6b77e5692ec112bb803202ae27f8c5d55de50:/refind/icns.c

diff --git a/refind/icns.c b/refind/icns.c
index e4738e2..a9b3f01 100644
--- a/refind/icns.c
+++ b/refind/icns.c
@@ -38,6 +38,7 @@
 #include "lib.h"
 #include "icns.h"
 #include "config.h"
+#include "../refind/screen.h"
 
 //
 // well-known icons
@@ -57,6 +58,8 @@ BUILTIN_ICON BuiltinIconTable[BUILTIN_ICON_COUNT] = {
    { NULL, L"tool_shell.icns", 48 },
    { NULL, L"tool_part.icns", 48 },
    { NULL, L"tool_rescue.icns", 48 },
+   { NULL, L"tool_apple_rescue.icns", 48 },
+   { NULL, L"tool_mok_tool.icns", 48 },
    { NULL, L"vol_internal.icns", 32 },
    { NULL, L"vol_external.icns", 32 },
    { NULL, L"vol_optical.icns", 32 },
@@ -100,9 +103,11 @@ EG_IMAGE * LoadOSIcon(IN CHAR16 *OSIconName OPTIONAL, IN CHAR16 *FallbackIconNam
 
         // try to load it
         Image = egLoadIcon(SelfDir, FileName, 128);
-        if (Image != NULL)
+        if (Image != NULL) {
+            MyFreePool(CutoutName);
             return Image;
-        FreePool(CutoutName);
+        }
+        MyFreePool(CutoutName);
     } // while
 
     // try the fallback name