-
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
218 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,181 @@ | ||
| #!/usr/bin/env python3 | ||
|
|
||
| import argparse | ||
| import asyncio | ||
| import os | ||
| import platform | ||
| import signal | ||
| 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: | ||
| print(f"Running command: {' '.join(cmd)} with kwargs: {kwargs}") | ||
| subprocess.check_call(cmd, **kwargs) | ||
|
|
||
| if platform.system() == "Linux": | ||
|
|
||
| async def run(): | ||
| args = ("cargo", "xtask", "run") | ||
| process = await asyncio.create_subprocess_exec( | ||
| *args, | ||
| cwd=project_dir, | ||
| stdin=subprocess.DEVNULL, | ||
| stdout=asyncio.subprocess.PIPE, | ||
| stderr=sys.stderr, | ||
| ) | ||
| assert process.stdout is not None | ||
| try: | ||
| while not process.stdout.at_eof(): | ||
| line = await process.stdout.readline() | ||
| line = line.decode() | ||
| sys.stdout.write(line) | ||
| if "Waiting for Ctrl-C" in line: | ||
| print("Sending SIGINT to the process") | ||
| process.send_signal(signal.SIGINT) | ||
| print("Sent SIGINT to the process") | ||
| finally: | ||
| process.terminate() | ||
|
|
||
| retcode = await process.wait() | ||
| if retcode != 0: | ||
| raise subprocess.CalledProcessError(retcode, args) | ||
|
|
||
| asyncio.run(asyncio.wait_for(run(), 30)) | ||
|
|
||
|
|
||
| if __name__ == "__main__": | ||
| main() |