{"componentChunkName":"component---src-templates-post-template-js","path":"/ctf-angr-hooking-en","result":{"data":{"markdownRemark":{"id":"c0240fc0-5ebc-536e-95e5-c17eddafe8a4","html":"
\n

This page has been machine-translated from the original page.

\n
\n

This article was written for CTF Advent Calendar 2023.

\n

The previous article was Edwow Math (N30Z30N)‘s ”Good CTFs, Bad CTFs, Ordinary CTFs (for CTF Advent Calendar 2023) - Learning cyber security by playing and enjoying CTFs.”

\n

As for the next article… unfortunately, it looks like nobody is scheduled to write one at the moment, so I will skip that part.

\n

This time, I will try using angr to hook arbitrary symbol functions so that I can partially modify their behavior or override them entirely.

\n\n

Table of Contents

\n\n

Using the hook_symbol Method in angr

\n

In angr, you can hook specific symbol functions by using the hook_symbol method.

\n

The basic way to override a symbol function with hook_symbol is to create any class that inherits from angr.SimProcedure, then call it while specifying the symbol name of the function you want to hook through its run method.

\n

At that time, you can also pass arbitrary arguments to the run method.

\n
from angr import Project, SimProcedure\nproject = Project('examples/fauxware/fauxware')\n\nclass BugFree(SimProcedure):\n   def run(self, argc, argv):\n       print('Program running with argc=%s and argv=%s' % (argc, argv))\n       return 0\n\n# this assumes we have symbols for the binary\nproject.hook_symbol('main', BugFree())\n\n# Run a quick execution!\nsimgr = project.factory.simulation_manager()\nsimgr.run()
\n

Reference: Hooks and SimProcedures - angr documentation

\n

By using SimProcedures and hook_symbol, you can, for example, override the behavior of arbitrary functions or library functions and recover the Flag.

\n

Overriding the Return Value of a Symbol Function in angr to Recover the Flag

\n

First, I will use angr to analyze a binary built from the following C code and recover the Flag.

\n

The program compiled from the code below takes a string from standard input and prints Success only if the input is Flag{angr} and the value generated by the rand() function is exactly 0x12345.

\n
// gcc sample1.c -o chal.bin\n#include <stdio.h>\n#include <string.h>\n#include <time.h>\n\nvoid main()\n{\n    char flag[16];\n    scanf(\"%15s\", flag);\n    srand(time(NULL));\n\n    if ((strcmp(flag, \"Flag{angr}\") == 0) && (rand() == 0x12345)) {\n        printf(\"Success\\n\");\n    } else {\n        printf(\"Failed\\n\");\n    }\n    return 0;\n}
\n

If you run it normally, the return value of rand() will almost never be exactly 0x12345, so the output is always Failed.

\n

In other words, unless the constraint “the return value of rand() is exactly 0x12345” is satisfied, angr cannot identify the correct Flag.

\n

In a case like this, you can recover the Flag by using a script like the following to hook rand and override its return value to 0x12345.

\n

In the example below, after setting the bitvector defined by the symbolic variable flag as standard input, I override the rand function with hook_symbol, which makes it easy to determine that the correct Flag is Flag{angr}.

\n
import angr\nimport claripy\nfrom logging import getLogger, WARN\n\ngetLogger(\"angr\").setLevel(WARN + 1)\n\nclass OverrideFunction(angr.SimProcedure):\n    def run(self, argc, argv):\n        data = (0x12345).to_bytes(4, byteorder='little')\n        data = int.from_bytes(data, byteorder='big')\n        return claripy.BVV(data, 32)\n\ndef correct(state):\n    if b\"Success\" in state.posix.dumps(1):\n        return True\n    return False\n\ndef failed(state):\n    if b\"Failed\" in state.posix.dumps(1):\n        return True\n    return False\n\nflag = claripy.BVS('flag', 16*8, explicit_name=True)\nproj = angr.Project(\"./chal.bin\", load_options={\"auto_load_libs\": False})\n\nstate = proj.factory.entry_state(stdin=flag)\nsimgr = proj.factory.simulation_manager(state)\nsimgr.explore(find=correct, avoid=failed)\n\nproj.hook_symbol(\"rand\", OverrideFunction())\n\ntry:\n    found = simgr.found[0]\n    # print(found.posix.dumps(0))\n    print(found.solver.eval(flag, cast_to=bytes))\nexcept IndexError:\n    print(\"Not Found\")
\n

Reference: Trying Various Things with angr [yoshi-camp notes] - Let’s Do CTF

\n

Here, the rand function is replaced with the behavior defined in the run method of the OverrideFunction class.

\n

When returning an integer, it seemed necessary to create a bitvector whose bytes are effectively in big-endian order, so the code below first converts the integer 0x12345 into bytes and then reads those bytes back as a big-endian integer before constructing the bitvector.

\n
data = (0x12345).to_bytes(4, byteorder='little')\ndata = int.from_bytes(data, byteorder='big')\nclaripy.BVV(data, 32)
\n

When you run this, the symbolic variable flag at the moment Success is ultimately printed is dumped, and you can identify the correct Flag.

\n

\n \n \n \n \n \n \n \n \n

\n

Replacing the Behavior of Statically Linked Library Functions

\n

It seems that angr bundles replacement implementations of common library functions such as those in libc.

\n

According to the angr documentation, angr uses these replacements so that even dynamically linked programs can be analyzed correctly.

\n

Reference: Extending the Environment Model - angr documentation

\n

Here, the blog post below mentions that when library functions are statically linked into a binary, angr analyzes those linked library functions too, which slows analysis down.

\n

In such cases, it seems possible to speed up analysis by binding library-function symbols to the replacement implementations bundled with angr, using the hook_symbol method as shown below.

\n
p.hook_symbol(\"__libc_start_main\", angr.SIM_PROCEDURES[\"glibc\"][\"__libc_start_main\"]())\np.hook_symbol(\"printf\", angr.procedures.libc.printf.printf())\np.hook_symbol(\"__isoc99_scanf\", angr.procedures.libc.scanf.scanf())\np.hook_symbol(\"strcmp\", angr.procedures.libc.strcmp.strcmp())\np.hook_symbol(\"puts\", angr.procedures.libc.puts.puts())
\n

Reference: Trying Various Things with angr [yoshi-camp notes] - Let’s Do CTF

\n

Using the hook Method in angr

\n

When using the SimProcedure described above, it seems that the entire function is hooked during analysis.

\n

On the other hand, if you use a user hook, you can hook a specific location in the code and do things such as rewriting registers.

\n

Reference: Hooks and SimProcedures - angr documentation

\n

Rewriting a Specific Register with the hook Method

\n

Now I will actually try angr’s hook method on a binary compiled from the following code.

\n
// gcc sample2.c -o chal.bin\n#include <stdio.h>\n#include <time.h>\n\nint func()\n{\n    return 0;\n}\n\nvoid main()\n{\n    int v = func();\n\n    char flag[16];\n    scanf(\"%15s\", flag);\n    \n    if ((strcmp(flag, \"Flag{angr}\") == 0) && (v == 1)) {\n        printf(\"Success\\n\");\n    } else {\n        printf(\"Failed\\n\");\n    }\n    return 0;\n}
\n

The program above checks whether the input string is Flag{angr}, but if the value of v is anything other than 1, Success will not be printed even when the correct Flag is entered.

\n

And the value of v is always 0 because of the func function.

\n

If you want to recover the Flag from a binary like this with angr, you could of course do it the same way as before by overriding func with the hook_symbol method.

\n

This time, however, I deliberately do not want to override the entire function, so I will try to recover the Flag by using the hook method to rewrite only the return-value register eax.

\n

First, use the objdmp command or similar in advance to check the address of the code that stores the return value of func into the variable v.

\n
11d8:       e8 cc ff ff ff          call   11a9 <func>\n11dd:       89 45 dc                mov    %eax,-0x24(%rbp)
\n

Next, recover the Flag with the following script by using the hook method to rewrite only the return-value register eax.

\n

The code below is similar to the previous example, but it overrides 5 bytes of processing starting at address 0x4011d8, where func is called, sets the value of the eax register to 1, and then resumes execution from 0x4011dd onward.

\n
import angr\nimport claripy\nfrom logging import getLogger, WARN\n\ngetLogger(\"angr\").setLevel(WARN + 1)\n\ndef correct(state):\n    if b\"Success\" in state.posix.dumps(1):\n        return True\n    return False\n\ndef failed(state):\n    if b\"Failed\" in state.posix.dumps(1):\n        return True\n    return False\n\nflag = claripy.BVS('flag', 16*8, explicit_name=True)\nproj = angr.Project(\"./chal.bin\", load_options={\"auto_load_libs\": False})\n\n@proj.hook(0x4011d8, length=5)\ndef set_eax(state):\n    state.regs.eax = 1\n\nstate = proj.factory.entry_state(stdin=flag)\nsimgr = proj.factory.simulation_manager(state)\nsimgr.explore(find=correct, avoid=failed)\n\ntry:\n    found = simgr.found[0]\n    # print(found.posix.dumps(0))\n    print(found.solver.eval(flag, cast_to=bytes))\nexcept IndexError:\n    print(\"Not Found\")
\n

By running this script, you can get past the v == 1 check, so angr is able to recover the correct Flag.

\n

\n \n \n \n \n \n \n \n \n

\n

Solving a CTF Challenge with hook_symbol

\n

Finally, I will solve an actual CTF challenge using angr’s hook_symbol method.

\n

The challenge I will solve with angr this time is SOP from Gracier CTF 2023.

\n

Reference: Gracier CTF 2023 Writeup SOP(Rev)

\n

The challenge itself is themed around SOP (probably Signal Oriented Programming or Sigreturn Oriented Programming). It begins by raising an exception at the end of the main function and then repeatedly calls handler functions defined with signal through the raise function.

\n

Because the program’s actual behavior (checking whether the input is the correct Flag) starts from the exception raised after main ends, normal dynamic analysis is not possible even with gdb, so this was a challenge where you needed to identify the Flag using a Frida hook or static analysis.

\n

This time, as an alternative solution, I will show how to recover the correct Flag by using angr’s hook_symbol to override the behavior of signal and raise, ignoring the binary’s actual behavior.

\n

How to Recover the SOP Flag with angr

\n

This binary is a program that checks whether the input string matches the correct Flag.

\n

For that reason, it is the kind of challenge that would normally be relatively easy to solve with angr’s SimulationManager.

\n

However, the Flag-verification logic in this binary starts from an exception after main finishes and is implemented by following the handler functions defined with signal, so you cannot solve it just by running SimulationManager as-is.

\n

Therefore, after analyzing the binary with Ghidra to identify the relationship between the signal numbers passed as arguments to raise and the handler functions defined in the binary, I use the hook_symbol method to override the program’s behavior so that SimulationManager can identify the Flag.

\n

This eliminates the need for angr to actually analyze the behavior of signal and raise, making it easy to identify the correct Flag.

\n

Analyzing the Binary

\n

To create a solver with angr, I first use Ghidra to identify the information needed to recover the Flag.

\n

First, in the main function, you can see that the read function takes 0x44 bytes of input from standard input, and the function returns the result of comparing DAT_001061c8 != 0x44.

\n
bool main(void)\n{\n  size_t sVar1;\n  ssize_t sVar2;\n  \n  sVar1 = strlen(&DAT_001060f0);\n  DAT_001061c8 = (int)sVar1;\n  if (DAT_001061c8 == 0) {\n    sVar2 = read(0,&DAT_001060f0,0x44);\n    DAT_001061c8 = (int)sVar2;\n  }\n  return DAT_001061c8 != 0x44;\n}
\n

This main function is called from __libc_start_main, and the return value of main called by __libc_start_main is passed directly to the exit function.

\n

Reference: _libcstart_main

\n

In other words, when the result of the comparison DAT_001061c8 != 0x44 becomes False, the program is regarded as having terminated abnormally, and the initial exception is triggered.

\n

Next, identify the relationship between signal numbers and handler functions.

\n

Looking through the decompilation results of the functions in Ghidra, you can confirm that the following signal definitions are present.

\n

For some reason there are multiple places that set a handler function for SIGSEGV (0xb), which suggests that the handler function is changed dynamically here.

\n
signal(0xb,FUN_001011e0);\nsignal(0xb,FUN_00101bc0);\nsignal(0xb,FUN_00102e60);\nsignal(0xe,FUN_00101bc0);\nsignal(0x10,FUN_00101350);\nsignal(0x11,FUN_001014a0);\nsignal(0x12,FUN_00102fb0);\nsignal(0x15,FUN_00101550);\nsignal(0x16,FUN_00102520);
\n

Next, identify the result when the correct Flag or an incorrect Flag is entered.

\n

This is easy to determine because if you just run the program with arbitrary input, it prints the string FAIL.

\n

\n \n \n \n \n \n \n \n \n

\n

As shown above, the output is SUCCESS when the Flag is correct and FAIL when the Flag is incorrect.

\n

Now let’s look more closely at the code where the Flag is verified.

\n
if (0x40 < DAT_001061c8) {\n    DAT_001061c8 = DAT_001061c8 - 0x40;\n    DAT_00106210 = DAT_00106210 + 0x40;\n    DAT_001061c0 = DAT_001061c0 + 0x40;\n    raise(0xb);\n    return;\n}\nif (DAT_001061c8 < 0x40) {\n    for (DAT_001061cc = 0; DAT_001061cc < DAT_001061c8; DAT_001061cc = DAT_001061cc + 1) {\n      *(undefined *)(DAT_00106138 + (ulong)DAT_001061cc) = DAT_00106210[DAT_001061cc];\n    }\n}\nDAT_001060d0 = DAT_001061a4;\nDAT_001060d4 = DAT_001061ac;\nmemcpy(local_58,&DAT_00104050,0x44);\nmemset(local_168,0,0x110);\nlocal_16c = 0;\ndo {\n    do {\n      if (local_16c == 0x44) {\n        printf(\"SUCCESS\\n\");\n                    /* WARNING: Subroutine does not return */\n        exit(0);\n      }\n      local_170 = 0;\n      getrandom(&local_170,4,2);\n      local_170 = local_170 % 0x44;\n    } while (local_168[local_170] == 1);\n    local_168[local_170] = 1;\n    local_16c = local_16c + 1;\n} while ((&DAT_00106220)[local_170] == local_58[local_170]);\n\nprintf(\"FAIL\\n\");
\n

Ignore the first half and focus on the part below.

\n

Here, after taking the random value obtained by the getrandom function modulo 0x44, the result is used as an index to check whether some transformed value derived from the input matches a hard-coded byte sequence used for verification.

\n
do {\ndo {\n**\n      local_170 = 0;\n      getrandom(&local_170,4,2);\n      local_170 = local_170 % 0x44;\n    } while (local_168[local_170] == 1);\n\nlocal_168[local_170] = 1;\nlocal_16c = local_16c + 1;\n\n**\n\n} while ((&DAT_00106220)[local_170] == local_58[local_170]);
\n

This is probably part of the dynamic-analysis countermeasures, and it also becomes a factor that interferes with analysis in angr.

\n

Finally, use strace -e trace=signal to trace how the callback functions behave when you actually give the program a string of 0x44 bytes.

\n
$ echo AAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAAA | strace -e trace=signal ./app\nrt_sigaction(SIGSEGV, {sa_handler=0x55a8e91ed1e0, sa_mask=[SEGV], sa_flags=SA_RESTORER|SA_RESTART, sa_restorer=0x7f047a881520}, {sa_handler=SIG_DFL, sa_mask=[], sa_flags=0}, 8) = 0\nrt_sigaction(SIGSTKFLT, {sa_handler=0x55a8e91ed350, sa_mask=[STKFLT], sa_flags=SA_RESTORER|SA_RESTART, sa_restorer=0x7f047a881520}, {sa_handler=SIG_DFL, sa_mask=[], sa_flags=0}, 8) = 0\nrt_sigaction(SIGCHLD, {sa_handler=0x55a8e91ed4a0, sa_mask=[CHLD], sa_flags=SA_RESTORER|SA_RESTART, sa_restorer=0x7f047a881520}, {sa_handler=SIG_DFL, sa_mask=[], sa_flags=0}, 8) = 0\nrt_sigaction(SIGCONT, {sa_handler=0x55a8e91eefb0, sa_mask=[CONT], sa_flags=SA_RESTORER|SA_RESTART, sa_restorer=0x7f047a881520}, {sa_handler=SIG_DFL, sa_mask=[], sa_flags=0}, 8) = 0\n--- SIGSEGV {si_signo=SIGSEGV, si_code=SEGV_MAPERR, si_addr=0xe91ef160} ---\ntgkill(29240, 29240, SIGSTKFLT)         = 0\n--- SIGSTKFLT {si_signo=SIGSTKFLT, si_code=SI_TKILL, si_pid=29240, si_uid=1000} ---\ntgkill(29240, 29240, SIGCHLD)           = 0\n--- SIGCHLD {si_signo=SIGCHLD, si_code=SI_TKILL, si_pid=29240, si_uid=1000} ---\ntgkill(29240, 29240, SIGCONT)           = 0\n--- SIGCONT {si_signo=SIGCONT, si_code=SI_TKILL, si_pid=29240, si_uid=1000} ---\nrt_sigaction(SIGSEGV, {sa_handler=0x55a8e91eee60, sa_mask=[SEGV], sa_flags=SA_RESTORER|SA_RESTART, sa_restorer=0x7f047a881520}, {sa_handler=0x55a8e91ed1e0, sa_mask=[SEGV], sa_flags=SA_RESTORER|SA_RESTART, sa_restorer=0x7f047a881520}, 8) = 0\ntgkill(29240, 29240, SIGSEGV)           = 0\nrt_sigreturn({mask=[SEGV STKFLT CHLD]}) = 0\nrt_sigreturn({mask=[SEGV STKFLT]})      = 0\nrt_sigreturn({mask=[SEGV]})             = 0\nrt_sigreturn({mask=[]})                 = 0\n--- SIGSEGV {si_signo=SIGSEGV, si_code=SI_TKILL, si_pid=29240, si_uid=1000} ---\nrt_sigaction(SIGTTOU, {sa_handler=0x55a8e91ee520, sa_mask=[TTOU], sa_flags=SA_RESTORER|SA_RESTART, sa_restorer=0x7f047a881520}, {sa_handler=SIG_DFL, sa_mask=[], sa_flags=0}, 8) = 0\nrt_sigaction(SIGALRM, {sa_handler=0x55a8e91edbc0, sa_mask=[ALRM], sa_flags=SA_RESTORER|SA_RESTART, sa_restorer=0x7f047a881520}, {sa_handler=SIG_DFL, sa_mask=[], sa_flags=0}, 8) = 0\nrt_sigaction(SIGTTIN, {sa_handler=0x55a8e91ed550, sa_mask=[TTIN], sa_flags=SA_RESTORER|SA_RESTART, sa_restorer=0x7f047a881520}, {sa_handler=SIG_DFL, sa_mask=[], sa_flags=0}, 8) = 0\n--- SIGALRM {si_signo=SIGALRM, si_code=SI_KERNEL} ---\nrt_sigaction(SIGSEGV, {sa_handler=0x55a8e91edbc0, sa_mask=[SEGV], sa_flags=SA_RESTORER|SA_RESTART, sa_restorer=0x7f047a881520}, {sa_handler=0x55a8e91eee60, sa_mask=[SEGV], sa_flags=SA_RESTORER|SA_RESTART, sa_restorer=0x7f047a881520}, 8) = 0\ntgkill(29240, 29240, SIGTTIN)           = 0\n--- SIGTTIN {si_signo=SIGTTIN, si_code=SI_TKILL, si_pid=29240, si_uid=1000} ---\ntgkill(29240, 29240, SIGTTOU)           = 0\n--- SIGTTOU {si_signo=SIGTTOU, si_code=SI_TKILL, si_pid=29240, si_uid=1000} ---\ntgkill(29240, 29240, SIGSEGV)           = 0\nrt_sigreturn({mask=[SEGV ALRM TTIN]})   = 0\nrt_sigreturn({mask=[SEGV ALRM]})        = 0\nrt_sigreturn({mask=[SEGV]})             = -1 EINTR (Interrupted system call)\nrt_sigreturn({mask=[]})                 = 0\n--- SIGSEGV {si_signo=SIGSEGV, si_code=SI_TKILL, si_pid=29240, si_uid=1000} ---\nrt_sigaction(SIGSEGV, {sa_handler=0x55a8e91edbc0, sa_mask=[SEGV], sa_flags=SA_RESTORER|SA_RESTART, sa_restorer=0x7f047a881520}, {sa_handler=0x55a8e91edbc0, sa_mask=[SEGV], sa_flags=SA_RESTORER|SA_RESTART, sa_restorer=0x7f047a881520}, 8) = 0\ntgkill(29240, 29240, SIGTTIN)           = 0\n--- SIGTTIN {si_signo=SIGTTIN, si_code=SI_TKILL, si_pid=29240, si_uid=1000} ---\ntgkill(29240, 29240, SIGTTOU)           = 0\n--- SIGTTOU {si_signo=SIGTTOU, si_code=SI_TKILL, si_pid=29240, si_uid=1000} ---\nFAIL\n+++ exited with 1 +++
\n

From this result, you can see that SIGSEGV (0xb) is called first when the main function finishes, and the RVA at that time is 0x11e0.

\n

After that, processing continues through SIGSTKFLT (0x10), SIGCHLD (0x11), and SIGCONT (0x12).

\n

The signals called throughout the whole process were the following.

\n

This matches the signals for which handler functions were defined in the signal calls checked earlier. (Only the handler function for SIGSEGV is reconfigured dynamically.)

\n
SIGSEGV(0xb)\nSIGALRM(0xe)\nSIGSTKFLT(0x10)\nSIGCHLD(0x11)\nSIGCONT(0x12)\nSIGTTIN(0x15)\nSIGTTOU(0x16)
\n

One thing worth noticing here is that SIGALRM, which is not triggered by raise, is still being called.

\n

It is a slightly tricky implementation, but by placing sleep(2) immediately after alarm(1), it seems to force SIGALRM to occur at that timing.

\n
signal(0x16,FUN_00102520);\nsignal(0xe,FUN_00101bc0);\nsignal(0x15,FUN_00101550);\nalarm(1);\nsleep(2);
\n

In analysis that uses symbolic execution like angr, it may be difficult to track time-based state changes such as alarm and sleep accurately.

\n

For that reason, on the solver side we need to override the sleep function and replace it with processing that forcibly calls the SIGALRM callback function.

\n

Building a Solver with hook_symbol

\n

Now that we have gathered all the necessary information, it is finally time to build a solver to obtain the Flag.

\n

This challenge can be solved with the following solver.

\n
import angr\nimport claripy\nfrom logging import getLogger, WARN\n\ngetLogger(\"angr\").setLevel(WARN + 1)\n\nproj = angr.Project(\"app\")\nflag = claripy.BVS(\"flag\", 0x44*8)\nstate = proj.factory.entry_state(stdin=flag)\n\nfor i in range(0x44):\n    state.solver.add(flag.get_byte(i) >= 0x21)\n    state.solver.add(flag.get_byte(i) <= 0x7f)\n\ndef correct(state):\n    if b\"SUCCESS\" in state.posix.dumps(1):\n        return True\n    return False\n\ndef failed(state):\n    if b\"FAIL\" in state.posix.dumps(1):\n        return True\n    return False\n\n\nclass OverrideSignal(angr.SimProcedure):\ndef run(self, sigid, handler):\nsigid = sigid.concrete_value\nhandler = handler.concrete_value\nself.state.globals[\"handlers\"] = self.state.globals[\"handlers\"].copy()\nself.state.globals[\"handlers\"][sigid] = handler\nreturn 0\n\nclass OverrideRaise(angr.SimProcedure):\ndef run(self, sigid):\nsigid = sigid.concrete_value\nself.call(self.state.globals[\"handlers\"][sigid], (sigid,), 0xFFFFFFFF)\n\nclass OverrideSleep(angr.SimProcedure):\ndef run(self, sigid):\nsigid = 14\nself.call(self.state.globals[\"handlers\"][sigid], (sigid,), 0xFFFFFFFF)\n\nclass OverrideGetRandom(angr.SimProcedure):\ndef run(self, val):\nres = self.state.globals[\"i\"]\nif res == 0x44:\nprint(self.state.posix.dumps(0))\ninput(\"\")\nself.state.globals[\"i\"] += 1\nself.state.mem[val].int = res\nreturn 0\n\n@proj.hook(0x4031F2)\ndef first_sigsegv(state):\nstate.regs.rsp -= 8\nstate.regs.rip = state.globals[\"handlers\"][11]\nstate.regs.rdi = 11\n\nstate.globals[\"handlers\"] = {\n11: 0x4011E0,\n14: 0x401bc0,\n16: 0x401350,\n17: 0x4014A0,\n18: 0x402FB0,\n21: 0x401550,\n22: 0x402520\n}\n\nstate.globals[\"i\"] = 0\nproj.hook_symbol(\"signal\", OverrideSignal(), replace=True)\nproj.hook_symbol(\"raise\", OverrideRaise(), replace=True)\nproj.hook_symbol(\"sleep\", OverrideSleep(), replace=True)\nproj.hook_symbol(\"getrandom\", OverrideGetRandom(), replace=True)\nsimgr = proj.factory.simulation_manager(state)\n\nwhile simgr.active:\n    simgr.explore(find=correct, avoid=failed)\n    if simgr.found:\n        print(simgr.found[0].solver.eval(flag, cast_to=bytes))\n        break
\n

The implementation looks complicated at first glance, but it is made up of the same things used up to this point.

\n

First, in the opening section of code below, as usual I define the Project, declare the symbolic variable flag, and define the expected output results for the cases where the input is correct and incorrect.

\n
import angr\nimport claripy\nfrom logging import getLogger, WARN\n\ngetLogger(\"angr\").setLevel(WARN + 1)\n\nproj = angr.Project(\"app\")\nflag = claripy.BVS(\"flag\", 0x44*8)\nstate = proj.factory.entry_state(stdin=flag)\n\nfor i in range(0x44):\n    state.solver.add(flag.get_byte(i) >= 0x21)\n    state.solver.add(flag.get_byte(i) <= 0x7f)\n\ndef correct(state):\n    if b\"SUCCESS\" in state.posix.dumps(1):\n        return True\n    return False\n\ndef failed(state):\n    if b\"FAIL\" in state.posix.dumps(1):\n        return True\n    return False
\n

In the next part, I define classes for the four functions that will be overridden with SimProcedure.

\n

This time, I override the signal, raise, sleep, and getrandom functions.

\n

Overriding signal would be unnecessary if all handler functions were fixed, but because the callback function for SIGSEGV is implemented so that it changes dynamically, I override it as well and add the handler function to the dictionary called handlers, which defines the mapping between handler functions and signal numbers.

\n

Also, the randomization of indices by getrandom, which was likely implemented as a dynamic-analysis countermeasure, is overridden so that it returns values incremented by 1 from the beginning, changing the comparison to check the correct Flag from the start in order.

\n
class OverrideSignal(angr.SimProcedure):\ndef run(self, sigid, handler):\nsigid = sigid.concrete_value\nhandler = handler.concrete_value\nself.state.globals[\"handlers\"] = self.state.globals[\"handlers\"].copy()\nself.state.globals[\"handlers\"][sigid] = handler\nreturn 0\n\nclass OverrideRaise(angr.SimProcedure):\ndef run(self, sigid):\nsigid = sigid.concrete_value\nself.call(self.state.globals[\"handlers\"][sigid], (sigid,), 0xFFFFFFFF)\n\nclass OverrideSleep(angr.SimProcedure):\ndef run(self, sigid):\nsigid = 14\nself.call(self.state.globals[\"handlers\"][sigid], (sigid,), 0xFFFFFFFF)\n\nclass OverrideGetRandom(angr.SimProcedure):\ndef run(self, val):\nres = self.state.globals[\"i\"]\nif res == 0x44:\nprint(self.state.posix.dumps(0))\ninput(\"\")\nself.state.globals[\"i\"] += 1\nself.state.mem[val].int = res\nreturn 0
\n

The section below uses a user hook to override the ret of the main function so that SIGSEGV can be caught.

\n
@proj.hook(0x4031F2)\ndef first_sigsegv(state):\nstate.regs.rsp -= 8\nstate.regs.rip = state.globals[\"handlers\"][11]\nstate.regs.rdi = 11
\n

\n \n