Replace internal attr string representation with an array

This ensures correct handling of attrnames with a dot in them and will
not throw errors about illegal attrnames.

Additionally this escapes any attributes containing dots in the JSON
output and adds another field called `attrPath` which contains the
attribute path as a list.

Example output:
```
{
  "attr": "hello",
  "attrPath": [
    "hello"
  ],
  "drvPath": "/nix/store/n204jib73z55cp9s0rmw1c5v5q528j7v-hello-2.12.drv",
  "name": "hello-2.12",
  "outputs": {
    "out": "/nix/store/h59dfk7dwrn7d2csykh9z9xm2miqmrnz-hello-2.12"
  },
  "system": "x86_64-linux"
}
```
This commit is contained in:
adisbladis 2022-04-26 17:11:17 +12:00
parent 2e58354a7e
commit 61c9f4cfcc
3 changed files with 43 additions and 26 deletions

View file

@ -28,6 +28,7 @@
#include <nlohmann/json.hpp> #include <nlohmann/json.hpp>
using namespace nix; using namespace nix;
using namespace nlohmann;
typedef enum { evalAuto, evalImpure, evalPure } pureEval; typedef enum { evalAuto, evalImpure, evalPure } pureEval;
@ -232,6 +233,17 @@ static void to_json(nlohmann::json & json, const Drv & drv) {
} }
std::string attrPathJoin(json input) {
return std::accumulate(input.begin(), input.end(), std::string(),
[](std::string ss, std::string s) {
// Escape token if containing dots
if (s.find(".") != std::string::npos) {
s = "\"" + s + "\"";
}
return ss.empty() ? s : ss + "." + s;
});
}
static void worker( static void worker(
EvalState & state, EvalState & state,
Bindings & autoArgs, Bindings & autoArgs,
@ -247,14 +259,15 @@ static void worker(
auto s = readLine(from.get()); auto s = readLine(from.get());
if (s == "exit") break; if (s == "exit") break;
if (!hasPrefix(s, "do ")) abort(); if (!hasPrefix(s, "do ")) abort();
std::string attrPath(s, 3); auto path = json::parse(s.substr(3));
auto attrPathS = attrPathJoin(path);
debug("worker process %d at '%s'", getpid(), attrPath); debug("worker process %d at '%s'", getpid(), path);
/* Evaluate it and send info back to the collector. */ /* Evaluate it and send info back to the collector. */
nlohmann::json reply = nlohmann::json{ { "attr", attrPath } }; json reply = json{ {"attr", attrPathS }, {"attrPath", path} };
try { try {
auto vTmp = findAlongAttrPath(state, attrPath, autoArgs, *vRoot).first; auto vTmp = findAlongAttrPath(state, attrPathS, autoArgs, *vRoot).first;
auto v = state.allocValue(); auto v = state.allocValue();
state.autoCallFunction(autoArgs, *vTmp, *v); state.autoCallFunction(autoArgs, *vTmp, *v);
@ -281,14 +294,10 @@ static void worker(
else if (v->type() == nAttrs) else if (v->type() == nAttrs)
{ {
auto attrs = nlohmann::json::array(); auto attrs = nlohmann::json::array();
bool recurse = attrPath == ""; // Dont require `recurseForDerivations = true;` for top-level attrset bool recurse = path.size() == 0; // Dont require `recurseForDerivations = true;` for top-level attrset
for (auto & i : v->attrs->lexicographicOrder()) { for (auto & i : v->attrs->lexicographicOrder()) {
std::string name(i->name); std::string name(i->name);
if (name.find('.') != std::string::npos || name.find(' ') != std::string::npos) {
printError("skipping job with illegal name '%s'", name);
continue;
}
attrs.push_back(name); attrs.push_back(name);
if (name == "recurseForDerivations") { if (name == "recurseForDerivations") {
@ -305,7 +314,7 @@ static void worker(
else if (v->type() == nNull) else if (v->type() == nNull)
; ;
else throw TypeError("attribute '%s' is %s, which is not supported", attrPath, showType(*v)); else throw TypeError("attribute '%s' is %s, which is not supported", path, showType(*v));
} catch (EvalError & e) { } catch (EvalError & e) {
auto err = e.info(); auto err = e.info();
@ -380,9 +389,9 @@ struct Proc {
struct State struct State
{ {
std::set<std::string> todo{""}; std::set<json> todo = json::array({ json::array() });
std::set<std::string> active; std::set<json> active;
std::exception_ptr exc; std::exception_ptr exc;
}; };
std::function<void()> collector(Sync<State> & state_, std::condition_variable & wakeup) { std::function<void()> collector(Sync<State> & state_, std::condition_variable & wakeup) {
@ -402,12 +411,12 @@ std::function<void()> collector(Sync<State> & state_, std::condition_variable &
proc_ = std::nullopt; proc_ = std::nullopt;
continue; continue;
} else if (s != "next") { } else if (s != "next") {
auto json = nlohmann::json::parse(s); auto json = json::parse(s);
throw Error("worker error: %s", (std::string) json["error"]); throw Error("worker error: %s", (std::string) json["error"]);
} }
/* Wait for a job name to become available. */ /* Wait for a job name to become available. */
std::string attrPath; json attrPath;
while (true) { while (true) {
checkInterrupt(); checkInterrupt();
@ -426,18 +435,19 @@ std::function<void()> collector(Sync<State> & state_, std::condition_variable &
} }
/* Tell the worker to evaluate it. */ /* Tell the worker to evaluate it. */
writeLine(proc->to.get(), "do " + attrPath); writeLine(proc->to.get(), "do " + attrPath.dump());
/* Wait for the response. */ /* Wait for the response. */
auto respString = readLine(proc->from.get()); auto respString = readLine(proc->from.get());
auto response = nlohmann::json::parse(respString); auto response = json::parse(respString);
/* Handle the response. */ /* Handle the response. */
StringSet newAttrs; std::vector<json> newAttrs;
if (response.find("attrs") != response.end()) { if (response.find("attrs") != response.end()) {
for (auto & i : response["attrs"]) { for (auto & i : response["attrs"]) {
auto s = (attrPath.empty() ? "" : attrPath + ".") + (std::string) i; json newAttr = json(response["attrPath"]);
newAttrs.insert(s); newAttr.emplace_back(i);
newAttrs.push_back(newAttr);
} }
} else { } else {
auto state(state_.lock()); auto state(state_.lock());
@ -450,8 +460,9 @@ std::function<void()> collector(Sync<State> & state_, std::condition_variable &
{ {
auto state(state_.lock()); auto state(state_.lock());
state->active.erase(attrPath); state->active.erase(attrPath);
for (auto & s : newAttrs) for (auto p : newAttrs) {
state->todo.insert(s); state->todo.insert(p);
}
wakeup.notify_all(); wakeup.notify_all();
} }
} }

View file

@ -31,4 +31,6 @@
}; };
}; };
"dotted.attr" = pkgs.hello;
} }

View file

@ -23,7 +23,7 @@ def common_test(extra_args: List[str]) -> None:
) )
results = [json.loads(r) for r in res.stdout.split("\n") if r] results = [json.loads(r) for r in res.stdout.split("\n") if r]
assert len(results) == 4 assert len(results) == 5
built_job = results[0] built_job = results[0]
assert built_job["attr"] == "builtJob" assert built_job["attr"] == "builtJob"
@ -32,14 +32,18 @@ def common_test(extra_args: List[str]) -> None:
assert built_job["drvPath"].endswith(".drv") assert built_job["drvPath"].endswith(".drv")
assert built_job["meta"]['broken'] is False assert built_job["meta"]['broken'] is False
recurse_drv = results[1] dotted_job = results[1]
assert dotted_job["attr"] == "\"dotted.attr\""
assert dotted_job["attrPath"] == [ "dotted.attr" ]
recurse_drv = results[2]
assert recurse_drv["attr"] == "recurse.drvB" assert recurse_drv["attr"] == "recurse.drvB"
assert recurse_drv["name"] == "drvB" assert recurse_drv["name"] == "drvB"
recurse_recurse_bool = results[2] recurse_recurse_bool = results[3]
assert "error" in recurse_recurse_bool assert "error" in recurse_recurse_bool
substituted_job = results[3] substituted_job = results[4]
assert substituted_job["attr"] == "substitutedJob" assert substituted_job["attr"] == "substitutedJob"
assert substituted_job["name"].startswith("hello-") assert substituted_job["name"].startswith("hello-")
assert substituted_job["meta"]['broken'] is False assert substituted_job["meta"]['broken'] is False