solver trusted peers

Author: almogdepaz

chia/solver/solver.py

@@ -14,6 +14,7 @@
 from chia.rpc.rpc_server import StateChangedProtocol, default_get_connections
 from chia.server.server import ChiaServer
 from chia.server.ws_connection import WSChiaConnection
+from chia.util.network import is_localhost
 
 log = logging.getLogger(__name__)
 
@@ -43,6 +44,7 @@ def server(self) -> ChiaServer:
     def __init__(self, root_path: Path, config: dict[str, Any], constants: ConsensusConstants):
         self.log = log
         self.root_path = root_path
+        self.config = config
         self._shut_down = False
         num_threads = config["num_threads"]
         self.log.info(f"Initializing solver with {num_threads} threads")
@@ -74,7 +76,16 @@ def get_connections(self, request_node_type: Optional[NodeType]) -> list[dict[st
         return default_get_connections(server=self.server, request_node_type=request_node_type)
 
     async def on_connect(self, connection: WSChiaConnection) -> None:
-        pass
+        if is_localhost(connection.peer_info.host):
+            self.log.info(f"Accepting localhost connection from {connection.connection_type}: {connection.get_peer_logging()}")
+            return
+
+        trusted_peers = self.config.get("trusted_peers", {})
+        if self.server.is_trusted_peer(connection, trusted_peers):
+            self.log.info(f"Accepting trusted peer connection from {connection.connection_type}: {connection.get_peer_logging()}")
+            return
+        self.log.warning(f"Rejecting untrusted connection from {connection.connection_type}: {connection.get_peer_logging()}")
+        await connection.close()
 
     async def on_disconnect(self, connection: WSChiaConnection) -> None:
         self.log.info(f"peer disconnected {connection.get_peer_logging()}")