import os
import fnmatch
from . import base
from tuned.utils.commands import commands
from tuned import logs
log = logs.get()
cmd = commands()
SYSFS_DIR = "/sys/devices/system/cpu/"
class package2cpus(base.Function):
"""
Provides cpu device list for a package (socket)
"""
def __init__(self):
super(package2cpus, self).__init__("package2cpus", 0)
def execute(self, args):
if not super(package2cpus, self).execute(args):
return None
if len(args) <= 0:
return None
try:
all_cpus = fnmatch.filter(os.listdir(SYSFS_DIR), "cpu[0-9]*")
except OSError:
return None
devices = []
for cpu in all_cpus:
f = SYSFS_DIR + cpu + "/topology/physical_package_id"
if not os.path.exists(f):
log.warning("File '%s' does not exist" % f)
continue
value = cmd.read_file(f)
try:
package_id = int(value)
except ValueError:
log.warning("Invalid package id '%s' for cpu '%s'" % (value, cpu))
continue
for package_pattern in args:
try:
this_package_id = int(package_pattern)
except ValueError:
if fnmatch.fnmatch(value, package_pattern):
devices.append(cpu)
else:
if package_id == this_package_id:
devices.append(cpu)
return ",".join(devices) if len(devices) > 0 else None
|