diff options
Diffstat (limited to 'third_party/nix/src/libexpr')
-rw-r--r-- | third_party/nix/src/libexpr/eval.cc | 15 | ||||
-rw-r--r-- | third_party/nix/src/libexpr/eval.hh | 8 |
2 files changed, 23 insertions, 0 deletions
diff --git a/third_party/nix/src/libexpr/eval.cc b/third_party/nix/src/libexpr/eval.cc index 5f272b62b8b8..f068b6ba6a48 100644 --- a/third_party/nix/src/libexpr/eval.cc +++ b/third_party/nix/src/libexpr/eval.cc @@ -379,6 +379,7 @@ EvalState::EvalState(const Strings& _searchPath, const ref<Store>& store) EvalState::~EvalState() = default; Path EvalState::checkSourcePath(const Path& path_) { + TraceFileAccess(path_); if (!allowedPaths) { return path_; } @@ -1819,6 +1820,20 @@ void EvalState::printStats() { } } +void EvalState::TraceFileAccess(const Path& realPath) { + if (file_access_trace_fn.has_value()) { + if (last_traced_file != realPath) { + (*file_access_trace_fn)(realPath); + // Basic deduplication. + last_traced_file = std::string(realPath); + } + } +} + +void EvalState::EnableFileAccessTracing(std::function<void(const Path&)> fn) { + file_access_trace_fn = fn; +} + size_t valueSize(const Value& v) { traceable_flat_hash_set<const Bindings*> seenBindings; traceable_flat_hash_set<const Env*> seenEnvs; diff --git a/third_party/nix/src/libexpr/eval.hh b/third_party/nix/src/libexpr/eval.hh index e08ec04ef598..c244b6eecc73 100644 --- a/third_party/nix/src/libexpr/eval.hh +++ b/third_party/nix/src/libexpr/eval.hh @@ -284,6 +284,10 @@ class EvalState : public gc { void realiseContext(const PathSet& context); + /* File access tracing. */ + void TraceFileAccess(const Path& path); + void EnableFileAccessTracing(std::function<void(const Path&)> fn); + private: unsigned long nrEnvs = 0; unsigned long nrValuesInEnvs = 0; @@ -299,6 +303,10 @@ class EvalState : public gc { bool countCalls; + std::optional<std::function<void(const Path&)>> file_access_trace_fn = + std::nullopt; + Path last_traced_file = ""; + typedef std::map<Symbol, size_t> PrimOpCalls; PrimOpCalls primOpCalls; |