관리-도구
편집 파일: commands.cpython-39.pyc
a ,�g�B � @ st d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dl Z d dl T d dl m Z ej �� ZG dd� d�ZdS )� N)�*)�TunedExceptionc @ s� e Zd Zdddd�Zdd� Zdd� Zdd � Zd d � Zd d � Zdedd�Z dfdd�Z dd� Z dd� Z dd� Z dgdd �Zdhd d �Zdid d!�Zdjd#d$�Zdkd%d&�Zdld'd(�Zdmd)d*�Zdnd+d,�Zd-d.� Zd/d0� Zdod1d2�Zd3d4� Zd5d6� Zdpd7d8�Zd di g d fd9d:�Z dqd;d<�Z d=d>� Z d?d@� Z dAdB� Z drdDdE�Z!dFdG� Z"dHdI� Z#dJdK� Z$dLdM� Z%dsdNdO�Z&dPdQ� Z'dRdS� Z(dTdU� Z)dVdW� Z*dXdY� Z+dZd[� Z,d\d]� Z-d^d_� Z.d`da� Z/dbdc� Z0dS )t�commandsTc C s || _ d S �N)�_logging)�selfZlogging� r �8/usr/lib/python3.9/site-packages/tuned/utils/commands.py�__init__ s zcommands.__init__c C s | j rt�|� d S r )r �log�error�r �msgr r r �_error s zcommands._errorc C s | j rt�|� d S r )r r �debugr r r r �_debug s zcommands._debugc C s. t |��� �� }ddddddddd��||�S )N�1�0)�YZYES�TZTRUE�NZNO�FZFALSE)�str�upper�strip�get)r �value�vr r r �get_bool s zcommands.get_boolc C s t �ddt|���� S )Nz\s+� )�re�subr r )r �sr r r � remove_ws s zcommands.remove_wsc C s t �dd|�S )Nz^"(.*)"$z\1)r r! )r r r r r �unquote" s zcommands.unquote�\c C s |� |d||f �S )Nz%s%s��replace)r r" Z what_escapeZ escape_byr r r �escape& s zcommands.escapec C s |� |d�S )N� r& )r r" Z escape_charr r r �unescape* s zcommands.unescapec C s |d|t |� | S )Nr )�len)r �s1�pos�s2r r r � align_str. s zcommands.align_strc C s. g }|d ur*t |�� �D ]}|t|�7 }q|S r )�sorted�items�list)r �d�l�ir r r � dict2list4 s zcommands.dict2listc C s( |d u r d S t �dd�t|�� �� �S )Nz(%s)z)|()r �compile�joinr2 �keys)r r3 r r r �re_lookup_compile< s zcommands.re_lookup_compileNr c sV � d u r|d u r.|S nt � �dks*|d u r.|S |d u r@| �� �}|�� fdd�||�S )Nr c s t � �� �| jd S )N� )r2 �values� lastindex)�mo�r3 r r �<lambda>M � z.commands.multiple_re_replace.<locals>.<lambda>)r+ r: r! )r r3 r" �r�flagsr r? r �multiple_re_replaceD s z commands.multiple_re_replacec C sR t |�dks|d u rd S |d u r*| �|�}|�|�}|rNt|�� �|jd S d S �Nr r; )r+ r: �searchr2 r< r= )r r3 r" rB r> r r r � re_lookupS s zcommands.re_lookupFc C s | � d||f � |r2tj�|�}tj�|�r2d}zh|rBt�|� |rt| j|dd��� t|�krt| � d| � W dS t |d�}|� t|�� |� � d}W n` t t fy� } zDd}t|t�r�|r�t|t�r�| j|vr�| �d|| f � W Y d} ~ n d} ~ 0 0 |S ) ah Write data to a file. Parameters: f -- filename where to write data -- data to write makedir -- if True and the path doesn't exist, it will be created no_error -- if True errors are silenced, it can be also list of ignored errnos ignore_same -- if True and the write would not change the file, it is skipped Return: bool -- True on success z Writing to file: '%s' < '%s'FT)�no_errorz=Skipping the write to file '%s', the content would not change�wz Writing to file '%s' error: '%s'N)r �os�path�dirname�isdir�makedirs� read_filer r �open�write�close�OSError�IOError� isinstance�boolr2 �errnor ) r �f�dataZmakedirrH Z ignore_samer3 �fd�rc�er r r � write_to_file] s0 ��(zcommands.write_to_filer) c C sx |}z t |d�}|�� }|�� W n> ttfy` } z"|sL| �d||f � W Y d }~n d }~0 0 | �d||f � |S )NrB z"Error when reading file '%s': '%s'z Read data from file: '%s' > '%s')rP �readrR rS rT r r )r rX �err_retrH � old_valuerZ r\ r r r rO � s (zcommands.read_filec C sp | � d| � tj�|�rlzt�||� W n@ tyj } z(|sTt�d|t |�f � W Y d }~dS d }~0 0 dS )NzRemoving tree: '%s'z cannot remove tree '%s': '%s'FT) r rJ rK �exists�shutil�rmtreerS r r r �r rX rH r r r r rc � s zcommands.rmtreec C sn | � d| � tj�|�rjzt�|� W n@ tyh } z(|sRt�d|t|�f � W Y d }~dS d }~0 0 dS )NzRemoving file: '%s'z cannot remove file '%s': '%s'FT) r rJ rK ra �unlinkrS r r r rd r r r re � s zcommands.unlinkc C sj | � d||f � zt�||� W nB tyd } z*|sNt�d||t|�f � W Y d }~dS d }~0 0 dS )NzRenaming file '%s' to '%s'z%cannot rename file '%s' to '%s': '%s'FT)r rJ �renamerS r r r )r �src�dstrH r r r r rf � s zcommands.renamec C sh z$t �d||f � t�||� W dS tyb } z&|sLt �d|||f � W Y d }~dS d }~0 0 d S )Nzcopying file '%s' to '%s'Tz!cannot copy file '%s' to '%s': %sF)r r rb �copyrT r )r rg rh rH r\ r r r ri � s z commands.copyc C s6 | � |�}t|�dkrdS | �|tj|||tjd��S �Nr F�rC )rO r+ r] r r! � MULTILINE)r rX �pattern�replrY r r r �replace_in_file� s zcommands.replace_in_filec C s4 | � |�}t|�dkrdS | �|| j||tjd��S rj )rO r+ r] rD r rl )r rX r3 rY r r r �multiple_replace_in_file� s z!commands.multiple_replace_in_filec C s� | � |�}|D ]�}t|�}t|| �}tjd| d |tjd�d u rz|r�t|�dkrh|d dkrh|d7 }|d||f 7 }qtjd| d d | �|� d |tjd�}q| �||�S ) Nz\bz\s*=.*$rk r ���� z%s="%s" z\b(z\s*=).*$z\1"�") rO r r rF rl r+ r! r( r] )r rX r3 �addrY �opt�or r r r �add_modify_option_in_file� s ,z"commands.add_modify_option_in_filec C s" | � |�}t�t|��d���� S �Nzutf-8)rO �hashlibZmd5r �encode� hexdigest�r rX rY r r r �md5sum� s zcommands.md5sumc C s" | � |�}t�t|��d���� S rx )rO ry Zsha256r rz r{ r| r r r � sha256sum� s zcommands.sha256sumc C s | � tj|��� S r )rO �constsZMACHINE_ID_FILEr )r rH r r r �get_machine_id� s zcommands.get_machine_idc C sX d}t j�� }d|d< |�|� | �dt|� � d} d } z�t|tt|||ddd�} | �� \} } | j }|r�||vr�d|vr�| d d� } t | �dkr�| d d� } d d � |�| f } |s�| � | � W nz t tf�y: } z\|jd ur�|j nd}t|�|v�r&d|v�r&d d � |�|f } |�s&| � | � W Y d }~n d }~0 0 |�rL|| | fS || fS d S ) Nr �C�LC_ALLz Executing %s.r) T)�stdout�stderr�env�shell�cwdZ close_fdsZuniversal_newlinesrq zExecuting '%s' error: %sr )rJ �environri �updater r �Popen�PIPEZ communicate� returncoder+ r8 r rS rT rW �abs)r �argsr� r� r� Z no_errorsZ return_errZretcodeZ _environment�out�err_msg�proc�errZerr_outr\ r r r �execute� s@ � zcommands.executec C s. t �d|�}|r|�d�S |r*|�� d S |S )Nz.*\[([^\]]+)\].*r; r )r �match�group�split)r �optionsZdosplit�mr r r �get_active_option s zcommands.get_active_optionc C s, t |�}|dkp*| jd| ddd��� dkS )Nr z$/sys/devices/system/cpu/cpu%s/onliner T)r_ rH )r rO r )r �cpuZscpur r r � is_cpu_online s zcommands.is_cpu_onlinec C s^ |d u r d S t |��dd�}zt|d�}W n( tyR t�dt |� � g Y S 0 | �|�S )N�,r) � z invalid hexadecimal mask '%s')r r' �int� ValueErrorr r �bitmask2cpulist)r �maskr� r r r � hex2cpulist s zcommands.hex2cpulistc C s8 d}g }|dkr4|d@ r"|� |� |dL }|d7 }q|S rE )�append)r r� r� �cpusr r r r� * s zcommands.bitmask2cpulist�'"c C s^ g }|d u r|S |}t |�turD|d ur6t|��|�}t|��d�}g }g }d}d}|D �]} t| �} |r�t| �dkr�d}|�|� d}n|| 7 }qX| dd� �� dkr�d}| }qX| �rT| d dks�| d d k�rT| d d � �d �} zPt| �d k�r |ttt | d �t | d �d ��7 }n|�t | d d � �� W n t �yP g Y S 0 qXt| �dkrX|�| � qXt|�dk�r�|�|� |D ]�} | �d �} | dd� �� dk�r�|| � | �7 }njzLt| �d k�r�|ttt | d �t | d �d ��7 }n|�t | d �� W n t �y$ g Y S 0 �q�t tt |���} |D ]}|| v �r>| �|� �q>| S ) Nr� Fr) r � �0xT�^�!r; �-)�typer2 r r r� r+ r� �lower�ranger� r� r� r0 �set�remove)r r4 Z strip_chars�rlZllZll2Z negation_listZhexmaskZhvr �sv�nlZvlZcpu_listr� r r r �cpulist_unpack= sj � ( zcommands.cpulist_unpackc C s� | � |�}|d u s t|�dkr"|S d}|}g }|d t|�k r�||d || dkr�||kr�|�t|| �d t|| � � n|�t|| �� |d }|d7 }q.|d t|�k r�|�t|| �d t|d � � n|�t|d �� |S )Nr r; r� rq )r� r+ r� r )r r4 r5 �jr� r r r � cpulist_pack� s" $ $zcommands.cpulist_packc C s. | � |�}| � | �d��}tt|�t|� �S )Nz /sys/devices/system/cpu/online)r� rO r2 r� )r r4 r� Zonliner r r �cpulist_invert� s zcommands.cpulist_invertc s� |d u r d S | � |�}|d u r"d S | �|�}d| � t� �}|d dkrX|d|d 7 }� �|�� d�� fdd�tdt� �d�D ��S )Nz%x� r r� c 3 s | ]}� ||d � V qdS )r� Nr )�.0r5 �r" r r � <genexpr>� rA z'commands.cpulist2hex.<locals>.<genexpr>)r� �cpulist2bitmaskr+ �zfillr8 r� )r r4 Zulr� Zlsr r� r � cpulist2hex� s zcommands.cpulist2hexc C s d}|D ]}|t d|�O }q|S )Nr r� )�pow)r r4 r� r r r r r� � s zcommands.cpulist2bitmaskc s d� � fdd�|D ��S )Nr� c 3 s | ]}� t |� V qd S r �r �r� r ��prefixr r r� � rA z*commands.cpulist2string.<locals>.<genexpr>)r8 )r r4 r� r r� r �cpulist2string� s zcommands.cpulist2stringc C sb |dd� � � dkrH|dd � }|dd� � � dkrHdd� | �|�D �S t�d|�}dd� |D �S )Nr r� zcpulist:c S s g | ]}d t |� �qS )r� r� r� r r r � <listcomp>� rA z(commands.devstr2devs.<locals>.<listcomp>z\s*(?<!\\),\s*c S s g | ]}t |��d d��qS )z\,r� )r r' r� r r r r� � rA )r� r� r r� )r r" r4 r r r � devstr2devs� s zcommands.devstr2devsc C sB z|� |d�W S ty< } z|� |�W Y d }~S d }~0 0 d S )NF)�wait� TypeError)r Z terminate�timer\ r r r r� � s z commands.waitc C s� t |��� �� }dD ]�}|�|�}|dkr,qzht|d |� �}||d � }|dkr\|d9 }n0|dkrn|d9 }n |dkr�|d9 }n |d kr�d }|W S ty� Y d S 0 qd S ) N)�KB�MB�GBr) rq r� i r� i r� i @r) )r r r �rfindr� r� )r r" ZunitZunit_ix�val�ur r r �get_size� s& zcommands.get_sizec C s� d}d}z<t tjd�� }|�� �� }W d � n1 s80 Y W nn ty� } z$|jtjkrltd| ��W Y d }~n<d }~0 t t fy� } ztd| ��W Y d }~n d }~0 0 z^t tj d��<}|�� �� }|dtj tj fvr�tdtj ��W d � n1 �s0 Y W nt t�yR } z&|jtjk�r>td| ��W Y d }~n>d }~0 t t f�y� } ztd| ��W Y d }~n d }~0 0 |dk�r�d }n |tj k}|dk�r�d }||fS )Nr) rB z!Failed to read active profile: %szInvalid value in file %s.z Failed to read profile mode: %s)rP r �ACTIVE_PROFILE_FILEr^ r rT rW �ENOENTr rS �EOFError�PROFILE_MODE_FILE�ACTIVE_PROFILE_AUTO�ACTIVE_PROFILE_MANUAL)r � profile_name�moderX r\ �manualr r r �get_active_profile� s4 . " 2 " zcommands.get_active_profilec C s zFt tjd��&}|d ur&|�|d � W d � n1 s:0 Y W n6 ttfy| } ztd|j ��W Y d }~n d }~0 0 zNt tjd��.}|r�tj ntj }|�|d � W d � n1 s�0 Y W n8 ttf�y } ztd|j ��W Y d }~n d }~0 0 d S )NrI rr z!Failed to save active profile: %sz Failed to save profile mode: %s) rP r r� rQ rS rT r �strerrorr� r� r� )r r� r� rX r\ r� r r r �save_active_profile� s 0$0z commands.save_active_profilec C s� d}z<t tjd�� }|�� �� }W d � n1 s40 Y W nn ty| } z$|jtjkrhtd| ��W Y d }~n<d }~0 t t fy� } ztd| ��W Y d }~n d }~0 0 |dkr�d }|S )Nr) rB z1Failed to read the active post-loaded profile: %s) rP r �POST_LOADED_PROFILE_FILEr^ r rT rW r� r rS r� �r r� rX r\ r r r �get_post_loaded_profile s . "z commands.get_post_loaded_profilec C s� zFt tjd��&}|d ur&|�|d � W d � n1 s:0 Y W n6 ttfy| } ztd|j ��W Y d }~n d }~0 0 d S )NrI rr z1Failed to save the active post-loaded profile: %s)rP r r� rQ rS rT r r� r� r r r �save_post_loaded_profile s 0z!commands.save_post_loaded_profilec C sB zt �||�}W n& ty6 dd l}|�||�}Y n0 |�|�S )Nr )r � maketrans�AttributeError�string� translate)r �textZ source_charsZ dest_charsZtransr� r r r �tr s z commands.trc C s& | p$t dd� |D ��o$t|�tjkS )Nc s s | ]}|t jv V qd S r )r ZNAMES_ALLOWED_CHARS)r� �cr r r r� * rA z)commands.is_valid_name.<locals>.<genexpr>)�allr+ r ZNAMES_MAX_LENGTH)r �namer r r � is_valid_name) s zcommands.is_valid_namec C s t d|g��� �� S )N�getconf)Z check_output�decoder )r �variabler r r r� , s zcommands.getconf)T)r% r% )r% )Nr )N)FFF)r) F)F)F)F)F)T)T)T)r� )r) )1�__name__� __module__� __qualname__r r r r r# r$ r( r* r/ r6 r: rD rG r] rO rc re rf ri ro rp rw r} r~ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r s\ # ' D r )rW ry Z tuned.logsZtunedri rJ rb Z tuned.constsr r � subprocessZtuned.exceptionsr Zlogsr r r r r r r �<module> s