Autorefresh Dev Documentation

Author: github-actions[bot]

dev/_modules/libdebug/ptrace/ptrace_interface.html

@@ -439,8 +439,8 @@ <h1>Source code for libdebug.ptrace.ptrace_interface</h1><div class="highlight">
 <span class="kn">from</span> <span class="nn">pathlib</span> <span class="kn">import</span> <span class="n">Path</span>
 <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.register_helper</span> <span class="kn">import</span> <span class="n">register_holder_provider</span>
 <span class="kn">from</span> <span class="nn">libdebug.architectures.call_utilities_provider</span> <span class="kn">import</span> <span class="n">call_utilities_provider</span>
+<span class="kn">from</span> <span class="nn">libdebug.architectures.register_helper</span> <span class="kn">import</span> <span class="n">register_holder_provider</span>
 <span class="kn">from</span> <span class="nn">libdebug.cffi</span> <span class="kn">import</span> <span class="n">_ptrace_cffi</span>
 <span class="kn">from</span> <span class="nn">libdebug.data.breakpoint</span> <span class="kn">import</span> <span class="n">Breakpoint</span>
 <span class="kn">from</span> <span class="nn">libdebug.debugger.internal_debugger_instance_manager</span> <span class="kn">import</span> <span class="p">(</span>
@@ -457,6 +457,7 @@ <h1>Source code for libdebug.ptrace.ptrace_interface</h1><div class="highlight">
 <span class="kn">from</span> <span class="nn">libdebug.utils.process_utils</span> <span class="kn">import</span> <span class="p">(</span>
     <span class="n">disable_self_aslr</span><span class="p">,</span>
     <span class="n">get_process_maps</span><span class="p">,</span>
+    <span class="n">get_process_tasks</span><span class="p">,</span>
     <span class="n">invalidate_process_cache</span><span class="p">,</span>
 <span class="p">)</span>
 
@@ -612,20 +613,27 @@ <h1>Source code for libdebug.ptrace.ptrace_interface</h1><div class="highlight">
         <span class="k">with</span> <span class="n">extend_internal_debugger</span><span class="p">(</span><span class="bp">self</span><span class="p">):</span>
             <span class="bp">self</span><span class="o">.</span><span class="n">status_handler</span> <span class="o">=</span> <span class="n">PtraceStatusHandler</span><span class="p">()</span>
 
-        <span class="n">res</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">lib_trace</span><span class="o">.</span><span class="n">ptrace_attach</span><span class="p">(</span><span class="n">pid</span><span class="p">)</span>
-        <span class="k">if</span> <span class="n">res</span> <span class="o">==</span> <span class="o">-</span><span class="mi">1</span><span class="p">:</span>
-            <span class="n">errno_val</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">ffi</span><span class="o">.</span><span class="n">errno</span>
-            <span class="k">raise</span> <span class="ne">OSError</span><span class="p">(</span><span class="n">errno_val</span><span class="p">,</span> <span class="n">errno</span><span class="o">.</span><span class="n">errorcode</span><span class="p">[</span><span class="n">errno_val</span><span class="p">])</span>
+        <span class="c1"># Attach to all the tasks of the process</span>
+        <span class="bp">self</span><span class="o">.</span><span class="n">_attach_to_all_tasks</span><span class="p">(</span><span class="n">pid</span><span class="p">)</span>
 
         <span class="bp">self</span><span class="o">.</span><span class="n">process_id</span> <span class="o">=</span> <span class="n">pid</span>
         <span class="bp">self</span><span class="o">.</span><span class="n">detached</span> <span class="o">=</span> <span class="kc">False</span>
         <span class="bp">self</span><span class="o">.</span><span class="n">_internal_debugger</span><span class="o">.</span><span class="n">process_id</span> <span class="o">=</span> <span class="n">pid</span>
-        <span class="bp">self</span><span class="o">.</span><span class="n">register_new_thread</span><span class="p">(</span><span class="n">pid</span><span class="p">)</span>
         <span class="c1"># If we are attaching to a process, we don&#39;t want to continue to the entry point</span>
         <span class="c1"># which we have probably already passed</span>
         <span class="bp">self</span><span class="o">.</span><span class="n">_setup_parent</span><span class="p">(</span><span class="kc">False</span><span class="p">)</span></div>
 
 
+    <span class="k">def</span> <span class="nf">_attach_to_all_tasks</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">PtraceStatusHandler</span><span class="p">,</span> <span class="n">pid</span><span class="p">:</span> <span class="nb">int</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>
+<span class="w">        </span><span class="sd">&quot;&quot;&quot;Attach to all the tasks of the process.&quot;&quot;&quot;</span>
+        <span class="n">tids</span> <span class="o">=</span> <span class="n">get_process_tasks</span><span class="p">(</span><span class="n">pid</span><span class="p">)</span>
+        <span class="k">for</span> <span class="n">tid</span> <span class="ow">in</span> <span class="n">tids</span><span class="p">:</span>
+            <span class="n">res</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">lib_trace</span><span class="o">.</span><span class="n">ptrace_attach</span><span class="p">(</span><span class="n">tid</span><span class="p">)</span>
+            <span class="k">if</span> <span class="n">res</span> <span class="o">==</span> <span class="o">-</span><span class="mi">1</span><span class="p">:</span>
+                <span class="n">errno_val</span> <span class="o">=</span> <span class="bp">self</span><span class="o">.</span><span class="n">ffi</span><span class="o">.</span><span class="n">errno</span>
+                <span class="k">raise</span> <span class="ne">OSError</span><span class="p">(</span><span class="n">errno_val</span><span class="p">,</span> <span class="n">errno</span><span class="o">.</span><span class="n">errorcode</span><span class="p">[</span><span class="n">errno_val</span><span class="p">])</span>
+            <span class="bp">self</span><span class="o">.</span><span class="n">register_new_thread</span><span class="p">(</span><span class="n">tid</span><span class="p">)</span>
+
 <div class="viewcode-block" id="PtraceInterface.detach">
 <a class="viewcode-back" href="../../../libdebug.ptrace.html#libdebug.ptrace.ptrace_interface.PtraceInterface.detach">[docs]</a>
     <span class="k">def</span> <span class="nf">detach</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">PtraceInterface</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>

dev/_modules/libdebug/ptrace/ptrace_status_handler.html

@@ -434,7 +434,6 @@ <h1>Source code for libdebug.ptrace.ptrace_status_handler</h1><div class="highli
 
 <span class="kn">import</span> <span class="nn">os</span>
 <span class="kn">import</span> <span class="nn">signal</span>
-<span class="kn">from</span> <span class="nn">pathlib</span> <span class="kn">import</span> <span class="n">Path</span>
 <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.ptrace_software_breakpoint_patcher</span> <span class="kn">import</span> <span class="p">(</span>
@@ -444,6 +443,7 @@ <h1>Source code for libdebug.ptrace.ptrace_status_handler</h1><div class="highli
 <span class="kn">from</span> <span class="nn">libdebug.liblog</span> <span class="kn">import</span> <span class="n">liblog</span>
 <span class="kn">from</span> <span class="nn">libdebug.ptrace.ptrace_constants</span> <span class="kn">import</span> <span class="n">SYSCALL_SIGTRAP</span><span class="p">,</span> <span class="n">StopEvents</span>
 <span class="kn">from</span> <span class="nn">libdebug.state.resume_context</span> <span class="kn">import</span> <span class="n">EventType</span>
+<span class="kn">from</span> <span class="nn">libdebug.utils.process_utils</span> <span class="kn">import</span> <span class="n">get_process_tasks</span>
 <span class="kn">from</span> <span class="nn">libdebug.utils.signal_utils</span> <span class="kn">import</span> <span class="n">resolve_signal_name</span>
 
 <span class="k">if</span> <span class="n">TYPE_CHECKING</span><span class="p">:</span>
@@ -884,10 +884,7 @@ <h1>Source code for libdebug.ptrace.ptrace_status_handler</h1><div class="highli
 <a class="viewcode-back" href="../../../libdebug.ptrace.html#libdebug.ptrace.ptrace_status_handler.PtraceStatusHandler.check_for_new_threads">[docs]</a>
     <span class="k">def</span> <span class="nf">check_for_new_threads</span><span class="p">(</span><span class="bp">self</span><span class="p">:</span> <span class="n">PtraceStatusHandler</span><span class="p">,</span> <span class="n">pid</span><span class="p">:</span> <span class="nb">int</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>
 <span class="w">        </span><span class="sd">&quot;&quot;&quot;Check for new threads in the process and register them.&quot;&quot;&quot;</span>
-        <span class="k">if</span> <span class="ow">not</span> <span class="n">Path</span><span class="p">(</span><span class="sa">f</span><span class="s2">&quot;/proc/</span><span class="si">{</span><span class="n">pid</span><span class="si">}</span><span class="s2">/task&quot;</span><span class="p">)</span><span class="o">.</span><span class="n">exists</span><span class="p">():</span>
-            <span class="k">return</span>
-
-        <span class="n">tids</span> <span class="o">=</span> <span class="p">[</span><span class="nb">int</span><span class="p">(</span><span class="n">x</span><span class="p">)</span> <span class="k">for</span> <span class="n">x</span> <span class="ow">in</span> <span class="n">os</span><span class="o">.</span><span class="n">listdir</span><span class="p">(</span><span class="sa">f</span><span class="s2">&quot;/proc/</span><span class="si">{</span><span class="n">pid</span><span class="si">}</span><span class="s2">/task&quot;</span><span class="p">)]</span>
+        <span class="n">tids</span> <span class="o">=</span> <span class="n">get_process_tasks</span><span class="p">(</span><span class="n">pid</span><span class="p">)</span>
         <span class="k">for</span> <span class="n">tid</span> <span class="ow">in</span> <span class="n">tids</span><span class="p">:</span>
             <span class="k">if</span> <span class="ow">not</span> <span class="bp">self</span><span class="o">.</span><span class="n">internal_debugger</span><span class="o">.</span><span class="n">get_thread_by_id</span><span class="p">(</span><span class="n">tid</span><span class="p">):</span>
                 <span class="bp">self</span><span class="o">.</span><span class="n">ptrace_interface</span><span class="o">.</span><span class="n">register_new_thread</span><span class="p">(</span><span class="n">tid</span><span class="p">,</span> <span class="bp">self</span><span class="o">.</span><span class="n">internal_debugger</span><span class="p">)</span>

dev/_modules/libdebug/utils/process_utils.html

@@ -473,6 +473,24 @@ <h1>Source code for libdebug.utils.process_utils</h1><div class="highlight"><pre
 
 
 
+<div class="viewcode-block" id="get_process_tasks">
+<a class="viewcode-back" href="../../../libdebug.utils.html#libdebug.utils.process_utils.get_process_tasks">[docs]</a>
+<span class="k">def</span> <span class="nf">get_process_tasks</span><span class="p">(</span><span class="n">process_id</span><span class="p">:</span> <span class="nb">int</span><span class="p">)</span> <span class="o">-&gt;</span> <span class="nb">list</span><span class="p">[</span><span class="nb">int</span><span class="p">]:</span>
+<span class="w">    </span><span class="sd">&quot;&quot;&quot;Returns the tasks of the specified process.</span>
+
+<span class="sd">    Args:</span>
+<span class="sd">        process_id (int): The PID of the process whose tasks should be returned.</span>
+
+<span class="sd">    Returns:</span>
+<span class="sd">        list: A list of integers, each representing a task of the specified process.</span>
+<span class="sd">    &quot;&quot;&quot;</span>
+    <span class="n">tids</span> <span class="o">=</span> <span class="p">[]</span>
+    <span class="k">if</span> <span class="n">Path</span><span class="p">(</span><span class="sa">f</span><span class="s2">&quot;/proc/</span><span class="si">{</span><span class="n">process_id</span><span class="si">}</span><span class="s2">/task&quot;</span><span class="p">)</span><span class="o">.</span><span class="n">exists</span><span class="p">():</span>
+        <span class="n">tids</span> <span class="o">=</span> <span class="p">[</span><span class="nb">int</span><span class="p">(</span><span class="n">task</span><span class="p">)</span> <span class="k">for</span> <span class="n">task</span> <span class="ow">in</span> <span class="n">os</span><span class="o">.</span><span class="n">listdir</span><span class="p">(</span><span class="sa">f</span><span class="s2">&quot;/proc/</span><span class="si">{</span><span class="n">process_id</span><span class="si">}</span><span class="s2">/task&quot;</span><span class="p">)]</span>
+    <span class="k">return</span> <span class="n">tids</span></div>
+
+
+
 <div class="viewcode-block" id="invalidate_process_cache">
 <a class="viewcode-back" href="../../../libdebug.utils.html#libdebug.utils.process_utils.invalidate_process_cache">[docs]</a>
 <span class="k">def</span> <span class="nf">invalidate_process_cache</span><span class="p">()</span> <span class="o">-&gt;</span> <span class="kc">None</span><span class="p">:</span>

dev/genindex.html

@@ -812,6 +812,8 @@ <h2 id="G">G</h2>
       <li><a href="libdebug.utils.html#libdebug.utils.platform_utils.get_platform_register_size">get_platform_register_size() (in module libdebug.utils.platform_utils)</a>
 </li>
       <li><a href="libdebug.utils.html#libdebug.utils.process_utils.get_process_maps">get_process_maps() (in module libdebug.utils.process_utils)</a>
+</li>
+      <li><a href="libdebug.utils.html#libdebug.utils.process_utils.get_process_tasks">get_process_tasks() (in module libdebug.utils.process_utils)</a>
 </li>
       <li><a href="libdebug.utils.html#libdebug.utils.syscall_utils.get_remote_definition_url">get_remote_definition_url() (in module libdebug.utils.syscall_utils)</a>
 </li>

dev/libdebug.html

@@ -1149,6 +1149,7 @@ <h2>Subpackages<a class="headerlink" href="#subpackages" title="Link to this hea
 <li class="toctree-l2"><a class="reference internal" href="libdebug.utils.html#module-libdebug.utils.process_utils">libdebug.utils.process_utils module</a><ul>
 <li class="toctree-l3"><a class="reference internal" href="libdebug.utils.html#libdebug.utils.process_utils.get_process_maps"><code class="docutils literal notranslate"><span class="pre">get_process_maps()</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="libdebug.utils.html#libdebug.utils.process_utils.get_open_fds"><code class="docutils literal notranslate"><span class="pre">get_open_fds()</span></code></a></li>
+<li class="toctree-l3"><a class="reference internal" href="libdebug.utils.html#libdebug.utils.process_utils.get_process_tasks"><code class="docutils literal notranslate"><span class="pre">get_process_tasks()</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="libdebug.utils.html#libdebug.utils.process_utils.invalidate_process_cache"><code class="docutils literal notranslate"><span class="pre">invalidate_process_cache()</span></code></a></li>
 <li class="toctree-l3"><a class="reference internal" href="libdebug.utils.html#libdebug.utils.process_utils.disable_self_aslr"><code class="docutils literal notranslate"><span class="pre">disable_self_aslr()</span></code></a></li>
 </ul>

dev/libdebug.utils.html

@@ -1163,6 +1163,23 @@ <h2>libdebug.utils.gdb module<a class="headerlink" href="#libdebug-utils-gdb-mod
 </dl>
 </dd></dl>
 
+<dl class="py function">
+<dt class="sig sig-object py" id="libdebug.utils.process_utils.get_process_tasks">
+<span class="sig-prename descclassname"><span class="pre">libdebug.utils.process_utils.</span></span><span class="sig-name descname"><span class="pre">get_process_tasks</span></span><span class="sig-paren">(</span><em class="sig-param"><span class="n"><span class="pre">process_id</span></span><span class="p"><span class="pre">:</span></span><span class="w"> </span><span class="n"><span class="pre">int</span></span></em><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">list</span><span class="p"><span class="pre">[</span></span><span class="pre">int</span><span class="p"><span class="pre">]</span></span></span></span><a class="reference internal" href="_modules/libdebug/utils/process_utils.html#get_process_tasks"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#libdebug.utils.process_utils.get_process_tasks" title="Link to this definition">#</a></dt>
+<dd><p>Returns the tasks of the specified process.</p>
+<dl class="field-list simple">
+<dt class="field-odd">Parameters<span class="colon">:</span></dt>
+<dd class="field-odd"><p><strong>process_id</strong> (<em>int</em>) – The PID of the process whose tasks should be returned.</p>
+</dd>
+<dt class="field-even">Returns<span class="colon">:</span></dt>
+<dd class="field-even"><p>A list of integers, each representing a task of the specified process.</p>
+</dd>
+<dt class="field-odd">Return type<span class="colon">:</span></dt>
+<dd class="field-odd"><p>list</p>
+</dd>
+</dl>
+</dd></dl>
+
 <dl class="py function">
 <dt class="sig sig-object py" id="libdebug.utils.process_utils.invalidate_process_cache">
 <span class="sig-prename descclassname"><span class="pre">libdebug.utils.process_utils.</span></span><span class="sig-name descname"><span class="pre">invalidate_process_cache</span></span><span class="sig-paren">(</span><span class="sig-paren">)</span> <span class="sig-return"><span class="sig-return-icon">&#x2192;</span> <span class="sig-return-typehint"><span class="pre">None</span></span></span><a class="reference internal" href="_modules/libdebug/utils/process_utils.html#invalidate_process_cache"><span class="viewcode-link"><span class="pre">[source]</span></span></a><a class="headerlink" href="#libdebug.utils.process_utils.invalidate_process_cache" title="Link to this definition">#</a></dt>
@@ -1409,6 +1426,7 @@ <h2>libdebug.utils.gdb module<a class="headerlink" href="#libdebug-utils-gdb-mod
 <li class="toc-h2 nav-item toc-entry"><a class="reference internal nav-link" href="#module-libdebug.utils.process_utils">libdebug.utils.process_utils module</a><ul class="nav section-nav flex-column">
 <li class="toc-h3 nav-item toc-entry"><a class="reference internal nav-link" href="#libdebug.utils.process_utils.get_process_maps"><code class="docutils literal notranslate"><span class="pre">get_process_maps()</span></code></a></li>
 <li class="toc-h3 nav-item toc-entry"><a class="reference internal nav-link" href="#libdebug.utils.process_utils.get_open_fds"><code class="docutils literal notranslate"><span class="pre">get_open_fds()</span></code></a></li>
+<li class="toc-h3 nav-item toc-entry"><a class="reference internal nav-link" href="#libdebug.utils.process_utils.get_process_tasks"><code class="docutils literal notranslate"><span class="pre">get_process_tasks()</span></code></a></li>
 <li class="toc-h3 nav-item toc-entry"><a class="reference internal nav-link" href="#libdebug.utils.process_utils.invalidate_process_cache"><code class="docutils literal notranslate"><span class="pre">invalidate_process_cache()</span></code></a></li>
 <li class="toc-h3 nav-item toc-entry"><a class="reference internal nav-link" href="#libdebug.utils.process_utils.disable_self_aslr"><code class="docutils literal notranslate"><span class="pre">disable_self_aslr()</span></code></a></li>
 </ul>