a
]i`C @ st d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZ d dl Z d dl
T d dl m Z ej
ZG dd dZdS ) N)*)TunedExceptionc @ s e Zd ZdfddZdd Zdd Zdd Zd
d Zd d
ZdgddZ dhddZ
dd Z dd Z dd Z
didd Zdjd d Zdkd d!Zdld#d$Zdmd%d&Zdnd'd(Zdod)d*Zdpd+d,Zd-d. Zd/d0 Zdqd1d2Zd3d4 Zd5d6 Zdrd7d8Zd di g d fd9d:Z dsd;d<Z d=d> Z d?d@ Z dAdB Z dtdDdEZ!dFdG Z"dHdI Z#dJdK Z$dLdM Z%dudNdOZ&dPdQ Z'dRdS Z(dTdU Z)dVdW Z*dXdY Z+dZd[ Z,d\d] Z-d^d_ Z.d`da Z/dbdc Z0ddde Z1dS )vcommandsTc 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 logerrorr 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 )N10)YZYESTZTRUENZNOFZFALSE)strupperstripget)r valuevr r r get_bool s zcommands.get_boolc C s t ddt| S )Nz\s+ )resubr 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%sreplace)r r" Z what_escapeZ escape_byr r r escape& s zcommands.escapec C s | |dS )N r& )r r" Z escape_charr r r unescape* s zcommands.unescapec C s |d|t | | S )Nr )len)r s1poss2r r r align_str. s zcommands.align_strc C s. g }|d ur*t | D ]}|t|7 }q|S r )sorteditemslist)r dlir r r dict2list4 s
zcommands.dict2listc C s( |d u r d S t ddt| S )Nz(%s)z)|()r compilejoinr2 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)mor3 r r M z.commands.multiple_re_replace..)r+ r: r! )r r3 r" rflagsr 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|tr|rt|tr| 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 changewz Writing to file '%s' error: '%s'N)r ospathdirnameisdirmakedirs read_filer r openwritecloseOSErrorIOError
isinstanceboolr2 errnor )
r fdataZmakedirrH Z ignore_samer3 fdrcer 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(|sTtd|t |f W Y d }~dS d }~0 0 dS )NzRemoving tree: '%s'z cannot remove tree '%s': '%s'FT)
r rJ rK existsshutilrmtreerS r r r r rX rH r r r r rc s zcommands.rmtreec
C sn | d| tj|rjzt| W n@ tyh } z(|sRtd|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*|sNtd||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 srcdstrH 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|||tjdS Nr FrC )rO r+ r] r r! MULTILINE)r rX patternreplrY r r r replace_in_file s
zcommands.replace_in_filec C s4 | |}t|dkrdS | || j||tjdS 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 |tjdd u rz|rt|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 optor r r r add_modify_option_in_file s
,z"commands.add_modify_option_in_filec C s" | |}tt|d S Nzutf-8)rO hashlibZmd5r encode hexdigestr rX rY r r r md5sum s
zcommands.md5sumc C s" | |}tt|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 }
zt|tt|||ddd} | \} } | j }|r||vrd|vr| d d }
t
|
dkr| d d }
d d
||
f }
|s| |
W nz t
tfy: } z\|jd ur|j nd}t||vr&d|vr&d d
||f }
|s&| |
W Y d }~n
d }~0 0 |rL|| |
fS || fS d S ) Nr CLC_ALLz
Executing %s.r) T)stdoutstderrenvshellcwdZ close_fdsZuniversal_newlinesrq zExecuting '%s' error: %sr )rJ environri updater r PopenPIPEZ communicate
returncoder+ r8 r rS rT rW abs)r argsr r r Z no_errorsZ
return_errZretcodeZ _environmentouterr_msgprocerrZerr_outr\ r r r execute s@
zcommands.executec C s. t d|}|r|dS |r*| d S |S )Nz.*\[([^\]]+)\].*r; r )r matchgroupsplit)r optionsZdosplitmr 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 tdt | 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| }
|rt|
dkrd}|| d}n||
7 }qX|
dd dkrd}|
}qX|
rT|
d dks|
d d krT|
d
d d } zPt| d
kr |ttt | d t | d
d
7 }n|t |
d
d W n t
yP g Y S 0 qXt|
dkrX||
qXt|dkr|| |D ]} | d } | dd dkr|| | 7 }njzLt| d
kr|ttt | d t | d
d
7 }n|t | d W n t
y$ g Y S 0 qt tt
|}
|D ]}||
v r>|
| q>|
S ) Nr Fr) r 0xT^!r; -)typer2 r r r r+ r lowerranger r r r0 setremove)r r4 Z strip_charsrlZllZll2Z
negation_listZhexmaskZhvr svnlZvlZcpu_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 fddtdt dD S )Nz%x r r c 3 s | ]} ||d V qdS )r Nr ).0r5 r" r r rA z'commands.cpulist2hex..)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..)r8 )r r4 r r r r cpulist2string s zcommands.cpulist2stringc C sb |dd dkrH|dd }|dd dkrHdd | |D S td|}dd |D S )Nr r zcpulist:c S s g | ]}d t | qS )r r r r r r
rA z(commands.devstr2devs..z\s*(?td| W Y d }~n>d }~0 t t
fy } ztd| W Y d }~n
d }~0 0 |dkrd }n
|tj
k}|dkrd }||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 EOFErrorPROFILE_MODE_FILEACTIVE_PROFILE_AUTOACTIVE_PROFILE_MANUAL)r profile_namemoderX 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.}|rtj ntj
}||d W d n1 s0 Y W n8 ttfy } 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.)allr+ r ZNAMES_MAX_LENGTH)r namer r r
is_valid_name) s zcommands.is_valid_namec C s t d|g S )Ngetconf)Z check_outputdecoder )r variabler r r r , s zcommands.getconfc C s | tj}|r| |S dgS r )rO r ZSYSFS_CPUS_PRESENT_PATHr )r r r r r get_cpus0 s zcommands.get_cpus)T)r% r% )r% )Nr )N)FFF)r) F)F)F)F)F)T)T)T)r )r) )2__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 r s^
#
'
D
r )rW ry Z
tuned.logsZtunedri rJ rb Z tuned.constsr r
subprocessZtuned.exceptionsr Zlogsr r r r r r r s