class ToolBag:
def __init__(self, options):
self.options = options
self.provider = ida.IDA()
# show splash screen
#
# this will only really be shown when toolbag is loaded for the first time
# as when we initialize our PluginForm-subclassed object (UI), it wipes it
if options['show_splash']:
splash_image = options['ida_user_dir'] + os.sep + "rsrc" + os.sep + "splash.png"
pixmap = QtGui.QPixmap(splash_image)
splash = QtGui.QSplashScreen(pixmap)
splash.show()
if options['file_system_type'] == 'netnode':
from fs_nn import FS
elif options['file_system_type'] == 'segment':
from fs import FS
else:
print "[!] Invalid 'file_system_type' specified in the config options"
return
# file system must be initialized first as db requires it
self.filesystem = FS(options)
print '[*] __init__.py: loading .DB from disk'
# try to see if its on disk already
try:
fh = open(options['full_file_name'], 'rb')
self.master = pickle.load(fh)
fh.close()
except Exception as detail:
print detail
print "[*] Creating a new database file on disk"
self.master = RefTree.MasterRefTree(options)
self.reftree = RefTree.RefTree(masterGraph=self.master)
self.ui = UI(self.reftree, self.filesystem, self.master, options)
self.ui.global_hook = self
self.retvals = {}
self.ui.Show("Toolbag")
if options['show_splash']:
try:
_script = __import__("utils")
except Exception as detail:
print detail
def getContext(self):
res = {}
res["ui"] = self.ui
res["master"] = self.master
res["fs"] = self.filesystem
res["reftree"] = self.reftree
return res
def reanalyze(self):
self.master = RefTree.MasterRefTree(self.options)
self.reftree = RefTree.RefTree(masterGraph=self.master, function_data=self.reftree.function_data)
self.ui.reftree = self.reftree
self.ui.master = self.master
import wx
from ui import UI
import logging
logging.basicConfig(filename="app.log",
filemode="w",
level=logging.INFO,
format='%(asctime)s - %(message)s',
datefmt='%m/%d/%Y %I:%M:%S %p')
logging.info("Running: ")
try:
if __name__ == "__main__":
app = wx.App()
frm = UI()
frm.Show()
app.MainLoop()
except Exception as e:
logging.error(f"main_ui, {e}")
