From 46cc9cd5b66487172281787c8b3e14a2cfca7558 Mon Sep 17 00:00:00 2001
From: ThomasV <thomasv@electrum.org>
Date: Wed, 10 Oct 2018 10:29:57 +0200
Subject: [PATCH] follow-up 1c8a4bcfa497b117e4511c2f108dbca8a1adb793

---
 electrum/lnbase.py | 33 +++++++++++----------------------
 1 file changed, 11 insertions(+), 22 deletions(-)

diff --git a/electrum/lnbase.py b/electrum/lnbase.py
index fab3baf4f..c6cad12c8 100644
--- a/electrum/lnbase.py
+++ b/electrum/lnbase.py
@@ -1042,28 +1042,17 @@ class Peer(PrintError):
         data = failure_msg.data
         self.print_error("UPDATE_FAIL_HTLC", repr(code), data)
         # handle some specific error codes
-        if code == OnionFailureCode.TEMPORARY_CHANNEL_FAILURE:
-            channel_update = (258).to_bytes(length=2, byteorder="big") + data[2:]
-            message_type, payload = decode_msg(channel_update)
-            self.on_channel_update(payload)
-        elif code == OnionFailureCode.AMOUNT_BELOW_MINIMUM:
-            channel_update = (258).to_bytes(length=2, byteorder="big") + data[10:]
-            message_type, payload = decode_msg(channel_update)
-            self.on_channel_update(payload)
-        elif code == OnionFailureCode.FEE_INSUFFICIENT:
-            channel_update = (258).to_bytes(length=2, byteorder="big") + data[10:]
-            message_type, payload = decode_msg(channel_update)
-            self.on_channel_update(payload)
-        elif code == OnionFailureCode.INCORRECT_CLTV_EXPIRY:
-            channel_update = (258).to_bytes(length=2, byteorder="big") + data[6:]
-            message_type, payload = decode_msg(channel_update)
-            self.on_channel_update(payload)
-        elif code == OnionFailureCode.EXPIRY_TOO_SOON:
-            channel_update = (258).to_bytes(length=2, byteorder="big") + data[2:]
-            message_type, payload = decode_msg(channel_update)
-            self.on_channel_update(payload)
-        elif code == OnionFailureCode.CHANNEL_DISABLED:
-            channel_update = (258).to_bytes(length=2, byteorder="big") + data[4:]
+        failure_codes = {
+            OnionFailureCode.TEMPORARY_CHANNEL_FAILURE: 2
+            OnionFailureCode.AMOUNT_BELOW_MINIMUM: 10
+            OnionFailureCode.FEE_INSUFFICIENT: 10
+            OnionFailureCode.INCORRECT_CLTV_EXPIRY: 6
+            OnionFailureCode.EXPIRY_TOO_SOON: 2
+            OnionFailureCode.CHANNEL_DISABLED: 4
+        }
+        offset = failure_codes.get(code)
+        if offset:
+            channel_update = (258).to_bytes(length=2, byteorder="big") + data[offset:]
             message_type, payload = decode_msg(channel_update)
             self.on_channel_update(payload)
         else: