2 use std
::path
::{Path, PathBuf}
;
3 use std
::os
::unix
::io
::AsRawFd
;
5 use anyhow
::{bail, format_err, Error}
;
8 use openssl
::ssl
::{SslMethod, SslAcceptor, SslFiletype}
;
10 use proxmox
::try_block
;
11 use proxmox
::api
::RpcEnvironmentType
;
23 rotate_task_log_archive
,
25 tools
::systemd
::time
::{
32 use proxmox_backup
::api2
::types
::Authid
;
33 use proxmox_backup
::configdir
;
34 use proxmox_backup
::buildcfg
;
35 use proxmox_backup
::server
;
36 use proxmox_backup
::auth_helpers
::*;
37 use proxmox_backup
::tools
::{
46 PROXMOX_BACKUP_TCP_KEEPALIVE_TIME
,
50 use proxmox_backup
::api2
::pull
::do_sync_job
;
51 use proxmox_backup
::server
::do_verification_job
;
52 use proxmox_backup
::server
::do_prune_job
;
54 fn main() -> Result
<(), Error
> {
55 proxmox_backup
::tools
::setup_safe_path_env();
57 let backup_uid
= proxmox_backup
::backup
::backup_user()?
.uid
;
58 let backup_gid
= proxmox_backup
::backup
::backup_group()?
.gid
;
59 let running_uid
= nix
::unistd
::Uid
::effective();
60 let running_gid
= nix
::unistd
::Gid
::effective();
62 if running_uid
!= backup_uid
|| running_gid
!= backup_gid
{
63 bail
!("proxy not running as backup user or group (got uid {} gid {})", running_uid
, running_gid
);
66 proxmox_backup
::tools
::runtime
::main(run())
69 async
fn run() -> Result
<(), Error
> {
70 if let Err(err
) = syslog
::init(
71 syslog
::Facility
::LOG_DAEMON
,
72 log
::LevelFilter
::Info
,
73 Some("proxmox-backup-proxy")) {
74 bail
!("unable to inititialize syslog - {}", err
);
77 // Note: To debug early connection error use
78 // PROXMOX_DEBUG=1 ./target/release/proxmox-backup-proxy
79 let debug
= std
::env
::var("PROXMOX_DEBUG").is_ok();
81 let _
= public_auth_key(); // load with lazy_static
82 let _
= csrf_secret(); // load with lazy_static
84 let mut config
= ApiConfig
::new(
85 buildcfg
::JS_DIR
, &proxmox_backup
::api2
::ROUTER
, RpcEnvironmentType
::PUBLIC
)?
;
87 config
.add_alias("novnc", "/usr/share/novnc-pve");
88 config
.add_alias("extjs", "/usr/share/javascript/extjs");
89 config
.add_alias("fontawesome", "/usr/share/fonts-font-awesome");
90 config
.add_alias("xtermjs", "/usr/share/pve-xtermjs");
91 config
.add_alias("locale", "/usr/share/pbs-i18n");
92 config
.add_alias("widgettoolkit", "/usr/share/javascript/proxmox-widget-toolkit");
93 config
.add_alias("css", "/usr/share/javascript/proxmox-backup/css");
94 config
.add_alias("docs", "/usr/share/doc/proxmox-backup/html");
96 let mut indexpath
= PathBuf
::from(buildcfg
::JS_DIR
);
97 indexpath
.push("index.hbs");
98 config
.register_template("index", &indexpath
)?
;
99 config
.register_template("console", "/usr/share/pve-xtermjs/index.html.hbs")?
;
101 let mut commando_sock
= server
::CommandoSocket
::new(server
::our_ctrl_sock());
103 config
.enable_file_log(buildcfg
::API_ACCESS_LOG_FN
, &mut commando_sock
)?
;
105 let rest_server
= RestServer
::new(config
);
107 //openssl req -x509 -newkey rsa:4096 -keyout /etc/proxmox-backup/proxy.key -out /etc/proxmox-backup/proxy.pem -nodes
108 let key_path
= configdir
!("/proxy.key");
109 let cert_path
= configdir
!("/proxy.pem");
111 let mut acceptor
= SslAcceptor
::mozilla_intermediate_v5(SslMethod
::tls()).unwrap();
112 acceptor
.set_private_key_file(key_path
, SslFiletype
::PEM
)
113 .map_err(|err
| format_err
!("unable to read proxy key {} - {}", key_path
, err
))?
;
114 acceptor
.set_certificate_chain_file(cert_path
)
115 .map_err(|err
| format_err
!("unable to read proxy cert {} - {}", cert_path
, err
))?
;
116 acceptor
.check_private_key().unwrap();
118 let acceptor
= Arc
::new(acceptor
.build());
120 let server
= daemon
::create_daemon(
121 ([0,0,0,0,0,0,0,0], 8007).into(),
124 let connections
= accept_connections(listener
, acceptor
, debug
);
125 let connections
= hyper
::server
::accept
::from_stream(connections
);
128 .and_then(|_
| hyper
::Server
::builder(connections
)
130 .with_graceful_shutdown(server
::shutdown_future())
131 .map_err(Error
::from
)
133 .map_err(|err
| eprintln
!("server error: {}", err
))
139 server
::write_pid(buildcfg
::PROXMOX_BACKUP_PROXY_PID_FN
)?
;
140 daemon
::systemd_notify(daemon
::SystemdNotify
::Ready
)?
;
142 let init_result
: Result
<(), Error
> = try_block
!({
143 server
::register_task_control_commands(&mut commando_sock
)?
;
144 commando_sock
.spawn()?
;
145 server
::server_state_init()?
;
149 if let Err(err
) = init_result
{
150 bail
!("unable to start daemon - {}", err
);
153 start_task_scheduler();
154 start_stat_generator();
157 log
::info
!("server shutting down, waiting for active workers to complete");
158 proxmox_backup
::server
::last_worker_future().await?
;
159 log
::info
!("done - exit server");
164 fn accept_connections(
165 mut listener
: tokio
::net
::TcpListener
,
166 acceptor
: Arc
<openssl
::ssl
::SslAcceptor
>,
168 ) -> tokio
::sync
::mpsc
::Receiver
<Result
<tokio_openssl
::SslStream
<tokio
::net
::TcpStream
>, Error
>> {
170 let (sender
, receiver
) = tokio
::sync
::mpsc
::channel(1024);
172 let accept_counter
= Arc
::new(());
174 const MAX_PENDING_ACCEPTS
: usize = 1024;
176 tokio
::spawn(async
move {
178 match listener
.accept().await
{
180 eprintln
!("error accepting tcp connection: {}", err
);
182 Ok((sock
, _addr
)) => {
183 sock
.set_nodelay(true).unwrap();
184 let _
= set_tcp_keepalive(sock
.as_raw_fd(), PROXMOX_BACKUP_TCP_KEEPALIVE_TIME
);
185 let acceptor
= Arc
::clone(&acceptor
);
186 let mut sender
= sender
.clone();
188 if Arc
::strong_count(&accept_counter
) > MAX_PENDING_ACCEPTS
{
190 eprintln
!("connection rejected - to many open connections");
195 let accept_counter
= accept_counter
.clone();
196 tokio
::spawn(async
move {
197 let accept_future
= tokio
::time
::timeout(
198 Duration
::new(10, 0), tokio_openssl
::accept(&acceptor
, sock
));
200 let result
= accept_future
.await
;
203 Ok(Ok(connection
)) => {
204 if let Err(_
) = sender
.send(Ok(connection
)).await
{
206 eprintln
!("detect closed connection channel");
212 eprintln
!("https handshake failed - {}", err
);
217 eprintln
!("https handshake timeout");
222 drop(accept_counter
); // decrease reference count
232 fn start_stat_generator() {
233 let abort_future
= server
::shutdown_future();
234 let future
= Box
::pin(run_stat_generator());
235 let task
= futures
::future
::select(future
, abort_future
);
236 tokio
::spawn(task
.map(|_
| ()));
239 fn start_task_scheduler() {
240 let abort_future
= server
::shutdown_future();
241 let future
= Box
::pin(run_task_scheduler());
242 let task
= futures
::future
::select(future
, abort_future
);
243 tokio
::spawn(task
.map(|_
| ()));
246 use std
::time
::{SystemTime, Instant, Duration, UNIX_EPOCH}
;
248 fn next_minute() -> Result
<Instant
, Error
> {
249 let now
= SystemTime
::now();
250 let epoch_now
= now
.duration_since(UNIX_EPOCH
)?
;
251 let epoch_next
= Duration
::from_secs((epoch_now
.as_secs()/60 + 1)*60);
252 Ok(Instant
::now() + epoch_next
- epoch_now
)
255 async
fn run_task_scheduler() {
257 let mut count
: usize = 0;
262 let delay_target
= match next_minute() { // try to run very minute
265 eprintln
!("task scheduler: compute next minute failed - {}", err
);
266 tokio
::time
::delay_until(tokio
::time
::Instant
::from_std(Instant
::now() + Duration
::from_secs(60))).await
;
271 if count
> 2 { // wait 1..2 minutes before starting
272 match schedule_tasks().catch_unwind().await
{
274 match panic
.downcast
::<&str>() {
276 eprintln
!("task scheduler panic: {}", msg
);
279 eprintln
!("task scheduler panic - unknown type");
284 eprintln
!("task scheduler failed - {:?}", err
);
290 tokio
::time
::delay_until(tokio
::time
::Instant
::from_std(delay_target
)).await
;
294 async
fn schedule_tasks() -> Result
<(), Error
> {
296 schedule_datastore_garbage_collection().await
;
297 schedule_datastore_prune().await
;
298 schedule_datastore_sync_jobs().await
;
299 schedule_datastore_verify_jobs().await
;
300 schedule_task_log_rotate().await
;
305 async
fn schedule_datastore_garbage_collection() {
307 use proxmox_backup
::config
::{
314 let config
= match datastore
::config() {
316 eprintln
!("unable to read datastore config - {}", err
);
319 Ok((config
, _digest
)) => config
,
322 for (store
, (_
, store_config
)) in config
.sections
{
323 let datastore
= match DataStore
::lookup_datastore(&store
) {
324 Ok(datastore
) => datastore
,
326 eprintln
!("lookup_datastore failed - {}", err
);
331 let store_config
: DataStoreConfig
= match serde_json
::from_value(store_config
) {
334 eprintln
!("datastore config from_value failed - {}", err
);
339 let event_str
= match store_config
.gc_schedule
{
340 Some(event_str
) => event_str
,
344 let event
= match parse_calendar_event(&event_str
) {
347 eprintln
!("unable to parse schedule '{}' - {}", event_str
, err
);
352 if datastore
.garbage_collection_running() { continue; }
354 let worker_type
= "garbage_collection";
356 let last
= match jobstate
::last_run_time(worker_type
, &store
) {
359 eprintln
!("could not get last run time of {} {}: {}", worker_type
, store
, err
);
364 let next
= match compute_next_event(&event
, last
, false) {
365 Ok(Some(next
)) => next
,
366 Ok(None
) => continue,
368 eprintln
!("compute_next_event for '{}' failed - {}", event_str
, err
);
373 let now
= proxmox
::tools
::time
::epoch_i64();
375 if next
> now { continue; }
377 let job
= match Job
::new(worker_type
, &store
) {
379 Err(_
) => continue, // could not get lock
382 let auth_id
= Authid
::backup_auth_id();
384 if let Err(err
) = crate::server
::do_garbage_collection_job(job
, datastore
, auth_id
, Some(event_str
), false) {
385 eprintln
!("unable to start garbage collection job on datastore {} - {}", store
, err
);
390 async
fn schedule_datastore_prune() {
392 use proxmox_backup
::{
402 let config
= match datastore
::config() {
404 eprintln
!("unable to read datastore config - {}", err
);
407 Ok((config
, _digest
)) => config
,
410 for (store
, (_
, store_config
)) in config
.sections
{
412 let store_config
: DataStoreConfig
= match serde_json
::from_value(store_config
) {
415 eprintln
!("datastore '{}' config from_value failed - {}", store
, err
);
420 let event_str
= match store_config
.prune_schedule
{
421 Some(event_str
) => event_str
,
425 let prune_options
= PruneOptions
{
426 keep_last
: store_config
.keep_last
,
427 keep_hourly
: store_config
.keep_hourly
,
428 keep_daily
: store_config
.keep_daily
,
429 keep_weekly
: store_config
.keep_weekly
,
430 keep_monthly
: store_config
.keep_monthly
,
431 keep_yearly
: store_config
.keep_yearly
,
434 if !prune_options
.keeps_something() { // no prune settings - keep all
438 let worker_type
= "prune";
439 if check_schedule(worker_type
, &event_str
, &store
) {
440 let job
= match Job
::new(worker_type
, &store
) {
442 Err(_
) => continue, // could not get lock
445 let auth_id
= Authid
::backup_auth_id().clone();
446 if let Err(err
) = do_prune_job(job
, prune_options
, store
.clone(), &auth_id
, Some(event_str
)) {
447 eprintln
!("unable to start datastore prune job {} - {}", &store
, err
);
453 async
fn schedule_datastore_sync_jobs() {
455 use proxmox_backup
::config
::sync
::{
460 let config
= match sync
::config() {
462 eprintln
!("unable to read sync job config - {}", err
);
465 Ok((config
, _digest
)) => config
,
468 for (job_id
, (_
, job_config
)) in config
.sections
{
469 let job_config
: SyncJobConfig
= match serde_json
::from_value(job_config
) {
472 eprintln
!("sync job config from_value failed - {}", err
);
477 let event_str
= match job_config
.schedule
{
478 Some(ref event_str
) => event_str
.clone(),
482 let worker_type
= "syncjob";
483 if check_schedule(worker_type
, &event_str
, &job_id
) {
484 let job
= match Job
::new(worker_type
, &job_id
) {
486 Err(_
) => continue, // could not get lock
489 let auth_id
= Authid
::backup_auth_id().clone();
490 if let Err(err
) = do_sync_job(job
, job_config
, &auth_id
, Some(event_str
)) {
491 eprintln
!("unable to start datastore sync job {} - {}", &job_id
, err
);
497 async
fn schedule_datastore_verify_jobs() {
499 use proxmox_backup
::config
::verify
::{
501 VerificationJobConfig
,
504 let config
= match verify
::config() {
506 eprintln
!("unable to read verification job config - {}", err
);
509 Ok((config
, _digest
)) => config
,
511 for (job_id
, (_
, job_config
)) in config
.sections
{
512 let job_config
: VerificationJobConfig
= match serde_json
::from_value(job_config
) {
515 eprintln
!("verification job config from_value failed - {}", err
);
519 let event_str
= match job_config
.schedule
{
520 Some(ref event_str
) => event_str
.clone(),
524 let worker_type
= "verificationjob";
525 let auth_id
= Authid
::backup_auth_id().clone();
526 if check_schedule(worker_type
, &event_str
, &job_id
) {
527 let job
= match Job
::new(&worker_type
, &job_id
) {
529 Err(_
) => continue, // could not get lock
531 if let Err(err
) = do_verification_job(job
, job_config
, &auth_id
, Some(event_str
)) {
532 eprintln
!("unable to start datastore verification job {} - {}", &job_id
, err
);
538 async
fn schedule_task_log_rotate() {
540 let worker_type
= "logrotate";
541 let job_id
= "access-log_and_task-archive";
543 // schedule daily at 00:00 like normal logrotate
544 let schedule
= "00:00";
546 if !check_schedule(worker_type
, schedule
, job_id
) {
547 // if we never ran the rotation, schedule instantly
548 match jobstate
::JobState
::load(worker_type
, job_id
) {
549 Ok(state
) => match state
{
550 jobstate
::JobState
::Created { .. }
=> {}
,
557 let mut job
= match Job
::new(worker_type
, job_id
) {
559 Err(_
) => return, // could not get lock
562 if let Err(err
) = WorkerTask
::new_thread(
565 Authid
::backup_auth_id().clone(),
568 job
.start(&worker
.upid().to_string())?
;
569 worker
.log(format
!("starting task log rotation"));
571 let result
= try_block
!({
572 let max_size
= 512 * 1024 - 1; // an entry has ~ 100b, so > 5000 entries/file
573 let max_files
= 20; // times twenty files gives > 100000 task entries
574 let has_rotated
= rotate_task_log_archive(max_size
, true, Some(max_files
))?
;
576 worker
.log(format
!("task log archive was rotated"));
578 worker
.log(format
!("task log archive was not rotated"));
581 let max_size
= 32 * 1024 * 1024 - 1;
583 let mut logrotate
= LogRotate
::new(buildcfg
::API_ACCESS_LOG_FN
, true)
584 .ok_or_else(|| format_err
!("could not get API access log file names"))?
;
586 let has_rotated
= logrotate
.rotate(max_size
, None
, Some(max_files
))?
;
588 println
!("rotated access log, telling daemons to re-open log file");
589 proxmox_backup
::tools
::runtime
::block_on(command_reopen_logfiles())?
;
591 worker
.log(format
!("API access log was rotated"));
593 worker
.log(format
!("API access log was not rotated"));
599 let status
= worker
.create_state(&result
);
601 if let Err(err
) = job
.finish(status
) {
602 eprintln
!("could not finish job state for {}: {}", worker_type
, err
);
608 eprintln
!("unable to start task log rotation: {}", err
);
613 async
fn command_reopen_logfiles() -> Result
<(), Error
> {
614 // only care about the most recent daemon instance for each, proxy & api, as other older ones
615 // should not respond to new requests anyway, but only finish their current one and then exit.
616 let sock
= server
::our_ctrl_sock();
617 server
::send_command(sock
, serde_json
::json
!({
618 "command": "api-access-log-reopen",
621 let pid
= server
::read_pid(buildcfg
::PROXMOX_BACKUP_API_PID_FN
)?
;
622 let sock
= server
::ctrl_sock_from_pid(pid
);
623 server
::send_command(sock
, serde_json
::json
!({
624 "command": "api-access-log-reopen",
629 async
fn run_stat_generator() {
634 let save
= if count
>= 6 { count = 0; true }
else { false }
;
636 let delay_target
= Instant
::now() + Duration
::from_secs(10);
638 generate_host_stats(save
).await
;
640 tokio
::time
::delay_until(tokio
::time
::Instant
::from_std(delay_target
)).await
;
646 fn rrd_update_gauge(name
: &str, value
: f64, save
: bool
) {
647 use proxmox_backup
::rrd
;
648 if let Err(err
) = rrd
::update_value(name
, value
, rrd
::DST
::Gauge
, save
) {
649 eprintln
!("rrd::update_value '{}' failed - {}", name
, err
);
653 fn rrd_update_derive(name
: &str, value
: f64, save
: bool
) {
654 use proxmox_backup
::rrd
;
655 if let Err(err
) = rrd
::update_value(name
, value
, rrd
::DST
::Derive
, save
) {
656 eprintln
!("rrd::update_value '{}' failed - {}", name
, err
);
660 async
fn generate_host_stats(save
: bool
) {
661 use proxmox
::sys
::linux
::procfs
::{
662 read_meminfo
, read_proc_stat
, read_proc_net_dev
, read_loadavg
};
663 use proxmox_backup
::config
::datastore
;
666 proxmox_backup
::tools
::runtime
::block_in_place(move || {
668 match read_proc_stat() {
670 rrd_update_gauge("host/cpu", stat
.cpu
, save
);
671 rrd_update_gauge("host/iowait", stat
.iowait_percent
, save
);
674 eprintln
!("read_proc_stat failed - {}", err
);
678 match read_meminfo() {
680 rrd_update_gauge("host/memtotal", meminfo
.memtotal
as f64, save
);
681 rrd_update_gauge("host/memused", meminfo
.memused
as f64, save
);
682 rrd_update_gauge("host/swaptotal", meminfo
.swaptotal
as f64, save
);
683 rrd_update_gauge("host/swapused", meminfo
.swapused
as f64, save
);
686 eprintln
!("read_meminfo failed - {}", err
);
690 match read_proc_net_dev() {
692 use proxmox_backup
::config
::network
::is_physical_nic
;
696 if !is_physical_nic(&item
.device
) { continue; }
697 netin
+= item
.receive
;
700 rrd_update_derive("host/netin", netin
as f64, save
);
701 rrd_update_derive("host/netout", netout
as f64, save
);
704 eprintln
!("read_prox_net_dev failed - {}", err
);
708 match read_loadavg() {
710 rrd_update_gauge("host/loadavg", loadavg
.0 as f64, save
);
713 eprintln
!("read_loadavg failed - {}", err
);
717 let disk_manager
= DiskManage
::new();
719 gather_disk_stats(disk_manager
.clone(), Path
::new("/"), "host", save
);
721 match datastore
::config() {
723 let datastore_list
: Vec
<datastore
::DataStoreConfig
> =
724 config
.convert_to_typed_array("datastore").unwrap_or(Vec
::new());
726 for config
in datastore_list
{
728 let rrd_prefix
= format
!("datastore/{}", config
.name
);
729 let path
= std
::path
::Path
::new(&config
.path
);
730 gather_disk_stats(disk_manager
.clone(), path
, &rrd_prefix
, save
);
734 eprintln
!("read datastore config failed - {}", err
);
741 fn check_schedule(worker_type
: &str, event_str
: &str, id
: &str) -> bool
{
742 let event
= match parse_calendar_event(event_str
) {
745 eprintln
!("unable to parse schedule '{}' - {}", event_str
, err
);
750 let last
= match jobstate
::last_run_time(worker_type
, &id
) {
753 eprintln
!("could not get last run time of {} {}: {}", worker_type
, id
, err
);
758 let next
= match compute_next_event(&event
, last
, false) {
759 Ok(Some(next
)) => next
,
760 Ok(None
) => return false,
762 eprintln
!("compute_next_event for '{}' failed - {}", event_str
, err
);
767 let now
= proxmox
::tools
::time
::epoch_i64();
771 fn gather_disk_stats(disk_manager
: Arc
<DiskManage
>, path
: &Path
, rrd_prefix
: &str, save
: bool
) {
773 match proxmox_backup
::tools
::disks
::disk_usage(path
) {
775 let rrd_key
= format
!("{}/total", rrd_prefix
);
776 rrd_update_gauge(&rrd_key
, status
.total
as f64, save
);
777 let rrd_key
= format
!("{}/used", rrd_prefix
);
778 rrd_update_gauge(&rrd_key
, status
.used
as f64, save
);
781 eprintln
!("read disk_usage on {:?} failed - {}", path
, err
);
785 match disk_manager
.find_mounted_device(path
) {
787 Ok(Some((fs_type
, device
, source
))) => {
788 let mut device_stat
= None
;
789 match fs_type
.as_str() {
791 if let Some(pool
) = source
{
792 match zfs_pool_stats(&pool
) {
793 Ok(stat
) => device_stat
= stat
,
794 Err(err
) => eprintln
!("zfs_pool_stats({:?}) failed - {}", pool
, err
),
799 if let Ok(disk
) = disk_manager
.clone().disk_by_dev_num(device
.into_dev_t()) {
800 match disk
.read_stat() {
801 Ok(stat
) => device_stat
= stat
,
802 Err(err
) => eprintln
!("disk.read_stat {:?} failed - {}", path
, err
),
807 if let Some(stat
) = device_stat
{
808 let rrd_key
= format
!("{}/read_ios", rrd_prefix
);
809 rrd_update_derive(&rrd_key
, stat
.read_ios
as f64, save
);
810 let rrd_key
= format
!("{}/read_bytes", rrd_prefix
);
811 rrd_update_derive(&rrd_key
, (stat
.read_sectors
*512) as f64, save
);
813 let rrd_key
= format
!("{}/write_ios", rrd_prefix
);
814 rrd_update_derive(&rrd_key
, stat
.write_ios
as f64, save
);
815 let rrd_key
= format
!("{}/write_bytes", rrd_prefix
);
816 rrd_update_derive(&rrd_key
, (stat
.write_sectors
*512) as f64, save
);
818 let rrd_key
= format
!("{}/io_ticks", rrd_prefix
);
819 rrd_update_derive(&rrd_key
, (stat
.io_ticks
as f64)/1000.0, save
);
823 eprintln
!("find_mounted_device failed - {}", err
);