def validate_gbs_file(self, gbs_file): file = open(gbs_file, 'rb') gbs = file.read() if len(constants.METADATA_GUID) >= len(gbs): raise Exception("Invalid GBS file") self.guid = gbs[:constants.GUID_LEN] if self.guid != constants.METADATA_GUID: raise Exception("Unsupported GBS format") metadata_index = constants.GUID_LEN + constants.SIZEOF_LEN_FIELD metadata_len = struct.unpack("<I", gbs[constants.GUID_LEN:metadata_index]) self.metadata_len = metadata_len[0] if self.metadata_len != 0: self.gbs_info = json.loads(gbs[metadata_index:(metadata_index + self.metadata_len)]) rbf_index = metadata_index + metadata_len[0] if rbf_index == len(gbs): raise Exception("No RBF in GBS file!") self.rbf = gbs[rbf_index:] self.metadata = metadata.get_metadata(self.gbs_info)
def validate_gbs_file(self, gbs_file): file = open(gbs_file, 'rb') gbs = file.read() if len(constants.METADATA_GUID) >= len(gbs): raise Exception("Invalid GBS file") self.guid = gbs[:constants.GUID_LEN] if self.guid != constants.METADATA_GUID: raise Exception("Unsupported GBS format") metadata_index = constants.GUID_LEN + constants.SIZEOF_LEN_FIELD metadata_len = struct.unpack( "<I", gbs[constants.GUID_LEN:metadata_index]) self.metadata_len = metadata_len[0] if self.metadata_len != 0: self.gbs_info = json.loads( gbs[metadata_index:(metadata_index + self.metadata_len)]) rbf_index = metadata_index + metadata_len[0] if rbf_index == len(gbs): raise Exception("No RBF in GBS file!") self.rbf = gbs[rbf_index:] self.metadata = metadata.get_metadata(self.gbs_info)
def test_get_metadata(get_os, is_freebsd, is_linux, config_get, time_time, uuid_getnode, platform_node): res = { 'agentVersion': '0.99.99', 'apiKey': 'my_api_key', 'uuid': '9ef19c3f1a4c5493825695cd864dc2c3', 'internalHostname': 'test', 'os': 'test_os', 'python': sys.version, 'systemStats': { 'machine': platform.machine(), 'platform': sys.platform, 'processor': platform.processor(), 'pythonV': platform.python_version(), 'cpuCores': psutil.cpu_count(), }, 'meta': { 'socket-hostname': socket.gethostname(), 'timezones': time.tzname, 'host_aliases': [], 'socket-fqdn': socket.getfqdn(), 'hostname': 'test', }, 'host-tags': {'system': ['tag1', 'tag2']}, 'resources': { 'meta': {'host': 'test'}, 'processes': {'snaps': []}, }, 'events': {}, } assert res == metadata.get_metadata('test', '0.99.99') res['events'].update({ 'System': [ { 'api_key': 'my_api_key', 'event_type': 'Agent Startup', 'host': 'test', 'msg_text': 'Version 0.99.99', 'timestamp': 1234567890.12 } ] }) assert res == metadata.get_metadata('test', '0.99.99', start_event=True)
def create_gbs_from_afu_info(cls, rbf_file, afu_json): gbs = cls() rbf = open(rbf_file, 'rb') rbf_content = rbf.read() gbs.guid = constants.METADATA_GUID gbs.metadata_len = len(afu_json) gbs.gbs_info = afu_json gbs.rbf = rbf_content gbs.metadata = metadata.get_metadata(afu_json) gbs.filename = os.path.splitext(os.path.basename(rbf_file))[0] return gbs
def create_gbs_from_afu_info(cls, rbf_file, afu_json): gbs = cls() rbf = open(rbf_file, 'rb') rbf_content = rbf.read() gbs.guid = constants.METADATA_GUID gbs.metadata_len = len(afu_json) gbs.gbs_info = afu_json gbs.rbf = rbf_content gbs.metadata = metadata.get_metadata(afu_json) gbs.filename = os.path.splitext(os.path.basename(rbf_file))[0] return gbs
def test_get_metadata(get_os, is_freebsd, is_linux, config_get, uuid_getnode, platform_node): res = { "agentVersion": "6.0.0", "apiKey": "my_api_key", "uuid": "9ef19c3f1a4c5493825695cd864dc2c3", "internalHostname": "test", "os": "test_os", "python": sys.version, "systemStats": { "machine": platform.machine(), "platform": sys.platform, "processor": platform.processor(), "pythonV": platform.python_version(), "cpuCores": psutil.cpu_count(), }, "meta": { "socket-hostname": socket.gethostname(), "timezones": time.tzname, "host_aliases": [], "socket-fqdn": socket.getfqdn(), "hostname": "test", }, "host-tags": { "system": ["tag1", "tag2"] }, "resources": { "meta": { "host": "test" }, "processes": { "snaps": [] }, }, } assert res == metadata.get_metadata("test")
def update_gbs_info(self, gbs_info): self.gbs_info = gbs_info self.metadata = metadata.get_metadata(self.gbs_info)
def update_gbs_info(self, gbs_info): self.gbs_info = gbs_info self.metadata = metadata.get_metadata(self.gbs_info)