o

���b:U�@s�dd
lZdd
l
Z
dd
lZdd
lZdd
lZdd
lZdd
lZdd
lZddlm	Z	
ddl
mZmZm
Z
mZmZmZmZ
ddlmZ
ddlmZmZmZmZ
ddlmZmZ
ddlmZ
ddlmZ 
dd	l!m"Z"
d
d�Z#id
d
dd
d
fdd�
Z$Gdd�d�Z%d
S)�N)
�getpwuid)�	NO_PRESET�GENERIC_PRESETS�PRESETS_PATH�PresetDefaults�DESC�NOTE�OPTS)
�PackageManager)�ImporterHelper�
import_module�get_human_readable�bold)�IndependentPlugin�ExperimentalPlugin)
�
SoSOptions)
�_sos)
�fillc
Cs*d
|}
zt|
t
�WSty


YdSw)Nzsos.policies.distros.%s)r�Policy�ImportError)�name�
policy_fqname�r�7/usr/lib/python3/dist-packages/sos/policies/__init__.py�
import_policys




�rT�c
Cs�d
|vr	|�d
�
Sddl
}t|jj�
}|��D]}t|�
D]}	|	j|d�
r/|	|
|||d�|d
<qqtj	dkr:t
d�
�
d
|v
rF|jj��|d
<|d
S)N�policyr
)
�remote)�sysroot�init�
probe_runtime�remote_exec�linuxz%SoS is not supported on this platform)�get�sos.policies.distrosr�policies�distros�get_modulesr�check�sys�platform�	Exception�GenericLinuxPolicy)
�cacherrr r!�remote_check�sos�helper�modulerrrr�loads$









���


r2c@s|
eZ
dZd
Zed�
ZdZdZdg
ZdZ	dZ
dZdZde
�i
ZeZdZdMdd�
ZdNd
d�
Zedd��
Zedd��
Zdd�Zdd�Zdd�Zdd�Zdd�Zdd�Zdd �Zd!d"�ZdOd#d$�
Z d%d&�Z!d'd(�Z"d)d*�Z#d+d,�Z$d-d.�Z%d/d0�Z&d1d2�Z'd3d4�Z(d5d6�Z)ed7d8��
Z*				dPd9d:�
Z+dQd;d<�
Z,d=d>�Z-d?d@�Z.dOdAdB�
Z/dCdD�Z0dEdF�Z1dRdGdH�
Z2d	d	d	e3�fdIdJ�
Z4dNdKdL�
Z5d	S)Sra.Policies represent distributions that sos supports, and define the way
    in which sos behaves on those distributions. A policy should define at
    minimum a way to identify the distribution, and a package manager to allow
    for package based plugin enablement.

    Policies also control preferred ContainerRuntime()'s, upload support to
    default locations for distribution vendors, disclaimer text, and default
    presets supported by that distribution or vendor's products.

    Every Policy will also need at least one "tagging class" for plugins.

    :param sysroot: Set the sysroot for the system, if not /
    :type sysroot: ``str`` or ``None``

    :param probe_runtime: Should the Policy try to load a ContainerRuntime
    :type probe_runtime: ``bool``

    :cvar distro: The name of the distribution the Policy represents
    :vartype distro: ``str``

    :cvar vendor: The name of the vendor producing the distribution
    :vartype vendor: ``str``

    :cvar vendor_urls: List of URLs for the vendor's website, or support portal
    :vartype vendor_urls: ``list`` of ``tuples`` formatted
        ``(``description``, ``url``)``

    :cvar vendor_text: Additional text to add to the banner message
    :vartype vendor_text: ``str``

    :cvar name_pattern: The naming pattern to be used for naming archives
                        generated by sos. Values of `legacy`, and `friendly`
                        are preset patterns. May also be set to an explicit
                        custom pattern, see `get_archive_name()`
    :vartype name_pattern: ``str``
    ap
This command will collect system configuration and diagnostic information from this %(distro)s system.

For more information on %(vendor)s visit:

  %(vendor_urls)s

The generated archive may contain data considered sensitive and its content should be reviewed by the originating organization before being passed to any third party.

%(changes_text)s

%(vendor_text)s
�Unknown)zExample URLzhttp://www.example.com/rz</usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin�legacyFNTcCsL|��
d
|_
||_t�|_tg
|_||_|js|��
|
|_	|�
t�

d
S)aJ
Subclasses that choose to override this initializer should call
        super() to ensure that they get the required platform bits attached.
        super(SubClass, self).__init__(). Policies that require runtime
        tests to construct PATH must call self.set_exec_path() after
        modifying PATH in their own initializer.N)�_parse_uname�case_idr r
�package_managerr�valid_subclassesr!�
set_exec_pathr�register_presetsr)�selfrr r!rrr�__init__ws








zPolicy.__init__c
C�d
S)a�

        This function is responsible for determining if the underlying system
        is supported by this policy.

        If `remote` is provided, it should be the contents of os-release from
        a remote host, or a similar vendor-specific file that can be used in
        place of a locally available file.

        :returns: ``True`` if the Policy should be loaded, else ``False``
        :rtype: ``bool``
        Fr)r;rrrrr(�szPolicy.checkc
CsFt|d
�sg|_
|jjD]}
t|
d�r|j
�|
���

qtt|j
�
�
S)a#
This property is used to determine the list of forbidden paths
        set by the policy. Note that this property will construct a
        *cumulative* list based on all subclasses of a given policy.

        :returns: All patterns of policy forbidden paths
        :rtype: ``list``
        �_forbidden_paths�set_forbidden_paths)�hasattrr>�	__class__�__mro__�extendr?�list�set)r;�clsrrr�forbidden_paths�s
	




�
zPolicy.forbidden_pathsc

Csgd
�
S)z�Use this to *append* policy-specifc forbidden paths that apply to
        all plugins. Setting this classmethod on an invidual policy will *not*
        override subclass-specific paths
        )z*.eggz*.pycz*.pyoz*.swpr)
rFrrrr?�szPolicy.set_forbidden_pathsc


C�|jS)
z�Are we running inside a container?

        :returns: ``True`` if in a container, else ``False``
        :rtype: ``bool``
        )
�
_in_container�
r;rrr�in_container�szPolicy.in_containerc


Cr=)z'
        Return the OS version
        NrrJrrr�dist_version��zPolicy.dist_versionc
Csd
dlm
}

|
S)zb
        Return the class object of the prefered archive format for this
        platform
        r
)
�TarFileArchive)�sos.archiverN)r;rNrrr�get_preferred_archive�s
zPolicy.get_preferred_archivec
Cs�|���
d
�
d}
|j}|jdj}d}d�dd�td�
D�
�
}|jdkr2d	}|r-d
|n
d}d
}n|jdkrLd}|r?d
|n
d}|rGd
|n
d}d}n|j}|j|
||||d�}|�	t
�|�
�
S)aq
        This function should return the filename of the archive without the
        extension.

        This uses the policy's `name_pattern` attribute to determine the name.
        There are two pre-defined naming patterns - `legacy` and `friendly`
        that give names like the following:

        * legacy - `sosreport-tux.123456-20171224185433`
        * friendly - `sosreport-tux-mylabel-123456-2017-12-24-ezcfcop.tar.xz`

        A custom name_pattern can be used by a policy provided that it
        defines name_pattern using a format() style string substitution.

        Usable substitutions are:

            * name  - the short hostname of the system
            * label - the label given by --label
            * case  - the case id given by --case-id or --ticker-number
            * rand  - a random string of 7 alpha characters

        Note that if a datestamp is needed, the substring should be set
        in `name_pattern` in the format accepted by ``strftime()``.

        :returns: A name to be used for the archive, as expanded from
                  the Policy `name_pattern`
        :rtype: ``str``
        �
.r
�cmdlineoptsrc
ss�|]	}
t�
tj�
V
qdS�
N)�random�choice�string�ascii_lowercase)�.0�
xrrr�	<genexpr>�s�z*Policy.get_archive_name.<locals>.<genexpr>�r4zsosreport-{name}{case}{date}z
-%Y%m%d%H%M%S�friendlyz*sosreport-{name}{label}{case}{date}-{rand}�
-z	-%Y-%m-%d)r�label�case�date�rand)�get_local_name�splitr6�commonsr^�join�range�name_pattern�format�sanitize_filename�time�strftime)r;rr_r^r`ra�nstrrrr�get_archive_name�s.


















�zPolicy.get_archive_namec
Cs|
SrSr)r;�binaryrrr�_get_pkg_name_for_binary
s
zPolicy._get_pkg_name_for_binarycCs|
st�
�S|
SrS)�tempfile�
gettempdir)r;�opt_tmp_dirrrr�get_tmp_dir
s


zPolicy.get_tmp_dirc


CrHrS)
�default_scl_prefixrJrrr�get_default_scl_prefix
s
zPolicy.get_default_scl_prefixcCs6t|
�
d
kr|
D]}t
||jd�r|
Sq|
dS)az
Determine what subclass of a Plugin should be used based on the
        tagging classes assigned to the Plugin

        :param plugin_classes: The classes that the Plugin subclasses
        :type plugin_classes: ``list``

        :returns: The first tagging class that matches one of the Policy's
                  `valid_subclasses`
        :rtype: ``PluginDistroTag``
        �
r
)�len�
issubclassr8)r;�plugin_classes�
prrr�match_plugin
s

�zPolicy.match_plugincs0tg
|j
}|r
|tg
7}t�f
d
d�|D�
�
S)a#

        Verifies that the plugin_class should execute under this policy

        :param plugin_class: The tagging class being checked
        :type plugin_class: ``PluginDistroTag``

        :returns: ``True`` if the `plugin_class` is allowed by the policy
        :rtype: ``bool``
        c
3s�|]}
t�|
�V
qdSrS)
rx)rX�class_�
�plugin_classrrrZ0
s�
�z)Policy.validate_plugin.<locals>.<genexpr>)rr8r�any)r;r~�experimentalr8rr}r�validate_plugin#
s





�zPolicy.validate_pluginc


Cr=)z>
        This function is called prior to collection.
        NrrJrrr�pre_work3
rMzPolicy.pre_workc


Cr=)zQ
        This function is called after the sosreport has been generated.
        NrrJrrr�	post_work9
rMzPolicy.post_workcCs|j�
|
�
S)
z�Wrapper to retrieve a package from the Policy's package manager

        :param pkg: The name of the package
        :type pkg: ``str``

        :returns: The first package that matches `pkg`
        :rtype: ``str``
        )r7�pkg_by_name)r;�pkgrrrr�?
s	zPolicy.pkg_by_namec
CsBt�
�\}
}}}}}|
|_||_||_|��d
dk|_||_dS)Nrv�SMP)r*�uname�system�hostname�releaserc�smp�machine)r;r��noder��versionr��	processorrrrr5J
s�






zPolicy._parse_unamecCs
|
|_d
S)z9Set common host data for the Policy to reference
        N)
rd)r;rdrrr�set_commonsS
s
zPolicy.set_commonscCs|
tj
d
<dS)N�PATH)�os�environ)r;�pathrrr�	_set_PATHX
s
zPolicy._set_PATHc

Cs|�|j
�

dSrS)r�r�rJrrrr9[
s
zPolicy.set_exec_pathc

Cst�
�d
kS)z�This method should return true if the user calling the script is
        considered to be a superuser

        :returns: ``True`` if user is superuser, else ``False``
        :rtype: ``bool``
        r
)r��getuidrJrrr�is_root^
szPolicy.is_rootc


Cr=)zRReturns the string name of the hashlib-supported checksum algorithm
        to use�sha256rrJrrr�get_preferred_hash_nameg
szPolicy.get_preferred_hash_namecCs�|
�d
�

|
�
d�

|
�
dtd�
td�
f�

|
�
d�

ddd	d
d�}|
�d�
}|�
d
�

|D]}|j
d�d|||�dd�
q/dS)NzSoS Policiesz�Policies help govern how SoS operates on across different distributions of Linux. They control aspects such as plugin enablement, $PATH determination, how/which package managers are queried, default upload specifications, and more.a�
When SoS intializes most functions, for example %s and %s, one of the first operations is to determine the correct policy to load for the local system. Policies will determine the proper package manager to use, any applicable container runtime(s), and init systems so that SoS and report plugins can properly function for collections. Generally speaking a single policy will map to a single distribution; for example there are separate policies for Debian, Ubuntu, RHEL, and Fedora.z
sos reportzsos collectzRIt is currently not possible for users to directly control which policy is loaded.z*The Google Cloud-Optimized OS distributionzThe Debian distributionz=Red Hat family distributions, not necessarily including forkszUbuntu/Canonical distributions)zpolicies.coszpolicies.debianzpolicies.redhatzpolicies.ubuntuzSee Alsoz9For more information on distribution policies, see below
z{:>8}{:<20}{:<30}�
 F)
�newline)�	set_title�add_textr�add_sectionrh)r;�section�pols�seealso�polrrr�display_helpl
s4


�
��
�

�


�

��zPolicy.display_helpcCs�|
s|sd
S|��
|r|�t
d|�
�

|
r<|jt
d�
|
dd�
|�t
d�
t|j�
�

|�t
d�
t|j�
j�

n|jt
d�
|dd�
|ra|�d	|��d
|�

|��
|�t
d�
�

|��
dS)
ahDisplay final information about a generated archive

        :param archive: The name of the archive that was generated
        :type archive: ``str``

        :param directory: The build directory for sos if --build was used
        :type directory: ``str``

        :param checksum: The checksum of the archive
        :type checksum: ``str``

        :param archivestat: stat() information for the archive
        :type archivestat: `os.stat_result`

        :param map_file: If sos clean was invoked, the location of the mapping
                         file for this run
        :type map_file: ``str``
        Fz5A mapping of obfuscated elements is available at
	%s
z4Your sosreport has been generated and saved in:
	%s
T)
�alwaysz Size	%sz	 Owner	%sz5Your sosreport build tree has been generated in:
	%s
r��
	z5Please send this file to your support representative.N)�_print�
_r
�st_sizer�st_uid�pw_namer�)r;�archive�	directory�checksum�archivestat�map_filerrr�display_results�
s:


�


�
�

�


�

�
�


zPolicy.display_resultscCs.|s|jd
j
s|
rt|
�

dSt�
dSdS)zSA wrapper around print that only prints if we are not running in
        quiet moderRN)rd�quiet�print)r;�msgr�rrrr��
s



�z
Policy._printc
Csn|jd
j
r	d}
nd}
d}|j|j|j|��|j|jd|
d�}d}|��D]
}|t||dd	�d
}q'|S)aU
This method is used to prepare the preamble text to display to
        the user in non-batch mode. If your policy sets self.distro that
        text will be substituted accordingly. You can also override this
        method to do something more complicated.

        :returns: Formatted banner message string
        :rtype: ``str``
        rRz,Changes CAN be made to system configuration.z0No changes will be made to system configuration.�H�tmpdir)�distro�vendor�vendor_urls�vendor_textr��changes_textrF)
�replace_whitespace�

)	rd�allow_system_changesr�r�r��_fmt_vendor_urlsr��
splitlinesr)r;r��width�_msg�_fmt�linerrr�get_msg�
s	






�


zPolicy.get_msgc

s.td
d�|j
D�
�
�d��f
dd�|j
D�
�
S)z�Formats all items in the ``vendor_urls`` class attr into a usable
        string for the banner message.

        :returns:   Formatted string of URLS
        :rtype:     ``str``
        c
Ssg|]}
t|
d�
�qS)
r
)
rw)rX�
vrrr�
<listcomp>�
sz+Policy._fmt_vendor_urls.<locals>.<listcomp>r�c
3s(�|]}
dj|
d
�|
dd�V
qdS)z	{desc:<{width}} : {url}r
rv)�descr��urlN)
rh)rX�
u�
r�rrrZ�
s��
�z*Policy._fmt_vendor_urls.<locals>.<genexpr>)�maxr�rerJrr�rr��
s
�zPolicy._fmt_vendor_urlscCs|ri|_|j�
|
�

d
S)aAdd new presets to this policy object.

            Merges the presets dictionary ``presets`` into this ``Policy``
            object, or replaces the current presets if ``replace`` is
            ``True``.

            ``presets`` should be a dictionary mapping ``str`` preset names
            to ``<class PresetDefaults>`` objects specifying the command
            line defaults.

            :param presets: dictionary of presets to add or replace
            :param replace: replace presets rather than merge new presets.
        N)�presets�update)r;r��replacerrrr:�
s

zPolicy.register_presetscCs*|j�
�D]
}||
kr|j|
Sqd
S)z�Find a preset profile matching the specified preset string.

            :param preset: a string containing a preset profile name.
            :returns: a matching PresetProfile.
        N)r��keys)r;�preset�matchrrr�find_preset	s


�zPolicy.find_presetc

Cs
|jt
S)
z�Return a ``PresetDefaults`` object matching the runing host.

            Stub method to be implemented by derived policy classes.

            :returns: a ``PresetDefaults`` object.
        )r�rrJrrr�probe_presets
zPolicy.probe_presetc	
Cs
|
p|j}
t
j�|
�
s
d
St
�|
�
D]z}t
j�|
|�}t|�
�}zt�|�
}Wnt	y6


YWd
�
qwWd
�
n1sAw



Y
|�
�D]A}t|t�d�}||}t
|vr_|t
n
d|_t|vrj|tn
d|_t|vr�|tD]
}t|j||t|�
qud|_||j|<qJqd
S)z�Load presets from disk.

            Read JSON formatted preset data from the specified path,
            or the default location at ``/var/lib/sos/presets``.

            :param presets_path: a directory containing JSON presets.
        N)
�optsrF)�presets_pathr�r��exists�listdirre�open�jsonr2�
ValueErrorr�rrrr�r�noter	�setattrr��builtinr�)	r;r��preset_path�pf�preset_datar��pd�data�argrrr�load_presetss4









���







��zPolicy.load_presetscCs\|j}|
s	t
d
�
�
|
|j��vrt
d|
�
�
t|
|||d�}d|_||j|j<|�|�

dS)z�Add a new on-disk preset and write it to the configured
            presets path.

            :param preset: the new PresetDefaults to add
        zPreset name cannot be emptyz&A preset with name '%s' already exists)rr�r�r�FN)r�r�r�r�rr�r�write)r;rr�r�r�r�r�rrr�
add_preset?s




zPolicy.add_presetcCsX|
r	|
|j�
�v
rtd
|
�
�
|j|
}|jrtd|j�
�
|�|j�

|j�|
�

dS)NzUnknown profile: '%s'z"Cannot delete built-in preset '%s')r�r�r�r�r�deleter��pop)r;rr�rrr�
del_presetRs




�
zPolicy.del_preset)NTN)
r)
F)NN)NFrS)6�__name__�
__module__�__qualname__�__doc__r�r�r�r�r�r�r�rtrgrr�rr�rIr<r(�propertyrG�classmethodr?rKrLrPrmrorsrur{r�r�r�r�r5r�r�r9r�r�r�r�r�r�r�r:r�r�r�rr�r�rrrrr5sf
%















;
		

*

�
2	


	 r)&r�r*rjr�rprTrVr)�pwdr�sos.presetsrrrrrrr	�sos.policies.package_managersr
�
sos.utilitiesrrr
r�sos.report.pluginsrr�sos.optionsrr/rr��textwraprrr2rrrrr�<module>
s*






$






�