o jGz@sdZddlmZddlZddlmZddlZddlZddlm Z ddl m Z m Z m Z e r2ddlmZddlmZdd lmZdd lmZdd lmZdd lmZmZdd lmZddl mZmZddl Z ddl!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'ddl(m)Z)m*Z*ddl+m,Z,ddl-m.Z.ddl/m0Z0m1Z2ddl3m1Z4ddl5m1Z6ddl7m8Z8m9Z9m:Z:ddl;mZ>ddl?m@Z@ddlAmBZBmCZCddlDmEZEddlFmGZGmHZHddlImJZJmKZKmLZLmMZMmNZNmOZOmPZPmQZQmRZRmSZSddlTZTddlUmVZVmWZWddlXmYZZm[Z\ddl]mYZ^m_Z_ddl`mambmcZbdd ldmeZemfZfdd!lgmYZhmiZidd"ljmkZkdd#llmmZmddlnZTdd$lompZpddlqZTepd%ard%Zsd&Ztd'ZuevewZxdd,d-Zyddd3d4Zzddd6d7Z{dd8d9Z|dd;d<Z}dd=d>Z~dd?d@ZddAdBZddDdEZddFdGZ ddd\d]Z        ^ ^ ^ _ ^        dddudvZdddddddd^d^d^d^deud^d^dddddd^dwdxdddd_d^fdddZedZdS)zLibp2p Python implementation.) annotationsN)Path)is_quic_multiaddr) TYPE_CHECKINGAnycastPortDemultiplexer)ed25519) serialization) QUICTransport)QUICTransportConfig)MappingSequence)version)LiteralOptional)IHostINetworkService IPeerRouting IPeerStoreISecureTransport ITransport)ChannelBridgeListenerEventBus)ResourceManager)KeyPair)Ed25519PrivateKeycreate_new_key_pair)r) TMuxerOptions TProtocolTSecurityOptions) BasicHost) RoutedHost)Swarm)ConnectionConfig RetryConfig)ID) PeerStorecreate_signed_peer_record) create_sync_peerstorecreate_async_peerstorecreate_sync_sqlite_peerstorecreate_async_sqlite_peerstorecreate_sync_memory_peerstorecreate_async_memory_peerstorecreate_sync_leveldb_peerstorecreate_async_leveldb_peerstorecreate_sync_rocksdb_peerstorecreate_async_rocksdb_peerstore)PLAINTEXT_PROTOCOL_IDInsecureTransport) PROTOCOL_ID Transport)r6 TLSTransport)MPLEX_PROTOCOL_IDMplex)r6Yamux)TCP)TransportUpgrader) setup_loggingZYAMUXMPLEX muxer_nameLiteral['YAMUX', 'MPLEX']returnNonecCs,|}|ttfvrtd|d|adS)z Set the default multiplexer protocol to use. :param muxer_name: Either "YAMUX" or "MPLEX" :raise ValueError: If an unsupported muxer name is provided Unknown muxer: . Use 'YAMUX' or 'MPLEX'.N)upper MUXER_YAMUX MUXER_MPLEX ValueError DEFAULT_MUXER)rAZ muxer_upperrL2/Users/sumanjeet/code/py-libp2p/libp2p/__init__.pyset_default_muxers rNr key_pairrtypestrcCsp|j}|dkr0t|tsJ|}tj|}|jtjj tj j t d}t jjj|dS td|)ab Persist a private key to disk in PEM format. Currently supports only Ed25519 keys. Writes the key to a predefined path for later retrieval. :param key_pair: KeyPair object containing private and public keys. :param type: Type of key to save (default: "ed25519"). :raises ValueError: if an unsupported key type is provided. r encodingformatZencryption_algorithmunsupported key type: N) private_key isinstancerto_bytesr Zfrom_private_bytes private_bytesr EncodingZPEM PrivateFormatZPKCS8 NoEncryptionlibp2putilspaths ED25519_PATH write_bytesrJ)rOrPpvt_keyraw crypto_keypemrLrLrM save_keypairs  rfKeyPair | NonecCs|dkr>tjjj}|sdS|}tj|dd}t|t j s"J|j tj j tjj td}t |}|}t||S td|)aJ Load a private key from disk and reconstruct its KeyPair. Currently supports only Ed25519 keys. Returns None if the key file does not exist. :param type: Type of key to load (default: "ed25519"). :return: KeyPair object if found, or None. :raises ValueError: if an unsupported key type is provided. r N)passwordrRrU)r]r^r_r`exists read_bytesr Zload_pem_private_keyrWr rrYrZZRawr[r\ from_bytesZget_public_keyrrJ)rPpathrerdrcrbZpub_keyrLrLrM load_keypairs*    rmcCstS)z_ Returns the currently selected default muxer. :return: Either "YAMUX" or "MPLEX" )rKrLrLrLrMget_default_muxersrnrcCttttttiS)zn Returns muxer options with Yamux as the primary choice. :return: Muxer options with Yamux first )r YAMUX_PROTOCOL_IDr;r9r:rLrLrLrMcreate_yamux_muxer_optionrqcCro)zn Returns muxer options with Mplex as the primary choice. :return: Muxer options with Mplex first )r r9r:rpr;rLrLrLrMcreate_mplex_muxer_optionrrrscCtSN)create_new_rsa_key_pairrLrLrLrMgenerate_new_rsa_identitysrwcCrt)z Generate a new Ed25519 identity key pair. Ed25519 is preferred for better interoperability with other libp2p implementations (e.g., Rust, Go) which often disable RSA support. )create_new_ed25519_key_pairrLrLrLrMgenerate_new_ed25519_identitysryr'cCs|j}t|Sru) public_keyr'Z from_pubkey)rOrzrLrLrMgenerate_peer_id_froms r{cCstdkrtStS)z Returns the default muxer options based on the current default muxer setting. :return: Muxer options with the preferred muxer first r?)rKrsrqrLrLrLrMget_default_muxer_optionssr| listen_addrs$Sequence[multiaddr.Multiaddr] | None transportsSequence[ITransport] | None enable_quicbool enable_webrtc enable_tcpenable_websocketenable_autotlsupgraderr= quic_configQUICTransportConfig | Nonetls_client_configssl.SSLContext | Nonetls_server_config quic_class type | Nonelist[ITransport]c Cs(|durt|S| dur| nt} g}|rt}|D]}dd|D}d}d|vr6d|vr6d|vr6t}nfd|vs>d|vrQ|durHtd q| |j| |d }nKd|vsYd|vrgd d lm }||| | d }n5d|vrd dl m }|dur{tdq||jd}nd|vrd dl m }|durtdq||jd}|durqt|}||vr||||q|rtdd|Ds| |j| |d g}|s|r|| |j| |d |rd d lm }|||| | d |rd dl m }|durtdn |||jd|s |s|t|S)u[ Build the ordered list of transports for the Swarm's TransportManager. Priority: 1. Explicit ``transports`` list — used as-is (highest priority). 2. ``listen_addrs`` inspection — auto-detects which transports are needed by inspecting **every** address (not just the first one). 3. ``enable_*`` flags — coarse-grained control when no addresses given. 4. Default fallback: TCP only. :param key_pair: The host's key pair (needed by QUIC for TLS). :param listen_addrs: The multiaddrs the host will listen on. :param transports: Explicit transport list, or ``None`` to auto-build. :param enable_quic: Whether to create a QUIC transport when auto-building. :param enable_webrtc: Whether to create a WebRTC transport when auto-building. :param enable_tcp: Whether to include a TCP transport when auto-building. :param enable_websocket: Whether to include a WebSocket transport. :param enable_autotls: Whether to enable AutoTLS in QUIC/WebSocket transports. :param upgrader: The upgrader passed to WebSocket transport at construction. :param quic_config: Optional QUIC transport configuration. :param tls_client_config: TLS client context for WebSocket. :param tls_server_config: TLS server context for WebSocket. :returns: Ordered list of :class:`~libp2p.abc.ITransport` instances. NcSsg|]}|jqSrLname.0prLrLrM Tz/_build_transports_for_swarm..tcpwswssZquiczquic-v1z.QUIC transport requires key_pair (private_key))configrr)WebsocketTransport)rrz webrtc-direct)WebRTCDirectTransportz0WebRTC transport requires key_pair (private_key))rVZwebrtc)WebRTCPrivateTransportcss|] }t|jdkVqdS)r NrP__name__rtrLrLrM sz._build_transports_for_swarm..)listr set protocolsr<loggerwarningrVZ$libp2p.transport.websocket.transportrZ!libp2p.transport.webrtc.transportrZ)libp2p.transport.webrtc.private_transportrrPaddappendany)rOr}rrrrrrrrrrrZ_QUICTransportresultZ seen_classesaddrrZ transport_objrrrclsrLrLrM_build_transports_for_swarm s)               rFT muxer_optTMuxerOptions | Nonesec_optTSecurityOptions | None peerstore_optIPeerStore | Nonemuxer_preference Literal['YAMUX', 'MPLEX'] | None retry_configRetryConfig | Noneconnection_config-ConnectionConfig | QUICTransportConfig | Noneresource_managerResourceManager | Nonepsk str | Nonemetric_send_channel"trio.MemorySendChannel[Any] | None event_busEventBus | Nonerc0Cs|durt}t|}t| tr| nd}t}|p7tt||jdtt || dt t j t |t tt||di}|durQ|}|ttfvrNtd|d|}nt}|durZ|}n |tkrbt}nt}t||d}t|||||| | | ||||td }td d d |D|pt}|||d}i}|rd d lm} i}!|D]F}"z;dd|" D}#d|#vrWq|"!dp|"!d}$|"!d}%|$dus|%durWqt"|$t#|%f}&|!$|&g%|#Wqt&yYqw|!'D].\\}'}(})t(dd|)D}*t(dd|)D}+|*r|+r| |'|(||'|(f<td|'|(qd dl)m*},|,|d}-t+||||| | ||||-d }.|durSz d dl,m-}/|/}Wn t&yRd}Ynw|dur]|..||.S)uN Create a swarm instance with multi-transport support. The swarm can listen on and dial over multiple transports simultaneously (TCP, WebSocket, QUIC), mirroring go-libp2p's architecture. Transport selection priority (highest to lowest): 1. **Explicit ``transports`` list** — used as-is; all other transport parameters are ignored. 2. **``listen_addrs`` inspection** — each address is inspected to determine which transports are needed (TCP, WebSocket, QUIC). All detected transport types are created and registered. 3. **``enable_*`` flags** — coarse-grained control when no addresses are provided (``enable_quic``, ``enable_websocket``, ``enable_tcp``). 4. **Default fallback** — TCP only. :param key_pair: optional choice of the ``KeyPair`` :param muxer_opt: optional choice of stream muxer :param sec_opt: optional choice of security upgrade :param peerstore_opt: optional peerstore :param muxer_preference: optional explicit muxer preference (``"YAMUX"`` or ``"MPLEX"``) :param listen_addrs: optional list of multiaddrs to listen on. **All** addresses are inspected to determine which transports to create. :param transports: explicit list of transport instances to register. When provided, all ``enable_*`` flags and ``listen_addrs``-based detection are bypassed. :param enable_quic: include a QUIC transport when auto-building (deprecated; prefer passing ``listen_addrs`` with QUIC addresses or ``transports``). :param enable_autotls: enable AutoTLS for QUIC / WebSocket transports. :param enable_tcp: include a TCP transport when auto-building (default True). :param enable_websocket: include a WebSocket transport when auto-building. :param retry_config: optional connection retry configuration. :param connection_config: optional connection configuration. :param tls_client_config: TLS client context for WebSocket transport. :param tls_server_config: TLS server context for WebSocket transport. :param resource_manager: optional resource manager for connection/stream limits. :param psk: optional pre-shared key for PSK encryption. :return: a Swarm instance implementing INetworkService. Examples:: # TCP only (default) swarm = new_swarm() # TCP + WebSocket + QUIC auto-detected from listen_addrs swarm = new_swarm(listen_addrs=[ Multiaddr("/ip4/0.0.0.0/tcp/4001"), Multiaddr("/ip4/0.0.0.0/tcp/4002/ws"), Multiaddr("/ip4/0.0.0.0/udp/4003/quic-v1"), ]) # Explicit transport list swarm = new_swarm(transports=[TCP(), QUICTransport(kp.private_key)]) Note: Yamux (/yamux/1.0.0) is the preferred stream multiplexer due to its improved performance and features. Mplex (/mplex/6.7.0) is retained for backward compatibility but may be deprecated in the future. Note: Ed25519 keys are used by default for better interoperability with other libp2p implementations (Rust, Go) which often disable RSA support. N)Z noise_privkey)r) peerstorerErF)secure_transports_by_protocolmuxer_transports_by_protocol) rOr}rrrrrrrrrrrznew_swarm: using transports: %scSsg|]}t|jqSrLrrrLrLrMrQsznew_swarm..rrcSsh|]}|jqSrLrrrLrLrM frznew_swarm..rZip4Zip6css |] }d|vo d|vVqdSrrNrLrpsrLrLrMrws znew_swarm..css |] }d|vp d|vVqdSrrLrrLrLrMrzsz:new_swarm: created PortDemultiplexer for shared port %s:%d)TransportManager) port_demuxers)rrrrrrtransport_managernew_resource_manager)/ryr{rWr create_new_x25519_key_pairNOISE_PROTOCOL_IDNoiseTransportrVTLS_PROTOCOL_IDr8r secior'r7r4r5rGrHrIrJrKrsrqr=rr rdebugr(Z add_key_pairlibp2p.transport.cmuxr rZvalue_for_protocolrQint setdefaultr ExceptionitemsrZlibp2p.transport.managerrr$ libp2p.rcmgrrZset_resource_manager)0rOrrrrr}rrrrrrrrrrrrrrZid_optquic_transport_optZnoise_key_pairrZ temp_prefZactive_preferencerrZtransport_listrZ port_demuxrZ_ConnMgrZ port_protosrprotosZhost_valZport_valkeyhostportZ proto_setsZ has_plain_tcpZhas_wsrrswarm_new_rmrLrLrM new_swarmsY                 rg$@disc_optIPeerRouting | None enable_mDNS enable_upnpenable_metrics bootstraplist[str] | Nonenegotiate_timeoutrrbootstrap_allow_ipv6bootstrap_dns_timeoutfloatbootstrap_dns_max_retriesConnectionConfig | Noneannounce_addrsrc"CsR| s |dur tdt}d\}}| r"td\}}|t||dur=z ddlm}|}Wn t y<d}Ynw| rF|durF|} n|} t did| d|d |d |d |d |d | d|d|d| d|d|d|d|d|d|d|d|d|}!|durt |!|||| ||||||d St |!|| || |||||||d S)u` Create a new libp2p host based on the given parameters. The host can listen on and dial over multiple transports simultaneously (TCP, WebSocket, QUIC), mirroring go-libp2p's architecture. Transport selection priority (highest to lowest): 1. ``transports`` — explicit list, used as-is. 2. ``listen_addrs`` inspection — all addresses are inspected. 3. ``enable_*`` flags. 4. Default: TCP only. :param key_pair: optional choice of the ``KeyPair`` :param muxer_opt: optional choice of stream muxer :param sec_opt: optional choice of security upgrade :param peerstore_opt: optional peerstore :param disc_opt: optional discovery :param muxer_preference: optional explicit muxer preference :param listen_addrs: optional list of multiaddrs to listen on. **All** addresses are inspected to determine which transports to create. :param enable_mDNS: whether to enable mDNS discovery :param bootstrap: optional list of bootstrap peer addresses as strings :param enable_quic: optional choice to use QUIC for transport :param enable_autotls: optional choice to use AutoTLS for security :param quic_transport_opt: optional configuration for quic transport :param tls_client_config: optional TLS client configuration for WebSocket transport :param tls_server_config: optional TLS server configuration for WebSocket transport :param resource_manager: optional resource manager for connection/stream limits :type resource_manager: :class:`libp2p.rcmgr.ResourceManager` or None :param psk: optional pre-shared key (PSK) :param bootstrap_allow_ipv6: if True, bootstrap accepts IPv6+TCP addresses :param bootstrap_dns_timeout: DNS resolution timeout in seconds per attempt :param bootstrap_dns_max_retries: max DNS resolution retries with backoff :param connection_config: optional connection configuration for connection manager :param announce_addrs: if set, these replace listen addrs in get_addrs() :param transports: explicit list of transport instances to register. When provided, all ``enable_*`` flags and ``listen_addrs``-based detection are bypassed. :param enable_tcp: include a TCP transport when auto-building (default True). :param enable_websocket: include a WebSocket transport when auto-building. :return: return a host instance Nz?QUIC config provided but QUIC not enabled, ignoring QUIC config)NNdrrrrrOrrrrrr}rrrrrrrrrr) networkrouterrrrrrrrrr) rrrrrrmetric_recv_channelrrrrrrL) rrrtrioopen_memory_channelZregister_listenerrrrrrr#r")"rOrrrrrr}rrrrrrrrrrrrrrrrrrrrrrrrrZeffective_configrrLrLrMnew_hosts L           rr])rArBrCrD)rOrrPrQrCrD)rPrQrCrg)rCrQ)rCr)rCr)rOrrCr'ru)rOrr}r~rrrrrrrrrrrrrr=rrrrrrrrrCr)NNNNNNNFFFTFNNNNNNNN)*rOrgrrrrrrrrr}r~rrrrrrrrrrrrrrrrrrrrrrrrrrrrrCr):rOrgrrrrrrrrrrr}r~rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr~rrrrrrrCr)__doc__ __future__rloggingpathlibrsslrZlibp2p.transport.quic.utilsrtypingrrrrr Z)cryptography.hazmat.primitives.asymmetricr Zcryptography.hazmat.primitivesr Zlibp2p.transport.quic.transportr Zlibp2p.transport.quic.configr collections.abcrrimportlib.metadatarZ __versionrrZ multiaddr libp2p.abcrrrrrrZ libp2p.eventsrrrrZlibp2p.crypto.keysrZlibp2p.crypto.ed25519rrrxZlibp2p.crypto.rsarvZlibp2p.crypto.x25519rZlibp2p.custom_typesrr r!Zlibp2p.host.basic_hostr"Zlibp2p.host.routed_hostr#Zlibp2p.network.swarmr$Zlibp2p.network.configr%r&Zlibp2p.peer.idr'Zlibp2p.peer.peerstorer(r)Zlibp2p.peer.persistentr*r+r,r-r.r/r0r1r2r3r]Z"libp2p.security.insecure.transportr4r5Zlibp2p.security.noise.transportr6rr7rZlibp2p.security.tls.transportrr8Zlibp2p.security.secio.transportsecurityr transportZlibp2p.stream_muxer.mplex.mplexr9r:Zlibp2p.stream_muxer.yamux.yamuxrpr;Zlibp2p.transport.tcp.tcpr<Zlibp2p.transport.upgraderr=Z libp2p.utilsZlibp2p.utils.loggingr>Zlibp2p.utils.pathsrKrHrIZDEFAULT_NEGOTIATE_TIMEOUT getLoggerrrrNrfrmrnrqrsrwryr{r|rrr __version__rLrLrLrMs                  0        (    x  $