def test_get_dirs(self):
     test_dirs = ["dir_1", "dir_2", "dir_1/sub_dir"]
     for item in test_dirs:
         os.mkdir(os.path.join(self.tmp_dir.name, item))
     result = get_dirs_in_dir(self.tmp_dir.name)
     self.assertEqual(sorted(result), [
         os.path.join(self.tmp_dir.name, "dir_1"),
         os.path.join(self.tmp_dir.name, "dir_2")
     ], "found dirs not correct")
Ejemplo n.º 2
0
def _get_plugin_src_dirs(base_dir):
    plug_in_base_path = Path(get_src_dir(), base_dir)
    plugin_dirs = get_dirs_in_dir(str(plug_in_base_path))
    plugins = []
    for plugin_path in plugin_dirs:
            plugin_code_dir = Path(plugin_path, 'code')
            if plugin_code_dir.is_dir():
                plugins.append(str(plugin_code_dir))
            else:
                logging.warning('Plugin has no code directory: {}'.format(plugin_path))
    return plugins
Ejemplo n.º 3
0
def main():
    _create_signature_dir()
    for plugin_dir in get_dirs_in_dir(os.path.join(get_src_dir(), 'plugins/analysis')):
        signature_dir = os.path.join(plugin_dir, 'signatures')
        if os.path.isdir(signature_dir):
            print('Compile signatures in {}'.format(signature_dir))
            with NamedTemporaryFile(mode='w') as tmp_file:
                _create_joint_signature_file(signature_dir, tmp_file)
                _create_compiled_signature_file(signature_dir, tmp_file)

    return 0
Ejemplo n.º 4
0
def _get_plugin_src_dirs(base_dir):
    plug_in_base_path = Path(get_src_dir(), base_dir)
    plugin_dirs = get_dirs_in_dir(str(plug_in_base_path))
    plugins = []
    for plugin_path in plugin_dirs:
        # Ignore cache directories
        if Path(plugin_path).name == '__pycache__':
            continue

        plugin_code_dir = Path(plugin_path, 'code')
        if plugin_code_dir.is_dir():
            plugins.append(str(plugin_code_dir))
        else:
            logging.warning(
                'Plugin has no code directory: {}'.format(plugin_path))
    return plugins
Ejemplo n.º 5
0
def _get_plugin_src_dirs(base_dir: str) -> List[str]:
    '''
    Returns a list of all plugin code directories.
    E.g. if base_dir contains the qemu_exec plugin it would return
    `base_dir`/qemu_exec/code.

    :param base_dir: The root directory of all plugins
    '''
    plug_in_base_path = Path(get_src_dir(), base_dir)
    plugin_dirs = get_dirs_in_dir(str(plug_in_base_path))
    plugins = []
    for plugin_path in plugin_dirs:
        if plugin_path.endswith('__pycache__'):
            continue
        plugin_code_dir = Path(plugin_path, 'code')
        if plugin_code_dir.is_dir():
            plugins.append(str(plugin_code_dir))
        else:
            logging.warning(
                'Plugin has no code directory: {}'.format(plugin_path))
    return plugins
 def test_get_dirs_in_dir_error(self):
     result = get_dirs_in_dir("/none_existing/dir")
     self.assertEqual(result, [], "error result should be an empty list")