0x1949 Team - FAZEMRX - MANAGER
Edit File: detect.cpython-310.pyc
o �g� � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl m Z da i a dZ G dd� de�Zdd � Zd\d d �Zd d � Zdd� Zdd� Zdd� Zdd� Zi e_dd� Zdd� Zdd� Zd d � Zd d � Z d d!� Z d"d#� Z d$d%� Z d&d'� Z d(d)� Z!d]d,d-�Z"d.d/� Z#d0d1� Z$d^d2d3�Z%d_d4d5�Z&d`d6d7�Z'dad8d9�Z(d:d;� Z)d<d=� Z*G d>d?� d?e�Z+d@dA� Z,dBdC� Z-dbdDdE�Z.d\dFdG�Z/dHdI� Z0dJdK� Z1dLdM� Z2dNdO� Z3dPdQ� Z4dRdS� Z5dTdU� Z6dVdW� Z7dXdY� Z8dZd[� Z9dS )czGHardware and driver package detection functionality for Ubuntu systems.� N)�kerneldetection� z /etc/custom_supported_gpus.jsonc @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dS )�NvidiaPkgNameInfoz%Class to process NVIDIA package namesc C s: || _ d| _d| _d| _d| _d| _d| _| �| j � d S )NF���r )� _pkg_name�_obsolete_name_scheme�_server�_open� _major_ver�_flavour�is_valid� _process_name)�self�pkg_name� r �6/usr/lib/python3/dist-packages/UbuntuDrivers/detect.py�__init__ s zNvidiaPkgNameInfo.__init__c C s� d|vr t �d|� d S t�d�}|�|�}|r*d| _|�d�| _| j| _d| _ d S t�d�}|�|�}|rf|�d�� d�d k| _ |�d�� d �d k| _ d |�d�| j rVdnd | j r\d nd f | _d| _ d S d S ) N�nvidiaz8NvidiaPkgNameInfo: %s is not an NVIDIA package. Skippingznvidia-([0-9]+)T� znvidia-driver-([0-9]+)(.*)r �-serverr �-openz%s%s%sr ) �logging�debug�re�compile�matchr �groupr r r �findr r )r �name�patternr r r r r ( s, � �z NvidiaPkgNameInfo._process_namec C � | j S �N)r �r r r r �has_obsolete_name_schemeD � z*NvidiaPkgNameInfo.has_obsolete_name_schemec C r r! )r r" r r r � is_serverG r$ zNvidiaPkgNameInfo.is_serverc C r r! )r r" r r r �is_openJ r$ zNvidiaPkgNameInfo.is_openc C r r! )r r" r r r �get_major_versionM r$ z#NvidiaPkgNameInfo.get_major_versionc C r r! )r r" r r r � get_flavourP r$ z NvidiaPkgNameInfo.get_flavourN) �__name__� __module__� __qualname__�__doc__r r r# r% r& r' r( r r r r r s r c C s t st�� d a t S )zCache system architecturer )�system_architecture�apt_pkg�get_architecturesr r r r � get_apt_archT s r0 c C sb i }| rd| p d}t �|�D ]�\}}}d}d|v rUz!tt j�|d���}|�� �� }W d � n1 s5w Y W nP tyT } z t� d||� W Y d}~qd}~ww d|v r�d|v r�tt j�|d��� } | D ]} | � d�r|| � d d �d �� } nqiW d � n1 s�w Y |s�qt j�|d �} t j�| d �} t j� | �r�t j� | �s�q|||< q|S ) u� Get 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/devicesN�modaliasz0system_modaliases(): Cannot read %s/modalias: %s�ssb�ueventz MODALIAS=�=r �driver�module) �os�walk�open�path�join�read�strip�IOErrorr r � startswith�split�islink) �sys_path�aliases�devicesr: �dirs�filesr1 �f�e�fd�line� driverlink�modlinkr r r �system_modaliases\ sD ��� �� �� � rM c C s� d }|j �d�r'|j �dd�}z| | }W n ty& t�d| � Y dS w t�| �}|�|�}d}z |j � d�D ]}|D ]\}} } |�d�rLd} nq>q:W n tt fyc t�d| � d}Y nw |spt�d |j � dS z| d } W n ty� t�d � Y dS w |�| �}|j D ] \} } }| �d�r�| }q�|r�| | }|j D ] }|jj |j kr� dS q�t�d |j |� t�d |dd� |j D �f � dS dS )Nznvidia-driver-zxserver-xorg-video-nvidia-z5Cannot find %s package in the cache. Cannot check ABITF�Dependszxorg-video-abi-z@The %s package seems to have no dependencies. Skipping ABI checkz;Skipping check for %s since it does not depend on video abizxserver-xorg-corez1xserver-xorg-core not available, cannot check ABIzBDriver package %s is incompatible with current X.org server ABI %sz%s is not in %sc S s g | ]}|j �qS r )� parent_pkg)�.0�xr r r � <listcomp>� s z+_check_video_abi_compat.<locals>.<listcomp>)r r? �replace�KeyErrorr r r. �DepCache�get_candidate_ver�depends_list_str�get� TypeError� provides_list�rev_depends_listrO )� apt_cache�package�xorg_video_abi�xorg_driver_name�depcache� candidate�needs_video_abi�dep_list�dep_name�dep_ver�dep_op� xorg_core� provides_name� provides_ver� p_version�abi_pkg�depr r r �_check_video_abi_compat� sf � ���� � � �� rm c C sp t �| �}t �| �}i }| jD ]�}z|�|�}|�|jd � |d }|s'W qW n ttt fy4 Y qw |j dt � fvr>qt | |�sDqz<|� d�D ]4}|�d�}|sTqJ|� d�\}} | � d�D ] } | �� } | � dd �d } |�| i ��| t� ��|j� q`qJW q ty� t�d |j|f � Y qw i } |�� D ]\} } t�d �d d � | �� D ��tj�}|| f| | < q�| S )aI Build a modalias map from an apt_pkg.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"). r � Modaliases�all�)�, �(�,�:r z*Package %s has invalid modalias header: %s�|c S s g | ]}t �|��qS r )�fnmatch� translate)rP �patr r r rR � s z+_apt_cache_modalias_map.<locals>.<listcomp>) r. rU �PackageRecords�packagesrV �lookup� file_listrT �AttributeError�UnicodeDecodeError� architecturer0 rm r@ r= � setdefault�set�addr � ValueErrorr �error�itemsr r r; �keys� IGNORECASE)r\ r` �records�resultr] ra �m�partr6 �lst�alias�bus�result2� alias_maprx r r r �_apt_cache_modalias_map� sV �� �� ���r� c C s t S r! )�custom_supported_gpus_jsonr r r r � path_get_custom_supported_gpus� s r� c C s� t � }d}z_t|d��O}z-tt�|�d �}|D ] }|d | kr7|d �d�d }t�d||d | f � nqW n tyG t� d � Y n w W d � W |S W d � W |S 1 s]w Y W |S tyu t� d | � Y |S w ) z|Get nvidia allowing driver for specific devices. did: 0x1234 Return the situable nvidia driver version for it. N�r�chips�devid�branch�.r z0Found a specific nv driver version %s for %s(%s)r z;package_get_nv_allowing_driver(): unexpected json detected.z3package_get_nv_allowing_driver(): unable to read %s) r� r9 �list�json�loadr@ r �info� Exceptionr )�didr: �version�stream�gpus�gpur r r � package_get_nv_allowing_driver� s@ ��� �� �� �� ��r� c sV t � }tdd� � jD ��}ztj| }W n ty& t� �}|tj|< Y nw |�|�dd�d di f�\}}t |�\}}d} d} |dkrzt d| �} d | } d | } � jD ]} | � � �d�d | krlt | g�|| < d} qV| durz| szt � d | � | s�|du s�|�|�s�g S |D ]}t�|�� |�� �r�|| D ]} |�| � q�q�� fd d �|D �S )z_Search packages which match the given modalias. Return a list of apt.Package objects. c s s � | ]}|� � V qd S r! )� get_fullname)rP r] r r r � <genexpr>! s � z(packages_for_modalias.<locals>.<genexpr>rt r r N�10DE�0xznvidia-driver-%szpci:v000010DEd0000%s*z %s is not in the package pool.c s g | ]}� | �qS r r �rP �p�r\ r r rR @ s z)packages_for_modalias.<locals>.<listcomp>)r� �hashrz �packages_for_modalias� cache_mapsrT r� rX r@ � _get_vendor_model_from_aliasr� r� r r r rv � fnmatchcase�lowerr� )r\ r1 �pkgs�apt_cache_hash� cache_maprx �bus_map�vidr� �nvamd�found�nvamdn�nvamdar� r� r r� r r� s@ � � �r� c C s� t �| �}|�|�}|d usJ �|j�d�rdS t �| �}|�|jd � |jd jD ]&\}}|j sIz |d }|dvW S t yH Y dS w |j dv rQ dS q+dS )Nr Fr � Component)� restricted� multiverseT) r. rU rV r r? ry r{ r| � version_list� componentrT )r\ �pkgr` ra r� �pfile�_r� r r r �_is_package_freeF s& � �r� c C sL t �| �}|�|�}|d u rdS z |jd d jdkW S ty% Y dS w )NFr �Ubuntu)r. rU rV r| �originrT )r\ r� r` ra r r r �_is_package_from_distroc s �r� c C s� t �| �}|�|�}t �| �}|�|jd � z|d }W n ttfy0 t� d|j � Y dS w |� d�}|dkrCt� d|j � dS |d|� }|S )z-Determine module name from apt Package objectr rn zM_pkg_get_module %s: package has no Modaliases header, cannot determine moduleNrr zR_pkg_get_module %s: package has invalid Modaliases header, cannot determine module) r. rU rV ry r{ r| rT r} r r r r )r\ r� r` ra r� r� �parenr6 r r r �_pkg_get_moduleo s � r� c C s� t �| �}|�|�}t �| �}|�|jd � z|d }W n ttfy0 t� d|j � Y dS w |dvr?t� d|j |� dS |S )z/Determine support level from apt Package objectr �SupportzR_pkg_get_support %s: package has no Support header, cannot determine support levelN)�PB�NFB�LTSB�LegacyzY_pkg_get_support %s: package has invalid Support %sheader, cannot determine support level) r. rU rV ry r{ r| rT r} r r r )r\ r� r` ra r� �supportr r r �_pkg_get_support� s ��r� c C sj t �d| �}|�d�}|�d�}|dkrdS d| }t� }z�t|d��v}zTtt�|�d �}|D ]F}|d |krvd |d v rv|d �d �d |krdt � d|d �d �d |f � W W d � W dS t � d| � W W d � W dS q0W n t y� t � d� Y n w W d � W dS W d � W dS 1 s�w Y W dS t y� t � d| � Y dS w )zAalias: e.g. pci:v000010DEd000024BAsv0000103Csd000089C6bc03sc00i00z pci:v0000(.*)d0000(.*)sv(.*)r � r� Fz0x%sr� r� r� � runtimepm�featuresr� r� r z)Candidate version does not match %s != %sNz Found runtimepm supports on %s.Tz?_is_nv_allowing_runtimepm_supported(): unexpected json detectedz8_is_nv_allowing_runtimepm_supported(): unable to read %s) r �searchr r� r9 r� r� r� r@ r r r� r� ) r� �verr� r� r� r: r� r� r� r r r �#_is_nv_allowing_runtimepm_supported� sR �� �� �� �� �� ��r� c s� z+t �| �}|�|�}t �| �}|�|jd � t �|j�}|j� d�d }|d }W n t t t fy8 Y dS w |� d�dkrBdS ||� d�d |� d�� } | � d �} t� |�r\d S t� fd d �| D ��S ) z>Check if the package supports runtimepm for the given modaliasr r� � PmAliasesFznvidia(rr r rp rq Tc 3 s$ � | ] }t � � �� |�� �V qd S r! )rv r� )rP �regex�r� r r r� � s �" z*_is_runtimepm_supported.<locals>.<genexpr>)r. rU rV ry r{ r| � TagSection�record�ver_strr@ rT r} r~ r r� �any) r\ r� r� r` ra r� �sectionr� r� �n� modaliasesr r� r �_is_runtimepm_supported� s$ � r� c C s t j�d�duS )z*Check if the current session in on Wayland�WAYLAND_DISPLAYN)r7 �environrX r r r r �is_wayland_session� s r� c C s� |j rdS |j�d�rd}n|j�d�rd}nt| |�}|s dS ztjd|gtjtjd�}|�� W n tt tj fyQ } zt � dt |�� W Y d}~dS d}~ww |jdkrat � d |j|� d S t � d |j|� dS ) zIDetermine if the kernel module from an apt.Package is manually installed.Fr �fglrx�modinfo)�stdout�stderrz _is_manual_install failed: %sNr zJ_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)� current_verr r? r� � subprocess�Popen�PIPE� communicate�OSError�FileNotFoundError�CalledProcessErrorr r �str� returncode)r\ r� r6 r� rH r r r �_is_manual_install� s8 � �� ��r� c C s� zt jddd| gdd�}W n tt jfy. } zt�d| |t|�� W Y d}~dS d}~ww t�d |� d}d}|�� D ]}|�d d �\}}d |v rM|}d |v rS|}q=t�d| |||� ||fS )zTReturn (vendor, model) names for given device. Values are None if unknown. �udevadm�hwdbz--test=T)�universal_newlinesz-_get_db_name(%s, %s): udevadm hwdb failed: %sN�NNz_get_db_name: output %s r4 r �_VENDOR�_MODELz-_get_db_name(%s, %s): vendor "%s", model "%s") r� � check_outputr� r� r r r� � splitlinesr@ ) �syspathr� �outrH �vendor�modelrJ �k�vr r r � _get_db_name� s. ��� ��r c C s, d}d| }t |d�}|�|� |�� dS )z Set KMS on or off for NVIDIAz /lib/modprobe.d/nvidia-kms.confz�# This file was generated by ubuntu-drivers # Set value to 1 to enable modesetting, 0 to disable it options nvidia-drm modeset=%d �wN)r9 �write�close)�value�nvidia_kms_file�kms_text�kms_fdr r r �set_nvidia_kms s � r FTc C s� t |�}| s'zt�d�} W n ty& } z t�|� i W Y d}~S d}~ww i }|�� D ]T\}}t| |�D ]J} |r@t| | �s@q6|sJt � | j d�rJq6||t| | �t | | �t | | �t | | |�d�|| j < t||�\} } | duru| || j d< | dur�| || j d< q6q-dd� |D �} | r�|�� D ] \} }| �d�r�|t| < q�| jt�t�d � | d }| D ] } | |k|| d < q�t| ��� D ]-\}}|D ]&} z| | }t| |�t | |�|d �|| < W q� ty� t�d | � Y q�w q�|S )u� Get 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_pkg.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. N� oem-*-meta)r1 r� �free� from_distror� r� r� r� c S � g | ] }|� d �r|�qS ��nvidia-�r? r� r r r rR c � z*system_driver_packages.<locals>.<listcomp>r ��keyr � recommended)r r �pluginz*Package %s plugin not available. Skipping.)rM r. �Cacher� r r� r� r� r� rv r r� r� r� r r? � lookup_cache�sort� functools� cmp_to_key�_cmp_gfx_alternatives�detect_plugin_packagesrT r )r\ rB �freeonly� include_oemr� �exrz r� r� r� r� r� �nvidia_packagesr r r r r� �apt_pr r r �system_driver_packages sj % �� ��� �� �� r! c C s@ t �d�}|�| �}|r |�d�dd � |�d�dd � fS dS )Nz$(.+):v(.+)d(.+)sv(.+)sd(.+)bc(.+)i.*r� � � r� )r r r r )r� �modalias_pattern�detailsr r r r� s $r� c C s� | d usJ �d } t �|�}| j}t|�}|js!t�d| � |S |�� r.t�d| � |S |�� }d| }z || }|� |�} |rO| j dt � fv rR|}W |S W |S W |S t y^ Y |S w )Nz)Unsupported driver detected: %s. Skipping�%Legacy driver detected: %s. Skipping.znvidia-headless-no-dkms-%sro ) r. rU r r r r r r# r( rV �archr0 rT ) r� r\ � metapackager` r � nvidia_info�candidate_flavourra r] �package_candidater r r �!_get_headless_no_dkms_metapackage� s: �� ��r, c C s� |si S t |�}| s+zt�d�} W n ty* } z t�|� i W Y d}~S d}~ww i }|�� D ])\}}t| |�D ] }t�|j d�sDq:||t | |�t | |�dt | |�d�||j < q:q1|S )u� Get 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_pkg.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. Nr T)r1 r� r r r r� ) rM r. r r� r r� r� r� rv r r� r� r� ) r\ rB r r� r rz r� r� r� r r r �#system_device_specific_metapackages� s2 ���� r- c C s� dg}t |�}| s*zt�d�} W n ty) } z t�|� i W Y d}~S d}~ww i }|�� D ]]\}}t| |�D ]S}t||�\} } t |�\} } | dur�| � � |v r�||t | |�t | |�t | |�d�||j< | durq| ||j d< | dur|| ||j d< t|| �} | dur�| ||j d< q9q0dd� |D �}|r�|�� D ] \}}|�d �r�|t|< q�|jt�t�d � |d }|D ] }||k|| d < q�|S ) u� Get 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_pkg.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. �10deN)r1 r� r r r� r� r� r( c S r r r r� r r r rR / r z0system_gpgpu_driver_packages.<locals>.<listcomp>r r r r )rM r. r r� r r� r� r� r r� r� r� r� r� r r, r? r r r r �_cmp_gfx_alternatives_gpgpu)r\ rB �vendors_whitelistr� r rz r� r� r� r� r� � vendor_id�model_idr( r r r r r r r � system_gpgpu_driver_packages� sT " �� � �� �r3 c C s2 i }| s%zt �d�} W n ty$ } z t�|� i W Y d}~S d}~ww t| ||d��� D ]G\}}d|v r;|d }n|d }|�|i � dD ]}||v rU|| || |< qG|| �di �} |d |d d �| |< d |v ru|d | | d < q.|�� D ]\} } | d D ] }t| | | �s� nq�d | d < qzt |� |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_pkg.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:...', <device info>, 'drivers': {'pkgname': {<driver package info>}} 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 <device info>: '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 <driver package info>: '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)r r� r )r1 r� r� �driversr r )r r r T�manual_install) r. r r� r r� r! r� r� r� � _add_builtins) r\ rB r r� r r� �pkginfo� device_name�opt_keyr4 r5 r� r r r �system_device_drivers> sH 5 ��� � �� ��r: c C s� t | |||d�}t||�}|st�d� |S t�| �}t�| �}g }|D ]V} | | } | jsz|�| � |� | �} |� | j d � |d r]zt dd�} | � d� | �� W n ty\ Y nw zt| | �} | ro| | jso|�| � W q$ tyy Y q$w q$|S )z4Return the list of packages that should be installed)r r z"No drivers found for installation.r r� � /run/nvidia_runtimepm_supportedr � )r! �auto_install_filterr r r. rU ry r� �appendrV r{ r| r9 r r �PermissionError� get_linux_modules_metapackagerT )r\ rB � free_onlyr � driver_stringrz r` r� � to_installr� � package_objra �pm_fd�modules_packager r r �get_desktop_package_list� sF � � � ��rG c C s0 | D ]}t |�}|�� dkrtd� dS qdS )zHApplies changes that need to happen before installing the NVIDIA driversi� r N)r r' r )rC � package_namer) r r r �$nvidia_desktop_pre_installation_hook� s ��rI c C sz t j�d�r;t�d� z t�ddg� W n ty Y nw tdd��} | � d� W d � d S 1 s4w Y d S d S )Nr; z,Trying to select the on-demand PRIME profilez/usr/bin/prime-selectz on-demandz$/etc/u-d-c-nvidia-runtimepm-overrider z!# File created by ubuntu-drivers ) r7 r: �isfiler r r� �callr� r9 r )rG r r r �%nvidia_desktop_post_installation_hook� s � "��rL c @ s e Zd Zddd�Zdd� ZdS )� _GpgpuDriverNc C s d| _ || _|| _d S )N)r )�_vendors_whitelistr� �flavour)r r� rO r r r r � s z_GpgpuDriver.__init__c C s, | j r t�| j| j �s dS | j o| j S )NF)r� rv �filterrN rO r" r r r r � s z_GpgpuDriver.is_validr� )r) r* r+ r r r r r r rM � s rM c C s� t � }t�d�}t�d�}t�d�}|�| �}|�| �}|�| �}|r8|�d�|_d|�d�|�d�f |_|S |r?| |_|S |rNd|�d�|�d�f |_|S )z Returns a _GpgpuDriver objectz(.+):([0-9]+)(.*)z([a-z]+)z ([0-9]+)(.*)r z%s%sr� r# )rM r r r r r� rO )�stringr5 � full_pattern�vendor_only_pattern�series_only_pattern� full_match� vendor_match� series_matchr r r �_process_driver_string� s" ��rX c C s� g }g }i } | s |S |r%|� d�D ]}t|�}|r#|�� r#|�|� qnt� }|�|� t|�dk r5|S d}|D ]}|jsCd|| _|d7 }q9d}g }|D ] }|j} |�| �rat� d� | S |�| � |d7 }qNd}|D ]}|j s||js|d|| _|d7 }qo|D ] }|j r�d|j|j f } nd|j } |� t � | | �� q�|D ]&} |D ]!}| �|j�r�|j r�| | || < n | | �d�r�| | || < nq�q�|S ) Nrs r r r z0Multiple nvidia versions passed at the same timez%s*%sz%s*r )r@ rX r r> rM �lenr� � __contains__r r rO �extendrv rP rX ) rz � drivers_strr4 �allowr� �itemr5 �it� vendors_tempr� r r� r r r �gpgpu_install_filter sh �� �� ra c C sp g d�}|r t | |�}|S g }|D ] }|�t�| |�� qi }|D ]}d| | vs/| | d r5| | ||< q!|S )a� Get 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*r r )ra r[ rv rP )rz r\ � whitelist�resultsr] r r� r� r r r r= j s �r= c C s� i }t j�dd�}t j�|�st�d|� |S | du r<zt�d�} W n t y; } z t� |� i W Y d}~S d}~ww t � |�D ]�}|� d�sIqAt j� ||�}t�d|� i }t|���}ztt|�� |d�|� |d | �}t�d ||� W n t y� t�d |� Y W d � qAw |du r� W d � qAt|�ttfvr�t� d |t|�� W d � qA|D ]&} z| | } t| | �r�|�|g ��| � W q� ty� t�d | |� Y q�w W d � n1 s�w Y qA|S ) aa 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_pkg.Cache() object, you can pass it as an argument for efficiency. Return pluginname -> [package, ...] map. �UBUNTU_DRIVERS_DETECT_DIRz(/usr/share/ubuntu-drivers-common/detect/z3Custom detection plugin directory %s does not existNz.pyz"Loading custom detection plugin %s�exec�detectzplugin %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)r7 r� rX r: �isdirr r r. r r� r� �listdir�endswithr; r9 re r r< � exception�typer� r� rm r� r> rT ) r\ rz � plugindirr �fnamer �symbrG r� r� r] r r r r � sf � �� �� � � �� ��r c C s t �| �r t �| ��d�S dS )z5Look up driver package and return their support levelr� N)r rX )rQ r r r �_pkg_support_from_cache� s ro c C s� | � d�r |� d�s dS | � d�s|� d�rdS | � d�r$|� d�s$dS | � d�s0|� d�r0dS ddg}d}d}t| �}t|�}||v rH|d7 }||v rP|d7 }| |krY|d7 }n| |k ra|d7 }|dksi|dkru||krodS ||k rudS t| �dkr�t|�dkr�dS t| �dkr�t|�dkr�dS t| �dkr�t|�dkr�dS t| �dkr�t|�dkr�dS t| �d kr�t|�d kr�dS t| �d kr�t|�d kr�dS t| �d kr�t|�d kr�dS t| �d kr�t|�d kr�dS | |k r�dS | |kr�dS | |ks�J �dS ) a Compare two graphics driver names in terms of preference. (desktop) -open always sorts after non-open. -server always sorts after non-server. LTSB (Long Term Support Branch) always sorts before NFB (New Feature Branch). Legacy always sorts before Beta. r r r r r� r� r �d r� �Beta�ri ro )rQ �y�preferred_support�x_score�y_score� x_support� y_supportr r r r � s` r c C sh | � d�r |� d�s dS | � d�s|� d�rdS | � d�r$|� d�s$dS | � d�s0|� d�r0dS t| �dkr>t|�dkr>dS t| �dkrLt|�dkrLdS t| �dkrZt|�dkrZdS t| �dkrht|�dkrhdS t| �dkrvt|�dkrvdS t| �dkr�t|�dkr�dS t| �dkr�t|�dkr�dS t| �dkr�t|�dkr�dS | |k r�dS | |kr�dS | |ks�J �d S ) a Compare two graphics driver names in terms of preference. (server) -open always sorts after non-open. -server always sorts before non-server. LTSB (Long Term Support Branch) always sorts before NFB (New Feature Branch). Legacy always sorts before Beta. r r r r r� r� r� rq r rr )rQ rs r r r r/ s< r/ c C s� | � � D ]>\}}|d D ]5}|�d�r ddddd�|d d< n"|�d�rA|d D ] }d|d | d< q)ddddd�|d d < nq qd S ) z Add builtin driver alternativesr4 r TF)r �builtinr r zxserver-xorg-video-nouveaur� r zxserver-xorg-video-atiN)r� r? )r4 �devicer� r� �dr r r r6 : s � ����r6 c C � t �| �}|�� S )z0Return the linux headers for the system's kernel)r �KernelDetection� get_linux_headers_metapackage�r\ �kernel_detectionr r r �get_linux_headersP � r� c C r| �z.Return the linux image for the system's kernel)r r} �get_linux_image_metapackager r r r �get_linux_imageV r� r� c C r| r� )r r} �get_linux_versionr r r r r� \ r� r� c C r| )z4Return the linux metapackage for the system's kernel)r r} �get_linux_metapackager r r r � get_linuxb r� r� c C s� t �| �}z |�| | �}W n ty t�d| � Y d S w z |j�d�D ]}|D ]\}}}|�d�r<| W S q+q'W d S tt fyS t�d| � Y d S w )NzNo candidate for %s foundrN � linux-image-z+Could not check dependencies for %s package) r. rU rV rT r r rW rX r? rY )r\ r� r` ra rc rd re rf r r r �get_linux_image_from_metah s( � �����r� c C s |dusJ �d}d}d}t �| �}t|�}|js!t�d| � |S |�� r.t�d| � |S t| �}t| |�}|r@|� dd�}n t�d| � |S |� � } d| |f } zB| | } |� | �} | r�| j dt � fv r�t| �} d| | f }t�d | � | | }|� |�}|r�|j dt � fv r�t�d | � | }W n ty� t�d | � Y nw |r�d d � | | jD �}d}d| t| �� dd�f }|D ]}||kr�|}q�|r�|}|S d| }t�d| � z | | } |� | �} | r�| j dt � fv r�|}W |S W |S W |S t�y t�d |� Y |S w )zDReturn the linux-modules-$driver metapackage for the system's kernelNr zNNon NVIDIA linux-modules packages are not supported at this time: %s. Skippingr& r� z-No linux-image can be found for %s. Skipping.zlinux-modules-nvidia-%s-%sro z linux_modules_abi_candidate: %szFound ABI compatible %szNo "%s" can be found.c S s" g | ] }|j j�d �r|j j�qS )zlinux-modules-nvidia-)rO r r? )rP rl r r r rR � s �z1get_linux_modules_metapackage.<locals>.<listcomp>znvidia-dkms-%szFalling back to %s)r. rU r r r r r# r� r� rS r( rV r' r0 r� rT r[ r� )r\ ra r( � linux_flavour�linux_modules_matchr` r) �linux_image_meta� linux_imager* �linux_modules_candidater] r+ � linux_version�linux_modules_abi_candidate� abi_specific�abi_specific_candidate� reverse_deps�pick�modules_candidaterl � dkms_packager r r r@ � s� � ��� ��� �r@ r! )NNFT)NNTr� )NNF)NFTr )r ):r, r7 r rv r� r r r� r. � UbuntuDriversr r- r r� �objectr r0 rM rm r� r� r� r� r� r� r� r� r� r� r� r� r� r r r! r� r, r- r3 r: rG rI rL rM rX ra r= r ro r r/ r6 r� r� r� r� r� r@ r r r r �<module> sp 8 /88) # ` % = R \- d >D*