yetangitu/ghidra
1
0
Fork 0
mirror of https://github.com/NationalSecurityAgency/ghidra.git synced 2025-10-04 02:09:44 +02:00
Code Issues Releases Wiki Activity

Candidate release of source code.

Browse source
This commit is contained in:
Dan 2019-03-26 13:45:32 -04:00
parent db81e6b3b0
commit 79d8f164f8
12449 changed files with 2800756 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

70
Ghidra/Features/Base/ghidra_scripts/CreateOperandReferencesInSelectionScript.java Normal file
View file

@ -0,0 +1,70 @@
/* ###
* IP: GHIDRA
* REVIEWED: YES
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
//This script creates references on all scalars contained in operands in the
// current selection. There must be a selection for this script to run.
//@category Analysis
import ghidra.app.script.GhidraScript;
import ghidra.program.model.address.Address;
import ghidra.program.model.address.AddressIterator;
import ghidra.program.model.listing.CodeUnit;
import ghidra.program.model.listing.Listing;
import ghidra.program.model.mem.Memory;
import ghidra.program.model.scalar.Scalar;
import ghidra.program.model.symbol.*;
public class CreateOperandReferencesInSelectionScript extends GhidraScript {
Listing listing;
Memory memory;
SymbolTable symbolTable;
@Override
public void run() throws Exception {
listing = currentProgram.getListing();
memory = currentProgram.getMemory();
symbolTable = currentProgram.getSymbolTable();
if(currentSelection == null) {
monitor.setMessage("You must have a selection for this script to run.");
return;
}
monitor.setMessage("Creating operand references...");
AddressIterator addrIt = currentSelection.getAddresses(true);
while(addrIt.hasNext()){
Address addr = addrIt.next();
CodeUnit cu = listing.getCodeUnitContaining(addr);
int numOps = cu.getNumOperands();
for(int i=0;i<numOps;i++){
Scalar scalar = cu.getScalar(i);
if(scalar != null){
//check to see if scalar value is a valid address in program memory
long scalarValue = scalar.getUnsignedValue();
Address testAddr = addr.getNewAddress(scalarValue);
if(memory.contains(testAddr)){
//if so, create the memory reference on the scalar operand
//TODO: not sure if the DATA type for the ref is correct
// RefTypeFactory.getDefaultMemoryRefType(instr, opIndex)
cu.addOperandReference(i, testAddr, RefType.DATA, SourceType.ANALYSIS);
}
}
}
}
}// end of run method
}