Extract preimage from revoked HTLC-Success to claim backward

Author: Antoine Riard

src/ln/channelmonitor.rs

@@ -1891,6 +1891,21 @@ impl ChannelMonitor {
 				}
 			}
 
+			macro_rules! check_htlc_valid_remote {
+				($remote_txid: expr, $htlc_output: expr) => {
+					if let &Some(txid) = $remote_txid {
+						for &(ref pending_htlc, ref pending_source) in self.remote_claimable_outpoints.get(&txid).unwrap() {
+							if pending_htlc.payment_hash == $htlc_output.payment_hash && pending_htlc.amount_msat == $htlc_output.amount_msat {
+								if let &Some(ref source) = pending_source {
+									payment_data = Some(((**source).clone(), $htlc_output.payment_hash));
+									break;
+								}
+							}
+						}
+					}
+				}
+			}
+
 			macro_rules! scan_commitment {
 				($htlcs: expr, $tx_info: expr, $local_tx: expr) => {
 					for (ref htlc_output, source_option) in $htlcs {
@@ -1903,6 +1918,10 @@ impl ChannelMonitor {
 								// has timed out, or we screwed up. In any case, we should now
 								// resolve the source HTLC with the original sender.
 								payment_data = Some(((*source).clone(), htlc_output.payment_hash));
+							} else if let Storage::Local { ref current_remote_commitment_txid, .. } = self.key_storage {
+								check_htlc_valid_remote!(current_remote_commitment_txid, htlc_output);
+							} else if let Storage::Local { ref prev_remote_commitment_txid, .. } = self.key_storage {
+								check_htlc_valid_remote!(prev_remote_commitment_txid, htlc_output);
 							} else {
 								log_claim!($tx_info, $local_tx, htlc_output, false);
 								continue 'outer_loop;