package eu.starsong.ghidra.endpoints; import com.google.gson.JsonObject; import com.sun.net.httpserver.HttpExchange; import com.sun.net.httpserver.HttpServer; import eu.starsong.ghidra.util.TransactionHelper; import eu.starsong.ghidra.util.TransactionHelper.TransactionException; import ghidra.program.model.address.Address; import ghidra.framework.plugintool.PluginTool; import ghidra.program.model.listing.Data; import ghidra.program.model.listing.DataIterator; import ghidra.program.model.listing.Listing; import ghidra.program.model.listing.Program; import ghidra.program.model.mem.MemoryBlock; import ghidra.program.model.symbol.SourceType; import ghidra.program.model.symbol.Symbol; import ghidra.program.model.symbol.SymbolTable; import ghidra.util.Msg; import java.io.IOException; import java.util.*; import java.util.concurrent.atomic.AtomicBoolean; import javax.swing.SwingUtilities; import java.lang.reflect.InvocationTargetException; public class DataEndpoints extends AbstractEndpoint { private PluginTool tool; // Updated constructor to accept port public DataEndpoints(Program program, int port) { super(program, port); // Call super constructor } public DataEndpoints(Program program, int port, PluginTool tool) { super(program, port); this.tool = tool; } @Override protected PluginTool getTool() { return tool; } @Override public void registerEndpoints(HttpServer server) { server.createContext("/data", this::handleData); } public void handleData(HttpExchange exchange) throws IOException { try { if ("GET".equals(exchange.getRequestMethod())) { handleListData(exchange); } else if ("POST".equals(exchange.getRequestMethod())) { handleRenameData(exchange); } else { sendErrorResponse(exchange, 405, "Method Not Allowed"); } } catch (Exception e) { Msg.error(this, "Error in /data endpoint", e); sendErrorResponse(exchange, 500, "Internal server error: " + e.getMessage()); } } private void handleListData(HttpExchange exchange) throws IOException { try { Map qparams = parseQueryParams(exchange); int offset = parseIntOrDefault(qparams.get("offset"), 0); int limit = parseIntOrDefault(qparams.get("limit"), 100); Program program = getCurrentProgram(); if (program == null) { sendErrorResponse(exchange, 400, "No program loaded", "NO_PROGRAM_LOADED"); return; } List> dataItems = new ArrayList<>(); for (MemoryBlock block : program.getMemory().getBlocks()) { DataIterator it = program.getListing().getDefinedData(block.getStart(), true); while (it.hasNext()) { Data data = it.next(); if (block.contains(data.getAddress())) { Map item = new HashMap<>(); item.put("address", data.getAddress().toString()); item.put("label", data.getLabel() != null ? data.getLabel() : "(unnamed)"); item.put("value", data.getDefaultValueRepresentation()); item.put("dataType", data.getDataType().getName()); // Add HATEOAS links Map links = new HashMap<>(); Map selfLink = new HashMap<>(); selfLink.put("href", "/data/" + data.getAddress().toString()); links.put("self", selfLink); item.put("_links", links); dataItems.add(item); } } } // Build response with HATEOAS links eu.starsong.ghidra.api.ResponseBuilder builder = new eu.starsong.ghidra.api.ResponseBuilder(exchange, port) .success(true); // Apply pagination and get paginated items List> paginated = applyPagination(dataItems, offset, limit, builder, "/data"); // Set the paginated result builder.result(paginated); // Add program link builder.addLink("program", "/program"); sendJsonResponse(exchange, builder.build(), 200); } catch (Exception e) { Msg.error(this, "Error listing data", e); sendErrorResponse(exchange, 500, "Error listing data: " + e.getMessage(), "INTERNAL_ERROR"); } } private void handleRenameData(HttpExchange exchange) throws IOException { try { Map params = parseJsonPostParams(exchange); final String addressStr = params.get("address"); final String newName = params.get("newName"); if (addressStr == null || addressStr.isEmpty() || newName == null || newName.isEmpty()) { sendErrorResponse(exchange, 400, "Missing required parameters: address, newName", "MISSING_PARAMETERS"); return; } Program program = getCurrentProgram(); if (program == null) { sendErrorResponse(exchange, 400, "No program loaded", "NO_PROGRAM_LOADED"); return; } try { TransactionHelper.executeInTransaction(program, "Rename Data", () -> { if (!renameDataAtAddress(program, addressStr, newName)) { throw new Exception("Rename data operation failed internally."); } return null; // Return null for void operation }); // Build HATEOAS response eu.starsong.ghidra.api.ResponseBuilder builder = new eu.starsong.ghidra.api.ResponseBuilder(exchange, port) .success(true) .result(Map.of("message", "Data renamed successfully", "address", addressStr, "name", newName)); // 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: Rename Data", e); sendErrorResponse(exchange, 500, "Failed to rename data: " + e.getMessage(), "TRANSACTION_ERROR"); } catch (Exception e) { // Catch potential AddressFormatException or other issues Msg.error(this, "Error during rename data operation", e); sendErrorResponse(exchange, 400, "Error renaming data: " + e.getMessage(), "INVALID_PARAMETER"); } } catch (IOException e) { Msg.error(this, "Error parsing POST params for data rename", e); sendErrorResponse(exchange, 400, "Invalid request body: " + e.getMessage(), "INVALID_REQUEST"); } catch (Exception e) { // Catch unexpected errors Msg.error(this, "Unexpected error renaming data", e); sendErrorResponse(exchange, 500, "Error renaming data: " + e.getMessage(), "INTERNAL_ERROR"); } } private boolean renameDataAtAddress(Program program, String addressStr, String newName) throws Exception { // This method now throws Exception to be caught by the transaction helper AtomicBoolean successFlag = new AtomicBoolean(false); try { Address addr = program.getAddressFactory().getAddress(addressStr); Listing listing = program.getListing(); Data data = listing.getDefinedDataAt(addr); if (data != null) { SymbolTable symTable = program.getSymbolTable(); Symbol symbol = symTable.getPrimarySymbol(addr); if (symbol != null) { symbol.setName(newName, SourceType.USER_DEFINED); successFlag.set(true); } else { // Create a new label if no primary symbol exists symTable.createLabel(addr, newName, SourceType.USER_DEFINED); successFlag.set(true); } } else { throw new Exception("No defined data found at address: " + addressStr); } } catch (ghidra.program.model.address.AddressFormatException afe) { throw new Exception("Invalid address format: " + addressStr, afe); } catch (ghidra.util.exception.InvalidInputException iie) { throw new Exception("Invalid name: " + newName, iie); } catch (Exception e) { // Catch other potential Ghidra exceptions throw new Exception("Failed to rename data at " + addressStr, e); } return successFlag.get(); } // parseIntOrDefault is inherited from AbstractEndpoint }