mirror of
https://github.com/NationalSecurityAgency/ghidra.git
synced 2025-10-03 09:49:23 +02:00
61 lines
2.5 KiB
Python
Executable file
61 lines
2.5 KiB
Python
Executable file
## ###
|
|
# IP: GHIDRA
|
|
#
|
|
# 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.
|
|
##
|
|
# Use the decompiler to generate signatures for the function at the current address, then dump the
|
|
# signature hashes and debug information to the console
|
|
# @category: BSim.python
|
|
|
|
import ghidra.app.decompiler.tracking.DecompInterfaceTracking as DecompInterfaceTracking
|
|
import ghidra.app.decompiler.DecompileOptions as DecompileOptions
|
|
import generic.lsh.vector.WeightedLSHCosineVectorFactory as WeightedLSHCosineVectorFactory
|
|
import ghidra.query.GenSignatures as GenSignatures
|
|
import ghidra.xml.NonThreadedXmlPullParserImpl as NonThreadedXmlPullParserImpl
|
|
import ghidra.util.xml.SpecXmlUtils as SpecXmlUtils
|
|
|
|
|
|
def processFunction(func):
|
|
decompiler = DecompInterfaceTracking()
|
|
options = DecompileOptions()
|
|
decompiler.setOptions(options)
|
|
decompiler.toggleSyntaxTree(False)
|
|
decompiler.setSignatureSettings(getSettings())
|
|
if not decompiler.openProgram(currentProgram):
|
|
print "Unable to initialize the Decompiler interface!"
|
|
print "%s" % decompiler.getLastMessage()
|
|
return
|
|
language = currentProgram.getLanguage()
|
|
sigres = decompiler.debugSignatures(func,10,None)
|
|
for i,res in enumerate(sigres):
|
|
buf = java.lang.StringBuffer()
|
|
sigres.get(i).printRaw(language,buf)
|
|
print "%s" % buf.toString()
|
|
decompiler.closeProgram()
|
|
decompiler.dispose()
|
|
|
|
def getSettings():
|
|
vectorFactory = WeightedLSHCosineVectorFactory()
|
|
id = currentProgram.getLanguageID()
|
|
defaultWeightsFile = GenSignatures.getWeightsFile(id,id)
|
|
input = defaultWeightsFile.getInputStream()
|
|
parser = NonThreadedXmlPullParserImpl(input,"Vector weights parser", SpecXmlUtils.getXmlHandler(),False)
|
|
vectorFactory.readWeights(parser)
|
|
input.close()
|
|
return vectorFactory.getSettings()
|
|
|
|
func = currentProgram.getFunctionManager().getFunctionContaining(currentAddress)
|
|
if func is None:
|
|
print "no function at current address"
|
|
else:
|
|
processFunction(func)
|