894b7a50ae
Some checks failed
CI Docker Mining Proof / Linux agent hashrate proof (push) Has been cancelled
Content-addressed shard DHT on seeder agents with subnet_primary_seeder election, cross-subnet fleet_torrent_gossip, BGP swarm magnets, C2 torrent manifest, and k-of-n peer fetch with C2 fallback.
88 lines
2.1 KiB
Go
88 lines
2.1 KiB
Go
package client
|
|
|
|
import (
|
|
"encoding/json"
|
|
"strings"
|
|
|
|
"crypto-miner-agent/deploy"
|
|
)
|
|
|
|
func (c *AgentClient) setFleetTorrentEnabled(enabled bool) {
|
|
c.mu.Lock()
|
|
c.fleetTorrentEnabled = enabled
|
|
c.mu.Unlock()
|
|
}
|
|
|
|
func (c *AgentClient) fleetTorrentEnabledSnapshot() bool {
|
|
c.mu.Lock()
|
|
defer c.mu.Unlock()
|
|
return c.fleetTorrentEnabled
|
|
}
|
|
|
|
func (c *AgentClient) setSubnetPrimarySeeder(primary bool) {
|
|
c.mu.Lock()
|
|
c.subnetPrimarySeeder = primary
|
|
c.cfg.SubnetPrimarySeeder = primary
|
|
c.mu.Unlock()
|
|
}
|
|
|
|
func (c *AgentClient) subnetPrimarySeederSnapshot() bool {
|
|
c.mu.Lock()
|
|
defer c.mu.Unlock()
|
|
return c.subnetPrimarySeeder
|
|
}
|
|
|
|
func (c *AgentClient) applyAuthFleetTorrent(resp AuthResponse) {
|
|
c.setFleetTorrentEnabled(resp.FleetTorrentEnabled)
|
|
if resp.SubnetPrimarySeeder != "" {
|
|
c.setSubnetPrimarySeeder(resp.SubnetPrimarySeeder == c.cfg.AgentID)
|
|
}
|
|
}
|
|
|
|
func (c *AgentClient) handleFleetTorrentGossip(payload json.RawMessage) {
|
|
var body struct {
|
|
Records []deploy.FleetGossipRecord `json:"records"`
|
|
}
|
|
if err := json.Unmarshal(payload, &body); err != nil || len(body.Records) == 0 {
|
|
return
|
|
}
|
|
deploy.FleetShardDHTSnapshot().MergeFleetGossipRecords(body.Records)
|
|
}
|
|
|
|
func (c *AgentClient) writeFleetTorrentGossip(records []deploy.FleetGossipRecord) {
|
|
if !c.fleetTorrentEnabledSnapshot() || len(records) == 0 {
|
|
return
|
|
}
|
|
c.mu.Lock()
|
|
agentID := c.cfg.AgentID
|
|
c.mu.Unlock()
|
|
localIP, _ := deploy.PrimaryLocalIPv4()
|
|
subnet := deploy.SubnetFromIP(localIP)
|
|
out := make([]deploy.FleetGossipRecord, 0, len(records))
|
|
for _, r := range records {
|
|
r.Kind = strings.TrimSpace(strings.ToLower(r.Kind))
|
|
if r.AgentID == "" {
|
|
r.AgentID = agentID
|
|
}
|
|
if r.Subnet == "" {
|
|
r.Subnet = subnet
|
|
}
|
|
out = append(out, r)
|
|
}
|
|
payload, err := json.Marshal(map[string]interface{}{"records": out})
|
|
if err != nil {
|
|
return
|
|
}
|
|
_ = c.write(Message{Type: "fleet_torrent_gossip", Payload: payload})
|
|
}
|
|
|
|
func (c *AgentClient) advertiseFleetTorrentHealthy() {
|
|
if !c.fleetTorrentEnabledSnapshot() {
|
|
return
|
|
}
|
|
c.writeFleetTorrentGossip([]deploy.FleetGossipRecord{{
|
|
Kind: deploy.FleetGossipHealthy,
|
|
Healthy: true,
|
|
}})
|
|
}
|