diff options
Diffstat (limited to 'internal/manager')
| -rw-r--r-- | internal/manager/helpers.go | 52 | ||||
| -rw-r--r-- | internal/manager/manager.go | 94 | 
2 files changed, 146 insertions, 0 deletions
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") +}  | 
