blob: 09429d39dbe5ced7d5ed306ae74d893523677d16 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
|
#!/usr/bin/env python3
import shutil
import subprocess
import sys
from argparse import ArgumentParser
from pathlib import Path
from tempfile import TemporaryDirectory
from urllib.request import urlopen
SRC_DIR = Path("snekbox").resolve(strict=True)
FILE_NAME = "config"
def compile_proto(path: Path) -> None:
"""Compile a protobuf file at `path` into Python code."""
protoc_bin = shutil.which("protoc")
if not protoc_bin:
print("protoc binary could not be found on PATH", file=sys.stderr)
sys.exit(1)
args = [protoc_bin, f"--proto_path={path.parent}", f"--python_out={SRC_DIR}", path]
result = subprocess.run(args)
if result.returncode != 0:
sys.exit(result.returncode)
def get_version() -> str:
"""Get the NsJail version from the command line arguments."""
parser = ArgumentParser(description="Compile an NsJail config protobuf into Python.")
parser.add_argument("version", help="the NsJail version from which to get the protobuf file")
args = parser.parse_args()
return args.version
def main() -> None:
"""Get a config.proto for NsJail and compile it into Python."""
version = get_version()
url = f"https://raw.githubusercontent.com/google/nsjail/{version}/config.proto"
with urlopen(url) as response:
if response.status >= 400:
print(f"Failed to retrieve config.proto: status {response.status}", file=sys.stderr)
sys.exit(1)
with TemporaryDirectory() as dir_name:
file_path = Path(dir_name) / f"{FILE_NAME}.proto"
with open(file_path, "wb") as file:
file.write(response.read())
compile_proto(file_path)
# Remove the _pb suffix from the generated Python file.
if generated_py := next(SRC_DIR.glob(f"{FILE_NAME}_pb*.py"), None):
generated_py.rename(generated_py.with_stem(FILE_NAME))
else:
print(f"Could not find the generated Python file in {SRC_DIR}.", file=sys.stderr)
sys.exit(1)
if __name__ == "__main__":
main()
|
