1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
|
#!/usr/bin/python3
# Helper script for realtime profiles provided by RT
import os
import sys
irqpath = "/proc/irq/"
def bitmasklist(line):
fields = line.strip().split(",")
bitmasklist = []
entry = 0
for i in range(len(fields) - 1, -1, -1):
mask = int(fields[i], 16)
while mask != 0:
if mask & 1:
bitmasklist.append(entry)
mask >>= 1
entry += 1
return bitmasklist
def get_cpumask(mask):
groups = []
comma = 0
while mask:
cpumaskstr = ''
m = mask & 0xffffffff
cpumaskstr += ('%x' % m)
if comma:
cpumaskstr += ','
comma = 1
mask >>= 32
groups.append(cpumaskstr)
string = ''
for i in reversed(groups):
string += i
return string
def parse_def_affinity(fname):
if os.getuid() != 0:
return
try:
with open(fname, 'r') as f:
line = f.readline()
return bitmasklist(line)
except IOError:
return [ 0 ]
def verify(shouldbemask):
inplacemask = 0
fname = irqpath + "default_smp_affinity";
cpulist = parse_def_affinity(fname)
for i in cpulist:
inplacemask = inplacemask | 1 << i;
if (inplacemask & ~shouldbemask):
sys.stderr.write("verify: failed: irqaffinity (%s) inplacemask=%x shouldbemask=%x\n" % (fname, inplacemask, shouldbemask))
sys.exit(1)
# now verify each /proc/irq/$num/smp_affinity
interruptdirs = [ f for f in os.listdir(irqpath) if os.path.isdir(os.path.join(irqpath,f)) ]
# IRQ 2 - cascaded signals from IRQs 8-15 (any devices configured to use IRQ 2 will actually be using IRQ 9)
try:
interruptdirs.remove("2")
except ValueError:
pass
# IRQ 0 - system timer (cannot be changed)
try:
interruptdirs.remove("0")
except ValueError:
pass
for i in interruptdirs:
inplacemask = 0
fname = irqpath + i + "/smp_affinity"
cpulist = parse_def_affinity(fname)
for i in cpulist:
inplacemask = inplacemask | 1 << i;
if (inplacemask & ~shouldbemask):
sys.stderr.write("verify: failed: irqaffinity (%s) inplacemask=%x shouldbemask=%x\n" % (fname, inplacemask, shouldbemask))
sys.exit(1)
sys.exit(0)
sys.stderr.write("defirqaffinity.py is deprecated. Use isolated_cores or other built-in functionality instead.\n")
# adjust default_smp_affinity
cpulist = parse_def_affinity(irqpath + "default_smp_affinity")
mask = 0
for i in cpulist:
mask = mask | 1 << i;
if len(sys.argv) < 3 or len(str(sys.argv[2])) == 0:
sys.stderr.write("%s: invalid arguments\n" % os.path.basename(sys.argv[0]))
sys.exit(1)
line = sys.argv[2]
fields = line.strip().split(",")
for i in fields:
if sys.argv[1] == "add":
mask = mask | 1 << int(i);
elif sys.argv[1] == "remove" or sys.argv[1] == "verify":
mask = mask & ~(1 << int(i));
if sys.argv[1] == "verify":
verify(mask)
string = get_cpumask(mask)
fo = open(irqpath + "default_smp_affinity", "w")
fo.write(string)
fo.close()
# now adjust each /proc/irq/$num/smp_affinity
interruptdirs = [ f for f in os.listdir(irqpath) if os.path.isdir(os.path.join(irqpath,f)) ]
# IRQ 2 - cascaded signals from IRQs 8-15 (any devices configured to use IRQ 2 will actually be using IRQ 9)
try:
interruptdirs.remove("2")
except ValueError:
pass
# IRQ 0 - system timer (cannot be changed)
try:
interruptdirs.remove("0")
except ValueError:
pass
ret = 0
for i in interruptdirs:
fname = irqpath + i + "/smp_affinity"
cpulist = parse_def_affinity(fname)
mask = 0
for j in cpulist:
mask = mask | 1 << j;
for j in fields:
if sys.argv[1] == "add":
mask = mask | 1 << int(j);
elif sys.argv[1] == "remove":
mask = mask & ~(1 << int(j));
string = get_cpumask(mask)
try:
fo = open(fname, "w")
fo.write(string)
fo.close()
except IOError as e:
sys.stderr.write('Failed to set smp_affinity for IRQ %s: %s\n' % (str(i), str(e)))
ret = 1
sys.exit(ret)
|
