Make the platform cfg loader to only accept hex based names

In the cfg folder, filter out the non-hex folder names, currently
we only filter out folders whose name starts with '__'

Signed-off-by: Ignacio Hernandez <ignacio.hernandez@intel.com>
This commit is contained in:
Ignacio Hernandez
2020-05-04 11:46:03 -07:00
committed by BrentHoltsclaw
parent 2c6528a38f
commit 33525e2ee0
2 changed files with 5 additions and 1 deletions
+2 -1
View File
@@ -35,6 +35,7 @@ from chipsec.hal import cpu, io, iobar, mmio, msgbus, msr, pci, physmem, ucode,
from chipsec.hal.pci import PCI_HDR_RID_OFF
from chipsec.logger import logger
from chipsec.defines import is_hex
import chipsec.file
@@ -341,7 +342,7 @@ class Chipset:
# find VID
_cfg_path = os.path.join( chipsec.file.get_main_dir(), 'chipsec', 'cfg' )
VID = [f for f in os.listdir(_cfg_path) if os.path.isdir(os.path.join(_cfg_path, f)) and not f[:2] == '__' ]
VID = [f for f in os.listdir(_cfg_path) if os.path.isdir(os.path.join(_cfg_path, f)) and is_hex(f) ]
# create dictionaries
for vid in VID:
self.chipset_dictionary[int(vid,16)] = collections.defaultdict(list)
+3
View File
@@ -180,6 +180,9 @@ def get_version():
def is_printable(seq):
return set(seq).issubset(set(string.printable))
def is_hex(maybe_hex):
return all(char in string.hexdigits for char in maybe_hex)
def get_message():
msg_str = ""
chipsec_folder = os.path.abspath(chipsec.file.get_main_dir())