mirror of https://github.com/xzeldon/htop.git
165 lines
4.5 KiB
C
165 lines
4.5 KiB
C
![]() |
/*
|
||
|
htop - TraceScreen.c
|
||
|
(C) 2005 Hisham H. Muhammad
|
||
|
Released under the GNU GPL, see the COPYING file
|
||
|
in the source distribution for its full text.
|
||
|
*/
|
||
|
|
||
|
#define _GNU_SOURCE
|
||
|
#include <stdio.h>
|
||
|
#include <unistd.h>
|
||
|
#include <stdbool.h>
|
||
|
#include <unistd.h>
|
||
|
#include <fcntl.h>
|
||
|
|
||
|
#include "TraceScreen.h"
|
||
|
#include "ProcessList.h"
|
||
|
#include "Process.h"
|
||
|
#include "ListItem.h"
|
||
|
|
||
|
/*{
|
||
|
|
||
|
typedef struct TraceScreen_ {
|
||
|
Process* process;
|
||
|
ListBox* display;
|
||
|
FunctionBar* bar;
|
||
|
bool tracing;
|
||
|
} TraceScreen;
|
||
|
|
||
|
}*/
|
||
|
|
||
|
/* private property */
|
||
|
static char* tbFunctions[3] = {"AutoScroll ", "Stop Tracing ", "Done "};
|
||
|
|
||
|
/* private property */
|
||
|
static char* tbKeys[3] = {"F4", "F5", "Esc"};
|
||
|
|
||
|
/* private property */
|
||
|
static int tbEvents[3] = {KEY_F(4), KEY_F(5), 27};
|
||
|
|
||
|
TraceScreen* TraceScreen_new(Process* process) {
|
||
|
TraceScreen* this = (TraceScreen*) malloc(sizeof(TraceScreen));
|
||
|
this->process = process;
|
||
|
this->display = ListBox_new(0, 1, COLS, LINES-2, LISTITEM_CLASS, true);
|
||
|
this->bar = FunctionBar_new(3, tbFunctions, tbKeys, tbEvents);
|
||
|
this->tracing = true;
|
||
|
return this;
|
||
|
}
|
||
|
|
||
|
void TraceScreen_delete(TraceScreen* this) {
|
||
|
ListBox_delete((Object*)this->display);
|
||
|
FunctionBar_delete((Object*)this->bar);
|
||
|
free(this);
|
||
|
}
|
||
|
|
||
|
void TraceScreen_draw(TraceScreen* this) {
|
||
|
attrset(CRT_colors[PANEL_HEADER_FOCUS]);
|
||
|
mvhline(0, 0, ' ', COLS);
|
||
|
mvprintw(0, 0, "Trace of process %d - %s", this->process->pid, this->process->comm);
|
||
|
attrset(CRT_colors[DEFAULT_COLOR]);
|
||
|
FunctionBar_draw(this->bar, NULL);
|
||
|
}
|
||
|
|
||
|
void TraceScreen_run(TraceScreen* this) {
|
||
|
// if (this->process->pid == getpid()) return;
|
||
|
char buffer[1001];
|
||
|
int fdpair[2];
|
||
|
int err = pipe(fdpair);
|
||
|
if (err == -1) return;
|
||
|
int child = fork();
|
||
|
if (child == -1) return;
|
||
|
if (child == 0) {
|
||
|
dup2(fdpair[1], STDERR_FILENO);
|
||
|
fcntl(fdpair[1], F_SETFL, O_NONBLOCK);
|
||
|
sprintf(buffer, "%d", this->process->pid);
|
||
|
execlp("strace", "strace", "-p", buffer, NULL);
|
||
|
exit(1);
|
||
|
}
|
||
|
fcntl(fdpair[0], F_SETFL, O_NONBLOCK);
|
||
|
FILE* strace = fdopen(fdpair[0], "r");
|
||
|
ListBox* lb = this->display;
|
||
|
int fd_strace = fileno(strace);
|
||
|
TraceScreen_draw(this);
|
||
|
CRT_disableDelay();
|
||
|
bool contLine = false;
|
||
|
bool follow = false;
|
||
|
bool looping = true;
|
||
|
while (looping) {
|
||
|
fd_set fds;
|
||
|
FD_ZERO(&fds);
|
||
|
FD_SET(fd_strace, &fds);
|
||
|
struct timeval tv;
|
||
|
tv.tv_sec = 0; tv.tv_usec = 500;
|
||
|
int ready = select(fd_strace+1, &fds, NULL, NULL, &tv);
|
||
|
int nread = 0;
|
||
|
if (ready > 0)
|
||
|
nread = fread(buffer, 1, 1000, strace);
|
||
|
if (nread && this->tracing) {
|
||
|
char* line = buffer;
|
||
|
buffer[nread] = '\0';
|
||
|
for (int i = 0; i < nread; i++) {
|
||
|
if (buffer[i] == '\n') {
|
||
|
buffer[i] = '\0';
|
||
|
if (contLine) {
|
||
|
ListItem_append((ListItem*)ListBox_get(lb,
|
||
|
ListBox_getSize(lb)-1), line);
|
||
|
contLine = false;
|
||
|
} else {
|
||
|
ListBox_add(lb, (Object*) ListItem_new(line, 0));
|
||
|
}
|
||
|
line = buffer+i+1;
|
||
|
}
|
||
|
}
|
||
|
if (line < buffer+nread) {
|
||
|
ListBox_add(lb, (Object*) ListItem_new(line, 0));
|
||
|
buffer[nread] = '\0';
|
||
|
contLine = true;
|
||
|
}
|
||
|
if (follow)
|
||
|
ListBox_setSelected(lb, ListBox_getSize(lb)-1);
|
||
|
ListBox_draw(lb, true);
|
||
|
}
|
||
|
int ch = getch();
|
||
|
if (ch == KEY_MOUSE) {
|
||
|
MEVENT mevent;
|
||
|
int ok = getmouse(&mevent);
|
||
|
if (ok == OK)
|
||
|
if (mevent.y >= lb->y && mevent.y < LINES - 1) {
|
||
|
ListBox_setSelected(lb, mevent.y - lb->y + lb->scrollV);
|
||
|
follow = false;
|
||
|
ch = 0;
|
||
|
} if (mevent.y == LINES - 1)
|
||
|
ch = FunctionBar_synthesizeEvent(this->bar, mevent.x);
|
||
|
}
|
||
|
switch(ch) {
|
||
|
case ERR:
|
||
|
continue;
|
||
|
case KEY_F(5):
|
||
|
this->tracing = !this->tracing;
|
||
|
FunctionBar_setLabel(this->bar, KEY_F(5), this->tracing?"Stop Tracing ":"Resume Tracing ");
|
||
|
TraceScreen_draw(this);
|
||
|
break;
|
||
|
case 'f':
|
||
|
case KEY_F(4):
|
||
|
follow = !follow;
|
||
|
if (follow)
|
||
|
ListBox_setSelected(lb, ListBox_getSize(lb)-1);
|
||
|
break;
|
||
|
case 'q':
|
||
|
case 27:
|
||
|
looping = false;
|
||
|
break;
|
||
|
case KEY_RESIZE:
|
||
|
ListBox_resize(lb, COLS, LINES-2);
|
||
|
TraceScreen_draw(this);
|
||
|
break;
|
||
|
default:
|
||
|
follow = false;
|
||
|
ListBox_onKey(lb, ch);
|
||
|
}
|
||
|
ListBox_draw(lb, true);
|
||
|
}
|
||
|
kill(child, SIGTERM);
|
||
|
fclose(strace);
|
||
|
}
|