Allow setting the priority of spawned subprocess

src/build.cc

@@ -483,7 +483,8 @@ bool RealCommandRunner::CanRunMore() const {
 
 bool RealCommandRunner::StartCommand(Edge* edge) {
   string command = edge->EvaluateCommand();
-  Subprocess* subproc = subprocs_.Add(command, edge->use_console());
+  Subprocess* subproc =
+      subprocs_.Add(command, config_.subprocess_priority, edge->use_console());
   if (!subproc)
     return false;
   subproc_to_edge_.insert(make_pair(subproc, edge));

src/build.h

@@ -15,6 +15,7 @@
 #ifndef NINJA_BUILD_H_
 #define NINJA_BUILD_H_
 
+#include <climits>
 #include <cstdio>
 #include <map>
 #include <memory>
@@ -155,8 +156,8 @@ struct CommandRunner {
 
 /// Options (e.g. verbosity, parallelism) passed to a build.
 struct BuildConfig {
-  BuildConfig() : verbosity(NORMAL), dry_run(false), parallelism(1),
-                  failures_allowed(1), max_load_average(-0.0f) {}
+  BuildConfig() : verbosity(NORMAL), dry_run(false), parallelism(1), failures_allowed(1),
+        subprocess_priority(INT_MIN), max_load_average(-0.0f) {}
 
   enum Verbosity {
     QUIET,  // No output -- used when testing.
@@ -168,6 +169,7 @@ struct BuildConfig {
   bool dry_run;
   int parallelism;
   int failures_allowed;
+  int subprocess_priority;
   /// The maximum load average we must not exceed. A negative value
   /// means that we do not have any limit.
   double max_load_average;

src/ninja.cc

@@ -223,18 +223,19 @@ void Usage(const BuildConfig& config) {
 "  -v, --verbose  show all command lines while building\n"
 "  --quiet        don't show progress status, just command output\n"
 "\n"
-"  -C DIR   change to DIR before doing anything else\n"
-"  -f FILE  specify input build file [default=build.ninja]\n"
+"  -C DIR         change to DIR before doing anything else\n"
+"  -f FILE        specify input build file [default=build.ninja]\n"
 "\n"
-"  -j N     run N jobs in parallel (0 means infinity) [default=%d on this system]\n"
-"  -k N     keep going until N jobs fail (0 means infinity) [default=1]\n"
-"  -l N     do not start new jobs if the load average is greater than N\n"
-"  -n       dry run (don't run commands but act like they succeeded)\n"
+"  -j N           run N jobs in parallel (0 means infinity) [default=%d on this system]\n"
+"  -k N           keep going until N jobs fail (0 means infinity) [default=1]\n"
+"  -l N           do not start new jobs if the load average is greater than N\n"
+"  -n             dry run (don't run commands but act like they succeeded)\n"
+"  -p PRIORITY    set the priority/niceness of spawned processes\n"
 "\n"
-"  -d MODE  enable debugging (use '-d list' to list modes)\n"
-"  -t TOOL  run a subtool (use '-t list' to list subtools)\n"
-"    terminates toplevel options; further flags are passed to the tool\n"
-"  -w FLAG  adjust warnings (use '-w list' to list warnings)\n",
+"  -d MODE        enable debugging (use '-d list' to list modes)\n"
+"  -t TOOL        run a subtool (use '-t list' to list subtools)\n"
+"                 terminates toplevel options; further flags are passed to the tool\n"
+"  -w FLAG        adjust warnings (use '-w list' to list warnings)\n",
           kNinjaVersion, config.parallelism);
 }
 
@@ -1428,7 +1429,7 @@ int ReadFlags(int* argc, char*** argv,
 
   int opt;
   while (!options->tool &&
-         (opt = getopt_long(*argc, *argv, "d:f:j:k:l:nt:vw:C:h", kLongOptions,
+         (opt = getopt_long(*argc, *argv, "d:f:j:k:l:p:nt:vw:C:h", kLongOptions,
                             NULL)) != -1) {
     switch (opt) {
       case 'd':
@@ -1473,6 +1474,15 @@ int ReadFlags(int* argc, char*** argv,
       case 'n':
         config->dry_run = true;
         break;
+      case 'p': {
+        char* end;
+        int value = strtol(optarg, &end, 10);
+        if (*end != 0)
+          Fatal("-p parameter not numeric: did you mean -p 0?");
+
+        config->subprocess_priority = value;
+        break;
+      }
       case 't':
         options->tool = ChooseTool(optarg);
         if (!options->tool)

src/subprocess-posix.cc

@@ -23,6 +23,8 @@
 #include <string.h>
 #include <sys/wait.h>
 #include <spawn.h>
+#include <sys/time.h>
+#include <sys/resource.h>
 
 #if defined(USE_PPOLL)
 #include <poll.h>
@@ -48,7 +50,8 @@ Subprocess::~Subprocess() {
     Finish();
 }
 
-bool Subprocess::Start(SubprocessSet* set, const string& command) {
+bool Subprocess::Start(SubprocessSet* set, const string& command,
+                       int priority) {
   int output_pipe[2];
   if (pipe(output_pipe) < 0)
     Fatal("pipe: %s", strerror(errno));
@@ -123,6 +126,12 @@ bool Subprocess::Start(SubprocessSet* set, const string& command) {
   if (err != 0)
     Fatal("posix_spawn: %s", strerror(err));
 
+  if(priority != INT_MIN) {
+    err = setpriority(PRIO_PROCESS, pid_, priority);
+    if(err != 0)
+      Fatal("setpriority: %s", strerror(errno));
+  }
+
   err = posix_spawnattr_destroy(&attr);
   if (err != 0)
     Fatal("posix_spawnattr_destroy: %s", strerror(err));
@@ -238,9 +247,9 @@ SubprocessSet::~SubprocessSet() {
     Fatal("sigprocmask: %s", strerror(errno));
 }
 
-Subprocess *SubprocessSet::Add(const string& command, bool use_console) {
+Subprocess *SubprocessSet::Add(const string& command, int priority, bool use_console) {
   Subprocess *subprocess = new Subprocess(use_console);
-  if (!subprocess->Start(this, command)) {
+  if (!subprocess->Start(this, command, priority)) {
     delete subprocess;
     return 0;
   }

src/subprocess-win32.cc

@@ -74,7 +74,52 @@ HANDLE Subprocess::SetupPipe(HANDLE ioport) {
   return output_write_child;
 }
 
-bool Subprocess::Start(SubprocessSet* set, const string& command) {
+static DWORD POSIXPriorityToWin32(int unixPriority) {
+  if (unixPriority <= -20)
+    return REALTIME_PRIORITY_CLASS;
+
+  switch (unixPriority) {
+  case -19:
+  case -18:
+  case -17:
+  case -16:
+  case -15:
+  case -14:
+  case -13:
+  case -12:
+  case -11:
+  case -10:
+    return HIGH_PRIORITY_CLASS;
+  case -9:
+  case -8:
+  case -7:
+  case -6:
+  case -5:
+  case -4:
+  case -3:
+  case -2:
+  case -1:
+    return ABOVE_NORMAL_PRIORITY_CLASS;
+  case 0:
+    return NORMAL_PRIORITY_CLASS;
+  case 1:
+  case 2:
+  case 3:
+  case 4:
+  case 5:
+  case 6:
+  case 7:
+  case 8:
+  case 9:
+  case 10:
+    return BELOW_NORMAL_PRIORITY_CLASS;
+  default:
+    return IDLE_PRIORITY_CLASS;
+  }
+}
+
+bool Subprocess::Start(SubprocessSet* set, const string& command,
+                       int priority) {
   HANDLE child_pipe = SetupPipe(set->ioport_);
 
   SECURITY_ATTRIBUTES security_attributes;
@@ -106,6 +151,8 @@ bool Subprocess::Start(SubprocessSet* set, const string& command) {
 
   // Ninja handles ctrl-c, except for subprocesses in console pools.
   DWORD process_flags = use_console_ ? 0 : CREATE_NEW_PROCESS_GROUP;
+  if (priority != INT_MIN)
+    process_flags |= POSIXPriorityToWin32(priority);
 
   // Do not prepend 'cmd /c' on Windows, this breaks command
   // lines greater than 8,191 chars.
@@ -238,9 +285,9 @@ BOOL WINAPI SubprocessSet::NotifyInterrupted(DWORD dwCtrlType) {
   return FALSE;
 }
 
-Subprocess *SubprocessSet::Add(const string& command, bool use_console) {
+Subprocess* SubprocessSet::Add(const string& command, int priority, bool use_console) {
   Subprocess *subprocess = new Subprocess(use_console);
-  if (!subprocess->Start(this, command)) {
+  if (!subprocess->Start(this, command, priority)) {
     delete subprocess;
     return 0;
   }

src/subprocess.h

@@ -52,7 +52,7 @@ struct Subprocess {
 
  private:
   Subprocess(bool use_console);
-  bool Start(struct SubprocessSet* set, const std::string& command);
+  bool Start(struct SubprocessSet* set, const std::string& command, int priority);
   void OnPipeReady();
 
   std::string buf_;
@@ -83,7 +83,8 @@ struct SubprocessSet {
   SubprocessSet();
   ~SubprocessSet();
 
-  Subprocess* Add(const std::string& command, bool use_console = false);
+  Subprocess* Add(const std::string& command, int priority = 0,
+                  bool use_console = false);
   bool DoWork();
   Subprocess* NextFinished();
   void Clear();