src/server/mod.rs

   1 //! Proxmox Server/Service framework
   2 //!
   3 //! This code provides basic primitives to build our REST API
   4 //! services. We want async IO, so this is built on top of
   5 //! tokio/hyper.
   6
   7 use anyhow::{format_err, Error};
   8 use lazy_static::lazy_static;
   9 use nix::unistd::Pid;
  10 use serde_json::Value;
  11
  12 use proxmox::sys::linux::procfs::PidStat;
  13 use proxmox::tools::fs::{create_path, CreateOptions};
  14
  15 use pbs_buildcfg;
  16
  17 lazy_static! {
  18     static ref PID: i32 = unsafe { libc::getpid() };
  19     static ref PSTART: u64 = PidStat::read_from_pid(Pid::from_raw(*PID)).unwrap().starttime;
  20 }
  21
  22 pub fn pid() -> i32 {
  23     *PID
  24 }
  25
  26 pub fn pstart() -> u64 {
  27     *PSTART
  28 }
  29
  30 pub fn write_pid(pid_fn: &str) -> Result<(), Error> {
  31     let pid_str = format!("{}\n", *PID);
  32     proxmox::tools::fs::replace_file(pid_fn, pid_str.as_bytes(), CreateOptions::new())
  33 }
  34
  35 pub fn read_pid(pid_fn: &str) -> Result<i32, Error> {
  36     let pid = proxmox::tools::fs::file_get_contents(pid_fn)?;
  37     let pid = std::str::from_utf8(&pid)?.trim();
  38     pid.parse().map_err(|err| format_err!("could not parse pid - {}", err))
  39 }
  40
  41 pub fn ctrl_sock_from_pid(pid: i32) -> String {
  42     format!("\0{}/control-{}.sock", pbs_buildcfg::PROXMOX_BACKUP_RUN_DIR, pid)
  43 }
  44
  45 pub fn our_ctrl_sock() -> String {
  46     ctrl_sock_from_pid(*PID)
  47 }
  48
  49 mod upid;
  50 pub use upid::*;
  51
  52 mod worker_task;
  53 pub use worker_task::*;
  54
  55 mod h2service;
  56 pub use h2service::*;
  57
  58 #[macro_use]
  59 pub mod rest;
  60
  61 pub mod jobstate;
  62
  63 mod verify_job;
  64 pub use verify_job::*;
  65
  66 mod prune_job;
  67 pub use prune_job::*;
  68
  69 mod gc_job;
  70 pub use gc_job::*;
  71
  72 mod email_notifications;
  73 pub use email_notifications::*;
  74
  75 mod report;
  76 pub use report::*;
  77
  78 pub mod ticket;
  79
  80 pub mod auth;
  81
  82 pub mod pull;
  83
  84 pub(crate) async fn reload_proxy_certificate() -> Result<(), Error> {
  85     let proxy_pid = crate::server::read_pid(pbs_buildcfg::PROXMOX_BACKUP_PROXY_PID_FN)?;
  86     let sock = crate::server::ctrl_sock_from_pid(proxy_pid);
  87     let _: Value = proxmox_rest_server::send_raw_command(sock, "{\"command\":\"reload-certificate\"}\n")
  88         .await?;
  89     Ok(())
  90 }
  91
  92 pub(crate) async fn notify_datastore_removed() -> Result<(), Error> {
  93     let proxy_pid = crate::server::read_pid(pbs_buildcfg::PROXMOX_BACKUP_PROXY_PID_FN)?;
  94     let sock = crate::server::ctrl_sock_from_pid(proxy_pid);
  95     let _: Value = proxmox_rest_server::send_raw_command(sock, "{\"command\":\"datastore-removed\"}\n")
  96         .await?;
  97     Ok(())
  98 }
  99
 100 /// Create the base run-directory.
 101 ///
 102 /// This exists to fixate the permissions for the run *base* directory while allowing intermediate
 103 /// directories after it to have different permissions.
 104 pub fn create_run_dir() -> Result<(), Error> {
 105     let backup_user = pbs_config::backup_user()?;
 106     let opts = CreateOptions::new()
 107         .owner(backup_user.uid)
 108         .group(backup_user.gid);
 109     let _: bool = create_path(pbs_buildcfg::PROXMOX_BACKUP_RUN_DIR_M!(), None, Some(opts))?;
 110     Ok(())
 111 }