mirror of
https://github.com/NVIDIA/OpenShell.git
synced 2026-10-20 06:00:06 +08:00
Drew Newberry
eeba0e7954
* refactor(runtime): confine and reduce unsafe code Signed-off-by: Drew Newberry <anewberry@nvidia.com> * fix(ci): make signal probe portable and gate Linux initialization Signed-off-by: Drew Newberry <anewberry@nvidia.com> * fix(cli): handle platform-specific SSH cancellation lints Signed-off-by: Drew Newberry <anewberry@nvidia.com> --------- Signed-off-by: Drew Newberry <anewberry@nvidia.com>
67 lines
2.4 KiB
Rust
67 lines
2.4 KiB
Rust
// SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
use std::env;
|
|
use std::path::PathBuf;
|
|
|
|
fn main() -> Result<(), Box<dyn std::error::Error>> {
|
|
// Preserve the vendored include-tree guarantee even when the caller has
|
|
// PROTOC_INCLUDE set: prost validates that ambient path independently of
|
|
// explicit includes. Re-enter with a child-only environment override.
|
|
if env::var_os("PROTOC_INCLUDE").is_some() {
|
|
let status = std::process::Command::new(env::current_exe()?)
|
|
.env_remove("PROTOC_INCLUDE")
|
|
.status()?;
|
|
if !status.success() {
|
|
return Err(format!("vendored protobuf build failed: {status}").into());
|
|
}
|
|
return Ok(());
|
|
}
|
|
|
|
let manifest_dir = PathBuf::from(env::var("CARGO_MANIFEST_DIR")?);
|
|
let storage_proto_dir = manifest_dir.join("proto");
|
|
let public_proto_dir = manifest_dir.join("../../proto");
|
|
let storage_proto = storage_proto_dir.join("storage.proto");
|
|
|
|
println!("cargo:rerun-if-changed={}", storage_proto.display());
|
|
for imported_proto in [
|
|
"datamodel.proto",
|
|
"openshell.proto",
|
|
"options.proto",
|
|
"sandbox.proto",
|
|
] {
|
|
println!(
|
|
"cargo:rerun-if-changed={}",
|
|
public_proto_dir.join(imported_proto).display()
|
|
);
|
|
}
|
|
|
|
// Configure the vendored compiler and well-known includes without changing
|
|
// the build process environment.
|
|
let mut proto_config = tonic_prost_build::Config::new();
|
|
proto_config.protoc_executable(protoc_bin_vendored::protoc_bin_path()?);
|
|
let proto_include = protoc_bin_vendored::include_path()?;
|
|
|
|
let descriptor_path = PathBuf::from(env::var("OUT_DIR")?).join("storage_descriptor.bin");
|
|
tonic_prost_build::configure()
|
|
.build_server(false)
|
|
.build_client(false)
|
|
.extern_path(".openshell.v1", "::openshell_core::proto")
|
|
.extern_path(
|
|
".openshell.datamodel.v1",
|
|
"::openshell_core::proto::datamodel::v1",
|
|
)
|
|
.extern_path(
|
|
".openshell.sandbox.v1",
|
|
"::openshell_core::proto::sandbox::v1",
|
|
)
|
|
.file_descriptor_set_path(&descriptor_path)
|
|
.compile_with_config(
|
|
proto_config,
|
|
&[storage_proto],
|
|
&[storage_proto_dir, public_proto_dir, proto_include],
|
|
)?;
|
|
|
|
Ok(())
|
|
}
|