binwalk.py

# Run binwalk on the current file and create bookmarks and EOL comments for findings.
#@author Thomas Roth code@stacksmashing.net
#@category Ghidra Ninja
#@keybinding 
#@menupath 
#@toolbar 


import subprocess
import tempfile
import os
import csv
from ghidra.program.model.listing import CodeUnit

def add_bookmark_comment(addr, text):
	cu = currentProgram.getListing().getCodeUnitAt(addr)
	createBookmark(addr, "binwalk", text)
	cu.setComment(CodeUnit.EOL_COMMENT, text)

file_location = currentProgram.getDomainFile().getMetadata()["Executable Location"]
_, result_file = tempfile.mkstemp()

try:
	subprocess.call(["binwalk", "-c", "-f", result_file, file_location])
	with open(result_file) as csvfile:
		reader = csv.reader(csvfile, delimiter=',', quotechar='"')
		for row in reader:
			try:
				addr = currentProgram.minAddress.add(int(row[0]))
			except:
				continue

			text = row[2]
			add_bookmark_comment(addr, text)
except Exception as e:
	print("Failed")
	print(e)

os.unlink(result_file)