def run(self):
changed = False
if self.dialog.run() == gtk.RESPONSE_OK:
new_policy = self.sched_policy.get_active()
new_prio = int(self.sched_pri.get_value())
new_affinity = self.affinity.get_text()
irq_re = tuna.threaded_irq_re(self.irq)
pids = self.ps.find_by_regex(irq_re)
if pids:
if gui.thread_set_attributes(self.ps[pids[0]], new_policy,
new_prio, new_affinity,
self.irqs.nr_cpus):
changed = True
try:
new_affinity = [int(a) for a in new_affinity.split(",")]
except:
try:
new_affinity = tuna.cpustring_to_list(new_affinity)
except:
new_affinity = procfs.bitmasklist(new_affinity,
self.irqs.nr_cpus)
new_affinity.sort()
curr_affinity = self.irqs[self.irq]["affinity"]
if curr_affinity != new_affinity:
tuna.set_irq_affinity(
self.irq, procfs.hexbitmask(new_affinity,
self.irqs.nr_cpus))
changed = True
self.dialog.destroy()
return changed
def on_affinity_text_changed(self):
new_affinity_text = self.affinity.get_text().strip()
if self.affinity_text != new_affinity_text:
try:
for cpu in new_affinity_text.strip(",").split(","):
new_affinity_cpu_entry = int(cpu, 16)
except:
try:
new_affinity = tuna.cpustring_to_list(new_affinity_text)
except:
if len(new_affinity_text) > 0 and new_affinity_text[-1] != '-' and new_affinity_text[0:2] not in ('0x', '0X'):
# print "not a hex number"
self.affinity.set_text(self.affinity_text)
return
self.affinity_text = new_affinity_text
def on_affinity_text_changed(self):
new_affinity_text = self.affinity.get_text().strip()
if self.affinity_text != new_affinity_text:
try:
for cpu in new_affinity_text.strip(",").split(","):
new_affinity_cpu_entry = int(cpu, 16)
except:
try:
new_affinity = tuna.cpustring_to_list(new_affinity_text)
except:
if len(new_affinity_text) > 0 and new_affinity_text[
-1] != '-' and new_affinity_text[0:2] not in ('0x',
'0X'):
# print "not a hex number"
self.affinity.set_text(self.affinity_text)
return
self.affinity_text = new_affinity_text
def run(self):
changed = False
if self.dialog.run() == gtk.RESPONSE_OK:
new_policy = self.sched_policy.get_active()
new_prio = int(self.sched_pri.get_value())
new_affinity = self.affinity.get_text()
irq_re = tuna.threaded_irq_re(self.irq)
pids = self.ps.find_by_regex(irq_re)
if pids:
if gui.thread_set_attributes(self.ps[pids[0]],
new_policy,
new_prio,
new_affinity,
self.irqs.nr_cpus):
changed = True
try:
new_affinity = [ int(a) for a in new_affinity.split(",") ]
except:
try:
new_affinity = tuna.cpustring_to_list(new_affinity)
except:
new_affinity = procfs.bitmasklist(new_affinity,
self.irqs.nr_cpus)
new_affinity.sort()
curr_affinity = self.irqs[self.irq]["affinity"]
if curr_affinity != new_affinity:
tuna.set_irq_affinity(self.irq,
procfs.hexbitmask(new_affinity,
self.irqs.nr_cpus))
changed = True
self.dialog.destroy()
return changed
thread_list_str = None
filter = False
affect_children = False
show_sockets = False
for o, a in opts:
if o in ("-h", "--help"):
usage()
return
elif o in ("-a", "--config_file_apply"):
apply_config(a)
elif o in ("-l", "--config_file_list"):
list_config()
elif o in ("-c", "--cpus"):
(op, a) = pick_op(a)
op_list = tuna.cpustring_to_list(a)
cpu_list = do_list_op(op, cpu_list, op_list)
elif o in ("-C", "--affect_children"):
affect_children = True
elif o in ("-G", "--cgroup"):
cgroups = True
elif o in ("-t", "--threads"):
(op, a) = pick_op(a)
op_list = reduce(lambda i, j: i + j,
map(thread_mapper, a.split(",")))
op_list = list(set(op_list))
thread_list = do_list_op(op, thread_list, op_list)
# Check if a process name was especified and no
# threads was found, which would result in an empty
# thread list, i.e. we would print all the threads
# in the system when we should print nothing.
def thread_set_attributes(pid_info, new_policy, new_prio, new_affinity,
nr_cpus):
pid = pid_info.pid
changed = False
curr_policy = schedutils.get_scheduler(pid)
curr_prio = int(pid_info["stat"]["rt_priority"])
if new_policy == schedutils.SCHED_OTHER:
new_prio = 0
if curr_policy != new_policy or curr_prio != new_prio:
try:
schedutils.set_scheduler(pid, new_policy, new_prio)
except:
dialog = gtk.MessageDialog(
None, gtk.DIALOG_MODAL | gtk.DIALOG_DESTROY_WITH_PARENT,
gtk.MESSAGE_WARNING, gtk.BUTTONS_OK, _("Invalid parameters!"))
dialog.run()
dialog.destroy()
return False
curr_policy = schedutils.get_scheduler(pid)
if curr_policy != new_policy:
print _("couldn't change pid %(pid)d from %(cpol)s(%(cpri)d) to %(npol)s(%(npri)d)!") % \
{ 'pid': pid, 'cpol': schedutils.schedstr(curr_policy),
'cpri': curr_prio,
'npol': schedutils.schedstr(new_policy),
'npri': new_prio}
else:
changed = True
try:
curr_affinity = schedutils.get_affinity(pid)
except (
SystemError, OSError
) as e: # (3, 'No such process') old python-schedutils incorrectly raised SystemError
if e[0] == 3:
return False
raise e
try:
new_affinity = [int(a) for a in new_affinity.split(",")]
except:
try:
new_affinity = tuna.cpustring_to_list(new_affinity)
except:
new_affinity = procfs.bitmasklist(new_affinity, nr_cpus)
new_affinity.sort()
if curr_affinity != new_affinity:
try:
schedutils.set_affinity(pid, new_affinity)
except:
return invalid_affinity()
try:
curr_affinity = schedutils.get_affinity(pid)
except (
SystemError, OSError
) as e: # (3, 'No such process') old python-schedutils incorrectly raised SystemError
if e[0] == 3:
return False
raise e
if curr_affinity != new_affinity:
print _("couldn't change pid %(pid)d from %(caff)s to %(naff)s!") % \
{ 'pid':pid, 'caff':curr_affinity, 'naff':new_affinity }
else:
changed = True
return changed
thread_list_str = None
filter = False
affect_children = False
show_sockets = False
for o, a in opts:
if o in ("-h", "--help"):
usage()
return
elif o in ("-a", "--config_file_apply"):
apply_config(a)
elif o in ("-l", "--config_file_list"):
list_config()
elif o in ("-c", "--cpus"):
(op, a) = pick_op(a)
op_list = tuna.cpustring_to_list(a)
cpu_list = do_list_op(op, cpu_list, op_list)
elif o in ("-C", "--affect_children"):
affect_children = True
elif o in ("-G", "--cgroup"):
cgroups = True
elif o in ("-t", "--threads"):
(op, a) = pick_op(a)
op_list = reduce(lambda i, j: i + j, map(thread_mapper, a.split(",")))
op_list = list(set(op_list))
thread_list = do_list_op(op, thread_list, op_list)
# Check if a process name was especified and no
# threads was found, which would result in an empty
# thread list, i.e. we would print all the threads
# in the system when we should print nothing.
if not op_list and type(a) == type(""):
def thread_set_attributes(pid_info, new_policy, new_prio, new_affinity, nr_cpus):
pid = pid_info.pid
changed = False
curr_policy = schedutils.get_scheduler(pid)
curr_prio = int(pid_info["stat"]["rt_priority"])
if new_policy == schedutils.SCHED_OTHER:
new_prio = 0
if curr_policy != new_policy or curr_prio != new_prio:
try:
schedutils.set_scheduler(pid, new_policy, new_prio)
except:
dialog = gtk.MessageDialog(None,
gtk.DIALOG_MODAL | gtk.DIALOG_DESTROY_WITH_PARENT,
gtk.MESSAGE_WARNING,
gtk.BUTTONS_OK,
_("Invalid parameters!"))
dialog.run()
dialog.destroy()
return False
curr_policy = schedutils.get_scheduler(pid)
if curr_policy != new_policy:
print _("couldn't change pid %(pid)d from %(cpol)s(%(cpri)d) to %(npol)s(%(npri)d)!") % \
{ 'pid': pid, 'cpol': schedutils.schedstr(curr_policy),
'cpri': curr_prio,
'npol': schedutils.schedstr(new_policy),
'npri': new_prio}
else:
changed = True
try:
curr_affinity = schedutils.get_affinity(pid)
except (SystemError, OSError) as e: # (3, 'No such process') old python-schedutils incorrectly raised SystemError
if e[0] == 3:
return False
raise e
try:
new_affinity = [ int(a) for a in new_affinity.split(",") ]
except:
try:
new_affinity = tuna.cpustring_to_list(new_affinity)
except:
new_affinity = procfs.bitmasklist(new_affinity, nr_cpus)
new_affinity.sort()
if curr_affinity != new_affinity:
try:
schedutils.set_affinity(pid, new_affinity)
except:
return invalid_affinity()
try:
curr_affinity = schedutils.get_affinity(pid)
except (SystemError, OSError) as e: # (3, 'No such process') old python-schedutils incorrectly raised SystemError
if e[0] == 3:
return False
raise e
if curr_affinity != new_affinity:
print _("couldn't change pid %(pid)d from %(caff)s to %(naff)s!") % \
{ 'pid':pid, 'caff':curr_affinity, 'naff':new_affinity }
else:
changed = True
return changed
