Connect to peers alt addr (if provided) when reconnecting. If both nodes

maxrantil · maxrantil · commit f6bacf08f91b · 2024-06-24T10:14:05.000+02:00
provide alt addrsses only the connection_out will utilize the alt addr.

Signed-off-by: Max Rantil &lt;rantil@pm.me&gt;
diff --git a/channeld/channeld.c b/channeld/channeld.c
@@ -196,11 +196,11 @@ struct peer {
 	bool experimental_upgrade;
 
 	/* Alt address for peer connections not publicly announced */
-	u8 *alt_addr;
+	u8 *our_alt_addr;
 };
 
 static void start_commit_timer(struct peer *peer);
-static void send_peer_alt_address(struct peer *peer);
+static void send_peer_our_alt_address(struct peer *peer);
 
 static void billboard_update(const struct peer *peer)
 {
@@ -542,11 +542,11 @@ static void handle_peer_splice_locked(struct peer *peer, const u8 *msg)
 	check_mutual_splice_locked(peer);
 }
 
-static void send_peer_alt_address(struct peer *peer) {
+static void send_peer_our_alt_address(struct peer *peer) {
 	struct pubkey node_id;
 
 	if (pubkey_from_node_id(&node_id, &peer->id)) {
-		u8 *msg = towire_peer_alt_address(peer, &node_id, peer->alt_addr);
+		u8 *msg = towire_peer_alt_address(peer, &node_id, peer->our_alt_addr);
 		peer_write(peer->pps, take(msg));
 	}
 }
@@ -4179,8 +4179,8 @@ static void peer_in(struct peer *peer, const u8 *msg)
 
 	check_tx_abort(peer, msg);
 
-	if (peer->alt_addr)
-		send_peer_alt_address(peer);
+	if (peer->our_alt_addr)
+		send_peer_our_alt_address(peer);
 
 	/* If we're in STFU mode and aren't waiting for a STFU mode
 	 * specific message, the only valid message was tx_abort */
@@ -5890,7 +5890,7 @@ static void init_channel(struct peer *peer)
 				    &peer->experimental_upgrade,
 				    &peer->splice_state->inflights,
 				    &peer->local_alias,
-				    &peer->alt_addr,
+				    &peer->our_alt_addr,
 				    &peer->id)) {
 		master_badmsg(WIRE_CHANNELD_INIT, msg);
 	}
diff --git a/connectd/connectd.c b/connectd/connectd.c
@@ -506,9 +506,30 @@ static struct io_plan *connection_in(struct io_conn *conn,
 
 	conn_in_arg.daemon = daemon;
 	conn_in_arg.is_websocket = false;
+
+	// HOW CAN WE GET THE peer->alt_addr AND USE THAT INSTEAD OF &conn_in_arg.addr????
+
 	return conn_in(conn, &conn_in_arg);
 }
 
+void handle_peer_alt_address(struct peer *peer, const u8 *msg)
+{
+	u8 *peer_alt_addr;
+	struct pubkey peer_id;
+	// u32 *timestamp = NULL;
+
+	if (!fromwire_peer_alt_address(peer, msg, &peer_id, &peer_alt_addr/* , timestamp */)) {
+		master_badmsg(WIRE_PEER_ALT_ADDRESS, msg);
+	}
+
+	// peer->alt_addr = tal_strdup(peer, (char *)peer_alt_addr);	// This is WIP, Wanted to utilize this in `connction_in()` but didn't manage that yet.
+
+	msg = towire_connectd_alt_address(NULL, &peer_id, peer_alt_addr);
+	daemon_conn_send(peer->daemon->master, take(msg));
+
+	tal_free(peer_alt_addr);
+}
+
 /*~ <hello>I speak web socket</hello>.
  *
  * Actually that's dumb, websocket (aka rfc6455) looks nothing like that. */
@@ -1414,7 +1435,8 @@ static void connect_init(struct daemon *daemon, const u8 *msg)
 		&daemon->dev_fast_gossip,
 		&dev_disconnect,
 		&daemon->dev_no_ping_timer,
-		&daemon->dev_handshake_no_reply)) {
+		&daemon->dev_handshake_no_reply/* , //This is WIP
+		&daemon->alt_addr */)) {
 		/* This is a helper which prints the type expected and the actual
 		 * message, then exits (it should never be called!). */
 		master_badmsg(WIRE_CONNECTD_INIT, msg);
@@ -1776,22 +1798,6 @@ static void try_connect_peer(struct daemon *daemon,
 	try_connect_one_addr(connect);
 }
 
-void handle_peer_alt_addr(struct peer *peer, const u8 *msg)
-{
-	u8 *alt_addr;
-	struct pubkey peer_id;
-
-	// u32 *timestamp = NULL;
-	if (!fromwire_peer_alt_address(peer, msg, &peer_id, &alt_addr/* , timestamp */)) {
-		master_badmsg(WIRE_PEER_ALT_ADDRESS, msg);
-	}
-
-	msg = towire_connectd_alt_address(NULL, &peer_id, alt_addr);
-	daemon_conn_send(peer->daemon->master, take(msg));
-
-	tal_free(alt_addr);
-}
-
 /* lightningd tells us to connect to a peer by id, with optional addr hint. */
 static void connect_to_peer(struct daemon *daemon, const u8 *msg)
 {
diff --git a/connectd/connectd.h b/connectd/connectd.h
@@ -61,6 +61,10 @@ struct peer {
 
 	/* The pubkey of the node */
 	struct node_id id;
+
+	//This is WIP
+	/* char *alt_addr; */
+
 	/* Counters and keys for symmetric crypto */
 	struct crypto_state cs;
 
@@ -225,6 +229,9 @@ struct daemon {
 	 * resort, but doing so leaks our address so can be disabled. */
 	bool use_dns;
 
+	// This is WIP
+	/* u8 *alt_addr; */
+
 	/* The address that the broken response returns instead of
 	 * NXDOMAIN. NULL if we have not detected a broken resolver. */
 	struct sockaddr *broken_resolver_response;
@@ -294,6 +301,6 @@ void destroy_peer(struct peer *peer);
 void close_random_connection(struct daemon *daemon);
 
 /* Handles alternative address message from peer. */
-void handle_peer_alt_addr(struct peer *peer, const u8 *msg);
+void handle_peer_alt_address(struct peer *peer, const u8 *msg);
 
 #endif /* LIGHTNING_CONNECTD_CONNECTD_H */
diff --git a/connectd/connectd_wire.csv b/connectd/connectd_wire.csv
@@ -27,6 +27,9 @@ msgdata,connectd_init,dev_disconnect,bool,
 msgdata,connectd_init,dev_no_ping_timer,bool,
 # Allow incoming connections, but don't talk.
 msgdata,connectd_init,dev_noreply,bool,
+# Use an alternative private address if provided by peer.
+#msgdata,connectd_init,alt_addr_len,u16,
+#msgdata,connectd_init,alt_addr,u8,alt_addr_len,
 
 # Connectd->master, here are the addresses I bound, can announce.
 msgtype,connectd_init_reply,2100
diff --git a/connectd/multiplex.c b/connectd/multiplex.c
@@ -773,7 +773,7 @@ static bool handle_message_locally(struct peer *peer, const u8 *msg)
 		handle_onion_message(peer->daemon, peer, msg);
 		return true;
 	} else if (type == WIRE_PEER_ALT_ADDRESS) { // IS THIS THE RIGHT PLACE ?? IT WORKS BUT...
-		handle_peer_alt_addr(peer, msg);
+		handle_peer_alt_address(peer, msg);
 		return true;
 	} else if (handle_custommsg(peer->daemon, peer, msg)) {
 		return true;
diff --git a/doc/schemas/lightning-listconfigs.json b/doc/schemas/lightning-listconfigs.json
@@ -1185,7 +1185,7 @@
             ],
             "properties": {
               "values_str": {
-								"added": "v24.05",
+                "added": "v24.05",
                 "type": "array",
                 "items": {
                   "type": "string",
diff --git a/lightningd/channel.h b/lightningd/channel.h
@@ -798,4 +798,5 @@ const u8 *channel_update_for_error(const tal_t *ctx,
 
 struct amount_msat htlc_max_possible_send(const struct channel *channel);
 
+
 #endif /* LIGHTNING_LIGHTNINGD_CHANNEL_H */
diff --git a/lightningd/channel_control.c b/lightningd/channel_control.c
@@ -1726,7 +1726,7 @@ bool peer_start_channeld(struct channel *channel,
 				       cast_const2(const struct inflight **,
 						   inflights),
 				       *channel->alias[LOCAL],
-				       ld->alt_addr,
+				       ld->our_alt_addr,
 				       &ld->id);
 
 	/* We don't expect a response: we are triggered by funding_depth_cb. */
diff --git a/lightningd/connect_control.c b/lightningd/connect_control.c
@@ -335,6 +335,12 @@ static void try_connect(const tal_t *ctx,
 {
 	struct delayed_reconnect *d;
 	struct peer *peer;
+	const struct wireaddr_internal *alt_addr;
+
+	alt_addr = wallet_get_peer_alt_addr(ld->wallet, id);
+	if (alt_addr) {
+		addrhint = alt_addr; //IS THIS CORRECT TO JUST OVERWRITE IT LIKE THIS??
+	}
 
 	/* Don't stack, unless this is an instant reconnect */
 	d = delayed_reconnect_map_get(ld->delayed_reconnect_map, id);
@@ -364,9 +370,6 @@ static void try_connect(const tal_t *ctx,
 	/* Update any channel billboards */
 	peer = peer_by_id(ld, id);
 	if (peer) {
-		// struct pubkey pubkey;
-		// if (pubkey_from_node_id(&pubkey, id)) {
-		// 	send_peer_alt_address(peer, &pubkey, (const u8 *)"127.21.21.21"); // THIS MIGHT BE THE RIGHT PLACE IN THE END, NEED TO SEND ANOTHER MSG FROM MASTER -> CHANNELD then????
 		struct channel *channel;
 		list_for_each(&peer->channels, channel, list) {
 			if (!channel_state_wants_peercomms(channel->state))
@@ -559,20 +562,21 @@ static void handle_custommsg_in(struct lightningd *ld, const u8 *msg)
 	plugin_hook_call_custommsg(ld, NULL, p);
 }
 
-static void handle_alt_addr_in(struct lightningd *ld, const u8 *msg)
+static void handle_peer_alt_addr_in(struct lightningd *ld, const u8 *msg)
 {
-	struct pubkey node_id;
-	struct node_id id;
-	u8 *alt_addr;
+	struct pubkey peer_node_id;
+	u8 *peer_alt_addr;
 
-	if (!fromwire_connectd_alt_address(tmpctx, msg, &node_id, &alt_addr)) {
+	if (!fromwire_connectd_alt_address(tmpctx, msg, &peer_node_id, &peer_alt_addr)) {
 		log_broken(ld->log, "Malformed peer_alt_addr_msg: %s",
 								tal_hex(tmpctx, msg));
 		return;
 	}
 
-	node_id_from_pubkey(&id, &node_id);
-	wallet_peer_alt_addr(ld->wallet->db, &id, (char *)alt_addr);
+	struct node_id id;
+	node_id_from_pubkey(&id, &peer_node_id);
+	wallet_add_peer_alt_addr(ld->wallet->db, &id, (char *)peer_alt_addr);
+	// SHOULD/CAN WE ALSO SAVE IT TO THE peer struct HERE, TO BE ABLE TO GET IN IT CONNECTD LATER FOR THE `connection_in`???
 }
 
 static void connectd_start_shutdown_reply(struct subd *connectd,
@@ -666,7 +670,7 @@ static unsigned connectd_msg(struct subd *connectd, const u8 *msg, const int *fd
 		break;
 
 	case WIRE_CONNECTD_ALT_ADDRESS:
-		handle_alt_addr_in(connectd->ld, msg);
+		handle_peer_alt_addr_in(connectd->ld, msg);
 		break;
 	}
 	return 0;
@@ -788,7 +792,8 @@ int connectd_init(struct lightningd *ld)
 	    ld->dev_fast_gossip,
 	    ld->dev_disconnect_fd >= 0,
 	    ld->dev_no_ping_timer,
-	    ld->dev_handshake_no_reply);
+	    ld->dev_handshake_no_reply/* , //This is WIP
+	    ld->our_alt_addr */);
 
 	subd_req(ld->connectd, ld->connectd, take(msg), -1, 0,
 		 connect_init_done, NULL);
diff --git a/lightningd/lightningd.h b/lightningd/lightningd.h
@@ -175,9 +175,6 @@ struct lightningd {
 	/* Do we want to reconnect to other peers having only unannouced channels with us? */
 	bool reconnect_private;
 
-	/* Alt address for peer connections not publicly announced */
-	u8 *alt_addr;
-
 	/* How many outstanding startup connection attempts? */
 	size_t num_startup_connects;
 
@@ -196,6 +193,9 @@ struct lightningd {
 	struct wireaddr_internal *binding;
 	struct wireaddr *announceable;
 
+	/* Alternative address for peer connections not publicly announced */
+	u8 *our_alt_addr;
+
 	/* Current node announcement (if any) */
 	const u8 *node_announcement;
 
diff --git a/lightningd/options.c b/lightningd/options.c
@@ -444,10 +444,10 @@ static char *opt_add_alt_addr(const char *arg, struct lightningd *ld)
 {
 	assert(arg != NULL);
 
-	ld->alt_addr = tal_free(ld->alt_addr);
+	ld->our_alt_addr = tal_free(ld->our_alt_addr);
 
-	ld->alt_addr = (u8 *)tal_strdup(ld, arg);
-	if (ld->alt_addr == NULL) {
+	ld->our_alt_addr = (u8 *)tal_strdup(ld, arg);
+	if (ld->our_alt_addr == NULL) {
 		return tal_fmt(tmpctx, "Failed to allocate memory for address: %s", arg);
 	}
 
diff --git a/lightningd/peer_control.c b/lightningd/peer_control.c
@@ -2250,6 +2250,13 @@ static void json_add_peer(struct lightningd *ld,
 					fmt_wireaddr(response, p->remote_addr));
 	}
 
+	struct wireaddr_internal *peer_alt_addr;
+	peer_alt_addr = wallet_get_peer_alt_addr(ld->wallet, &p->id);
+	if (peer_alt_addr) {
+		json_add_string(response, "alt_addr",
+				fmt_wireaddr_internal(tmpctx, peer_alt_addr));
+	}
+
 	/* Note: If !PEER_CONNECTED, peer may use different features on reconnect */
 	json_add_hex_talarr(response, "features", p->their_features);
 
diff --git a/pytest.ini b/pytest.ini
diff --git a/wallet/wallet.c b/wallet/wallet.c
@@ -1027,7 +1027,7 @@ static struct peer *wallet_peer_load(struct wallet *w, const u64 dbid)
 	return peer;
 }
 
-void wallet_peer_alt_addr(struct db *db, const struct node_id *node_id, const char *alt_addr)
+void wallet_add_peer_alt_addr(struct db *db, const struct node_id *node_id, const char *alt_addr)
 {
 	struct db_stmt *stmt;
 
@@ -1041,6 +1041,61 @@ void wallet_peer_alt_addr(struct db *db, const struct node_id *node_id, const ch
 	db_exec_prepared_v2(take(stmt));
 }
 
+struct wireaddr_internal *wallet_get_peer_alt_addr(struct wallet *w, const struct node_id *node_id) {
+	struct db_stmt *stmt;
+	struct wireaddr_internal *alt_address = NULL;
+	bool result;
+	bool transaction_started = false;
+
+	if (!db_in_transaction(w->db)) {
+		db_begin_transaction(w->db);
+		transaction_started = true;
+	}
+
+	stmt = db_prepare_v2(w->db, SQL("SELECT alt_addr FROM peers WHERE node_id=?"));
+	if (!stmt) {
+		log_broken(w->log, "Failed to prepare statement for node_id %s", fmt_node_id(tmpctx, node_id));
+		return NULL;
+	}
+
+	db_bind_node_id(stmt, node_id);
+	db_query_prepared(stmt);
+
+	result = db_step(stmt);
+	if (!result) {
+		log_broken(w->log, "No alternative address found for peer %s", fmt_node_id(tmpctx, node_id));
+	} else {
+		const char *addr_str = db_col_strdup(tmpctx, stmt, "alt_addr");
+		if (!addr_str) {
+			log_broken(w->log, "No address string retrieved for peer %s", fmt_node_id(tmpctx, node_id));
+		} else if (*addr_str == '\0') {
+			log_broken(w->log, "Empty address string retrieved for peer %s", fmt_node_id(tmpctx, node_id));
+			tal_free(addr_str);
+		} else {
+			alt_address = tal(tmpctx, struct wireaddr_internal);
+			if (!alt_address) {
+				log_broken(w->log, "Memory allocation failed for alternative address of peer %s", fmt_node_id(tmpctx, node_id));
+			} else {
+				const char *err = parse_wireaddr_internal(tmpctx, addr_str, 0, false, alt_address);
+				if (err) {
+					log_broken(w->log, "Invalid alternative address %s for peer %s: %s",
+							       addr_str, fmt_node_id(tmpctx, node_id), err);
+					tal_free(alt_address);
+					alt_address = NULL;
+				}
+			}
+			tal_free(addr_str);
+		}
+	}
+
+	if (transaction_started) {
+		db_commit_transaction(w->db);
+	}
+
+	tal_free(stmt);
+	return alt_address;
+	}
+
 static struct bitcoin_signature *
 wallet_htlc_sigs_load(const tal_t *ctx, struct wallet *w, u64 channelid,
 		      bool option_anchors)
diff --git a/wallet/wallet.h b/wallet/wallet.h

Original file line number	Diff line number	Diff line change
`@@ -798,4 +798,5 @@ const u8 channel_update_for_error(const tal_t ctx,`
`798`	`798`
`799`	`799`	`struct amount_msat htlc_max_possible_send(const struct channel *channel);`
`800`	`800`
	`801`	`+`
`801`	`802`	`#endif /* LIGHTNING_LIGHTNINGD_CHANNEL_H */`