SFT-2235: added error handling to showing the signed QR

Foundation-Devices · Jul 20, 2023 · 7ddc843 · 7ddc843
1 parent 5c3d576
commit 7ddc843
Showing 1 changed file with 11 additions and 5 deletions.
diff --git a/ports/stm32/boards/Passport/modules/flows/sign_psbt_qr_flow.py b/ports/stm32/boards/Passport/modules/flows/sign_psbt_qr_flow.py
@@ -137,7 +137,7 @@ async def get_signed_bytes(self):
         self.goto(self.show_signed_transaction)
 
     async def show_signed_transaction(self):
-        from pages import ShowQRPage
+        from pages import ShowQRPage, ErrorPage
         from data_codecs.qr_type import QRType
         from ubinascii import hexlify as b2a_hex
         import microns
@@ -155,10 +155,16 @@ async def show_signed_transaction(self):
             else:
                 raise RuntimeError('Unknown UR type: {}'.format(self.ur_type))
 
-        result = await ShowQRPage(qr_type=qr_type,
-                                  qr_data=qr_data,
-                                  left_micron=microns.MicroSD,
-                                  right_micron=microns.Checkmark).show()
+        result = False
+        try:
+            result = await ShowQRPage(qr_type=qr_type,
+                                      qr_data=qr_data,
+                                      left_micron=microns.MicroSD,
+                                      right_micron=microns.Checkmark).show()
+        except MemoryError as e:
+            await ErrorPage(text='Transaction is too complex: {}'.format(e)).show()
+            self.set_result(False)
+            return
 
         if not result:
             self.goto(self.save_to_microsd)