BaseTools: Fix BPDG tool print traceback info issue

[mirror_edk2.git] / BaseTools / Source / Python / BPDG / BPDG.py

diff --git a/BaseTools/Source/Python/BPDG/BPDG.py b/BaseTools/Source/Python/BPDG/BPDG.py
index 2ec1516c0a08da6b52ae6d0b4aafd91e0d82811d..c30e062a6953e12ecb827003c2042ee5553234dd 100644 (file)
--- a/BaseTools/Source/Python/BPDG/BPDG.py
+++ b/BaseTools/Source/Python/BPDG/BPDG.py
@@ -153,7 +153,10 @@ def StartBpdg(InputFileName, MapFileName, VpdFileName, Force):
     EdkLogger.info("- Vpd pcd fixed done! -")\r
 \r
 if __name__ == '__main__':\r
-    r = main()\r
+    try:\r
+        r = main()\r
+    except FatalError as e:\r
+        r = e\r
     ## 0-127 is a safe return range, and 1 is a standard default error\r
     if r < 0 or r > 127: r = 1\r
     sys.exit(r)\r