sunflower-seedmind/standalone.py at dev · emptyiscolor/sunflower-seedmind · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
# Run SeedGen as a standalone pipeline on oss-fuzz project with local source repo
# Usage: python3 standalone.py <project_name> <path_to_fuzz_tooling> <path_to_src_dir> [--mini]

import os
import argparse

from utils.task import TaskData
from infra.aixcc import (
    validate_environment,
    load_project_config,
    print_project_info,
    run_mini_mode,
    run_mcp_mode,
    run_full_mode
)


def parse_args():
    parser = argparse.ArgumentParser(
        description="Run SeedGen on an OSS-Fuzz project")
    parser.add_argument("project_name", type=str,
                        help="Name of the OSS-Fuzz project")
    parser.add_argument(
        "fuzz_tooling",
        type=str,
        help="Path to the fuzz tooling directory (oss-fuzz)",
    )
    parser.add_argument(
        "src_path",
        type=str,
        help="Path to the local project source directory",
    )
    parser.add_argument(
        "gen_model",
        type=str,
        help="model name",
    )
    parser.add_argument(
        "--mini",
        action="store_true",
        help="Run seedgen in mini mode",
    )
    parser.add_argument(
        "--mcp",
        action="store_true",
        help="Run seedgen in MCP mode",
    )
    return parser.parse_args()


def build_and_run_targets(project_name, src_path, fuzz_tooling, gen_model, mini=False, mcp_mode=False):
    os.makedirs(".tmp", exist_ok=True)

    project_yaml_path = validate_environment(fuzz_tooling, project_name)
    project_config = load_project_config(project_yaml_path)
    print_project_info(project_name, project_config)

    is_java = project_config["language"] in ["jvm", "java"]

    mock_task = TaskData(
        task_id="test-1234-5678",
        task_type="delta",
        project_name=project_name,
        focus="",
        repo=[],
        fuzz_tooling="",
        diff=""
    )

    if mcp_mode:
        run_mcp_mode(project_name, project_config,
                     src_path, fuzz_tooling, task=mock_task)
    elif is_java or mini:
        run_mini_mode(project_name, project_config,
                      src_path, fuzz_tooling, task=mock_task)
    else:
        run_full_mode(project_name, project_config,
                      src_path, fuzz_tooling, gen_model, task=mock_task)


def main():
    args = parse_args()
    project_name = args.project_name
    src_path = args.src_path
    fuzz_tooling = args.fuzz_tooling
    mini = args.mini
    mcp_mode = args.mcp
    gen_model = args.gen_model

    build_and_run_targets(project_name, src_path, fuzz_tooling, mini, mcp_mode)


if __name__ == "__main__":
    main()