-
Notifications
You must be signed in to change notification settings - Fork 42
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Updates the README to use cargo target config instead of RUSTFLAGS to avoid setting the linker for ebpf in cargo-in-cargo. Rewrite test.sh in python.
- Loading branch information
Showing
5 changed files
with
215 additions
and
98 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,172 @@ | ||
| #!/usr/bin/env python3 | ||
|
|
||
| import argparse | ||
| import os | ||
| import platform | ||
| import subprocess | ||
| import sys | ||
| import tempfile | ||
| from typing import TypedDict | ||
|
|
||
|
|
||
| class SubprocessArgs(TypedDict, total=False): | ||
| cwd: str | ||
| env: dict[str, str] | ||
|
|
||
|
|
||
| def main() -> None: | ||
| parser = argparse.ArgumentParser( | ||
| description="Generate and build a Rust project using cargo." | ||
| ) | ||
| parser.add_argument("template_dir", help="Template directory") | ||
| parser.add_argument("program_type", help="Program type") | ||
| args = parser.parse_args() | ||
|
|
||
| match args.program_type: | ||
| case "cgroup_sockopt": | ||
| additional_args = ["-d", "sockopt_target=getsockopt"] | ||
| case "classifier" | "cgroup_skb": | ||
| additional_args = ["-d", "direction=Ingress"] | ||
| case "fentry" | "fexit": | ||
| additional_args = ["-d", "fn_name=try_to_wake_up"] | ||
| case "kprobe" | "kretprobe": | ||
| additional_args = ["-d", "kprobe=do_unlinkat"] | ||
| case "lsm": | ||
| additional_args = ["-d", "lsm_hook=file_open"] | ||
| case "raw_tracepoint": | ||
| additional_args = ["-d", "tracepoint_name=sys_enter"] | ||
| case "sk_msg": | ||
| additional_args = ["-d", "sock_map=SOCK_MAP"] | ||
| case "tp_btf": | ||
| additional_args = ["-d", "tracepoint_name=net_dev_queue"] | ||
| case "tracepoint": | ||
| additional_args = [ | ||
| "-d", | ||
| "tracepoint_category=net", | ||
| "-d", | ||
| "tracepoint_name=net_dev_queue", | ||
| ] | ||
| case "uprobe" | "uretprobe": | ||
| additional_args = [ | ||
| "-d", | ||
| "uprobe_target=/proc/self/exe", | ||
| "-d", | ||
| "uprobe_fn_name=main", | ||
| ] | ||
| case _: | ||
| additional_args = [] | ||
|
|
||
| CRATE_NAME = "aya-test-crate" | ||
| with tempfile.TemporaryDirectory() as tmp_dir: | ||
| cmds: list[tuple[list[str], SubprocessArgs]] = [ | ||
| ( | ||
| [ | ||
| "cargo", | ||
| "generate", | ||
| "--path", | ||
| args.template_dir, | ||
| "-n", | ||
| CRATE_NAME, | ||
| "-d", | ||
| f"program_type={args.program_type}", | ||
| ] | ||
| + additional_args, | ||
| {"cwd": tmp_dir}, | ||
| ), | ||
| ] | ||
| project_dir = os.path.join(tmp_dir, CRATE_NAME) | ||
| match platform.system(): | ||
| case "Linux": | ||
| cmds.extend( | ||
| (cmd, {"cwd": project_dir}) | ||
| for cmd in ( | ||
| ["cargo", "+nightly", "fmt", "--all", "--", "--check"], | ||
| ["cargo", "build", "--package", CRATE_NAME], | ||
| ["cargo", "build", "--package", CRATE_NAME, "--release"], | ||
| # We cannot run clippy over the whole workspace at once due to feature unification. | ||
| # Since both ${CRATE_NAME} and ${CRATE_NAME}-ebpf depend on ${CRATE_NAME}-common and | ||
| # ${CRATE_NAME} activates ${CRATE_NAME}-common's aya dependency, we end up trying to | ||
| # compile the panic handler twice: once from the bpf program, and again from std via | ||
| # aya. | ||
| [ | ||
| "cargo", | ||
| "clippy", | ||
| "--exclude", | ||
| f"{CRATE_NAME}-ebpf", | ||
| "--all-targets", | ||
| "--workspace", | ||
| "--", | ||
| "--deny", | ||
| "warnings", | ||
| ], | ||
| [ | ||
| "cargo", | ||
| "clippy", | ||
| "--package", | ||
| f"{CRATE_NAME}-ebpf", | ||
| "--all-targets", | ||
| "--", | ||
| "--deny", | ||
| "warnings", | ||
| ], | ||
| ) | ||
| ) | ||
| case "Darwin": | ||
| arch = platform.machine() | ||
| if arch == "arm64": | ||
| arch = "aarch64" | ||
| target = f"{arch}-unknown-linux-musl" | ||
| cmds.append( | ||
| ( | ||
| [ | ||
| "cargo", | ||
| "build", | ||
| "--package", | ||
| CRATE_NAME, | ||
| "--release", | ||
| "--target", | ||
| target, | ||
| "--config", | ||
| f'target.{target}.linker = "rust-lld"', | ||
| ], | ||
| { | ||
| "cwd": project_dir, | ||
| "env": os.environ | ||
| | { | ||
| "AYA_BUILD_EBPF": "true", | ||
| "CC": f"{arch}-linux-musl-gcc", | ||
| }, | ||
| }, | ||
| ) | ||
| ) | ||
|
|
||
| for cmd, kwargs in cmds: | ||
| sys.stdout.write(f"{cmd=} {kwargs=}\n") | ||
| sys.stdout.flush() | ||
| subprocess.check_call(cmd, **kwargs) | ||
|
|
||
| if platform.system() == "Linux": | ||
| with subprocess.Popen( | ||
| "cargo xtask run", | ||
| cwd=project_dir, | ||
| shell=True, | ||
| stdin=subprocess.PIPE, | ||
| stdout=subprocess.PIPE, | ||
| stderr=sys.stderr, | ||
| ) as process: | ||
| assert process.stdin is not None | ||
| assert process.stdout is not None | ||
| for line in process.stdout: | ||
| sys.stdout.buffer.writelines([line]) | ||
| sys.stdout.buffer.flush() | ||
| if "Waiting for Ctrl-C..." in line.decode("utf-8"): | ||
| process.stdin.write(b"\x03") | ||
| process.stdin.flush() | ||
|
|
||
| retcode = process.returncode | ||
| if retcode != 0: | ||
| raise subprocess.CalledProcessError(retcode, process.args) | ||
|
|
||
|
|
||
| if __name__ == "__main__": | ||
| main() |