@@ -3,14 +3,11 @@ use std::{
33 time:: { Duration , Instant , SystemTime , UNIX_EPOCH } ,
44} ;
55
6- use bitcoin:: {
7- consensus,
8- p2p:: {
9- message:: { CommandString , NetworkMessage , RawNetworkMessage } ,
10- message_network:: VersionMessage ,
11- Address , Magic , ServiceFlags ,
12- } ,
13- FeeRate , Network ,
6+ use bitcoin:: { consensus, FeeRate , Network } ;
7+ use p2p:: {
8+ message:: { CommandString , NetworkMessage , RawNetworkMessage } ,
9+ message_network:: VersionMessage ,
10+ Address , Magic , ServiceFlags ,
1411} ;
1512use validation:: ValidationExt ;
1613
@@ -648,7 +645,7 @@ macro_rules! define_version_message_logic {
648645 }
649646
650647 let mut negotiation = Negotiation :: default ( ) ;
651- let magic = Magic :: from_params( $conn. network) ;
648+ let magic = Magic :: from_params( $conn. network) . expect ( "unknown network" ) ;
652649 let mut write_half = WriteHalf :: V1 ( magic) ;
653650 let mut read_half = ReadHalf :: V1 ( magic) ;
654651 let nonce = rand:: random( ) ;
@@ -744,6 +741,7 @@ macro_rules! define_version_message_logic {
744741 } } ;
745742}
746743
744+ #[ cfg( feature = "tokio" ) ]
747745macro_rules! async_awaiter {
748746 ( $e: expr) => {
749747 $e. await
@@ -756,6 +754,7 @@ macro_rules! blocking_awaiter {
756754 } ;
757755}
758756
757+ #[ cfg( feature = "tokio" ) ]
759758macro_rules! read_message_async {
760759 ( $reader: expr, $magic: expr) => {
761760 $crate:: define_read_message_logic!( async_awaiter, $reader, $magic)
@@ -774,15 +773,19 @@ macro_rules! version_handshake_blocking {
774773 } ;
775774}
776775
776+ #[ cfg( feature = "tokio" ) ]
777777macro_rules! version_handshake_async {
778778 ( $reader: expr, $conn: ident) => {
779779 $crate:: define_version_message_logic!( async_awaiter, $reader, $conn)
780780 } ;
781781}
782782
783+ #[ cfg( feature = "tokio" ) ]
783784pub ( crate ) use async_awaiter;
784785pub ( crate ) use blocking_awaiter;
786+ #[ cfg( feature = "tokio" ) ]
785787pub ( crate ) use read_message_async;
786788pub ( crate ) use read_message_blocking;
789+ #[ cfg( feature = "tokio" ) ]
787790pub ( crate ) use version_handshake_async;
788791pub ( crate ) use version_handshake_blocking;
0 commit comments