main.c

Go to the documentation of this file.

 
#define TEST_APIC_MODE APIC_MODE_XAPIC
 
#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;"
 
    "mov %esp, %eax;"
    "call do_exception;"
 
    "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_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_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:
 */