From 0be46c952901aaafe76e1fb5a8faecb38323d549 Mon Sep 17 00:00:00 2001
From: Orfeas <38209077+0xfea5@users.noreply.github.com>
Date: Mon, 15 Apr 2024 19:38:05 +0300
Subject: Attaching & Detaching from process

---
 src/main.c | 34 +++++++++++++++++++++++++++++++++-
 1 file changed, 33 insertions(+), 1 deletion(-)

(limited to 'src')

diff --git a/src/main.c b/src/main.c
index b54dbf1..1d23902 100644
--- a/src/main.c
+++ b/src/main.c
@@ -1,6 +1,38 @@
+#include <stdlib.h>
 #include <stdio.h>
+#include <sys/ptrace.h>
+#include <sys/wait.h>
+#include <unistd.h>
+
+#define ERROR(...) \
+  do { \
+    fprintf(stderr, __VA_ARGS__); \
+    exit(1); \
+  } while (0)
+
+#define LOG(...) \
+  fprintf(stderr, __VA_ARGS__)
 
 int main(int argc, char *argv[])
 {
-  
+  if (argc < 2) {
+    ERROR("Usage: %s <tracee_pid>\n", argv[0]);
+  }
+
+  int pid;
+  if ((pid = atoi(argv[1])) == 0) {
+    ERROR("Invalid pid '%s'\n", argv[1]);
+  }
+
+  ptrace(PTRACE_ATTACH, pid, NULL, NULL);
+  waitpid(pid, NULL, __WALL);
+  LOG("Attached to process %d\n", pid);
+
+  /* Do stuff ... */
+  sleep(5);
+
+  ptrace(PTRACE_DETACH, pid, NULL, NULL);
+  LOG("Detached from process %d\n", pid);
+
+  return 0;
 }
-- 
cgit v1.2.3