From: Farhat-Abbas <[email protected]>

Signed-off-by: Farhat-Abbas <[email protected]>
---
 scripts/lib/wic/plugins/source/efibootguard-boot.py | 4 ++--
 scripts/lib/wic/plugins/source/efibootguard-efi.py  | 3 ++-
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/scripts/lib/wic/plugins/source/efibootguard-boot.py 
b/scripts/lib/wic/plugins/source/efibootguard-boot.py
index f4326ae..6b1760f 100644
--- a/scripts/lib/wic/plugins/source/efibootguard-boot.py
+++ b/scripts/lib/wic/plugins/source/efibootguard-boot.py
@@ -30,6 +30,7 @@ import os
 import shutil
 import logging
 
+from wic import WicError
 from wic.pluginbase import SourcePlugin
 from wic.engine import get_custom_config
 from wic.misc import exec_cmd, exec_native_cmd, get_bitbake_var, \
@@ -85,8 +86,7 @@ class EfibootguardBootPlugin(SourcePlugin):
         if not kernel_dir:
             kernel_dir = get_bitbake_var("DEPLOY_DIR_IMAGE")
             if not kernel_dir:
-                msger.error("DEPLOY_DIR_IMAGE not set, exiting\n")
-                os.exit(1)
+                raise WicError("DEPLOY_DIR_IMAGE not set, exiting\n")
 
         staging_kernel_dir = kernel_dir
 
diff --git a/scripts/lib/wic/plugins/source/efibootguard-efi.py 
b/scripts/lib/wic/plugins/source/efibootguard-efi.py
index 62e283a..7a3aef2 100644
--- a/scripts/lib/wic/plugins/source/efibootguard-efi.py
+++ b/scripts/lib/wic/plugins/source/efibootguard-efi.py
@@ -31,6 +31,7 @@ import os.path
 import shutil
 import logging
 
+from wic import WicError
 from wic.pluginbase import SourcePlugin
 from wic.engine import get_custom_config
 from wic.misc import exec_cmd, exec_native_cmd, get_bitbake_var, \
@@ -58,7 +59,7 @@ class EfibootguardEFIPlugin(SourcePlugin):
         if not kernel_dir:
             kernel_dir = get_bitbake_var("DEPLOY_DIR_IMAGE")
             if not kernel_dir:
-                msger.error("DEPLOY_DIR_IMAGE not set, exiting\n")
+                raise WicError("DEPLOY_DIR_IMAGE not set, exiting\n")
 
         staging_kernel_dir = kernel_dir
 
-- 
2.7.4

-- 
You received this message because you are subscribed to the Google Groups "EFI 
Boot Guard" group.
To unsubscribe from this group and stop receiving emails from it, send an email 
to [email protected].
To post to this group, send email to [email protected].
To view this discussion on the web visit 
https://groups.google.com/d/msgid/efibootguard-dev/1519216052-8012-3-git-send-email-Farhat_Abbas%40mentor.com.
For more options, visit https://groups.google.com/d/optout.

Reply via email to