Your IP : 3.145.69.185
3
�<�e3 � @ s� d dl Z d dlZd dlZd dlZd dlZd dlmZmZ yd dlZ dZ
W n dZ
Y nX d dl j Z d dl
Zd dlmZ ejj� ZG dd� d�ZdS )� N)� ConfigParser�ErrorTF)�commandsc @ s0 e Zd Zd dd�Zdd� Zd
dd�Zd d
� Zd S )�ProfileRecommenderFc C s || _ t� | _d | _d S )N)�
_is_hardcodedr � _commands�
_chassis_type)�selfZ is_hardcoded� r
�)/usr/lib/python3.6/profile_recommender.py�__init__ s zProfileRecommender.__init__c
C s t j}| jr|S tj� dk}|s*tjd� | jt j|d�}|d k rF|S i }x�t j D ]|}g }ytj
|�}W n@ t k
r� } z$|j t j
kr�tjd||f � W Y d d }~X nX x"|D ]}tjj||�} | ||< q�W qRW x6t|j� �D ]&}|| } | j| |d�}|d k r�|S q�W |S )Nr ztProfile recommender is running without root privileges. Profiles with virt recommendation condition will be omitted.)�has_rootzerror accessing %s: %s)�constsZDEFAULT_PROFILEr �os�geteuid�logZwarning�process_configZRECOMMEND_CONF_FILEZRECOMMEND_DIRECTORIES�listdir�OSError�errno�ENOENT�error�path�join�sorted�keys)
r Zprofiler
Zmatching�filesZ directory�contents�e�namer r
r
r � recommend s6
$
z ProfileRecommender.recommendTc ! C s� d }d}�y�t jj|�s d S tdddd�}t|_t|��}|j||� W d Q R X �x:|j� D �],}d}�x|j |�D �]�} |j
|| dd�}
|
dkr�d}
| d kr�|s�d}P t j |
| j
jd
g�d t j�s�d}qt| d k� r�t j |
| j
jtjdd
�t j��sjd}qt| d dk�r<t jj| � �s6t j |
| j
j| �t j� �rjd}qt| dd� dk�r~tj� } | j� t| jt j|
���dk�rjd}qt| dk�r�| j� } t j |
| t j��sjd}qt| dkrtd}
t�r:t j j t j j dd�}y|j� |j d }
W nR t!t"t#fk
�r6 } z.t$|d��r&|j%t%j&k�r&t'j(d| � W Y d d }~X nX n|�sRt'j(d| � d}t j |
|
t j�d krtd}qtW |r\t jd�}|j)d|�}P q\W W n< t!t"t*fk
�r� } zt'j(d||f � W Y d d }~X nX |S )NF�=�#)Z
delimitersZinline_comment_prefixes�strictT)�raw� z^$Zvirtz virt-what� �system)Zno_errorr �/� Zprocess� chassis_typeZsyspurpose_role)Zraise_on_error�roler z/Failed to load the syspurpose file: %sz\Failed to process 'syspurpose_role' in '%s' , the syspurpose module is not availablez,[^,]*$zerror processing '%s', %s)+r r �isfiler �strZ optionxform�openZ read_fileZsectionsZoptions�get�re�matchr Zexecute�Sr ZSYSTEM_RELEASE_FILE�exists�procfsZpidstatsZreload_threads�lenZ
find_by_regex�compile�_get_chassis_type�
IGNORECASE�have_syspurpose�
syspurposer ZSyspurposeStoreZUSER_SYSPURPOSEr �IOErrorr �KeyError�hasattrr r r r �subr )r Zfnamer
Zmatching_profileZsyspurpose_error_logged�config�fZsectionr1 Zoption�valueZpsr* r+ �sr �rr
r
r r 9 s�
$z!ProfileRecommender.process_configc % C s� | j d k r tjd| j � | j S dddddddd d
d d d
ddddddddddddddd d d d d d!d"d#d$d%d&g%}y0td'd(��}t|j� �}W d Q R X || | _ W nT tk
r� tjd)| � Y n4 tt fk
r� } ztj
d*| � W Y d d }~X nX | j �rtjd+| j � | j S yXt j d,d-d.gt j
t j
d/d0�}|j� \}}|j�rZtjd1|j|f � n|j� j� | _ W n6 tt fk
�r� } ztj
d2| � W Y d d }~X nX | j �s�tjd3� d| _ ntjd+| j � | j S )4Nz"returning cached chassis type '%s'r% ZOtherZUnknownZDesktopzLow Profile Desktopz Pizza Boxz
Mini TowerZTowerZPortableZLaptopZNotebookz Hand HeldzDocking Stationz
All In Onez Sub Notebookz Space-savingz Lunch BoxzMain Server ChassiszExpansion Chassisz Sub ChassiszBus Expansion ChassiszPeripheral Chassisz RAID ChassiszRack Mount ChassiszSealed-case PCz Multi-systemZ
CompactPCIZ AdvancedTCAZBladezBlade EnclosingZTabletZ ConvertibleZ
Detachablez IoT Gatewayz Embedded PCzMini PCzStick PCz(/sys/devices/virtual/dmi/id/chassis_typerC z/Unknown chassis type id read from dmi sysfs: %dz"error accessing dmi sysfs file: %szchassis type - %sZ dmidecodez-sz chassis-typeT)�stdout�stderrZ close_fdsz,dmidecode finished with error (ret %d): '%s'z#error executing dmidecode tool : %sz!could not determine chassis type.)r r �debugr. �int�read�
IndexErrorr r r; �warn�
subprocess�Popen�PIPEZ communicate�
returncode�strip�decode)r ZDMI_CHASSIS_TYPESZsysfs_chassis_typeZchassis_type_idr Zp_dmiZ
dmi_outputZ dmi_errorr
r
r r7 � sF
z$ProfileRecommender._get_chassis_typeN)F)T)�__name__�
__module__� __qualname__r r r r7 r
r
r
r r s
Ir )r r0 r r4 rK Ztuned.utils.config_parserr r Zsyspurpose.filesr: r9 Z tuned.constsr Z
tuned.logsZtunedZtuned.utils.commandsr Zlogsr/ r r r
r
r
r �<module> s