Edit file

File name : oem-getlogs

Content :#!/usr/bin/python3 from apport import hookutils from glob import glob from io import BytesIO from problem_report import CompressedValue import apport import os import re import shutil import subprocess import sys import tempfile import time import zipfile opt_debug = False # Apport helper routines def debug(text): if opt_debug: print("%s\n" % (text)) def attach_command_output(report, command_list, key): debug("%s" % (' '.join(command_list))) log = hookutils.command_output(command_list) if not log or log[:5] == "Error": return report[key] = log def attach_pathglob_as_zip(report, pathglob, key, data_filter=None, type="b"): """Use zip file here because tarfile module in linux can't properly handle file size 0 with content in /sys directory like edid file. zipfile module works fine here. So we use it. type: a: for ascii type of data b: for binary type of data """ if data_filter is None: data_filter = lambda x: x filelist = [] for pg in pathglob: for file in glob(pg): filelist.append(file) zipf = BytesIO() with zipfile.ZipFile(zipf, mode='w', compression=zipfile.ZIP_DEFLATED) as \ zipobj: for f in filelist: if opt_debug: print(key, f) if not os.path.isfile(f): if opt_debug: print(f, "is not a file") continue if type == "a": with open(f) as f_fd: data = f_fd.read() zipobj.writestr(f, data_filter(data)) else: zipobj.write(f) cvalue = CompressedValue() cvalue.set_value(zipf.getbuffer()) report[key + ".zip"] = cvalue def attach_nvidia_debug_logs(report, keep_locale=False): # check if nvidia-bug-report.sh exists nv_debug_command = 'nvidia-bug-report.sh' if shutil.which(nv_debug_command) is None: if opt_debug: print(nv_debug_command, "does not exist.") return env = os.environ.copy() if not keep_locale: env['LC_MESSAGES'] = 'C' # output result to temp directory nv_tempdir = tempfile.mkdtemp() nv_debug_file = 'nvidia-bug-report' nv_debug_fullfile = os.path.join(nv_tempdir, nv_debug_file) nv_debug_cmd = [nv_debug_command, '--output-file', nv_debug_fullfile] try: with open(os.devnull, 'w') as devnull: subprocess.run(nv_debug_cmd, env=env, stdout=devnull, stderr=devnull) nv_debug_fullfile_gz = nv_debug_fullfile + ".gz" hookutils.attach_file_if_exists(report, nv_debug_fullfile_gz, 'nvidia-bug-report.gz') os.unlink(nv_debug_fullfile_gz) os.rmdir(nv_tempdir) except OSError as e: print("Error:", str(e)) print("Fail on cleanup", nv_tempdir, ". Please file a bug for it.") def dot(): print(".", end="", flush=True) def build_packages(): # related packages packages = ['apt', 'grub2'] # display packages.append('xorg') packages.append('gnome-shell') # audio packages.append('alsa-base') # hotkey and hotplugs packages.append('udev') # networking issues packages.append('network-manager') return packages def helper_url_credential_filter(string_with_urls): return re.sub(r"://\w+?:\w+?@", "://USER:SECRET@", string_with_urls) def add_info(report): # Check if the DCD file is exist in the installer. attach_command_output(report, ['ubuntu-report', 'show'], 'UbuntuReport') dot() hookutils.attach_file_if_exists(report, '/etc/buildstamp', 'BuildStamp') dot() attach_pathglob_as_zip(report, ['/sys/firmware/acpi/tables/*', '/sys/firmware/acpi/tables/*/*'], "acpitables") dot() # Basic hardare information hookutils.attach_hardware(report) dot() hookutils.attach_wifi(report) dot() hwe_system_commands = {'lspci--xxxx': ['lspci', '-xxxx'], 'lshw.json': ['lshw', '-json', '-numeric'], 'dmidecode': ['dmidecode'], 'fwupdmgr_get-devices': ['fwupdmgr', 'get-devices', '--show-all-devices', '--no-unreported-check'], 'boltctl-list': ['boltctl', 'list'], 'mokutil---sb-state': ['mokutil', '--sb-state'], 'tlp-stat': ['tlp-stat'] } for name in hwe_system_commands: attach_command_output(report, hwe_system_commands[name], name) dot() # More audio related hookutils.attach_alsa(report) dot() audio_system_commands = {'pactl-list': ['pactl', 'list'], 'aplay-l': ['aplay', '-l'], 'aplay-L': ['aplay', '-L'], 'arecord-l': ['arecord', '-l'], 'arecord-L': ['arecord', '-L'] } for name in audio_system_commands: attach_command_output(report, audio_system_commands[name], name) dot() attach_pathglob_as_zip(report, ['/usr/share/alsa/ucm/*/*'], "ALSA-UCM") dot() # FIXME: should be included in xorg in the future gfx_system_commands = {'glxinfo': ['glxinfo'], 'xrandr': ['xrandr'], 'xinput': ['xinput'] } for name in gfx_system_commands: attach_command_output(report, gfx_system_commands[name], name) dot() attach_pathglob_as_zip(report, ['/sys/devices/*/*/drm/card?/*/edid'], "EDID") dot() # nvidia-bug-reports.sh attach_nvidia_debug_logs(report) dot() # FIXME: should be included in thermald in the future attach_pathglob_as_zip(report, ["/etc/thermald/*", "/sys/devices/virtual/thermal/*", "/sys/class/thermal/*"], "THERMALD") dot() # all kernel and system messages attach_pathglob_as_zip(report, ["/var/log/*", "/var/log/*/*"], "VAR_LOG") dot() # apt configs attach_pathglob_as_zip(report, [ "/etc/apt/apt.conf.d/*", "/etc/apt/sources.list", "/etc/apt/sources.list.d/*.list", "/etc/apt/preferences.d/*"], "APT_CONFIGS", type="a", data_filter=helper_url_credential_filter) dot() # TODO: debug information for suspend or hibernate # packages installed. attach_command_output(report, ['dpkg', '-l'], 'dpkg-l') dot() # FIXME: should be included in bluez in the future attach_command_output(report, ['hciconfig', '-a'], 'hciconfig-a') dot() # FIXME: should be included in dkms in the future attach_command_output(report, ['dkms', 'status'], 'dkms_status') dot() # enable when the feature to include data from package hooks exists. # packages = build_packages() # attach_related_packages(report, packages) if __name__ == '__main__': from argparse import ArgumentParser import gzip parser = ArgumentParser(prog="oem-getlogs", usage="Useage: sudo -E oem-getlogs [-c CASE_ID]", description="Get Hardware Enablement related logs") parser.add_argument("-c", "--case-id", help="optional CASE_ID", dest="cid", default="") args = parser.parse_args() # check if we got root permission if os.geteuid() != 0: print("Error: you need to run this program as root") parser.print_help() sys.exit(1) print("Start to collect logs: ", end="", flush=True) # create report report = apport.Report() add_info(report) # generate filename hostname = os.uname()[1] date_time = time.strftime("%Y%m%d%H%M%S%z", time.localtime()) filename_lst = ["oemlogs", hostname] if (len(args.cid) > 0): filename_lst.append(args.cid) filename_lst.append(date_time + ".apport.gz") filename = "-".join(filename_lst) with gzip.open(filename, 'wb') as f: report.write(f) print("\nSaved log to", filename) print("The owner of the file is root. You might want to") print(" chown [user].[group]", filename)

Save