mirror of
https://github.com/chipsec/chipsec
synced 2026-06-08 13:31:00 +00:00
LeaMessenger
72419667f5
* Updated info, cpuid, and cr to argparse * Updated to argparse Signed-off-by: Lea Messenger <lea.messenger@intel.com>
150 lines
6.7 KiB
Python
150 lines
6.7 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
|
|
#
|
|
|
|
from time import time
|
|
from argparse import ArgumentParser
|
|
|
|
import chipsec.hal.cpu
|
|
from chipsec.hal.cpu import CPU, CPURuntimeError
|
|
from chipsec.command import BaseCommand
|
|
|
|
# ###################################################################
|
|
#
|
|
# CPU utility
|
|
#
|
|
# ###################################################################
|
|
class CPUCommand(BaseCommand):
|
|
"""
|
|
>>> chipsec_util cpu info
|
|
>>> chipsec_util cpu cr <thread> <cr_number> [value]
|
|
>>> chipsec_util cpu cpuid <eax> [ecx]
|
|
>>> chipsec_util cpu pt [paging_base_cr3]
|
|
|
|
Examples:
|
|
|
|
>>> chipsec_util cpu info
|
|
>>> chipsec_util cpu cr 0 0
|
|
>>> chipsec_util cpu cr 0 4 0x0
|
|
>>> chipsec_util cpu cpuid 0x40000000
|
|
>>> chipsec_util cpu pt
|
|
"""
|
|
|
|
def requires_driver(self):
|
|
parser = ArgumentParser(usage=CPUCommand.__doc__)
|
|
subparsers = parser.add_subparsers()
|
|
parser_info = subparsers.add_parser('info')
|
|
parser_cr = subparsers.add_parser('cr')
|
|
parser_cpuid = subparsers.add_parser('cpuid')
|
|
parser_pt = subparsers.add_parser('pt')
|
|
parser_info.set_defaults(func=self.cpu_info)
|
|
parser_cr.set_defaults(func=self.cpu_cr)
|
|
parser_cpuid.set_defaults(func=self.cpu_cpuid)
|
|
parser_pt.set_defaults(func=self.cpu_pt)
|
|
parser_cr.add_argument('thread',type=int,nargs='?',default=None)
|
|
parser_cr.add_argument('cr_number',type=int,nargs='?',default=None)
|
|
parser_cr.add_argument('value',type=lambda x: int(x,0),nargs='?',default=None)
|
|
parser_cpuid.add_argument('eax',type=lambda x: int(x,0))
|
|
parser_cpuid.add_argument('ecx',type=lambda x: int(x,0),nargs='?',default=0)
|
|
parser_pt.add_argument('cr3',type=lambda x: int(x,0),nargs='?',default=None)
|
|
|
|
parser.parse_args(self.argv[2:],namespace=CPUCommand)
|
|
|
|
return True
|
|
|
|
def cpu_info(self):
|
|
self.logger.log( "[CHIPSEC] CPU information:" )
|
|
ht = self.cs.cpu.is_HT_active()
|
|
threads_per_core = self.cs.cpu.get_number_logical_processor_per_core()
|
|
threads_per_pkg = self.cs.cpu.get_number_logical_processor_per_package()
|
|
cores_per_pkg = self.cs.cpu.get_number_physical_processor_per_package()
|
|
self.logger.log( " Hyper-Threading : {}".format('Enabled' if ht else 'Disabled') )
|
|
self.logger.log( " CPU cores per package : {:d}".format(cores_per_pkg) )
|
|
self.logger.log( " CPU threads per core : {:d}".format(threads_per_core) )
|
|
self.logger.log( " CPU threads per package : {:d}".format(threads_per_pkg) )
|
|
try:
|
|
threads_count = self.cs.cpu.get_number_threads_from_APIC_table()
|
|
sockets_count = self.cs.cpu.get_number_sockets_from_APIC_table()
|
|
self.logger.log( " Number of sockets : {:d}".format(sockets_count) )
|
|
self.logger.log( " Number of CPU threads : {:d}".format(threads_count) )
|
|
except:
|
|
pass
|
|
|
|
def cpu_cr(self):
|
|
if self.value is not None:
|
|
self.logger.log( "[CHIPSEC] CPU{:d}: write CR{:d} <- 0x{:08X}".format(self.thread, self.cr_number, self.value) )
|
|
self.cs.cpu.write_cr( self.thread, self.cr_number, self.value )
|
|
return True
|
|
elif self.cr_number is not None:
|
|
value = self.cs.cpu.read_cr( self.thread, self.cr_number )
|
|
self.logger.log( "[CHIPSEC] CPU{:d}: read CR{:d} -> 0x{:08X}".format(self.thread, self.cr_number, value) )
|
|
return value
|
|
else:
|
|
for tid in range(self.cs.msr.get_cpu_thread_count()):
|
|
cr0 = self.cs.cpu.read_cr( tid, 0 )
|
|
cr2 = self.cs.cpu.read_cr( tid, 2 )
|
|
cr3 = self.cs.cpu.read_cr( tid, 3 )
|
|
cr4 = self.cs.cpu.read_cr( tid, 4 )
|
|
cr8 = self.cs.cpu.read_cr( tid, 8 )
|
|
self.logger.log( "[CHIPSEC][cpu{:d}] x86 Control Registers:".format(tid) )
|
|
self.logger.log( " CR0: 0x{:016X}".format(cr0) )
|
|
self.logger.log( " CR2: 0x{:016X}".format(cr2) )
|
|
self.logger.log( " CR3: 0x{:016X}".format(cr3) )
|
|
self.logger.log( " CR4: 0x{:016X}".format(cr4) )
|
|
self.logger.log( " CR8: 0x{:016X}".format(cr8) )
|
|
|
|
def cpu_cpuid(self):
|
|
self.logger.log( "[CHIPSEC] CPUID < EAX: 0x{:08X}".format(self.eax))
|
|
self.logger.log( "[CHIPSEC] ECX: 0x{:08X}".format(self.ecx))
|
|
|
|
(_eax,_ebx,_ecx,_edx) = self.cs.cpu.cpuid( self.eax, self.ecx )
|
|
|
|
self.logger.log( "[CHIPSEC] CPUID > EAX: 0x%08X" % _eax )
|
|
self.logger.log( "[CHIPSEC] EBX: 0x%08X" % _ebx )
|
|
self.logger.log( "[CHIPSEC] ECX: 0x%08X" % _ecx )
|
|
self.logger.log( "[CHIPSEC] EDX: 0x%08X" % _edx )
|
|
|
|
def cpu_pt(self):
|
|
if self.cr3 is not None:
|
|
pt_fname = 'pt_{:08X}'.format(cr3)
|
|
self.logger.log( "[CHIPSEC] paging physical base (CR3): 0x{:016X}".format(cr3) )
|
|
self.logger.log( "[CHIPSEC] dumping paging hierarchy to '{}'...".format(pt_fname) )
|
|
self.cs.cpu.dump_page_tables( self.cr3, pt_fname )
|
|
else:
|
|
for tid in range(self.cs.msr.get_cpu_thread_count()):
|
|
cr3 = self.cs.cpu.read_cr( tid, 3 )
|
|
pt_fname = 'cpu{:d}_pt_{:08X}'.format(tid,cr3)
|
|
self.logger.log( "[CHIPSEC][cpu{:d}] paging physical base (CR3): 0x{:016X}".format(tid,cr3) )
|
|
self.logger.log( "[CHIPSEC][cpu{:d}] dumping paging hierarchy to '{}'...".format(tid,pt_fname) )
|
|
self.cs.cpu.dump_page_tables( cr3, pt_fname )
|
|
|
|
def run(self):
|
|
t = time()
|
|
try:
|
|
self._cpu = CPU(self.cs)
|
|
except CPURuntimeError as msg:
|
|
print(msg)
|
|
return
|
|
|
|
self.func()
|
|
self.logger.log( "[CHIPSEC] (cpu) time elapsed {:.3f}".format(time()-t) )
|
|
|
|
commands = { 'cpu': CPUCommand }
|