Imported Upstream version 0.6upstream/0.6

1 files changed, 164 insertions, 0 deletions

diff --git a/TraceScreen.c b/TraceScreen.c
new file mode 100644
index 0000000..9628290
--- /dev/null
+++ b/TraceScreen.c
@@ -0,0 +1,164 @@
+/*
+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);
+}