def GetWMIData(): if WMI_SW: return WMI_SW # Load some wmi data parser = wmi_parser.WMIInstalledSoftwareParser() test_data = os.path.join(CHECKS_DIR, "data/wmi_sw.yaml") with open(test_data, "rb") as f: wmi = yaml.safe_load(f) for sw in wmi: WMI_SW.extend(parser.Parse(None, sw, None))
from grr.parsers import wmi_parser CHECKS_DIR = os.path.join(config_lib.CONFIG["Test.data_dir"], "checks") TRIGGER_1 = ("DebianPackagesStatus", "Linux", None, None) TRIGGER_2 = ("WMIInstalledSoftware", "Windows", None, None) TRIGGER_3 = ("DebianPackagesStatus", None, None, "foo") # Load some dpkg data parser = linux_cmd_parser.DpkgCmdParser() test_data = os.path.join(config_lib.CONFIG["Test.data_dir"], "dpkg.out") with open(test_data) as f: DPKG_SW = list( parser.Parse("/usr/bin/dpkg", ["-l"], f.read(), "", 0, 5, None)) # Load some wmi data parser = wmi_parser.WMIInstalledSoftwareParser() test_data = os.path.join(config_lib.CONFIG["Test.data_dir"], "wmi_sw.yaml") WMI_SW = [] with open(test_data) as f: wmi = yaml.safe_load(f) for sw in wmi: WMI_SW.extend(parser.Parse(None, sw, None)) # Load an sshd config parser = config_file_parsers.SshdConfigParser() test_data = os.path.join(config_lib.CONFIG["Test.data_dir"], "sshd_config") with open(test_data) as f: SSHD_CFG = list(parser.Parse(None, f, None)) def _LoadCheck(cfg_file, check_id):