/* ******************************************************************
osi_linux extra API
****************************************************************** */
char *osi_linux_resolve_fd(CPUState *env, OsiProc *p, int fd) {
char *name = get_fd_name(env, p->offset, fd);
if (unlikely(name == NULL)) goto make_name;
name = g_strchug(name);
if (unlikely(g_strcmp0(name, "") == 0)) goto make_name;
return name;
make_name:
g_free(name);
name = g_strdup_printf("FD%d_%d", fd, (int)p->pid);
return name;
}
char *osi_linux_fd_to_filename(CPUState *env, OsiProc *p, int fd) {
// target_ulong asid = panda_current_asid(env);
PTR ts_current = 0;
ts_current = p->offset;
if (ts_current == 0) {
if (debug) printf ("osi_linux_fd_to_filename(pid=%d, fd=%d) -- can't get task\n", (int)p->pid, fd);
return NULL;
}
char *name = get_fd_name(env, ts_current, fd);
if (unlikely(name == NULL)) {
if (debug) printf ("osi_linux_fd_to_filename(pid=%d, fd=%d) -- can't get filename\n", (int)p->pid, fd);
return NULL;
}
name = g_strchug(name);
if (unlikely(g_strcmp0(name, "") == 0)) {
if (debug) printf ("osi_linux_fd_to_filename(pid=%d, fd=%d) -- filename is empty\n", (int)p->pid, fd);
g_free(name);
return NULL;
}
return name;
}
char *osi_linux_fd_to_filename(CPUState *env, OsiProc *p, int fd) {
// target_ulong asid = panda_current_asid(env);
PTR ts_current = 0;
ts_current = p->offset;
if (ts_current == 0) {
if (debug) printf ("osi_linux_fd_to_filename -- cant get task\n");
return NULL;
}
char *name = get_fd_name(env, ts_current, fd);
if (unlikely(name == NULL)) {
if (debug) printf ("osi_linux_fd_to_filename -- can't get filename\n");
goto make_name;
}
name = g_strchug(name);
if (unlikely(g_strcmp0(name, "") == 0)) {
if (debug) printf ("osi_linux_fd_to_filename -- can't get filename 2\n");
goto make_name;
}
return name;
make_name:
return NULL;
}
/**
* @brief PPP callback to retrieve process list from the running OS.
*/
void on_get_processes(CPUState *env, OsiProcs **out_ps) {
PTR ts_first, ts_current;
OsiProcs *ps;
OsiProc *p;
uint32_t ps_capacity = 16;
ts_first = ts_current = get_task_struct(env, (_ESP & THREADINFO_MASK));
if (ts_current == (PTR)NULL) goto error0;
// When thread_group points to itself, the task_struct belongs to a thread
// (see kernel_structs.md for details). This will trigger an infinite loop
// in the traversal loop.
// Following next will lead us to a task_struct belonging to a process and
// help avoid the condition.
if (ts_current+ki.task.thread_group_offset != get_thread_group(env, ts_current)) {
ts_first = ts_current = get_task_struct_next(env, ts_current);
}
ps = (OsiProcs *)g_malloc0(sizeof(OsiProcs));
ps->proc = g_new(OsiProc, ps_capacity);
do {
if (ps->num == ps_capacity) {
ps_capacity *= 2;
ps->proc = g_renew(OsiProc, ps->proc, ps_capacity);
}
p = &ps->proc[ps->num++];
// Garbage in p->name will cause fill_osiproc() to segfault.
memset(p, 0, sizeof(OsiProc));
fill_osiproc(env, p, ts_current);
#if 0
/*********************************************************/
// Test of fd -> name resolution.
/*********************************************************/
for (int fdn=0; fdn<10; fdn++) {
char *s = get_fd_name(env, ts_current, fdn);
LOG_INFO("%s fd%d -> %s", p->name, fdn, s);
g_free(s);
}
/*********************************************************/
#endif
ts_current = get_task_struct_next(env, ts_current);
} while(ts_current != (PTR)NULL && ts_current != ts_first);
// memory read error
if (ts_current == (PTR)NULL) goto error1;
*out_ps = ps;
return;
error1:
do {
ps->num--;
g_free(ps->proc[ps->num].name);
} while (ps->num != 0);
g_free(ps->proc);
g_free(ps);
error0:
*out_ps = NULL;
return;
}
/**
* @brief PPP callback to retrieve process list from the running OS.
*/
void on_get_processes(CPUState *env, OsiProcs **out_ps) {
PTR ts_first, ts_current;
OsiProcs *ps;
OsiProc *p;
uint32_t ps_capacity = 16;
#ifdef OSI_LINUX_LIST_THREADS
PTR tg_first, tg_next;
#endif
// Get a task_struct of a process to start iterating the process list. If
// current task is a thread (ts->t_group != &ts->t_group), follow ts->next
// to get to a process.
// Always starting the traversal with a process has the benefits of:
// a. Simplifying the traversal when OSI_LINUX_LIST_THREADS is disabled.
// b. Avoiding an infinite loop when OSI_LINUX_LIST_THREADS is enabled and
// the current task is a thread.
// See kernel_structs.md for details.
ts_first = ts_current = get_task_struct(env, (_ESP & THREADINFO_MASK));
if (ts_current == (PTR)NULL) goto error0;
if (ts_current+ki.task.thread_group_offset != get_thread_group(env, ts_current)) {
ts_first = ts_current = get_task_struct_next(env, ts_current);
}
ps = (OsiProcs *)g_malloc0(sizeof(OsiProcs));
ps->proc = g_new(OsiProc, ps_capacity);
do {
if (ps->num == ps_capacity) {
ps_capacity *= 2;
ps->proc = g_renew(OsiProc, ps->proc, ps_capacity);
}
p = &ps->proc[ps->num++];
memset(p, 0, sizeof(OsiProc)); // fill_osiproc() expects p to be zeroed-out.
fill_osiproc(env, p, ts_current);
#ifdef OSI_LINUX_LIST_THREADS
// Traverse thread group list.
// It is assumed that ts_current is a thread group leader.
tg_first = ts_current+ki.task.thread_group_offset;
while ((tg_next = get_thread_group(env, ts_current)) != tg_first) {
ts_current = tg_next-ki.task.thread_group_offset;
if (ps->num == ps_capacity) {
ps_capacity *= 2;
ps->proc = g_renew(OsiProc, ps->proc, ps_capacity);
}
p = &ps->proc[ps->num++];
memset(p, 0, sizeof(OsiProc)); // fill_osiproc() expects p to be zeroed-out.
fill_osiproc(env, p, ts_current);
}
ts_current = tg_first-ki.task.thread_group_offset;
#endif
#if 0
/*********************************************************/
// Test of fd -> name resolution.
/*********************************************************/
for (int fdn=0; fdn<256; fdn++) {
char *s = get_fd_name(env, ts_current, fdn);
LOG_INFO("%s fd%d -> %s", p->name, fdn, s);
g_free(s);
}
/*********************************************************/
#endif
ts_current = get_task_struct_next(env, ts_current);
} while(ts_current != (PTR)NULL && ts_current != ts_first);
// memory read error
if (ts_current == (PTR)NULL) goto error1;
*out_ps = ps;
return;
error1:
do {
ps->num--;
g_free(ps->proc[ps->num].name);
} while (ps->num != 0);
g_free(ps->proc);
g_free(ps);
error0:
*out_ps = NULL;
return;
}
