Compare commits
2 commits
62e7fc4d88
...
a0784bf027
Author | SHA1 | Date | |
---|---|---|---|
Skyler Grey | a0784bf027 | ||
Skyler Grey | 46800094bb |
|
@ -310,16 +310,58 @@ class UpdateBuildOutput(steps.BuildStep):
|
||||||
self.project = project
|
self.project = project
|
||||||
self.path = path
|
self.path = path
|
||||||
|
|
||||||
|
def join_traversalsafe(self, root: Path, joined: Path) -> Path:
|
||||||
|
root = root.resolve()
|
||||||
|
|
||||||
|
for part in joined.parts:
|
||||||
|
new_root = (root / part).resolve()
|
||||||
|
|
||||||
|
if not new_root.is_relative_to(root):
|
||||||
|
msg = f"Joined path attempted to traverse upwards when processing {root} against {part} (gave {new_root})"
|
||||||
|
raise ValueError(msg)
|
||||||
|
|
||||||
|
root = new_root
|
||||||
|
|
||||||
|
return root
|
||||||
|
|
||||||
|
def join_all_traversalsafe(self, root: Path, *paths: Path) -> Path:
|
||||||
|
for path in paths:
|
||||||
|
root = self.join_traversalsafe(root, path)
|
||||||
|
|
||||||
|
return root
|
||||||
|
|
||||||
|
@defer.inlineCallbacks
|
||||||
def run(self) -> Generator[Any, object, Any]:
|
def run(self) -> Generator[Any, object, Any]:
|
||||||
props = self.build.getProperties()
|
props = self.build.getProperties()
|
||||||
if props.getProperty("branch") != self.project.default_branch:
|
|
||||||
|
pr = props.getProperty("pr_number")
|
||||||
|
|
||||||
|
if not pr and props.getProperty("branch") != self.project.default_branch:
|
||||||
return util.SKIPPED
|
return util.SKIPPED
|
||||||
|
|
||||||
attr = Path(props.getProperty("attr")).name
|
|
||||||
out_path = props.getProperty("out_path")
|
out_path = props.getProperty("out_path")
|
||||||
# XXX don't hardcode this
|
|
||||||
self.path.mkdir(parents=True, exist_ok=True)
|
if not out_path: # if, e.g., the build fails and doesn't produce an output
|
||||||
(self.path / attr).write_text(out_path)
|
return util.SKIPPED
|
||||||
|
|
||||||
|
owner = Path(props.getProperty("owner"))
|
||||||
|
repo = Path(props.getProperty("repository_name"))
|
||||||
|
|
||||||
|
target = Path(props.getProperty("branch")) if not pr else Path(f"pulls/{pr}")
|
||||||
|
|
||||||
|
attr = Path(props.getProperty("attr"))
|
||||||
|
|
||||||
|
try:
|
||||||
|
file = self.join_all_traversalsafe(self.path, owner, repo, target, attr)
|
||||||
|
except ValueError as e:
|
||||||
|
error_log: StreamLog = yield self.addLog("path_error")
|
||||||
|
error_log.addStderr(f"Path traversal prevented ... skipping update: {e}")
|
||||||
|
return util.FAILURE
|
||||||
|
|
||||||
|
file.parent.mkdir(parents=True, exist_ok=True)
|
||||||
|
|
||||||
|
file.write_text(out_path)
|
||||||
|
|
||||||
return util.SUCCESS
|
return util.SUCCESS
|
||||||
|
|
||||||
|
|
||||||
|
@ -563,6 +605,7 @@ def nix_build_config(
|
||||||
def nix_skipped_build_config(
|
def nix_skipped_build_config(
|
||||||
project: GitProject,
|
project: GitProject,
|
||||||
worker_names: list[str],
|
worker_names: list[str],
|
||||||
|
outputs_path: Path | None = None,
|
||||||
) -> BuilderConfig:
|
) -> BuilderConfig:
|
||||||
"""Dummy builder that is triggered when a build is skipped."""
|
"""Dummy builder that is triggered when a build is skipped."""
|
||||||
factory = util.BuildFactory()
|
factory = util.BuildFactory()
|
||||||
|
@ -601,6 +644,14 @@ def nix_skipped_build_config(
|
||||||
set_properties={"report_status": False},
|
set_properties={"report_status": False},
|
||||||
),
|
),
|
||||||
)
|
)
|
||||||
|
if outputs_path is not None:
|
||||||
|
factory.addStep(
|
||||||
|
UpdateBuildOutput(
|
||||||
|
project=project,
|
||||||
|
name="Update build output",
|
||||||
|
path=outputs_path,
|
||||||
|
),
|
||||||
|
)
|
||||||
return util.BuilderConfig(
|
return util.BuilderConfig(
|
||||||
name=f"{project.name}/nix-skipped-build",
|
name=f"{project.name}/nix-skipped-build",
|
||||||
project=project.name,
|
project=project.name,
|
||||||
|
@ -759,7 +810,7 @@ def config_for_project(
|
||||||
retries=build_retries,
|
retries=build_retries,
|
||||||
post_build_steps=post_build_steps,
|
post_build_steps=post_build_steps,
|
||||||
),
|
),
|
||||||
nix_skipped_build_config(project, [SKIPPED_BUILDER_NAME]),
|
nix_skipped_build_config(project, [SKIPPED_BUILDER_NAME], outputs_path),
|
||||||
nix_register_gcroot_config(project, worker_names),
|
nix_register_gcroot_config(project, worker_names),
|
||||||
nix_build_combined_config(project, worker_names),
|
nix_build_combined_config(project, worker_names),
|
||||||
],
|
],
|
||||||
|
|
Loading…
Reference in a new issue