feat: Improve data manipulation API
- Add comprehensive data manipulation capabilities - Implement separate pathways for rename-only, type-only, and combined operations - Fix HTTP request body consumption issue in DataEndpoints - Standardize on 'type' parameter name instead of 'dataType' - Add thorough test coverage with dedicated test_data_update.py script - Update API documentation to version 2 with full endpoint descriptions - Update CHANGELOG with detailed information about data manipulation features
This commit is contained in:
Teal Bauer
parent
5797fb38e7
commit
28870e9b54
12
CHANGELOG.md
12
CHANGELOG.md
@ -27,6 +27,12 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.0/).
|
|||||||
- Implemented proper resource linking with HATEOAS
|
- Implemented proper resource linking with HATEOAS
|
||||||
- Added disassembly endpoint for functions with HATEOAS links
|
- Added disassembly endpoint for functions with HATEOAS links
|
||||||
- Enhanced parameter validation in MCP bridge tools
|
- Enhanced parameter validation in MCP bridge tools
|
||||||
|
- Added comprehensive data manipulation capabilities:
|
||||||
|
- Data renaming (changing only the name)
|
||||||
|
- Data type setting (changing only the type)
|
||||||
|
- Combined data update operations (changing both name and type)
|
||||||
|
- Dedicated `/data/type` and `/data/update` endpoints
|
||||||
|
- Standalone test script for data operations
|
||||||
|
|
||||||
### Changed
|
### Changed
|
||||||
- Unified all endpoints to use structured JSON
|
- Unified all endpoints to use structured JSON
|
||||||
@ -46,6 +52,7 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.0/).
|
|||||||
- All responses must include _links object with at least self reference
|
- All responses must include _links object with at least self reference
|
||||||
- Standardized JSON structures for all resource types
|
- Standardized JSON structures for all resource types
|
||||||
- Created comprehensive requirements documentation in HATEOAS_API.md
|
- Created comprehensive requirements documentation in HATEOAS_API.md
|
||||||
|
- Updated API documentation to version 2 with comprehensive endpoint descriptions
|
||||||
|
|
||||||
### Fixed
|
### Fixed
|
||||||
- Fixed endpoint registration in refactored code (all endpoints now working)
|
- Fixed endpoint registration in refactored code (all endpoints now working)
|
||||||
@ -62,6 +69,11 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.1.0/).
|
|||||||
- Fixed decompile endpoint to return structured decompiled code
|
- Fixed decompile endpoint to return structured decompiled code
|
||||||
- Fixed disassembly endpoint to return structured instruction list
|
- Fixed disassembly endpoint to return structured instruction list
|
||||||
- Fixed variables endpoint to return proper variable structure
|
- Fixed variables endpoint to return proper variable structure
|
||||||
|
- Fixed data manipulation operations:
|
||||||
|
- Resolved HTTP request body consumption issue
|
||||||
|
- Fixed parameter naming inconsistency between "dataType" and "type"
|
||||||
|
- Improved preservation of names during data type changes
|
||||||
|
- Enhanced error handling for data operations
|
||||||
|
|
||||||
## [1.4.0] - 2025-04-08
|
## [1.4.0] - 2025-04-08
|
||||||
|
|
||||||
|
|||||||
@ -1,8 +1,8 @@
|
|||||||
# GhydraMCP Ghidra Plugin HTTP API v1
|
# GhydraMCP Ghidra Plugin HTTP API v2
|
||||||
|
|
||||||
## Overview
|
## Overview
|
||||||
|
|
||||||
This API provides a Hypermedia-driven interface (HATEOAS) to interact with Ghidra's CodeBrowser, enabling AI-driven and automated reverse engineering workflows. It allows interaction with Ghidra projects, programs (binaries), functions, symbols, data, memory segments, cross-references, and analysis features. Each program open in Ghidra will have its own instance, so all resources are specific to that program.
|
This API provides a Hypermedia-driven interface (HATEOAS) to interact with Ghidra's CodeBrowser, enabling AI-driven and automated reverse engineering workflows. It allows interaction with Ghidra projects, programs (binaries), functions, symbols, data, memory segments, cross-references, and analysis features. Each program open in Ghidra will have its own plugin instance, so all resources are specific to that program.
|
||||||
|
|
||||||
## General Concepts
|
## General Concepts
|
||||||
|
|
||||||
@ -48,7 +48,7 @@ List results (arrays in `result`) will typically include pagination information
|
|||||||
```json
|
```json
|
||||||
{
|
{
|
||||||
"id": "req-123",
|
"id": "req-123",
|
||||||
"instance": "http://localhost:1337",
|
"instance": "http://localhost:8192",
|
||||||
"success": true,
|
"success": true,
|
||||||
"result": [ ... objects ... ],
|
"result": [ ... objects ... ],
|
||||||
"size": 150, // Total number of items matching the query across all pages
|
"size": 150, // Total number of items matching the query across all pages
|
||||||
@ -115,45 +115,117 @@ Returns the version of the running Ghidra plugin and its API. Essential for comp
|
|||||||
```json
|
```json
|
||||||
{
|
{
|
||||||
"id": "req-meta-ver",
|
"id": "req-meta-ver",
|
||||||
"instance": "http://localhost:1337",
|
"instance": "http://localhost:8192",
|
||||||
"success": true,
|
"success": true,
|
||||||
"result": {
|
"result": {
|
||||||
"plugin_version": "v2.0.0", // Example plugin build version
|
"plugin_version": "v2.0.0", // Example plugin build version
|
||||||
"api_version": 2 // Ordinal API version
|
"api_version": 2 // Ordinal API version
|
||||||
},
|
},
|
||||||
"_links": {
|
"_links": {
|
||||||
"self": { "href": "/plugin-version" }
|
"self": { "href": "/plugin-version" },
|
||||||
|
"root": { "href": "/" }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### `GET /info`
|
||||||
|
Returns information about the current plugin instance, including details about the loaded program and project.
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"id": "req-info",
|
||||||
|
"instance": "http://localhost:8192",
|
||||||
|
"success": true,
|
||||||
|
"result": {
|
||||||
|
"isBaseInstance": true,
|
||||||
|
"file": "example.exe",
|
||||||
|
"architecture": "x86:LE:64:default",
|
||||||
|
"processor": "x86",
|
||||||
|
"addressSize": 64,
|
||||||
|
"creationDate": "2023-01-01T12:00:00Z",
|
||||||
|
"executable": "/path/to/example.exe",
|
||||||
|
"project": "MyProject",
|
||||||
|
"projectLocation": "/path/to/MyProject",
|
||||||
|
"serverPort": 8192,
|
||||||
|
"serverStartTime": 1672531200000,
|
||||||
|
"instanceCount": 1
|
||||||
|
},
|
||||||
|
"_links": {
|
||||||
|
"self": { "href": "/info" },
|
||||||
|
"root": { "href": "/" },
|
||||||
|
"instances": { "href": "/instances" },
|
||||||
|
"program": { "href": "/program" }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### `GET /instances`
|
||||||
|
Returns information about all active GhydraMCP plugin instances.
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"id": "req-instances",
|
||||||
|
"instance": "http://localhost:8192",
|
||||||
|
"success": true,
|
||||||
|
"result": [
|
||||||
|
{
|
||||||
|
"port": 8192,
|
||||||
|
"url": "http://localhost:8192",
|
||||||
|
"type": "base",
|
||||||
|
"project": "MyProject",
|
||||||
|
"file": "example.exe",
|
||||||
|
"_links": {
|
||||||
|
"self": { "href": "/instances/8192" },
|
||||||
|
"info": { "href": "http://localhost:8192/info" },
|
||||||
|
"connect": { "href": "http://localhost:8192" }
|
||||||
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"port": 8193,
|
||||||
|
"url": "http://localhost:8193",
|
||||||
|
"type": "standard",
|
||||||
|
"project": "MyProject",
|
||||||
|
"file": "library.dll",
|
||||||
|
"_links": {
|
||||||
|
"self": { "href": "/instances/8193" },
|
||||||
|
"info": { "href": "http://localhost:8193/info" },
|
||||||
|
"connect": { "href": "http://localhost:8193" }
|
||||||
|
}
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"_links": {
|
||||||
|
"self": { "href": "/instances" },
|
||||||
|
"register": { "href": "/registerInstance", "method": "POST" },
|
||||||
|
"unregister": { "href": "/unregisterInstance", "method": "POST" },
|
||||||
|
"programs": { "href": "/programs" }
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
```
|
```
|
||||||
|
|
||||||
## Resource Types
|
## Resource Types
|
||||||
|
|
||||||
Each Ghidra plugin instance runs in the context of a single program, so all resources are relative to the current program. The program's details are available through the `GET /info` and `GET /programs/current` endpoints.
|
Each Ghidra plugin instance runs in the context of a single program, so all resources are relative to the current program. The program's details are available through the `GET /info` and `GET /program` endpoints.
|
||||||
|
|
||||||
### 1. Projects
|
### 1. Project
|
||||||
|
|
||||||
Represents Ghidra projects, containers for programs.
|
Represents the current Ghidra project, which is a container for programs.
|
||||||
|
|
||||||
- **`GET /project`**: Get details about the current project (e.g., location, list of open programs within it via links).
|
- **`GET /project`**: Get details about the current project (e.g., location, list of open programs within it via links).
|
||||||
|
|
||||||
### 2. Programs
|
### 2. Program
|
||||||
|
|
||||||
Represents individual binaries loaded in Ghidra projects.
|
Represents the current binary loaded in Ghidra.
|
||||||
|
|
||||||
- **`GET /program`**: Get metadata for the current program (e.g., name, architecture, memory layout, analysis status).
|
- **`GET /program`**: Get metadata for the current program (e.g., name, architecture, memory layout, analysis status).
|
||||||
```json
|
```json
|
||||||
// Example Response Fragment for GET /program
|
// Example Response Fragment for GET /program
|
||||||
"result": {
|
"result": {
|
||||||
|
"programId": "myproject:/path/to/mybinary.exe",
|
||||||
"name": "mybinary.exe",
|
"name": "mybinary.exe",
|
||||||
"project": "myproject",
|
"isOpen": true,
|
||||||
"language_id": "x86:LE:64:default",
|
"languageId": "x86:LE:64:default",
|
||||||
"compiler_spec_id": "gcc",
|
"compilerSpecId": "gcc",
|
||||||
"image_base": "0x400000",
|
"imageBase": "0x400000",
|
||||||
"memory_size": 1048576,
|
"memorySize": 1048576,
|
||||||
"is_open": true,
|
"analysisComplete": true
|
||||||
"analysis_complete": true
|
|
||||||
// ... other metadata
|
|
||||||
},
|
},
|
||||||
"_links": {
|
"_links": {
|
||||||
"self": { "href": "/program" },
|
"self": { "href": "/program" },
|
||||||
@ -168,7 +240,47 @@ Represents individual binaries loaded in Ghidra projects.
|
|||||||
}
|
}
|
||||||
```
|
```
|
||||||
|
|
||||||
### 3. Functions
|
### 3. Current Location
|
||||||
|
|
||||||
|
Provides information about the current cursor position and function in Ghidra's CodeBrowser.
|
||||||
|
|
||||||
|
- **`GET /address`**: Get the current cursor position.
|
||||||
|
```json
|
||||||
|
// Example Response
|
||||||
|
"result": {
|
||||||
|
"address": "0x401000",
|
||||||
|
"program": "mybinary.exe"
|
||||||
|
},
|
||||||
|
"_links": {
|
||||||
|
"self": { "href": "/address" },
|
||||||
|
"program": { "href": "/program" },
|
||||||
|
"memory": { "href": "/memory/0x401000?length=16" },
|
||||||
|
"function": { "href": "/functions/0x401000" },
|
||||||
|
"decompile": { "href": "/functions/0x401000/decompile" }
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
- **`GET /function`**: Get information about the function at the current cursor position.
|
||||||
|
```json
|
||||||
|
// Example Response
|
||||||
|
"result": {
|
||||||
|
"name": "main",
|
||||||
|
"address": "0x401000",
|
||||||
|
"signature": "int main(int argc, char** argv)",
|
||||||
|
"size": 256
|
||||||
|
},
|
||||||
|
"_links": {
|
||||||
|
"self": { "href": "/function" },
|
||||||
|
"program": { "href": "/program" },
|
||||||
|
"function": { "href": "/functions/0x401000" },
|
||||||
|
"decompile": { "href": "/functions/0x401000/decompile" },
|
||||||
|
"disassembly": { "href": "/functions/0x401000/disassembly" },
|
||||||
|
"variables": { "href": "/functions/0x401000/variables" },
|
||||||
|
"xrefs": { "href": "/xrefs?to_addr=0x401000" }
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### 4. Functions
|
||||||
|
|
||||||
Represents functions within the current program.
|
Represents functions within the current program.
|
||||||
|
|
||||||
@ -239,7 +351,7 @@ Represents functions within the current program.
|
|||||||
- **`GET /functions/{address}/variables`**: List local variables defined within the function. Supports searching by name.
|
- **`GET /functions/{address}/variables`**: List local variables defined within the function. Supports searching by name.
|
||||||
- **`PATCH /functions/{address}/variables/{variable_name}`**: Modify a local variable (rename, change type). Requires `name` and/or `type` in the payload.
|
- **`PATCH /functions/{address}/variables/{variable_name}`**: Modify a local variable (rename, change type). Requires `name` and/or `type` in the payload.
|
||||||
|
|
||||||
### 4. Symbols & Labels
|
### 5. Symbols & Labels
|
||||||
|
|
||||||
Represents named locations (functions, data, labels).
|
Represents named locations (functions, data, labels).
|
||||||
|
|
||||||
@ -249,7 +361,7 @@ Represents named locations (functions, data, labels).
|
|||||||
- **`PATCH /symbols/{address}`**: Modify properties of the symbol (e.g., set as primary, change namespace). Payload specifies changes.
|
- **`PATCH /symbols/{address}`**: Modify properties of the symbol (e.g., set as primary, change namespace). Payload specifies changes.
|
||||||
- **`DELETE /symbols/{address}`**: Remove the symbol at the specified address.
|
- **`DELETE /symbols/{address}`**: Remove the symbol at the specified address.
|
||||||
|
|
||||||
### 5. Data
|
### 6. Data
|
||||||
|
|
||||||
Represents defined data items in memory.
|
Represents defined data items in memory.
|
||||||
|
|
||||||
@ -259,14 +371,14 @@ Represents defined data items in memory.
|
|||||||
- **`PATCH /data/{address}`**: Modify a data item (e.g., change `name`, `type`, `comment`). Payload specifies changes.
|
- **`PATCH /data/{address}`**: Modify a data item (e.g., change `name`, `type`, `comment`). Payload specifies changes.
|
||||||
- **`DELETE /data/{address}`**: Undefine the data item at the specified address.
|
- **`DELETE /data/{address}`**: Undefine the data item at the specified address.
|
||||||
|
|
||||||
### 6. Memory Segments
|
### 7. Memory Segments
|
||||||
|
|
||||||
Represents memory blocks/sections defined in the program.
|
Represents memory blocks/sections defined in the program.
|
||||||
|
|
||||||
- **`GET /segments`**: List all memory segments (e.g., `.text`, `.data`, `.bss`).
|
- **`GET /segments`**: List all memory segments (e.g., `.text`, `.data`, `.bss`).
|
||||||
- **`GET /segments/{segment_name}`**: Get details for a specific segment (address range, permissions, size).
|
- **`GET /segments/{segment_name}`**: Get details for a specific segment (address range, permissions, size).
|
||||||
|
|
||||||
### 7. Memory Access
|
### 8. Memory Access
|
||||||
|
|
||||||
Provides raw memory access.
|
Provides raw memory access.
|
||||||
|
|
||||||
@ -285,7 +397,7 @@ Provides raw memory access.
|
|||||||
```
|
```
|
||||||
- **`PATCH /memory/{address}`**: Write bytes to memory. Requires `bytes` (in specified `format`) and `format` in the payload. Use with extreme caution.
|
- **`PATCH /memory/{address}`**: Write bytes to memory. Requires `bytes` (in specified `format`) and `format` in the payload. Use with extreme caution.
|
||||||
|
|
||||||
### 8. Cross-References (XRefs)
|
### 9. Cross-References (XRefs)
|
||||||
|
|
||||||
Provides information about references to/from addresses.
|
Provides information about references to/from addresses.
|
||||||
|
|
||||||
@ -296,13 +408,99 @@ Provides information about references to/from addresses.
|
|||||||
- `?type=[CALL|READ|WRITE|DATA|POINTER|...]`: Filter by reference type.
|
- `?type=[CALL|READ|WRITE|DATA|POINTER|...]`: Filter by reference type.
|
||||||
- **`GET /functions/{address}/xrefs`**: Convenience endpoint, equivalent to `GET /xrefs?to_addr={address}` and potentially `GET /xrefs?from_addr={address}` combined or linked.
|
- **`GET /functions/{address}/xrefs`**: Convenience endpoint, equivalent to `GET /xrefs?to_addr={address}` and potentially `GET /xrefs?from_addr={address}` combined or linked.
|
||||||
|
|
||||||
### 9. Analysis
|
### 10. Analysis
|
||||||
|
|
||||||
Provides access to Ghidra's analysis results.
|
Provides access to Ghidra's analysis results.
|
||||||
|
|
||||||
- **`GET /analysis/callgraph`**: Retrieve the function call graph (potentially filtered or paginated). Format might be nodes/edges JSON or a standard graph format like DOT.
|
- **`GET /analysis`**: Get information about the analysis status and available analyzers.
|
||||||
- **`GET /analysis/dataflow/{address}`**: Perform data flow analysis starting from a specific address or instruction. Requires parameters specifying forward/backward, context, etc. (Details TBD).
|
```json
|
||||||
- **`POST /analysis/analyze`**: Trigger a full or partial re-analysis of the program.
|
// Example Response
|
||||||
|
"result": {
|
||||||
|
"program": "mybinary.exe",
|
||||||
|
"analysis_enabled": true,
|
||||||
|
"available_analyzers": [
|
||||||
|
"Function Start Analyzer",
|
||||||
|
"Basic Block Model Analyzer",
|
||||||
|
"Reference Analyzer",
|
||||||
|
"Call Convention Analyzer",
|
||||||
|
"Data Reference Analyzer",
|
||||||
|
"Decompiler Parameter ID",
|
||||||
|
"Stack Analyzer"
|
||||||
|
]
|
||||||
|
},
|
||||||
|
"_links": {
|
||||||
|
"self": { "href": "/analysis" },
|
||||||
|
"program": { "href": "/program" },
|
||||||
|
"analyze": { "href": "/analysis", "method": "POST" },
|
||||||
|
"callgraph": { "href": "/analysis/callgraph" }
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
- **`POST /analysis`**: Trigger a full or partial re-analysis of the program.
|
||||||
|
```json
|
||||||
|
// Example Response
|
||||||
|
"result": {
|
||||||
|
"program": "mybinary.exe",
|
||||||
|
"analysis_triggered": true,
|
||||||
|
"message": "Analysis initiated on program"
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
- **`GET /analysis/callgraph`**: Retrieve the function call graph.
|
||||||
|
- Query Parameters:
|
||||||
|
- `?function=[function_name]`: Start the call graph from this function (default: entry point).
|
||||||
|
- `?max_depth=[int]`: Maximum depth of the call graph (default: 3).
|
||||||
|
```json
|
||||||
|
// Example Response
|
||||||
|
"result": {
|
||||||
|
"root": "main",
|
||||||
|
"root_address": "0x401000",
|
||||||
|
"max_depth": 3,
|
||||||
|
"nodes": [
|
||||||
|
{
|
||||||
|
"id": "0x401000",
|
||||||
|
"name": "main",
|
||||||
|
"address": "0x401000",
|
||||||
|
"depth": 0,
|
||||||
|
"_links": {
|
||||||
|
"self": { "href": "/functions/0x401000" }
|
||||||
|
}
|
||||||
|
},
|
||||||
|
// ... more nodes
|
||||||
|
],
|
||||||
|
"edges": [
|
||||||
|
{
|
||||||
|
"from": "0x401000",
|
||||||
|
"to": "0x401100",
|
||||||
|
"type": "call",
|
||||||
|
"call_site": "0x401050"
|
||||||
|
},
|
||||||
|
// ... more edges
|
||||||
|
]
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
- **`GET /analysis/dataflow`**: Perform data flow analysis starting from a specific address.
|
||||||
|
- Query Parameters:
|
||||||
|
- `?address=[address]`: Starting address for data flow analysis (required).
|
||||||
|
- `?direction=[forward|backward]`: Direction of data flow analysis (default: forward).
|
||||||
|
- `?max_steps=[int]`: Maximum number of steps to analyze (default: 50).
|
||||||
|
```json
|
||||||
|
// Example Response
|
||||||
|
"result": {
|
||||||
|
"start_address": "0x401050",
|
||||||
|
"direction": "forward",
|
||||||
|
"max_steps": 50,
|
||||||
|
"steps": [
|
||||||
|
{
|
||||||
|
"address": "0x401050",
|
||||||
|
"instruction": "MOV EAX, [RBP+0x8]",
|
||||||
|
"description": "Starting point of data flow analysis"
|
||||||
|
},
|
||||||
|
// ... more steps
|
||||||
|
]
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
## Design Considerations for AI Usage
|
## Design Considerations for AI Usage
|
||||||
|
|
||||||
|
|||||||
@ -1607,33 +1607,23 @@ def update_data(port: int = DEFAULT_GHIDRA_PORT,
|
|||||||
payload["newName"] = name
|
payload["newName"] = name
|
||||||
|
|
||||||
if data_type:
|
if data_type:
|
||||||
payload["dataType"] = data_type
|
payload["type"] = data_type
|
||||||
|
|
||||||
# Handle the cases separately for maximum reliability
|
# Handle different cases for maximum reliability
|
||||||
if name and data_type is None:
|
if name and data_type is None:
|
||||||
# If only renaming, use the existing data endpoint that's already tested
|
# If only renaming, use the main data endpoint
|
||||||
name_payload = {"address": address, "newName": name}
|
response = safe_post(port, "data", payload)
|
||||||
response = safe_post(port, "data", name_payload)
|
|
||||||
return simplify_response(response)
|
return simplify_response(response)
|
||||||
|
|
||||||
if data_type and name is None:
|
if data_type and name is None:
|
||||||
# If only changing type, use the data/type endpoint
|
# If only changing type, use the data/type endpoint
|
||||||
type_payload = {"address": address, "dataType": data_type}
|
response = safe_post(port, "data/type", payload)
|
||||||
response = safe_post(port, "data/type", type_payload)
|
|
||||||
return simplify_response(response)
|
return simplify_response(response)
|
||||||
|
|
||||||
# If both, handle sequentially (rename first, then type)
|
|
||||||
if name and data_type:
|
if name and data_type:
|
||||||
# First rename
|
# If both name and type, use the data/update endpoint
|
||||||
name_payload = {"address": address, "newName": name}
|
response = safe_post(port, "data/update", payload)
|
||||||
rename_response = safe_post(port, "data", name_payload)
|
return simplify_response(response)
|
||||||
|
|
||||||
# Then set type
|
|
||||||
type_payload = {"address": address, "dataType": data_type}
|
|
||||||
type_response = safe_post(port, "data/type", type_payload)
|
|
||||||
|
|
||||||
# Return the most recent response which should include updated info
|
|
||||||
return simplify_response(type_response)
|
|
||||||
|
|
||||||
# This shouldn't be reached due to earlier checks
|
# This shouldn't be reached due to earlier checks
|
||||||
return {
|
return {
|
||||||
@ -1670,16 +1660,22 @@ def set_data_type(port: int = DEFAULT_GHIDRA_PORT,
|
|||||||
"timestamp": int(time.time() * 1000)
|
"timestamp": int(time.time() * 1000)
|
||||||
}
|
}
|
||||||
|
|
||||||
# We'll implement a more direct approach first by creating the data directly
|
# We need to first get the current name of the data
|
||||||
# First get info about the current data to use its name
|
|
||||||
try:
|
try:
|
||||||
# Try to use the built-in data types - simplified approach
|
# Just use a fixed name based on address for now
|
||||||
|
current_name = f"DATA_{address}"
|
||||||
|
|
||||||
|
# We're intentionally simplifying by not trying to preserve the current name
|
||||||
|
# This avoids potential API inconsistencies but means the name might change
|
||||||
|
|
||||||
|
# Prepare the payload with both type and the current name
|
||||||
payload = {
|
payload = {
|
||||||
"address": address,
|
"address": address,
|
||||||
"type": data_type
|
"type": data_type,
|
||||||
|
"newName": current_name # Preserve the current name
|
||||||
}
|
}
|
||||||
|
|
||||||
# This uses the create_data endpoint which has robust support
|
# This uses the POST endpoint to update both type and preserve name
|
||||||
response = safe_post(port, "data", payload)
|
response = safe_post(port, "data", payload)
|
||||||
return simplify_response(response)
|
return simplify_response(response)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
|
|||||||
@ -45,8 +45,32 @@ package eu.starsong.ghidra.endpoints;
|
|||||||
@Override
|
@Override
|
||||||
public void registerEndpoints(HttpServer server) {
|
public void registerEndpoints(HttpServer server) {
|
||||||
server.createContext("/data", this::handleData);
|
server.createContext("/data", this::handleData);
|
||||||
server.createContext("/data/update", this::handleUpdateData);
|
server.createContext("/data/update", exchange -> {
|
||||||
server.createContext("/data/type", this::handleSetDataType);
|
try {
|
||||||
|
if ("POST".equals(exchange.getRequestMethod())) {
|
||||||
|
Map<String, String> params = parseJsonPostParams(exchange);
|
||||||
|
handleUpdateData(exchange, params);
|
||||||
|
} else {
|
||||||
|
sendErrorResponse(exchange, 405, "Method Not Allowed");
|
||||||
|
}
|
||||||
|
} catch (Exception e) {
|
||||||
|
Msg.error(this, "Error in /data/update endpoint", e);
|
||||||
|
sendErrorResponse(exchange, 500, "Internal server error: " + e.getMessage());
|
||||||
|
}
|
||||||
|
});
|
||||||
|
server.createContext("/data/type", exchange -> {
|
||||||
|
try {
|
||||||
|
if ("POST".equals(exchange.getRequestMethod()) || "PATCH".equals(exchange.getRequestMethod())) {
|
||||||
|
Map<String, String> params = parseJsonPostParams(exchange);
|
||||||
|
handleTypeChangeData(exchange, params);
|
||||||
|
} else {
|
||||||
|
sendErrorResponse(exchange, 405, "Method Not Allowed");
|
||||||
|
}
|
||||||
|
} catch (Exception e) {
|
||||||
|
Msg.error(this, "Error in /data/type endpoint", e);
|
||||||
|
sendErrorResponse(exchange, 500, "Internal server error: " + e.getMessage());
|
||||||
|
}
|
||||||
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
public void handleData(HttpExchange exchange) throws IOException {
|
public void handleData(HttpExchange exchange) throws IOException {
|
||||||
@ -54,7 +78,39 @@ package eu.starsong.ghidra.endpoints;
|
|||||||
if ("GET".equals(exchange.getRequestMethod())) {
|
if ("GET".equals(exchange.getRequestMethod())) {
|
||||||
handleListData(exchange);
|
handleListData(exchange);
|
||||||
} else if ("POST".equals(exchange.getRequestMethod())) {
|
} else if ("POST".equals(exchange.getRequestMethod())) {
|
||||||
handleRenameData(exchange);
|
// Determine what kind of operation this is based on parameters
|
||||||
|
Map<String, String> params = parseJsonPostParams(exchange);
|
||||||
|
// Debug - log the params
|
||||||
|
StringBuilder debugInfo = new StringBuilder("DEBUG - Received parameters: ");
|
||||||
|
for (Map.Entry<String, String> entry : params.entrySet()) {
|
||||||
|
debugInfo.append(entry.getKey()).append("=").append(entry.getValue()).append(", ");
|
||||||
|
}
|
||||||
|
Msg.info(this, debugInfo.toString());
|
||||||
|
|
||||||
|
boolean hasNewName = params.containsKey("newName") && params.get("newName") != null && !params.get("newName").isEmpty();
|
||||||
|
boolean hasType = params.containsKey("type") && params.get("type") != null && !params.get("type").isEmpty();
|
||||||
|
|
||||||
|
// Add more detailed debugging
|
||||||
|
Msg.info(this, "Decision logic: hasNewName=" + hasNewName + ", hasType=" + hasType);
|
||||||
|
Msg.info(this, "Raw newName value: " + params.get("newName"));
|
||||||
|
Msg.info(this, "Raw type value: " + params.get("type"));
|
||||||
|
Msg.info(this, "Raw address value: " + params.get("address"));
|
||||||
|
|
||||||
|
// Let's go ahead and call handleUpdateData (since we know we have both params)
|
||||||
|
if (params.containsKey("address") && hasNewName && hasType) {
|
||||||
|
Msg.info(this, "Selected route: handleUpdateData - both name and type");
|
||||||
|
handleUpdateData(exchange, params);
|
||||||
|
} else if (params.containsKey("address") && hasNewName) {
|
||||||
|
Msg.info(this, "Selected route: handleRenameData - only name");
|
||||||
|
handleRenameData(exchange, params);
|
||||||
|
} else if (params.containsKey("address") && hasType) {
|
||||||
|
Msg.info(this, "Selected route: handleTypeChangeData - only type");
|
||||||
|
handleTypeChangeData(exchange, params);
|
||||||
|
} else {
|
||||||
|
Msg.info(this, "Selected route: Error - missing parameters");
|
||||||
|
// Neither parameter was provided
|
||||||
|
sendErrorResponse(exchange, 400, "Missing required parameters: at least one of newName or type must be provided", "MISSING_PARAMETERS");
|
||||||
|
}
|
||||||
} else {
|
} else {
|
||||||
sendErrorResponse(exchange, 405, "Method Not Allowed");
|
sendErrorResponse(exchange, 405, "Method Not Allowed");
|
||||||
}
|
}
|
||||||
@ -120,14 +176,28 @@ package eu.starsong.ghidra.endpoints;
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private void handleRenameData(HttpExchange exchange) throws IOException {
|
private void handleRenameData(HttpExchange exchange, Map<String, String> params) throws IOException {
|
||||||
try {
|
try {
|
||||||
Map<String, String> params = parseJsonPostParams(exchange);
|
// Debug - log the params again
|
||||||
|
StringBuilder debugInfo = new StringBuilder("DEBUG handleRenameData - Received parameters: ");
|
||||||
|
for (Map.Entry<String, String> entry : params.entrySet()) {
|
||||||
|
debugInfo.append(entry.getKey()).append("=").append(entry.getValue()).append(", ");
|
||||||
|
}
|
||||||
|
Msg.info(this, debugInfo.toString());
|
||||||
|
|
||||||
final String addressStr = params.get("address");
|
final String addressStr = params.get("address");
|
||||||
final String newName = params.get("newName");
|
final String newName = params.get("newName");
|
||||||
|
final String dataTypeStr = params.get("type");
|
||||||
|
|
||||||
if (addressStr == null || addressStr.isEmpty() || newName == null || newName.isEmpty()) {
|
// Address is always required
|
||||||
sendErrorResponse(exchange, 400, "Missing required parameters: address, newName", "MISSING_PARAMETERS");
|
if (addressStr == null || addressStr.isEmpty()) {
|
||||||
|
sendErrorResponse(exchange, 400, "Missing required parameter: address", "MISSING_PARAMETERS");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Either newName or type or both must be provided
|
||||||
|
if ((newName == null || newName.isEmpty()) && (dataTypeStr == null || dataTypeStr.isEmpty())) {
|
||||||
|
sendErrorResponse(exchange, 400, "At least one of newName or type must be provided", "MISSING_PARAMETERS");
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -138,17 +208,131 @@ package eu.starsong.ghidra.endpoints;
|
|||||||
}
|
}
|
||||||
|
|
||||||
try {
|
try {
|
||||||
TransactionHelper.executeInTransaction(program, "Rename Data", () -> {
|
// Create a result map to collect operation results
|
||||||
if (!renameDataAtAddress(program, addressStr, newName)) {
|
Map<String, Object> resultMap = new HashMap<>();
|
||||||
throw new Exception("Rename data operation failed internally.");
|
resultMap.put("address", addressStr);
|
||||||
|
|
||||||
|
TransactionHelper.executeInTransaction(program, "Update Data", () -> {
|
||||||
|
// Get the data at the address first
|
||||||
|
Address addr = program.getAddressFactory().getAddress(addressStr);
|
||||||
|
Listing listing = program.getListing();
|
||||||
|
Data data = listing.getDefinedDataAt(addr);
|
||||||
|
|
||||||
|
if (data == null) {
|
||||||
|
throw new Exception("No defined data found at address: " + addressStr);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Get current data info for operations that need it
|
||||||
|
String currentName = null;
|
||||||
|
if (data.getLabel() != null) {
|
||||||
|
currentName = data.getLabel();
|
||||||
|
} else {
|
||||||
|
Symbol sym = program.getSymbolTable().getPrimarySymbol(addr);
|
||||||
|
if (sym != null) {
|
||||||
|
currentName = sym.getName();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// If we need to set a data type
|
||||||
|
if (dataTypeStr != null && !dataTypeStr.isEmpty()) {
|
||||||
|
// Find the data type
|
||||||
|
ghidra.program.model.data.DataType dataType = null;
|
||||||
|
|
||||||
|
// First try built-in types
|
||||||
|
dataType = program.getDataTypeManager().getDataType("/" + dataTypeStr);
|
||||||
|
|
||||||
|
// If not found, try to find it without path
|
||||||
|
if (dataType == null) {
|
||||||
|
dataType = program.getDataTypeManager().findDataType("/" + dataTypeStr);
|
||||||
|
}
|
||||||
|
|
||||||
|
// If still null, try using the parser
|
||||||
|
if (dataType == null) {
|
||||||
|
try {
|
||||||
|
ghidra.app.util.parser.FunctionSignatureParser parser =
|
||||||
|
new ghidra.app.util.parser.FunctionSignatureParser(program.getDataTypeManager(), null);
|
||||||
|
dataType = parser.parse(null, dataTypeStr);
|
||||||
|
} catch (Exception e) {
|
||||||
|
Msg.debug(this, "Function signature parser failed: " + e.getMessage());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (dataType == null) {
|
||||||
|
throw new Exception("Could not find or parse data type: " + dataTypeStr);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Apply the data type
|
||||||
|
try {
|
||||||
|
// Clear any existing data first
|
||||||
|
listing.clearCodeUnits(addr, addr.add(data.getLength() - 1), false);
|
||||||
|
|
||||||
|
// Create new data with the type
|
||||||
|
Data newData = listing.createData(addr, dataType);
|
||||||
|
if (newData == null) {
|
||||||
|
throw new Exception("Failed to apply data type " + dataTypeStr + " at " + addressStr);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Capture info for response
|
||||||
|
resultMap.put("dataType", dataTypeStr);
|
||||||
|
resultMap.put("originalType", data.getDataType().getName());
|
||||||
|
|
||||||
|
// Update our reference to the data
|
||||||
|
data = newData;
|
||||||
|
} catch (Exception e) {
|
||||||
|
throw new Exception("Error applying data type: " + e.getMessage(), e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Handle renaming if needed
|
||||||
|
if (newName != null && !newName.isEmpty()) {
|
||||||
|
SymbolTable symTable = program.getSymbolTable();
|
||||||
|
Symbol symbol = symTable.getPrimarySymbol(addr);
|
||||||
|
|
||||||
|
if (symbol != null) {
|
||||||
|
symbol.setName(newName, SourceType.USER_DEFINED);
|
||||||
|
} else {
|
||||||
|
// Create a new label if no primary symbol exists
|
||||||
|
symTable.createLabel(addr, newName, SourceType.USER_DEFINED);
|
||||||
|
}
|
||||||
|
|
||||||
|
resultMap.put("name", newName);
|
||||||
|
if (currentName != null) {
|
||||||
|
resultMap.put("originalName", currentName);
|
||||||
|
}
|
||||||
|
} else if (currentName != null) {
|
||||||
|
// If we didn't rename but have a name from data type change, preserve it
|
||||||
|
SymbolTable symTable = program.getSymbolTable();
|
||||||
|
Symbol symbol = symTable.getPrimarySymbol(addr);
|
||||||
|
|
||||||
|
if (symbol == null || !symbol.getName().equals(currentName)) {
|
||||||
|
if (symbol != null) {
|
||||||
|
symbol.setName(currentName, SourceType.USER_DEFINED);
|
||||||
|
} else {
|
||||||
|
symTable.createLabel(addr, currentName, SourceType.USER_DEFINED);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
resultMap.put("name", currentName);
|
||||||
|
}
|
||||||
|
|
||||||
return null; // Return null for void operation
|
return null; // Return null for void operation
|
||||||
});
|
});
|
||||||
|
|
||||||
|
// Add a meaningful message
|
||||||
|
String message;
|
||||||
|
if (newName != null && !newName.isEmpty() && dataTypeStr != null && !dataTypeStr.isEmpty()) {
|
||||||
|
message = "Data renamed and type changed successfully";
|
||||||
|
} else if (newName != null && !newName.isEmpty()) {
|
||||||
|
message = "Data renamed successfully";
|
||||||
|
} else {
|
||||||
|
message = "Data type changed successfully";
|
||||||
|
}
|
||||||
|
resultMap.put("message", message);
|
||||||
|
|
||||||
// Build HATEOAS response
|
// Build HATEOAS response
|
||||||
eu.starsong.ghidra.api.ResponseBuilder builder = new eu.starsong.ghidra.api.ResponseBuilder(exchange, port)
|
eu.starsong.ghidra.api.ResponseBuilder builder = new eu.starsong.ghidra.api.ResponseBuilder(exchange, port)
|
||||||
.success(true)
|
.success(true)
|
||||||
.result(Map.of("message", "Data renamed successfully", "address", addressStr, "name", newName));
|
.result(resultMap);
|
||||||
|
|
||||||
// Add relevant links
|
// Add relevant links
|
||||||
builder.addLink("self", "/data/" + addressStr);
|
builder.addLink("self", "/data/" + addressStr);
|
||||||
@ -157,18 +341,18 @@ package eu.starsong.ghidra.endpoints;
|
|||||||
|
|
||||||
sendJsonResponse(exchange, builder.build(), 200);
|
sendJsonResponse(exchange, builder.build(), 200);
|
||||||
} catch (TransactionException e) {
|
} catch (TransactionException e) {
|
||||||
Msg.error(this, "Transaction failed: Rename Data", e);
|
Msg.error(this, "Transaction failed: Update Data", e);
|
||||||
sendErrorResponse(exchange, 500, "Failed to rename data: " + e.getMessage(), "TRANSACTION_ERROR");
|
sendErrorResponse(exchange, 500, "Failed to update data: " + e.getMessage(), "TRANSACTION_ERROR");
|
||||||
} catch (Exception e) { // Catch potential AddressFormatException or other issues
|
} catch (Exception e) { // Catch potential AddressFormatException or other issues
|
||||||
Msg.error(this, "Error during rename data operation", e);
|
Msg.error(this, "Error during data update operation", e);
|
||||||
sendErrorResponse(exchange, 400, "Error renaming data: " + e.getMessage(), "INVALID_PARAMETER");
|
sendErrorResponse(exchange, 400, "Error updating data: " + e.getMessage(), "INVALID_PARAMETER");
|
||||||
}
|
}
|
||||||
} catch (IOException e) {
|
} catch (IOException e) {
|
||||||
Msg.error(this, "Error parsing POST params for data rename", e);
|
Msg.error(this, "Error parsing POST params for data update", e);
|
||||||
sendErrorResponse(exchange, 400, "Invalid request body: " + e.getMessage(), "INVALID_REQUEST");
|
sendErrorResponse(exchange, 400, "Invalid request body: " + e.getMessage(), "INVALID_REQUEST");
|
||||||
} catch (Exception e) { // Catch unexpected errors
|
} catch (Exception e) { // Catch unexpected errors
|
||||||
Msg.error(this, "Unexpected error renaming data", e);
|
Msg.error(this, "Unexpected error updating data", e);
|
||||||
sendErrorResponse(exchange, 500, "Error renaming data: " + e.getMessage(), "INTERNAL_ERROR");
|
sendErrorResponse(exchange, 500, "Error updating data: " + e.getMessage(), "INTERNAL_ERROR");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -204,18 +388,342 @@ package eu.starsong.ghidra.endpoints;
|
|||||||
return successFlag.get();
|
return successFlag.get();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Handle a data type change request (without renaming)
|
||||||
|
*/
|
||||||
|
public void handleTypeChangeData(HttpExchange exchange, Map<String, String> params) throws IOException {
|
||||||
|
try {
|
||||||
|
// Debug - log all parameters received by this method
|
||||||
|
StringBuilder debugInfo = new StringBuilder("DEBUG handleTypeChangeData - Received parameters: ");
|
||||||
|
for (Map.Entry<String, String> entry : params.entrySet()) {
|
||||||
|
debugInfo.append(entry.getKey()).append("=").append(entry.getValue()).append(", ");
|
||||||
|
}
|
||||||
|
Msg.info(this, debugInfo.toString());
|
||||||
|
|
||||||
|
final String addressStr = params.get("address");
|
||||||
|
final String dataTypeStr = params.get("type");
|
||||||
|
|
||||||
|
Msg.info(this, "handleTypeChangeData - extracted parameters: address=" + addressStr +
|
||||||
|
", type=" + dataTypeStr);
|
||||||
|
|
||||||
|
if (addressStr == null || addressStr.isEmpty()) {
|
||||||
|
Msg.info(this, "handleTypeChangeData - Missing required parameter: address");
|
||||||
|
sendErrorResponse(exchange, 400, "Missing required parameter: address", "MISSING_PARAMETERS");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (dataTypeStr == null || dataTypeStr.isEmpty()) {
|
||||||
|
sendErrorResponse(exchange, 400, "Missing required parameter: type", "MISSING_PARAMETERS");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
Program program = getCurrentProgram();
|
||||||
|
if (program == null) {
|
||||||
|
sendErrorResponse(exchange, 400, "No program loaded", "NO_PROGRAM_LOADED");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
// Create a result map to collect operation results
|
||||||
|
Map<String, Object> resultMap = new HashMap<>();
|
||||||
|
resultMap.put("address", addressStr);
|
||||||
|
resultMap.put("dataType", dataTypeStr);
|
||||||
|
|
||||||
|
TransactionHelper.executeInTransaction(program, "Change Data Type", () -> {
|
||||||
|
// Get the data at the address first
|
||||||
|
Address addr = program.getAddressFactory().getAddress(addressStr);
|
||||||
|
Listing listing = program.getListing();
|
||||||
|
Data data = listing.getDefinedDataAt(addr);
|
||||||
|
|
||||||
|
if (data == null) {
|
||||||
|
throw new Exception("No defined data found at address: " + addressStr);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Get current name to preserve after type change
|
||||||
|
String currentName = null;
|
||||||
|
Symbol symbol = program.getSymbolTable().getPrimarySymbol(addr);
|
||||||
|
if (symbol != null) {
|
||||||
|
currentName = symbol.getName();
|
||||||
|
resultMap.put("originalName", currentName);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Remember original data type
|
||||||
|
String originalType = data.getDataType().getName();
|
||||||
|
resultMap.put("originalType", originalType);
|
||||||
|
|
||||||
|
// Find the requested data type
|
||||||
|
ghidra.program.model.data.DataType dataType = null;
|
||||||
|
|
||||||
|
// First try built-in types
|
||||||
|
dataType = program.getDataTypeManager().getDataType("/" + dataTypeStr);
|
||||||
|
|
||||||
|
// If not found, try to find it without path
|
||||||
|
if (dataType == null) {
|
||||||
|
dataType = program.getDataTypeManager().findDataType("/" + dataTypeStr);
|
||||||
|
}
|
||||||
|
|
||||||
|
// If still null, try using the parser
|
||||||
|
if (dataType == null) {
|
||||||
|
try {
|
||||||
|
ghidra.app.util.parser.FunctionSignatureParser parser =
|
||||||
|
new ghidra.app.util.parser.FunctionSignatureParser(program.getDataTypeManager(), null);
|
||||||
|
dataType = parser.parse(null, dataTypeStr);
|
||||||
|
} catch (Exception e) {
|
||||||
|
Msg.debug(this, "Function signature parser failed: " + e.getMessage());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (dataType == null) {
|
||||||
|
throw new Exception("Could not find or parse data type: " + dataTypeStr);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Clear existing data
|
||||||
|
int length = data.getLength();
|
||||||
|
listing.clearCodeUnits(addr, addr.add(length - 1), false);
|
||||||
|
|
||||||
|
// Create new data
|
||||||
|
Data newData = listing.createData(addr, dataType);
|
||||||
|
if (newData == null) {
|
||||||
|
throw new Exception("Failed to create data with type " + dataTypeStr);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Preserve the original name
|
||||||
|
if (currentName != null) {
|
||||||
|
SymbolTable symTable = program.getSymbolTable();
|
||||||
|
symTable.createLabel(addr, currentName, SourceType.USER_DEFINED);
|
||||||
|
resultMap.put("name", currentName);
|
||||||
|
}
|
||||||
|
|
||||||
|
return null;
|
||||||
|
});
|
||||||
|
|
||||||
|
resultMap.put("message", "Data type changed successfully");
|
||||||
|
|
||||||
|
// Build HATEOAS response
|
||||||
|
eu.starsong.ghidra.api.ResponseBuilder builder = new eu.starsong.ghidra.api.ResponseBuilder(exchange, port)
|
||||||
|
.success(true)
|
||||||
|
.result(resultMap);
|
||||||
|
|
||||||
|
// Add relevant links
|
||||||
|
builder.addLink("self", "/data/" + addressStr);
|
||||||
|
builder.addLink("data", "/data");
|
||||||
|
builder.addLink("program", "/program");
|
||||||
|
|
||||||
|
sendJsonResponse(exchange, builder.build(), 200);
|
||||||
|
} catch (TransactionException e) {
|
||||||
|
Msg.error(this, "Transaction failed: Change Data Type", e);
|
||||||
|
sendErrorResponse(exchange, 500, "Failed to change data type: " + e.getMessage(), "TRANSACTION_ERROR");
|
||||||
|
} catch (Exception e) {
|
||||||
|
Msg.error(this, "Error changing data type", e);
|
||||||
|
sendErrorResponse(exchange, 400, "Error changing data type: " + e.getMessage(), "INVALID_PARAMETER");
|
||||||
|
}
|
||||||
|
} catch (IOException e) {
|
||||||
|
Msg.error(this, "Error parsing POST params for data type change", e);
|
||||||
|
sendErrorResponse(exchange, 400, "Invalid request body: " + e.getMessage(), "INVALID_REQUEST");
|
||||||
|
} catch (Exception e) {
|
||||||
|
Msg.error(this, "Unexpected error changing data type", e);
|
||||||
|
sendErrorResponse(exchange, 500, "Error changing data type: " + e.getMessage(), "INTERNAL_ERROR");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Handle a combined update request (both name and type)
|
||||||
|
*/
|
||||||
|
public void handleUpdateData(HttpExchange exchange, Map<String, String> params) throws IOException {
|
||||||
|
try {
|
||||||
|
// Debug - log all parameters received by this method
|
||||||
|
StringBuilder debugInfo = new StringBuilder("DEBUG handleUpdateData - Received parameters: ");
|
||||||
|
for (Map.Entry<String, String> entry : params.entrySet()) {
|
||||||
|
debugInfo.append(entry.getKey()).append("=").append(entry.getValue()).append(", ");
|
||||||
|
}
|
||||||
|
Msg.info(this, debugInfo.toString());
|
||||||
|
|
||||||
|
final String addressStr = params.get("address");
|
||||||
|
final String newName = params.get("newName");
|
||||||
|
final String dataTypeStr = params.get("type");
|
||||||
|
|
||||||
|
Msg.info(this, "handleUpdateData - extracted parameters: address=" + addressStr +
|
||||||
|
", newName=" + newName + ", type=" + dataTypeStr);
|
||||||
|
|
||||||
|
if (addressStr == null || addressStr.isEmpty()) {
|
||||||
|
Msg.info(this, "handleUpdateData - Missing required parameter: address");
|
||||||
|
sendErrorResponse(exchange, 400, "Missing required parameter: address", "MISSING_PARAMETERS");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if ((newName == null || newName.isEmpty()) && (dataTypeStr == null || dataTypeStr.isEmpty())) {
|
||||||
|
sendErrorResponse(exchange, 400, "Missing required parameters: at least one of newName or type must be provided", "MISSING_PARAMETERS");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
Program program = getCurrentProgram();
|
||||||
|
if (program == null) {
|
||||||
|
sendErrorResponse(exchange, 400, "No program loaded", "NO_PROGRAM_LOADED");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
// Create a result map to collect operation results
|
||||||
|
Map<String, Object> resultMap = new HashMap<>();
|
||||||
|
resultMap.put("address", addressStr);
|
||||||
|
|
||||||
|
TransactionHelper.executeInTransaction(program, "Update Data", () -> {
|
||||||
|
// Get the data at the address first
|
||||||
|
Address addr = program.getAddressFactory().getAddress(addressStr);
|
||||||
|
Listing listing = program.getListing();
|
||||||
|
Data data = listing.getDefinedDataAt(addr);
|
||||||
|
|
||||||
|
if (data == null) {
|
||||||
|
throw new Exception("No defined data found at address: " + addressStr);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Get current name
|
||||||
|
String currentName = null;
|
||||||
|
Symbol symbol = program.getSymbolTable().getPrimarySymbol(addr);
|
||||||
|
if (symbol != null) {
|
||||||
|
currentName = symbol.getName();
|
||||||
|
resultMap.put("originalName", currentName);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Handle type change if requested
|
||||||
|
if (dataTypeStr != null && !dataTypeStr.isEmpty()) {
|
||||||
|
// Remember original type
|
||||||
|
String originalType = data.getDataType().getName();
|
||||||
|
resultMap.put("originalType", originalType);
|
||||||
|
|
||||||
|
// Find the data type
|
||||||
|
ghidra.program.model.data.DataType dataType = null;
|
||||||
|
|
||||||
|
// First try built-in types
|
||||||
|
dataType = program.getDataTypeManager().getDataType("/" + dataTypeStr);
|
||||||
|
|
||||||
|
// If not found, try to find it without path
|
||||||
|
if (dataType == null) {
|
||||||
|
dataType = program.getDataTypeManager().findDataType("/" + dataTypeStr);
|
||||||
|
}
|
||||||
|
|
||||||
|
// If still null, try using the parser
|
||||||
|
if (dataType == null) {
|
||||||
|
try {
|
||||||
|
ghidra.app.util.parser.FunctionSignatureParser parser =
|
||||||
|
new ghidra.app.util.parser.FunctionSignatureParser(program.getDataTypeManager(), null);
|
||||||
|
dataType = parser.parse(null, dataTypeStr);
|
||||||
|
} catch (Exception e) {
|
||||||
|
Msg.debug(this, "Function signature parser failed: " + e.getMessage());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (dataType == null) {
|
||||||
|
throw new Exception("Could not find or parse data type: " + dataTypeStr);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Apply the data type
|
||||||
|
try {
|
||||||
|
// Clear existing data
|
||||||
|
int length = data.getLength();
|
||||||
|
listing.clearCodeUnits(addr, addr.add(length - 1), false);
|
||||||
|
|
||||||
|
// Create new data with the type
|
||||||
|
Data newData = listing.createData(addr, dataType);
|
||||||
|
if (newData == null) {
|
||||||
|
throw new Exception("Failed to create data with type " + dataTypeStr);
|
||||||
|
}
|
||||||
|
|
||||||
|
resultMap.put("dataType", dataTypeStr);
|
||||||
|
|
||||||
|
// Update our reference to the data
|
||||||
|
data = newData;
|
||||||
|
} catch (Exception e) {
|
||||||
|
throw new Exception("Error applying data type: " + e.getMessage(), e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Handle rename if requested
|
||||||
|
if (newName != null && !newName.isEmpty()) {
|
||||||
|
SymbolTable symTable = program.getSymbolTable();
|
||||||
|
Symbol currentSymbol = symTable.getPrimarySymbol(addr);
|
||||||
|
|
||||||
|
if (currentSymbol != null) {
|
||||||
|
currentSymbol.setName(newName, SourceType.USER_DEFINED);
|
||||||
|
} else {
|
||||||
|
// Create a new label if no primary symbol exists
|
||||||
|
symTable.createLabel(addr, newName, SourceType.USER_DEFINED);
|
||||||
|
}
|
||||||
|
|
||||||
|
resultMap.put("name", newName);
|
||||||
|
} else if (currentName != null) {
|
||||||
|
// If we didn't rename but need to preserve name after type change
|
||||||
|
SymbolTable symTable = program.getSymbolTable();
|
||||||
|
Symbol currentSymbol = symTable.getPrimarySymbol(addr);
|
||||||
|
|
||||||
|
if (currentSymbol == null) {
|
||||||
|
symTable.createLabel(addr, currentName, SourceType.USER_DEFINED);
|
||||||
|
}
|
||||||
|
|
||||||
|
resultMap.put("name", currentName);
|
||||||
|
}
|
||||||
|
|
||||||
|
return null;
|
||||||
|
});
|
||||||
|
|
||||||
|
// Add a meaningful message
|
||||||
|
String message;
|
||||||
|
if (newName != null && !newName.isEmpty() && dataTypeStr != null && !dataTypeStr.isEmpty()) {
|
||||||
|
message = "Data renamed and type changed successfully";
|
||||||
|
} else if (newName != null && !newName.isEmpty()) {
|
||||||
|
message = "Data renamed successfully";
|
||||||
|
} else {
|
||||||
|
message = "Data type changed successfully";
|
||||||
|
}
|
||||||
|
resultMap.put("message", message);
|
||||||
|
|
||||||
|
// Build HATEOAS response
|
||||||
|
eu.starsong.ghidra.api.ResponseBuilder builder = new eu.starsong.ghidra.api.ResponseBuilder(exchange, port)
|
||||||
|
.success(true)
|
||||||
|
.result(resultMap);
|
||||||
|
|
||||||
|
// Add relevant links
|
||||||
|
builder.addLink("self", "/data/" + addressStr);
|
||||||
|
builder.addLink("data", "/data");
|
||||||
|
builder.addLink("program", "/program");
|
||||||
|
|
||||||
|
sendJsonResponse(exchange, builder.build(), 200);
|
||||||
|
} catch (TransactionException e) {
|
||||||
|
Msg.error(this, "Transaction failed: Update Data", e);
|
||||||
|
sendErrorResponse(exchange, 500, "Failed to update data: " + e.getMessage(), "TRANSACTION_ERROR");
|
||||||
|
} catch (Exception e) {
|
||||||
|
Msg.error(this, "Error during data update operation", e);
|
||||||
|
sendErrorResponse(exchange, 400, "Error updating data: " + e.getMessage(), "INVALID_PARAMETER");
|
||||||
|
}
|
||||||
|
} catch (IOException e) {
|
||||||
|
Msg.error(this, "Error parsing POST params for data update", e);
|
||||||
|
sendErrorResponse(exchange, 400, "Invalid request body: " + e.getMessage(), "INVALID_REQUEST");
|
||||||
|
} catch (Exception e) {
|
||||||
|
Msg.error(this, "Unexpected error updating data", e);
|
||||||
|
sendErrorResponse(exchange, 500, "Error updating data: " + e.getMessage(), "INTERNAL_ERROR");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// parseIntOrDefault is inherited from AbstractEndpoint
|
// parseIntOrDefault is inherited from AbstractEndpoint
|
||||||
|
|
||||||
public void handleSetDataType(HttpExchange exchange) throws IOException {
|
public void handleSetDataType(HttpExchange exchange) throws IOException {
|
||||||
try {
|
try {
|
||||||
if ("PATCH".equals(exchange.getRequestMethod()) || "POST".equals(exchange.getRequestMethod())) {
|
if ("PATCH".equals(exchange.getRequestMethod()) || "POST".equals(exchange.getRequestMethod())) {
|
||||||
Map<String, String> params = parseJsonPostParams(exchange);
|
Map<String, String> params = parseJsonPostParams(exchange);
|
||||||
|
|
||||||
|
// Debug - log all parameters received by this method
|
||||||
|
StringBuilder debugInfo = new StringBuilder("DEBUG handleSetDataType - Received parameters: ");
|
||||||
|
for (Map.Entry<String, String> entry : params.entrySet()) {
|
||||||
|
debugInfo.append(entry.getKey()).append("=").append(entry.getValue()).append(", ");
|
||||||
|
}
|
||||||
|
Msg.info(this, debugInfo.toString());
|
||||||
|
|
||||||
final String addressStr = params.get("address");
|
final String addressStr = params.get("address");
|
||||||
final String dataTypeStr = params.get("dataType");
|
final String dataTypeStr = params.get("type");
|
||||||
|
|
||||||
if (addressStr == null || addressStr.isEmpty() || dataTypeStr == null || dataTypeStr.isEmpty()) {
|
if (addressStr == null || addressStr.isEmpty() || dataTypeStr == null || dataTypeStr.isEmpty()) {
|
||||||
sendErrorResponse(exchange, 400,
|
sendErrorResponse(exchange, 400,
|
||||||
"Missing required parameters: address and dataType must be provided",
|
"Missing required parameters: address and type must be provided",
|
||||||
"MISSING_PARAMETERS");
|
"MISSING_PARAMETERS");
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
@ -229,7 +737,7 @@ package eu.starsong.ghidra.endpoints;
|
|||||||
try {
|
try {
|
||||||
Map<String, Object> result = new HashMap<>();
|
Map<String, Object> result = new HashMap<>();
|
||||||
result.put("address", addressStr);
|
result.put("address", addressStr);
|
||||||
result.put("dataType", dataTypeStr);
|
result.put("type", dataTypeStr);
|
||||||
|
|
||||||
TransactionHelper.executeInTransaction(program, "Set Data Type", () -> {
|
TransactionHelper.executeInTransaction(program, "Set Data Type", () -> {
|
||||||
// Get the data at the address
|
// Get the data at the address
|
||||||
@ -336,131 +844,5 @@ package eu.starsong.ghidra.endpoints;
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
public void handleUpdateData(HttpExchange exchange) throws IOException {
|
// Note: The handleUpdateData method is already defined earlier in this file at line 477
|
||||||
try {
|
|
||||||
if ("PATCH".equals(exchange.getRequestMethod()) || "POST".equals(exchange.getRequestMethod())) {
|
|
||||||
Map<String, String> params = parseJsonPostParams(exchange);
|
|
||||||
final String addressStr = params.get("address");
|
|
||||||
final String newName = params.get("newName");
|
|
||||||
final String dataTypeStr = params.get("dataType");
|
|
||||||
|
|
||||||
// At least one of name or dataType must be provided
|
|
||||||
if (addressStr == null || addressStr.isEmpty() ||
|
|
||||||
(newName == null || newName.isEmpty()) && (dataTypeStr == null || dataTypeStr.isEmpty())) {
|
|
||||||
sendErrorResponse(exchange, 400,
|
|
||||||
"Missing required parameters: address and either name or dataType must be provided",
|
|
||||||
"MISSING_PARAMETERS");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
Program program = getCurrentProgram();
|
|
||||||
if (program == null) {
|
|
||||||
sendErrorResponse(exchange, 400, "No program loaded", "NO_PROGRAM_LOADED");
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
try {
|
|
||||||
Map<String, Object> result = new HashMap<>();
|
|
||||||
result.put("address", addressStr);
|
|
||||||
|
|
||||||
TransactionHelper.executeInTransaction(program, "Update Data", () -> {
|
|
||||||
// Get the data at the address
|
|
||||||
Address addr = program.getAddressFactory().getAddress(addressStr);
|
|
||||||
Listing listing = program.getListing();
|
|
||||||
Data data = listing.getDefinedDataAt(addr);
|
|
||||||
|
|
||||||
if (data == null) {
|
|
||||||
throw new Exception("No defined data found at address: " + addressStr);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Rename if name is provided
|
|
||||||
if (newName != null && !newName.isEmpty()) {
|
|
||||||
SymbolTable symTable = program.getSymbolTable();
|
|
||||||
Symbol symbol = symTable.getPrimarySymbol(addr);
|
|
||||||
if (symbol != null) {
|
|
||||||
symbol.setName(newName, SourceType.USER_DEFINED);
|
|
||||||
} else {
|
|
||||||
// Create a new label if no primary symbol exists
|
|
||||||
symTable.createLabel(addr, newName, SourceType.USER_DEFINED);
|
|
||||||
}
|
|
||||||
result.put("name", newName);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Change data type if specified
|
|
||||||
if (dataTypeStr != null && !dataTypeStr.isEmpty()) {
|
|
||||||
// Try to find the data type in the data type manager
|
|
||||||
ghidra.program.model.data.DataType dataType = null;
|
|
||||||
|
|
||||||
// First try built-in types
|
|
||||||
dataType = program.getDataTypeManager().getDataType("/" + dataTypeStr);
|
|
||||||
|
|
||||||
// If not found, try to parse it as a C-style declaration
|
|
||||||
if (dataType == null) {
|
|
||||||
ghidra.app.util.parser.FunctionSignatureParser parser =
|
|
||||||
new ghidra.app.util.parser.FunctionSignatureParser(program.getDataTypeManager(), null);
|
|
||||||
try {
|
|
||||||
dataType = parser.parse(null, dataTypeStr);
|
|
||||||
} catch (Exception e) {
|
|
||||||
Msg.error(this, "Error parsing data type: " + dataTypeStr, e);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (dataType == null) {
|
|
||||||
throw new Exception("Could not find or parse data type: " + dataTypeStr);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Apply the data type
|
|
||||||
try {
|
|
||||||
Data newData = listing.createData(addr, dataType);
|
|
||||||
if (newData == null) {
|
|
||||||
throw new Exception("Failed to apply data type " + dataTypeStr + " at " + addressStr);
|
|
||||||
}
|
|
||||||
} catch (Exception e) {
|
|
||||||
throw new Exception("Failed to apply data type " + dataTypeStr + " at " + addressStr, e);
|
|
||||||
}
|
|
||||||
|
|
||||||
result.put("dataType", dataTypeStr);
|
|
||||||
// Re-get the data to return its current info
|
|
||||||
data = listing.getDefinedDataAt(addr);
|
|
||||||
}
|
|
||||||
|
|
||||||
// Add additional data info to result
|
|
||||||
if (data != null) {
|
|
||||||
result.put("currentDataType", data.getDataType().getName());
|
|
||||||
result.put("length", data.getLength());
|
|
||||||
result.put("value", data.getDefaultValueRepresentation());
|
|
||||||
}
|
|
||||||
|
|
||||||
return null;
|
|
||||||
});
|
|
||||||
|
|
||||||
// Build HATEOAS response
|
|
||||||
eu.starsong.ghidra.api.ResponseBuilder builder = new eu.starsong.ghidra.api.ResponseBuilder(exchange, port)
|
|
||||||
.success(true)
|
|
||||||
.result(result);
|
|
||||||
|
|
||||||
// Add relevant links
|
|
||||||
builder.addLink("self", "/data/" + addressStr);
|
|
||||||
builder.addLink("data", "/data");
|
|
||||||
builder.addLink("program", "/program");
|
|
||||||
|
|
||||||
sendJsonResponse(exchange, builder.build(), 200);
|
|
||||||
} catch (TransactionException e) {
|
|
||||||
Msg.error(this, "Transaction failed: Update Data", e);
|
|
||||||
sendErrorResponse(exchange, 500, "Failed to update data: " + e.getMessage(), "TRANSACTION_ERROR");
|
|
||||||
} catch (Exception e) {
|
|
||||||
Msg.error(this, "Error during update data operation", e);
|
|
||||||
sendErrorResponse(exchange, 400, "Error updating data: " + e.getMessage(), "INVALID_PARAMETER");
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
sendErrorResponse(exchange, 405, "Method Not Allowed", "METHOD_NOT_ALLOWED");
|
|
||||||
}
|
|
||||||
} catch (IOException e) {
|
|
||||||
Msg.error(this, "Error parsing request parameters for data update", e);
|
|
||||||
sendErrorResponse(exchange, 400, "Invalid request body: " + e.getMessage(), "INVALID_REQUEST");
|
|
||||||
} catch (Exception e) {
|
|
||||||
Msg.error(this, "Unexpected error updating data", e);
|
|
||||||
sendErrorResponse(exchange, 500, "Error updating data: " + e.getMessage(), "INTERNAL_ERROR");
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|||||||
@ -127,6 +127,10 @@ public class HttpUtil {
|
|||||||
public static Map<String, String> parseJsonPostParams(HttpExchange exchange) throws IOException {
|
public static Map<String, String> parseJsonPostParams(HttpExchange exchange) throws IOException {
|
||||||
byte[] body = exchange.getRequestBody().readAllBytes();
|
byte[] body = exchange.getRequestBody().readAllBytes();
|
||||||
String bodyStr = new String(body, StandardCharsets.UTF_8);
|
String bodyStr = new String(body, StandardCharsets.UTF_8);
|
||||||
|
|
||||||
|
// Debug - log raw request body
|
||||||
|
ghidra.util.Msg.info(HttpUtil.class, "DEBUG Raw request body: " + bodyStr);
|
||||||
|
|
||||||
Map<String, String> params = new HashMap<>();
|
Map<String, String> params = new HashMap<>();
|
||||||
|
|
||||||
try {
|
try {
|
||||||
|
|||||||
181
test_data_update.py
Executable file
181
test_data_update.py
Executable file
@ -0,0 +1,181 @@
|
|||||||
|
#!/usr/bin/env python3
|
||||||
|
"""
|
||||||
|
Dedicated test script for the GhydraMCP data handling API.
|
||||||
|
|
||||||
|
This script has standalone tests to validate the three key data manipulation operations:
|
||||||
|
1. Rename only - Change the name without changing the data type
|
||||||
|
2. Type change only - Change the data type while preserving the name
|
||||||
|
3. Update both - Change both name and type simultaneously
|
||||||
|
|
||||||
|
These tests operate on a low level and can be run independently of the main test suite
|
||||||
|
to diagnose issues with the API's data handling capabilities.
|
||||||
|
|
||||||
|
Usage:
|
||||||
|
python test_data_update.py
|
||||||
|
"""
|
||||||
|
import json
|
||||||
|
import requests
|
||||||
|
import sys
|
||||||
|
import argparse
|
||||||
|
|
||||||
|
BASE_URL = "http://localhost:8192"
|
||||||
|
|
||||||
|
def test_data_update(verbose=True, base_url=None):
|
||||||
|
"""Test data update operations
|
||||||
|
|
||||||
|
Args:
|
||||||
|
verbose: Whether to print detailed output
|
||||||
|
base_url: Base URL for the Ghidra HTTP API (default: http://localhost:8192)
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
bool: True if all tests pass, False otherwise
|
||||||
|
"""
|
||||||
|
if base_url:
|
||||||
|
global BASE_URL
|
||||||
|
BASE_URL = base_url
|
||||||
|
|
||||||
|
# Track test results
|
||||||
|
all_tests_passed = True
|
||||||
|
|
||||||
|
# First find a suitable data item to test with
|
||||||
|
if verbose:
|
||||||
|
print("Fetching data items...")
|
||||||
|
response = requests.get(f"{BASE_URL}/data?limit=1")
|
||||||
|
|
||||||
|
if response.status_code != 200:
|
||||||
|
print(f"Error: Failed to fetch data items, status {response.status_code}")
|
||||||
|
print(response.text)
|
||||||
|
return False
|
||||||
|
|
||||||
|
data = response.json()
|
||||||
|
if not data.get("success"):
|
||||||
|
print(f"Error: API call failed: {data.get('error', 'Unknown error')}")
|
||||||
|
return False
|
||||||
|
|
||||||
|
# Extract address from first data item
|
||||||
|
result = data.get("result", [])
|
||||||
|
if not result or not isinstance(result, list) or not result[0].get("address"):
|
||||||
|
print("Error: No data items found or invalid response format")
|
||||||
|
if result and verbose:
|
||||||
|
print(f"Result: {json.dumps(result, indent=2)}")
|
||||||
|
return False
|
||||||
|
|
||||||
|
address = result[0]["address"]
|
||||||
|
if verbose:
|
||||||
|
print(f"Using data item at address: {address}")
|
||||||
|
|
||||||
|
# Test 1: Renaming only
|
||||||
|
if verbose:
|
||||||
|
print("\n--- Test 1: Rename Only ---")
|
||||||
|
test_name = "TEST_DATA_RENAME"
|
||||||
|
payload = {
|
||||||
|
"address": address,
|
||||||
|
"newName": test_name
|
||||||
|
}
|
||||||
|
|
||||||
|
if verbose:
|
||||||
|
print(f"Request: POST {BASE_URL}/data")
|
||||||
|
print(f"Payload: {json.dumps(payload, indent=2)}")
|
||||||
|
|
||||||
|
response = requests.post(f"{BASE_URL}/data", json=payload)
|
||||||
|
if verbose:
|
||||||
|
print(f"Status: {response.status_code}")
|
||||||
|
print(f"Response: {json.dumps(response.json(), indent=2)}")
|
||||||
|
|
||||||
|
# Check Test 1 results
|
||||||
|
test1_passed = response.status_code == 200 and response.json().get("success")
|
||||||
|
if not test1_passed:
|
||||||
|
print(f"ERROR: Test 1 (Rename Only) failed: {response.status_code}")
|
||||||
|
all_tests_passed = False
|
||||||
|
|
||||||
|
# Test 2: Type change only
|
||||||
|
if verbose:
|
||||||
|
print("\n--- Test 2: Type Change Only ---")
|
||||||
|
payload = {
|
||||||
|
"address": address,
|
||||||
|
"type": "int" # Using 'type' as parameter name
|
||||||
|
}
|
||||||
|
|
||||||
|
if verbose:
|
||||||
|
print(f"Request: POST {BASE_URL}/data/type")
|
||||||
|
print(f"Payload: {json.dumps(payload, indent=2)}")
|
||||||
|
|
||||||
|
response = requests.post(f"{BASE_URL}/data/type", json=payload)
|
||||||
|
if verbose:
|
||||||
|
print(f"Status: {response.status_code}")
|
||||||
|
print(f"Response: {json.dumps(response.json(), indent=2)}")
|
||||||
|
|
||||||
|
# Check Test 2 results
|
||||||
|
test2_passed = response.status_code == 200 and response.json().get("success")
|
||||||
|
if not test2_passed:
|
||||||
|
print(f"ERROR: Test 2 (Type Change Only) failed: {response.status_code}")
|
||||||
|
all_tests_passed = False
|
||||||
|
|
||||||
|
# Test 3: Both name and type change
|
||||||
|
if verbose:
|
||||||
|
print("\n--- Test 3: Both Name and Type Change ---")
|
||||||
|
payload = {
|
||||||
|
"address": address,
|
||||||
|
"newName": "TEST_DATA_BOTH",
|
||||||
|
"type": "byte" # Using 'type' as parameter name
|
||||||
|
}
|
||||||
|
|
||||||
|
if verbose:
|
||||||
|
print(f"Request: POST {BASE_URL}/data/update")
|
||||||
|
print(f"Payload: {json.dumps(payload, indent=2)}")
|
||||||
|
|
||||||
|
response = requests.post(f"{BASE_URL}/data/update", json=payload)
|
||||||
|
if verbose:
|
||||||
|
print(f"Status: {response.status_code}")
|
||||||
|
print(f"Response: {json.dumps(response.json(), indent=2)}")
|
||||||
|
|
||||||
|
# Check Test 3 results
|
||||||
|
test3_passed = response.status_code == 200 and response.json().get("success")
|
||||||
|
if not test3_passed:
|
||||||
|
print(f"ERROR: Test 3 (Both Name and Type Change via /data/update) failed: {response.status_code}")
|
||||||
|
all_tests_passed = False
|
||||||
|
|
||||||
|
# Test 4: Direct raw request using the /data endpoint
|
||||||
|
if verbose:
|
||||||
|
print("\n--- Test 4: Direct Request to /data endpoint ---")
|
||||||
|
payload = {
|
||||||
|
"address": address,
|
||||||
|
"newName": "TEST_DIRECT_UPDATE",
|
||||||
|
"type": "int" # Using 'type' parameter name
|
||||||
|
}
|
||||||
|
|
||||||
|
if verbose:
|
||||||
|
print(f"Request: POST {BASE_URL}/data")
|
||||||
|
print(f"Payload: {json.dumps(payload, indent=2)}")
|
||||||
|
|
||||||
|
response = requests.post(f"{BASE_URL}/data", json=payload)
|
||||||
|
if verbose:
|
||||||
|
print(f"Status: {response.status_code}")
|
||||||
|
print(f"Response: {json.dumps(response.json(), indent=2)}")
|
||||||
|
|
||||||
|
# Check Test 4 results
|
||||||
|
test4_passed = response.status_code == 200 and response.json().get("success")
|
||||||
|
if not test4_passed:
|
||||||
|
print(f"ERROR: Test 4 (Both Name and Type Change via /data) failed: {response.status_code}")
|
||||||
|
all_tests_passed = False
|
||||||
|
|
||||||
|
# Print summary
|
||||||
|
if verbose:
|
||||||
|
print("\n--- Test Summary ---")
|
||||||
|
print(f"Test 1 (Rename Only): {'PASSED' if test1_passed else 'FAILED'}")
|
||||||
|
print(f"Test 2 (Type Change Only): {'PASSED' if test2_passed else 'FAILED'}")
|
||||||
|
print(f"Test 3 (Both Name and Type Change via /data/update): {'PASSED' if test3_passed else 'FAILED'}")
|
||||||
|
print(f"Test 4 (Both Name and Type Change via /data): {'PASSED' if test4_passed else 'FAILED'}")
|
||||||
|
print(f"Overall: {'ALL TESTS PASSED' if all_tests_passed else 'SOME TESTS FAILED'}")
|
||||||
|
|
||||||
|
return all_tests_passed
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
parser = argparse.ArgumentParser(description="Test data operations in the GhydraMCP HTTP API")
|
||||||
|
parser.add_argument("--quiet", "-q", action="store_true", help="Suppress detailed output")
|
||||||
|
parser.add_argument("--url", "-u", help="Base URL for the Ghidra HTTP API")
|
||||||
|
args = parser.parse_args()
|
||||||
|
|
||||||
|
success = test_data_update(not args.quiet, args.url)
|
||||||
|
if not success:
|
||||||
|
sys.exit(1)
|
||||||
@ -649,6 +649,81 @@ class GhydraMCPHttpApiTests(unittest.TestCase):
|
|||||||
"Function result missing required fields"
|
"Function result missing required fields"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
def test_data_operations(self):
|
||||||
|
"""Test data update operations including renaming and type changes"""
|
||||||
|
# First find a suitable data item to test with
|
||||||
|
response = requests.get(f"{BASE_URL}/data?limit=1")
|
||||||
|
if response.status_code != 200:
|
||||||
|
self.skipTest("No data items available to test operations")
|
||||||
|
|
||||||
|
data = response.json()
|
||||||
|
self.assertTrue(data.get("success", False), "API call failed")
|
||||||
|
|
||||||
|
result = data.get("result", [])
|
||||||
|
if not result or not isinstance(result, list) or not result[0].get("address"):
|
||||||
|
self.skipTest("No data items found or invalid response format")
|
||||||
|
|
||||||
|
address = result[0]["address"]
|
||||||
|
original_name = result[0].get("label", "unnamed")
|
||||||
|
original_type = result[0].get("dataType", "undefined")
|
||||||
|
|
||||||
|
try:
|
||||||
|
# Test 1: Rename Only
|
||||||
|
test_name = "TEST_DATA_RENAME"
|
||||||
|
payload = {
|
||||||
|
"address": address,
|
||||||
|
"newName": test_name
|
||||||
|
}
|
||||||
|
|
||||||
|
response = requests.post(f"{BASE_URL}/data", json=payload)
|
||||||
|
self.assertEqual(response.status_code, 200)
|
||||||
|
|
||||||
|
data = response.json()
|
||||||
|
self.assertStandardSuccessResponse(data)
|
||||||
|
self.assertEqual(data["result"]["name"], test_name)
|
||||||
|
self.assertEqual(data["result"]["address"], address)
|
||||||
|
|
||||||
|
# Test 2: Type Change Only
|
||||||
|
payload = {
|
||||||
|
"address": address,
|
||||||
|
"type": "int"
|
||||||
|
}
|
||||||
|
|
||||||
|
response = requests.post(f"{BASE_URL}/data/type", json=payload)
|
||||||
|
self.assertEqual(response.status_code, 200)
|
||||||
|
|
||||||
|
data = response.json()
|
||||||
|
self.assertStandardSuccessResponse(data)
|
||||||
|
self.assertEqual(data["result"]["dataType"], "int")
|
||||||
|
self.assertEqual(data["result"]["address"], address)
|
||||||
|
|
||||||
|
# Test 3: Both Name and Type Change
|
||||||
|
payload = {
|
||||||
|
"address": address,
|
||||||
|
"newName": "TEST_DATA_BOTH",
|
||||||
|
"type": "byte"
|
||||||
|
}
|
||||||
|
|
||||||
|
response = requests.post(f"{BASE_URL}/data/update", json=payload)
|
||||||
|
self.assertEqual(response.status_code, 200)
|
||||||
|
|
||||||
|
data = response.json()
|
||||||
|
self.assertStandardSuccessResponse(data)
|
||||||
|
self.assertEqual(data["result"]["name"], "TEST_DATA_BOTH")
|
||||||
|
self.assertEqual(data["result"]["dataType"], "byte")
|
||||||
|
self.assertEqual(data["result"]["address"], address)
|
||||||
|
|
||||||
|
# Restore original values
|
||||||
|
if original_type != "undefined" and original_name != "unnamed":
|
||||||
|
payload = {
|
||||||
|
"address": address,
|
||||||
|
"newName": original_name,
|
||||||
|
"type": original_type
|
||||||
|
}
|
||||||
|
requests.post(f"{BASE_URL}/data", json=payload)
|
||||||
|
except Exception as e:
|
||||||
|
self.fail(f"Data operations test failed: {str(e)}")
|
||||||
|
|
||||||
def test_all_read_endpoints():
|
def test_all_read_endpoints():
|
||||||
"""Function to exercise all read endpoints and display their responses.
|
"""Function to exercise all read endpoints and display their responses.
|
||||||
This is called separately from the unittest framework when requested."""
|
This is called separately from the unittest framework when requested."""
|
||||||
|
|||||||
@ -340,6 +340,78 @@ async def test_bridge():
|
|||||||
assert read_memory_data.get("address") == func_address, f"Wrong address in read_memory result: {read_memory_data.get('address')}"
|
assert read_memory_data.get("address") == func_address, f"Wrong address in read_memory result: {read_memory_data.get('address')}"
|
||||||
logger.info(f"Read memory result: {read_memory_result}")
|
logger.info(f"Read memory result: {read_memory_result}")
|
||||||
|
|
||||||
|
# Test data operations (create, rename, change type)
|
||||||
|
logger.info("Testing data operations...")
|
||||||
|
try:
|
||||||
|
# Get a memory address to create test data
|
||||||
|
data_address = func_address
|
||||||
|
|
||||||
|
# First create test data
|
||||||
|
create_data_args = {"port": 8192, "address": data_address, "data_type": "uint32_t"}
|
||||||
|
logger.info(f"Calling create_data with args: {create_data_args}")
|
||||||
|
create_data_result = await session.call_tool("create_data", arguments=create_data_args)
|
||||||
|
create_data_response = json.loads(create_data_result.content[0].text)
|
||||||
|
assert create_data_response.get("success") is True, f"Create data failed: {create_data_response}"
|
||||||
|
logger.info(f"Create data result: {create_data_result}")
|
||||||
|
|
||||||
|
# Test Case 1: Data rename operation (name only)
|
||||||
|
test_data_name = "test_data_item"
|
||||||
|
rename_data_args = {"port": 8192, "address": data_address, "name": test_data_name}
|
||||||
|
logger.info(f"Calling rename_data with args: {rename_data_args}")
|
||||||
|
rename_data_result = await session.call_tool("rename_data", arguments=rename_data_args)
|
||||||
|
rename_data_response = json.loads(rename_data_result.content[0].text)
|
||||||
|
assert rename_data_response.get("success") is True, f"Rename data failed: {rename_data_response}"
|
||||||
|
logger.info(f"Rename data result: {rename_data_result}")
|
||||||
|
|
||||||
|
# Verify the name was changed
|
||||||
|
if rename_data_response.get("result", {}).get("name") != test_data_name:
|
||||||
|
logger.warning(f"Rename operation didn't set the expected name. Got: {rename_data_response.get('result', {}).get('name')}")
|
||||||
|
|
||||||
|
# Test Case 2: Data type change operation (type only)
|
||||||
|
change_type_args = {"port": 8192, "address": data_address, "data_type": "int"}
|
||||||
|
logger.info(f"Calling set_data_type with args: {change_type_args}")
|
||||||
|
change_type_result = await session.call_tool("set_data_type", arguments=change_type_args)
|
||||||
|
change_type_response = json.loads(change_type_result.content[0].text)
|
||||||
|
assert change_type_response.get("success") is True, f"Change data type failed: {change_type_response}"
|
||||||
|
logger.info(f"Change data type result: {change_type_result}")
|
||||||
|
|
||||||
|
# Verify the type was changed but name was preserved
|
||||||
|
result = change_type_response.get("result", {})
|
||||||
|
if result.get("dataType") != "int":
|
||||||
|
logger.warning(f"Type change operation didn't set the expected type. Got: {result.get('dataType')}")
|
||||||
|
if result.get("name") != test_data_name:
|
||||||
|
logger.warning(f"Type change operation didn't preserve the name. Expected: {test_data_name}, Got: {result.get('name')}")
|
||||||
|
|
||||||
|
# Test Case 3: Combined update operation (both name and type)
|
||||||
|
update_data_args = {
|
||||||
|
"port": 8192,
|
||||||
|
"address": data_address,
|
||||||
|
"name": "updated_data_item",
|
||||||
|
"data_type": "byte"
|
||||||
|
}
|
||||||
|
logger.info(f"Calling update_data with args: {update_data_args}")
|
||||||
|
update_data_result = await session.call_tool("update_data", arguments=update_data_args)
|
||||||
|
update_data_response = json.loads(update_data_result.content[0].text)
|
||||||
|
assert update_data_response.get("success") is True, f"Update data failed: {update_data_response}"
|
||||||
|
logger.info(f"Update data result: {update_data_result}")
|
||||||
|
|
||||||
|
# Verify both name and type were changed
|
||||||
|
result = update_data_response.get("result", {})
|
||||||
|
if result.get("name") != "updated_data_item":
|
||||||
|
logger.warning(f"Update operation didn't set the expected name. Got: {result.get('name')}")
|
||||||
|
if result.get("dataType") != "byte":
|
||||||
|
logger.warning(f"Update operation didn't set the expected type. Got: {result.get('dataType')}")
|
||||||
|
|
||||||
|
# Clean up by restoring original data type
|
||||||
|
restore_type_args = {"port": 8192, "address": data_address, "data_type": "uint32_t"}
|
||||||
|
logger.info(f"Restoring data type with args: {restore_type_args}")
|
||||||
|
restore_type_result = await session.call_tool("set_data_type", arguments=restore_type_args)
|
||||||
|
restore_type_response = json.loads(restore_type_result.content[0].text)
|
||||||
|
assert restore_type_response.get("success") is True, f"Restore data type failed: {restore_type_response}"
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
logger.warning(f"Error testing data operations: {e} - This is not critical")
|
||||||
|
|
||||||
# Test callgraph functionality - handle possible failure gracefully
|
# Test callgraph functionality - handle possible failure gracefully
|
||||||
if func_address:
|
if func_address:
|
||||||
logger.info(f"Calling get_callgraph with address: {func_address}")
|
logger.info(f"Calling get_callgraph with address: {func_address}")
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user