-
Notifications
You must be signed in to change notification settings - Fork 173
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
11 changed files
with
530 additions
and
2 deletions.
There are no files selected for viewing
Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.
Oops, something went wrong.
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,40 @@ | ||
| # | ||
| # Copyright (c) 2024 ZettaScale Technology | ||
| # | ||
| # This program and the accompanying materials are made available under the | ||
| # terms of the Eclipse Public License 2.0 which is available at | ||
| # http://www.eclipse.org/legal/epl-2.0, or the Apache License, Version 2.0 | ||
| # which is available at https://www.apache.org/licenses/LICENSE-2.0. | ||
| # | ||
| # SPDX-License-Identifier: EPL-2.0 OR Apache-2.0 | ||
| # | ||
| # Contributors: | ||
| # ZettaScale Zenoh Team, <[email protected]> | ||
| # | ||
| [package] | ||
| rust-version = { workspace = true } | ||
| name = "zenoh-link-vsock" | ||
| version = { workspace = true } | ||
| repository = { workspace = true } | ||
| homepage = { workspace = true } | ||
| authors = { workspace = true } | ||
| edition = { workspace = true } | ||
| license = { workspace = true } | ||
| categories = { workspace = true } | ||
| description = "Internal crate for zenoh." | ||
| # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html | ||
|
|
||
| [dependencies] | ||
| async-trait = { workspace = true } | ||
| tokio = { workspace = true, features = ["net", "io-util", "rt", "time"] } | ||
| tokio-util = { workspace = true, features = ["rt"] } | ||
| tokio-vsock = { workspace = true } | ||
| log = { workspace = true } | ||
| libc = { workspace = true } | ||
| zenoh-core = { workspace = true } | ||
| zenoh-link-commons = { workspace = true } | ||
| zenoh-protocol = { workspace = true } | ||
| zenoh-result = { workspace = true } | ||
| zenoh-sync = { workspace = true } | ||
| zenoh-util = { workspace = true } | ||
| zenoh-runtime = { workspace = true } |
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,98 @@ | ||
| // | ||
| // Copyright (c) 2024 ZettaScale Technology | ||
| // | ||
| // This program and the accompanying materials are made available under the | ||
| // terms of the Eclipse Public License 2.0 which is available at | ||
| // http://www.eclipse.org/legal/epl-2.0, or the Apache License, Version 2.0 | ||
| // which is available at https://www.apache.org/licenses/LICENSE-2.0. | ||
| // | ||
| // SPDX-License-Identifier: EPL-2.0 OR Apache-2.0 | ||
| // | ||
| // Contributors: | ||
| // ZettaScale Zenoh Team, <[email protected]> | ||
| // | ||
|
|
||
| //! ⚠️ WARNING ⚠️ | ||
| //! | ||
| //! This crate is intended for Zenoh's internal use. | ||
| //! | ||
| //! [Click here for Zenoh's documentation](../zenoh/index.html) | ||
| use async_trait::async_trait; | ||
| use libc::VMADDR_PORT_ANY; | ||
| use tokio_vsock::{ | ||
| VsockAddr, VMADDR_CID_ANY, VMADDR_CID_HOST, VMADDR_CID_HYPERVISOR, VMADDR_CID_LOCAL, | ||
| }; | ||
| use zenoh_core::zconfigurable; | ||
| use zenoh_link_commons::LocatorInspector; | ||
| use zenoh_protocol::core::{endpoint::Address, Locator}; | ||
| use zenoh_result::{bail, ZResult}; | ||
|
|
||
| mod unicast; | ||
| pub use unicast::*; | ||
|
|
||
| pub const VSOCK_LOCATOR_PREFIX: &str = "vsock"; | ||
|
|
||
| pub const VSOCK_VMADDR_CID_ANY: &str = "VMADDR_CID_ANY"; | ||
| pub const VSOCK_VMADDR_CID_HYPERVISOR: &str = "VMADDR_CID_HYPERVISOR"; | ||
| pub const VSOCK_VMADDR_CID_LOCAL: &str = "VMADDR_CID_LOCAL"; | ||
| pub const VSOCK_VMADDR_CID_HOST: &str = "VMADDR_CID_HOST"; | ||
|
|
||
| pub const VSOCK_VMADDR_PORT_ANY: &str = "VMADDR_PORT_ANY"; | ||
|
|
||
| #[derive(Default, Clone, Copy)] | ||
| pub struct VsockLocatorInspector; | ||
| #[async_trait] | ||
| impl LocatorInspector for VsockLocatorInspector { | ||
| fn protocol(&self) -> &str { | ||
| VSOCK_LOCATOR_PREFIX | ||
| } | ||
|
|
||
| async fn is_multicast(&self, _locator: &Locator) -> ZResult<bool> { | ||
| Ok(false) | ||
| } | ||
| } | ||
|
|
||
| zconfigurable! { | ||
| // Default MTU in bytes. | ||
| static ref VSOCK_DEFAULT_MTU: u16 = u16::MAX; | ||
| // Amount of time in microseconds to throttle the accept loop upon an error. | ||
| // Default set to 100 ms. | ||
| static ref VSOCK_ACCEPT_THROTTLE_TIME: u64 = 100_000; | ||
| } | ||
|
|
||
| pub fn get_vsock_addr(address: Address<'_>) -> ZResult<VsockAddr> { | ||
| let parts: Vec<&str> = address.as_str().split(':').collect(); | ||
|
|
||
| if parts.len() != 2 { | ||
| bail!("Incorrect vsock address: {:?}", address); | ||
| } | ||
|
|
||
| let cid = match parts[0].to_uppercase().as_str() { | ||
| VSOCK_VMADDR_CID_HYPERVISOR => VMADDR_CID_HYPERVISOR, | ||
| VSOCK_VMADDR_CID_HOST => VMADDR_CID_HOST, | ||
| VSOCK_VMADDR_CID_LOCAL => VMADDR_CID_LOCAL, | ||
| VSOCK_VMADDR_CID_ANY => VMADDR_CID_ANY, | ||
| "-1" => VMADDR_CID_ANY, | ||
| _ => { | ||
| if let Ok(cid) = parts[0].parse::<u32>() { | ||
| cid | ||
| } else { | ||
| bail!("Incorrect vsock cid: {:?}", parts[0]); | ||
| } | ||
| } | ||
| }; | ||
|
|
||
| let port = match parts[1].to_uppercase().as_str() { | ||
| VSOCK_VMADDR_PORT_ANY => VMADDR_PORT_ANY, | ||
| "-1" => VMADDR_PORT_ANY, | ||
| _ => { | ||
| if let Ok(cid) = parts[1].parse::<u32>() { | ||
| cid | ||
| } else { | ||
| bail!("Incorrect vsock port: {:?}", parts[1]); | ||
| } | ||
| } | ||
| }; | ||
|
|
||
| Ok(VsockAddr::new(cid, port)) | ||
| } |
Oops, something went wrong.