projection: begin the exact-path public epoch
The public tree and its history contain only the listed paths. Earlier projection history remains preserved internally. Source-Sha: 913fde029b935671833254797f0f20f1eb9fabba Policy-Sha: 913fde029b935671833254797f0f20f1eb9fabba Tree-Digest: 180ae530c1058a2a5c89837bdce2d323ae83e669e38590ca72e75b8d92b7262f
This commit is contained in:
commit
67e26be2d9
838 changed files with 157302 additions and 0 deletions
107
server/internal/command/factory.go
Normal file
107
server/internal/command/factory.go
Normal file
|
|
@ -0,0 +1,107 @@
|
|||
package command
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"time"
|
||||
|
||||
"github.com/Fimeg/RedFlag/server/internal/database/queries"
|
||||
"github.com/Fimeg/RedFlag/server/internal/models"
|
||||
"github.com/gofrs/uuid/v5"
|
||||
)
|
||||
|
||||
// Factory creates validated AgentCommand instances
|
||||
type Factory struct {
|
||||
validator *Validator
|
||||
commandQueries *queries.CommandQueries
|
||||
}
|
||||
|
||||
// NewFactory creates a new command factory
|
||||
func NewFactory(commandQueries *queries.CommandQueries) *Factory {
|
||||
return &Factory{
|
||||
validator: NewValidator(),
|
||||
commandQueries: commandQueries,
|
||||
}
|
||||
}
|
||||
|
||||
// Create generates a new validated AgentCommand with unique ID
|
||||
func (f *Factory) Create(agentID uuid.UUID, commandType string, params map[string]interface{}) (*models.AgentCommand, error) {
|
||||
cmd := &models.AgentCommand{
|
||||
ID: uuid.Must(uuid.NewV4()),
|
||||
AgentID: agentID,
|
||||
CommandType: commandType,
|
||||
Status: "pending",
|
||||
Source: determineSource(commandType),
|
||||
Params: params,
|
||||
CreatedAt: time.Now().UTC(),
|
||||
UpdatedAt: time.Now().UTC(),
|
||||
}
|
||||
|
||||
if err := f.validator.Validate(cmd); err != nil {
|
||||
return nil, fmt.Errorf("command validation failed: %w", err)
|
||||
}
|
||||
|
||||
return cmd, nil
|
||||
}
|
||||
|
||||
// CreateWithIdempotency generates a command with idempotency protection
|
||||
// If a command with the same idempotency key exists, returns it instead of creating a duplicate
|
||||
func (f *Factory) CreateWithIdempotency(agentID uuid.UUID, commandType string, params map[string]interface{}, idempotencyKey string) (*models.AgentCommand, error) {
|
||||
// If no idempotency key provided, create normally
|
||||
if idempotencyKey == "" {
|
||||
return f.Create(agentID, commandType, params)
|
||||
}
|
||||
|
||||
// Check for existing command with same idempotency key
|
||||
existing, err := f.commandQueries.GetCommandByIdempotencyKey(agentID, idempotencyKey)
|
||||
if err != nil {
|
||||
// If no existing command found, proceed with creation
|
||||
if err.Error() == "sql: no rows in result set" || err.Error() == "command not found" {
|
||||
cmd := &models.AgentCommand{
|
||||
ID: uuid.Must(uuid.NewV4()),
|
||||
AgentID: agentID,
|
||||
CommandType: commandType,
|
||||
Status: "pending",
|
||||
Source: determineSource(commandType),
|
||||
IdempotencyKey: &idempotencyKey,
|
||||
Params: params,
|
||||
CreatedAt: time.Now().UTC(),
|
||||
UpdatedAt: time.Now().UTC(),
|
||||
}
|
||||
|
||||
if err := f.validator.Validate(cmd); err != nil {
|
||||
return nil, fmt.Errorf("command validation failed: %w", err)
|
||||
}
|
||||
return cmd, nil
|
||||
}
|
||||
return nil, fmt.Errorf("failed to check idempotency: %w", err)
|
||||
}
|
||||
|
||||
// Return existing command
|
||||
return existing, nil
|
||||
}
|
||||
|
||||
// determineSource classifies command source based on type
|
||||
func determineSource(commandType string) string {
|
||||
if isSystemCommand(commandType) {
|
||||
return "system"
|
||||
}
|
||||
return "manual"
|
||||
}
|
||||
|
||||
func isSystemCommand(commandType string) bool {
|
||||
systemCommands := []string{
|
||||
"enable_heartbeat",
|
||||
"disable_heartbeat",
|
||||
"update_check",
|
||||
"cleanup_old_logs",
|
||||
"heartbeat_on",
|
||||
"heartbeat_off",
|
||||
}
|
||||
|
||||
for _, cmd := range systemCommands {
|
||||
if commandType == cmd {
|
||||
return true
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
123
server/internal/command/validator.go
Normal file
123
server/internal/command/validator.go
Normal file
|
|
@ -0,0 +1,123 @@
|
|||
package command
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"fmt"
|
||||
|
||||
"github.com/gofrs/uuid/v5"
|
||||
"github.com/Fimeg/RedFlag/server/internal/models"
|
||||
)
|
||||
|
||||
// Validator validates command parameters
|
||||
type Validator struct {
|
||||
minCheckInSeconds int
|
||||
maxCheckInSeconds int
|
||||
minScannerMinutes int
|
||||
maxScannerMinutes int
|
||||
}
|
||||
|
||||
// NewValidator creates a new command validator
|
||||
func NewValidator() *Validator {
|
||||
return &Validator{
|
||||
minCheckInSeconds: 60, // 1 minute minimum
|
||||
maxCheckInSeconds: 3600, // 1 hour maximum
|
||||
minScannerMinutes: 1, // 1 minute minimum
|
||||
maxScannerMinutes: 1440, // 24 hours maximum
|
||||
}
|
||||
}
|
||||
|
||||
// Validate performs comprehensive command validation
|
||||
func (v *Validator) Validate(cmd *models.AgentCommand) error {
|
||||
if cmd == nil {
|
||||
return errors.New("command cannot be nil")
|
||||
}
|
||||
|
||||
if cmd.ID == uuid.Nil {
|
||||
return errors.New("command ID cannot be zero UUID")
|
||||
}
|
||||
|
||||
if cmd.AgentID == uuid.Nil {
|
||||
return errors.New("agent ID is required")
|
||||
}
|
||||
|
||||
if cmd.CommandType == "" {
|
||||
return errors.New("command type is required")
|
||||
}
|
||||
|
||||
if cmd.Status == "" {
|
||||
return errors.New("status is required")
|
||||
}
|
||||
|
||||
validStatuses := []string{"pending", "running", "completed", "failed", "cancelled"}
|
||||
if !contains(validStatuses, cmd.Status) {
|
||||
return fmt.Errorf("invalid status: %s", cmd.Status)
|
||||
}
|
||||
|
||||
if cmd.Source != "manual" && cmd.Source != "system" {
|
||||
return fmt.Errorf("source must be 'manual' or 'system', got: %s", cmd.Source)
|
||||
}
|
||||
|
||||
// Validate command type format
|
||||
if err := v.validateCommandType(cmd.CommandType); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// ValidateSubsystemAction validates subsystem-specific actions
|
||||
func (v *Validator) ValidateSubsystemAction(subsystem string, action string) error {
|
||||
validActions := map[string][]string{
|
||||
"storage": {"trigger", "enable", "disable", "set_interval"},
|
||||
"system": {"trigger", "enable", "disable", "set_interval"},
|
||||
"docker": {"trigger", "enable", "disable", "set_interval"},
|
||||
"updates": {"trigger", "enable", "disable", "set_interval"},
|
||||
}
|
||||
|
||||
actions, ok := validActions[subsystem]
|
||||
if !ok {
|
||||
return fmt.Errorf("unknown subsystem: %s", subsystem)
|
||||
}
|
||||
|
||||
if !contains(actions, action) {
|
||||
return fmt.Errorf("invalid action '%s' for subsystem '%s'", action, subsystem)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
// ValidateInterval ensures scanner intervals are within bounds
|
||||
func (v *Validator) ValidateInterval(subsystem string, minutes int) error {
|
||||
if minutes < v.minScannerMinutes {
|
||||
return fmt.Errorf("interval %d minutes below minimum %d for subsystem %s",
|
||||
minutes, v.minScannerMinutes, subsystem)
|
||||
}
|
||||
|
||||
if minutes > v.maxScannerMinutes {
|
||||
return fmt.Errorf("interval %d minutes above maximum %d for subsystem %s",
|
||||
minutes, v.maxScannerMinutes, subsystem)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (v *Validator) validateCommandType(commandType string) error {
|
||||
validPrefixes := []string{"scan_", "install_", "update_", "enable_", "disable_", "reboot"}
|
||||
|
||||
for _, prefix := range validPrefixes {
|
||||
if len(commandType) >= len(prefix) && commandType[:len(prefix)] == prefix {
|
||||
return nil
|
||||
}
|
||||
}
|
||||
|
||||
return fmt.Errorf("invalid command type format: %s", commandType)
|
||||
}
|
||||
|
||||
func contains(slice []string, item string) bool {
|
||||
for _, s := range slice {
|
||||
if s == item {
|
||||
return true
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
Loading…
Reference in a new issue