Rusty Russell
6 years ago
committed by
Christian Decker
Christian Decker
4 changed files with 253 additions and 4 deletions
@ -0,0 +1,220 @@ |
|||||
|
/* Simple tool to route gossip from a peer. */ |
||||
|
#include <ccan/err/err.h> |
||||
|
#include <ccan/io/io.h> |
||||
|
#include <ccan/opt/opt.h> |
||||
|
#include <ccan/read_write_all/read_write_all.h> |
||||
|
#include <common/crypto_sync.h> |
||||
|
#include <common/dev_disconnect.h> |
||||
|
#include <common/peer_failed.h> |
||||
|
#include <common/status.h> |
||||
|
#include <netdb.h> |
||||
|
#include <secp256k1_ecdh.h> |
||||
|
#include <wire/peer_wire.h> |
||||
|
|
||||
|
#define io_write_ simple_write |
||||
|
#define io_read_ simple_read |
||||
|
|
||||
|
static struct io_plan *simple_write(struct io_conn *conn, |
||||
|
const void *data, size_t len, |
||||
|
struct io_plan *(*next)(struct io_conn *, void *), |
||||
|
void *arg); |
||||
|
|
||||
|
static struct io_plan *simple_read(struct io_conn *conn, |
||||
|
void *data, size_t len, |
||||
|
struct io_plan *(*next)(struct io_conn *, void *), |
||||
|
void *next_arg); |
||||
|
|
||||
|
#include "../connectd/handshake.c" |
||||
|
|
||||
|
/* This makes the handshake prototypes work. */ |
||||
|
struct io_conn { |
||||
|
int fd; |
||||
|
}; |
||||
|
|
||||
|
static struct secret notsosecret; |
||||
|
static bool initial_sync = false; |
||||
|
static unsigned long max_messages = -1UL; |
||||
|
|
||||
|
/* Empty stubs to make us compile */ |
||||
|
void status_peer_io(enum log_level iodir, const u8 *p) |
||||
|
{ |
||||
|
} |
||||
|
|
||||
|
void status_fmt(enum log_level level, const char *fmt, ...) |
||||
|
{ |
||||
|
} |
||||
|
|
||||
|
#if DEVELOPER |
||||
|
void dev_sabotage_fd(int fd) |
||||
|
{ |
||||
|
abort(); |
||||
|
} |
||||
|
|
||||
|
void dev_blackhole_fd(int fd) |
||||
|
{ |
||||
|
abort(); |
||||
|
} |
||||
|
|
||||
|
enum dev_disconnect dev_disconnect(int pkt_type) |
||||
|
{ |
||||
|
return DEV_DISCONNECT_NORMAL; |
||||
|
} |
||||
|
#endif |
||||
|
|
||||
|
void peer_failed_connection_lost(void) |
||||
|
{ |
||||
|
exit(0); |
||||
|
} |
||||
|
|
||||
|
bool hsm_do_ecdh(struct secret *ss, const struct pubkey *point) |
||||
|
{ |
||||
|
if (secp256k1_ecdh(secp256k1_ctx, ss->data, &point->pubkey, |
||||
|
notsosecret.data) != 1) |
||||
|
errx(1, "ECDH failed"); |
||||
|
return true; |
||||
|
} |
||||
|
|
||||
|
/* We don't want to discard *any* messages. */ |
||||
|
bool is_unknown_msg_discardable(const u8 *cursor) |
||||
|
{ |
||||
|
return false; |
||||
|
} |
||||
|
|
||||
|
static struct io_plan *simple_write(struct io_conn *conn, |
||||
|
const void *data, size_t len, |
||||
|
struct io_plan *(*next)(struct io_conn *, void *), |
||||
|
void *arg) |
||||
|
{ |
||||
|
if (!write_all(conn->fd, data, len)) |
||||
|
err(1, "Writing data"); |
||||
|
return next(conn, arg); |
||||
|
} |
||||
|
|
||||
|
static struct io_plan *simple_read(struct io_conn *conn, |
||||
|
void *data, size_t len, |
||||
|
struct io_plan *(*next)(struct io_conn *, void *), |
||||
|
void *next_arg) |
||||
|
{ |
||||
|
if (!read_all(conn->fd, data, len)) |
||||
|
err(1, "Reading data"); |
||||
|
return next(conn, next_arg); |
||||
|
} |
||||
|
|
||||
|
static struct io_plan *handshake_success(struct io_conn *conn, |
||||
|
const struct pubkey *them, |
||||
|
const struct wireaddr_internal *addr, |
||||
|
const struct crypto_state *orig_cs, |
||||
|
void *unused) |
||||
|
{ |
||||
|
u8 *msg; |
||||
|
struct crypto_state cs = *orig_cs; |
||||
|
u8 *local_features; |
||||
|
|
||||
|
if (initial_sync) { |
||||
|
local_features = tal(conn, u8); |
||||
|
local_features[0] = (1 << 3); |
||||
|
} else |
||||
|
local_features = NULL; |
||||
|
|
||||
|
msg = towire_init(NULL, NULL, local_features); |
||||
|
|
||||
|
sync_crypto_write(&cs, conn->fd, take(msg)); |
||||
|
/* Ignore their init message. */ |
||||
|
tal_free(sync_crypto_read(NULL, &cs, conn->fd)); |
||||
|
|
||||
|
/* Now write out whatever we get. */ |
||||
|
while ((msg = sync_crypto_read(NULL, &cs, conn->fd)) != NULL) { |
||||
|
be16 len = cpu_to_be16(tal_bytelen(msg)); |
||||
|
|
||||
|
if (!write_all(STDOUT_FILENO, &len, sizeof(len)) |
||||
|
|| !write_all(STDOUT_FILENO, msg, tal_bytelen(msg))) |
||||
|
err(1, "Writing out msg"); |
||||
|
tal_free(msg); |
||||
|
|
||||
|
if (--max_messages == 0) |
||||
|
exit(0); |
||||
|
} |
||||
|
err(1, "Reading msg"); |
||||
|
} |
||||
|
|
||||
|
int main(int argc, char *argv[]) |
||||
|
{ |
||||
|
struct io_conn *conn = tal(NULL, struct io_conn); |
||||
|
struct wireaddr_internal addr; |
||||
|
int af; |
||||
|
struct pubkey us, them; |
||||
|
const char *err_msg; |
||||
|
const char *at; |
||||
|
struct addrinfo *ai; |
||||
|
|
||||
|
setup_locale(); |
||||
|
secp256k1_ctx = secp256k1_context_create(SECP256K1_CONTEXT_VERIFY | |
||||
|
SECP256K1_CONTEXT_SIGN); |
||||
|
|
||||
|
opt_register_noarg("--initial-sync", opt_set_bool, &initial_sync, |
||||
|
"Stream complete gossip history at start"); |
||||
|
opt_register_arg("--max-messages", opt_set_ulongval, opt_show_ulongval, |
||||
|
&max_messages, |
||||
|
"Terminate after reading this many messages (> 0)"); |
||||
|
opt_register_noarg("--help|-h", opt_usage_and_exit, |
||||
|
"id@addr[:port]\n" |
||||
|
"Connect to a lightning peer and relay gossip messages from it", |
||||
|
"Print this message."); |
||||
|
|
||||
|
opt_parse(&argc, argv, opt_log_stderr_exit); |
||||
|
if (argc != 2) |
||||
|
opt_usage_exit_fail("Need an id@addr to connect to"); |
||||
|
at = strchr(argv[1], '@'); |
||||
|
if (!at) |
||||
|
opt_usage_exit_fail("Need id@addr"); |
||||
|
|
||||
|
if (!pubkey_from_hexstr(argv[1], at - argv[1], &them)) |
||||
|
opt_usage_exit_fail("Invalid id %.*s", |
||||
|
(int)(at - argv[1]), argv[1]); |
||||
|
|
||||
|
if (!parse_wireaddr_internal(at+1, &addr, DEFAULT_PORT, NULL, |
||||
|
true, false, &err_msg)) |
||||
|
opt_usage_exit_fail("%s '%s'", err_msg, argv[1]); |
||||
|
|
||||
|
switch (addr.itype) { |
||||
|
case ADDR_INTERNAL_SOCKNAME: |
||||
|
af = AF_LOCAL; |
||||
|
ai = wireaddr_internal_to_addrinfo(conn, &addr); |
||||
|
break; |
||||
|
case ADDR_INTERNAL_ALLPROTO: |
||||
|
case ADDR_INTERNAL_AUTOTOR: |
||||
|
case ADDR_INTERNAL_FORPROXY: |
||||
|
opt_usage_exit_fail("Don't support proxy use"); |
||||
|
|
||||
|
case ADDR_INTERNAL_WIREADDR: |
||||
|
switch (addr.u.wireaddr.type) { |
||||
|
case ADDR_TYPE_TOR_V2: |
||||
|
case ADDR_TYPE_TOR_V3: |
||||
|
opt_usage_exit_fail("Don't support proxy use"); |
||||
|
break; |
||||
|
case ADDR_TYPE_IPV4: |
||||
|
af = AF_INET; |
||||
|
break; |
||||
|
case ADDR_TYPE_IPV6: |
||||
|
af = AF_INET6; |
||||
|
break; |
||||
|
case ADDR_TYPE_PADDING: |
||||
|
abort(); |
||||
|
} |
||||
|
ai = wireaddr_to_addrinfo(tmpctx, &addr.u.wireaddr); |
||||
|
} |
||||
|
conn->fd = socket(af, SOCK_STREAM, 0); |
||||
|
if (conn->fd < 0) |
||||
|
err(1, "Creating socket"); |
||||
|
|
||||
|
memset(¬sosecret, 0x42, sizeof(notsosecret)); |
||||
|
if (!pubkey_from_secret(¬sosecret, &us)) |
||||
|
errx(1, "Creating pubkey"); |
||||
|
|
||||
|
if (connect(conn->fd, ai->ai_addr, ai->ai_addrlen) != 0) |
||||
|
err(1, "Connecting to %s", at+1); |
||||
|
|
||||
|
initiator_handshake(conn, &us, &them, &addr, handshake_success, NULL); |
||||
|
exit(0); |
||||
|
} |
||||
|
|
||||
Loading…
Reference in new issue