Files
chipsec-chipsec/chipsec/hal/io.py
T
2024-03-22 14:26:59 -07:00

112 lines
3.9 KiB
Python

# CHIPSEC: Platform Security Assessment Framework
# Copyright (c) 2010-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.
#
# Contact information:
# chipsec@intel.com
#
"""
Access to Port I/O
usage:
>>> read_port_byte( 0x61 )
>>> read_port_word( 0x61 )
>>> read_port_dword( 0x61 )
>>> write_port_byte( 0x71, 0 )
>>> write_port_word( 0x71, 0 )
>>> write_port_dword( 0x71, 0 )
"""
from typing import List
from chipsec.library.logger import logger
class PortIO:
def __init__(self, cs):
self.helper = cs.helper
self.cs = cs
def _read_port(self, io_port: int, size: int) -> int:
value = self.helper.read_io_port(io_port, size)
if logger().HAL:
logger().log(f"[io] IN 0x{io_port:04X}: value = 0x{value:08X}, size = 0x{size:02X}")
return value
def _write_port(self, io_port: int, value: int, size: int) -> int:
if logger().HAL:
logger().log(f"[io] OUT 0x{io_port:04X}: value = 0x{value:08X}, size = 0x{size:02X}")
status = self.helper.write_io_port(io_port, value, size)
return status
def read_port_dword(self, io_port: int) -> int:
value = self.helper.read_io_port(io_port, 4)
if logger().HAL:
logger().log(f"[io] reading dword from I/O port 0x{io_port:04X} -> 0x{value:08X}")
return value
def read_port_word(self, io_port: int) -> int:
value = self.helper.read_io_port(io_port, 2)
if logger().HAL:
logger().log(f"[io] reading word from I/O port 0x{io_port:04X} -> 0x{value:04X}")
return value
def read_port_byte(self, io_port: int) -> int:
value = self.helper.read_io_port(io_port, 1)
if logger().HAL:
logger().log(f"[io] reading byte from I/O port 0x{io_port:04X} -> 0x{value:02X}")
return value
def write_port_byte(self, io_port: int, value: int) -> int:
if logger().HAL:
logger().log(f"[io] writing byte to I/O port 0x{io_port:04X} <- 0x{value:02X}")
status = self.helper.write_io_port(io_port, value, 1)
return status
def write_port_word(self, io_port: int, value: int) -> int:
if logger().HAL:
logger().log(f"[io] writing word to I/O port 0x{io_port:04X} <- 0x{value:04X}")
status = self.helper.write_io_port(io_port, value, 2)
return status
def write_port_dword(self, io_port: int, value: int) -> int:
if logger().HAL:
logger().log(f"[io] writing dword to I/O port 0x{io_port:04X} <- 0x{value:08X}")
status = self.helper.write_io_port(io_port, value, 4)
return status
#
# Read registers from I/O range
#
def read_IO(self, range_base: int, range_size: int, size: int = 1) -> List[int]:
n = range_size // size
io_ports = []
for i in range(n):
io_ports.append(self._read_port(range_base + i * size, size))
return io_ports
#
# Dump I/O range
#
def dump_IO(self, range_base: int, range_size: int, size: int = 1) -> None:
n = range_size // size
fmt = f'0{size * 2:d}X'
logger().log(f"[io] I/O register range [0x{range_base:04X}:0x{range_base:04X}+{range_size:04X}]:")
for i in range(n):
reg = self._read_port(range_base + i * size, size)
logger().log(f'+{size * i:04X}: {reg:{fmt}}')