github/kata-containers
1
0
Fork 1
mirror of https://github.com/kata-containers/kata-containers.git synced 2025-06-27 07:48:55 +00:00
Code Issues Projects Releases Wiki Activity

agent: Add underscore for constants

Browse Source 
To make them follow the rust convention and easier to read.

Fixes: #1235

Signed-off-by: Tim Zhang <tim@hyper.sh>
This commit is contained in:
Tim Zhang 2021-01-20 16:49:59 +08:00
parent c3a4180ae3
commit 14a63cce22
3 changed files with 35 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/agent/src/device.rs
View File

@ -13,7 +13,7 @@ use std::sync::Arc;
use tokio::sync::Mutex; use tokio::sync::Mutex;
use crate::linux_abi::*; use crate::linux_abi::*;
use crate::mount::{DRIVERBLKTYPE, DRIVERMMIOBLKTYPE, DRIVERNVDIMMTYPE, DRIVERSCSITYPE}; use crate::mount::{DRIVER_BLK_TYPE, DRIVER_MMIO_BLK_TYPE, DRIVER_NVDIMM_TYPE, DRIVER_SCSI_TYPE};
use crate::sandbox::Sandbox; use crate::sandbox::Sandbox;
use crate::{AGENT_CONFIG, GLOBAL_DEVICE_WATCHER}; use crate::{AGENT_CONFIG, GLOBAL_DEVICE_WATCHER};
use anyhow::{anyhow, Result}; use anyhow::{anyhow, Result};
@ -401,10 +401,10 @@ async fn add_device(
} }
match device.field_type.as_str() { match device.field_type.as_str() {
DRIVERBLKTYPE => virtio_blk_device_handler(device, spec, sandbox, devidx).await, DRIVER_BLK_TYPE => virtio_blk_device_handler(device, spec, sandbox, devidx).await,
DRIVERMMIOBLKTYPE => virtiommio_blk_device_handler(device, spec, sandbox, devidx).await, DRIVER_MMIO_BLK_TYPE => virtiommio_blk_device_handler(device, spec, sandbox, devidx).await,
DRIVERNVDIMMTYPE => virtio_nvdimm_device_handler(device, spec, sandbox, devidx).await, DRIVER_NVDIMM_TYPE => virtio_nvdimm_device_handler(device, spec, sandbox, devidx).await,
DRIVERSCSITYPE => virtio_scsi_device_handler(device, spec, sandbox, devidx).await, DRIVER_SCSI_TYPE => virtio_scsi_device_handler(device, spec, sandbox, devidx).await,
_ => Err(anyhow!("Unknown device type {}", device.field_type)), _ => Err(anyhow!("Unknown device type {}", device.field_type)),
} }
} }

54
src/agent/src/mount.rs
View File

@ -31,16 +31,16 @@ use crate::Sandbox;
use anyhow::{anyhow, Context, Result}; use anyhow::{anyhow, Context, Result};
use slog::Logger; use slog::Logger;
pub const DRIVER9PTYPE: &str = "9p"; pub const DRIVER_9P_TYPE: &str = "9p";
pub const DRIVERVIRTIOFSTYPE: &str = "virtio-fs"; pub const DRIVER_VIRTIOFS_TYPE: &str = "virtio-fs";
pub const DRIVERBLKTYPE: &str = "blk"; pub const DRIVER_BLK_TYPE: &str = "blk";
pub const DRIVERMMIOBLKTYPE: &str = "mmioblk"; pub const DRIVER_MMIO_BLK_TYPE: &str = "mmioblk";
pub const DRIVERSCSITYPE: &str = "scsi"; pub const DRIVER_SCSI_TYPE: &str = "scsi";
pub const DRIVERNVDIMMTYPE: &str = "nvdimm"; pub const DRIVER_NVDIMM_TYPE: &str = "nvdimm";
pub const DRIVEREPHEMERALTYPE: &str = "ephemeral"; pub const DRIVER_EPHEMERAL_TYPE: &str = "ephemeral";
pub const DRIVERLOCALTYPE: &str = "local"; pub const DRIVER_LOCAL_TYPE: &str = "local";
pub const TYPEROOTFS: &str = "rootfs"; pub const TYPE_ROOTFS: &str = "rootfs";
#[rustfmt::skip] #[rustfmt::skip]
lazy_static! { lazy_static! {
@ -125,14 +125,14 @@ lazy_static! {
} }
pub const STORAGE_HANDLER_LIST: [&str; 8] = [ pub const STORAGE_HANDLER_LIST: [&str; 8] = [
DRIVERBLKTYPE, DRIVER_BLK_TYPE,
DRIVER9PTYPE, DRIVER_9P_TYPE,
DRIVERVIRTIOFSTYPE, DRIVER_VIRTIOFS_TYPE,
DRIVEREPHEMERALTYPE, DRIVER_EPHEMERAL_TYPE,
DRIVERMMIOBLKTYPE, DRIVER_MMIO_BLK_TYPE,
DRIVERLOCALTYPE, DRIVER_LOCAL_TYPE,
DRIVERSCSITYPE, DRIVER_SCSI_TYPE,
DRIVERNVDIMMTYPE, DRIVER_NVDIMM_TYPE,
]; ];
#[derive(Debug, Clone)] #[derive(Debug, Clone)]
@ -381,7 +381,7 @@ fn mount_storage(logger: &Logger, storage: &Storage) -> Result<()> {
let logger = logger.new(o!("subsystem" => "mount")); let logger = logger.new(o!("subsystem" => "mount"));
match storage.fstype.as_str() { match storage.fstype.as_str() {
DRIVER9PTYPE | DRIVERVIRTIOFSTYPE => { DRIVER_9P_TYPE | DRIVER_VIRTIOFS_TYPE => {
let dest_path = Path::new(storage.mount_point.as_str()); let dest_path = Path::new(storage.mount_point.as_str());
if !dest_path.exists() { if !dest_path.exists() {
fs::create_dir_all(dest_path).context("Create mount destination failed")?; fs::create_dir_all(dest_path).context("Create mount destination failed")?;
@ -457,20 +457,22 @@ pub async fn add_storages(
"storage-type" => handler_name.to_owned())); "storage-type" => handler_name.to_owned()));
let res = match handler_name.as_str() { let res = match handler_name.as_str() {
DRIVERBLKTYPE => virtio_blk_storage_handler(&logger, &storage, sandbox.clone()).await, DRIVER_BLK_TYPE => virtio_blk_storage_handler(&logger, &storage, sandbox.clone()).await,
DRIVER9PTYPE => virtio9p_storage_handler(&logger, &storage, sandbox.clone()).await, DRIVER_9P_TYPE => virtio9p_storage_handler(&logger, &storage, sandbox.clone()).await,
DRIVERVIRTIOFSTYPE => { DRIVER_VIRTIOFS_TYPE => {
virtiofs_storage_handler(&logger, &storage, sandbox.clone()).await virtiofs_storage_handler(&logger, &storage, sandbox.clone()).await
} }
DRIVEREPHEMERALTYPE => { DRIVER_EPHEMERAL_TYPE => {
ephemeral_storage_handler(&logger, &storage, sandbox.clone()).await ephemeral_storage_handler(&logger, &storage, sandbox.clone()).await
} }
DRIVERMMIOBLKTYPE => { DRIVER_MMIO_BLK_TYPE => {
virtiommio_blk_storage_handler(&logger, &storage, sandbox.clone()).await virtiommio_blk_storage_handler(&logger, &storage, sandbox.clone()).await
} }
DRIVERLOCALTYPE => local_storage_handler(&logger, &storage, sandbox.clone()).await, DRIVER_LOCAL_TYPE => local_storage_handler(&logger, &storage, sandbox.clone()).await,
DRIVERSCSITYPE => virtio_scsi_storage_handler(&logger, &storage, sandbox.clone()).await, DRIVER_SCSI_TYPE => {
DRIVERNVDIMMTYPE => nvdimm_storage_handler(&logger, &storage, sandbox.clone()).await, virtio_scsi_storage_handler(&logger, &storage, sandbox.clone()).await
}
DRIVER_NVDIMM_TYPE => nvdimm_storage_handler(&logger, &storage, sandbox.clone()).await,
_ => { _ => {
return Err(anyhow!( return Err(anyhow!(
"Failed to find the storage handler {}", "Failed to find the storage handler {}",

4
src/agent/src/sandbox.rs
View File

@ -4,7 +4,7 @@
// //
use crate::linux_abi::*; use crate::linux_abi::*;
use crate::mount::{get_mount_fs_type, remove_mounts, TYPEROOTFS}; use crate::mount::{get_mount_fs_type, remove_mounts, TYPE_ROOTFS};
use crate::namespace::Namespace; use crate::namespace::Namespace;
use crate::network::Network; use crate::network::Network;
use anyhow::{anyhow, Context, Result}; use anyhow::{anyhow, Context, Result};
@ -71,7 +71,7 @@ impl Sandbox {
sandbox_pidns: None, sandbox_pidns: None,
storages: HashMap::new(), storages: HashMap::new(),
running: false, running: false,
no_pivot_root: fs_type.eq(TYPEROOTFS), no_pivot_root: fs_type.eq(TYPE_ROOTFS),
sender: None, sender: None,
rtnl: Some(RtnlHandle::new(NETLINK_ROUTE, 0).unwrap()), rtnl: Some(RtnlHandle::new(NETLINK_ROUTE, 0).unwrap()),
hooks: None, hooks: None,