summaryrefslogtreecommitdiff
path: root/wrapper/ptrace.c
blob: caf510f77f11d54b6398d642fab176188ffe135a (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
/* Copyright (C) 2017 Jakob Kreuze, All Rights Reserved.

   This file is part of Hypodermic.

   Hypodermic is free software: you can redistribute it and/or modify it
   under the terms of the GNU General Public License as published by the
   Free Software Foundation, either version 3 of the License, or (at
   your option) any later version.

   Hypodermic is distributed in the hope that it will be useful, but
   WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
   General Public License for more details.

   You should have received a copy of the GNU General Public License
   along with Hypodermic. If not, see <http://www.gnu.org/licenses/>. */

#include <sys/ptrace.h>
#include <sys/types.h>
#include <sys/wait.h>

#include <stddef.h>
#include <unistd.h>


static void run_target(const char *path) {
    if (ptrace(PTRACE_TRACEME, 0, NULL, NULL) < 0) {
        return;
    }

    execl(path, path, 0);
}


int new_proc(const char *path) {
    int   wait_stat;
    pid_t pid;

    if (path == NULL) {
        return -1;
    }

    pid = fork();

    if (pid == 0) {
        run_target(path);
    } else if (pid > 0) {
        wait(&wait_stat);
    } else {
        return -1;
    }

    return pid;
}


int attach(int pid) {
    if ((ptrace(PTRACE_ATTACH, pid, NULL, NULL)) < 0) {
        return -1;
    }

    waitpid(pid, NULL, WUNTRACED);
    return 0;
}


int detach(int pid) {
    return ptrace(PTRACE_DETACH, pid, NULL, NULL) < 0;
}


int cont(int pid) {
    int s;

    if ((ptrace(PTRACE_CONT, pid, NULL, NULL)) < 0) {
        return -1;
    }

    while (!WIFSTOPPED(s)) {
        waitpid(pid, &s, WNOHANG);
    }

    return 0;
}