run_exploit_simulations.go

package main

import (
	"context"
	"encoding/json"
	"fmt"
	"os"
	"time"

	"github.com/aethelred/aethelred/tools/tools/loadtest"
)

type ExploitSimulationOutput struct {
	GeneratedAt string        `json:"generated_at"`
	Scenarios   []interface{} `json:"scenarios"`
}

func main() {
	scenarios := loadtest.GetNetworkScenarios()
	var results []interface{}

	validators := make([]*loadtest.SimulatedValidator, 100)
	for i := 0; i < 100; i++ {
		validators[i] = &loadtest.SimulatedValidator{
			ID:          i,
			VotingPower: 1000,
			IsActive:    true,
		}
	}

	for _, s := range scenarios {
		fmt.Printf("Running network scenario: %s\n", s.Name)
		runner := loadtest.NewNetworkScenarioRunner(&s, validators, nil)
		res, err := runner.Run(context.Background())
		if err != nil {
			fmt.Printf("Scenario %s failed: %v\n", s.Name, err)
			continue
		}

		results = append(results, map[string]interface{}{
			"name":                    res.ScenarioName,
			"description":             s.Description,
			"attack_resilience_grade": res.AttackResilienceGrade,
			"recovery_grade":          res.RecoveryGrade,
			"overall_grade":           res.OverallGrade,
			"consensus_failures":      res.ConsensusFailures,
			"split_brain_events":      res.SplitBrainEvents,
			"recommendations":         res.Recommendations,
			"duration_ms":             res.Duration.Milliseconds(),
		})
	}

	out := ExploitSimulationOutput{
		GeneratedAt: time.Now().UTC().Format(time.RFC3339),
		Scenarios:   results,
	}

	data, _ := json.MarshalIndent(out, "", "  ")
	if err := os.MkdirAll("test-results", 0755); err != nil {
		fmt.Printf("Failed to create test-results directory: %v\n", err)
		return
	}
	if err := os.WriteFile("test-results/dynamic-exploit-simulations.json", data, 0644); err != nil {
		fmt.Printf("Failed to write results: %v\n", err)
		return
	}
	fmt.Println("Exploit simulations finished. Results written to test-results/dynamic-exploit-simulations.json")
}