Adding upstream version 3.2.2.upstream/3.2.2upstream

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>

1 files changed, 193 insertions, 0 deletions

diff --git a/TraceScreen.c b/TraceScreen.c
new file mode 100644
index 0000000..2aa0781
--- /dev/null
+++ b/TraceScreen.c
@@ -0,0 +1,193 @@
+/*
+htop - TraceScreen.c
+(C) 2005-2006 Hisham H. Muhammad
+Released under the GNU GPLv2+, see the COPYING file
+in the source distribution for its full text.
+*/
+
+#include "config.h" // IWYU pragma: keep
+
+#include "TraceScreen.h"
+
+#include <assert.h>
+#include <errno.h>
+#include <fcntl.h>
+#include <signal.h>
+#include <stdbool.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <sys/select.h>
+#include <sys/wait.h>
+
+#include "CRT.h"
+#include "FunctionBar.h"
+#include "Panel.h"
+#include "ProvideCurses.h"
+#include "XUtils.h"
+
+
+static const char* const TraceScreenFunctions[] = {"Search ", "Filter ", "AutoScroll ", "Stop Tracing   ", "Done   ", NULL};
+
+static const char* const TraceScreenKeys[] = {"F3", "F4", "F8", "F9", "Esc"};
+
+static const int TraceScreenEvents[] = {KEY_F(3), KEY_F(4), KEY_F(8), KEY_F(9), 27};
+
+TraceScreen* TraceScreen_new(const 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));
+   this->tracing = true;
+   FunctionBar* fuBar = FunctionBar_new(TraceScreenFunctions, TraceScreenKeys, TraceScreenEvents);
+   CRT_disableDelay();
+   return (TraceScreen*) InfoScreen_init(&this->super, process, fuBar, LINES - 2, " ");
+}
+
+void TraceScreen_delete(Object* cast) {
+   TraceScreen* this = (TraceScreen*) cast;
+   if (this->child > 0) {
+      kill(this->child, SIGTERM);
+      while (waitpid(this->child, NULL, 0) == -1)
+         if (errno != EINTR)
+            break;
+   }
+
+   if (this->strace) {
+      fclose(this->strace);
+   }
+
+   CRT_enableDelay();
+   free(InfoScreen_done((InfoScreen*)this));
+}
+
+static void TraceScreen_draw(InfoScreen* this) {
+   InfoScreen_drawTitled(this, "Trace of process %d - %s", this->process->pid, Process_getCommand(this->process));
+}
+
+bool TraceScreen_forkTracer(TraceScreen* this) {
+   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]);
+
+      char buffer[32] = {0};
+      xSnprintf(buffer, sizeof(buffer), "%d", this->super.process->pid);
+      // Use of NULL in variadic functions must have a pointer cast.
+      // The NULL constant is not required by standard to have a pointer type.
+      execlp("strace", "strace", "-T", "-tt", "-s", "512", "-p", buffer, (char *)NULL);
+
+      // Should never reach here, unless execlp fails ...
+      const char* message = "Could not execute 'strace'. Please make sure it is available in your $PATH.";
+      (void)! write(STDERR_FILENO, message, strlen(message));
+
+      exit(127);
+   }
+
+   FILE* fd = fdopen(fdpair[0], "r");
+   if (!fd)
+      goto err;
+
+   close(fdpair[1]);
+
+   this->child = child;
+   this->strace = fd;
+   return true;
+
+err:
+   close(fdpair[1]);
+   close(fdpair[0]);
+   return false;
+}
+
+static void TraceScreen_updateTrace(InfoScreen* super) {
+   TraceScreen* this = (TraceScreen*) super;
+   char buffer[1025];
+
+   int fd_strace = fileno(this->strace);
+   assert(fd_strace != -1);
+
+   fd_set fds;
+   FD_ZERO(&fds);
+// FD_SET(STDIN_FILENO, &fds);
+   FD_SET(fd_strace, &fds);
+
+   struct timeval tv = { .tv_sec = 0, .tv_usec = 500 };
+   int ready = select(fd_strace + 1, &fds, NULL, NULL, &tv);
+
+   size_t nread = 0;
+   if (ready > 0 && FD_ISSET(fd_strace, &fds))
+      nread = fread(buffer, 1, sizeof(buffer) - 1, this->strace);
+
+   if (nread && this->tracing) {
+      const char* line = buffer;
+      buffer[nread] = '\0';
+      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;
+         }
+      }
+      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);
+      }
+   }
+}
+
+static 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;
+   }
+   this->follow = false;
+   return false;
+}
+
+const InfoScreenClass TraceScreen_class = {
+   .super = {
+      .extends = Class(Object),
+      .delete = TraceScreen_delete
+   },
+   .draw = TraceScreen_draw,
+   .onErr = TraceScreen_updateTrace,
+   .onKey = TraceScreen_onKey,
+};