gitea-mirror/capa
1
0
Fork 0
mirror of https://github.com/mandiant/capa.git synced 2025-12-21 23:00:29 -08:00
Code Issues Packages Projects Releases Wiki Activity

ida: use ida-settings to persist rules directory

Browse Source 
closes #268
This commit is contained in:
Willi Ballenthin
2020-09-01 16:12:50 -06:00
parent 2bd0c03f70
commit b611ddeb6e
2 changed files with 16 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
capa/ida/plugin/form.py
View File

@@ -12,6 +12,7 @@ import logging
import collections import collections
import idaapi import idaapi
import ida_settings
from PyQt5 import QtGui, QtCore, QtWidgets from PyQt5 import QtGui, QtCore, QtWidgets
import capa.main import capa.main
@@ -25,6 +26,7 @@ from capa.ida.plugin.model import CapaExplorerDataModel
from capa.ida.plugin.proxy import CapaExplorerSortFilterProxyModel from capa.ida.plugin.proxy import CapaExplorerSortFilterProxyModel
logger = logging.getLogger("capa") logger = logging.getLogger("capa")
settings = ida_settings.IDASettings("capa")
class CapaExplorerForm(idaapi.PluginForm): class CapaExplorerForm(idaapi.PluginForm):
@@ -303,12 +305,16 @@ class CapaExplorerForm(idaapi.PluginForm):
def load_capa_results(self): def load_capa_results(self):
""" run capa analysis and render results in UI """ """ run capa analysis and render results in UI """
if not self.rule_path: if not self.rule_path:
rule_path = self.ask_user_directory() if "rule_path" in settings:
if not rule_path: self.rule_path = settings["rule_path"]
capa.ida.helpers.inform_user_ida_ui("You must select a rules directory to use for analysis.") else:
logger.warning("no rules directory selected. nothing to do.") rule_path = self.ask_user_directory()
return if not rule_path:
self.rule_path = rule_path capa.ida.helpers.inform_user_ida_ui("You must select a rules directory to use for analysis.")
logger.warning("no rules directory selected. nothing to do.")
return
self.rule_path = rule_path
settings.user["rule_path"] = rule_path
logger.info("-" * 80) logger.info("-" * 80)
logger.info(" Using rules from %s." % self.rule_path) logger.info(" Using rules from %s." % self.rule_path)
@@ -520,6 +526,9 @@ class CapaExplorerForm(idaapi.PluginForm):
if not rule_path: if not rule_path:
logger.warning("no rules directory selected. nothing to do.") logger.warning("no rules directory selected. nothing to do.")
return return
self.rule_path = rule_path self.rule_path = rule_path
settings.user["rule_path"] = rule_path
if 1 == idaapi.ask_yn(1, "Run analysis now?"): if 1 == idaapi.ask_yn(1, "Run analysis now?"):
self.reload() self.reload()

2
setup.py
View File

@@ -12,7 +12,7 @@ import sys
import setuptools import setuptools
# halo==0.0.30 is the last version to support py2.7 # halo==0.0.30 is the last version to support py2.7
requirements = ["six", "tqdm", "pyyaml", "tabulate", "colorama", "termcolor", "ruamel.yaml", "wcwidth", "halo==0.0.30"] requirements = ["six", "tqdm", "pyyaml", "tabulate", "colorama", "termcolor", "ruamel.yaml", "wcwidth", "halo==0.0.30", "ida-settings=1.0.1"]
if sys.version_info >= (3, 0): if sys.version_info >= (3, 0):
# py3 # py3