Autorefresh Dev Documentation

Author: github-actions[bot]

dev/_modules/libdebug/architectures/aarch64/aarch64_stack_unwinder.html

@@ -435,8 +435,10 @@ <h1>Source code for libdebug.architectures.aarch64.aarch64_stack_unwinder</h1><d
 <span class="kn">from</span> <span class="nn">typing</span> <span class="kn">import</span> <span class="n">TYPE_CHECKING</span>
 
 <span class="kn">from</span> <span class="nn">libdebug.architectures.stack_unwinding_manager</span> <span class="kn">import</span> <span class="n">StackUnwindingManager</span>
+<span class="kn">from</span> <span class="nn">libdebug.liblog</span> <span class="kn">import</span> <span class="n">liblog</span>
 
 <span class="k">if</span> <span class="n">TYPE_CHECKING</span><span class="p">:</span>
+    <span class="kn">from</span> <span class="nn">libdebug.data.memory_map</span> <span class="kn">import</span> <span class="n">MemoryMap</span>
     <span class="kn">from</span> <span class="nn">libdebug.state.thread_context</span> <span class="kn">import</span> <span class="n">ThreadContext</span>
 
 
@@ -459,10 +461,18 @@ <h1>Source code for libdebug.architectures.aarch64.aarch64_stack_unwinder</h1><d
         <span class="k">assert</span> <span class="nb">hasattr</span><span class="p">(</span><span class="n">target</span><span class="o">.</span><span class="n">regs</span><span class="p">,</span> <span class="s2">&quot;pc&quot;</span><span class="p">)</span>
 
         <span class="n">frame_pointer</span> <span class="o">=</span> <span class="n">target</span><span class="o">.</span><span class="n">regs</span><span class="o">.</span><span class="n">x29</span>
-        <span class="n">initial_link_register</span> <span class="o">=</span> <span class="n">target</span><span class="o">.</span><span class="n">regs</span><span class="o">.</span><span class="n">x30</span>
-        <span class="n">stack_trace</span> <span class="o">=</span> <span class="p">[</span><span class="n">target</span><span class="o">.</span><span class="n">regs</span><span class="o">.</span><span class="n">pc</span><span class="p">,</span> <span class="n">initial_link_register</span><span class="p">]</span>
 
         <span class="n">vmaps</span> <span class="o">=</span> <span class="n">target</span><span class="o">.</span><span class="n">_internal_debugger</span><span class="o">.</span><span class="n">debugging_interface</span><span class="o">.</span><span class="n">maps</span><span class="p">()</span>
+        <span class="n">initial_link_register</span> <span class="o">=</span> <span class="kc">None</span>
+
+        <span class="k">try</span><span class="p">:</span>
+            <span class="n">initial_link_register</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_return_address</span><span class="p">(</span><span class="n">target</span><span class="p">,</span> <span class="n">vmaps</span><span class="p">)</span>
+        <span class="k">except</span> <span class="ne">ValueError</span><span class="p">:</span>
+            <span class="n">liblog</span><span class="o">.</span><span class="n">warning</span><span class="p">(</span>
+                <span class="s2">&quot;Failed to get the return address. Check stack frame registers (e.g., base pointer). The stack trace may be incomplete.&quot;</span><span class="p">,</span>
+            <span class="p">)</span>
+
+        <span class="n">stack_trace</span> <span class="o">=</span> <span class="p">[</span><span class="n">target</span><span class="o">.</span><span class="n">regs</span><span class="o">.</span><span class="n">pc</span><span class="p">,</span> <span class="n">initial_link_register</span><span class="p">]</span> <span class="k">if</span> <span class="n">initial_link_register</span> <span class="k">else</span> <span class="p">[</span><span class="n">target</span><span class="o">.</span><span class="n">regs</span><span class="o">.</span><span class="n">pc</span><span class="p">]</span>
 
         <span class="c1"># Follow the frame chain</span>
         <span class="k">while</span> <span class="n">frame_pointer</span><span class="p">:</span>
@@ -489,16 +499,22 @@ <h1>Source code for libdebug.architectures.aarch64.aarch64_stack_unwinder</h1><d
 
 <div class="viewcode-block" id="Aarch64StackUnwinder.get_return_address">
 <a class="viewcode-back" href="../../../../libdebug.architectures.aarch64.html#libdebug.architectures.aarch64.aarch64_stack_unwinder.Aarch64StackUnwinder.get_return_address">[docs]</a>
-    <span class="k">def</span> <span class="nf">get_return_address</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Aarch64StackUnwinder</span><span class="p">,</span> <span class="n">target</span><span class="p">:</span> <span class="n">ThreadContext</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">int</span><span class="p">:</span>
+    <span class="k">def</span> <span class="nf">get_return_address</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Aarch64StackUnwinder</span><span class="p">,</span> <span class="n">target</span><span class="p">:</span> <span class="n">ThreadContext</span><span class="p">,</span> <span class="n">vmaps</span><span class="p">:</span> <span class="nb">list</span><span class="p">[</span><span class="n">MemoryMap</span><span class="p">])</span> <span class="o">-&gt;</span> <span class="nb">int</span><span class="p">:</span>
 <span class="w">        </span><span class="sd">&quot;&quot;&quot;Get the return address of the current function.</span>
 
 <span class="sd">        Args:</span>
 <span class="sd">            target (ThreadContext): The target ThreadContext.</span>
+<span class="sd">            vmaps (list[MemoryMap]): The memory maps of the process.</span>
 
 <span class="sd">        Returns:</span>
 <span class="sd">            int: The return address.</span>
 <span class="sd">        &quot;&quot;&quot;</span>
-        <span class="k">return</span> <span class="n">target</span><span class="o">.</span><span class="n">regs</span><span class="o">.</span><span class="n">x30</span></div>
+        <span class="n">return_address</span> <span class="o">=</span> <span class="n">target</span><span class="o">.</span><span class="n">regs</span><span class="o">.</span><span class="n">x30</span>
+
+        <span class="k">if</span> <span class="ow">not</span> <span class="nb">any</span><span class="p">(</span><span class="n">vmap</span><span class="o">.</span><span class="n">start</span> <span class="o">&lt;=</span> <span class="n">return_address</span> <span class="o">&lt;</span> <span class="n">vmap</span><span class="o">.</span><span class="n">end</span> <span class="k">for</span> <span class="n">vmap</span> <span class="ow">in</span> <span class="n">vmaps</span><span class="p">):</span>
+            <span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span><span class="s2">&quot;Return address not in any valid memory map&quot;</span><span class="p">)</span>
+
+        <span class="k">return</span> <span class="n">return_address</span></div>
 </div>
 
 </pre></div>

dev/_modules/libdebug/architectures/amd64/amd64_stack_unwinder.html

@@ -435,13 +435,13 @@ <h1>Source code for libdebug.architectures.amd64.amd64_stack_unwinder</h1><div c
 <span class="kn">from</span> <span class="nn">typing</span> <span class="kn">import</span> <span class="n">TYPE_CHECKING</span>
 
 <span class="kn">from</span> <span class="nn">libdebug.architectures.stack_unwinding_manager</span> <span class="kn">import</span> <span class="n">StackUnwindingManager</span>
-<span class="kn">from</span> <span class="nn">libdebug.liblog</span> <span class="kn">import</span> <span class="n">logging</span>
+<span class="kn">from</span> <span class="nn">libdebug.liblog</span> <span class="kn">import</span> <span class="n">liblog</span>
 
 <span class="k">if</span> <span class="n">TYPE_CHECKING</span><span class="p">:</span>
+    <span class="kn">from</span> <span class="nn">libdebug.data.memory_map</span> <span class="kn">import</span> <span class="n">MemoryMap</span>
     <span class="kn">from</span> <span class="nn">libdebug.state.thread_context</span> <span class="kn">import</span> <span class="n">ThreadContext</span>
 
 
-
 <div class="viewcode-block" id="Amd64StackUnwinder">
 <a class="viewcode-back" href="../../../../libdebug.architectures.amd64.html#libdebug.architectures.amd64.amd64_stack_unwinder.Amd64StackUnwinder">[docs]</a>
 <span class="k">class</span> <span class="nc">Amd64StackUnwinder</span><span class="p">(</span><span class="n">StackUnwindingManager</span><span class="p">):</span>
@@ -484,25 +484,29 @@ <h1>Source code for libdebug.architectures.amd64.amd64_stack_unwinder</h1><div c
         <span class="c1"># If we are in the prolouge of a function, we need to get the return address from the stack</span>
         <span class="c1"># using a slightly more complex method</span>
         <span class="k">try</span><span class="p">:</span>
-            <span class="n">first_return_address</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_return_address</span><span class="p">(</span><span class="n">target</span><span class="p">)</span>
+            <span class="n">first_return_address</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">get_return_address</span><span class="p">(</span><span class="n">target</span><span class="p">,</span> <span class="n">vmaps</span><span class="p">)</span>
 
-            <span class="k">if</span> <span class="n">first_return_address</span> <span class="o">!=</span> <span class="n">stack_trace</span><span class="p">[</span><span class="mi">1</span><span class="p">]:</span>
-                <span class="n">stack_trace</span><span class="o">.</span><span class="n">insert</span><span class="p">(</span><span class="mi">1</span><span class="p">,</span> <span class="n">first_return_address</span><span class="p">)</span>
+            <span class="k">if</span> <span class="nb">len</span><span class="p">(</span><span class="n">stack_trace</span><span class="p">)</span> <span class="o">&gt;</span> <span class="mi">1</span><span class="p">:</span>
+                <span class="k">if</span> <span class="n">first_return_address</span> <span class="o">!=</span> <span class="n">stack_trace</span><span class="p">[</span><span class="mi">1</span><span class="p">]:</span>
+                    <span class="n">stack_trace</span><span class="o">.</span><span class="n">insert</span><span class="p">(</span><span class="mi">1</span><span class="p">,</span> <span class="n">first_return_address</span><span class="p">)</span>
+            <span class="k">else</span><span class="p">:</span>
+                <span class="n">stack_trace</span><span class="o">.</span><span class="n">append</span><span class="p">(</span><span class="n">first_return_address</span><span class="p">)</span>
         <span class="k">except</span> <span class="p">(</span><span class="ne">OSError</span><span class="p">,</span> <span class="ne">ValueError</span><span class="p">):</span>
-            <span class="n">logging</span><span class="o">.</span><span class="n">WARNING</span><span class="p">(</span>
-                <span class="s2">&quot;Failed to get the return address from the stack. Check stack frame registers (e.g., base pointer). The stack trace may be incomplete.&quot;</span><span class="p">,</span>
+            <span class="n">liblog</span><span class="o">.</span><span class="n">warning</span><span class="p">(</span>
+                <span class="s2">&quot;Failed to get the return address. Check stack frame registers (e.g., base pointer). The stack trace may be incomplete.&quot;</span><span class="p">,</span>
             <span class="p">)</span>
 
         <span class="k">return</span> <span class="n">stack_trace</span></div>
 
 
 <div class="viewcode-block" id="Amd64StackUnwinder.get_return_address">
 <a class="viewcode-back" href="../../../../libdebug.architectures.amd64.html#libdebug.architectures.amd64.amd64_stack_unwinder.Amd64StackUnwinder.get_return_address">[docs]</a>
-    <span class="k">def</span> <span class="nf">get_return_address</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Amd64StackUnwinder</span><span class="p">,</span> <span class="n">target</span><span class="p">:</span> <span class="n">ThreadContext</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">int</span><span class="p">:</span>
+    <span class="k">def</span> <span class="nf">get_return_address</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Amd64StackUnwinder</span><span class="p">,</span> <span class="n">target</span><span class="p">:</span> <span class="n">ThreadContext</span><span class="p">,</span> <span class="n">vmaps</span><span class="p">:</span> <span class="nb">list</span><span class="p">[</span><span class="n">MemoryMap</span><span class="p">])</span> <span class="o">-&gt;</span> <span class="nb">int</span><span class="p">:</span>
 <span class="w">        </span><span class="sd">&quot;&quot;&quot;Get the return address of the current function.</span>
 
 <span class="sd">        Args:</span>
 <span class="sd">            target (ThreadContext): The target ThreadContext.</span>
+<span class="sd">            vmaps (list[MemoryMap]): The memory maps of the process.</span>
 
 <span class="sd">        Returns:</span>
 <span class="sd">            int: The return address.</span>
@@ -519,7 +523,12 @@ <h1>Source code for libdebug.architectures.amd64.amd64_stack_unwinder</h1><div c
         <span class="k">else</span><span class="p">:</span>
             <span class="n">return_address</span> <span class="o">=</span> <span class="n">target</span><span class="o">.</span><span class="n">memory</span><span class="p">[</span><span class="n">target</span><span class="o">.</span><span class="n">regs</span><span class="o">.</span><span class="n">rsp</span> <span class="o">+</span> <span class="mi">8</span><span class="p">,</span> <span class="mi">8</span><span class="p">,</span> <span class="s2">&quot;absolute&quot;</span><span class="p">]</span>
 
-        <span class="k">return</span> <span class="nb">int</span><span class="o">.</span><span class="n">from_bytes</span><span class="p">(</span><span class="n">return_address</span><span class="p">,</span> <span class="n">byteorder</span><span class="o">=</span><span class="s2">&quot;little&quot;</span><span class="p">)</span></div>
+        <span class="n">return_address</span> <span class="o">=</span> <span class="nb">int</span><span class="o">.</span><span class="n">from_bytes</span><span class="p">(</span><span class="n">return_address</span><span class="p">,</span> <span class="n">byteorder</span><span class="o">=</span><span class="s2">&quot;little&quot;</span><span class="p">)</span>
+
+        <span class="k">if</span> <span class="ow">not</span> <span class="nb">any</span><span class="p">(</span><span class="n">vmap</span><span class="o">.</span><span class="n">start</span> <span class="o">&lt;=</span> <span class="n">return_address</span> <span class="o">&lt;</span> <span class="n">vmap</span><span class="o">.</span><span class="n">end</span> <span class="k">for</span> <span class="n">vmap</span> <span class="ow">in</span> <span class="n">vmaps</span><span class="p">):</span>
+            <span class="k">raise</span> <span class="ne">ValueError</span><span class="p">(</span><span class="s2">&quot;Return address not in any valid memory map&quot;</span><span class="p">)</span>
+
+        <span class="k">return</span> <span class="n">return_address</span></div>
 
 
     <span class="k">def</span> <span class="nf">_preamble_state</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">Amd64StackUnwinder</span><span class="p">,</span> <span class="n">instruction_window</span><span class="p">:</span> <span class="nb">bytes</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">int</span><span class="p">:</span>

dev/_modules/libdebug/architectures/stack_unwinding_manager.html

@@ -436,6 +436,7 @@ <h1>Source code for libdebug.architectures.stack_unwinding_manager</h1><div clas
 <span class="kn">from</span> <span class="nn">typing</span> <span class="kn">import</span> <span class="n">TYPE_CHECKING</span>
 
 <span class="k">if</span> <span class="n">TYPE_CHECKING</span><span class="p">:</span>
+    <span class="kn">from</span> <span class="nn">libdebug.data.memory_map</span> <span class="kn">import</span> <span class="n">MemoryMap</span>
     <span class="kn">from</span> <span class="nn">libdebug.state.thread_context</span> <span class="kn">import</span> <span class="n">ThreadContext</span>
 
 
@@ -454,7 +455,7 @@ <h1>Source code for libdebug.architectures.stack_unwinding_manager</h1><div clas
 <div class="viewcode-block" id="StackUnwindingManager.get_return_address">
 <a class="viewcode-back" href="../../../libdebug.architectures.html#libdebug.architectures.stack_unwinding_manager.StackUnwindingManager.get_return_address">[docs]</a>
     <span class="nd">@abstractmethod</span>
-    <span class="k">def</span> <span class="nf">get_return_address</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">StackUnwindingManager</span><span class="p">,</span> <span class="n">target</span><span class="p">:</span> <span class="n">ThreadContext</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">int</span><span class="p">:</span>
+    <span class="k">def</span> <span class="nf">get_return_address</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">StackUnwindingManager</span><span class="p">,</span> <span class="n">target</span><span class="p">:</span> <span class="n">ThreadContext</span><span class="p">,</span> <span class="n">vmaps</span><span class="p">:</span> <span class="nb">list</span><span class="p">[</span><span class="n">MemoryMap</span><span class="p">])</span> <span class="o">-&gt;</span> <span class="nb">int</span><span class="p">:</span>
 <span class="w">        </span><span class="sd">&quot;&quot;&quot;Get the return address of the current function.&quot;&quot;&quot;</span></div>
 </div>
 

dev/_modules/libdebug/ptrace/ptrace_interface.html

@@ -764,7 +764,7 @@ <h1>Source code for libdebug.ptrace.ptrace_interface</h1><div class="highlight">
             <span class="n">invalidate_process_cache</span><span class="p">()</span>
         <span class="k">elif</span> <span class="n">heuristic</span> <span class="o">==</span> <span class="s2">&quot;backtrace&quot;</span><span class="p">:</span>
             <span class="c1"># Breakpoint to return address</span>
-            <span class="n">last_saved_instruction_pointer</span> <span class="o">=</span> <span class="n">thread</span><span class="o">.</span><span class="n">current_return_address</span><span class="p">()</span>
+            <span class="n">last_saved_instruction_pointer</span> <span class="o">=</span> <span class="n">thread</span><span class="o">.</span><span class="n">saved_ip</span>
 
             <span class="c1"># If a breakpoint already exists at the return address, we don&#39;t need to set a new one</span>
             <span class="n">found</span> <span class="o">=</span> <span class="kc">False</span>