| /* |
| htop - htop.c |
| (C) 2004-2011 Hisham H. Muhammad |
| Copyright 2015-2023 Rivoreo |
| Released under the GNU GPL, see the COPYING file |
| in the source distribution for its full text. |
| */ |
| |
| #include "config.h" |
| |
| #if defined DISK_STATS && !defined HAVE_GETOPT_LONG |
| #error "Disk statistics mode requires getopt_long(3)" |
| #endif |
| |
| #include "FunctionBar.h" |
| #include "Hashtable.h" |
| #include "ColumnsPanel.h" |
| #include "CRT.h" |
| #include "MainPanel.h" |
| #include "ProcessList.h" |
| #include "ScreenManager.h" |
| #include "Settings.h" |
| #include "UsersTable.h" |
| #include "Platform.h" |
| #ifdef DISK_STATS |
| #include "DiskPanel.h" |
| #include "DiskList.h" |
| #include "Disk.h" |
| #endif |
| #if defined HAVE_GETOPT_H && defined HAVE_GETOPT_LONG |
| #include <getopt.h> |
| #endif |
| #include <locale.h> |
| #include <stdio.h> |
| #include <stdlib.h> |
| #include <string.h> |
| #include <time.h> |
| #include <unistd.h> |
| |
| static void printVersionFlag() { |
| fputs("htop " VERSION "\n" COPYRIGHT "\n" |
| "Released under the GNU GPL any version.\n", |
| stdout); |
| } |
| |
| static void print_usage(FILE *f, const char *name) { |
| fprintf(f, |
| "Usage: %s [OPTION]...\n\n" |
| "Options:\n" |
| #ifdef HAVE_GETOPT_LONG |
| " -C, --no-color Use a monochrome color scheme\n" |
| " -d, --delay=DELAY Set the delay between updates, in tenths of\n" |
| " seconds\n" |
| " -h, --help Print this help screen\n" |
| " -s, --sort-key=COLUMN Sort by COLUMN (try --sort-key=help for a list)\n" |
| " -t, --tree Show the tree view by default\n" |
| " -u, --user=USERNAME Show only processes of a given user\n" |
| " -p, --pid=PID[,PID,PID...] Show only the given PIDs\n" |
| #ifdef DISK_STATS |
| " --disk Show disk instead of process statistics\n" |
| #endif |
| " --explicit-delay Explicitly delay between updates\n" |
| " -v, --version Print version info\n" |
| "\nArguments to long options are required for short options too.\n\n" |
| #else |
| " -C Use a monochrome color scheme\n" |
| " -d DELAY Set the delay between updates, in tenths of seconds\n" |
| " -h Print this help screen\n" |
| " -s COLUMN Sort by COLUMN (try --sort-key=help for a list)\n" |
| " -t Show the tree view by default\n" |
| " -u USERNAME Show only processes of a given user\n" |
| " -p PID[,PID,PID...] Show only the given PIDs\n" |
| " -v Print version info\n" |
| "\n" |
| #endif |
| "Press F1 inside htop for online help.\n" |
| "See 'man htop' for more information.\n", |
| name); |
| } |
| |
| // ---------------------------------------- |
| |
| typedef struct CommandLineSettings_ { |
| Hashtable* pidWhiteList; |
| uid_t userId; |
| int sortKey; |
| int delay; |
| bool useColors; |
| bool treeView; |
| bool explicit_delay; |
| #ifdef DISK_STATS |
| bool disk; |
| #endif |
| } CommandLineSettings; |
| |
| static CommandLineSettings parseArguments(int argc, char** argv) { |
| const char *sort_key = NULL; |
| CommandLineSettings flags = { |
| .pidWhiteList = NULL, |
| .userId = -1, // -1 is guaranteed to be an invalid uid_t (see setreuid(2)) |
| .sortKey = 0, |
| .delay = -1, |
| .useColors = true, |
| .treeView = false, |
| .explicit_delay = false, |
| }; |
| |
| #define HTOP_LONG_OPTION_EXPLICIT_DELAY (1 << 8) |
| #define HTOP_LONG_OPTION_DISK (2 << 8) |
| #ifdef HAVE_GETOPT_LONG |
| static struct option long_opts[] = { |
| { "help", no_argument, NULL, 'h' }, |
| { "version", no_argument, NULL, 'v' }, |
| { "delay", required_argument, NULL, 'd' }, |
| { "sort-key", required_argument, NULL, 's' }, |
| { "user", required_argument, NULL, 'u' }, |
| { "no-color", no_argument, NULL, 'C' }, |
| { "no-colour", no_argument, NULL, 'C' }, |
| { "tree", no_argument, NULL, 't' }, |
| { "pid", required_argument, NULL, 'p' }, |
| { "explicit-delay", no_argument, NULL, HTOP_LONG_OPTION_EXPLICIT_DELAY }, |
| #ifdef DISK_STATS |
| { "disk", no_argument, NULL, HTOP_LONG_OPTION_DISK }, |
| #endif |
| { NULL, 0, NULL, 0 } |
| }; |
| #endif |
| /* Parse options */ |
| while(true) { |
| #ifdef HAVE_GETOPT_LONG |
| int opt = getopt_long(argc, argv, "hvCs:td:u:p:", long_opts, NULL); |
| #else |
| int opt = getopt(argc, argv, "hvCs:td:u:p:"); |
| #endif |
| if (opt == -1) break; |
| switch (opt) { |
| char *p; |
| case HTOP_LONG_OPTION_EXPLICIT_DELAY: |
| flags.explicit_delay = true; |
| break; |
| #ifdef DISK_STATS |
| case HTOP_LONG_OPTION_DISK: |
| flags.disk = true; |
| break; |
| #endif |
| case 'h': |
| print_usage(stdout, argv[0]); |
| exit(0); |
| case 'v': |
| printVersionFlag(); |
| exit(0); |
| case 's': |
| sort_key = optarg; |
| break; |
| case 'd': |
| 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 (!Action_setUserOnly(optarg, &(flags.userId))) { |
| fprintf(stderr, "Error: invalid user \"%s\".\n", optarg); |
| exit(-1); |
| } |
| break; |
| case 'C': |
| flags.useColors = false; |
| break; |
| case 't': |
| flags.treeView = true; |
| break; |
| case 'p': |
| if(!flags.pidWhiteList) { |
| flags.pidWhiteList = Hashtable_new(8, false); |
| } |
| p = optarg; |
| do { |
| long int pid = strtol(p, &p, 0); |
| if(*p && *p != ',') { |
| fprintf(stderr, "Error: invalid PID list \"%s\".\n", optarg); |
| exit(-1); |
| } |
| if(pid < 0) { |
| fprintf(stderr, "Error: invalid PID %ld.\n", pid); |
| exit(-1); |
| } |
| Hashtable_put(flags.pidWhiteList, pid, (void *) 1); |
| } while(*p++); |
| break; |
| default: |
| #ifdef __ANDROID__ |
| fputc('\n', stderr); |
| #endif |
| exit(-1); |
| } |
| } |
| #ifdef DISK_STATS |
| if(flags.disk) { |
| if(flags.pidWhiteList) { |
| fputs("Error: cannot specific PIDs for disk statistics mode\n", stderr); |
| exit(-1); |
| } |
| if(flags.userId != (uid_t)-1) { |
| fputs("Error: cannot specific user for disk statistics mode\n", stderr); |
| exit(-1); |
| } |
| if(flags.treeView) { |
| fputs("Error: disk statistics mode doesn't support tree view\n", stderr); |
| exit(-1); |
| } |
| } |
| #endif |
| if(sort_key) { |
| #ifdef DISK_STATS |
| const FieldData *field_data = flags.disk ? Disk_fields : Process_fields; |
| unsigned int nfields = flags.disk ? Disk_field_count : Platform_numberOfFields; |
| #else |
| const FieldData *field_data = Process_fields; |
| unsigned int nfields = Platform_numberOfFields; |
| #endif |
| if (strcmp(sort_key, "help") == 0) { |
| for (unsigned int j = 1; j < nfields; j++) { |
| const char* name = field_data[j].name; |
| if (name) puts(name); |
| } |
| exit(0); |
| } |
| flags.sortKey = ColumnsPanel_fieldNameToIndex(field_data, nfields, sort_key); |
| if (flags.sortKey == -1) { |
| fprintf(stderr, "Error: invalid column \"%s\".\n", sort_key); |
| exit(-1); |
| } |
| } |
| return flags; |
| } |
| |
| static void millisleep(unsigned long millisec) { |
| #ifdef HAVE_NANOSLEEP |
| struct timespec req = { |
| .tv_sec = 0, |
| .tv_nsec = millisec * 1000000L |
| }; |
| while(nanosleep(&req,&req)==-1) { |
| continue; |
| } |
| #else |
| unsigned int sec = millisec / 1000; |
| if(sec) sleep(sec); |
| millisec %= 1000; |
| usleep(millisec); |
| #endif |
| } |
| |
| int main(int argc, char** argv) { |
| const char *lc_ctype = getenv("LC_CTYPE"); |
| if(!lc_ctype) { |
| lc_ctype = getenv("LC_ALL"); |
| if(!lc_ctype) lc_ctype = ""; |
| } |
| setlocale(LC_CTYPE, lc_ctype); |
| |
| CommandLineSettings flags = parseArguments(argc, argv); // may exit(3) |
| |
| #ifdef HAVE_PROC |
| if (access(PROCDIR, R_OK) < 0) { |
| fprintf(stderr, "Error: could not read procfs (compiled to look in %s).\n", PROCDIR); |
| return 1; |
| } |
| #endif |
| CRT_initColorSchemes(); |
| Process_setupColumnWidths(); |
| UsersTable* ut = UsersTable_new(); |
| ProcessList* pl = ProcessList_new(ut, flags.pidWhiteList, flags.userId); |
| Settings* settings = Settings_new(pl->cpuCount, Platform_haveSwap()); |
| pl->settings = settings; |
| |
| Header* header = Header_new(pl, settings, 2); |
| |
| Header_populateFromSettings(header); |
| |
| if (flags.delay != -1) settings->delay = flags.delay; |
| // Use built-in MONOCHROME_COLOR_SCHEME for '--no-color' |
| if (!flags.useColors) settings->colorScheme = MONOCHROME_COLOR_SCHEME; |
| if (flags.treeView) settings->treeView = true; |
| if(flags.explicit_delay) settings->explicit_delay = true; |
| |
| CRT_init(settings); |
| CRT_setMouse(settings->use_mouse); |
| CRT_setExplicitDelay(settings->explicit_delay); |
| |
| Panel *panel; |
| #ifdef DISK_STATS |
| DiskList *disk_list = NULL; |
| settings->disk_mode = flags.disk; |
| if(flags.disk) { |
| disk_list = DiskList_new(settings); |
| panel = (Panel *)DiskPanel_new(settings, disk_list, header); |
| DiskList_setPanel(disk_list, panel); |
| header->disk_list = disk_list; |
| } else |
| #endif |
| { |
| MainPanel *process_panel = MainPanel_new(); |
| panel = (Panel *)process_panel; |
| ProcessList_setPanel(pl, panel); |
| MainPanel_updateTreeFunctions(process_panel, settings->treeView); |
| } |
| |
| if (flags.sortKey > 0) { |
| #ifdef DISK_STATS |
| if(flags.disk) settings->disk_sort_key = flags.sortKey; |
| else |
| #endif |
| { |
| settings->sortKey = flags.sortKey; |
| settings->treeView = false; |
| } |
| settings->direction = 1; |
| } |
| #ifdef DISK_STATS |
| if(flags.disk) DiskList_printHeader(disk_list, Panel_getHeader(panel)); |
| else |
| #endif |
| ProcessList_printHeader(pl, Panel_getHeader(panel)); |
| |
| #ifdef DISK_STATS |
| if(!flags.disk) |
| #endif |
| { |
| State state = { |
| .settings = settings, |
| .ut = ut, |
| .pl = pl, |
| .panel = panel, |
| .header = header, |
| .repeat = 1 |
| }; |
| MainPanel_setState((MainPanel *)panel, &state); |
| } |
| |
| ScreenManager* scr = ScreenManager_new(0, header->height, 0, -1, HORIZONTAL, header, settings, true); |
| ScreenManager_add(scr, panel, -1); |
| |
| #ifdef DISK_STATS |
| if(flags.disk) { |
| DiskList_scan(disk_list, 0); |
| ProcessList_scan(pl, true); |
| } else |
| #endif |
| ProcessList_scan(pl, false); |
| millisleep(75); |
| #ifdef DISK_STATS |
| ProcessList_scan(pl, flags.disk); |
| #else |
| ProcessList_scan(pl, false); |
| #endif |
| |
| ScreenManager_run(scr, NULL, NULL); |
| attron(CRT_colors[HTOP_DEFAULT_COLOR]); |
| mvhline(LINES-1, 0, ' ', COLS); |
| attroff(CRT_colors[HTOP_DEFAULT_COLOR]); |
| refresh(); |
| CRT_done(); |
| if (settings->changed) Settings_write(settings); |
| Header_delete(header); |
| ProcessList_delete(pl); |
| #ifdef DISK_STATS |
| if(flags.disk) DiskList_delete(disk_list); |
| #endif |
| |
| ScreenManager_delete(scr); |
| UsersTable_delete(ut); |
| Settings_delete(settings); |
| if(flags.pidWhiteList) { |
| Hashtable_delete(flags.pidWhiteList); |
| } |
| |
| return 0; |
| } |