pkg/aflow: support setting a custom repository

For situations where the user wants to reproduce bugs against a
different repository than mainline.

Author: FlorentRevest

pkg/aflow/flow/patching/base_commit.go

@@ -14,6 +14,7 @@ var baseCommitPicker = aflow.NewFuncAction("base-commit-picker", pickBaseCommit)
 type baseCommitArgs struct {
 	// Can be used to override the selected base commit (for manual testing).
 	FixedBaseCommit string
+	Repository      string
 }
 
 type baseCommitResult struct {
@@ -41,6 +42,9 @@ func pickBaseCommit(ctx *aflow.Context, args baseCommitArgs) (baseCommitResult,
 		KernelRepo:   baseRepo,
 		KernelCommit: args.FixedBaseCommit,
 	}
+	if args.Repository != "" {
+		res.KernelRepo = args.Repository
+	}
 	if args.FixedBaseCommit != "" {
 		return res, nil
 	}

pkg/aflow/flow/patching/patching.go

@@ -29,6 +29,7 @@ type Inputs struct {
 
 	// Use this fixed based kernel commit (for testing/local running).
 	FixedBaseCommit string
+	Repository      string
 }
 
 type Outputs struct {

syz-agent/agent.go

@@ -47,6 +47,8 @@ type Config struct {
 	CacheSize uint64 `json:"cache_size"`
 	// Use fixed base commit for patching jobs (for testing).
 	FixedBaseCommit string `json:"fixed_base_commit"`
+	// Use a different repo than torvalds's mainline.
+	Repository string `json:"repo"`
 	// Use this LLM model (for testing, if empty use workflow-default model).
 	Model string `json:"model"`
 }
@@ -221,6 +223,7 @@ func (s *Server) executeJob(ctx context.Context, req *dashapi.AIJobPollResp) (ma
 		"Type":              s.cfg.Type,
 		"VM":                s.cfg.VM,
 		"FixedBaseCommit":   s.cfg.FixedBaseCommit,
+		"Repository":        s.cfg.Repository,
 	}
 	maps.Insert(inputs, maps.All(req.Args))
 	onEvent := func(span *trajectory.Span) error {