vector-uefi/fd/tool/chipsec/utilcmd/interrupts_cmd.py
#!/usr/local/bin/python
#CHIPSEC: Platform Security Assessment Framework
#Copyright (c) 2010-2014, 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
#
#
# usage as a standalone utility:
#
## \addtogroup standalone
#chipsec_util smi / chipsec_util nmi
#-----------
#~~~
#chipsec_util smi <SMI_code> <SMI_data> [RAX] [RBX] [RCX] [RDX] [RSI] [RDI]
#chipsec_util nmi
#''
# Examples:
#''
# chipsec_util smi 0xDE 0x0
# chipsec_util nmi
#~~~
__version__ = '1.0'
import os
import sys
import time
import chipsec_util
#from chipsec_util import global_usage, chipsec_util_commands, _cs
from chipsec_util import chipsec_util_commands, _cs
from chipsec.logger import *
from chipsec.file import *
from chipsec.hal.interrupts import Interrupts
#_cs = cs()
usage = "chipsec_util smi <SMI_code> <SMI_data> [RAX] [RBX] [RCX] [RDX] [RSI] [RDI]\n\n" + \
"chipsec_util nmi\n" + \
"Examples:\n" + \
" chipsec_util smi 0xDE 0x0\n" + \
" chipsec_util nmi\n\n"
chipsec_util.global_usage += usage
# ###################################################################
#
# CPU Interrupts
#
# ###################################################################
def smi(argv):
try:
interrupts = Interrupts( _cs )
except RuntimeError, msg:
print msg
return
SMI_code_port_value = 0xF
SMI_data_port_value = 0x0
if (2 == len(argv)):
pass
elif (3 < len(argv)):
SMI_code_port_value = int(argv[2],16)
SMI_data_port_value = int(argv[3],16)
logger().log( "[CHIPSEC] Sending SW SMI (code: 0x%02X, data: 0x%02X).." % (SMI_code_port_value, SMI_data_port_value) )
if (4 == len(argv)):
interrupts.send_SMI_APMC( SMI_code_port_value, SMI_data_port_value )
elif (10 == len(argv)):
_rax = int(argv[4],16)
_rbx = int(argv[5],16)
_rcx = int(argv[6],16)
_rdx = int(argv[7],16)
_rsi = int(argv[8],16)
_rdi = int(argv[9],16)
logger().log( " RAX: 0x%016X (AX will be overwridden with values of SW SMI ports B2/B3)" % _rax )
logger().log( " RBX: 0x%016X" % _rbx )
logger().log( " RCX: 0x%016X" % _rcx )
logger().log( " RDX: 0x%016X (DX will be overwridden with 0x00B2)" % _rdx )
logger().log( " RSI: 0x%016X" % _rsi )
logger().log( " RDI: 0x%016X" % _rdi )
interrupts.send_SW_SMI( SMI_code_port_value, SMI_data_port_value, _rax, _rbx, _rcx, _rdx, _rsi, _rdi )
else: print usage
else: print usage
def nmi(argv):
if 2 < len(argv):
print usage
try:
interrupts = Interrupts( _cs )
except RuntimeError, msg:
print msg
return
logger().log( "[CHIPSEC] Sending NMI#.." )
interrupts.send_NMI()
chipsec_util_commands['nmi'] = {'func' : nmi, 'start_driver' : True }
chipsec_util_commands['smi'] = {'func' : smi, 'start_driver' : True }