101 lines
3.1 KiB
Python
101 lines
3.1 KiB
Python
# Copyright (C) 2019 Intel Corporation. All rights reserved.
|
|
#
|
|
# SPDX-License-Identifier: BSD-3-Clause
|
|
#
|
|
|
|
import parser_lib
|
|
|
|
RDT_TYPE = {
|
|
"L2":4,
|
|
"L3":2,
|
|
"MBA":8
|
|
}
|
|
|
|
|
|
def dump_cpuid_reg(cmd, reg):
|
|
"""execute the cmd of cpuid, and return the register value by reg
|
|
:param cmd: command what can be executed in shell
|
|
:param reg: register name
|
|
"""
|
|
res = parser_lib.cmd_execute(cmd)
|
|
if reg == "eax":
|
|
idx = 2
|
|
if reg == "ebx":
|
|
idx = 3
|
|
if reg == "edx":
|
|
idx = 5
|
|
|
|
while True:
|
|
line = parser_lib.decode_stdout(res)
|
|
|
|
if not line:
|
|
break
|
|
|
|
if len(line.split()) <= 2:
|
|
continue
|
|
|
|
reg_value = line.split()[idx].split('=')[1]
|
|
|
|
if reg == "eax":
|
|
res_info = int(reg_value, 16) + 1
|
|
break
|
|
elif reg == "ebx":
|
|
res_info = []
|
|
if int(reg_value, 16) & RDT_TYPE['L2'] != 0:
|
|
res_info.append("L2")
|
|
if int(reg_value, 16) & RDT_TYPE['L3'] != 0:
|
|
res_info.append("L3")
|
|
if int(reg_value, 16) & RDT_TYPE['MBA'] != 0:
|
|
res_info.append("MBA")
|
|
break
|
|
elif reg == "edx":
|
|
res_info = int(reg_value, 16) + 1
|
|
break
|
|
|
|
return res_info
|
|
|
|
|
|
def get_clos_info():
|
|
"""Get max clos, mask supported and clos cache type"""
|
|
rdt_res = []
|
|
rdt_clos_max = []
|
|
rdt_mask_max = []
|
|
cmd = "cpuid -1 -r -l 0x10"
|
|
rdt_res = dump_cpuid_reg(cmd, "ebx")
|
|
|
|
if len(rdt_res) == 0:
|
|
parser_lib.print_yel("Resource Allocation is not supported!")
|
|
else:
|
|
for i in range(len(rdt_res)):
|
|
if rdt_res[i] == "L2":
|
|
cmd = "cpuid -1 -r -l 0x10 --subleaf 2"
|
|
rdt_clos_max.append(dump_cpuid_reg(cmd, "edx"))
|
|
l2_info = dump_cpuid_reg(cmd, "eax")
|
|
rdt_mask_max.append(hex((1 << l2_info) - 1))
|
|
if rdt_res[i] == "L3":
|
|
cmd = "cpuid -1 -r -l 0x10 --subleaf 1"
|
|
rdt_clos_max.append(dump_cpuid_reg(cmd, "edx"))
|
|
l3_info = dump_cpuid_reg(cmd, "eax")
|
|
rdt_mask_max.append(hex((1 << l3_info) - 1))
|
|
if rdt_res[i] == "MBA":
|
|
cmd = "cpuid -1 -r -l 0x10 --subleaf 3"
|
|
rdt_clos_max.append(dump_cpuid_reg(cmd, "edx"))
|
|
rdt_mask_max.append(hex(dump_cpuid_reg(cmd, "eax")))
|
|
|
|
return (rdt_res, rdt_clos_max, rdt_mask_max)
|
|
|
|
|
|
def generate_info(board_info):
|
|
"""Generate clos information
|
|
:param board_info: this is the file which stores the hardware board information
|
|
"""
|
|
(rdt_res, rdt_res_clos_max, rdt_res_mask_max) = get_clos_info()
|
|
|
|
with open(board_info, 'a+') as config:
|
|
print("\t<CLOS_INFO>", file=config)
|
|
if ((len(rdt_res) != 0) and (len(rdt_res_clos_max) != 0)):
|
|
print("\trdt resources supported:", ', '.join(rdt_res), file=config)
|
|
print("\trdt resource clos max:",str(rdt_res_clos_max).strip('[]'), file=config)
|
|
print("\trdt resource mask max:",str(rdt_res_mask_max).strip('[]'), file=config)
|
|
print("\t</CLOS_INFO>\n", file=config)
|