proxmox-backup/src/api2/node/time.rs

use failure::*;

use crate::tools;
use crate::api_schema::*;
use crate::api_schema::router::*;
use crate::api2::types::*;

use serde_json::{json, Value};

use chrono::prelude::*;

fn read_etc_localtime() -> Result<String, Error> {
    // use /etc/timezone
    if let Ok(line) = tools::file_read_firstline("/etc/timezone") {
        return Ok(line.trim().to_owned());
    }

    // otherwise guess from the /etc/localtime symlink
    let mut buf: [u8; 64] = unsafe { std::mem::uninitialized() };
    let len = unsafe {
        libc::readlink("/etc/localtime".as_ptr() as *const _, buf.as_mut_ptr() as *mut _, buf.len())
    };
    if len <= 0 {
        bail!("failed to guess timezone");
    }
    let len = len as usize;
    buf[len] = 0;
    let link = std::str::from_utf8(&buf[..len])?;
    match link.rfind("/zoneinfo/") {
        Some(pos) => Ok(link[(pos + 10)..].to_string()),
        None => Ok(link.to_string()),
    }
}

fn get_time(
    _param: Value,
    _info: &ApiMethod,
    _rpcenv: &mut dyn RpcEnvironment,
) -> Result<Value, Error> {

    let datetime = Local::now();
    let offset = datetime.offset();
    let time = datetime.timestamp();
    let localtime = time + (offset.fix().local_minus_utc() as i64);

    Ok(json!({
        "timezone": read_etc_localtime()?,
        "time": time,
        "localtime": localtime,
    }))
}

fn set_timezone(
    param: Value,
    _info: &ApiMethod,
    _rpcenv: &mut dyn RpcEnvironment,
) -> Result<Value, Error> {

    let timezone = tools::required_string_param(&param, "timezone")?;

    let path = std::path::PathBuf::from(format!("/usr/share/zoneinfo/{}", timezone));

    if !path.exists() {
        bail!("No such timezone.");
    }

    tools::file_set_contents("/etc/timezone", timezone.as_bytes(), None)?;

    let _ = std::fs::remove_file("/etc/localtime");

    use std::os::unix::fs::symlink;
    symlink(path, "/etc/localtime")?;

    Ok(Value::Null)
}

pub fn router() -> Router {

    let route = Router::new()
        .get(
            ApiMethod::new(
                get_time,
                ObjectSchema::new("Read server time and time zone settings.")
                    .required("node", NODE_SCHEMA.clone())
             ).returns(
                ObjectSchema::new("Returns server time and timezone.")
                    .required("timezone", StringSchema::new("Time zone"))
                    .required("time", IntegerSchema::new("Seconds since 1970-01-01 00:00:00 UTC.")
                              .minimum(1297163644))
                    .required("localtime", IntegerSchema::new("Seconds since 1970-01-01 00:00:00 UTC. (local time)")
                              .minimum(1297163644))
            )
        )
        .put(
            ApiMethod::new(
                set_timezone,
                ObjectSchema::new("Set time zone.")
                    .required("node", NODE_SCHEMA.clone())
                    .required("timezone", StringSchema::new(
                        "Time zone. The file '/usr/share/zoneinfo/zone.tab' contains the list of valid names."))
            ).protected(true).reload_timezone(true)
        );


    route
}