Skip to content

Commit b8ba7e3

Browse files
committed
Add 'has_all_fields' function to parent register class and store vid in pci device
Signed-off-by: Nathaniel Mitchell <nathaniel.p.mitchell@intel.com>
1 parent 5f3c7e5 commit b8ba7e3

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

chipsec/cfg/parsers/ip/pci_device.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -179,6 +179,7 @@ def __init__(self, cfg_obj: Dict[str, Any]):
179179
try:
180180
# Handle device ID and name
181181
self.did: Optional[int] = cfg_obj.get('did', None)
182+
self.vid: Optional[int] = cfg_obj.get('vid', None)
182183
if 'name' not in cfg_obj and self.did is not None:
183184
cfg_obj['name'] = str(self.did)
184185
self.__name_updated = False

chipsec/library/register.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -502,6 +502,9 @@ def get_field(
502502
def has_field(self, field_name: str) -> bool:
503503
return self.fields.get(field_name, None) is not None
504504

505+
def has_all_fields(self, field_names: List[str]) -> bool:
506+
return all(self.has_field(name) for name in field_names)
507+
505508
def get_mask(self) -> int:
506509
mask = make_mask(self.size * 8)
507510
return mask

0 commit comments

Comments
 (0)