From: Matt Corallo <git@bluematt.me>
Date: Wed, 24 Oct 2018 04:44:15 +0000 (-0400)
Subject: Fix XXXs that slipped into router and handle HTLCFailCHannelUpdates
X-Git-Tag: v0.0.12~288^2~1
X-Git-Url: http://git.bitcoin.ninja/?a=commitdiff_plain;h=c662dd3e946749f2c4a4c7e3169963d4e7906048;p=rust-lightning

Fix XXXs that slipped into router and handle HTLCFailCHannelUpdates
---

diff --git a/src/ln/router.rs b/src/ln/router.rs
index 4c3bcb980..0d049f5a7 100644
--- a/src/ln/router.rs
+++ b/src/ln/router.rs
@@ -349,18 +349,25 @@ impl RoutingMessageHandler for Router {
 			&msgs::HTLCFailChannelUpdate::ChannelUpdateMessage { ref msg } => {
 				let _ = self.handle_channel_update(msg);
 			},
-			&msgs::HTLCFailChannelUpdate::ChannelClosed { ref short_channel_id, is_permanent:_ } => {
-//XXX
+			&msgs::HTLCFailChannelUpdate::ChannelClosed { ref short_channel_id, ref is_permanent } => {
 				let mut network = self.network_map.write().unwrap();
-				if let Some(chan) = network.channels.remove(short_channel_id) {
-					Self::remove_channel_in_nodes(&mut network.nodes, &chan, *short_channel_id);
+				if *is_permanent {
+					if let Some(chan) = network.channels.remove(short_channel_id) {
+						Self::remove_channel_in_nodes(&mut network.nodes, &chan, *short_channel_id);
+					}
+				} else {
+					if let Some(chan) = network.channels.get_mut(short_channel_id) {
+						chan.one_to_two.enabled = false;
+						chan.two_to_one.enabled = false;
+					}
 				}
 			},
-			&msgs::HTLCFailChannelUpdate::NodeFailure { ref node_id, is_permanent:_ } => {
-//XXX
-				//let mut network = self.network_map.write().unwrap();
-				//TODO: check _blamed_upstream_node
-				self.mark_node_bad(node_id, false);
+			&msgs::HTLCFailChannelUpdate::NodeFailure { ref node_id, ref is_permanent } => {
+				if *is_permanent {
+					//TODO: Wholly remove the node
+				} else {
+					self.mark_node_bad(node_id, false);
+				}
 			},
 		}
 	}