o O6uf@s6ddlZddlZddlmZmZGdddeeZdS)N)PluginIndependentPluginc@s8eZdZdZdZdZdZdZddZdd Z d d Z d S) MellanoxFirmwarez&Nvidia(Mellanox) firmware tools outputmellanox_firmware)hardwaresystem)mstmstflintzMellanox TechnologiescCs$|d}|ddko|j|dvS)z Checks if this plugin should be executed at all. We will only enable the plugin if there is a Mellanox Technologies network adapter lspci -D -d 15b3::0200statusroutput)exec_cmd MLNX_STRING)selflspcirF/usr/lib/python3/dist-packages/sos/report/plugins/mellanox_firmware.py check_enableds zMellanoxFirmware.check_enabledcCs|ds |ddS|d}|ddkrdS|d}|ddkr&dS|d|jdd d |jd d d td }g}|D] }d |vrM||qB|D]}|jd|dd d |jd|dd d qP|jddddS)Nallow_system_changesztSkipping mst/mlx cable commands as system changeswould be made. Use --allow-system-changes toenable this collection.flint --versionr rz mst startz mst cable addz mst status -v timeout mlxcablesz/dev/mstcablez mlxcables -d z --DDMz --dumpzmst stopT)changes) get_option _log_infor collect_cmd_outputoslistdirappend)rcoutrdevicemlxcablerrrcollect#s4         zMellanoxFirmware.collectc Cs g}|d}|ddkrdS|dD] }||dddqt|}g}|d}|ddkrFdd gd d gd d gd dgddgg}nddgdd gdd gdd gddgddgddgg}|D].}|D]}|j|dd|d|dddqctdD]}|d|tdq}q_dS)Nr r rr z00.0rz mstconfig -d z -e qz mstflint -d z dcz qz mstreg -d z --reg_name ROCE_ACCL --getz mstlink -d z mlxdump -d z pcie_uc --allz flint -d z mlxreg -d z mlxlink -d z fwtrace -d z -i all --tracer_mode FIFO rzmstdump ) r splitlinesr!setr add_cmd_outputrangetimesleep) rdevices device_listlinecommandsr"r#command_rrrsetupCsH        zMellanoxFirmware.setupN) __name__ __module__ __qualname__ short_desc plugin_nameprofilespackagesrrr%r8rrrrrs r)rr0sos.report.pluginsrrrrrrrs