mirror of
https://github.com/chipsec/chipsec
synced 2026-06-08 13:31:00 +00:00
BrentHoltsclaw
118 lines
4.0 KiB
Python
118 lines
4.0 KiB
Python
#!/usr/bin/python
|
|
#CHIPSEC: Platform Security Assessment Framework
|
|
#Copyright (c) 2010-2019, 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.
|
|
#
|
|
#Contact information:
|
|
#chipsec@intel.com
|
|
#
|
|
|
|
|
|
import time
|
|
|
|
from chipsec.command import BaseCommand
|
|
from chipsec.hal import smbus, spd
|
|
|
|
class SPDCommand(BaseCommand):
|
|
"""
|
|
>>> chipsec_util spd detect
|
|
>>> chipsec_util spd dump [device_addr]
|
|
>>> chipsec_util spd read <device_addr> <offset>
|
|
>>> chipsec_util spd write <device_addr> <offset> <byte_val>
|
|
|
|
Examples:
|
|
|
|
>>> chipsec_util spd detect
|
|
>>> chipsec_util spd dump DIMM0
|
|
>>> chipsec_util spd read 0xA0 0x0
|
|
>>> chipsec_util spd write 0xA0 0x0 0xAA
|
|
"""
|
|
|
|
def requires_driver(self):
|
|
# No driver required when printing the util documentation
|
|
if len(self.argv) < 3:
|
|
return False
|
|
return True
|
|
|
|
def run(self):
|
|
if len(self.argv) < 3:
|
|
print (SPDCommand.__doc__)
|
|
return
|
|
|
|
try:
|
|
_smbus = smbus.SMBus( self.cs )
|
|
_spd = spd.SPD( _smbus )
|
|
except BaseException as msg:
|
|
print (msg)
|
|
return
|
|
|
|
op = self.argv[2]
|
|
t = time.time()
|
|
|
|
if not _smbus.is_SMBus_supported():
|
|
self.logger.log( "[CHIPSEC] SMBus controller is not supported" )
|
|
return
|
|
#smbus.display_SMBus_info()
|
|
|
|
dev_addr = spd.SPD_SMBUS_ADDRESS
|
|
|
|
if( 'detect' == op ):
|
|
|
|
self.logger.log( "[CHIPSEC] Searching for DIMMs with SPD.." )
|
|
_dimms = _spd.detect()
|
|
if _dimms is not None:
|
|
self.logger.log( "Detected the following SPD devices:" )
|
|
for _dimm in _dimms: self.logger.log( "{}: 0x{:02X}".format(spd.SPD_DIMMS[_dimm],_dimm) )
|
|
|
|
elif( 'dump' == op ):
|
|
|
|
if len(self.argv) > 3:
|
|
dev = self.argv[3].upper()
|
|
dev_addr = spd.SPD_DIMM_ADDRESSES[ dev ] if dev in spd.SPD_DIMM_ADDRESSES else int(self.argv[3],16)
|
|
if not _spd.isSPDPresent( dev_addr ):
|
|
self.logger.log( "[CHIPSEC] SPD for DIMM 0x{:X} is not found".format(dev_addr) )
|
|
return
|
|
_spd.decode( dev_addr )
|
|
else:
|
|
_dimms = _spd.detect()
|
|
for d in _dimms: _spd.decode( d )
|
|
|
|
elif( 'read' == op ) or ( 'write' == op ):
|
|
|
|
if len(self.argv) > 3:
|
|
dev = self.argv[3].upper()
|
|
dev_addr = spd.SPD_DIMM_ADDRESSES[ dev ] if dev in spd.SPD_DIMM_ADDRESSES else int(self.argv[3],16)
|
|
if not _spd.isSPDPresent( dev_addr ):
|
|
self.logger.log( "[CHIPSEC] SPD for DIMM 0x{:X} is not found".format(dev_addr) )
|
|
return
|
|
|
|
off = int(self.argv[4],16)
|
|
if( 'read' == op ):
|
|
val = _spd.read_byte( off, dev_addr )
|
|
self.logger.log( "[CHIPSEC] SPD read: offset 0x{:X} = 0x{:X}".format(off, val) )
|
|
elif( 'write' == op ):
|
|
val = int(self.argv[5],16)
|
|
self.logger.log( "[CHIPSEC] SPD write: offset 0x{:X} = 0x{:X}".format(off, val) )
|
|
_spd.write_byte( off, val, dev_addr )
|
|
|
|
else:
|
|
self.logger.error( "unknown command-line option '{:.32}'".format(op) )
|
|
self.logger.log( SPDCommand.__doc__ )
|
|
return
|
|
|
|
self.logger.log( "[CHIPSEC] (spd) time elapsed {:.3f}".format(time.time()-t) )
|
|
|
|
commands = { 'spd': SPDCommand }
|