191. Инструкция rti

реализовать поведение RTI.

Урок 191 из 356 · tests/chapter_01_cpu/test_191_instruction_rti.py

На этом шаге измените только `emulator/cpu/instructions.py, добавив rti(cpu)`. Предварительное условие: шаг 188 ввёл необходимые API для прерываний и флага Break.

Зачем нужен этот шаг

При входе в прерывание в следующие три слота стека записываются статус, младший байт адреса возврата и старший байт адреса возврата. RTI должен извлечь эти байты в этом порядке LIFO и восстановить точный PC; в отличие от RTS, он не должен прибавлять единицу.

Предлагаемая реализация

cpu.s = (cpu.s + 1) & 0xFF
flags = cpu.bus.read(0x0100 | cpu.s)
cpu.p = flags & 0b1100_1111

cpu.s = (cpu.s + 1) & 0xFF
low = cpu.bus.read(0x0100 | cpu.s)

cpu.s = (cpu.s + 1) & 0xFF
high = cpu.bus.read(0x0100 | cpu.s)

cpu.pc = (high << 8) | low

Поместите эти инструкции в `def rti(cpu: CPU)`.

Инварианты: увеличивайте 8-битный S перед каждым чтением; читайте только страницу стека $0100; заменяйте P после маскирования битов 4 и 5; извлекайте младший байт перед старшим; увеличивайте S три раза; оставляйте восстановленный PC без инкремента.

Заблуждение: RTI — это не RTS для прерываний. Применение финального `+ 1` из RTS приводит к возврату не на ту инструкцию.

Вне рамок: импорт/регистрация опкода по адресу $40 относится к шагу 192. Поведение NMI относится к более поздним шагам и не должно добавляться здесь.

Запустить этот урок

uv run pytest tests/chapter_01_cpu/test_191_instruction_rti.py -v

Пример реализации

Предлагаемая реализация

cpu.s = (cpu.s + 1) & 0xFF
flags = cpu.bus.read(0x0100 | cpu.s)
cpu.p = flags & 0b1100_1111

cpu.s = (cpu.s + 1) & 0xFF
low = cpu.bus.read(0x0100 | cpu.s)

cpu.s = (cpu.s + 1) & 0xFF
high = cpu.bus.read(0x0100 | cpu.s)

cpu.pc = (high << 8) | low

Показать код тестаtest_191_instruction_rti.py

from emulator.cpu.instructions import rti
from tests.helpers import make_cpu


CARRY_FLAG = 1 << 0
ZERO_FLAG = 1 << 1
INTERRUPT_DISABLE_FLAG = 1 << 2
BREAK_FLAG = 1 << 4
ONE_FLAG = 1 << 5
OVERFLOW_FLAG = 1 << 6
NEGATIVE_FLAG = 1 << 7
STACK_BASE = 0x0100


def test_rti_restores_program_counter_from_stack_without_adding_one():
    """
    Objective:
    RTI restores the exact PC from the stack.

    If the stack contains $1234, PC becomes $1234, not $1235.
    """
    cpu = make_cpu()
    cpu.s = 0xFA
    cpu.bus.write(STACK_BASE | 0xFB, 0x00)
    cpu.bus.write(STACK_BASE | 0xFC, 0x34)
    cpu.bus.write(STACK_BASE | 0xFD, 0x12)

    rti(cpu)

    assert cpu.pc == 0x1234


def test_rti_pulls_status_then_pc_low_then_pc_high():
    """
    Objective:
    RTI must pull stack bytes in interrupt-return order:

        status -> PC low -> PC high

    This is the reverse of what BRK/interrupt entry pushed.
    """
    cpu = make_cpu()
    cpu.s = 0xFA
    cpu.bus.write(STACK_BASE | 0xFB, CARRY_FLAG | NEGATIVE_FLAG)
    cpu.bus.write(STACK_BASE | 0xFC, 0xCD)
    cpu.bus.write(STACK_BASE | 0xFD, 0xAB)

    rti(cpu)

    assert cpu.p == (CARRY_FLAG | NEGATIVE_FLAG)
    assert cpu.pc == 0xABCD


def test_rti_increments_stack_pointer_three_times():
    """Objective: RTI pulls status, PC low, and PC high, so S increases by 3."""
    cpu = make_cpu()
    cpu.s = 0xFA
    cpu.bus.write(STACK_BASE | 0xFB, 0x00)
    cpu.bus.write(STACK_BASE | 0xFC, 0x34)
    cpu.bus.write(STACK_BASE | 0xFD, 0x12)

    rti(cpu)

    assert cpu.s == 0xFD


def test_rti_restores_status_flags_from_stack():
    """
    Objective:
    RTI restores saved processor status flags from the stack.

    These flags affect later branches and arithmetic, so RTI must not leave the
    interrupt handler's temporary flags behind.
    """
    cpu = make_cpu()
    cpu.s = 0xFA
    saved_flags = CARRY_FLAG | ZERO_FLAG | INTERRUPT_DISABLE_FLAG | OVERFLOW_FLAG | NEGATIVE_FLAG
    cpu.bus.write(STACK_BASE | 0xFB, saved_flags)
    cpu.bus.write(STACK_BASE | 0xFC, 0x34)
    cpu.bus.write(STACK_BASE | 0xFD, 0x12)

    rti(cpu)

    assert cpu.p == saved_flags


def test_rti_clears_break_and_unused_status_bits_from_pulled_status():
    """
    Objective:
    In this emulator model, status bits 4 and 5 are not kept as persistent CPU
    state after RTI.

    Why:
    BRK may push a status byte with Break set, but Break is metadata in the
    saved stack byte, not a normal long-lived CPU flag here.
    """
    cpu = make_cpu()
    cpu.s = 0xFA
    saved_flags_with_break_and_unused = 0b0011_0000 | CARRY_FLAG | NEGATIVE_FLAG
    cpu.bus.write(STACK_BASE | 0xFB, saved_flags_with_break_and_unused)
    cpu.bus.write(STACK_BASE | 0xFC, 0x34)
    cpu.bus.write(STACK_BASE | 0xFD, 0x12)

    rti(cpu)

    assert (cpu.p & BREAK_FLAG) == 0
    assert (cpu.p & ONE_FLAG) == 0
    assert cpu.p == (CARRY_FLAG | NEGATIVE_FLAG)


def test_rti_stack_pointer_wraps_to_8_bits():
    """Objective: S is an 8-bit stack pointer and wraps while pulling bytes."""
    cpu = make_cpu()
    cpu.s = 0xFE
    cpu.bus.write(STACK_BASE | 0xFF, CARRY_FLAG)
    cpu.bus.write(STACK_BASE | 0x00, 0x78)
    cpu.bus.write(STACK_BASE | 0x01, 0x56)

    rti(cpu)

    assert cpu.p == CARRY_FLAG
    assert cpu.pc == 0x5678
    assert cpu.s == 0x01

Тестовые функции (6)

test_rti_restores_program_counter_from_stack_without_adding_one
test_rti_pulls_status_then_pc_low_then_pc_high
test_rti_increments_stack_pointer_three_times
test_rti_restores_status_flags_from_stack
test_rti_clears_break_and_unused_status_bits_from_pulled_status
test_rti_stack_pointer_wraps_to_8_bits