def bapida(idapatch, popenpatch, monkeypatch, idadir):
from bap.utils import config
ida = Ida()
bap = Bap(BAP_PATH)
def run_bap(args):
if args[0] == BAP_PATH:
res = bap.call(args) or 0
if isinstance(res, int):
return 'exit ' + str(res)
else:
return str(res)
config.set('bap_executable_path', bap.path)
monkeypatch.setattr('os.access', lambda p, x: p == BAP_PATH)
idapatch({
'register_timer': ida.register_timer,
'get_input_ida_file_path': lambda: 'true'
})
idapatch(ns='idc',
attrs={
'Message': ida.message,
'Warning': ida.warning,
'SetStatus': ida.set_status,
})
popenpatch(run_bap)
monkeypatch.setattr('bap.utils.ida.output_symbols', lambda out: None)
return (bap, ida)
def bapida(idapatch, popenpatch, monkeypatch, idadir):
from bap.utils import config
ida = Ida()
bap = Bap(BAP_PATH)
def run_bap(args):
if args[0] == BAP_PATH:
res = bap.call(args) or 0
if isinstance(res, int):
return 'exit ' + str(res)
else:
return str(res)
config.set('bap_executable_path', bap.path)
monkeypatch.setattr('os.access', lambda p, x: p == BAP_PATH)
idapatch({
'register_timer': ida.register_timer,
'get_input_ida_file_path': lambda: 'true'
})
idapatch(ns='idc', attrs={
'Message': ida.message,
'Warning': ida.warning,
'SetStatus': ida.set_status,
})
popenpatch(run_bap)
monkeypatch.setattr('bap.utils.ida.output_symbols', lambda out: None)
return (bap, ida)
def test_set_and_get(idadir):
from bap.utils.config import get, set, is_set
for path in ('foo', 'foo.bar'):
assert get(path) is None
set(path, 'hello')
assert get(path) == 'hello'
assert not is_set(path)
set(path, 'true')
assert is_set(path)
def config_path():
if config.get('bap_executable_path') is not None:
return
default_bap_path = ''
from subprocess import check_output, CalledProcessError
import os
try:
default_bap_path = check_output(['which', 'bap']).strip()
except (OSError, CalledProcessError) as e:
# Cannot run 'which' command OR
# 'which' could not find 'bap'
try:
default_bap_path = os.path.join(
check_output(['opam', 'config', 'var', 'bap:bin']).strip(),
'bap'
)
except OSError:
# Cannot run 'opam'
pass
if not default_bap_path.endswith('bap'):
default_bap_path = ''
def confirm(msg):
from idaapi import askyn_c, ASKBTN_CANCEL, ASKBTN_YES
return askyn_c(ASKBTN_CANCEL, msg) == ASKBTN_YES
while True:
bap_path = idaapi.askfile_c(False, default_bap_path, 'Path to bap')
if bap_path is None:
if confirm('Are you sure you don\'t want to set path?'):
return
else:
continue
if not bap_path.endswith('bap'):
if not confirm("Path does not end with bap. Confirm?"):
continue
if not os.path.isfile(bap_path):
if not confirm("Path does not point to a file. Confirm?"):
continue
break
config.set('bap_executable_path', bap_path)
def config_path():
if config.get('bap_executable_path') is not None:
return
default_bap_path = ''
from subprocess import check_output, CalledProcessError
import os
try:
default_bap_path = check_output(['which', 'bap']).strip()
except (OSError, CalledProcessError) as e:
# Cannot run 'which' command OR
# 'which' could not find 'bap'
try:
default_bap_path = os.path.join(
check_output(['opam', 'config', 'var', 'bap:bin']).strip(),
'bap')
except OSError:
# Cannot run 'opam'
pass
if not default_bap_path.endswith('bap'):
default_bap_path = ''
def confirm(msg):
from idaapi import askyn_c, ASKBTN_CANCEL, ASKBTN_YES
return askyn_c(ASKBTN_CANCEL, msg) == ASKBTN_YES
while True:
bap_path = idaapi.askfile_c(False, default_bap_path, 'Path to bap')
if bap_path is None:
if confirm('Are you sure you don\'t want to set path?'):
return
else:
continue
if not bap_path.endswith('bap'):
if not confirm("Path does not end with bap. Confirm?"):
continue
if not os.path.isfile(bap_path):
if not confirm("Path does not point to a file. Confirm?"):
continue
break
config.set('bap_executable_path', bap_path)
def check_and_configure_bap():
"""Ensures that bap location is known."""
if not config.get('bap_executable_path'):
path = ask_user(bap.find())
if path and len(path) > 0:
config.set('bap_executable_path', path)
def config_bap_api():
if config.get('enabled', section='bap_api') is None:
config.set('enabled', '1', section='bap_api')
def config_bap_api():
if config.get('enabled', section='bap_api') is None:
config.set('enabled', '1', section='bap_api')
def check_and_configure_bap():
"""Ensures that bap location is known."""
if not config.get('bap_executable_path'):
path = ask_user(bap.find())
if path and len(path) > 0:
config.set('bap_executable_path', path)