mirror of
https://github.com/NVIDIA/OpenShell.git
synced 2026-10-19 05:50:41 +08:00
Simon Scatton
1959ea19be
Establish the Phase 1 reference implementation proposed by RFC 0012 while retaining the existing Cargo and Mise workflows during evaluation. - pin Bazel 9.1.1 and configure Bzlmod, rules_rs, LLVM, protobuf, and Rust 1.95 toolchains - import third-party crates from Cargo metadata and propagate the workspace version into Bazel targets - add library, binary, proc-macro, unit-test, and integration-test targets across the supported Rust workspace crates and drivers - generate protobuf Rust sources and descriptor sets under Bazel while preserving Cargo-compatible generated-code imports - annotate aws-lc-sys and zstd-sys native dependencies, build Z3 4.15.2 from source, and generate z3-sys bindings - make CLI and procfs test fixtures available as explicit Bazel inputs without relying on fixed host binary paths - define optimized release targets for Linux x86_64 and aarch64 CLI, sandbox, and gateway binaries, plus macOS aarch64 artifacts - add Bazel, buildifier, and lcov to the Nix development environment RFC: 0012 (rfc12 branch) Refs: #2491 Signed-off-by: Simon Scatton <sscatton@nvidia.com>
65 lines
1.9 KiB
Rust
65 lines
1.9 KiB
Rust
// SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
use std::net::TcpListener;
|
|
use std::process::Command;
|
|
use std::thread;
|
|
use std::time::Duration;
|
|
|
|
fn main() {
|
|
if std::env::args_os().next().as_deref() != Some(std::ffi::OsStr::new("ssh")) {
|
|
reexec_as_ssh();
|
|
}
|
|
|
|
match std::env::var("OPENSHELL_FAKE_FORWARD_MODE").as_deref() {
|
|
Ok("listen") => run_listener(),
|
|
Ok("sleep") => loop {
|
|
thread::sleep(Duration::from_secs(60));
|
|
},
|
|
_ => std::process::exit(2),
|
|
}
|
|
}
|
|
|
|
fn reexec_as_ssh() -> ! {
|
|
use std::os::unix::process::CommandExt;
|
|
|
|
let executable = std::env::current_exe().expect("fake forward must resolve its executable");
|
|
let error = Command::new(executable)
|
|
.arg0("ssh")
|
|
.args(std::env::args_os().skip(1))
|
|
.exec();
|
|
panic!("fake forward failed to re-execute as ssh: {error}");
|
|
}
|
|
|
|
fn run_listener() {
|
|
let port = forward_port().expect("fake forward must receive an SSH -L argument");
|
|
let listener = TcpListener::bind(("127.0.0.1", port)).expect("fake forward must bind");
|
|
for stream in listener.incoming() {
|
|
let _ = stream;
|
|
}
|
|
}
|
|
|
|
fn forward_port() -> Option<u16> {
|
|
let args = std::env::args().skip(1).collect::<Vec<_>>();
|
|
let mut index = 0;
|
|
while index < args.len() {
|
|
let arg = &args[index];
|
|
if arg == "-L" {
|
|
return args.get(index + 1).and_then(|value| local_port(value));
|
|
}
|
|
if let Some(value) = arg.strip_prefix("-L").filter(|value| !value.is_empty()) {
|
|
return local_port(value);
|
|
}
|
|
index += 1;
|
|
}
|
|
None
|
|
}
|
|
|
|
fn local_port(forward: &str) -> Option<u16> {
|
|
let (first, rest) = forward.split_once(':')?;
|
|
if first.bytes().all(|byte| byte.is_ascii_digit()) {
|
|
return first.parse().ok();
|
|
}
|
|
rest.split_once(':')?.0.parse().ok()
|
|
}
|