htop/TraceScreen.c

193 lines
5.1 KiB
C
Raw Normal View History

2006-03-04 18:16:49 +00:00
/*
htop - TraceScreen.c
2006-03-23 18:55:29 +00:00
(C) 2005-2006 Hisham H. Muhammad
Released under the GNU GPLv2, see the COPYING file
2006-03-04 18:16:49 +00:00
in the source distribution for its full text.
*/
#include "config.h" // IWYU pragma: keep
2011-12-26 21:35:57 +00:00
#include "TraceScreen.h"
2011-12-26 21:35:57 +00:00
#include <assert.h>
#include <fcntl.h>
#include <signal.h>
#include <stdbool.h>
2006-03-04 18:16:49 +00:00
#include <stdio.h>
2011-12-26 21:35:57 +00:00
#include <stdlib.h>
#include <string.h>
2006-03-04 18:16:49 +00:00
#include <unistd.h>
#include <sys/select.h>
#include <sys/time.h>
#include <sys/wait.h>
#include "CRT.h"
#include "FunctionBar.h"
#include "IncSet.h"
#include "Panel.h"
#include "ProvideCurses.h"
#include "XUtils.h"
2006-03-04 18:16:49 +00:00
static const char* const TraceScreenFunctions[] = {"Search ", "Filter ", "AutoScroll ", "Stop Tracing ", "Done ", NULL};
2006-03-04 18:16:49 +00:00
static const char* const TraceScreenKeys[] = {"F3", "F4", "F8", "F9", "Esc"};
2006-03-04 18:16:49 +00:00
2015-03-23 18:26:56 +00:00
static int TraceScreenEvents[] = {KEY_F(3), KEY_F(4), KEY_F(8), KEY_F(9), 27};
2006-03-04 18:16:49 +00:00
2020-10-05 11:19:50 +00:00
const InfoScreenClass TraceScreen_class = {
.super = {
.extends = Class(Object),
.delete = TraceScreen_delete
},
.draw = TraceScreen_draw,
.onErr = TraceScreen_updateTrace,
.onKey = TraceScreen_onKey,
};
2006-03-04 18:16:49 +00:00
TraceScreen* TraceScreen_new(Process* process) {
// This initializes all TraceScreen variables to "false" so only default = true ones need to be set below
TraceScreen* this = xCalloc(1, sizeof(TraceScreen));
Object_setClass(this, Class(TraceScreen));
2006-03-04 18:16:49 +00:00
this->tracing = true;
FunctionBar* fuBar = FunctionBar_new(TraceScreenFunctions, TraceScreenKeys, TraceScreenEvents);
CRT_disableDelay();
return (TraceScreen*) InfoScreen_init(&this->super, process, fuBar, LINES-2, "");
2006-03-04 18:16:49 +00:00
}
void TraceScreen_delete(Object* cast) {
TraceScreen* this = (TraceScreen*) cast;
if (this->child > 0) {
kill(this->child, SIGTERM);
waitpid(this->child, NULL, 0);
}
2020-10-17 20:32:43 +00:00
if (this->strace)
fclose(this->strace);
CRT_enableDelay();
2020-10-17 20:32:43 +00:00
free(InfoScreen_done((InfoScreen*)this));
2006-03-04 18:16:49 +00:00
}
void TraceScreen_draw(InfoScreen* this) {
2006-03-04 18:16:49 +00:00
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]);
IncSet_drawBar(this->inc);
}
bool TraceScreen_forkTracer(TraceScreen* this) {
2020-10-17 20:32:43 +00:00
int fdpair[2] = {0, 0};
if (pipe(fdpair) == -1)
return false;
if(fcntl(fdpair[0], F_SETFL, O_NONBLOCK) < 0)
goto err;
if(fcntl(fdpair[1], F_SETFL, O_NONBLOCK) < 0)
goto err;
pid_t child = fork();
if (child == -1)
goto err;
if (child == 0) {
close(fdpair[0]);
dup2(fdpair[1], STDOUT_FILENO);
dup2(fdpair[1], STDERR_FILENO);
close(fdpair[1]);
CRT_dropPrivileges();
2020-10-17 20:32:43 +00:00
char buffer[32] = {0};
xSnprintf(buffer, sizeof(buffer), "%d", this->super.process->pid);
execlp("strace", "strace", "-T", "-tt", "-s", "512", "-p", buffer, NULL);
// Should never reach here, unless execlp fails ...
const char* message = "Could not execute 'strace'. Please make sure it is available in your $PATH.";
2020-10-17 20:32:43 +00:00
ssize_t written = write(STDERR_FILENO, message, strlen(message));
(void) written;
2020-10-17 20:32:43 +00:00
exit(127);
2006-03-04 18:16:49 +00:00
}
2020-10-17 20:32:43 +00:00
FILE* fd = fdopen(fdpair[0], "r");
if (!fd)
goto err;
close(fdpair[1]);
this->child = child;
this->strace = fd;
return true;
2020-10-17 20:32:43 +00:00
err:
close(fdpair[1]);
close(fdpair[0]);
return false;
}
void TraceScreen_updateTrace(InfoScreen* super) {
TraceScreen* this = (TraceScreen*) super;
2020-09-15 20:08:43 +00:00
char buffer[1025];
2020-10-17 20:32:43 +00:00
int fd_strace = fileno(this->strace);
assert(fd_strace != -1);
fd_set fds;
FD_ZERO(&fds);
// FD_SET(STDIN_FILENO, &fds);
2020-10-17 20:32:43 +00:00
FD_SET(fd_strace, &fds);
struct timeval tv;
tv.tv_sec = 0; tv.tv_usec = 500;
2020-10-17 20:32:43 +00:00
int ready = select(fd_strace+1, &fds, NULL, NULL, &tv);
2020-10-15 20:35:44 +00:00
size_t nread = 0;
2020-10-17 20:32:43 +00:00
if (ready > 0 && FD_ISSET(fd_strace, &fds))
2020-09-15 20:08:43 +00:00
nread = fread(buffer, 1, sizeof(buffer) - 1, this->strace);
if (nread && this->tracing) {
2020-10-15 20:35:44 +00:00
const char* line = buffer;
buffer[nread] = '\0';
2020-10-15 20:35:44 +00:00
for (size_t i = 0; i < nread; i++) {
if (buffer[i] == '\n') {
buffer[i] = '\0';
if (this->contLine) {
InfoScreen_appendLine(&this->super, line);
this->contLine = false;
} else {
InfoScreen_addLine(&this->super, line);
}
line = buffer+i+1;
2006-03-04 18:16:49 +00:00
}
}
if (line < buffer+nread) {
InfoScreen_addLine(&this->super, line);
buffer[nread] = '\0';
this->contLine = true;
}
if (this->follow)
Panel_setSelected(this->super.display, Panel_size(this->super.display)-1);
}
}
bool TraceScreen_onKey(InfoScreen* super, int ch) {
TraceScreen* this = (TraceScreen*) super;
switch(ch) {
case 'f':
case KEY_F(8):
this->follow = !(this->follow);
if (this->follow)
Panel_setSelected(super->display, Panel_size(super->display)-1);
return true;
case 't':
case KEY_F(9):
this->tracing = !this->tracing;
FunctionBar_setLabel(super->display->defaultBar, KEY_F(9), this->tracing?"Stop Tracing ":"Resume Tracing ");
InfoScreen_draw(this);
return true;
2006-03-04 18:16:49 +00:00
}
this->follow = false;
return false;
2006-03-04 18:16:49 +00:00
}