3 J_@sdZddlZddlZddlZddlZddlZddlZddlZddlm Z ej j dZ ia dIddZddZdd Zd d Zie_d d ZddZddZddZddZddZddZdJddZddZd d!ZdKd"d#ZdLd$d%ZdMd&d'ZGd(d)d)e Z!d*d+Z"d,d-Z#dNd/d0Z$dOd1d2Z%d3d4Z&d5d6Z'd7d8Z(d9d:Z)d;d<Z*d=d>Z+d?d@Z,dAdBZ-dCdDZ.dEdFZ/dGdHZ0dS)PzGHardware and driver package detection functionality for Ubuntu systems.N)kerneldetectionc CsNi}|rd|pd}x2tj|D]"\}}}d}d|kry.ttjj|d}|jj}WdQRXWqtk r}ztj d||w"WYdd}~XqXn\d|krd|krttjj|d4} x,| D]$} | j dr| j d d d j}PqWWdQRX|sq"tjj|d } tjj| d } tjj | r>tjj |  r>q"|||<q"W|S) uGet modaliases present in the system. This ignores devices whose drivers are statically built into the kernel, as you cannot replace them with other driver packages anyway. Return a modalias → sysfs path map. z %s/devicesz /sys/devicesNmodaliasz0system_modaliases(): Cannot read %s/modalias: %sZssbZueventz MODALIAS==drivermodule) oswalkopenpathjoinreadstripIOErrorloggingdebug startswithsplitislink) sys_pathaliasesZdevicesr dirsfilesrfefdlineZ driverlinkZmodlinkr6/usr/lib/python3/dist-packages/UbuntuDrivers/detect.pysystem_modaliasess4   rcCsd}y*x$|djjD]}|jdr|}PqWWn"ttfk rPtjddSX|sZdSy |d}Wntk rzdSXd|kr||krtjd|d|dSdS) Nzxserver-xorg-corezxorg-video-abi-zL_check_video_abi_compat(): xserver-xorg-core not available, cannot check ABITFDependszBDriver package %s is incompatible with current X.org server ABI %sZPackage) candidateZprovidesrAttributeErrorKeyErrorrr) apt_cacherecordZxorg_video_abipdepsrrr_check_video_abi_compatIs(     r(c Csi}x|D]}y|jjd}Wntttfk r>w YnX|j s |jjdtfkrZq t||jjsjq yxz|jdD]l}|j d}|sqx|jd\}}xH|jdD]:}|j }|jddd }|j |ij |t j |j qWqxWWq tk rtjd |j |fYq Xq W|S) aEBuild a modalias map from an apt.Cache object. This filters out uninstallable video drivers (i. e. which depend on a video ABI that xserver-xorg-core does not provide). Return a map bus -> modalias -> [package, ...], where "bus" is the prefix of the modalias up to the first ':' (e. g. "pci" or "usb"). Modaliasesall)z, (,:rrz*Package %s has invalid modalias header: %s)r!r%r#r"UnicodeDecodeError architecturesystem_architecturer(rr setdefaultsetaddname ValueErrorrerror) r$resultpackagempartrZlstaliasZbusrrr_apt_cache_modalias_mapds2  *r=c st}t}ytj|}Wn&tk rBt}|tj|<YnX|j|jdddi}x<|D]4}tj|j |j rbx||D]}|j |qWqbWfdd|DS)z_Search packages which match the given modalias. Return a list of apt.Package objects. r.rrcsg|] }|qSrr).0r&)r$rr sz)packages_for_modalias..) r3hashpackages_for_modalias cache_mapsr#r=getrfnmatchlowerr4)r$rpkgsZapt_cache_hashZ cache_mapZbus_mapr<r&r)r$rrAs rAcCsB|jdk st|jjdrdSx|jjD]}|jdkr(dSq(WdS)NnvidiaF restricted multiverseT)rHrI)r!AssertionErrorr5roriginsZ component)pkgorrr_is_package_frees  rNcCs2|jdkrdSx|jjD]}|jdkrdSqWdS)NFZUbuntuT)r!rKorigin)rLrMrrr_is_package_from_distros   rPc Csly|jjd}Wn&ttfk r6tjd|jdSX|jd}|dkr\tjd|jdS|d|}|S)z-Determine module name from apt Package objectr)zM_pkg_get_module %s: package has no Modaliases header, cannot determine moduleNr,rzR_pkg_get_module %s: package has invalid Modaliases header, cannot determine module) r!r%r#r"rrr5findwarning)rLr:Zparenrrrr_pkg_get_modules  rSc CsXy|jjd}Wn&ttfk r6tjd|jdSX|dkrTtjd|j|dS|S)z/Determine support level from apt Package objectZSupportzR_pkg_get_support %s: package has no Support header, cannot determine support levelNNFBLTSBzY_pkg_get_support %s: package has invalid Support %sheader, cannot determine support level)rTrU)r!r%r#r"rrr5rR)rLsupportrrr_pkg_get_supports rWc s~y|jjd}Wntttfk r*dSX|jddkr>dS||jdd|jd}|jd}tfd d |DSd S) z>Check if the package supports runtimepm for the given modaliasZ PmAliasesFznvidia(rr,rr+z, c3s"|]}tjj|jVqdS)N)rDrE)r>Zregex)r<rr sz*_is_runtimepm_supported..N)r!r%r#r"r/rQrany)rLr<r:n modaliasesr)r<r_is_runtimepm_supporteds r\cCs|jr dS|jjdrd}n|jjdr.d}nt|}|s>dSy$tjd|gtjtjd}|jWn8tt tj fk r}zt j dt |dSd}~Xn X|jdkrt j d |j|d St j d |j|dS) zIDetermine if the kernel module from an apt.Package is manually installed.FrGfglrxmodinfo)stdoutstderrz_is_manual_install failed: %sNrzJ_is_manual_install %s: builds module %s which is available, manual installTzQ_is_manual_install %s: builds module %s which is not available, no manual install) installedr5rrS subprocessPopenPIPEZ communicateOSErrorFileNotFoundErrorCalledProcessErrorrrstr returncode)rLrr^rrrr_is_manual_installs.       rjc Csytjddd|gdd}Wn:ttjfk rV}ztjd||t|dSd}~XnXtjd|d}d}x8|jD],}|jd d \}}d |kr|}d |krv|}qvWtjd ||||||fS)zTReturn (vendor, model) names for given device. Values are None if unknown. ZudevadmZhwdbz--test=T)Zuniversal_newlinesz-_get_db_name(%s, %s): udevadm hwdb failed: %sNz_get_db_name: output %s rrZ_VENDORZ_MODELz-_get_db_name(%s, %s): vendor "%s", model "%s")NN) rbZ check_outputrergrrrh splitlinesr) syspathr<outrvendormodelrkvrrr _get_db_names$   rrFTcCst|}|sDy tj}Wn*tk rB}ztj|iSd}~XnXi}x|jD]\}}xt||D]} |rzt|  rzqf| rt j | j drqf||t| t | t | t | |d|| j <t||\} } | dk r| || j d<| dk rf| || j d<qfWqRWdd|D} | rvx*|jD]\} }| jdr|t| <qW| jtjtd | d }x| D]} | |k|| d <q\WxJt|jD]:\}}x.|D]&} || }t|t ||d || <qWqW|S)uGet driver packages that are available for the system. This calls system_modaliases() to determine the system's hardware and then queries apt about which packages provide drivers for those. It also adds available packages from detect_plugin_packages(). If you already have an apt.Cache() object, you should pass it as an argument for efficiency. If not given, this function creates a temporary one by itself. If freeonly is set to True, only free packages (from main and universe) are considered Return a dictionary which maps package names to information about them: driver_package → {'modalias': 'pci:...', ...} Available information keys are: 'modalias': Modalias for the device that needs this driver (not for drivers from detect plugins) 'syspath': sysfs directory for the device that needs this driver (not for drivers from detect plugins) 'plugin': Name of plugin that detected this package (only for drivers from detect plugins) 'free': Boolean flag whether driver is free, i. e. in the "main" or "universe" component. 'from_distro': Boolean flag whether the driver is shipped by the distro; if not, it comes from a (potentially less tested/trusted) third party source. 'vendor': Human readable vendor name, if available. 'model': Human readable product name, if available. 'recommended': Some drivers (nvidia, fglrx) come in multiple variants and versions; these have this flag, where exactly one has recommended == True, and all others False. Nz oem-*-meta)rrlfree from_distrorVZ runtimepmrnrocSsg|]}|jdr|qS)znvidia-)r)r>r&rrrr?xsz*system_driver_packages..znvidia-)keyr recommended)rsrtplugin)raptCache Exceptionrr7itemsrArNrDr5rPrWr\rrr lookup_cachesort functools cmp_to_key_cmp_gfx_alternativesdetect_plugin_packages)r$rfreeonly include_oemr[expackagesr<rlr&rnronvidia_packagesruvaluervrwrFZapt_prrrsystem_driver_packages4sR%     rcCs@tjd}|j|}|r<|jddd|jdddfSdS)Nz$(.+):v(.+)d(.+)sv(.+)sd(.+)bc(.+)i.*)NN)recompilematchgroup)r<Zmodalias_patternZdetailsrrr_get_vendor_model_from_aliass   $rc Cs|jdk std}|j}tjd}|j|}|r@tjd||Stjd}|j|}|shtjd||Sd|jd|jdf}d|}y(|j |}|jr|jj d t fkr|}Wnt k rYnX|S) Nznvidia-([0-9]+)z%Legacy driver detected: %s. Skipping.znvidia-driver-([0-9]+)(.*)z(No flavour can be found in %s. Skipping.z%s%srrznvidia-headless-no-dkms-%sr*) r!rJZ shortnamerrrrrr __getitem__r0r1r#) rLr$ metapackager5patternrcandidate_flavourr!r9rrr!_get_headless_no_dkms_metapackages.       rc Cs|siSt|}|sLy tj}Wn*tk rJ}ztj|iSd}~XnXi}x\|jD]P\}}xFt||D]8}tj|j dsqn||t |t |dt |d||j <qnWqZW|S)uGet device specific metapackages for this system This calls system_modaliases() to determine the system's hardware and then queries apt about which packages provide hardware enablement support for those. If you already have an apt.Cache() object, you should pass it as an argument for efficiency. If not given, this function creates a temporary one by itself. Return a dictionary which maps package names to information about them: driver_package → {'modalias': 'pci:...', ...} Available information keys are: 'modalias': Modalias for the device that needs this driver (not for drivers from detect plugins) 'syspath': sysfs directory for the device that needs this driver (not for drivers from detect plugins) 'plugin': Name of plugin that detected this package (only for drivers from detect plugins) 'free': Boolean flag whether driver is free, i. e. in the "main" or "universe" component. 'from_distro': Boolean flag whether the driver is shipped by the distro; if not, it comes from a (potentially less tested/trusted) third party source. 'vendor': Human readable vendor name, if available. 'model': Human readable product name, if available. 'recommended': Always True; we always recommend you install these packages. Nz oem-*-metaT)rrlrsrtrvrV) rryrzr{rr7r|rArDr5rNrPrW) r$rrr[rrr<rlr&rrr#system_device_specific_metapackagess*   rcCsdg}t|}|sJy tj}Wn*tk rH}ztj|iSd}~XnXi}x|jD]\}}xt||D]}t||\} } t |\} } | dk o| j |krl||t |t |t |d||j<| dk r| ||jd<| dk r| ||jd<t||} | dk rl| ||jd<qlWqXWdd|D}|rx*|jD]\}}|jd r2|t|<q2W|jtjtd |d }x|D]}||k||d <qtW|S)uGet driver packages, for gpgpu purposes, that are available for the system. This calls system_modaliases() to determine the system's hardware and then queries apt about which packages provide drivers for those. Finally, it looks for the correct metapackage, by calling _get_headless_no_dkms_metapackage(). If you already have an apt.Cache() object, you should pass it as an argument for efficiency. If not given, this function creates a temporary one by itself. Return a dictionary which maps package names to information about them: driver_package → {'modalias': 'pci:...', ...} Available information keys are: 'modalias': Modalias for the device that needs this driver (not for drivers from detect plugins) 'syspath': sysfs directory for the device that needs this driver (not for drivers from detect plugins) 'plugin': Name of plugin that detected this package (only for drivers from detect plugins) 'free': Boolean flag whether driver is free, i. e. in the "main" or "universe" component. 'from_distro': Boolean flag whether the driver is shipped by the distro; if not, it comes from a (potentially less tested/trusted) third party source. 'vendor': Human readable vendor name, if available. 'model': Human readable product name, if available. 'recommended': Some drivers (nvidia, fglrx) come in multiple variants and versions; these have this flag, where exactly one has recommended == True, and all others False. Z10deN)rrlrsrtrVrnrorcSsg|]}|jdr|qS)znvidia-)r)r>r&rrrr?Bsz0system_gpgpu_driver_packages..znvidia-)rurrvrx)rryrzr{rr7r|rArrrrErNrPrWr5rrr}r~rr_cmp_gfx_alternatives_gpgpu)r$rZvendors_whitelistr[rrr<rlr&rnroZ vendor_idZmodel_idrrrurrvrrrsystem_gpgpu_driver_packagessH"      rc Cs:i}|s@y tj}Wn*tk r>}ztj|iSd}~XnXxt|||djD]\}}d|krn|d}n|d}|j|ix$dD]}||kr|||||<qW||jdi} |d |d d | |<d |krT|d | |d <qTWxB|jD]6\} } x*| dD]}t||sPqWd | d<qWt ||S)u Get by-device driver packages that are available for the system. This calls system_modaliases() to determine the system's hardware and then queries apt about which packages provide drivers for each of those. It also adds available packages from detect_plugin_packages(), using the name of the detction plugin as device name. If you already have an apt.Cache() object, you should pass it as an argument for efficiency. If not given, this function creates a temporary one by itself. If freeonly is set to True, only free packages (from main and universe) are considered Return a dictionary which maps devices to available drivers: device_name → {'modalias': 'pci:...', , 'drivers': {'pkgname': {}} A key (device name) is either the sysfs path (for drivers detected through modaliases) or the detect plugin name (without the full path). Available keys in : 'modalias': Modalias for the device that needs this driver (not for drivers from detect plugins) 'vendor': Human readable vendor name, if available. 'model': Human readable product name, if available. 'drivers': Driver package map for this device, see below. Installing any of the drivers in that map will make this particular device work. The keys are the package names of the driver packages; note that this can be an already installed default package such as xserver-xorg-video-nouveau which provides a free alternative to the proprietary NVidia driver; these will have the 'builtin' flag set. 'manual_install': None of the driver packages are installed, but the kernel module that it provides is available; this usually means that the user manually installed the driver from upstream. Aavailable keys in : 'builtin': The package is shipped by default in Ubuntu and MUST NOT be uninstalled. This usually applies to free drivers like xserver-xorg-video-nouveau. 'free': Boolean flag whether driver is free, i. e. in the "main" or "universe" component. 'from_distro': Boolean flag whether the driver is shipped by the distro; if not, it comes from a (potentially less tested/trusted) third party source. 'recommended': Some drivers (nvidia, fglrx) come in multiple variants and versions; these have this flag, where exactly one has recommended == True, and all others False. N)rrlrwrrnrodriversrsrt)rsrtrvTZmanual_install)rrnro) ryrzr{rr7rr|r2rj _add_builtins) r$rrr8rrLZpkginfoZ device_nameZopt_keyrrinforrrsystem_device_driversQs65     rc@seZdZdddZddZdS) _GpgpuDriverNcCsd|_||_||_dS)NrG)rG)_vendors_whitelistrnflavour)selfrnrrrr__init__sz_GpgpuDriver.__init__cCs,|jrtj|j|jsdS|j o(|j S)NF)rnrDfilterrr)rrrris_validsz_GpgpuDriver.is_valid)NN)__name__ __module__ __qualname__rrrrrrrs rcCst}tjd}tjd}tjd}|j|}|j|}|j|}|rn|jd|_d|jd|jdf|_n*|rz||_n|rd|jd|jdf|_|S)zReturns a _GpgpuDriver objectz(.+):([0-9]+)(.*)z([a-z]+)z ([0-9]+)(.*)rz%s%srr)rrrrrrnr)stringrZ full_patternZvendor_only_patternZseries_only_patternZ full_matchZ vendor_matchZ series_matchrrr_process_driver_strings       rc Csg}g}i}|s|S|rLxB|jdD]"}t|}|r$|jr$|j|q$Wnt}|j|t|dkrl|Sd}x$|D]}|jsd||_|d7}qvWd}g}x<|D]4}|j} |j| rtj d|S|j| |d7}qWd}x0|D](}|j o|j rd||_|d7}qWxD|D]<}|j r6d|j|j f} n d|j} |j t j || qWx`|D]X} xP|D]H}| j|jrh|j r|| || <n|| jdr|| || <PqhWq^W|S) Nr-rrrGz0Multiple nvidia versions passed at the same timez%s*%sz%s*rv)rrrappendrlenrn __contains__rrrextendrDrrC) r drivers_strrallowr8itemritZ vendors_temprnrr&rrrgpgpu_install_filters^                   rcCsddddddg}|r"t||}|Sg}x|D]}|jtj||q,Wi}x0|D](}d||ksl||drP||||<qPW|S)aGet packages which are appropriate for automatic installation. Return the subset of the given list of packages which are appropriate for automatic installation by the installer. This applies to e. g. the Broadcom Wifi driver (as there is no alternative), but not to the FGLRX proprietary graphics driver (as the free driver works well and FGLRX does not provide KMS). zbcmwl*z pvr-omap*zvirtualbox-guest*znvidia-*zopen-vm-tools*z oem-*-metarv)rrrDr)rrZ whitelistZresultsrrr8r&rrrauto_install_filter7s    rc #Csi}tjjdd}tjj|s.tjd||S|dkrny tj}Wn*t k rl}ztj |iSd}~XnXx:tj |D]*}|j dsq|tjj ||}tjd|i}t|}y4tt|j|d||d|}tjd ||Wn$t k rtjd |w|YnX|dkrw|t|ttfkrBtj d |t|w|xZ|D]R} | |kr|| jrt||| jjr|j|gj| ntjd | |qHWWdQRXq|W|S) a]Get driver packages from custom detection plugins. Some driver packages cannot be identified by modaliases, but need some custom code for determining whether they apply to the system. Read all *.py files in /usr/share/ubuntu-drivers-common/detect/ or $UBUNTU_DRIVERS_DETECT_DIR and call detect(apt_cache) on them. Filter the returned lists for packages which are available for installation, and return the joined results. If you already have an existing apt.Cache() object, you can pass it as an argument for efficiency. Return pluginname -> [package, ...] map. ZUBUNTU_DRIVERS_DETECT_DIRz(/usr/share/ubuntu-drivers-common/detect/z3Custom detection plugin directory %s does not existNz.pyz"Loading custom detection plugin %sexecZdetectzplugin %s return value: %szplugin %s failed:z6plugin %s returned a bad type %s (must be list or set)z.Ignoring unavailable package %s from plugin %s)renvironrCr isdirrrryrzr{r7listdirendswithr r rrr Z exceptiontypelistr3r!r(r%r2r) r$rZ plugindirrfnamerwZsymbrr8rLrrrrTsJ           "rcCstj|rtj|jdSdS)z5Look up driver package and return their support levelrVN)r}rC)xrrr_pkg_support_from_caches rcCs|jdr|jd rdS|jd r4|jdr4dSt|dkrPt|dkrPdSt|dkrlt|dkrldSt|dkrt|dkrd St|dkrt|dkrdSt|dkrt|dkrd St|dkrt|dkrdS||krd S||krdS||kstdS) zCompare two graphics driver names in terms of preference. (desktop) -server always sorts after non-server. LTSB (Long Term Support Branch) always sorts before NFB (New Feature Branch). Legacy always sorts before Beta. z-serverrrULegacyBetarrxrxrxrxrx)rrrJ)ryrrrrs,rcCs|jdr|jd rdS|jd r4|jdr4dSt|dkrPt|dkrPdSt|dkrlt|dkrldSt|dkrt|dkrd St|dkrt|dkrdSt|dkrt|dkrd St|dkrt|dkrdS||krd S||krdS||kstdS) zCompare two graphics driver names in terms of preference. (server) -server always sorts before non-server. LTSB (Long Term Support Branch) always sorts before NFB (New Feature Branch). Legacy always sorts before Beta. z-serverrrUrrrrxrxrxrxrx)rrrJ)rrrrrrs,rcCsx|jD]\}}xv|dD]j}|jdrBddddd|dd<P|jdrx |dD]}d|d|d<qVWddddd|dd <PqWq Wd S) zAdd builtin driver alternativesrrGTF)rsZbuiltinrtrvzxserver-xorg-video-nouveaur]rvzxserver-xorg-video-atiN)r|r)rZdevicerrLdrrrrs  rcCstj|}|jS)z0Return the linux headers for the system's kernel)rKernelDetectionZget_linux_headers_metapackage)r$kernel_detectionrrrget_linux_headerss rcCstj|}|jS)z.Return the linux image for the system's kernel)rrZget_linux_image_metapackage)r$rrrrget_linux_images rcCstj|}|jS)z.Return the linux image for the system's kernel)rrget_linux_version)r$rrrrrs rcCstj|}|jS)z4Return the linux metapackage for the system's kernel)rrZget_linux_metapackage)r$rrrr get_linuxs rcCst}x|D]z}|jj|r g}|jr4|j|jj|jrH|j|jjx<|D]4}x.|D]&}|jdkrhqX|j|krX|j|jqXWqNWq Wt |S)z-Return the reverse dependencies for a packager ) r3r5rr!r dependenciesraZrawtyper4r)r$r9prefixr'rLrZordepdeprrrfind_reverse_dependenciess      rc Csf||jr||jj}y |d}Wntk r6dSX|jjd}x|D]}|jdrL|SqLWdS)Nr z, z linux-image-)r!r%r#rrr)r$rLr%r'Z deps_listrrrrget_linux_image_from_metas     rcCs&|dk s td}d}d}d|kr0tjd||Stjd}|j|}|rXtjd||St|}t||}|r||jdd}ntjd||Stjd }|j|}|stjd ||Sd |j d |j d f} d| |f} y~|j | } | j o| j j dt fkrTt|} d| | f} tjd| |j | }|j rT|j j dt fkrTtjd| | }Wn"tk rxtjd| YnX|rt||d}d}x|D]}||kr|}qW|r|}|Sd| }tjd|y,|j |} | j r| j j dt fkr|}Wn"tk r tjd|YnX|S)zDReturn the linux-modules-$driver metapackage for the system's kernelNrrGzNNon NVIDIA linux-modules packages are not supported at this time: %s. Skippingznvidia-([0-9]+)z%Legacy driver detected: %s. Skipping.z linux-image-z-No linux-image can be found for %s. Skipping.znvidia-.*-([0-9]+)(.*)z(No flavour can be found in %s. Skipping.z%s%srrzlinux-modules-nvidia-%s-%sr*zlinux_modules_abi_candidate: %szFound ABI compatible %szNo "%s" can be found.zlinux-modules-nvidia-znvidia-dkms-%szFalling back to %s)rJrrrrrrrreplacerrr!r0r1rr#rr7)r$r!rZ linux_flavourZlinux_modules_matchrrZlinux_image_metaZ linux_imagerZlinux_modules_candidater9Z linux_versionZlinux_modules_abi_candidateZ abi_specificZ reverse_depsZpickrZ dkms_packagerrrget_linux_modules_metapackage+sx                     r)N)NNFT)NNT)NN)NNF)r)N)1__doc__rrrDrbrrryZ UbuntuDriversrapt_pkgZget_architecturesr1r}rr(r=rArBrNrPrSrWr\rjrrrrrrrrobjectrrrrrrrrrrrrrrrrrrrrsX   /, # ] & = R \d  =