HEX

File: //usr/lib/python3/dist-packages/UbuntuDrivers/__pycache__/detect.cpython-310.pyc
o

�:9iB��@sdZddlZddlZddlZddlZddlZddlZddlZddlZddl	m
Z
ddlmZdaia
dZGdd�de�Zd	d
�Zdgdd�Zd
d�Zdd�Zdd�Zdd�Zdd�Zie_dd�Zdd�Zdd�Zdd�Zdd �Zd!d"�Zd#d$�Zd%d&�Z d'd(�Z!d)d*�Z"d+d,�Z#d-d.�Z$dhd1d2�Z%d3d4�Z&d5d6�Z'did7d8�Z(djd9d:�Z)dkd;d<�Z*	0	/dld=d>�Z+d?d@�Z,dAdB�Z-GdCdD�dDe�Z.dEdF�Z/dmdGdH�Z0dndIdJ�Z1dgdKdL�Z2dMdN�Z3dOdP�Z4dQdR�Z5dSdT�Z6dUdV�Z7dWdX�Z8dYdZ�Z9d[d\�Z:d]d^�Z;d_d`�Z<dadb�Z=dcdd�Z>dedf�Z?dS)ozGHardware and driver package detection functionality for Ubuntu systems.�N)�kerneldetection)�
cmp_to_key�z/etc/custom_supported_gpus.jsonc@sHeZdZdZdd�Zdd�Zdd�Zdd	�Zd
d�Zdd
�Z	dd�Z
dS)�NvidiaPkgNameInfoz%Class to process NVIDIA package namescCs:||_d|_d|_d|_d|_d|_d|_|�|j�dS)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__szNvidiaPkgNameInfo.__init__cCsdd|vrt�d|�dSt�d�}|�|�}|r,d|_t|�d��|_|j|_	d|_
dSt�d�}|�|�}|rn|�d��d�d	k|_|�d��d
�d	k|_
d|�d�|jrXdnd|j
r^d
ndf|_	t|�d��|_d|_
t�d
�}|�|�}|r�|�d��d�d	k|_|�d��d
�d	k|_
d|�d�|jr�dnd|j
r�d
ndf|_	|�d�|_d|_
dSdS)N�nvidiaz8NvidiaPkgNameInfo: %s is not an NVIDIA package. Skippingznvidia-([0-9]+)T�znvidia-driver-([0-9]+)(.*)rz-serverrz-openz%s%s%srz nvidia-headless(.*)-([0-9]+)(.*)�)�logging�debug�re�compile�matchr�int�grouprrr
�findr	r
)r�name�patternrrrrr)sD




�


�
�zNvidiaPkgNameInfo._process_namecC�|jS�N)r�rrrr�has_obsolete_name_schemeS�z*NvidiaPkgNameInfo.has_obsolete_name_schemecCr!r")r	r#rrr�	is_serverVr%zNvidiaPkgNameInfo.is_servercCr!r")r
r#rrr�is_openYr%zNvidiaPkgNameInfo.is_opencCr!r")rr#rrr�get_major_version\r%z#NvidiaPkgNameInfo.get_major_versioncCr!r")rr#rrr�get_flavour_r%zNvidiaPkgNameInfo.get_flavourN)�__name__�
__module__�__qualname__�__doc__rrr$r&r'r(r)rrrrrs
*rcCstst��datS)zCache system architecturer)�system_architecture�apt_pkg�get_architecturesrrrr�get_apt_archcsr1c
Csbi}|rd|p	d}t�|�D]�\}}}d}d|vrUz!ttj�|d���}|����}Wd�n1s5wYWnPtyT}z
t�	d||�WYd}~qd}~wwd|vr�d|vr�ttj�|d���}	|	D]}
|
�
d�r||
�d	d
�d
��}nqiWd�n1s�wY|s�qtj�|d�}tj�|d�}tj�|�r�tj�|�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�IOErrorrr�
startswith�split�islink)
�sys_path�aliases�devicesr;�dirs�filesr2�f�e�fd�line�
driverlink�modlinkrrr�system_modaliasesksD�����
���
rNc	Cs�d}|j�d�r'|j�dd�}z||}Wnty&t�d|�YdSwt�|�}|�|�}d}z|j	�
d�D]}|D]\}}	}
|�d�rLd}nq>q:Wnttfyct�d|�d}Ynw|spt�d	|j�dSz|d
}Wnty�t�d�YdSw|�|�}|jD]\}}
}|�d�r�|}q�|r�||}|j
D]}|jj|jkr�dSq�t�d|j|�t�d
|dd�|j
D�f�dSdS)N�nvidia-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 %scSsg|]}|j�qSr)�
parent_pkg)�.0�xrrr�
<listcomp>��z+_check_video_abi_compat.<locals>.<listcomp>)rr@�replace�KeyErrorrrr/�DepCache�get_candidate_ver�depends_list_str�get�	TypeError�
provides_list�rev_depends_listrQ)�	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�deprrr�_check_video_abi_compat�sf�


����
�

�
��rpcCspt�|�}t�|�}i}|jD]�}z|�|�}|�|jd�|d}|s'WqWnttt	fy4Yqw|j
dt�fvr>qt||�sDqz<|�
d�D]4}|�d�}|sTqJ|�
d�\}}	|	�
d�D]}
|
��}
|
�
dd	�d}|�|i��|
t���|j�q`qJWqty�t�d
|j|f�Yqwi}|��D]\}}
t�d�dd
�|
��D��tj�}||
f||<q�|S)aIBuild 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�)�, �(�,�:rz*Package %s has invalid modalias header: %s�|cSsg|]}t�|��qSr)�fnmatch�	translate)rR�patrrrrTsz+_apt_cache_modalias_map.<locals>.<listcomp>)r/rX�PackageRecords�packagesrY�lookup�	file_listrW�AttributeError�UnicodeDecodeError�architecturer1rprAr>�
setdefault�set�addr�
ValueErrorr�error�itemsrrr<�keys�
IGNORECASE)r_rc�records�resultr`rd�m�partr7�lst�alias�bus�result2�	alias_mapr{rrr�_apt_cache_modalias_map�sV
	


��

 ��	���r�cCstSr")�custom_supported_gpus_jsonrrrr�path_get_custom_supported_gpus
sr�c	Cs�t�}d}z_t|d��O}z-tt�|�d�}|D]}|d|kr7|d�d�d}t�d||d	|f�nqWntyGt�	d
�Yn
wWd�W|SWd�W|S1s]wYW|Styut�	d|�Y|Sw)z|Get nvidia allowing driver for specific devices.

    did: 0x1234
    Return the situable nvidia driver version for it.
    N�r�chips�devid�branch�.rz0Found a specific nv driver version %s for %s(%s)rz;package_get_nv_allowing_driver(): unexpected json detected.z3package_get_nv_allowing_driver(): unable to read %s)
r�r:�list�json�loadrAr�info�	Exceptionr)�didr;�version�stream�gpus�gpurrr�package_get_nv_allowing_drivers@���
��������r�csVt�}tdd��jD��}ztj|}Wnty&t��}|tj|<Ynw|�|�dd�ddif�\}}t	|�\}}d}	d}
|dkrzt
d|�}	d	|	}d
|}�jD]}
|
���d�d|krlt|g�||<d}
qV|	durz|
szt�
d|�|
s�|dus�|�|�s�gS|D]}t�|��|���r�||D]}
|�|
�q�q��fdd
�|D�S)z_Search packages which match the given modalias.

    Return a list of apt.Package objects.
    css�|]}|��VqdSr")�get_fullname)rRr`rrr�	<genexpr>0s�z(packages_for_modalias.<locals>.<genexpr>rwrrN�10DE�0xznvidia-driver-%szpci:v000010DEd0000%s*z%s is not in the package pool.csg|]}�|�qSrr�rR�p�r_rrrTOsz)packages_for_modalias.<locals>.<listcomp>)r��hashr}�packages_for_modalias�
cache_mapsrWr�r[rA�_get_vendor_model_from_aliasr�r�rrrry�fnmatchcase�lowerr�)r_r2�pkgs�apt_cache_hash�	cache_mapr{�bus_map�vidr��nvamd�found�nvamdn�nvamdar�r�rr�rr�)s@� 
��r�c	Cs�t�|�}|�|�}|dusJ�|j�d�rdSt�|�}|�|jd�|jdjD]&\}}|j	sIz|d}|dvWSt
yHYdSw|j	dvrQdSq+dS)NrFr�	Component)�
restricted�
multiverseT)r/rXrYrr@r|r~r�version_list�	componentrW)r_�pkgrcrdr��pfile�_r�rrr�_is_package_freeUs&


�
�r�cCsLt�|�}|�|�}|durdSz|jddjdkWSty%YdSw)NFr�Ubuntu)r/rXrYr�originrW)r_r�rcrdrrr�_is_package_from_distrors

�r�c	Cs�t�|�}|�|�}t�|�}|�|jd�z|d}Wnttfy0t�	d|j
�YdSw|dvr?t�	d|j
|�dS|S)z>Determine if -open package is prefered from apt Package objectrzPrefer-VariantzM_pkg_get_open_preference %s: package has Prefer-Open header, cannot determineF)�Open�Closedzb_pkg_get_open_preference %s: package has invalid Prefer-Open %sheader, cannot determine preferenceN�r/rXrYr|r~rrWr�rrr)r_r�rcrdr��
preferencerrr�_pkg_get_open_preference~s 


��r�c	Cs�t�|�}|�|�}t�|�}|�|jd�z|d}Wnttfy0t�	d|j
�YdSw|�d�}|dkrCt�	d|j
�dS|d|�}|S)z-Determine module name from apt Package objectrrqzM_pkg_get_module %s: package has no Modaliases header, cannot determine moduleNruzR_pkg_get_module %s: package has invalid Modaliases header, cannot determine module)r/rXrYr|r~rrWr�rrrr)r_r�rcrdr�r��parenr7rrr�_pkg_get_module�s 


�
r�c	Cs�t�|�}|�|�}t�|�}|�|jd�z|d}Wnttfy0t�	d|j
�YdSw|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 levelr�)r_r�rcrdr��supportrrr�_pkg_get_support�s 


��r�c		Csjt�d|�}|�d�}|�d�}|dkrdSd|}t�}z�t|d��v}zTtt�|�d�}|D]F}|d	|krvd
|dvrv|d�d
�d|krdt	�
d|d�d
�d|f�WWd�WdSt	�d|�WWd�WdSq0Wnty�t	�
d�Yn
wWd�WdSWd�WdS1s�wYWdSty�t	�
d|�YdSw)zAalias: e.g. pci:v000010DEd000024BAsv0000103Csd000089C6bc03sc00i00zpci:v0000(.*)d0000(.*)sv(.*)rrr�Fz0x%sr�r�r��	runtimepm�featuresr�r�rz)Candidate version does not match %s != %sNzFound runtimepm supports on %s.Tz?_is_nv_allowing_runtimepm_supported(): unexpected json detectedz8_is_nv_allowing_runtimepm_supported(): unable to read %s)
r�searchrr�r:r�r�r�rArrr�r�)	r��verr�r�r�r;r�r�r�rrr�#_is_nv_allowing_runtimepm_supported�sR

��	��
��������r�c
s�z+t�|�}|�|�}t�|�}|�|jd�t�|j�}|j�	d�d}|d}Wn
t
ttfy8YdSw|�
d�dkrBdS||�
d�d|�
d��}	|	�	d	�}
t�|�r\d
St�fdd�|
D��S)
z>Check if the package supports runtimepm for the given modaliasrr��	PmAliasesFznvidia(rurrsrtTc3s$�|]
}t����|���VqdSr")ryr�)rR�regex�r�rrr��s�"z*_is_runtimepm_supported.<locals>.<genexpr>)r/rXrYr|r~r�
TagSection�record�ver_strrArWr�r�rr��any)r_r�r�rcrdr��sectionr�r��n�
modaliasesrr�r�_is_runtimepm_supported�s$


�

r�cCstj�d�duS)z*Check if the current session in on Wayland�WAYLAND_DISPLAYN)r8�environr[rrrr�is_wayland_session�sr�c
Cs�|jrdS|j�d�rd}n|j�d�rd}nt||�}|s dSztjd|gtjtjd�}|��Wntt	tj
fyQ}zt�dt
|��WYd}~dSd}~ww|jdkrat�d	|j|�d
St�d|j|�dS)zIDetermine if the kernel module from an apt.Package is manually installed.Fr�fglrx�modinfo)�stdout�stderrz_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)�current_verrr@r��
subprocess�Popen�PIPE�communicate�OSError�FileNotFoundError�CalledProcessErrorrr�str�
returncode)r_r�r7r�rIrrr�_is_manual_install�s8
���
��r�c	
Cs�ztjddd|gdd�}Wn ttjfy.}zt�d||t|��WYd}~dSd}~wwt�d	|�d}d}|��D]}|�d
d�\}}d|vrM|}d
|vrS|}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
r5r�_VENDOR�_MODELz-_get_db_name(%s, %s): vendor "%s", model "%s")	r��check_outputr�r�rrr��
splitlinesrA)	�syspathr��outrI�vendor�modelrK�k�vrrr�_get_db_names.
�����rcCsx|j�d�sdSt||�}|dur&t|j�}|��dkr%t�d|j�dSn
|dkr3t�d|j�dSt�d|j�dS)N�nvidia-Fi0z_is_open_prefered(%s): TrueTr�z_is_open_prefered(%s): False)rr@r�rr(rr)r_r�r��nvidia_inforrr�_is_open_prefered8s

�r
cCs,d}d|}t|d�}|�|�|��dS)zSet 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)r:�write�close)�value�nvidia_kms_file�kms_text�kms_fdrrr�set_nvidia_kmsJs�

rFTcCs�t|�}|s'zt�d�}Wnty&}z
t�|�iWYd}~Sd}~wwi}|��D]X\}}t||�D]N}	|r@t||	�s@q6|sJt	�	|	j
d�rJq6||t||	�t||	�t||	�t
||	|�t||	�d�||	j
<t||�\}
}|
dury|
||	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�||	<Wq�ty�t�d
|	�Yq�wq�|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)r2r�free�from_distror�r��open_preferredrrcS�g|]	}|�d�r|�qS�r�r@r�rrrrT��z*system_driver_packages.<locals>.<listcomp>r��keyr�recommended)rr�pluginz*Package %s plugin not available. Skipping.)rNr/�Cacher�rr�r�r�r�ryrr�r�r�r
rr@�lookup_cache�sort�	functoolsr�_cmp_gfx_alternatives�detect_plugin_packagesrWr)r_rC�freeonly�include_oemr��exr}r�rr�rr�nvidia_packagesrrrrr��apt_prrr�system_driver_packagesUsl%
��
�	��
����r*cCs@t�d�}|�|�}|r|�d�dd�|�d�dd�fSdS)Nz$(.+):v(.+)d(.+)sv(.+)sd(.+)bc(.+)i.*r��r�)rrrr)r��modalias_pattern�detailsrrrr��s


$r�c
Cs�|dusJ�d}	t�|�}|j}t|�}|js!t�d|�|S|��r.t�d|�|S|��}d|}z||}|�	|�}	|rO|	j
dt�fvrR|}W|SW|SW|Sty^Y|Sw)Nz)Unsupported driver detected: %s. Skipping�%Legacy driver detected: %s. Skipping.znvidia-headless-no-dkms-%srr)
r/rXrrr
rrr$r)rY�archr1rW)
r�r_�metapackagercrr	�candidate_flavourrdr`�package_candidaterrr�!_get_headless_no_dkms_metapackage�s:

����r4c	Cs�|siSt|�}|s+zt�d�}Wnty*}z
t�|�iWYd}~Sd}~wwi}|��D]-\}}t||�D]#}t�|j	d�sDq:||t
||�t||�dt||�t
||�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.
    NrT)r2rrrrr�r)rNr/rr�rr�r�r�ryrr�r�r�r
)	r_rCr&r�r'r}r�rr�rrr�#system_device_specific_metapackages�s4 
����
r5c
Cs�dg}t|�}|s*zt�d�}Wnty)}z
t�|�iWYd}~Sd}~wwi}|��D]a\}}t||�D]W}t||�\}	}
t	|�\}}|dur�|�
�|vr�||t||�t||�t
||�t||�d�||j<|	duru|	||jd<|
dur�|
||jd<t||�}
|
dur�|
||jd<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)r2rrrr�rrrr1cSrrrr�rrrrThrz0system_gpgpu_driver_packages.<locals>.<listcomp>rrrr)rNr/rr�rr�r�r�rr�r�r�r�r�r
rr4r@r r!r"r�_cmp_gfx_alternatives_gpgpu)r_rC�vendors_whitelistr�r'r}r�rr�rr�	vendor_id�model_idr1r(rrrrrr�system_gpgpu_driver_packages$sV"
���
��
�r;c
Cs2i}|s%zt�d�}Wnty$}z
t�|�iWYd}~Sd}~wwt|||d���D]G\}}d|vr;|d}n|d}|�|i�dD]}||vrU|||||<qG||�di�}	|d|dd	�|	|<d
|vru|d
|	|d
<q.|��D]\}
}|dD]}t|||�s�nq�d|d<qzt	|�|S)
uGet 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%rr)r2rr�driversrr)rrrT�manual_install)
r/rr�rr�r*r�r�r��
_add_builtins)r_rCr%r�r'r��pkginfo�device_name�opt_keyr<r6r�rrr�system_device_driverswsH5
���
�
����rBc	CsXt||||d�}t||dd�}|st�d�|St�|�}t�|�}g}	t|��t	dd��dd�D]x\}
}||
}|j
s�|�|�}
|�|
j
d	�|
�d
�rXtdd�|	D��rXq1|	�|
�|d
r{ztdd�}|�d�|��Wn	tyzYnwz$t||
�}|r�||j
s�|s�d|vr�|	�|
�Wq1|	�|�W|	SWq1ty�Yq1wq1|	S)z4Return the list of packages that should be installed)r%r&F)�get_recommendedz"No drivers found for installation.cSst|d|d�S)Nr)r#)�left�rightrrr�<lambda>�rUz*get_desktop_package_list.<locals>.<lambda>T)r�reverserrOcss�|]}|�d�VqdS)rONr)rRr�rrrr��s�z+get_desktop_package_list.<locals>.<genexpr>r��/run/nvidia_runtimepm_supportedr�
�dkms)r*�auto_install_filterrrr/rXr|�sortedr�rr�rYr~rr@r��appendr:rr
�PermissionError�get_linux_modules_metapackage�removerW)r_rC�	free_onlyr&�
driver_string�include_dkmsr}rcr��
to_installr�r��package_objrd�pm_fd�modules_packagerrr�get_desktop_package_list�s\�



�




�


����#rXcCs0|D]}t|�}|��dkrtd�dSqdS)zHApplies changes that need to happen before installing the NVIDIA driversi�rN)rr(r)rT�package_namer	rrr�$nvidia_desktop_pre_installation_hooks��rZcCsztj�d�r;t�d�z	t�ddg�Wn	tyYnwtdd��}|�	d�Wd�dS1s4wYdSdS)NrHz,Trying to select the on-demand PRIME profilez/usr/bin/prime-selectz	on-demandz$/etc/u-d-c-nvidia-runtimepm-overriderz!# File created by ubuntu-drivers
)
r8r;�isfilerrr��callr�r:r)rHrrr�%nvidia_desktop_post_installation_hooks
�"��r]c@seZdZddd�Zdd�ZdS)�_GpgpuDriverNcCsd|_||_||_dS)N)r)�_vendors_whitelistr�flavour)rrr`rrrr)s
z_GpgpuDriver.__init__cCs,|jr
t�|j|j�s
dS|jo|jS)NF)rry�filterr_r`r#rrrr
.sz_GpgpuDriver.is_validr�)r*r+r,rr
rrrrr^'s
r^cCs�t�}t�d�}t�d�}t�d�}|�|�}|�|�}|�|�}|r8|�d�|_d|�d�|�d�f|_|S|r?||_|S|rNd|�d�|�d�f|_|S)zReturns a _GpgpuDriver objectz(.+):([0-9]+)(.*)z([a-z]+)z([0-9]+)(.*)rz%s%srr,)r^rrrrrr`)�stringr6�full_pattern�vendor_only_pattern�series_only_pattern�
full_match�vendor_match�series_matchrrr�_process_driver_string6s"





��ric
Cs�g}g}i}	|s|S|r%|�d�D]}t|�}|r#|��r#|�|�qnt�}|�|�t|�dkr5|Sd}|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|r�||�d�r�||||<n||||<nq�q�|S)	Nrvrrrz0Multiple nvidia versions passed at the same timez%s*%sz%s*r)rArir
rMr^�lenr�__contains__rrr`�extendryrar[)
r}�drivers_strrCr<�allowr��itemr6�it�vendors_temprr r�rrr�gpgpu_install_filterMsn
��









���
rrc	Cs�gd�}|r
t||�}|Sg}|D]}|�t�||��qi}|D]}|r8d||vs1||dr7||||<q!||||<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*rr)rrrlryra)	r}rmrC�	whitelist�resultsrnr r�r�rrrrK�s

�rKcCs�i}tj�dd�}tj�|�st�d|�|S|dur<zt�d�}Wnt	y;}z
t�
|�iWYd}~Sd}~wwt�|�D]�}|�d�sIqAtj�
||�}t�d|�i}t|���}ztt|��|d�|�|d|�}t�d	||�Wnt	y�t�d
|�YWd�qAw|dur�	Wd�qAt|�ttfvr�t�
d|t|��	Wd�qA|D]&}	z||	}
t||
�r�|�|g��|	�Wq�ty�t�d|	|�Yq�wWd�n1s�wYqA|S)
aaGet 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)r8r�r[r;�isdirrrr/rr�r��listdir�endswithr<r:rvrr=�	exception�typer�r�rpr�rMrW)r_r}�	plugindirr'�fnamer�symbrHr�r�r`rrrr$�sf�
��

���
�����r$cC�t�|�r
t�|��d�SdS)z5Look up driver package and return their support levelr�N�r r[�rSrrr�_pkg_support_from_cache�
r�cCr�)z6Look up driver and return if open package is preferredrFr�r�rrr�_pkg_open_preferred_from_cacher�r�cC�t|�r&t�d|�rdSt�d|�rdSt�d|�rdSt�d|�r$dSdSt�d|�r.dSt�d|�r6dSt�d|�r>dSt�d|�rFdSdS)	z4decide how well the package fits desktop environment�
(\d{3})-open$r,�(\d{3})$r�(\d{3})-server-open$r�(\d{3})-server$r�r�rr�r�rrr�_get_fit_level �&
�r�cCr�)	z3decide how well the package fits server environmentr�r,r�rr�rr�rr�r�rrr�_get_fit_level_gpgpuAr�r�cCs|�d�r�|�d�r�t|�t|�krdSt|�t|�krdSddg}d}d}t|�}t|�}||vr6|d7}||vr>|d7}||krG|d7}n||krO|d7}|dksW|dkrc||kr]dS||krcdSt|�dkrqt|�dkrqdSt|�dkrt|�dkrdSt|�dkr�t|�dkr�dSt|�dkr�t|�dkr�dSt|�dkr�t|�dkr�dSt|�dkr�t|�dkr�dSt|�d	kr�t|�d	kr�dSt|�d	kr�t|�d	kr�dSt|�d
kr�t|�d
kr�dSt|�d
kr�t|�d
kr�dS||kr�dS||kr�dS||k�sJ�dS)a<Compare two graphics driver names in terms of preference. (desktop)

    -open by default sorts after non-open, unless a driver preferes open variant
    -server always sorts after non-server.
    LTSB (Long Term Support Branch) always sorts before NFB (New Feature Branch).
    Legacy always sorts before Beta.
    rrrr�r�r�dr�r��Beta)r@r�r�)rS�y�preferred_support�x_score�y_score�	x_support�	y_supportrrrr#bsb	
r#cCs||�d�r�|�d�r�t|�t|�krdSt|�t|�krdSt|�dkr,t|�dkr,dSt|�dkr:t|�dkr:dSt|�dkrHt|�dkrHdSt|�dkrVt|�dkrVdSt|�dkrdt|�dkrddSt|�dkrrt|�dkrrdSt|�dkr�t|�dkr�dSt|�dkr�t|�dkr�dSt|�dkr�t|�dkr�dSt|�dkr�t|�dkr�dS||kr�dS||kr�dS||ks�J�d	S)
auCompare two graphics driver names in terms of preference. (server)

    -open by default sorts after non-open, unless a driver preferes open variant
    -server always sorts before non-server.
    non server package always sorts after server package
    LTSB (Long Term Support Branch) always sorts before NFB (New Feature Branch).
    Legacy always sorts before Beta.
    rrrr�r�r�r�r�r)r@r�r�)rSr�rrrr7�s>
r7cCs�|��D]>\}}|dD]5}|�d�r ddddd�|dd<n"|�d�rA|dD]
}d|d|d<q)ddddd�|dd	<nqqd
S)zAdd builtin driver alternativesr<rTF)r�builtinrrzxserver-xorg-video-nouveaur�rzxserver-xorg-video-atiN)r�r@)r<�devicer�r��drrrr>�s
�
����r>cC�t�|�}|��S)z0Return the linux headers for the system's kernel)r�KernelDetection�get_linux_headers_metapackage�r_�kernel_detectionrrr�get_linux_headers��
r�cCr��z.Return the linux image for the system's kernel)rr��get_linux_image_metapackager�rrr�get_linux_image�r�r�cCr�r�)rr��get_linux_versionr�rrrr��r�r�cCr�)z4Return the linux metapackage for the system's kernel)rr��get_linux_metapackager�rrr�	get_linux�r�r�c	Cs�t�|�}z	|�||�}Wntyt�d|�YdSwz |j�d�D]}|D]\}}}|�d�r<|WSq+q'WdStt	fySt�d|�YdSw)NzNo candidate for %s foundrP�linux-image-z+Could not check dependencies for %s package)
r/rXrYrWrrrZr[r@r\)r_r�rcrdrfrgrhrirrr�get_linux_image_from_metas(
�
�����r�cCs&|dusJ�d}d}d}t�|�}t|�}|js!t�d|�|S|��r.t�d|�|St|�}t||�}|r@|�	dd�}n	t�d|�|S|�
�}	d|	|f}
zB||
}|�|�}|r�|jdt
�fvr�t|�}
d|	|
f}t�d	|�||}|�|�}|r�|jdt
�fvr�t�d
|�|
}Wnty�t�d|
�Ynw|r�dd
�||jD�}d}t|��	dd�}|dkr�d}d|	|f}|D]}||kr�|}q�|r�|}|Sd|	}t�d|�z||}|�|�}|r�|jdt
�fvr�|}W|SW|SW|St�yt�d|�Y|Sw)zDReturn the linux-modules-$driver metapackage for the system's kernelNrzNNon 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-%srrzlinux_modules_abi_candidate: %szFound ABI compatible %szNo "%s" can be found.cSs"g|]
}|jj�d�r|jj�qS)zlinux-modules-nvidia-)rQrr@)rRrorrrrT]s�z1get_linux_modules_metapackage.<locals>.<listcomp>�virtual�genericznvidia-dkms-%szFalling back to %s)r/rXrr
rrr$r�r�rVr)rYr0r1r�rWr^r�)r_rdr1�
linux_flavour�linux_modules_matchrcr	�linux_image_meta�linux_imager2�linux_modules_candidater`r3�
linux_version�linux_modules_abi_candidate�abi_specific�abi_specific_candidate�reverse_deps�pick�candidate_suffix�modules_candidatero�dkms_packagerrrrO"s�



���
����rOr")NNFT)NNTr�)NNF)NFTrF)T)rT)@r-r8rryr�r"rr�r/�
UbuntuDriversrrr.r r��objectrr1rNrpr�r�r�r�r�r�r�r�r�r�r�r�r�r�rr
rr*r�r4r5r;rBrXrZr]r^rirrrKr$r�r�r�r�r#r7r>r�r�r�r�r�rOrrrr�<module>s�
F
/88)#
a
%
>
S]
�;


g
 >!!E-