p f(pddlZddlZddlZddlZddlmcmZddlmcmcm Z ddl m Z Gdde Z y)N) RDMAHandlerceZdZdZdZdZdZdZdZdZ dZ dZ d Z e d Zd Zd Zd Ze dZe dZdZdZe dZdZe dZdZdZy)CentOSRDMAHandlerzmicrosoft-hyper-v-rdmazkmod-microsoft-hyper-v-rdmazmsft-rdma-drivers hypervkvpdzmicrosoft-hyper-vNc|jd}t|dkrtd|z|d|dc|_|_y)N.zUnexpected centos version: %sr)splitlen Exception version_major version_minor)selfdistro_versionvs @/usr/lib/python3/dist-packages/azurelinuxagent/pa/rdma/centos.py__init__zCentOSRDMAHandler.__init__(sH   % q6A:;nLM M121qt.D.ctjdtj}t j d|z|s|j tjd|j}|s tdt j dj||j|}|j}|rst j dj||j||rt j dyt j d|j|yt j d |j|y) zo Install the KVP daemon and the appropriate RDMA driver package for the RDMA firmware. zRDMA: kvp daemon running: %sz&Cannot determine RDMA firmware versionz!RDMA: found firmware version: {0}z!RDMA: driver package present: {0}z"RDMA: driver package is up-to-dateNz#RDMA: driver package needs updatingz%RDMA: driver package is NOT installed)timesleepris_kvp_daemon_runningloggerinfocheck_or_install_kvp_daemonget_rdma_versionr formatget_int_rdma_versionget_rdma_package_infois_rdma_package_up_to_dateupdate_rdma_package)r kvpd_running fw_version installed_pkgs rinstall_driverz CentOSRDMAHandler.install_driver.s 2"88:  2\AB  , , . 2**, DE E 7>>zJK..z: 224  KK3::=I K..}jI @A AB((4 KK? @  $ $Z 0rcfdj|j|}tj||S)Nz{0}-(\d+\.){{3,}}({1})-)rrdma_user_mode_package_namerematch)rpkgr%patterns rr"z,CentOSRDMAHandler.is_rdma_package_up_to_dateQs.-33D4T4TV`axx%%rcf|jd}t|dk(rtd|z|dS)Nrrz&Unexpected RDMA firmware version: "%s")r r r )versionss rr z&CentOSRDMAHandler.get_int_rdma_versionXs5 MM#  q6Q;DwNO Ot rc`tjd|jzd\}}|dk7ry|S)zA Returns the installed rdma package name or None rpm -q %sFchk_errrN) shellutilrun_get_outputr))rretoutputs rr!z'CentOSRDMAHandler.get_rdma_package_info_s8 .. $:: :EK V !8 rctjd|j|j|j|j |y)NzRDMA: updating RDMA packages)rr refresh_reposforce_install_packagerdma_wrapper_package_nameinstall_rdma_drivers)rr%s rr#z%CentOSRDMAHandler.update_rdma_packageis= 23  ""4#A#AB !!*-rctjd|z|j|dk7rtjd|j|dk7rt dj |tjd|zy)zN Attempts to remove existing package and installs the package z"RDMA: Force installing package: %srz.RDMA: Erasing package failed but will continuezFailed to install package "{0}"z RDMA: installation completed: %sN)rruninstall_packageinstall_packager r)rpkg_names rr;z'CentOSRDMAHandler.force_install_packageosr  88CD  ! !( +q 0 KKH I    )Q .=DDXNO O 6ABrcJtjdj|S)Nzyum erase -y -q {0}r5runrrAs rr?z#CentOSRDMAHandler.uninstall_packagezs}}299(CDDrcJtjdj|S)Nzyum install -y -q {0}rCrEs rr@z!CentOSRDMAHandler.install_package~s}}4;;HEFFrctjdtjddk7r t dtjddk7r t dtjdy)NzRDMA: refreshing yum reposz yum clean allrz Cleaning yum repositories failedzyum updateinfoz,Failed to act on yum repo update informationzRDMA: repositories refreshed)rrr5rDr )rs rr:zCentOSRDMAHandler.refresh_repossW 01 == )Q .>? ? ==) *a /JK K 23rcdj|j|j}tjdj|t j j|std|zt j|}tjdt|z|j|j||jd|d}|s tdt j j||}|jd|j||j||j d|d}|s td t j j||}|jd |j |tjd |j#r|j%s&tjd |j'ytjd y)z Installs the drivers from /opt/rdma/rhel[Major][Minor] directory, particularly the microsoft-hyper-v-rdma-* kmod-* and (no debuginfo or src). Tries to uninstall them first. z/opt/microsoft/rdma/rhel{0}{1}zRDMA: pkgs dir: {0}z%RDMA packages directory %s is missingz)RDMA: found %d files in package directoryz -(\d+\.){3,}(z)-\d{8}\.x86_64.rpmz"RDMA kernel mode package not foundz kernel modez RDMA user mode package not foundz user modezRDMA: driver packages installedz2RDMA: driver module is not loaded; reboot requiredzRDMA: kernel module is loadedN)rrrrrospathisdirr listdirr &uninstall_kvp_driver_package_if_existsget_file_by_patternrdma_kernel_mode_package_namejoinuninstall_pkg_and_install_fromr)load_driver_moduleis_driver_loaded reboot_system)rr%pkg_dirpkgskmod_pkg kmod_pkg_pathumod_pkg umod_pkg_paths rr=z&CentOSRDMAHandler.install_rdma_driverss 399    2 24 )009:ww}}W%CgMN Nzz'" ?#d)KL 335++ T=_=_akln@A A Wh7  ++ 4==} N++ T=]=]_ijl>? ? Wh7  ++ 99= J  56&&(0E0E0G KKL M    KK7 8rcF|D]}tj||s|cSy)N)r*r+) file_listr-ls rrNz%CentOSRDMAHandler.get_file_by_patterns) Axx# rc|tjdj||tjd|z|j|dk(rtjd|ztjdj|||j |dk7rt dj|y)Nz RDMA: Processing {0} driver: {1}z+RDMA: Try to uninstall existing version: %srz RDMA: Successfully uninstaled %sz%RDMA: Installing {0} package from {1}z"Failed to install RDMA {0} package)rrrr?r@r )rpkg_typerApkg_paths rrQz0CentOSRDMAHandler.uninstall_pkg_and_install_froms . 5 5h I K AHLM  ! !( +q 0 KK:XE F 3 : :8X N P    )Q .4;;HEG G /rc<tjd|zddk(S)zPRuns rpm -q and checks return code to find out if a package is installedr2Fr3r)r5rD)r,s ris_package_installedz&CentOSRDMAHandler.is_package_installeds}}[3.>!CCrctjd|j|jg}|D]}|j |stjd|z-tjd|zt j d|zddk(rtjdxtjd y) Nz+RDMA: deleting existing kvp driver packagesz-RDMA: kvp package %s does not exist, skippingzRDMA: erasing kvp package "%s"zyum erase -q -y %sFr3rz!RDMA: successfully erased packagezRDMA: failed to erase package)rrhyper_v_package_namehyper_v_package_name_newrbr5rDerror)rkvp_pkgskvp_pkgs rrMz8CentOSRDMAHandler.uninstall_kvp_driver_package_if_existss AB--113 BG,,W5 CgMO ASST    2 3q 8MPbbc c ?BTTU 9: r)__name__ __module__ __qualname__r)rOr<rdrerrrr'r" staticmethodr r!r#r;r?r@r:r=rNrQrbrMrrrrrs":$A! 3'2MM< !1F& . CEEGG4)9V GDD B"rr) globrIr*razurelinuxagent.common.loggercommonr&azurelinuxagent.common.utils.shellutilutilsr5azurelinuxagent.common.rdmarrrprrrws0( ..::3W Wr