From 895d5fea41177e444c18f4fdc820fffa5f67d5bf Mon Sep 17 00:00:00 2001 From: Rasmus Dahlberg Date: Sat, 9 Dec 2023 17:08:45 +0100 Subject: Add drafty skeleton --- internal/manager/helpers.go | 52 +++++++++++++++++++++++++ internal/manager/manager.go | 94 +++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 146 insertions(+) create mode 100644 internal/manager/helpers.go create mode 100644 internal/manager/manager.go (limited to 'internal/manager') diff --git a/internal/manager/helpers.go b/internal/manager/helpers.go new file mode 100644 index 0000000..a9a2158 --- /dev/null +++ b/internal/manager/helpers.go @@ -0,0 +1,52 @@ +package manager + +import ( + "crypto/sha256" + "encoding/base64" + "fmt" + + ct "github.com/google/certificate-transparency-go" + "gitlab.torproject.org/rgdd/ct/pkg/metadata" + "rgdd.se/silent-ct/pkg/monitor" +) + +func selectLogs(m metadata.Metadata) []monitor.MessageLogConfig { + var logs []monitor.MessageLogConfig + for _, operator := range m.Operators { + for _, log := range operator.Logs { + if log.State == nil { + continue // ignore logs without a state (should not happen) + } + if log.State.Name == metadata.LogStatePending { + continue // log is not yet relevant + } + if log.State.Name == metadata.LogStateRetired { + continue // log is not expected to be reachable + } + if log.State.Name == metadata.LogStateRejected { + continue // log is not expected to be reachable + } + + // FIXME: remove me instead of hard coding Argon 2024 + id, _ := log.Key.ID() + got := fmt.Sprintf("%s", base64.StdEncoding.EncodeToString(id[:])) + want := "7s3QZNXbGs7FXLedtM0TojKHRny87N7DUUhZRnEftZs=" + if got != want { + continue + } + + logs = append(logs, monitor.MessageLogConfig{ + Metadata: log, + State: monitor.MonitorState{ + LogState: monitor.LogState{ct.SignedTreeHead{ + SHA256RootHash: [sha256.Size]byte{47, 66, 110, 15, 246, 154, 8, 100, 150, 140, 206, 208, 17, 57, 112, 116, 210, 3, 19, 55, 46, 63, 209, 12, 234, 130, 225, 124, 237, 2, 64, 228}, + TreeSize: 610650601, + Timestamp: 1702108968538, + }}, + NextIndex: 388452203, + }, + }) + } + } + return logs +} diff --git a/internal/manager/manager.go b/internal/manager/manager.go new file mode 100644 index 0000000..2210c9b --- /dev/null +++ b/internal/manager/manager.go @@ -0,0 +1,94 @@ +package manager + +import ( + "context" + "encoding/json" + "fmt" + "os" + "time" + + "gitlab.torproject.org/rgdd/ct/pkg/metadata" + "rgdd.se/silent-ct/pkg/monitor" + "rgdd.se/silent-ct/pkg/server" +) + +const ( + DefaultStateDir = "/home/rgdd/.local/share/silent-ct" // FIXME + DefaultMetadataRefreshInterval = 1 * time.Hour +) + +type Config struct { + StateDir string + Nodes server.Nodes + + MetadataRefreshInterval time.Duration +} + +type Manager struct { + Config +} + +func New(cfg Config) (Manager, error) { + if cfg.StateDir == "" { + cfg.StateDir = DefaultStateDir + } + if cfg.MetadataRefreshInterval == 0 { + cfg.MetadataRefreshInterval = DefaultMetadataRefreshInterval + } + return Manager{Config: cfg}, nil +} + +func (mgr *Manager) Run(ctx context.Context, + serverCh chan server.MessageNodeSubmission, + monitorCh chan monitor.MessageLogProgress, + configCh chan []monitor.MessageLogConfig, + errorCh chan error) error { + + md, err := mgr.metadataRead() + if err != nil { + return fmt.Errorf("read metadata: %v\n", err) + } + configCh <- selectLogs(md) + + ticker := time.NewTicker(mgr.MetadataRefreshInterval) + defer ticker.Stop() + + for { + select { + case <-ctx.Done(): + return nil + case <-ticker.C: + mu, err := mgr.metadataUpdate(ctx, md) + if err != nil { + continue + } + if mu.Version.Major <= md.Version.Major { + continue + } + md = mu + configCh <- selectLogs(md) + case ev := <-monitorCh: + fmt.Printf("DEBUG: received event from monitor with %d matches\n", len(ev.Matches)) + case ev := <-serverCh: + fmt.Printf("DEBUG: received event from server\n: %v", ev) + case err := <-errorCh: + fmt.Printf("DEBUG: received error: %v\n", err) + } + } +} + +func (mgr *Manager) metadataRead() (metadata.Metadata, error) { + b, err := os.ReadFile(mgr.StateDir + "/metadata.json") + if err != nil { + return metadata.Metadata{}, err + } + var md metadata.Metadata + if err := json.Unmarshal(b, &md); err != nil { + return metadata.Metadata{}, err + } + return md, nil +} + +func (mgr *Manager) metadataUpdate(ctx context.Context, old metadata.Metadata) (metadata.Metadata, error) { + return metadata.Metadata{}, fmt.Errorf("TODO: update metadata") +} -- cgit v1.2.3