//! Tools and utilities
//!
//! This is a collection of small and useful tools.
-
use failure::*;
use nix::unistd;
use nix::sys::stat;
+use nix::{convert_ioctl_res, request_code_read, ioc};
use lazy_static::lazy_static;
use std::io::Read;
use std::io::ErrorKind;
use std::time::Duration;
+use std::any::Any;
+
+use std::os::unix::io::{AsRawFd, FromRawFd, IntoRawFd, RawFd};
-use std::os::unix::io::AsRawFd;
+use std::collections::HashMap;
use serde_json::Value;
+use proxmox::tools::vec;
+
+pub mod async_mutex;
pub mod timer;
+pub mod wrapped_reader_stream;
+#[macro_use]
+pub mod common_regex;
+pub mod ticket;
+pub mod borrow;
+pub mod fs;
+pub mod tty;
+pub mod signalfd;
+pub mod daemon;
+pub mod procfs;
+pub mod acl;
+pub mod xattr;
+pub mod futures;
+
+mod process_locker;
+pub use process_locker::*;
+
+#[macro_use]
+mod file_logger;
+pub use file_logger::*;
+
+mod broadcast_future;
+pub use broadcast_future::*;
+
+/// Macro to write error-handling blocks (like perl eval {})
+///
+/// #### Example:
+/// ```
+/// # #[macro_use] extern crate proxmox_backup;
+/// # use failure::*;
+/// # let some_condition = false;
+/// let result = try_block!({
+/// if (some_condition) {
+/// bail!("some error");
+/// }
+/// Ok(())
+/// })
+/// .map_err(|e| format_err!("my try block returned an error - {}", e));
+/// ```
+
+#[macro_export]
+macro_rules! try_block {
+ { $($token:tt)* } => {{ (|| -> Result<_,_> { $($token)* })() }}
+}
+
-/// The `BufferedReader` trait provides a single function
+/// The `BufferedRead` trait provides a single function
/// `buffered_read`. It returns a reference to an internal buffer. The
/// purpose of this traid is to avoid unnecessary data copies.
-pub trait BufferedReader {
+pub trait BufferedRead {
/// This functions tries to fill the internal buffers, then
/// returns a reference to the available data. It returns an empty
/// buffer if `offset` points to the end of the file.
Ok(unsafe { &mut * (buffer.as_ptr() as *mut T) })
}
-/// Atomically write a file. We first create a temporary file, which
-/// is then renamed.
+pub fn file_read_firstline<P: AsRef<Path>>(path: P) -> Result<String, Error> {
+
+ let path = path.as_ref();
+
+ try_block!({
+ let file = std::fs::File::open(path)?;
+
+ use std::io::{BufRead, BufReader};
+
+ let mut reader = BufReader::new(file);
+
+ let mut line = String::new();
+
+ let _ = reader.read_line(&mut line)?;
+
+ Ok(line)
+ }).map_err(|err: Error| format_err!("unable to read {:?} - {}", path, err))
+}
+
+pub fn file_get_contents<P: AsRef<Path>>(path: P) -> Result<Vec<u8>, Error> {
+
+ let path = path.as_ref();
+
+ try_block!({
+ std::fs::read(path)
+ }).map_err(|err| format_err!("unable to read {:?} - {}", path, err))
+}
+
+pub fn file_get_json<P: AsRef<Path>>(path: P, default: Option<Value>) -> Result<Value, Error> {
+
+ let path = path.as_ref();
+
+ let raw = match std::fs::read(path) {
+ Ok(v) => v,
+ Err(err) => {
+ if err.kind() == std::io::ErrorKind::NotFound {
+ if let Some(v) = default {
+ return Ok(v);
+ }
+ }
+ bail!("unable to read json {:?} - {}", path, err);
+ }
+ };
+
+ try_block!({
+ let data = String::from_utf8(raw)?;
+ let json = serde_json::from_str(&data)?;
+ Ok(json)
+ }).map_err(|err: Error| format_err!("unable to parse json from {:?} - {}", path, err))
+}
+
+/// Atomically write a file
+///
+/// We first create a temporary file, which is then renamed.
pub fn file_set_contents<P: AsRef<Path>>(
path: P,
data: &[u8],
perm: Option<stat::Mode>,
) -> Result<(), Error> {
+ file_set_contents_full(path, data, perm, None, None)
+}
+
+/// Atomically write a file with owner and group
+pub fn file_set_contents_full<P: AsRef<Path>>(
+ path: P,
+ data: &[u8],
+ perm: Option<stat::Mode>,
+ owner: Option<unistd::Uid>,
+ group: Option<unistd::Gid>,
+) -> Result<(), Error> {
let path = path.as_ref();
bail!("fchmod {:?} failed: {}", tmp_path, err);
}
- use std::os::unix::io::FromRawFd;
+ if owner != None || group != None {
+ if let Err(err) = fchown(fd, owner, group) {
+ let _ = unistd::unlink(tmp_path);
+ bail!("fchown {:?} failed: {}", tmp_path, err);
+ }
+ }
+
let mut file = unsafe { File::from_raw_fd(fd) };
if let Err(err) = file.write_all(data) {
file: &mut F,
exclusive: bool,
timeout: Option<Duration>,
- ) -> Result<(), Error>
-{
+) -> Result<(), Error> {
let lockarg =
if exclusive {
nix::fcntl::FlockArg::LockExclusive
if chunk_size > READ_BUFFER_SIZE { bail!("chunk size too large!"); }
- let mut buf = vec![0u8; READ_BUFFER_SIZE];
+ let mut buf = vec::undefined(READ_BUFFER_SIZE);
let mut pos = 0;
let mut file_pos = 0;
Ok(())
}
+/// Returns the Unix uid/gid for the sepcified system user.
+pub fn getpwnam_ugid(username: &str) -> Result<(libc::uid_t,libc::gid_t), Error> {
+ let info = unsafe { libc::getpwnam(std::ffi::CString::new(username).unwrap().as_ptr()) };
+ if info == std::ptr::null_mut() {
+ bail!("getwpnam '{}' failed", username);
+ }
+
+ let info = unsafe { *info };
+
+ Ok((info.pw_uid, info.pw_gid))
+}
+
+/// Creates directory at the provided path with specified ownership
+///
+/// Simply returns if the directory already exists.
+pub fn create_dir_chown<P: AsRef<Path>>(
+ path: P,
+ perm: Option<stat::Mode>,
+ owner: Option<unistd::Uid>,
+ group: Option<unistd::Gid>,
+) -> Result<(), nix::Error>
+{
+ let mode : stat::Mode = perm.unwrap_or(stat::Mode::from_bits_truncate(0o770));
+
+ let path = path.as_ref();
+
+ match nix::unistd::mkdir(path, mode) {
+ Ok(()) => {},
+ Err(nix::Error::Sys(nix::errno::Errno::EEXIST)) => {
+ return Ok(());
+ },
+ err => return err,
+ }
+
+ unistd::chown(path, owner, group)?;
+
+ Ok(())
+}
+
+/// Change ownership of an open file handle
+pub fn fchown(
+ fd: RawFd,
+ owner: Option<nix::unistd::Uid>,
+ group: Option<nix::unistd::Gid>
+) -> Result<(), Error> {
+
+ // According to the POSIX specification, -1 is used to indicate that owner and group
+ // are not to be changed. Since uid_t and gid_t are unsigned types, we have to wrap
+ // around to get -1 (copied fron nix crate).
+ let uid = owner.map(Into::into).unwrap_or((0 as libc::uid_t).wrapping_sub(1));
+ let gid = group.map(Into::into).unwrap_or((0 as libc::gid_t).wrapping_sub(1));
+
+ let res = unsafe { libc::fchown(fd, uid, gid) };
+ nix::errno::Errno::result(res)?;
+
+ Ok(())
+}
+
+// Returns the hosts node name (UTS node name)
pub fn nodename() -> &'static str {
lazy_static!{
&NODENAME
}
+pub fn json_object_to_query(data: Value) -> Result<String, Error> {
+
+ let mut query = url::form_urlencoded::Serializer::new(String::new());
+
+ let object = data.as_object().ok_or_else(|| {
+ format_err!("json_object_to_query: got wrong data type (expected object).")
+ })?;
+
+ for (key, value) in object {
+ match value {
+ Value::Bool(b) => { query.append_pair(key, &b.to_string()); }
+ Value::Number(n) => { query.append_pair(key, &n.to_string()); }
+ Value::String(s) => { query.append_pair(key, &s); }
+ Value::Array(arr) => {
+ for element in arr {
+ match element {
+ Value::Bool(b) => { query.append_pair(key, &b.to_string()); }
+ Value::Number(n) => { query.append_pair(key, &n.to_string()); }
+ Value::String(s) => { query.append_pair(key, &s); }
+ _ => bail!("json_object_to_query: unable to handle complex array data types."),
+ }
+ }
+ }
+ _ => bail!("json_object_to_query: unable to handle complex data types."),
+ }
+ }
+
+ Ok(query.finish())
+}
+
pub fn required_string_param<'a>(param: &'a Value, name: &str) -> Result<&'a str, Error> {
match param[name].as_str() {
Some(s) => Ok(s),
None => bail!("missing parameter '{}'", name),
}
}
+
+pub fn required_array_param<'a>(param: &'a Value, name: &str) -> Result<Vec<Value>, Error> {
+ match param[name].as_array() {
+ Some(s) => Ok(s.to_vec()),
+ None => bail!("missing parameter '{}'", name),
+ }
+}
+
+pub fn complete_file_name(arg: &str, _param: &HashMap<String, String>) -> Vec<String> {
+
+ let mut result = vec![];
+
+ use nix::fcntl::OFlag;
+ use nix::sys::stat::Mode;
+ use nix::fcntl::AtFlags;
+
+ let mut dirname = std::path::PathBuf::from(if arg.len() == 0 { "./" } else { arg });
+
+ let is_dir = match nix::sys::stat::fstatat(libc::AT_FDCWD, &dirname, AtFlags::empty()) {
+ Ok(stat) => (stat.st_mode & libc::S_IFMT) == libc::S_IFDIR,
+ Err(_) => false,
+ };
+
+ if !is_dir {
+ if let Some(parent) = dirname.parent() {
+ dirname = parent.to_owned();
+ }
+ }
+
+ let mut dir = match nix::dir::Dir::openat(libc::AT_FDCWD, &dirname, OFlag::O_DIRECTORY, Mode::empty()) {
+ Ok(d) => d,
+ Err(_) => return result,
+ };
+
+ for item in dir.iter() {
+ if let Ok(entry) = item {
+ if let Ok(name) = entry.file_name().to_str() {
+ if name == "." || name == ".." { continue; }
+ let mut newpath = dirname.clone();
+ newpath.push(name);
+
+ if let Ok(stat) = nix::sys::stat::fstatat(libc::AT_FDCWD, &newpath, AtFlags::empty()) {
+ if (stat.st_mode & libc::S_IFMT) == libc::S_IFDIR {
+ newpath.push("");
+ if let Some(newpath) = newpath.to_str() {
+ result.push(newpath.to_owned());
+ }
+ continue;
+ }
+ }
+ if let Some(newpath) = newpath.to_str() {
+ result.push(newpath.to_owned());
+ }
+
+ }
+ }
+ }
+
+ result
+}
+
+/// Scan directory for matching file names.
+///
+/// Scan through all directory entries and call `callback()` function
+/// if the entry name matches the regular expression. This function
+/// used unix `openat()`, so you can pass absolute or relative file
+/// names. This function simply skips non-UTF8 encoded names.
+pub fn scandir<P, F>(
+ dirfd: RawFd,
+ path: &P,
+ regex: ®ex::Regex,
+ mut callback: F
+) -> Result<(), Error>
+ where F: FnMut(RawFd, &str, nix::dir::Type) -> Result<(), Error>,
+ P: ?Sized + nix::NixPath,
+{
+ for entry in self::fs::scan_subdir(dirfd, path, regex)? {
+ let entry = entry?;
+ let file_type = match entry.file_type() {
+ Some(file_type) => file_type,
+ None => bail!("unable to detect file type"),
+ };
+
+ callback(entry.parent_fd(), unsafe { entry.file_name_utf8_unchecked() }, file_type)?;
+ }
+ Ok(())
+}
+
+pub fn get_hardware_address() -> Result<String, Error> {
+
+ static FILENAME: &str = "/etc/ssh/ssh_host_rsa_key.pub";
+
+ let contents = file_get_contents(FILENAME)?;
+ let digest = md5::compute(contents);
+
+ Ok(format!("{:0x}", digest))
+}
+
+
+pub fn assert_if_modified(digest1: &str, digest2: &str) -> Result<(), Error> {
+ if digest1 != digest2 {
+ bail!("detected modified configuration - file changed by other user? Try again.");
+ }
+ Ok(())
+}
+
+/// Extract authentication cookie from cookie header.
+/// We assume cookie_name is already url encoded.
+pub fn extract_auth_cookie(cookie: &str, cookie_name: &str) -> Option<String> {
+
+ for pair in cookie.split(';') {
+
+ let (name, value) = match pair.find('=') {
+ Some(i) => (pair[..i].trim(), pair[(i + 1)..].trim()),
+ None => return None, // Cookie format error
+ };
+
+ if name == cookie_name {
+ use url::percent_encoding::percent_decode;
+ if let Ok(value) = percent_decode(value.as_bytes()).decode_utf8() {
+ return Some(value.into());
+ } else {
+ return None; // Cookie format error
+ }
+ }
+ }
+
+ None
+}
+
+pub fn join(data: &Vec<String>, sep: char) -> String {
+
+ let mut list = String::new();
+
+ for item in data {
+ if list.len() != 0 { list.push(sep); }
+ list.push_str(item);
+ }
+
+ list
+}
+
+/// normalize uri path
+///
+/// Do not allow ".", "..", or hidden files ".XXXX"
+/// Also remove empty path components
+pub fn normalize_uri_path(path: &str) -> Result<(String, Vec<&str>), Error> {
+
+ let items = path.split('/');
+
+ let mut path = String::new();
+ let mut components = vec![];
+
+ for name in items {
+ if name.is_empty() { continue; }
+ if name.starts_with(".") {
+ bail!("Path contains illegal components.");
+ }
+ path.push('/');
+ path.push_str(name);
+ components.push(name);
+ }
+
+ Ok((path, components))
+}
+
+pub fn fd_change_cloexec(fd: RawFd, on: bool) -> Result<(), Error> {
+ use nix::fcntl::{fcntl, F_GETFD, F_SETFD, FdFlag};
+ let mut flags = FdFlag::from_bits(fcntl(fd, F_GETFD)?)
+ .ok_or_else(|| format_err!("unhandled file flags"))?; // nix crate is stupid this way...
+ flags.set(FdFlag::FD_CLOEXEC, on);
+ fcntl(fd, F_SETFD(flags))?;
+ Ok(())
+}
+
+
+static mut SHUTDOWN_REQUESTED: bool = false;
+
+pub fn request_shutdown() {
+ unsafe { SHUTDOWN_REQUESTED = true; }
+ crate::server::server_shutdown();
+}
+
+#[inline(always)]
+pub fn shutdown_requested() -> bool {
+ unsafe { SHUTDOWN_REQUESTED }
+}
+
+pub fn fail_on_shutdown() -> Result<(), Error> {
+ if shutdown_requested() {
+ bail!("Server shutdown requested - aborting task");
+ }
+ Ok(())
+}
+
+/// Guard a raw file descriptor with a drop handler. This is mostly useful when access to an owned
+/// `RawFd` is required without the corresponding handler object (such as when only the file
+/// descriptor number is required in a closure which may be dropped instead of being executed).
+pub struct Fd(pub RawFd);
+
+impl Drop for Fd {
+ fn drop(&mut self) {
+ if self.0 != -1 {
+ unsafe {
+ libc::close(self.0);
+ }
+ }
+ }
+}
+
+impl AsRawFd for Fd {
+ fn as_raw_fd(&self) -> RawFd {
+ self.0
+ }
+}
+
+impl IntoRawFd for Fd {
+ fn into_raw_fd(mut self) -> RawFd {
+ let fd = self.0;
+ self.0 = -1;
+ fd
+ }
+}
+
+impl FromRawFd for Fd {
+ unsafe fn from_raw_fd(fd: RawFd) -> Self {
+ Self(fd)
+ }
+}
+
+// wrap nix::unistd::pipe2 + O_CLOEXEC into something returning guarded file descriptors
+pub fn pipe() -> Result<(Fd, Fd), Error> {
+ let (pin, pout) = nix::unistd::pipe2(nix::fcntl::OFlag::O_CLOEXEC)?;
+ Ok((Fd(pin), Fd(pout)))
+}
+
+/// An easy way to convert types to Any
+///
+/// Mostly useful to downcast trait objects (see RpcEnvironment).
+pub trait AsAny {
+ fn as_any(&self) -> &dyn Any;
+}
+
+impl<T: Any> AsAny for T {
+ fn as_any(&self) -> &dyn Any { self }
+}
+
+
+// /usr/include/linux/fs.h: #define BLKGETSIZE64 _IOR(0x12,114,size_t)
+// return device size in bytes (u64 *arg)
+nix::ioctl_read!(blkgetsize64, 0x12, 114, u64);
+
+/// Return file or block device size
+pub fn image_size(path: &Path) -> Result<u64, Error> {
+
+ use std::os::unix::fs::FileTypeExt;
+
+ let file = std::fs::File::open(path)?;
+ let metadata = file.metadata()?;
+ let file_type = metadata.file_type();
+
+ if file_type.is_block_device() {
+ let mut size : u64 = 0;
+ let res = unsafe { blkgetsize64(file.as_raw_fd(), &mut size) };
+
+ if let Err(err) = res {
+ bail!("blkgetsize64 failed for {:?} - {}", path, err);
+ }
+ Ok(size)
+ } else if file_type.is_file() {
+ Ok(metadata.len())
+ } else {
+ bail!("image size failed - got unexpected file type {:?}", file_type);
+ }
+}