htop.c

/*
htop - htop.c
(C) 2004-2011 Hisham H. Muhammad
Released under the GNU GPL, see the COPYING file
in the source distribution for its full text.
*/

#include "config.h"

#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"

#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>

//#link m

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"
         "   -C, --no-color              Use a monochrome color scheme\n"
         "   -d, --delay=DELAY           Set the delay between updates, in tenths of 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"
         "   -v, --version               Print version info\n"
         "\n"
         "Arguments to long options are required for short options too.\n\n"
         "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;
} CommandLineSettings;

static CommandLineSettings parseArguments(int argc, char** argv) {

   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,
   };

#ifdef HAVE_GETOPT_LONG
   static struct option long_opts[] =
   {
      {"help",     no_argument,         0, 'h'},
      {"version",  no_argument,         0, 'v'},
      {"delay",    required_argument,   0, 'd'},
      {"sort-key", required_argument,   0, 's'},
      {"user",     required_argument,   0, 'u'},
      {"no-color", no_argument,         0, 'C'},
      {"no-colour",no_argument,         0, 'C'},
      {"tree",     no_argument,         0, 't'},
      {"pid",      required_argument,   0, 'p'},
      {0,0,0,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 'h':
            print_usage(stdout, argv[0]);
            exit(0);
         case 'v':
            printVersionFlag();
            exit(0);
         case 's':
            if (strcmp(optarg, "help") == 0) {
               for (int j = 1; j < Platform_numberOfFields; j++) {
                  const char* name = Process_fields[j].name;
                  if (name) puts(name);
               }
               exit(0);
            }
            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", &(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);
      }
   }
   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);
   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;

   CRT_init(settings);
   CRT_setMouse(settings->use_mouse);
   MainPanel* panel = MainPanel_new();
   ProcessList_setPanel(pl, (Panel*) panel);

   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,
      .repeat = 1
   };
   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);
   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);

   ScreenManager_delete(scr);
   UsersTable_delete(ut);
   Settings_delete(settings);
   if(flags.pidWhiteList) {
      Hashtable_delete(flags.pidWhiteList);
   }
   return 0;
}