OwlCyberSecurity - MANAGER
Edit File: ioprio.cpython-38.pyc
U $�DY � @ s� d dl Z d dlZd dlZd dlZdddddddd d d d d dg Zdddddddddddddg Zd d � ZG d d � d e�Zee�Z ee�Z ze j dd d!�Z W n e k r� e � d�Z Y nX e jZdd"d#d$gZd%Zd&Zd%e> d% Zd'd(� Zd)d*� Zd+d,� Ze jZed-�\ZZZZZ Z e j Z d Z d.d/� Z d0d1� Z!d2d3� Z"d4d5� Z#d6d7� Z$e%d8k�r�d dl&Z&e'e&j(�d9k�rre)e&j(d% �Z*ne�+� Z*e,d:e*� e,d;e!e*�� dS )<� N)�alpha�*i� )�arm*r i; )�i*86r �" )�ia64*r i� )�mips*�32biti� )r �64biti� )�parisc*r i )�powerpc*r i )�s390*r i )�sparc*r �� )�sh*r �! )�x86_64*r r )r r � )r r i� )r r i: )r r r )r r i� )r r i� )r r i� )r r i )r r i )r r i )r r �� )r r i )r r r )r r � c C sL t �� d }t�� d }| D ]*\}}}t�||�r t�||�r | S q d S )N� r )�os�uname�platformZ architecture�fnmatch)Z syscall_listZarch�bitsZcandidate_archZcandidate_bitsZ syscall_nr� r �./usr/lib/python3/dist-packages/iotop/ioprio.py�find_ioprio_syscall_number> s �r c @ s e Zd Zdd� ZdS )�IoprioSetErrorc C s0 zt �|�| _W n tk r* || _Y nX d S �N)r �strerror�err� TypeError)�selfr" r r r �__init__I s zIoprioSetError.__init__N)�__name__� __module__� __qualname__r% r r r r r H s r T�Z use_errnoZrtZbe�idle� � c C s: zt �| �} W n tk r, t �d �} Y nX | t> |B S r )�PRIORITY_CLASSES�index� ValueError�IOPRIO_CLASS_SHIFT)� ioprio_class� ioprio_datar r r � ioprio_value` s r3 c C s t | t? S r )r- r0 ��ioprior r r r1 h s r1 c C s | t @ S r )�IOPRIO_PRIO_MASKr4 r r r r2 l s r2 � c C sJ t | �}tt| �}|d d }|ttfkr2d| S |tkr>dS d| S d S )N� � zrt/%dr* zbe/%d)�sched_getscheduler� getpriority� PRIO_PROCESS� SCHED_FIFO�SCHED_RR� SCHED_IDLE)�pidZ scheduler�niceZ ioprio_nicer r r �get_ioprio_from_schedw s rB c C sT t d kr dS tt t| �}|dk r$dS t|�}|s8t| �S |dkrD|S d|t|�f S )Nz?sysr z?errr* z%s/%d)�__NR_ioprio_get�syscall�IOPRIO_WHO_PROCESSr1 rB r2 )r@ r5 Z prio_classr r r �get� s rF c C sf t d krtd��t||�}tt | ||dd�}|dk rbz t�� }W n tk rX d}Y nX t|��d S )NzNo ioprio_set syscall foundTr) r z<Unknown error (errno support not available before Python2.6))�__NR_ioprio_setr r3 rD �ctypesZ get_errno�AttributeError)ZwhichZwhor1 r2 Z ioprio_valZretr" r r r � set_ioprio� s �rJ c C st | d dkrt | d � S d| krT| �d�r2d}n| �d�r@d}t| �d�d �}n| dkrdd}d}d|t > | S ) Nr �?r+ �/zrt/zbe/r* � )�ord� startswith�int�splitr0 )�key�shiftZprior r r �sort_key� s rT c C s6 d| kr&| � d�}|d t|d �fS | dkr2dS dS )NrL r r+ r* )r* r )NN)rQ rP )Z ioprio_strrQ r r r �to_class_and_data� s rU �__main__rM zpid:zioprio:)-rH r r r ZIOPRIO_GET_ARCH_SYSCALLZIOPRIO_SET_ARCH_SYSCALLr � Exceptionr rC rG ZCDLLZ ctypes_handler# rD r- rE r0 r6 r3 r1 r2 r: �range� SCHED_OTHERr= r> � SCHED_BATCHZ SCHED_ISOr? r; r< rB rF rJ rT rU r& �sys�len�argvrP r@ �getpid�printr r r r �<module> s� �� �