From ff9409b1737627857eb47f64f536a3f66b6a09a4 Mon Sep 17 00:00:00 2001 From: Daniel Lange Date: Mon, 11 Apr 2016 13:01:07 +0200 Subject: Imported Upstream version 2.0.0 --- htop.c | 899 ++++++++++++++--------------------------------------------------- 1 file changed, 193 insertions(+), 706 deletions(-) (limited to 'htop.c') diff --git a/htop.c b/htop.c index b343809..9e82a12 100644 --- a/htop.c +++ b/htop.c @@ -5,51 +5,35 @@ Released under the GNU GPL, see the COPYING file in the source distribution for its full text. */ -#include "ProcessList.h" +#include "config.h" -#include "CRT.h" -#include "Panel.h" -#include "UsersTable.h" -#include "RichString.h" -#include "Settings.h" -#include "ScreenManager.h" #include "FunctionBar.h" -#include "ListItem.h" -#include "String.h" +#include "Hashtable.h" #include "ColumnsPanel.h" -#include "CategoriesPanel.h" -#include "SignalsPanel.h" -#include "TraceScreen.h" -#include "OpenFilesScreen.h" -#include "AffinityPanel.h" -#include "IOPriorityPanel.h" -#include "IncSet.h" +#include "CRT.h" +#include "MainPanel.h" +#include "ProcessList.h" +#include "ScreenManager.h" +#include "Settings.h" +#include "UsersTable.h" +#include "Platform.h" -#include -#include -#include -#include -#include -#include #include -#include +#include +#include +#include #include -#include -#include #include +#include //#link m -#define COPYRIGHT "(C) 2004-2012 Hisham Muhammad" - static void printVersionFlag() { fputs("htop " VERSION " - " COPYRIGHT "\n" "Released under the GNU GPL.\n\n", stdout); exit(0); } - -static const char* defaultFunctions[] = {"Help ", "Setup ", "Search", "Filter", "Tree ", "SortBy", "Nice -", "Nice +", "Kill ", "Quit ", NULL}; static void printHelpFlag() { fputs("htop " VERSION " - " COPYRIGHT "\n" @@ -60,7 +44,7 @@ static void printHelpFlag() { "-s --sort-key=COLUMN Sort by COLUMN (try --sort-key=help for a list)\n" "-u --user=USERNAME Show only processes of a given user\n" "-p --pid=PID,[,PID,PID...] Show only the given PIDs\n" - "-v --version Print version info\n" + "-v --version Print version info\n" "\n" "Long options may be passed with a single dash.\n\n" "Press F1 inside htop for online help.\n" @@ -69,298 +53,26 @@ static void printHelpFlag() { exit(0); } -static struct { const char* key; const char* info; } helpLeft[] = { - { .key = " Arrows: ", .info = "scroll process list" }, - { .key = " Digits: ", .info = "incremental PID search" }, - { .key = " F3 /: ", .info = "incremental name search" }, - { .key = " F4 \\: ",.info = "incremental name filtering" }, - { .key = " F5 t: ", .info = "tree view" }, - { .key = " u: ", .info = "show processes of a single user" }, - { .key = " H: ", .info = "hide/show user threads" }, - { .key = " K: ", .info = "hide/show kernel threads" }, - { .key = " F: ", .info = "cursor follows process" }, - { .key = " F6 + -: ", .info = "expand/collapse tree" }, - { .key = " P M T: ", .info = "sort by CPU%, MEM% or TIME" }, - { .key = " I: ", .info = "invert sort order" }, - { .key = " F6 >: ", .info = "select sort column" }, - { .key = NULL, .info = NULL } -}; - -static struct { const char* key; const char* info; } helpRight[] = { - { .key = " Space: ", .info = "tag process" }, - { .key = " c: ", .info = "tag process and its children" }, - { .key = " U: ", .info = "untag all processes" }, - { .key = " F9 k: ", .info = "kill process/tagged processes" }, - { .key = " F7 ]: ", .info = "higher priority (root only)" }, - { .key = " F8 [: ", .info = "lower priority (+ nice)" }, -#if (HAVE_LIBHWLOC || HAVE_NATIVE_AFFINITY) - { .key = " a: ", .info = "set CPU affinity" }, -#endif - { .key = " i: ", .info = "set IO prority" }, - { .key = " l: ", .info = "list open files with lsof" }, - { .key = " s: ", .info = "trace syscalls with strace" }, - { .key = " ", .info = "" }, - { .key = " F2 S: ", .info = "setup" }, - { .key = " F1 h: ", .info = "show this help screen" }, - { .key = " F10 q: ", .info = "quit" }, - { .key = NULL, .info = NULL } -}; - -static void showHelp(ProcessList* pl) { - clear(); - attrset(CRT_colors[HELP_BOLD]); - - for (int i = 0; i < LINES-1; i++) - mvhline(i, 0, ' ', COLS); - - mvaddstr(0, 0, "htop " VERSION " - " COPYRIGHT); - mvaddstr(1, 0, "Released under the GNU GPL. See 'man' page for more info."); - - attrset(CRT_colors[DEFAULT_COLOR]); - mvaddstr(3, 0, "CPU usage bar: "); - #define addattrstr(a,s) attrset(a);addstr(s) - addattrstr(CRT_colors[BAR_BORDER], "["); - if (pl->detailedCPUTime) { - addattrstr(CRT_colors[CPU_NICE_TEXT], "low"); addstr("/"); - addattrstr(CRT_colors[CPU_NORMAL], "normal"); addstr("/"); - addattrstr(CRT_colors[CPU_KERNEL], "kernel"); addstr("/"); - addattrstr(CRT_colors[CPU_IRQ], "irq"); addstr("/"); - addattrstr(CRT_colors[CPU_SOFTIRQ], "soft-irq"); addstr("/"); - addattrstr(CRT_colors[CPU_STEAL], "steal"); addstr("/"); - addattrstr(CRT_colors[CPU_GUEST], "guest"); addstr("/"); - addattrstr(CRT_colors[CPU_IOWAIT], "io-wait"); - addattrstr(CRT_colors[BAR_SHADOW], " used%"); - } else { - addattrstr(CRT_colors[CPU_NICE_TEXT], "low-priority"); addstr("/"); - addattrstr(CRT_colors[CPU_NORMAL], "normal"); addstr("/"); - addattrstr(CRT_colors[CPU_KERNEL], "kernel"); addstr("/"); - addattrstr(CRT_colors[CPU_STEAL], "virtualiz"); - addattrstr(CRT_colors[BAR_SHADOW], " used%"); - } - addattrstr(CRT_colors[BAR_BORDER], "]"); - attrset(CRT_colors[DEFAULT_COLOR]); - mvaddstr(4, 0, "Memory bar: "); - addattrstr(CRT_colors[BAR_BORDER], "["); - addattrstr(CRT_colors[MEMORY_USED], "used"); addstr("/"); - addattrstr(CRT_colors[MEMORY_BUFFERS_TEXT], "buffers"); addstr("/"); - addattrstr(CRT_colors[MEMORY_CACHE], "cache"); - addattrstr(CRT_colors[BAR_SHADOW], " used/total"); - addattrstr(CRT_colors[BAR_BORDER], "]"); - attrset(CRT_colors[DEFAULT_COLOR]); - mvaddstr(5, 0, "Swap bar: "); - addattrstr(CRT_colors[BAR_BORDER], "["); - addattrstr(CRT_colors[SWAP], "used"); - addattrstr(CRT_colors[BAR_SHADOW], " used/total"); - addattrstr(CRT_colors[BAR_BORDER], "]"); - attrset(CRT_colors[DEFAULT_COLOR]); - mvaddstr(6,0, "Type and layout of header meters are configurable in the setup screen."); - if (CRT_colorScheme == COLORSCHEME_MONOCHROME) { - mvaddstr(7, 0, "In monochrome, meters are displayed through different chars, in order: |#*@$%&"); - } - mvaddstr( 8, 0, " Status: R: running; S: sleeping; T: traced/stopped; Z: zombie; D: disk sleep"); - for (int i = 0; helpLeft[i].info; i++) { mvaddstr(9+i, 9, helpLeft[i].info); } - for (int i = 0; helpRight[i].info; i++) { mvaddstr(9+i, 49, helpRight[i].info); } - attrset(CRT_colors[HELP_BOLD]); - for (int i = 0; helpLeft[i].key; i++) { mvaddstr(9+i, 0, helpLeft[i].key); } - for (int i = 0; helpRight[i].key; i++) { mvaddstr(9+i, 40, helpRight[i].key); } - - attrset(CRT_colors[HELP_BOLD]); - mvaddstr(23,0, "Press any key to return."); - attrset(CRT_colors[DEFAULT_COLOR]); - refresh(); - CRT_readKey(); - clear(); -} - -static const char* CategoriesFunctions[] = {" ", " ", " ", " ", " ", " ", " ", " ", " ", "Done ", NULL}; - -static void Setup_run(Settings* settings, const Header* header) { - ScreenManager* scr = ScreenManager_new(0, header->height, 0, -1, HORIZONTAL, header, true); - CategoriesPanel* panelCategories = CategoriesPanel_new(settings, scr); - ScreenManager_add(scr, (Panel*) panelCategories, FunctionBar_new(CategoriesFunctions, NULL, NULL), 16); - CategoriesPanel_makeMetersPage(panelCategories); - Panel* panelFocus; - int ch; - ScreenManager_run(scr, &panelFocus, &ch); - ScreenManager_delete(scr); -} - -typedef bool(*ForeachProcessFn)(Process*, size_t); - -static bool foreachProcess(Panel* panel, ForeachProcessFn fn, int arg, bool* wasAnyTagged) { - bool ok = true; - bool anyTagged = false; - for (int i = 0; i < Panel_size(panel); i++) { - Process* p = (Process*) Panel_get(panel, i); - if (p->tag) { - ok = fn(p, arg) && ok; - anyTagged = true; - } - } - if (!anyTagged) { - Process* p = (Process*) Panel_getSelected(panel); - if (p) ok = fn(p, arg) && ok; - } - if (wasAnyTagged) - *wasAnyTagged = anyTagged; - return ok; -} - -static bool changePriority(Panel* panel, int delta) { - bool anyTagged; - bool ok = foreachProcess(panel, (ForeachProcessFn) Process_changePriorityBy, delta, &anyTagged); - if (!ok) - beep(); - return anyTagged; -} - -static int selectedPid(Panel* panel) { - Process* p = (Process*) Panel_getSelected(panel); - if (p) { - return p->pid; - } - return -1; -} - -static Object* pickFromVector(Panel* panel, Panel* list, int x, int y, const char** keyLabels, FunctionBar* prevBar, Header* header) { - const char* fuKeys[] = {"Enter", "Esc", NULL}; - int fuEvents[] = {13, 27}; - ScreenManager* scr = ScreenManager_new(0, y, 0, -1, HORIZONTAL, header, false); - scr->allowFocusChange = false; - ScreenManager_add(scr, list, FunctionBar_new(keyLabels, fuKeys, fuEvents), x - 1); - ScreenManager_add(scr, panel, NULL, -1); - Panel* panelFocus; - int ch; - bool unfollow = false; - int pid = selectedPid(panel); - if (header->pl->following == -1) { - header->pl->following = pid; - unfollow = true; - } - ScreenManager_run(scr, &panelFocus, &ch); - if (unfollow) { - header->pl->following = -1; - } - ScreenManager_delete(scr); - Panel_move(panel, 0, y); - Panel_resize(panel, COLS, LINES-y-1); - FunctionBar_draw(prevBar, NULL); - if (panelFocus == list && ch == 13) { - Process* selected = (Process*)Panel_getSelected(panel); - if (selected && selected->pid == pid) - return Panel_getSelected(list); - else - beep(); - } - return NULL; -} - -static void addUserToVector(int key, void* userCast, void* panelCast) { - char* user = (char*) userCast; - Panel* panel = (Panel*) panelCast; - Panel_add(panel, (Object*) ListItem_new(user, key)); -} - -static bool setUserOnly(const char* userName, bool* userOnly, uid_t* userId) { - struct passwd* user = getpwnam(userName); - if (user) { - *userOnly = true; - *userId = user->pw_uid; - return true; - } - return false; -} - -static void setTreeView(ProcessList* pl, FunctionBar* fuBar, bool mode) { - if (mode) { - FunctionBar_setLabel(fuBar, KEY_F(5), "Sorted"); - FunctionBar_setLabel(fuBar, KEY_F(6), "Collap"); - } else { - FunctionBar_setLabel(fuBar, KEY_F(5), "Tree "); - FunctionBar_setLabel(fuBar, KEY_F(6), "SortBy"); - } - if (mode != pl->treeView) { - FunctionBar_draw(fuBar, NULL); - } - pl->treeView = mode; -} +// ---------------------------------------- -static inline void setSortKey(ProcessList* pl, FunctionBar* fuBar, ProcessField sortKey, Panel* panel, Settings* settings) { - pl->sortKey = sortKey; - pl->direction = 1; - setTreeView(pl, fuBar, false); - settings->changed = true; - ProcessList_printHeader(pl, Panel_getHeader(panel)); -} +typedef struct CommandLineSettings_ { + Hashtable* pidWhiteList; + uid_t userId; + int sortKey; + int delay; + bool useColors; +} CommandLineSettings; -static const char* getMainPanelValue(Panel* panel, int i) { - Process* p = (Process*) Panel_get(panel, i); - if (p) - return p->comm; - return ""; -} +static CommandLineSettings parseArguments(int argc, char** argv) { -static void tagAllChildren(Panel* panel, Process* parent) { - parent->tag = true; - pid_t ppid = parent->pid; - for (int i = 0; i < Panel_size(panel); i++) { - Process* p = (Process*) Panel_get(panel, i); - if (!p->tag && p->ppid == ppid) { - tagAllChildren(panel, p); - } - } -} - -static bool expandCollapse(Panel* panel) { - Process* p = (Process*) Panel_getSelected(panel); - if (!p) return false; - p->showChildren = !p->showChildren; - return true; -} - -void sortBy(Panel* panel, ProcessList* pl, Settings* settings, int headerHeight, FunctionBar* defaultBar, Header* header) { - Panel* sortPanel = Panel_new(0, 0, 0, 0, true, Class(ListItem)); - Panel_setHeader(sortPanel, "Sort by"); - const char* fuFunctions[] = {"Sort ", "Cancel ", NULL}; - ProcessField* fields = pl->fields; - for (int i = 0; fields[i]; i++) { - char* name = String_trim(Process_fieldNames[fields[i]]); - Panel_add(sortPanel, (Object*) ListItem_new(name, fields[i])); - if (fields[i] == pl->sortKey) - Panel_setSelected(sortPanel, i); - free(name); - } - ListItem* field = (ListItem*) pickFromVector(panel, sortPanel, 15, headerHeight, fuFunctions, defaultBar, header); - if (field) { - settings->changed = true; - setSortKey(pl, defaultBar, field->key, panel, settings); - } else { - ProcessList_printHeader(pl, Panel_getHeader(panel)); - } - Object_delete(sortPanel); -} - -static void millisleep(unsigned long millisec) { - struct timespec req = { - .tv_sec = 0, - .tv_nsec = millisec * 1000000L + CommandLineSettings flags = { + .pidWhiteList = NULL, + .userId = -1, // -1 is guaranteed to be an invalid uid_t (see setreuid(2)) + .sortKey = 0, + .delay = -1, + .useColors = true, }; - while(nanosleep(&req,&req)==-1) { - continue; - } -} - -int main(int argc, char** argv) { - int delay = -1; - bool userOnly = false; - uid_t userId = 0; - int usecolors = 1; - char *argCopy; - char *pid; - Hashtable *pidWhiteList = NULL; - - int opt, opti=0; static struct option long_opts[] = { {"help", no_argument, 0, 'h'}, @@ -371,20 +83,13 @@ int main(int argc, char** argv) { {"no-color", no_argument, 0, 'C'}, {"no-colour",no_argument, 0, 'C'}, {"pid", required_argument, 0, 'p'}, + {"io", no_argument, 0, 'i'}, {0,0,0,0} }; - int sortKey = 0; - - char *lc_ctype = getenv("LC_CTYPE"); - if(lc_ctype != NULL) - setlocale(LC_CTYPE, lc_ctype); - else if ((lc_ctype = getenv("LC_ALL"))) - setlocale(LC_CTYPE, lc_ctype); - else - setlocale(LC_CTYPE, ""); + int opt, opti=0; /* Parse arguments */ - while ((opt = getopt_long(argc, argv, "hvCs:d:u:p:", long_opts, &opti))) { + while ((opt = getopt_long(argc, argv, "hvCs:d:u:p:i", long_opts, &opti))) { if (opt == EOF) break; switch (opt) { case 'h': @@ -395,47 +100,45 @@ int main(int argc, char** argv) { break; case 's': if (strcmp(optarg, "help") == 0) { - for (int j = 1; j < LAST_PROCESSFIELD; j++) - printf ("%s\n", Process_fieldNames[j]); + for (int j = 1; j < Platform_numberOfFields; j++) { + const char* name = Process_fields[j].name; + if (name) printf ("%s\n", name); + } exit(0); } - - sortKey = ColumnsPanel_fieldNameToIndex(optarg); - if (sortKey == -1) { + flags.sortKey = ColumnsPanel_fieldNameToIndex(optarg); + if (flags.sortKey == -1) { fprintf(stderr, "Error: invalid column \"%s\".\n", optarg); - exit(1); } break; case 'd': - if (sscanf(optarg, "%16d", &delay) == 1) { - if (delay < 1) delay = 1; - if (delay > 100) delay = 100; + if (sscanf(optarg, "%16d", &(flags.delay)) == 1) { + if (flags.delay < 1) flags.delay = 1; + if (flags.delay > 100) flags.delay = 100; } else { fprintf(stderr, "Error: invalid delay value \"%s\".\n", optarg); - exit(1); } break; case 'u': - if (!setUserOnly(optarg, &userOnly, &userId)) { + if (!Action_setUserOnly(optarg, &(flags.userId))) { fprintf(stderr, "Error: invalid user \"%s\".\n", optarg); - exit(1); } break; case 'C': - usecolors=0; + flags.useColors = false; break; - case 'p': { - argCopy = strdup(optarg); + case 'p': { + char* argCopy = xStrdup(optarg); char* saveptr; - pid = strtok_r(argCopy, ",", &saveptr); + char* pid = strtok_r(argCopy, ",", &saveptr); - if( !pidWhiteList ) { - pidWhiteList = Hashtable_new(8, false); + if(!flags.pidWhiteList) { + flags.pidWhiteList = Hashtable_new(8, false); } - while( pid ) { + while(pid) { unsigned int num_pid = atoi(pid); - Hashtable_put(pidWhiteList, num_pid, (void *) 1); + Hashtable_put(flags.pidWhiteList, num_pid, (void *) 1); pid = strtok_r(NULL, ",", &saveptr); } free(argCopy); @@ -446,70 +149,87 @@ int main(int argc, char** argv) { exit(1); } } + return flags; +} + +static void millisleep(unsigned long millisec) { + struct timespec req = { + .tv_sec = 0, + .tv_nsec = millisec * 1000000L + }; + while(nanosleep(&req,&req)==-1) { + continue; + } +} + +int main(int argc, char** argv) { + char *lc_ctype = getenv("LC_CTYPE"); + if(lc_ctype != NULL) + setlocale(LC_CTYPE, lc_ctype); + else if ((lc_ctype = getenv("LC_ALL"))) + setlocale(LC_CTYPE, lc_ctype); + else + setlocale(LC_CTYPE, ""); + + CommandLineSettings flags = parseArguments(argc, argv); // may exit() +#ifdef HAVE_PROC if (access(PROCDIR, R_OK) != 0) { - fprintf(stderr, "Error: could not read procfs (compiled to look in %s).\n", PROCDIR); - exit(1); + fprintf("Error: could not read procfs (compiled to look in %s).\n", PROCDIR); } - - int quit = 0; - int refreshTimeout = 0; - int resetRefreshTimeout = 5; - bool doRefresh = true; - bool doRecalculate = false; - Settings* settings; +#endif + + Process_setupColumnWidths(); - ProcessList* pl = NULL; UsersTable* ut = UsersTable_new(); + ProcessList* pl = ProcessList_new(ut, flags.pidWhiteList, flags.userId); + + Settings* settings = Settings_new(pl->cpuCount); + pl->settings = settings; -#ifdef HAVE_LIBNCURSESW - char *locale = setlocale(LC_ALL, NULL); - if (locale == NULL || locale[0] == '\0') - locale = setlocale(LC_CTYPE, NULL); - if (locale != NULL && - (strstr(locale, "UTF-8") || - strstr(locale, "utf-8") || - strstr(locale, "UTF8") || - strstr(locale, "utf8"))) - CRT_utf8 = true; - else - CRT_utf8 = false; -#endif + Header* header = Header_new(pl, settings, 2); - pl = ProcessList_new(ut, pidWhiteList); - Process_getMaxPid(); - - Header* header = Header_new(pl); - settings = Settings_new(pl, header, pl->cpuCount); - int headerHeight = Header_calculateHeight(header); - - // FIXME: move delay code to settings - if (delay != -1) - settings->delay = delay; - if (!usecolors) + Header_populateFromSettings(header); + + if (flags.delay != -1) + settings->delay = flags.delay; + if (!flags.useColors) settings->colorScheme = COLORSCHEME_MONOCHROME; CRT_init(settings->delay, settings->colorScheme); - - Panel* panel = Panel_new(0, headerHeight, COLS, LINES - headerHeight - 2, false, &Process_class); - ProcessList_setPanel(pl, panel); - FunctionBar* defaultBar = FunctionBar_new(defaultFunctions, NULL, NULL); - setTreeView(pl, defaultBar, pl->treeView); - - if (sortKey > 0) { - pl->sortKey = sortKey; - setTreeView(pl, defaultBar, false); - pl->direction = 1; - } - ProcessList_printHeader(pl, Panel_getHeader(panel)); + MainPanel* panel = MainPanel_new(); + ProcessList_setPanel(pl, (Panel*) panel); - IncSet* inc = IncSet_new(defaultBar); + MainPanel_updateTreeFunctions(panel, settings->treeView); + + if (flags.sortKey > 0) { + settings->sortKey = flags.sortKey; + settings->treeView = false; + settings->direction = 1; + } + ProcessList_printHeader(pl, Panel_getHeader((Panel*)panel)); + + State state = { + .settings = settings, + .ut = ut, + .pl = pl, + .panel = (Panel*) panel, + .header = header, + }; + MainPanel_setState(panel, &state); + + ScreenManager* scr = ScreenManager_new(0, header->height, 0, -1, HORIZONTAL, header, settings, true); + ScreenManager_add(scr, (Panel*) panel, -1); ProcessList_scan(pl); millisleep(75); + ProcessList_scan(pl); + + ScreenManager_run(scr, NULL, NULL); + /* FunctionBar_draw(defaultBar, NULL); int acc = 0; @@ -521,34 +241,45 @@ int main(int argc, char** argv) { int ch = ERR; int closeTimeout = 0; - bool idle = false; + bool drawPanel = true; bool collapsed = false; + Htop_Action keys[KEY_MAX] = { NULL }; + setBindings(keys); + Platform_setBindings(keys); + + bool quit = false; + int sortTimeout = 0; + int resetSortTimeout = 5; + bool doRefresh = true; + bool forceRecalculate = false; + while (!quit) { gettimeofday(&tv, NULL); double newTime = ((double)tv.tv_sec * 10) + ((double)tv.tv_usec / 100000); - bool recalculate = (newTime - oldTime > settings->delay); - int following = follow ? selectedPid(panel) : -1; - if (recalculate) { + bool timeToRecalculate = (newTime - oldTime > settings->delay); + if (newTime < oldTime) timeToRecalculate = true; // clock was adjusted? + int following = follow ? MainPanel_selectedPid((MainPanel*)panel) : -1; + if (timeToRecalculate) { Header_draw(header); oldTime = newTime; } if (doRefresh) { - if (recalculate || doRecalculate) { + if (timeToRecalculate || forceRecalculate) { ProcessList_scan(pl); - doRecalculate = false; + forceRecalculate = false; } - if (refreshTimeout == 0 || pl->treeView) { + if (sortTimeout == 0 || settings->treeView) { ProcessList_sort(pl); - refreshTimeout = 1; + sortTimeout = 1; } - ProcessList_rebuildPanel(pl, true, following, userOnly, userId, IncSet_filter(inc)); - idle = false; + ProcessList_rebuildPanel(pl, true, following, IncSet_filter(inc)); + drawPanel = true; } doRefresh = true; - if (pl->treeView) { + if (settings->treeView) { Process* p = (Process*) Panel_getSelected(panel); if (p) { if (!p->showChildren && !collapsed) { @@ -560,9 +291,11 @@ int main(int argc, char** argv) { } collapsed = !p->showChildren; } + } else { + collapsed = false; } - if (!idle) { + if (drawPanel) { Panel_draw(panel, true); } @@ -573,18 +306,20 @@ int main(int argc, char** argv) { if (ch == ERR) { if (!inc->active) - refreshTimeout--; - if (prev == ch && !recalculate) { + sortTimeout--; + if (prev == ch && !timeToRecalculate) { closeTimeout++; if (closeTimeout == 100) { break; } } else closeTimeout = 0; - idle = true; + drawPanel = false; continue; } - idle = false; + drawPanel = true; + + Htop_Reaction reaction = HTOP_OK; if (ch == KEY_MOUSE) { MEVENT mevent; @@ -594,20 +329,19 @@ int main(int argc, char** argv) { if (mevent.y == panel->y) { int x = panel->scrollH + mevent.x + 1; ProcessField field = ProcessList_keyAt(pl, x); - if (field == pl->sortKey) { - ProcessList_invertSortOrder(pl); - setTreeView(pl, defaultBar, false); + if (field == settings->sortKey) { + Settings_invertSortOrder(settings); + settings->treeView = false; + reaction |= HTOP_REDRAW_BAR; } else { - setSortKey(pl, defaultBar, field, panel, settings); + reaction |= setSortKey(settings, field); } - refreshTimeout = 0; - continue; + sortTimeout = 0; + ch = ERR; } else if (mevent.y >= panel->y + 1 && mevent.y < LINES - 1) { Panel_setSelected(panel, mevent.y - panel->y + panel->scrollV - 1); - doRefresh = false; - refreshTimeout = resetRefreshTimeout; follow = true; - continue; + ch = ERR; } if (mevent.y == LINES - 1) { ch = FunctionBar_synthesizeEvent(inc->bar, mevent.x); } @@ -623,10 +357,13 @@ int main(int argc, char** argv) { if (inc->active) { doRefresh = IncSet_handleKey(inc, ch, panel, getMainPanelValue, NULL); + if (!inc->active) { + follow = true; + } continue; } - if (isdigit((char)ch)) { + if (ch < 255 && isdigit((char)ch)) { if (Panel_size(panel) == 0) continue; pid_t pid = ch-48 + acc; for (int i = 0; i < ProcessList_size(pl); i++) { @@ -644,298 +381,42 @@ int main(int argc, char** argv) { acc = 0; } - switch (ch) { - case KEY_RESIZE: - Panel_resize(panel, COLS, LINES-headerHeight-1); - IncSet_drawBar(inc); - break; - case 'M': - { - refreshTimeout = 0; - setSortKey(pl, defaultBar, PERCENT_MEM, panel, settings); - break; - } - case 'T': - { - refreshTimeout = 0; - setSortKey(pl, defaultBar, TIME, panel, settings); - break; - } - case 'c': - { - Process* p = (Process*) Panel_getSelected(panel); - if (!p) break; - tagAllChildren(panel, p); - break; - } - case 'U': - { - for (int i = 0; i < Panel_size(panel); i++) { - Process* p = (Process*) Panel_get(panel, i); - p->tag = false; - } - doRefresh = true; - break; - } - case 'P': - { - refreshTimeout = 0; - setSortKey(pl, defaultBar, PERCENT_CPU, panel, settings); - break; - } - case KEY_F(1): - case 'h': - case '?': - { - showHelp(pl); - FunctionBar_draw(defaultBar, NULL); - refreshTimeout = 0; - break; - } - case '\014': // Ctrl+L - { - clear(); - FunctionBar_draw(defaultBar, NULL); - refreshTimeout = 0; - break; - } - case ' ': - { - Process* p = (Process*) Panel_getSelected(panel); - if (!p) break; - Process_toggleTag(p); - Panel_onKey(panel, KEY_DOWN); - break; - } - case 's': - { - Process* p = (Process*) Panel_getSelected(panel); - if (!p) break; - TraceScreen* ts = TraceScreen_new(p); - TraceScreen_run(ts); - TraceScreen_delete(ts); - clear(); - FunctionBar_draw(defaultBar, NULL); - refreshTimeout = 0; - CRT_enableDelay(); - break; - } - case 'l': - { - Process* p = (Process*) Panel_getSelected(panel); - if (!p) break; - OpenFilesScreen* ts = OpenFilesScreen_new(p); - OpenFilesScreen_run(ts); - OpenFilesScreen_delete(ts); - clear(); - FunctionBar_draw(defaultBar, NULL); - refreshTimeout = 0; - CRT_enableDelay(); - break; - } - case 'S': - case 'C': - case KEY_F(2): - { - Setup_run(settings, header); - // TODO: shouldn't need this, colors should be dynamic - ProcessList_printHeader(pl, Panel_getHeader(panel)); - headerHeight = Header_calculateHeight(header); - Panel_move(panel, 0, headerHeight); - Panel_resize(panel, COLS, LINES-headerHeight-1); - FunctionBar_draw(defaultBar, NULL); - refreshTimeout = 0; - break; - } - case 'F': - { - follow = true; - continue; - } - case 'u': - { - Panel* usersPanel = Panel_new(0, 0, 0, 0, true, Class(ListItem)); - Panel_setHeader(usersPanel, "Show processes of:"); - UsersTable_foreach(ut, addUserToVector, usersPanel); - Vector_insertionSort(usersPanel->items); - ListItem* allUsers = ListItem_new("All users", -1); - Panel_insert(usersPanel, 0, (Object*) allUsers); - const char* fuFunctions[] = {"Show ", "Cancel ", NULL}; - ListItem* picked = (ListItem*) pickFromVector(panel, usersPanel, 20, headerHeight, fuFunctions, defaultBar, header); - if (picked) { - if (picked == allUsers) { - userOnly = false; - } else { - setUserOnly(ListItem_getRef(picked), &userOnly, &userId); - } - } - Panel_delete((Object*)usersPanel); - break; - } - case '+': - case '=': - case '-': - { - if (expandCollapse(panel)) { - doRecalculate = true; - refreshTimeout = 0; - } - break; - } - case KEY_F(9): - case 'k': - { - Panel* signalsPanel = (Panel*) SignalsPanel_new(); - const char* fuFunctions[] = {"Send ", "Cancel ", NULL}; - ListItem* sgn = (ListItem*) pickFromVector(panel, signalsPanel, 15, headerHeight, fuFunctions, defaultBar, header); - if (sgn) { - if (sgn->key != 0) { - Panel_setHeader(panel, "Sending..."); - Panel_draw(panel, true); - refresh(); - foreachProcess(panel, (ForeachProcessFn) Process_sendSignal, (size_t) sgn->key, NULL); - napms(500); - } - } - ProcessList_printHeader(pl, Panel_getHeader(panel)); - Panel_delete((Object*)signalsPanel); - refreshTimeout = 0; - break; + if(ch != ERR && keys[ch]) { + reaction |= (keys[ch])(&state); + } else { + doRefresh = false; + sortTimeout = resetSortTimeout; + Panel_onKey(panel, ch); } -#if (HAVE_LIBHWLOC || HAVE_NATIVE_AFFINITY) - case 'a': - { - if (pl->cpuCount == 1) - break; + + // Reaction handlers: - Process* p = (Process*) Panel_getSelected(panel); - if (!p) break; - Affinity* affinity = Process_getAffinity(p); - if (!affinity) break; - Panel* affinityPanel = AffinityPanel_new(pl, affinity); - Affinity_delete(affinity); - - const char* fuFunctions[] = {"Set ", "Cancel ", NULL}; - void* set = pickFromVector(panel, affinityPanel, 15, headerHeight, fuFunctions, defaultBar, header); - if (set) { - Affinity* affinity = AffinityPanel_getAffinity(affinityPanel); - bool ok = foreachProcess(panel, (ForeachProcessFn) Process_setAffinity, (size_t) affinity, NULL); - if (!ok) beep(); - Affinity_delete(affinity); - } - Panel_delete((Object*)affinityPanel); - ProcessList_printHeader(pl, Panel_getHeader(panel)); - refreshTimeout = 0; - break; - } -#endif - case KEY_F(10): - case 'q': - quit = 1; - break; - case '<': - case ',': - case '>': - case '.': - { - sortBy(panel, pl, settings, headerHeight, defaultBar, header); - refreshTimeout = 0; - break; - } - case KEY_F(18): - case KEY_F(6): - { - if (pl->treeView) { - if (expandCollapse(panel)) { - doRecalculate = true; - refreshTimeout = 0; - } - } else { - sortBy(panel, pl, settings, headerHeight, defaultBar, header); - refreshTimeout = 0; - } - break; + if (reaction & HTOP_REDRAW_BAR) { + updateTreeFunctions(defaultBar, settings->treeView); + IncSet_drawBar(inc); } - case 'i': - { - Process* p = (Process*) Panel_getSelected(panel); - if (!p) break; - IOPriority ioprio = p->ioPriority; - Panel* ioprioPanel = IOPriorityPanel_new(ioprio); - const char* fuFunctions[] = {"Set ", "Cancel ", NULL}; - void* set = pickFromVector(panel, ioprioPanel, 21, headerHeight, fuFunctions, defaultBar, header); - if (set) { - IOPriority ioprio = IOPriorityPanel_getIOPriority(ioprioPanel); - bool ok = foreachProcess(panel, (ForeachProcessFn) Process_setIOPriority, (size_t) ioprio, NULL); - if (!ok) - beep(); - } - Panel_delete((Object*)ioprioPanel); + if (reaction & HTOP_UPDATE_PANELHDR) { ProcessList_printHeader(pl, Panel_getHeader(panel)); - refreshTimeout = 0; - break; - } - case 'I': - { - refreshTimeout = 0; - settings->changed = true; - ProcessList_invertSortOrder(pl); - break; } - case KEY_F(8): - case '[': - { - doRefresh = changePriority(panel, 1); - break; - } - case KEY_F(7): - case ']': - { - doRefresh = changePriority(panel, -1); - break; - } - case KEY_F(3): - case '/': - IncSet_activate(inc, INC_SEARCH); - break; - case KEY_F(4): - case '\\': - IncSet_activate(inc, INC_FILTER); - refreshTimeout = 0; + if (reaction & HTOP_REFRESH) { doRefresh = true; - continue; - case 't': - case KEY_F(5): - refreshTimeout = 0; - collapsed = false; - setTreeView(pl, defaultBar, !pl->treeView); - if (pl->treeView) pl->direction = 1; - ProcessList_printHeader(pl, Panel_getHeader(panel)); - ProcessList_expandTree(pl); - settings->changed = true; - if (following != -1) continue; - break; - case 'H': - doRecalculate = true; - refreshTimeout = 0; - pl->hideUserlandThreads = !pl->hideUserlandThreads; - pl->hideThreads = pl->hideUserlandThreads; - settings->changed = true; - break; - case 'K': - doRecalculate = true; - refreshTimeout = 0; - pl->hideKernelThreads = !pl->hideKernelThreads; + sortTimeout = 0; + } + if (reaction & HTOP_RECALCULATE) { + forceRecalculate = true; + sortTimeout = 0; + } + if (reaction & HTOP_SAVE_SETTINGS) { settings->changed = true; - break; - default: - doRefresh = false; - refreshTimeout = resetRefreshTimeout; - Panel_onKey(panel, ch); - break; } - follow = false; + if (reaction & HTOP_QUIT) { + quit = true; + } + follow = (reaction & HTOP_KEEP_FOLLOWING); } + + */ + attron(CRT_colors[RESET_COLOR]); mvhline(LINES-1, 0, ' ', COLS); attroff(CRT_colors[RESET_COLOR]); @@ -946,13 +427,19 @@ int main(int argc, char** argv) { Settings_write(settings); Header_delete(header); ProcessList_delete(pl); - IncSet_delete(inc); + + /* FunctionBar_delete((Object*)defaultBar); Panel_delete((Object*)panel); + */ + + ScreenManager_delete(scr); + UsersTable_delete(ut); Settings_delete(settings); - if(pidWhiteList) { - Hashtable_delete(pidWhiteList); + + if(flags.pidWhiteList) { + Hashtable_delete(flags.pidWhiteList); } return 0; } -- cgit v1.2.3