diff --git a/CMakeLists.txt b/CMakeLists.txt index 78243b7bac..79b99c2e3b 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -138,6 +138,7 @@ add_library(libninja OBJECT src/eval_env.cc src/graph.cc src/graphviz.cc + src/jobserver.cc src/json.cc src/line_printer.cc src/manifest_parser.cc @@ -153,6 +154,7 @@ add_library(libninja OBJECT if(WIN32) target_sources(libninja PRIVATE src/subprocess-win32.cc + src/jobserver-win32.cc src/includes_normalize-win32.cc src/msvc_helper-win32.cc src/msvc_helper_main-win32.cc @@ -164,7 +166,10 @@ if(WIN32) # compiler may build ninja. set_source_files_properties(src/getopt.c PROPERTIES LANGUAGE CXX) else() - target_sources(libninja PRIVATE src/subprocess-posix.cc) + target_sources(libninja PRIVATE + src/subprocess-posix.cc + src/jobserver-posix.cc + ) if(CMAKE_SYSTEM_NAME STREQUAL "OS400" OR CMAKE_SYSTEM_NAME STREQUAL "AIX") target_sources(libninja PRIVATE src/getopt.c) # Build getopt.c, which can be compiled as either C or C++, as C++ diff --git a/configure.py b/configure.py index 2b16618c43..c5473617d4 100755 --- a/configure.py +++ b/configure.py @@ -542,6 +542,7 @@ def has_re2c() -> bool: 'eval_env', 'graph', 'graphviz', + 'jobserver', 'json', 'line_printer', 'manifest_parser', @@ -556,6 +557,7 @@ def has_re2c() -> bool: objs += cxx(name, variables=cxxvariables) if platform.is_windows(): for name in ['subprocess-win32', + 'jobserver-win32', 'includes_normalize-win32', 'msvc_helper-win32', 'msvc_helper_main-win32']: @@ -565,6 +567,7 @@ def has_re2c() -> bool: objs += cc('getopt') else: objs += cxx('subprocess-posix') + objs += cxx('jobserver-posix') if platform.is_aix(): objs += cc('getopt') if platform.is_msvc(): diff --git a/src/build.cc b/src/build.cc index e52345b2a2..667880c778 100644 --- a/src/build.cc +++ b/src/build.cc @@ -50,7 +50,7 @@ struct DryRunCommandRunner : public CommandRunner { virtual ~DryRunCommandRunner() {} // Overridden from CommandRunner: - virtual size_t CanRunMore() const; + virtual size_t CanRunMore(bool jobserver_enabled) const; virtual bool StartCommand(Edge* edge); virtual bool WaitForCommand(Result* result); @@ -58,7 +58,7 @@ struct DryRunCommandRunner : public CommandRunner { queue finished_; }; -size_t DryRunCommandRunner::CanRunMore() const { +size_t DryRunCommandRunner::CanRunMore(bool jobserver_enabled) const { return SIZE_MAX; } @@ -163,6 +163,11 @@ Edge* Plan::FindWork() { if (ready_.empty()) return NULL; + // Don't initiate more work if the jobserver cannot acquire more tokens + if (jobserver_.Enabled() && !jobserver_.Acquire()) { + return NULL; + } + Edge* work = ready_.top(); ready_.pop(); return work; @@ -200,6 +205,11 @@ bool Plan::EdgeFinished(Edge* edge, EdgeResult result, string* err) { edge->pool()->EdgeFinished(*edge); edge->pool()->RetrieveReadyEdges(&ready_); + // Return the token for acquired for this very edge to the jobserver + if (jobserver_.Enabled()) { + jobserver_.Release(); + } + // The rest of this function only applies to successful commands. if (result != kEdgeSucceeded) return true; @@ -578,6 +588,7 @@ void Plan::ScheduleInitialEdges() { } void Plan::PrepareQueue() { + jobserver_.Init(); ComputeCriticalPath(); ScheduleInitialEdges(); } @@ -595,7 +606,7 @@ void Plan::Dump() const { struct RealCommandRunner : public CommandRunner { explicit RealCommandRunner(const BuildConfig& config) : config_(config) {} virtual ~RealCommandRunner() {} - virtual size_t CanRunMore() const; + virtual size_t CanRunMore(bool jobserver_enabled) const; virtual bool StartCommand(Edge* edge); virtual bool WaitForCommand(Result* result); virtual vector GetActiveEdges(); @@ -618,12 +629,17 @@ void RealCommandRunner::Abort() { subprocs_.Clear(); } -size_t RealCommandRunner::CanRunMore() const { +size_t RealCommandRunner::CanRunMore(bool jobserver_enabled) const { size_t subproc_number = subprocs_.running_.size() + subprocs_.finished_.size(); int64_t capacity = config_.parallelism - subproc_number; + // Return "infinite" capacity if a jobserver is used to limit the number + // of parallel subprocesses instead. + if (jobserver_enabled) + return SIZE_MAX; + if (config_.max_load_average > 0.0f) { int load_capacity = config_.max_load_average - GetLoadAverage(); if (load_capacity < capacity) @@ -789,7 +805,7 @@ bool Builder::Build(string* err) { while (plan_.more_to_do()) { // See if we can start any more commands. if (failures_allowed) { - size_t capacity = command_runner_->CanRunMore(); + size_t capacity = command_runner_->CanRunMore(plan_.JobserverEnabled()); while (capacity > 0) { Edge* edge = plan_.FindWork(); if (!edge) @@ -817,7 +833,7 @@ bool Builder::Build(string* err) { --capacity; // Re-evaluate capacity. - size_t current_capacity = command_runner_->CanRunMore(); + size_t current_capacity = command_runner_->CanRunMore(plan_.JobserverEnabled()); if (current_capacity < capacity) capacity = current_capacity; } diff --git a/src/build.h b/src/build.h index 471f0b2caa..d822119ea0 100644 --- a/src/build.h +++ b/src/build.h @@ -24,6 +24,7 @@ #include "depfile_parser.h" #include "graph.h" #include "exit_status.h" +#include "jobserver.h" #include "util.h" // int64_t struct BuildLog; @@ -51,6 +52,9 @@ struct Plan { /// Returns true if there's more work to be done. bool more_to_do() const { return wanted_edges_ > 0 && command_edges_ > 0; } + /// Jobserver status used to skip capacity based on load average + bool JobserverEnabled() const { return jobserver_.Enabled(); } + /// Dumps the current state of the plan. void Dump() const; @@ -138,6 +142,9 @@ struct Plan { /// Total remaining number of wanted edges. int wanted_edges_; + + /// Jobserver client + Jobserver jobserver_; }; /// CommandRunner is an interface that wraps running the build @@ -145,7 +152,7 @@ struct Plan { /// RealCommandRunner is an implementation that actually runs commands. struct CommandRunner { virtual ~CommandRunner() {} - virtual size_t CanRunMore() const = 0; + virtual size_t CanRunMore(bool jobserver_enabled) const = 0; virtual bool StartCommand(Edge* edge) = 0; /// The result of waiting for a command. diff --git a/src/build_test.cc b/src/build_test.cc index c84190a040..84d9ceed1e 100644 --- a/src/build_test.cc +++ b/src/build_test.cc @@ -521,7 +521,7 @@ struct FakeCommandRunner : public CommandRunner { max_active_edges_(1), fs_(fs) {} // CommandRunner impl - virtual size_t CanRunMore() const; + virtual size_t CanRunMore(bool jobserver_enabled) const; virtual bool StartCommand(Edge* edge); virtual bool WaitForCommand(Result* result); virtual vector GetActiveEdges(); @@ -622,10 +622,13 @@ void BuildTest::RebuildTarget(const string& target, const char* manifest, builder.command_runner_.release(); } -size_t FakeCommandRunner::CanRunMore() const { +size_t FakeCommandRunner::CanRunMore(bool jobserver_enabled) const { if (active_edges_.size() < max_active_edges_) return SIZE_MAX; + if (jobserver_enabled) + return SIZE_MAX; + return 0; } diff --git a/src/jobserver-posix.cc b/src/jobserver-posix.cc new file mode 100644 index 0000000000..cf8ffa05de --- /dev/null +++ b/src/jobserver-posix.cc @@ -0,0 +1,72 @@ +// Copyright 2024 Google Inc. All Rights Reserved. +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. + +#include +#include +#include +#include + +#include "jobserver.h" +#include "util.h" + +Jobserver::Jobserver() : token_count_(0), fd_(-1) +{} + +void Jobserver::Init() { + assert(fd_ < 0); + + if (!ParseJobserverAuth("fifo")) { + return; + } + + const char *jobserver = jobserver_name_.c_str(); + + fd_ = open(jobserver, O_NONBLOCK | O_CLOEXEC | O_RDWR); + if (fd_ < 0) { + Fatal("failed to open jobserver: %s: %s", jobserver, strerror(errno)); + } + + Info("Using jobserver: %s", jobserver); +} + +Jobserver::~Jobserver() { + assert(token_count_ == 0); + + if (fd_ >= 0) { + close(fd_); + } +} + +bool Jobserver::Enabled() const { + return fd_ >= 0; +} + +bool Jobserver::AcquireToken() { + char token; + int res = read(fd_, &token, 1); + if (res < 0 && errno != EAGAIN && errno != EWOULDBLOCK) { + Fatal("failed to read jobserver token: %s", strerror(errno)); + } + + return res > 0; +} + +void Jobserver::ReleaseToken() { + char token = '+'; + int res = write(fd_, &token, 1); + if (res != 1) { + Fatal("failed to write token: %s: %s", jobserver_name_.c_str(), + strerror(errno)); + } +} diff --git a/src/jobserver-win32.cc b/src/jobserver-win32.cc new file mode 100644 index 0000000000..b1a9de0c2a --- /dev/null +++ b/src/jobserver-win32.cc @@ -0,0 +1,61 @@ +// Copyright 2024 Google Inc. All Rights Reserved. +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. + +#include +#include + +#include "jobserver.h" +#include "util.h" + +Jobserver::Jobserver() : token_count_(0), sem_(NULL) +{} + +void Jobserver::Init() { + assert(sem_ == NULL); + + if (!ParseJobserverAuth("sem")) { + return; + } + + const char *name = jobserver_name_.c_str(); + + sem_ = OpenSemaphore(SEMAPHORE_ALL_ACCESS, false, name); + if (sem_ == NULL) { + Win32Fatal("OpenSemaphore"); + } + + Info("using jobserver: %s", name); +} + +Jobserver::~Jobserver() { + assert(token_count_ == 0); + + if (sem_ != NULL) { + CloseHandle(sem_); + } +} + +bool Jobserver::Enabled() const { + return sem_ != NULL; +} + +bool Jobserver::AcquireToken() { + return WaitForSingleObject(sem_, 0) == WAIT_OBJECT_0; +} + +void Jobserver::ReleaseToken() { + if (ReleaseSemaphore(sem_, 1, NULL)) { + Win32Fatal("ReleaseSemaphore"); + } +} diff --git a/src/jobserver.cc b/src/jobserver.cc new file mode 100644 index 0000000000..4b501353a8 --- /dev/null +++ b/src/jobserver.cc @@ -0,0 +1,94 @@ +// Copyright 2024 Google Inc. All Rights Reserved. +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. + +#include +#include + +#include "jobserver.h" + +bool Jobserver::Acquire() { + // The first token is implicitly handed to the ninja process, so don't + // acquire it from the jobserver + if (token_count_ == 0 || AcquireToken()) { + token_count_++; + return true; + } + + return false; +} + +void Jobserver::Release() { + assert(token_count_ >= 1); + token_count_--; + + // Don't return first token to the jobserver, as it is implicitly handed + // to the ninja process + if (token_count_ > 0) { + ReleaseToken(); + } +} + +bool Jobserver::ParseJobserverAuth(const char *type) { + const char *jobserver_auth = "--jobserver-auth="; + const char *makeflags = getenv("MAKEFLAGS"); + + // Return early if no make flags are passed in the environment + if (makeflags == NULL) { + return false; + } + + // Search the make flags for a jobserver-auth argument + const char *str_begin = strstr(makeflags, jobserver_auth); + if (str_begin == NULL) { + return false; + } + + // Advance the string pointer to just past the = character + str_begin += strlen(jobserver_auth); + + // Find the length of the type value by searching for the following colon + const char *str_end = strchr(str_begin, ':'); + if (str_end == NULL) { + Warning("invalid --jobserver-auth value: '%s'", str_begin); + return false; + } + + // Ignore the argument if the length or the value of the type value doesn't + // match the requested type (i.e. "fifo" on posix or "sem" on windows). + if (strlen(type) != static_cast(str_end - str_begin) || + strncmp(str_begin, type, str_end - str_begin)) { + Warning("invalid jobserver type: got %.*s; expected %s", + str_end - str_begin, str_begin, type); + return false; + } + + // Advance the string pointer to just after the : character + str_begin = str_end + 1; + + // Find the length of the jobserver path/name value by searching for the + // following space or the end of the string. + str_end = strchr(str_begin, ' '); + if (str_end == NULL) { + jobserver_name_ = std::string(str_begin); + } else { + jobserver_name_ = std::string(str_begin, str_end - str_begin); + } + + if (jobserver_name_.empty()) { + Warning("invalid --jobserver-auth value: ''"); + return false; + } + + return true; +} diff --git a/src/jobserver.h b/src/jobserver.h new file mode 100644 index 0000000000..365bc20c55 --- /dev/null +++ b/src/jobserver.h @@ -0,0 +1,47 @@ +// Copyright 2024 Google Inc. All Rights Reserved. +// +// Licensed under the Apache License, Version 2.0 (the "License"); +// you may not use this file except in compliance with the License. +// You may obtain a copy of the License at +// +// http://www.apache.org/licenses/LICENSE-2.0 +// +// Unless required by applicable law or agreed to in writing, software +// distributed under the License is distributed on an "AS IS" BASIS, +// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +// See the License for the specific language governing permissions and +// limitations under the License. + +#pragma once + +#if defined( _WIN32) +#include +#endif + +#include +#include + +#include "util.h" + +struct Jobserver { + Jobserver(); + ~Jobserver(); + void Init(); + bool Enabled() const; + bool Acquire(); + void Release(); + +private: + bool ParseJobserverAuth(const char *type); + bool AcquireToken(); + void ReleaseToken(); + + std::string jobserver_name_; + size_t token_count_; + +#ifdef _WIN32 + HANDLE sem_; +#else + int fd_; +#endif +};