Skip to content

Commit 89b2a8e

Browse files
author
dudcom
committed
saving fog tool call location revert to orginal
1 parent d38cac4 commit 89b2a8e

3 files changed

Lines changed: 329 additions & 54 deletions

File tree

‎Sources/Agentic_System/agents/EBG_crash.py‎

Lines changed: 59 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -93,9 +93,19 @@ def setup_agents(self, crash_program_hash: Optional[str] = None):
9393
prompt="Complete the delegated task.",
9494
system_prompt=v8_sys,
9595
tools=_tools(
96-
fuzzy_finder, ripgrep, tree, read_rag_db_id, write_rag_db_id, read_file, get_realpath,
97-
get_runtime_db_ids, get_cfg_for, get_call_graph_hashmap, find_functions_by_simple_name,
98-
find_functions_by_fully_qualified_name, get_call_graph_node,
96+
fuzzy_finder,
97+
ripgrep,
98+
tree,
99+
read_rag_db_id,
100+
write_rag_db_id,
101+
read_file,
102+
get_realpath,
103+
get_runtime_db_ids,
104+
get_cfg_for,
105+
get_call_graph_hashmap,
106+
find_functions_by_simple_name,
107+
find_functions_by_fully_qualified_name,
108+
get_call_graph_node,
99109
),
100110
model_id=WORKER_MODEL,
101111
api_key=self.api_key,
@@ -112,10 +122,20 @@ def setup_agents(self, crash_program_hash: Optional[str] = None):
112122
prompt="Complete the delegated task.",
113123
system_prompt=db_prompt,
114124
tools=_tools(
115-
base64_program_to_js, db_query, db_list_programs, db_get_fuzzer_performance_summary,
116-
db_list_fuzzers, db_get_crash_diversity, db_get_mutator_effectiveness, db_get_program_grouping,
117-
db_get_execution_outcome_distribution, read_from_generate_folder, write_to_generate_folder,
118-
delete_files_from_generate_folder, list_generate_folder, create_generate_folder,
125+
base64_program_to_js,
126+
db_query,
127+
db_list_programs,
128+
db_get_fuzzer_performance_summary,
129+
db_list_fuzzers,
130+
db_get_crash_diversity,
131+
db_get_mutator_effectiveness,
132+
db_get_program_grouping,
133+
db_get_execution_outcome_distribution,
134+
read_from_generate_folder,
135+
write_to_generate_folder,
136+
delete_files_from_generate_folder,
137+
list_generate_folder,
138+
create_generate_folder,
119139
),
120140
model_id=WORKER_MODEL,
121141
api_key=self.api_key,
@@ -128,10 +148,27 @@ def setup_agents(self, crash_program_hash: Optional[str] = None):
128148
prompt="Complete the delegated task.",
129149
system_prompt=self.get_prompt("debugger.txt"),
130150
tools=_tools(
131-
execute_javascript_program, list_d8_flags, list_v8_trace_options, trace_v8_analysis,
132-
get_program_js_from_hash, read_from_generate_folder, list_generate_folder,
133-
start_mi_debug_session, stop_mi_debug_session, mi_exec, mi_run, mi_step, mi_next, mi_continue,
134-
gdb_run_command, gdb_set_breakpoint, gdb_print_value, pwndbg_context, pwndbg_vmmap, pwndbg_regs, pwndbg_nearpc,
151+
execute_javascript_program,
152+
list_d8_flags,
153+
list_v8_trace_options,
154+
trace_v8_analysis,
155+
get_program_js_from_hash,
156+
read_from_generate_folder,
157+
list_generate_folder,
158+
start_mi_debug_session,
159+
stop_mi_debug_session,
160+
mi_exec,
161+
mi_run,
162+
mi_step,
163+
mi_next,
164+
mi_continue,
165+
gdb_run_command,
166+
gdb_set_breakpoint,
167+
gdb_print_value,
168+
pwndbg_context,
169+
pwndbg_vmmap,
170+
pwndbg_regs,
171+
pwndbg_nearpc,
135172
),
136173
model_id=WORKER_MODEL,
137174
api_key=self.api_key,
@@ -155,8 +192,11 @@ def setup_agents(self, crash_program_hash: Optional[str] = None):
155192
prompt="Complete the delegated task.",
156193
system_prompt=self.get_prompt("runtime_analyzer.txt"),
157194
tools=_tools(
158-
execute_javascript_program, list_d8_flags, list_v8_trace_options,
159-
read_from_generate_folder, list_generate_folder,
195+
execute_javascript_program,
196+
list_d8_flags,
197+
list_v8_trace_options,
198+
read_from_generate_folder,
199+
list_generate_folder,
160200
),
161201
model_id=MANAGER_MODEL,
162202
api_key=self.api_key,
@@ -170,8 +210,12 @@ def setup_agents(self, crash_program_hash: Optional[str] = None):
170210
prompt="Complete the delegated task.",
171211
system_prompt=self.get_prompt("variant_analysis.txt"),
172212
tools=_tools(
173-
execute_javascript_program, list_d8_flags, list_v8_trace_options, trace_v8_analysis,
174-
read_from_generate_folder, list_generate_folder,
213+
execute_javascript_program,
214+
list_d8_flags,
215+
list_v8_trace_options,
216+
trace_v8_analysis,
217+
read_from_generate_folder,
218+
list_generate_folder,
175219
),
176220
model_id=MANAGER_MODEL,
177221
api_key=self.api_key,

‎Sources/Agentic_System/agents/EBG_plateau.py‎

Lines changed: 98 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -95,9 +95,19 @@ def setup_agents(self, fuzzer_id: Optional[str] = None):
9595
prompt="Complete the delegated task.",
9696
system_prompt=v8_sys,
9797
tools=_tools(
98-
fuzzy_finder, ripgrep, tree, read_rag_db_id, write_rag_db_id, read_file, get_realpath,
99-
get_runtime_db_ids, get_cfg_for, get_call_graph_hashmap, find_functions_by_simple_name,
100-
find_functions_by_fully_qualified_name, get_call_graph_node,
98+
fuzzy_finder,
99+
ripgrep,
100+
tree,
101+
read_rag_db_id,
102+
write_rag_db_id,
103+
read_file,
104+
get_realpath,
105+
get_runtime_db_ids,
106+
get_cfg_for,
107+
get_call_graph_hashmap,
108+
find_functions_by_simple_name,
109+
find_functions_by_fully_qualified_name,
110+
get_call_graph_node,
101111
),
102112
model_id=WORKER_MODEL,
103113
api_key=self.api_key,
@@ -111,11 +121,35 @@ def setup_agents(self, fuzzer_id: Optional[str] = None):
111121
prompt="Complete the delegated task.",
112122
system_prompt=self.get_prompt("validator.txt"),
113123
tools=_tools(
114-
execute_javascript_program, list_d8_flags, list_v8_trace_options, trace_v8_analysis,
115-
write_to_generate_folder, read_from_generate_folder, list_generate_folder,
116-
start_mi_debug_session, stop_mi_debug_session, mi_exec, mi_run, mi_step, mi_next, mi_continue,
117-
gdb_run_command, gdb_set_breakpoint, gdb_print_value, pwndbg_context, pwndbg_vmmap, pwndbg_regs, pwndbg_nearpc,
118-
read_file, get_realpath, fuzzy_finder, ripgrep, tree, search_v8_source_rag, search_v8_source_rag_hybrid, get_v8_source_rag_doc,
124+
execute_javascript_program,
125+
list_d8_flags,
126+
list_v8_trace_options,
127+
trace_v8_analysis,
128+
write_to_generate_folder,
129+
read_from_generate_folder,
130+
list_generate_folder,
131+
start_mi_debug_session,
132+
stop_mi_debug_session,
133+
mi_exec,
134+
mi_run,
135+
mi_step,
136+
mi_next,
137+
mi_continue,
138+
gdb_run_command,
139+
gdb_set_breakpoint,
140+
gdb_print_value,
141+
pwndbg_context,
142+
pwndbg_vmmap,
143+
pwndbg_regs,
144+
pwndbg_nearpc,
145+
read_file,
146+
get_realpath,
147+
fuzzy_finder,
148+
ripgrep,
149+
tree,
150+
search_v8_source_rag,
151+
search_v8_source_rag_hybrid,
152+
get_v8_source_rag_doc,
119153
),
120154
model_id=WORKER_MODEL,
121155
api_key=self.api_key,
@@ -133,10 +167,19 @@ def setup_agents(self, fuzzer_id: Optional[str] = None):
133167
prompt="Complete the delegated task.",
134168
system_prompt=db_prompt,
135169
tools=_tools(
136-
base64_program_to_js, db_query, db_list_programs, db_get_fuzzer_performance_summary,
137-
db_list_fuzzers, db_get_crash_diversity, db_get_mutator_effectiveness, db_get_program_grouping,
138-
db_get_execution_outcome_distribution, read_from_generate_folder, write_to_generate_folder,
139-
delete_files_from_generate_folder, list_generate_folder,
170+
base64_program_to_js,
171+
db_query,
172+
db_list_programs,
173+
db_get_fuzzer_performance_summary,
174+
db_list_fuzzers,
175+
db_get_crash_diversity,
176+
db_get_mutator_effectiveness,
177+
db_get_program_grouping,
178+
db_get_execution_outcome_distribution,
179+
read_from_generate_folder,
180+
write_to_generate_folder,
181+
delete_files_from_generate_folder,
182+
list_generate_folder,
140183
),
141184
model_id=WORKER_MODEL,
142185
api_key=self.api_key,
@@ -150,10 +193,28 @@ def setup_agents(self, fuzzer_id: Optional[str] = None):
150193
prompt="Complete the delegated task.",
151194
system_prompt=self.get_prompt("debugger.txt"),
152195
tools=_tools(
153-
execute_javascript_program, list_d8_flags, list_v8_trace_options, trace_v8_analysis,
154-
get_program_js_from_hash, read_from_generate_folder, write_to_generate_folder, list_generate_folder,
155-
start_mi_debug_session, stop_mi_debug_session, mi_exec, mi_run, mi_step, mi_next, mi_continue,
156-
gdb_run_command, gdb_set_breakpoint, gdb_print_value, pwndbg_context, pwndbg_vmmap, pwndbg_regs, pwndbg_nearpc,
196+
execute_javascript_program,
197+
list_d8_flags,
198+
list_v8_trace_options,
199+
trace_v8_analysis,
200+
get_program_js_from_hash,
201+
read_from_generate_folder,
202+
write_to_generate_folder,
203+
list_generate_folder,
204+
start_mi_debug_session,
205+
stop_mi_debug_session,
206+
mi_exec,
207+
mi_run,
208+
mi_step,
209+
mi_next,
210+
mi_continue,
211+
gdb_run_command,
212+
gdb_set_breakpoint,
213+
gdb_print_value,
214+
pwndbg_context,
215+
pwndbg_vmmap,
216+
pwndbg_regs,
217+
pwndbg_nearpc,
157218
),
158219
model_id=WORKER_MODEL,
159220
api_key=self.api_key,
@@ -167,9 +228,17 @@ def setup_agents(self, fuzzer_id: Optional[str] = None):
167228
prompt="Complete the delegated task.",
168229
system_prompt=self.get_prompt("JS_generator.txt"),
169230
tools=_tools(
170-
db_store_generated_program, db_list_programs, db_query, execute_javascript_program,
171-
list_d8_flags, list_v8_trace_options, trace_v8_analysis, get_program_js_from_hash,
172-
read_from_generate_folder, list_generate_folder, write_and_execute_js,
231+
db_store_generated_program,
232+
db_list_programs,
233+
db_query,
234+
execute_javascript_program,
235+
list_d8_flags,
236+
list_v8_trace_options,
237+
trace_v8_analysis,
238+
get_program_js_from_hash,
239+
read_from_generate_folder,
240+
list_generate_folder,
241+
write_and_execute_js,
173242
),
174243
model_id=MANAGER_MODEL,
175244
api_key=self.api_key,
@@ -184,9 +253,16 @@ def setup_agents(self, fuzzer_id: Optional[str] = None):
184253
prompt="Complete the delegated task.",
185254
system_prompt=self.get_prompt("runtime_analyzer.txt"),
186255
tools=_tools(
187-
execute_javascript_program, list_d8_flags, list_v8_trace_options, trace_v8_analysis,
188-
get_program_js_from_hash, read_from_generate_folder, list_generate_folder,
189-
search_v8_source_rag, search_v8_source_rag_hybrid, get_v8_source_rag_doc,
256+
execute_javascript_program,
257+
list_d8_flags,
258+
list_v8_trace_options,
259+
trace_v8_analysis,
260+
get_program_js_from_hash,
261+
read_from_generate_folder,
262+
list_generate_folder,
263+
search_v8_source_rag,
264+
search_v8_source_rag_hybrid,
265+
get_v8_source_rag_doc,
190266
),
191267
model_id=MANAGER_MODEL,
192268
api_key=self.api_key,

0 commit comments

Comments
 (0)