main.c

Go to the documentation of this file.

#include <xtf.h>

const char test_title[] = "Test nmi-taskswitch-priv";

static uint8_t nmi_stack[PAGE_SIZE] __page_aligned_bss;

void entry_NMI_task(void);
asm("exit_NMI_task:"
    "iretl;"
    "entry_NMI_task:;"

    "push $0;"
    "push $" STR(X86_EXC_NMI) ";"

    "push %es;"
    "push %ds;"

    "push %edi;"
    "push %esi;"
    "push %edx;"
    "push %ecx;"
    "push %eax;"
    "push %ebx;"
    "push %ebp;"

    "push %esp;"
    "call do_exception;"
    "add $1*4, %esp;"

    "pop %ebp;"
    "pop %ebx;"
    "pop %eax;"
    "pop %ecx;"
    "pop %edx;"
    "pop %esi;"
    "pop %edi;"

    "pop %ds;"
    "pop %es;"

    "add $2*4, %esp;" /* entry_vector/error_code. */

    "jmp exit_NMI_task;");


static env_tss nmi_tss __aligned(16) =
{
    .eip    = _u(entry_NMI_task),
    .cs     = __KERN_CS,
    .eflags = X86_EFLAGS_MBS,
    .esp    = _u(nmi_stack + PAGE_SIZE),
    .ss     = __KERN_DS,

    .ds     = __USER_DS,
    .es     = __USER_DS,
    .fs     = __USER_DS,
    .gs     = __USER_DS,

    .cr3    = _u(cr3_target),
    .iopb   = X86_TSS_INVALID_IO_BITMAP,
};

bool do_unhandled_exception(struct cpu_regs *regs)
{
    if ( regs->entry_vector != X86_EXC_NMI )
        return false;

    unsigned int curr_ts = str();
    if ( curr_ts != GDTE_AVAIL0 * 8 )
        xtf_failure("Fail: Running NMI handler with unexpected %%tr\n"
                    "  Expected %04x, got %04x\n", GDTE_AVAIL0 * 8, curr_ts);

    /* Follow the TSS link pointer to get the interrupted state. */
    env_tss *t = _p(user_desc_base(&gdt[nmi_tss.link >> 3]));

    printk("  NMI at %04x:%p, stack %04x:%p\n",
           t->cs, _p(t->eip), t->ss, _p(t->esp));

    return true;
}

static void __user_text user_inject_nmi(void)
{
    apic_mmio_icr_write(APIC_DEST_SELF | APIC_DM_NMI);
}

void test_main(void)
{
    unsigned int curr_ts;
    int rc = apic_init(APIC_MODE_XAPIC);

    if ( rc )
        return xtf_error("Error: Unable to set up xapic mode: %d\n", rc);

    /*
     * Remap the Local APIC MMIO window as USER, so user_inject_nmi() can send
     * a self-NMI.  No INVLPG, as this is a strict relaxing of permissions.
     */
    pae_l2_identmap[APIC_DEFAULT_BASE >> PAE_L2_PT_SHIFT] |= _PAGE_USER;

    /*
     * Set up NMI handling to be a task gate.
     */
    pack_tss_desc(&gdt[GDTE_AVAIL0], &nmi_tss);
    pack_task_gate(&idt[X86_EXC_NMI], GDTE_AVAIL0 * 8);

    /*
     * Send an NMI from supervisor mode, checking that we task switch back to
     * the expected TSS.
     */
    printk("First self-nmi, from supervisor mode\n");
    apic_mmio_icr_write(APIC_DEST_SELF | APIC_DM_NMI);

    if ( (curr_ts = str()) != TSS_SEL )
        xtf_failure("Fail: Running main task with unexpected %%tr\n"
                    "  Expected %04x, got %04x\n", TSS_SEL, curr_ts);

    /*
     * Send an NMI from user mode, and again check that we are in the expected
     * task.
     */
    printk("Second self-nmi, from user mode\n");
    exec_user_void(user_inject_nmi);

    if ( (curr_ts = str()) != TSS_SEL )
        xtf_failure("Fail: Running main task with unexpected %%tr\n"
                    "  Expected %04x, got %04x\n", TSS_SEL, curr_ts);

    /*
     * If Xen is still alive, it handled the user=>supervisor task switch
     * properly.
     */
    xtf_success(NULL);
}

/*
 * Local variables:
 * mode: C
 * c-file-style: "BSD"
 * c-basic-offset: 4
 * tab-width: 4
 * indent-tabs-mode: nil
 * End:
 */