diff --git a/README.md b/README.md index 7349f46..73c0b19 100644 --- a/README.md +++ b/README.md @@ -40,7 +40,7 @@ Attempts to identify the purpose of the current function and potential open sour Attempts to discover all potential vulnerabilities in the current function with the help of ChatGPT. The response from ChatGPT will be added as a comment above the function. **Beautify Function** (Tools > GhidraChatGPT > Beautify Function) -Attempts to beautify the current function with the help of ChatGPT. +Attempts to beautify the current function by identifying and applying more meaningful variable and function names with the help of ChatGPT. ### Settings **Update OpenAI Token** (Tools > GhidraChatGPT > Settings > Update OpenAI Token) diff --git a/assets/ghidrachatgpt.png b/assets/ghidrachatgpt.png index 9a8a5c6..0e635d5 100644 Binary files a/assets/ghidrachatgpt.png and b/assets/ghidrachatgpt.png differ diff --git a/ghidrachatgpt/src/main/java/ghidrachatgpt/GhidraChatGPTComponent.java b/ghidrachatgpt/src/main/java/ghidrachatgpt/GhidraChatGPTComponent.java index 118e8af..86ab46a 100644 --- a/ghidrachatgpt/src/main/java/ghidrachatgpt/GhidraChatGPTComponent.java +++ b/ghidrachatgpt/src/main/java/ghidrachatgpt/GhidraChatGPTComponent.java @@ -81,6 +81,7 @@ public void actionPerformed(ActionContext context) { action.setEnabled(true); action.setDescription("Identify the function with the help of ChatGPT"); action.setMenuBarData(new MenuData(new String[] { ToolConstants.MENU_TOOLS, FUNCTION_ID_NAME, "Identify Function" })); + action.setKeyBindingData(new KeyBindingData(KeyEvent.VK_I, InputEvent.SHIFT_DOWN_MASK | InputEvent.ALT_DOWN_MASK | InputEvent.CTRL_DOWN_MASK)); dockingTool.addAction(action); // Find vulnerabilities @@ -96,6 +97,7 @@ public void actionPerformed(ActionContext context) { action.setEnabled(true); action.setDescription("Find vulnerabilities in the function with the help of ChatGPT"); action.setMenuBarData(new MenuData(new String[] { ToolConstants.MENU_TOOLS, FUNCTION_ID_NAME, "Find Vulnerabilities" })); + action.setKeyBindingData(new KeyBindingData(KeyEvent.VK_V, InputEvent.SHIFT_DOWN_MASK | InputEvent.ALT_DOWN_MASK | InputEvent.CTRL_DOWN_MASK)); dockingTool.addAction(action); // Beautify function @@ -111,6 +113,7 @@ public void actionPerformed(ActionContext context) { action.setEnabled(true); action.setDescription("Beautify the function with the help of ChatGPT"); action.setMenuBarData(new MenuData(new String[] { ToolConstants.MENU_TOOLS, FUNCTION_ID_NAME, "Beautify Function" })); + action.setKeyBindingData(new KeyBindingData(KeyEvent.VK_B, InputEvent.SHIFT_DOWN_MASK | InputEvent.ALT_DOWN_MASK | InputEvent.CTRL_DOWN_MASK)); dockingTool.addAction(action); // Update OpenAI Token