Line data Source code
1 : use anyhow::Context;
2 : use tracing::instrument;
3 :
4 : pub const DISK_QUOTA_BIN: &str = "/neonvm/bin/set-disk-quota";
5 :
6 : /// If size_bytes is 0, it disables the quota. Otherwise, it sets filesystem quota to size_bytes.
7 : /// `fs_mountpoint` should point to the mountpoint of the filesystem where the quota should be set.
8 : #[instrument]
9 : pub fn set_disk_quota(size_bytes: u64, fs_mountpoint: &str) -> anyhow::Result<()> {
10 : let size_kb = size_bytes / 1024;
11 : // run `/neonvm/bin/set-disk-quota {size_kb} {mountpoint}`
12 : let child_result = std::process::Command::new("/usr/bin/sudo")
13 : .arg(DISK_QUOTA_BIN)
14 : .arg(size_kb.to_string())
15 : .arg(fs_mountpoint)
16 : .spawn();
17 :
18 : child_result
19 : .context("spawn() failed")
20 0 : .and_then(|mut child| child.wait().context("wait() failed"))
21 0 : .and_then(|status| match status.success() {
22 0 : true => Ok(()),
23 0 : false => Err(anyhow::anyhow!("process exited with {status}")),
24 0 : })
25 : // wrap any prior error with the overall context that we couldn't run the command
26 0 : .with_context(|| format!("could not run `/usr/bin/sudo {DISK_QUOTA_BIN}`"))
27 : }
|