mirror of
https://github.com/chipsec/chipsec
synced 2026-06-08 13:31:00 +00:00
William Leara
5ffa64907d
This commit fixes all the E302 errors. They were found by running "flake8 ." from the root directory. This is an error defined by "pycodestyle", and has no functional impact to the source. This is simply fixing whitespace as defined by CHIPSEC's .flake8 configuration. background: https://www.flake8rules.com/rules/E302.html https://pycodestyle.pycqa.org/en/latest/intro.html#error-codes:~:text=line%2C%20found%200-,E302,-expected%202%20blank versions: flake8 v7.2.0, python v3.12.6 Signed-off-by: William Leara <william.leara@dell.com>
79 lines
2.4 KiB
Python
79 lines
2.4 KiB
Python
# CHIPSEC: Platform Security Assessment Framework
|
|
# Copyright (c) 2016, Google
|
|
# Copyright (c) 2019-2021, Intel Corporation
|
|
#
|
|
# This program is free software; you can redistribute it and/or
|
|
# modify it under the terms of the GNU General Public License
|
|
# as published by the Free Software Foundation; Version 2.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with this program; if not, write to the Free Software
|
|
# Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
#
|
|
|
|
|
|
from enum import Enum
|
|
import traceback
|
|
|
|
from chipsec.library.logger import logger
|
|
from chipsec.library.defines import CHIPSET_CODE_UNKNOWN
|
|
from chipsec.testcase import ExitCode
|
|
|
|
|
|
class BaseCommand:
|
|
|
|
def __init__(self, argv, cs=None):
|
|
self.argv = argv
|
|
self.logger = logger()
|
|
self.cs = cs
|
|
self.ExitCode = ExitCode.OK
|
|
|
|
def run(self) -> None:
|
|
try:
|
|
self.func()
|
|
except Exception:
|
|
self.logger.log_error('An error occured during the execution of the command!')
|
|
self.logger.log_error('Please run with the debug option for further details')
|
|
if logger().DEBUG:
|
|
traceback.print_exc()
|
|
|
|
def prerun(self) -> bool:
|
|
if self.requirements().load_config() and self.cs.Cfg.code == CHIPSET_CODE_UNKNOWN:
|
|
self.logger.log_error('This module requires a configuration to be loaded. Please use the "-p XXX" flag to specify a platform configuration to load.')
|
|
return False
|
|
return True
|
|
|
|
def set_up(self) -> None:
|
|
pass
|
|
|
|
def tear_down(self) -> None:
|
|
pass
|
|
|
|
def parse_arguments(self) -> None:
|
|
raise NotImplementedError('sub class should overwrite the parse_arguments() method')
|
|
|
|
def requirements(self) -> 'toLoad':
|
|
raise NotImplementedError('sub class should overwrite the requirements() method')
|
|
|
|
|
|
class toLoad(Enum):
|
|
Nil = 0
|
|
Config = 1
|
|
Driver = 2
|
|
All = 3
|
|
|
|
def load_config(self) -> bool:
|
|
if self in [self.Config, self.All]:
|
|
return True
|
|
return False
|
|
|
|
def load_driver(self) -> bool:
|
|
if self in [self.Driver, self.All]:
|
|
return True
|
|
return False
|