#!/usr/local/bin/python
#CHIPSEC: Platform Security Assessment Framework
#Copyright (c) 2010-2015, 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
#



__version__ = '1.0'

import time

from chipsec.command     import BaseCommand
from chipsec.hal.smbus   import *
from chipsec.hal.spd     import *


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):
        if len(self.argv) < 1:
            return False
        return True

    def run(self):
        if 1 > len(self.argv):
            print SPDCommand.__doc__
            return

        try:
            _smbus = SMBus( self.cs )
            _spd   = SPD( _smbus )
        except BaseException, msg:
            print msg
            return

        op = self.argv[0]
        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_SMBUS_ADDRESS

        if( 'detect' == op ):

            self.logger.log( "[CHIPSEC] Searching for DIMMs with SPD.." )
            _spd.detect()

        elif( 'dump' == op ):

            if len(self.argv) > 1:
                dev = self.argv[1].upper()
                dev_addr = chipsec.hal.spd.SPD_DIMM_ADDRESSES[ dev ] if dev in chipsec.hal.spd.SPD_DIMM_ADDRESSES else int(self.argv[1],16)
                if not _spd.isSPDPresent( dev_addr ):
                    self.logger.log( "[CHIPSEC] SPD for DIMM 0x%X is not found" % 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) > 1:
                dev = self.argv[1].upper()
                dev_addr = chipsec.hal.spd.SPD_DIMM_ADDRESSES[ dev ] if dev in chipsec.hal.spd.SPD_DIMM_ADDRESSES else int(self.argv[1],16)
            if not _spd.isSPDPresent( dev_addr ):
                self.logger.log( "[CHIPSEC] SPD for DIMM 0x%X is not found" % dev_addr )
                return

            off = int(self.argv[2],16)
            if( 'read' == op ):
                val      = _spd.read_byte( off, dev_addr )
                self.logger.log( "[CHIPSEC] SPD read: offset 0x%X = 0x%X" % (off, val) )
            elif( 'write' == op ):
                val      = int(self.argv[3],16)
                self.logger.log( "[CHIPSEC] SPD write: offset 0x%X = 0x%X" % (off, val) )
                _spd.write_byte( off, val, dev_addr )

        else:
            self.logger.error( "unknown command-line option '%.32s'" % op )
            self.logger.log( SPDCommand.__doc__ )
            return

        self.logger.log( "[CHIPSEC] (spd) time elapsed %.3f" % (time.time()-t) )

commands = { 'spd': SPDCommand }
