mirror of
https://github.com/Chia-Network/chia-blockchain.git
synced 2026-08-28 18:14:19 -05:00
Matt HauffandGitHub
40db4635a8
* Enable PEP604 Ruff rules * Fix harcoded signature in test * Hack CLVMStreamable test with note to fast follow
89 lines
2.7 KiB
Python
89 lines
2.7 KiB
Python
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass
|
|
from enum import IntEnum
|
|
|
|
from chia_rs.sized_ints import int16, uint8, uint16
|
|
|
|
from chia.protocols.outbound_message import NodeType
|
|
from chia.util.streamable import Streamable, streamable
|
|
|
|
protocol_version = {
|
|
NodeType.FULL_NODE: "0.0.36",
|
|
NodeType.HARVESTER: "0.0.37",
|
|
NodeType.FARMER: "0.0.37",
|
|
NodeType.TIMELORD: "0.0.36",
|
|
NodeType.INTRODUCER: "0.0.36",
|
|
NodeType.WALLET: "0.0.38",
|
|
NodeType.DATA_LAYER: "0.0.36",
|
|
NodeType.SOLVER: "0.0.37",
|
|
}
|
|
|
|
"""
|
|
Handshake when establishing a connection between two servers.
|
|
Note: When changing this file, also change protocol_message_types.py
|
|
"""
|
|
|
|
|
|
# Capabilities can be added here when new features are added to the protocol
|
|
# These are passed in as uint16 into the Handshake
|
|
class Capability(IntEnum):
|
|
BASE = 1 # Base capability just means it supports the chia protocol at mainnet
|
|
# introduces RequestBlockHeaders, which is a faster API for fetching header blocks
|
|
# !! the old API is *RequestHeaderBlock* !!
|
|
BLOCK_HEADERS = 2
|
|
# Specifies support for v1 and v2 versions of rate limits. Peers will use the lowest shared capability:
|
|
# if peer A support v3 and peer B supports v2, they should send:
|
|
# (BASE, RATE_LIMITS_V2, RATE_LIMITS_V3), and (BASE, RATE_LIMITS_V2) respectively. They will use the V2 limits.
|
|
RATE_LIMITS_V2 = 3
|
|
|
|
# a node can handle a None response and not wait the full timeout
|
|
# capability removed but functionality is still supported
|
|
NONE_RESPONSE = 4
|
|
|
|
# Opts in to receiving mempool updates for subscribed transactions
|
|
# This is between a full node and receiving wallet
|
|
MEMPOOL_UPDATES = 5
|
|
|
|
|
|
# These are the default capabilities used in all outgoing handshakes.
|
|
# "1" means the capability is supported and enabled.
|
|
_capabilities: list[tuple[uint16, str]] = [
|
|
(uint16(Capability.BASE.value), "1"),
|
|
(uint16(Capability.BLOCK_HEADERS.value), "1"),
|
|
(uint16(Capability.RATE_LIMITS_V2.value), "1"),
|
|
]
|
|
_mempool_updates = [
|
|
(uint16(Capability.MEMPOOL_UPDATES.value), "1"),
|
|
]
|
|
|
|
default_capabilities = {
|
|
NodeType.FULL_NODE: _capabilities + _mempool_updates,
|
|
NodeType.HARVESTER: _capabilities,
|
|
NodeType.FARMER: _capabilities,
|
|
NodeType.TIMELORD: _capabilities,
|
|
NodeType.INTRODUCER: _capabilities,
|
|
NodeType.WALLET: _capabilities,
|
|
NodeType.DATA_LAYER: _capabilities,
|
|
NodeType.SOLVER: _capabilities,
|
|
}
|
|
|
|
|
|
@streamable
|
|
@dataclass(frozen=True)
|
|
class Handshake(Streamable):
|
|
network_id: str
|
|
protocol_version: str
|
|
software_version: str
|
|
server_port: uint16
|
|
node_type: uint8
|
|
capabilities: list[tuple[uint16, str]]
|
|
|
|
|
|
@streamable
|
|
@dataclass(frozen=True)
|
|
class Error(Streamable):
|
|
code: int16 # Err
|
|
message: str
|
|
data: bytes | None = None
|