Namecoin / AuxPoW: Improve exception safety of auxpow.fast_tx_deseria…

…lize.
namecoin · Jul 4, 2018 · fbfaef2 · fbfaef2
1 parent fb60f94
commit fbfaef2
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/lib/auxpow.py b/lib/auxpow.py
@@ -318,12 +318,12 @@ def stub_parse_output(vds, i):
 def fast_tx_deserialize(tx):
     # Monkeypatch output address parsing with a stub, since we only care about
     # inputs.
-    real_parse_output = transaction.parse_output
-    transaction.parse_output = stub_parse_output
+    real_parse_output, transaction.parse_output = transaction.parse_output, stub_parse_output
 
-    result = tx.deserialize()
-
-    # Restore the real output address parser.
-    transaction.parse_output = real_parse_output
+    try:
+        result = tx.deserialize()
+    finally:
+        # Restore the real output address parser.
+        transaction.parse_output = real_parse_output
 
     return result