Change variable names to allow for modularity.

Given the addition of renaming expressions, it didn't make sense to just have one prompt. This has been changed to be more meaningful.

Author: WhatTheFuzz

src/agent.py

@@ -9,7 +9,7 @@
 from binaryninja.function import Function
 from binaryninja.lowlevelil import LowLevelILFunction
 from binaryninja.mediumlevelil import MediumLevelILFunction
-from binaryninja.highlevelil import HighLevelILFunction
+from binaryninja.highlevelil import HighLevelILFunction, HighLevelILInstruction
 from binaryninja.settings import Settings
 from binaryninja import log, BinaryView
 
@@ -19,11 +19,16 @@
 
 class Agent:
 
- question: str = '''
+ function_question: str = '''
  This is a function that was decompiled with Binary Ninja.
  It is in IL_FORM. What does this function do?
  '''
 
+ rename_variable_question: str = '''
+ In one word, what should the variable name be for the variable that is
+ assigned to the result of the C expression:\n
+ '''
+
  # A mapping of IL forms to their names.
  il_name: dict[type, str] = {
  LowLevelILFunction: 'Low Level Intermediate Language',
@@ -140,18 +145,25 @@ def generate_query(self, function: Union[Function,
  LowLevelILFunction,
  MediumLevelILFunction,
  HighLevelILFunction]) -> str:
- '''Generates a query string given a BNIL function. Reads the file
- prompt.txt and replaces the IL form with the name of the IL form.
+ '''Generates a query string given a BNIL function. Returns the query as
+ a string.
  '''
- prompt: str = self.question
+ prompt: str = self.function_question
  # Read the prompt from the text file.
- prompt = self.question.replace('IL_FORM', self.il_name[type(function)])
+ prompt = prompt.replace('IL_FORM', self.il_name[type(function)])
  # Add some new lines. Maybe not necessary.
  prompt += '\n\n'
  # Add the instructions to the prompt.
  prompt += '\n'.join(self.instruction_list(function))
  return prompt
 
+ def generate_rename_expression_query(
+ instruction: HighLevelILInstruction) -> str:
+ '''Generates a query string given a BNIL instruction. Returns the query
+ as a string.
+ '''
+ pass
+
  def send_query(self, query: str) -> None:
  '''Sends a query to the engine and prints the response.'''
  query = Query(query_string=query,

src/entry.py

@@ -14,4 +14,8 @@ def check_function(bv: BinaryView, func: Function) -> bool:
  agent.send_query(query)
 
 def rename_expression(bv: BinaryView, instruction: HighLevelILInstruction) -> bool:
- pass
+ agent: Agent = Agent(
+ bv=bv,
+ path_to_api_key=API_KEY_PATH
+ )
+ query: str = agent.generate_rename_expression_query(instruction)