pstree/proc_psinfo.c at master · janmojzis/pstree · GitHub

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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
/*
Jan Mojzis
Public domain.
20091020
*/

/* tested on sunos, aix */

#include "proc.h"

#ifdef PROCTYPE_PSINFO

#include "stralloc.h"
#include "openreadclose.h"
#include "error.h"
#include "scan.h"

static stralloc taskname = {0};
static stralloc taskdir  = {0};
static int taskppid = -1;

int proc_add_thread(const char *procdir, const char *piddir){

    unsigned long ul;
    unsigned int i;
    proc_entry *new;
    stralloc *name;

    /* assuming that thread name is same as process name */
    i = scan_ulong(piddir,&ul);
    if (!i) return 0; /* XXX */
    if (ul == 1) return 0; /* master thread */

    /* allocate memory for new process */
    new = proc_new();
    if (!new) return -1;
    name = &(new->name);

    if (!stralloc_copys(name, "{")) return -1;
    if (!stralloc_cat(name, &taskname)) return -1;
    if (!stralloc_cats(name, "}")) return -1;
    new->fakepid = ul;
    new->fakepid <<= 32;
    new->fakepid += taskppid;
    new->fakeppid = taskppid;

    /* add to tree */
    if (!ctx->first) ctx->first = new;
    if (ctx->last) ctx->last->next = new;
    ctx->last = new;

    return 0;
}

static stralloc buf = {0};
static stralloc fn = {0};

int proc_add(const char *procdir ,const char *piddir){

    psinfo_t *ps;
    lwpsinfo_t *lwps;
    proc_entry *new;
    stralloc *name;

    /* allocate memory for new process */
    new = proc_new();
    if (!new) return -1;
    name = &(new->name);

    /* psinfo filename */
    if (!stralloc_copys(&fn, procdir)) return -1;
    if (!stralloc_copys(&fn, "/")) return -1;
    if (!stralloc_copys(&fn, piddir)) return -1;
    if (!stralloc_cats(&fn,"/psinfo")) return -1;
    if (!stralloc_0(&fn)) return -1;

    /* read */
    if (openreadclose(fn.s, &buf, sizeof(psinfo_t)) <= 0) return -1;

    if (buf.len != sizeof(psinfo_t)) {errno = error_proto; return -1;}
    ps = (psinfo_t *)buf.s;

    if (!stralloc_copys(name, ps->pr_fname)) return -1;
    if (!stralloc_copys(&taskname, ps->pr_fname)) return -1;
    new->fakepid = ps->pr_pid;
    taskppid     = ps->pr_pid;
    new->fakeppid = ps->pr_ppid;

    /* add to tree */
    if (!ctx->first) ctx->first = new;
    if (ctx->last) ctx->last->next = new;
    ctx->last = new;

    /* add tasks */
    if (!stralloc_copys(&taskdir, procdir)) return -1;
    if (!stralloc_cats(&taskdir, "/")) return -1;
    if (!stralloc_cats(&taskdir, piddir)) return -1;
    if (!stralloc_cats(&taskdir,"/lwp")) return -1;
    if (!stralloc_0(&taskdir)) return -1;
    if (proc_read(procdir, taskdir.s, proc_add_thread) == -1) return -1;


  return 0;
}

int proc_get(void){
    return proc_read("/proc", "/proc", proc_add);
}


#endif /* PROCTYPE_PSINFO */