1 # This program is free software; you can redistribute it and/or modify
2 # it under the terms of the GNU General Public License as published by
3 # the Free Software Foundation; either version 2 of the License, or
4 # (at your option) any later version.
6 # This program is distributed in the hope that it will be useful,
7 # but WITHOUT ANY WARRANTY; without even the implied warranty of
8 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
9 # GNU Library General Public License for more details.
11 # You should have received a copy of the GNU General Public License
12 # along with this program; if not, write to the Free Software
13 # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
15 # See the COPYING file for license information.
17 # Copyright (c) 2007 Guillaume Chazarain <guichaz@gmail.com>
25 # From http://git.kernel.org/?p=utils/util-linux-ng/util-linux-ng.git;a=blob;
26 # f=configure.ac;h=770eb45ae85d32757fc3cff1d70a7808a627f9f7;hb=HEAD#l354
27 # i386 bit userspace under an x86_64 kernel will have its uname() appear as
28 # 'x86_64' but it will use the i386 syscall number, that's why we consider both
29 # the architecture name and the word size.
30 IOPRIO_GET_ARCH_SYSCALL
= [
34 ('powerpc*', '*', 274),
38 ('x86_64*', '32bit', 290),
39 ('x86_64*', '64bit', 252),
42 IOPRIO_SET_ARCH_SYSCALL
= [
46 ('powerpc*', '*', 273),
50 ('x86_64*', '32bit', 289),
51 ('x86_64*', '64bit', 251),
54 def find_ioprio_syscall_number(syscall_list
):
56 bits
= platform
.architecture()[0]
58 for candidate_arch
, candidate_bits
, syscall_nr
in syscall_list
:
59 if fnmatch
.fnmatch(arch
, candidate_arch
) and \
60 fnmatch
.fnmatch(bits
, candidate_bits
):
63 class IoprioSetError(Exception):
64 def __init__(self
, err
):
66 self
.err
= os
.strerror(err
)
70 __NR_ioprio_get
= find_ioprio_syscall_number(IOPRIO_GET_ARCH_SYSCALL
)
71 __NR_ioprio_set
= find_ioprio_syscall_number(IOPRIO_SET_ARCH_SYSCALL
)
74 ctypes_handle
= ctypes
.CDLL(None, use_errno
=True)
76 ctypes_handle
= ctypes
.CDLL(None)
78 syscall
= ctypes_handle
.syscall
80 PRIORITY_CLASSES
= [None, 'rt', 'be', 'idle']
82 IOPRIO_WHO_PROCESS
= 1
83 IOPRIO_CLASS_SHIFT
= 13
84 IOPRIO_PRIO_MASK
= (1 << IOPRIO_CLASS_SHIFT
) - 1
86 def ioprio_value(ioprio_class
, ioprio_data
):
88 ioprio_class
= PRIORITY_CLASSES
.index(ioprio_class
)
90 ioprio_class
= PRIORITY_CLASSES
.index(None)
91 return (ioprio_class
<< IOPRIO_CLASS_SHIFT
) | ioprio_data
93 def ioprio_class(ioprio
):
94 return PRIORITY_CLASSES
[ioprio
>> IOPRIO_CLASS_SHIFT
]
96 def ioprio_data(ioprio
):
97 return ioprio
& IOPRIO_PRIO_MASK
99 sched_getscheduler
= ctypes_handle
.sched_getscheduler
100 SCHED_OTHER
, SCHED_FIFO
, SCHED_RR
, SCHED_BATCH
, SCHED_ISO
, SCHED_IDLE
= range(6)
102 getpriority
= ctypes_handle
.getpriority
105 def get_ioprio_from_sched(pid
):
106 scheduler
= sched_getscheduler(pid
)
107 nice
= getpriority(PRIO_PROCESS
, pid
)
108 ioprio_nice
= (nice
+ 20) / 5
110 if scheduler
in (SCHED_FIFO
, SCHED_RR
):
111 return 'rt/%d' % ioprio_nice
112 elif scheduler
== SCHED_IDLE
:
115 return 'be/%d' % ioprio_nice
118 if __NR_ioprio_get
is None:
121 ioprio
= syscall(__NR_ioprio_get
, IOPRIO_WHO_PROCESS
, pid
)
125 prio_class
= ioprio_class(ioprio
)
127 return get_ioprio_from_sched(pid
)
128 if prio_class
== 'idle':
130 return '%s/%d' % (prio_class
, ioprio_data(ioprio
))
132 def set_ioprio(which
, who
, ioprio_class
, ioprio_data
):
133 if __NR_ioprio_set
is None:
134 raise IoprioSetError('No ioprio_set syscall found')
136 ioprio_val
= ioprio_value(ioprio_class
, ioprio_data
)
137 ret
= syscall(__NR_ioprio_set
, which
, who
, ioprio_val
, use_errno
=True)
140 err
= ctypes
.get_errno()
141 except AttributeError:
142 err
= 'Unknown error (errno support not available before Python2.6)'
143 raise IoprioSetError(err
)
150 if key
.startswith('rt/'):
152 elif key
.startswith('be/'):
154 prio
= int(key
.split('/')[1])
159 return (1 << (shift
* IOPRIO_CLASS_SHIFT
)) + prio
161 def to_class_and_data(ioprio_str
):
162 if '/' in ioprio_str
:
163 split
= ioprio_str
.split('/')
164 return (split
[0], int(split
[1]))
165 elif ioprio_str
== 'idle':
169 if __name__
== '__main__':
171 if len(sys
.argv
) == 2:
172 pid
= int(sys
.argv
[1])
176 print 'ioprio:', get(pid
)