82 lines
2.7 KiB
Python
82 lines
2.7 KiB
Python
|
from multiaddr import Multiaddr
|
||
|
|
from pexpect.spawnbase import SpawnBase
|
||
|
|
|
||
|
|
from p2pclient import Client
|
||
|
|
|
||
|
|
from libp2p.peer.peerinfo import info_from_p2p_addr, PeerInfo
|
||
|
|
from libp2p.peer.id import ID
|
||
|
|
|
||
|
|
from .constants import PEXPECT_NEW_LINE, LOCALHOST_IP
|
||
|
|
from .envs import GO_BIN_PATH
|
||
|
|
|
||
|
|
P2PD_PATH = GO_BIN_PATH / "p2pd"
|
||
|
|
|
||
|
|
|
||
|
|
class Daemon:
|
||
|
|
proc: SpawnBase
|
||
|
|
control: Client
|
||
|
|
peer_info: PeerInfo
|
||
|
|
|
||
|
|
def __init__(self, proc: SpawnBase, control: Client, peer_info: PeerInfo) -> None:
|
||
|
|
self.proc = proc
|
||
|
|
self.control = control
|
||
|
|
self.peer_info = peer_info
|
||
|
|
|
||
|
|
def __repr__(self) -> str:
|
||
|
|
return f"<Daemon {self.peer_id.to_string()[2:8]}>"
|
||
|
|
|
||
|
|
@property
|
||
|
|
def peer_id(self) -> ID:
|
||
|
|
return self.peer_info.peer_id
|
||
|
|
|
||
|
|
@property
|
||
|
|
def listen_maddr(self) -> Multiaddr:
|
||
|
|
return self.peer_info.addrs[0]
|
||
|
|
|
||
|
|
|
||
|
|
async def make_p2pd(
|
||
|
|
proc_factory,
|
||
|
|
unused_tcp_port_factory,
|
||
|
|
is_secure: bool,
|
||
|
|
is_pubsub_enabled=True,
|
||
|
|
is_gossipsub=True,
|
||
|
|
is_pubsub_signing=False,
|
||
|
|
is_pubsub_signing_strict=False,
|
||
|
|
) -> Daemon:
|
||
|
|
control_maddr = Multiaddr(f"/ip4/{LOCALHOST_IP}/tcp/{unused_tcp_port_factory()}")
|
||
|
|
args = [f"-listen={str(control_maddr)}"]
|
||
|
|
# NOTE: To support `-insecure`, we need to hack `go-libp2p-daemon`.
|
||
|
|
if not is_secure:
|
||
|
|
args.append("-insecure=true")
|
||
|
|
if is_pubsub_enabled:
|
||
|
|
args.append("-pubsub")
|
||
|
|
if is_gossipsub:
|
||
|
|
args.append("-pubsubRouter=gossipsub")
|
||
|
|
else:
|
||
|
|
args.append("-pubsubRouter=floodsub")
|
||
|
|
if not is_pubsub_signing:
|
||
|
|
args.append("-pubsubSign=false")
|
||
|
|
if not is_pubsub_signing_strict:
|
||
|
|
args.append("-pubsubSignStrict=false")
|
||
|
|
# NOTE:
|
||
|
|
# Two other params are possibly what we want to configure:
|
||
|
|
# - gossipsubHeartbeatInterval: GossipSubHeartbeatInitialDelay = 100 * time.Millisecond
|
||
|
|
# - gossipsubHeartbeatInitialDelay: GossipSubHeartbeatInterval = 1 * time.Second
|
||
|
|
# Referece: https://github.com/libp2p/go-libp2p-daemon/blob/b95e77dbfcd186ccf817f51e95f73f9fd5982600/p2pd/main.go#L348-L353 # noqa: E501
|
||
|
|
proc = proc_factory(str(P2PD_PATH), args)
|
||
|
|
await proc.expect(r"Peer ID:\s+(\w+)" + PEXPECT_NEW_LINE, async_=True)
|
||
|
|
peer_id_base58 = proc.match.group(1)
|
||
|
|
await proc.expect(r"Peer Addrs:", async_=True)
|
||
|
|
await proc.expect(
|
||
|
|
rf"(/ip4/{LOCALHOST_IP}/tcp/[\w]+)" + PEXPECT_NEW_LINE, async_=True
|
||
|
|
)
|
||
|
|
daemon_listener_maddr = Multiaddr(proc.match.group(1))
|
||
|
|
daemon_pinfo = info_from_p2p_addr(
|
||
|
|
daemon_listener_maddr.encapsulate(f"/p2p/{peer_id_base58}")
|
||
|
|
)
|
||
|
|
client_callback_maddr = Multiaddr(
|
||
|
|
f"/ip4/{LOCALHOST_IP}/tcp/{unused_tcp_port_factory()}"
|
||
|
|
)
|
||
|
|
p2pc = Client(control_maddr, client_callback_maddr)
|
||
|
|
return Daemon(proc, p2pc, daemon_pinfo)
|