mirror of
https://github.com/uroni/urbackup_backend.git
synced 2025-10-26 11:36:50 +00:00
(cherry picked from commit 6bc27673b43cd81e7f6a4b1affc053ec01fd277c)
(cherry picked from commit 5fb8d84b28)
4460 lines
115 KiB
C++
4460 lines
115 KiB
C++
/*************************************************************************
|
|
* UrBackup - Client/Server backup system
|
|
* Copyright (C) 2011-2021 Martin Raiber
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU Affero General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
**************************************************************************/
|
|
#include "ClientMain.h"
|
|
#include "server_ping.h"
|
|
#include "database.h"
|
|
#include "../stringtools.h"
|
|
#include "../urbackupcommon/fileclient/FileClient.h"
|
|
#include "../Interface/Server.h"
|
|
#include "../Interface/ThreadPool.h"
|
|
#include "../urbackupcommon/fileclient/tcpstack.h"
|
|
#include "../common/data.h"
|
|
#include "../urbackupcommon/settingslist.h"
|
|
#include "server_channel.h"
|
|
#include "server_log.h"
|
|
#include "ServerDownloadThread.h"
|
|
#include "InternetServiceConnector.h"
|
|
#include "server_update_stats.h"
|
|
#include "../urbackupcommon/escape.h"
|
|
#include "../common/adler32.h"
|
|
#include "server_running.h"
|
|
#include "server_cleanup.h"
|
|
#include "treediff/TreeDiff.h"
|
|
#include "../urlplugin/IUrlFactory.h"
|
|
#include "../urbackupcommon/mbrdata.h"
|
|
#include "../Interface/PipeThrottler.h"
|
|
#include "snapshot_helper.h"
|
|
#include "../cryptoplugin/ICryptoFactory.h"
|
|
#include "server_dir_links.h"
|
|
#include "server.h"
|
|
#include "../urbackupcommon/filelist_utils.h"
|
|
#include "server_continuous.h"
|
|
#include <algorithm>
|
|
#include <memory.h>
|
|
#include <time.h>
|
|
#include <stdio.h>
|
|
#include <limits.h>
|
|
#include <memory>
|
|
#include <assert.h>
|
|
#include <math.h>
|
|
#include <errno.h>
|
|
#include <string.h>
|
|
#include "create_files_index.h"
|
|
#include <stack>
|
|
#include "FullFileBackup.h"
|
|
#include "IncrFileBackup.h"
|
|
#include "LocalBackup.h"
|
|
#include "ImageBackup.h"
|
|
#include "ContinuousBackup.h"
|
|
#include "ThrottleUpdater.h"
|
|
#include "../fileservplugin/IFileServ.h"
|
|
#include "DataplanDb.h"
|
|
#include "ImageMount.h"
|
|
#include "../urbackupcommon/InternetServicePipe2.h"
|
|
#include "../urbackupcommon/CompressedPipe2.h"
|
|
#include "../urbackupcommon/CompressedPipeZstd.h"
|
|
|
|
extern IUrlFactory *url_fak;
|
|
extern ICryptoFactory *crypto_fak;
|
|
extern std::string server_identity;
|
|
extern std::string server_token;
|
|
extern IFileServ* fileserv;
|
|
|
|
const unsigned short serviceport=35623;
|
|
const unsigned int check_time_intervall=5*60*1000;
|
|
const unsigned int status_update_intervall=1000;
|
|
const unsigned int eta_update_intervall=60000;
|
|
const size_t minfreespace_min=50*1024*1024;
|
|
const unsigned int ident_err_retry_time=1*60*1000;
|
|
const unsigned int ident_err_retry_time_retok=10*60*1000;
|
|
const unsigned int c_filesrv_connect_timeout=10000;
|
|
const unsigned int c_internet_fileclient_timeout=30*60*1000;
|
|
const unsigned int c_sleeptime_failed_imagebackup=20*60;
|
|
const unsigned int c_sleeptime_failed_filebackup=20*60;
|
|
const unsigned int c_exponential_backoff_div=2;
|
|
|
|
const int64 max_ecdh_key_age = 6 * 60 * 60 * 1000; //6h
|
|
|
|
|
|
int ClientMain::running_backups=0;
|
|
int ClientMain::running_file_backups=0;
|
|
IMutex *ClientMain::running_backup_mutex=NULL;
|
|
IMutex *ClientMain::tmpfile_mutex=NULL;
|
|
size_t ClientMain::tmpfile_num=0;
|
|
IMutex* ClientMain::cleanup_mutex = NULL;
|
|
std::map<int, std::vector<SShareCleanup> > ClientMain::cleanup_shares;
|
|
int ClientMain::restore_client_id = -1;
|
|
bool ClientMain::running_backups_allowed=true;
|
|
IMutex* ClientMain::ecdh_key_exchange_mutex = NULL;
|
|
std::vector<std::pair<IECDHKeyExchange*, int64> > ClientMain::ecdh_key_exchange_buffer;
|
|
IMutex* ClientMain::client_uid_reset_mutex = NULL;
|
|
ICondition* ClientMain::client_uid_reset_cond = NULL;
|
|
|
|
ClientMain::ClientMain(IPipe *pPipe, FileClient::SAddrHint pAddr, const std::string &pName,
|
|
const std::string& pSubName, const std::string& pMainName, int filebackup_group_offset, bool internet_connection,
|
|
bool use_file_snapshots, bool use_image_snapshots, bool use_reflink)
|
|
: internet_connection(internet_connection), client_throttler(NULL),
|
|
use_file_snapshots(use_file_snapshots), use_image_snapshots(use_image_snapshots), use_reflink(use_reflink),
|
|
backup_dao(NULL), client_updated_time(0), continuous_backup(NULL),
|
|
clientsubname(pSubName), filebackup_group_offset(filebackup_group_offset), needs_authentification(false),
|
|
restore_mutex(Server->createMutex())
|
|
{
|
|
q_update_lastseen=NULL;
|
|
pipe=pPipe;
|
|
clientaddr=pAddr;
|
|
clientaddr_mutex=Server->createMutex();
|
|
clientname=pName;
|
|
clientmainname=pMainName;
|
|
clientid=0;
|
|
|
|
do_full_backup_now=false;
|
|
do_incr_backup_now=false;
|
|
do_update_settings=false;
|
|
do_full_image_now=false;
|
|
do_incr_image_now=false;
|
|
do_update_access_key = false;
|
|
cdp_needs_sync=true;
|
|
|
|
can_backup_images=true;
|
|
|
|
protocol_versions.file_protocol_version=1;
|
|
update_version=0;
|
|
|
|
tcpstack.setAddChecksum(internet_connection);
|
|
tcpstack_checksum.setAddChecksum(true);
|
|
|
|
last_backup_try = 0;
|
|
|
|
last_image_backup_try=0;
|
|
count_image_backup_try=0;
|
|
|
|
last_file_backup_try=0;
|
|
count_file_backup_try=0;
|
|
|
|
last_cdp_backup_try=0;
|
|
count_cdp_backup_try=0;
|
|
|
|
continuous_mutex=Server->createMutex();
|
|
throttle_mutex=Server->createMutex();
|
|
|
|
curr_image_version=1;
|
|
last_incr_freq=-1;
|
|
last_startup_backup_delay = -1;
|
|
|
|
curr_server_token = server_token;
|
|
if(filebackup_group_offset>0)
|
|
{
|
|
curr_server_token+=convert(filebackup_group_offset/c_group_size);
|
|
}
|
|
|
|
session_identity_refreshtime = 0;
|
|
connection_metered = false;
|
|
do_reauthenticate = false;
|
|
windows_locked = false;
|
|
update_capa = false;
|
|
}
|
|
|
|
ClientMain::~ClientMain(void)
|
|
{
|
|
if(q_update_lastseen!=NULL)
|
|
unloadSQL();
|
|
|
|
Server->destroy(clientaddr_mutex);
|
|
|
|
if(client_throttler!=NULL)
|
|
{
|
|
Server->destroy(client_throttler);
|
|
}
|
|
|
|
Server->destroy(continuous_mutex);
|
|
Server->destroy(throttle_mutex);
|
|
}
|
|
|
|
void ClientMain::init_mutex(void)
|
|
{
|
|
running_backup_mutex=Server->createMutex();
|
|
tmpfile_mutex=Server->createMutex();
|
|
cleanup_mutex=Server->createMutex();
|
|
ecdh_key_exchange_mutex = Server->createMutex();
|
|
client_uid_reset_cond = Server->createCondition();
|
|
client_uid_reset_mutex = Server->createMutex();
|
|
}
|
|
|
|
void ClientMain::destroy_mutex(void)
|
|
{
|
|
Server->destroy(running_backup_mutex);
|
|
Server->destroy(tmpfile_mutex);
|
|
Server->destroy(cleanup_mutex);
|
|
Server->destroy(ecdh_key_exchange_mutex);
|
|
Server->destroy(client_uid_reset_cond);
|
|
Server->destroy(client_uid_reset_mutex);
|
|
}
|
|
|
|
void ClientMain::wakeupClientUidReset()
|
|
{
|
|
IScopedLock lock(client_uid_reset_mutex);
|
|
client_uid_reset_cond->notify_all();
|
|
}
|
|
|
|
void ClientMain::unloadSQL(void)
|
|
{
|
|
db->destroyQuery(q_update_lastseen);
|
|
db->destroyQuery(q_update_setting);
|
|
db->destroyQuery(q_insert_setting);
|
|
db->destroyQuery(q_get_setting);
|
|
db->destroyQuery(q_get_unsent_logdata);
|
|
db->destroyQuery(q_set_logdata_sent);
|
|
}
|
|
|
|
void ClientMain::operator ()(void)
|
|
{
|
|
db = Server->getDatabase(Server->getThreadID(), URBACKUPDB_SERVER);
|
|
DBScopedFreeMemory free_db_memory(db);
|
|
|
|
server_settings.reset(new ServerSettings(db));
|
|
|
|
if (tooManyClients(db, clientname, server_settings.get()))
|
|
{
|
|
ServerStatus::setStatusError(clientname, se_too_many_clients);
|
|
Server->Log("client_main Thread for client " + clientname + " finished, because there were too many clients (1)", LL_INFO);
|
|
|
|
Server->wait(10 * 60 * 1000); //10min
|
|
|
|
BackupServer::forceOfflineClient(clientname);
|
|
pipe->Write("ok");
|
|
ServerLogger::reset(logid);
|
|
delete this;
|
|
return;
|
|
}
|
|
|
|
if(!sendServerIdentity(true))
|
|
{
|
|
pipe->Write("ok");
|
|
Server->Log("client_main Thread for client \"" + clientname + "\" finished and the identity was not recognized", LL_INFO);
|
|
delete this;
|
|
return;
|
|
}
|
|
|
|
if( clientname.find("##restore##")==0 )
|
|
{
|
|
{
|
|
IScopedLock lock(cleanup_mutex);
|
|
clientid = restore_client_id--;
|
|
}
|
|
|
|
ServerChannelThread channel_thread(this, clientname, clientid, internet_connection,
|
|
true, curr_server_token, clientsubname, NULL);
|
|
THREADPOOL_TICKET channel_thread_id=Server->getThreadPool()->execute(&channel_thread, "client channel");
|
|
|
|
while(true)
|
|
{
|
|
std::string msg;
|
|
pipe->Read(&msg);
|
|
if(msg=="exit" || msg=="exitnow" )
|
|
break;
|
|
}
|
|
|
|
channel_thread.doExit();
|
|
Server->getThreadPool()->waitFor(channel_thread_id);
|
|
|
|
cleanupShares();
|
|
|
|
pipe->Write("ok");
|
|
Server->Log("client_main Thread for client "+clientname+" finished, restore thread");
|
|
delete this;
|
|
return;
|
|
}
|
|
else
|
|
{
|
|
if(!authenticateIfNeeded(true, false))
|
|
{
|
|
pipe->Write("ok");
|
|
delete this;
|
|
return;
|
|
}
|
|
}
|
|
|
|
ServerStatus::setStatusError(clientname, se_startup);
|
|
|
|
std::string identity = getIdentity();
|
|
|
|
std::unique_ptr<ServerBackupDao> local_server_backup_dao(new ServerBackupDao(db));
|
|
backup_dao = local_server_backup_dao.get();
|
|
|
|
clientid=getClientID(db, clientname, server_settings.get(), NULL, nullptr, nullptr, &perm_uid);
|
|
|
|
if(clientid==-1)
|
|
{
|
|
ServerStatus::setStatusError(clientname, se_too_many_clients);
|
|
Server->Log("client_main Thread for client "+clientname+" finished, because there were too many clients (2)", LL_INFO);
|
|
|
|
Server->wait(10*60*1000); //10min
|
|
|
|
BackupServer::forceOfflineClient(clientname);
|
|
pipe->Write("ok");
|
|
ServerLogger::reset(logid);
|
|
delete this;
|
|
return;
|
|
}
|
|
else if(!clientsubname.empty())
|
|
{
|
|
backup_dao->setVirtualMainClient(clientmainname, clientid);
|
|
}
|
|
|
|
logid = ServerLogger::getLogId(clientid);
|
|
|
|
server_settings.reset(new ServerSettings(db, clientid));
|
|
|
|
if(!createDirectoryForClient())
|
|
{
|
|
int64 starttime = Server->getTimeMS();
|
|
int64 ctime;
|
|
bool has_dir=false;
|
|
while( (ctime=Server->getTimeMS())>=starttime && ctime-starttime<10*60*1000)
|
|
{
|
|
Server->wait(1000);
|
|
|
|
if(createDirectoryForClient())
|
|
{
|
|
has_dir=true;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if(!has_dir)
|
|
{
|
|
BackupServer::forceOfflineClient(clientname);
|
|
pipe->Write("ok");
|
|
delete this;
|
|
return;
|
|
}
|
|
}
|
|
|
|
if(server_settings->getSettings()->computername.empty() || !clientsubname.empty())
|
|
{
|
|
server_settings->getSettings()->computername=clientname;
|
|
}
|
|
|
|
if(server_settings->getImageFileFormat()==image_file_format_cowraw)
|
|
{
|
|
curr_image_version = 0;
|
|
}
|
|
else if (server_settings->getImageFileFormat() == image_file_format_vhdx
|
|
|| server_settings->getImageFileFormat() == image_file_format_vhdxz)
|
|
{
|
|
curr_image_version = 2;
|
|
}
|
|
else
|
|
{
|
|
curr_image_version = 1;
|
|
}
|
|
|
|
prepareSQL();
|
|
|
|
int64 lastseen = Server->getTimeSeconds();
|
|
updateLastseen(lastseen);
|
|
|
|
bool needs_restart = false;
|
|
if(!updateCapabilities(&needs_restart) || needs_restart)
|
|
{
|
|
if (!needs_restart)
|
|
{
|
|
Server->Log("Could not get client capabilities", LL_ERROR);
|
|
|
|
IScopedLock lock(client_uid_reset_mutex);
|
|
client_uid_reset_cond->wait(&lock, 5 * 60 * 1000); //5min
|
|
}
|
|
|
|
pipe->Write("ok");
|
|
BackupServer::forceOfflineClient(clientname);
|
|
delete this;
|
|
return;
|
|
}
|
|
|
|
int64 next_capa_update = 0;
|
|
if (protocol_versions.update_capa_interval > 0)
|
|
{
|
|
next_capa_update = Server->getTimeMS() + protocol_versions.update_capa_interval;
|
|
}
|
|
|
|
ServerStatus::setClientId(clientname, clientid);
|
|
|
|
bool use_reflink=false;
|
|
#ifndef _WIN32
|
|
if( use_file_snapshots )
|
|
use_reflink=true;
|
|
#endif
|
|
if (BackupServer::canReflink())
|
|
use_reflink = true;
|
|
|
|
use_tmpfiles=server_settings->getSettings()->use_tmpfiles;
|
|
use_tmpfiles_images=server_settings->getSettings()->use_tmpfiles_images;
|
|
if(!use_tmpfiles)
|
|
{
|
|
tmpfile_path=server_settings->getSettings()->backupfolder+os_file_sep()+"urbackup_tmp_files";
|
|
os_create_dir(tmpfile_path);
|
|
if(!os_directory_exists(tmpfile_path))
|
|
{
|
|
Server->Log("Could not create or see temporary folder in backuppath", LL_ERROR);
|
|
use_tmpfiles=true;
|
|
}
|
|
}
|
|
|
|
ServerChannelThread channel_thread(this, clientname, clientid, internet_connection,
|
|
true, curr_server_token, clientsubname, NULL);
|
|
THREADPOOL_TICKET channel_thread_id=Server->getThreadPool()->execute(&channel_thread, "client channel");
|
|
|
|
bool received_client_settings=true;
|
|
ServerLogger::Log(logid, "Getting client settings...", LL_DEBUG);
|
|
ServerStatus::setStatusError(clientname, se_settings);
|
|
bool settings_doesnt_exist=false;
|
|
if(server_settings->getSettings()->allow_overwrite && !getClientSettings(settings_doesnt_exist))
|
|
{
|
|
if(!settings_doesnt_exist)
|
|
{
|
|
ServerLogger::Log(logid, "Getting client settings failed. Retrying...", LL_INFO);
|
|
Server->wait(200000);
|
|
if(!getClientSettings(settings_doesnt_exist))
|
|
{
|
|
ServerLogger::Log(logid, "Getting client settings failed -1", LL_ERROR);
|
|
received_client_settings=false;
|
|
}
|
|
else
|
|
{
|
|
do_update_settings = true;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
ServerLogger::Log(logid, "Getting client settings failed. Not retrying because settings do not exist.", LL_INFO);
|
|
}
|
|
}
|
|
|
|
if(received_client_settings || settings_doesnt_exist)
|
|
{
|
|
sendSettings();
|
|
}
|
|
|
|
ServerStatus::setStatusError(clientname, se_startup);
|
|
|
|
updateVirtualClients();
|
|
|
|
ServerLogger::Log(logid, "Sending backup incr interval...", LL_DEBUG);
|
|
sendClientBackupIncrIntervall();
|
|
|
|
int64 last_client_update_check = 0;
|
|
if(server_settings->getSettings()->autoupdate_clients)
|
|
{
|
|
checkClientVersion();
|
|
last_client_update_check = Server->getTimeMS();
|
|
}
|
|
|
|
sendClientLogdata();
|
|
|
|
curr_image_format = server_settings->getImageFileFormat();
|
|
|
|
ServerStatus::setCommPipe(clientname, pipe);
|
|
|
|
bool skip_checking=false;
|
|
|
|
ServerStatus::setStatusError(clientname, se_none);
|
|
|
|
if( server_settings->getSettings()->startup_backup_delay>0
|
|
&& (!do_full_backup_now && !do_incr_backup_now
|
|
&& !do_incr_backup_now && !do_full_image_now) )
|
|
{
|
|
pipe->isReadable(server_settings->getSettings()->startup_backup_delay*1000);
|
|
skip_checking=true;
|
|
}
|
|
|
|
ServerSettings server_settings_updated(db, clientid);
|
|
|
|
bool do_exit_now=false;
|
|
|
|
do_full_backup_now = true;
|
|
|
|
while(true)
|
|
{
|
|
if(!skip_checking)
|
|
{
|
|
timeoutRestores();
|
|
|
|
bool send_logdata = false;
|
|
|
|
for (size_t i = 0; i<backup_queue.size();)
|
|
{
|
|
if (backup_queue[i].ticket != ILLEGAL_THREADPOOL_TICKET
|
|
|| backup_queue[i].running)
|
|
{
|
|
if (!backup_queue[i].backup->isFileBackup())
|
|
{
|
|
ImageBackup* ibackup = dynamic_cast<ImageBackup*>(backup_queue[i].backup);
|
|
if (ibackup != NULL)
|
|
{
|
|
bool in_image_group = false;
|
|
for (size_t j = 0; j < running_image_groups.size();++j)
|
|
{
|
|
if (running_image_groups[j].find(ibackup) != running_image_groups[j].end())
|
|
{
|
|
in_image_group = true;
|
|
break;
|
|
}
|
|
}
|
|
|
|
std::vector<ImageBackup::SImageDependency> dependencies = ibackup->getDependencies(true);
|
|
if (!in_image_group
|
|
&& !dependencies.empty())
|
|
{
|
|
std::map<ImageBackup*, bool> new_running_image_group;
|
|
new_running_image_group[ibackup] = false;
|
|
|
|
for (size_t j = 0; j < dependencies.size(); ++j)
|
|
{
|
|
std::string vols_list = ibackup->getLetter();
|
|
for (size_t k = 0; k < dependencies.size(); ++k)
|
|
{
|
|
if (j == k) continue;
|
|
vols_list += ", " + dependencies[k].volume;
|
|
}
|
|
|
|
std::string letter = normalizeVolumeUpper(dependencies[j].volume);
|
|
|
|
SRunningBackup backup;
|
|
ImageBackup* idep = new ImageBackup(this, clientid, clientname, clientsubname, LogAction_LogIfNotDisabled,
|
|
ibackup->isIncrementalBackup(), letter, curr_server_token, letter, false,
|
|
dependencies[j].snapshot_id, vols_list, ibackup->getBackupStarttime(), ibackup->isScheduled());
|
|
backup.backup = idep;
|
|
backup.letter = letter;
|
|
|
|
new_running_image_group[idep] = false;
|
|
|
|
backup_queue.push_back(backup);
|
|
}
|
|
|
|
running_image_groups.push_back(new_running_image_group);
|
|
}
|
|
}
|
|
}
|
|
|
|
if ( isBackupFinished(backup_queue[i])
|
|
&& (dynamic_cast<ImageBackup*>(backup_queue[i].backup)==NULL
|
|
|| dynamic_cast<ImageBackup*>(backup_queue[i].backup)->getDependencies(false).empty()) )
|
|
{
|
|
if (backup_queue[i].running)
|
|
stopBackupRunning(backup_queue[i].backup->isFileBackup());
|
|
|
|
ServerStatus::subRunningJob(clientmainname);
|
|
|
|
if (!backup_queue[i].backup->getResult() &&
|
|
backup_queue[i].backup->shouldBackoff())
|
|
{
|
|
if (backup_queue[i].backup->isFileBackup())
|
|
{
|
|
last_file_backup_try = Server->getTimeMS();
|
|
++count_file_backup_try;
|
|
ServerLogger::Log(logid, "Exponential backoff: Waiting at least " + PrettyPrintTime(exponentialBackoffTimeFile() * 1000) + " before next file backup", LL_WARNING);
|
|
}
|
|
else
|
|
{
|
|
last_image_backup_try = Server->getTimeMS();
|
|
++count_image_backup_try;
|
|
ServerLogger::Log(logid, "Exponential backoff: Waiting at least " + PrettyPrintTime(exponentialBackoffTimeImage() * 1000) + " before next image backup", LL_WARNING);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (backup_queue[i].backup->isFileBackup())
|
|
{
|
|
count_file_backup_try = 0;
|
|
}
|
|
else
|
|
{
|
|
count_image_backup_try = 0;
|
|
}
|
|
|
|
last_backup_try = Server->getTimeMS();
|
|
}
|
|
|
|
bool del_backup = true;
|
|
|
|
if (!backup_queue[i].backup->isFileBackup())
|
|
{
|
|
ImageBackup* ibackup = dynamic_cast<ImageBackup*>(backup_queue[i].backup);
|
|
if (ibackup != NULL)
|
|
{
|
|
for (size_t j = 0; j < running_image_groups.size();)
|
|
{
|
|
{
|
|
std::map<ImageBackup*, bool>::iterator it = running_image_groups[j].find(ibackup);
|
|
if (it != running_image_groups[j].end())
|
|
{
|
|
ServerCleanupThread::lockImageFromCleanup(ibackup->getBackupId());
|
|
del_backup = false;
|
|
it->second = true;
|
|
}
|
|
}
|
|
|
|
bool finished = true;
|
|
bool success = true;
|
|
for (std::map<ImageBackup*, bool>::iterator it = running_image_groups[j].begin();
|
|
it != running_image_groups[j].end(); ++it)
|
|
{
|
|
if (!it->second)
|
|
{
|
|
finished = false;
|
|
}
|
|
else if (!it->first->getResult())
|
|
{
|
|
success = false;
|
|
}
|
|
}
|
|
|
|
if (finished)
|
|
{
|
|
for (std::map<ImageBackup*, bool>::iterator it = running_image_groups[j].begin();
|
|
it != running_image_groups[j].end(); ++it)
|
|
{
|
|
if (success)
|
|
{
|
|
backup_dao->setImageBackupComplete(it->first->getBackupId());
|
|
backup_dao->updateClientLastImageBackup(it->first->getBackupId(), clientid);
|
|
}
|
|
ServerCleanupThread::unlockImageFromCleanup(it->first->getBackupId());
|
|
delete it->first;
|
|
}
|
|
|
|
running_image_groups.erase(running_image_groups.begin() + j);
|
|
continue;
|
|
}
|
|
|
|
++j;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (del_backup)
|
|
{
|
|
delete backup_queue[i].backup;
|
|
}
|
|
|
|
send_logdata = true;
|
|
backup_queue.erase(backup_queue.begin() + i);
|
|
continue;
|
|
}
|
|
}
|
|
|
|
++i;
|
|
}
|
|
|
|
if (send_logdata)
|
|
{
|
|
sendClientLogdata();
|
|
}
|
|
|
|
{
|
|
bool received_client_settings=true;
|
|
bool settings_updated=false;
|
|
server_settings_updated.getSettings(&settings_updated);
|
|
bool settings_dont_exist=false;
|
|
SStatusError curr_status = se_unknown;
|
|
if(do_update_settings || settings_updated)
|
|
{
|
|
curr_status = ServerStatus::getStatus(clientname).status_error;
|
|
ServerStatus::setStatusError(clientname, se_settings);
|
|
ServerLogger::Log(logid, "Getting client settings...", LL_DEBUG);
|
|
do_update_settings=false;
|
|
if(server_settings->getSettings()->allow_overwrite && !getClientSettings(settings_dont_exist))
|
|
{
|
|
ServerLogger::Log(logid, "Getting client settings failed -2", LL_ERROR);
|
|
received_client_settings=false;
|
|
do_update_settings = true;
|
|
}
|
|
|
|
if(server_settings->getImageFileFormat()==image_file_format_cowraw)
|
|
{
|
|
curr_image_version = 0;
|
|
}
|
|
else if (server_settings->getImageFileFormat() == image_file_format_vhdxz
|
|
|| server_settings->getImageFileFormat() == image_file_format_vhdx)
|
|
{
|
|
curr_image_version = 2;
|
|
}
|
|
else
|
|
{
|
|
curr_image_version = 1;
|
|
}
|
|
|
|
updateVirtualClients();
|
|
|
|
ServerStatus::setStatusError(clientname, curr_status);
|
|
}
|
|
|
|
if(settings_updated && (received_client_settings || settings_dont_exist) )
|
|
{
|
|
sendSettings();
|
|
}
|
|
|
|
if(settings_updated)
|
|
{
|
|
sendClientBackupIncrIntervall();
|
|
}
|
|
|
|
if (curr_status != se_unknown)
|
|
{
|
|
ServerStatus::setStatusError(clientname, curr_status);
|
|
}
|
|
}
|
|
|
|
if( (client_updated_time!=0 && Server->getTimeMS()-client_updated_time>6*60*1000)
|
|
|| update_capa )
|
|
{
|
|
update_capa = false;
|
|
|
|
updateCapabilities(NULL);
|
|
client_updated_time=0;
|
|
session_identity_refreshtime = 0;
|
|
if (!authenticateIfNeeded(true, false))
|
|
{
|
|
skip_checking = true;
|
|
}
|
|
}
|
|
|
|
if ((next_capa_update != 0 && next_capa_update <= Server->getTimeMS()))
|
|
{
|
|
updateCapabilities(NULL);
|
|
|
|
if (protocol_versions.update_capa_interval > 0)
|
|
{
|
|
next_capa_update = Server->getTimeMS() + protocol_versions.update_capa_interval;
|
|
}
|
|
}
|
|
|
|
if (do_reauthenticate)
|
|
{
|
|
do_reauthenticate = false;
|
|
if ( !sendServerIdentity(true)
|
|
|| !authenticateIfNeeded(true, true) )
|
|
{
|
|
skip_checking = true;
|
|
}
|
|
}
|
|
|
|
if (Server->getTimeMS()- last_client_update_check > 24*60*60*1000
|
|
&& server_settings->getSettings()->autoupdate_clients)
|
|
{
|
|
checkClientVersion();
|
|
last_client_update_check = Server->getTimeMS();
|
|
}
|
|
|
|
if (Server->getTimeSeconds() - lastseen > 5 * 60
|
|
&& ServerStatus::getLastseen(clientname)>lastseen)
|
|
{
|
|
lastseen = ServerStatus::getLastseen(clientname);
|
|
updateLastseen(lastseen);
|
|
}
|
|
|
|
curr_image_format = server_settings->getImageFileFormat();
|
|
|
|
|
|
bool internet_no_full_file=(internet_connection && !server_settings->getSettings()->internet_full_file_backups );
|
|
bool internet_no_images=(internet_connection && !server_settings->getSettings()->internet_image_backups );
|
|
|
|
if(do_incr_image_now)
|
|
{
|
|
if(!can_backup_images)
|
|
ServerLogger::Log(logid, "Cannot do image backup because can_backup_images=false", LL_DEBUG);
|
|
if(server_settings->getSettings()->no_images)
|
|
ServerLogger::Log(logid, "Cannot do image backup because no_images=true", LL_DEBUG);
|
|
if(!isBackupsRunningOkay(false))
|
|
ServerLogger::Log(logid, "Cannot do image backup because isBackupsRunningOkay()=false", LL_DEBUG);
|
|
if(!internet_no_images )
|
|
ServerLogger::Log(logid, "Cannot do image backup because internet_no_images=true", LL_DEBUG);
|
|
}
|
|
|
|
if(do_incr_backup_now)
|
|
{
|
|
if(server_settings->getSettings()->no_file_backups)
|
|
ServerLogger::Log(logid, "Cannot do incremental file backup because no_file_backups=true", LL_DEBUG);
|
|
if(!isBackupsRunningOkay(true))
|
|
ServerLogger::Log(logid, "Cannot do incremental file backup because isBackupsRunningOkay()=false", LL_DEBUG);
|
|
}
|
|
|
|
if (do_update_access_key)
|
|
{
|
|
do_update_access_key = false;
|
|
updateClientAccessKey();
|
|
}
|
|
|
|
if( !server_settings->getSettings()->no_file_backups && (!internet_no_full_file || do_full_backup_now) &&
|
|
( (isUpdateFull(filebackup_group_offset + c_group_default) && ServerSettings::isInTimeSpan(server_settings->getBackupWindowFullFile())
|
|
&& exponentialBackoffFile() && pauseRetryBackup() && isDataplanOkay(true) && isOnline(channel_thread) ) || do_full_backup_now )
|
|
&& isBackupsRunningOkay(true) && !do_full_image_now && !do_incr_image_now && !do_incr_backup_now
|
|
&& (!isRunningFileBackup(filebackup_group_offset + c_group_default) || do_full_backup_now) )
|
|
{
|
|
SRunningBackup backup;
|
|
if (server_settings->getSettings()->backup_dest_url.empty())
|
|
{
|
|
backup.backup = new FullFileBackup(this, clientid, clientname, clientsubname,
|
|
do_full_backup_now ? LogAction_AlwaysLog : LogAction_LogIfNotDisabled, filebackup_group_offset + c_group_default, use_tmpfiles,
|
|
tmpfile_path, use_reflink, use_file_snapshots, curr_server_token, convert(c_group_default), !do_full_backup_now);
|
|
}
|
|
else
|
|
{
|
|
backup.backup = new LocalBackup(this, clientid, clientname, clientsubname,
|
|
do_full_backup_now ? LogAction_AlwaysLog : LogAction_LogIfNotDisabled, true, false,
|
|
server_token, convert(c_group_default), !do_full_backup_now, filebackup_group_offset + c_group_default);
|
|
}
|
|
|
|
backup.group = filebackup_group_offset + c_group_default;
|
|
|
|
backup_queue.push_back(backup);
|
|
|
|
do_full_backup_now=false;
|
|
}
|
|
else if( !server_settings->getSettings()->no_file_backups
|
|
&& ( (isUpdateIncr(filebackup_group_offset + c_group_default) && ServerSettings::isInTimeSpan(server_settings->getBackupWindowIncrFile())
|
|
&& exponentialBackoffFile() && pauseRetryBackup() && isDataplanOkay(true) && isOnline(channel_thread) ) || do_incr_backup_now )
|
|
&& isBackupsRunningOkay(true) && !do_full_image_now && !do_incr_image_now
|
|
&& (!isRunningFileBackup(filebackup_group_offset + c_group_default) || do_incr_backup_now) )
|
|
{
|
|
SRunningBackup backup;
|
|
if (server_settings->getSettings()->backup_dest_url.empty())
|
|
{
|
|
backup.backup = new IncrFileBackup(this, clientid, clientname, clientsubname,
|
|
do_incr_backup_now ? LogAction_AlwaysLog : LogAction_LogIfNotDisabled, filebackup_group_offset + c_group_default, use_tmpfiles,
|
|
tmpfile_path, use_reflink, use_file_snapshots, curr_server_token, convert(c_group_default), !do_incr_backup_now);
|
|
}
|
|
else
|
|
{
|
|
backup.backup = new LocalBackup(this, clientid, clientname, clientsubname,
|
|
do_incr_backup_now ? LogAction_AlwaysLog : LogAction_LogIfNotDisabled, true, true,
|
|
server_token, convert(c_group_default), !do_incr_backup_now, filebackup_group_offset + c_group_default);
|
|
}
|
|
backup.group=filebackup_group_offset + c_group_default;
|
|
|
|
backup_queue.push_back(backup);
|
|
|
|
do_incr_backup_now=false;
|
|
}
|
|
else if(can_backup_images && !server_settings->getSettings()->no_images && (!internet_no_images || do_full_image_now)
|
|
&& ( (isUpdateFullImage() && ServerSettings::isInTimeSpan(server_settings->getBackupWindowFullImage())
|
|
&& exponentialBackoffImage() && pauseRetryBackup() && isDataplanOkay(false) && isOnline(channel_thread) ) || do_full_image_now)
|
|
&& isBackupsRunningOkay(false) && !do_incr_image_now)
|
|
{
|
|
if (protocol_versions.update_vols > 0)
|
|
{
|
|
updateCapabilities(NULL);
|
|
}
|
|
|
|
std::vector<std::string> vols=server_settings->getBackupVolumes(all_volumes, all_nonusb_volumes);
|
|
for(size_t i=0;i<vols.size();++i)
|
|
{
|
|
std::string letter=normalizeVolumeUpper(vols[i]);
|
|
if( ( (isUpdateFullImage(letter) && !isRunningImageBackup(letter) && isBackupsRunningOkay(false)) || do_full_image_now)
|
|
&& !isImageGroupQueued(letter, true) )
|
|
{
|
|
SRunningBackup backup;
|
|
backup.backup = new ImageBackup(this, clientid, clientname, clientsubname,
|
|
do_full_image_now?LogAction_AlwaysLog:LogAction_LogIfNotDisabled,
|
|
false, letter, curr_server_token, letter, true, 0, std::string(), 0, !do_full_image_now);
|
|
backup.letter=letter;
|
|
|
|
backup_queue.push_back(backup);
|
|
}
|
|
}
|
|
|
|
do_full_image_now=false;
|
|
}
|
|
else if(can_backup_images && !server_settings->getSettings()->no_images && (!internet_no_images || do_incr_image_now)
|
|
&& ((isUpdateIncrImage() && ServerSettings::isInTimeSpan(server_settings->getBackupWindowIncrImage())
|
|
&& exponentialBackoffImage() && pauseRetryBackup() && isDataplanOkay(false) && isOnline(channel_thread) ) || do_incr_image_now)
|
|
&& isBackupsRunningOkay(false) )
|
|
{
|
|
if (protocol_versions.update_vols > 0)
|
|
{
|
|
updateCapabilities(NULL);
|
|
}
|
|
|
|
std::vector<std::string> vols=server_settings->getBackupVolumes(all_volumes, all_nonusb_volumes);
|
|
for(size_t i=0;i<vols.size();++i)
|
|
{
|
|
std::string letter= normalizeVolumeUpper(vols[i]);
|
|
if( ((isUpdateIncrImage(letter) && !isRunningImageBackup(letter) && isBackupsRunningOkay(false) ) || do_incr_image_now)
|
|
&& !isImageGroupQueued(letter, false) )
|
|
{
|
|
SRunningBackup backup;
|
|
backup.backup = new ImageBackup(this, clientid, clientname, clientsubname, do_incr_image_now ?LogAction_AlwaysLog:LogAction_LogIfNotDisabled,
|
|
true, letter, curr_server_token, letter, true, 0, std::string(), 0, !do_incr_image_now);
|
|
backup.letter=letter;
|
|
|
|
backup_queue.push_back(backup);
|
|
}
|
|
}
|
|
|
|
do_incr_image_now=false;
|
|
}
|
|
else if(protocol_versions.cdp_version>0 && cdp_needs_sync && !isRunningFileBackup(filebackup_group_offset + c_group_continuous))
|
|
{
|
|
cdp_needs_sync=false;
|
|
|
|
SRunningBackup backup;
|
|
backup.backup = new ContinuousBackup(this, clientid, clientname, clientsubname,
|
|
LogAction_LogIfNotDisabled, filebackup_group_offset + c_group_continuous, use_tmpfiles,
|
|
tmpfile_path, use_reflink, use_file_snapshots, convert(c_group_continuous), false);
|
|
backup.group=filebackup_group_offset + c_group_continuous;
|
|
|
|
backup_queue.push_back(backup);
|
|
}
|
|
|
|
bool can_start=false;
|
|
size_t running_jobs=0;
|
|
for(size_t i=0;i<backup_queue.size();++i)
|
|
{
|
|
if(backup_queue[i].ticket==ILLEGAL_THREADPOOL_TICKET)
|
|
{
|
|
can_start=true;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (can_start)
|
|
{
|
|
if (!checkClientName(can_start))
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
|
|
if(can_start)
|
|
{
|
|
while(ServerStatus::numRunningJobs(clientmainname)<server_settings->getSettings()->max_running_jobs_per_client)
|
|
{
|
|
bool started_job=false;
|
|
for(size_t i=0;i<backup_queue.size();++i)
|
|
{
|
|
bool filebackup = backup_queue[i].backup->isFileBackup();
|
|
|
|
if( backup_queue[i].ticket==ILLEGAL_THREADPOOL_TICKET
|
|
&& !backup_queue[i].running
|
|
&& (!backup_queue[i].backup->isScheduled() || inBackupWindow(backup_queue[i].backup) )
|
|
&& (!filebackup || !isRunningFileBackup(backup_queue[i].group, false) ) )
|
|
{
|
|
ServerStatus::addRunningJob(clientmainname);
|
|
if(ServerStatus::numRunningJobs(clientmainname)<=server_settings->getSettings()->max_running_jobs_per_client
|
|
&& isBackupsRunningOkay(filebackup, true))
|
|
{
|
|
std::string tname = "backup main";
|
|
if (filebackup)
|
|
{
|
|
tname = "fbackup main";
|
|
}
|
|
else if (dynamic_cast<ImageBackup*>(backup_queue[i].backup) != NULL)
|
|
{
|
|
tname = "ibackup main";
|
|
}
|
|
|
|
LocalBackup* localBackup = dynamic_cast<LocalBackup*>(backup_queue[i].backup);
|
|
if (localBackup == nullptr)
|
|
{
|
|
backup_queue[i].ticket = Server->getThreadPool()->execute(backup_queue[i].backup, tname);
|
|
}
|
|
else
|
|
{
|
|
backup_queue[i].running = true;
|
|
|
|
if (!localBackup->doBackup())
|
|
{
|
|
localBackup->setResult(false, false);
|
|
}
|
|
|
|
running_local_backups.push_back(SRunningLocalBackup(localBackup->getLogId(),
|
|
localBackup->getStatusId(), localBackup->getBackupId()));
|
|
}
|
|
started_job=true;
|
|
}
|
|
else
|
|
{
|
|
ServerStatus::subRunningJob(clientmainname);
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
|
|
if(!started_job)
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
std::string msg;
|
|
pipe->Read(&msg, skip_checking?0:check_time_intervall);
|
|
|
|
skip_checking=false;
|
|
if(msg=="exit")
|
|
break;
|
|
else if(msg=="exitnow")
|
|
{
|
|
do_exit_now=true;
|
|
break;
|
|
}
|
|
else if(msg=="START BACKUP INCR") do_incr_backup_now=true;
|
|
else if(msg=="START BACKUP FULL") do_full_backup_now=true;
|
|
else if(msg=="UPDATE SETTINGS") do_update_settings=true;
|
|
else if(msg=="START IMAGE INCR") do_incr_image_now=true;
|
|
else if(msg=="START IMAGE FULL") do_full_image_now=true;
|
|
else if (msg == "UPDATE ACCESS KEY") do_update_access_key = true;
|
|
else if(next(msg, 0, "address"))
|
|
{
|
|
updateClientAddress(msg.substr(7));
|
|
|
|
tcpstack.setAddChecksum(internet_connection);
|
|
|
|
if(!authenticateIfNeeded(true, true))
|
|
{
|
|
skip_checking=true;
|
|
}
|
|
}
|
|
else if(msg=="WAKEUP")
|
|
{
|
|
|
|
}
|
|
else if(next(msg, 0, "RESTORE"))
|
|
{
|
|
std::string data_str = msg.substr(7);
|
|
CRData rdata(&data_str);
|
|
|
|
std::string restore_identity;
|
|
rdata.getStr(&restore_identity);
|
|
int64 restore_id;
|
|
rdata.getInt64(&restore_id);
|
|
int64 status_id;
|
|
rdata.getInt64(&status_id);
|
|
logid_t log_id = logid_t();
|
|
rdata.getInt64(&log_id.first);
|
|
std::string restore_token;
|
|
rdata.getStr(&restore_token);
|
|
char single_file = 0;
|
|
rdata.getChar(&single_file);
|
|
char clean_other = 0;
|
|
rdata.getChar(&clean_other);
|
|
char ignore_other_fs = 0;
|
|
rdata.getChar(&ignore_other_fs);
|
|
int64 restore_flags = 0;
|
|
rdata.getInt64(&restore_flags);
|
|
char restore_client_access_encryption = 1;
|
|
rdata.getChar(&restore_client_access_encryption);
|
|
|
|
std::string restore_path = ServerStatus::getProcess(clientname, status_id).details;
|
|
|
|
bool server_confirms;
|
|
if(ServerStatus::canRestore(clientname, server_confirms))
|
|
{
|
|
if(server_confirms || !restore_token.empty())
|
|
{
|
|
ServerLogger::Log(log_id, "Starting restore of path \""+ restore_path + "\"...", LL_INFO);
|
|
}
|
|
else
|
|
{
|
|
ServerLogger::Log(log_id, "Starting restore of path \"" + restore_path + "\". Waiting for client confirmation...", LL_INFO);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
ServerLogger::Log(log_id, "Starting restore of path \"" + restore_path + "\". But client may be offline...", LL_INFO);
|
|
}
|
|
|
|
std::string client_token_key;
|
|
if (restore_client_access_encryption)
|
|
{
|
|
client_token_key = "client_token";
|
|
if (crypto_fak == NULL)
|
|
{
|
|
ServerLogger::Log(log_id, "Cannot restore without crypto plugin", LL_ERROR);
|
|
restore_identity.clear();
|
|
}
|
|
else
|
|
{
|
|
restore_identity = base64_encode_dash(crypto_fak->encryptAuthenticatedAES(restore_identity, server_settings->getSettings()->client_access_key, 1));
|
|
}
|
|
}
|
|
else
|
|
{
|
|
client_token_key = "client_token_d";
|
|
}
|
|
|
|
std::string ret = sendClientMessageRetry("FILE RESTORE "+ client_token_key +"="+restore_identity+"&server_token="+curr_server_token+
|
|
"&id="+convert(restore_id)+"&status_id="+convert(status_id)+
|
|
"&log_id="+convert(log_id.first)+(restore_token.empty()?"":"&restore_token="+restore_token)+
|
|
"&restore_path="+EscapeParamString(restore_path)+
|
|
"&single_file="+ convert(static_cast<int>(single_file))+
|
|
"&clean_other=" + convert(static_cast<int>(clean_other))+
|
|
"&ignore_other_fs=" + convert(static_cast<int>(ignore_other_fs))+
|
|
"&restore_flags="+convert(restore_flags)+
|
|
"&tgroup=0"
|
|
"&clientsubname="+EscapeParamString(clientsubname),
|
|
"Starting restore failed", 10000, 10, true, LL_ERROR);
|
|
|
|
if (ret != "ok")
|
|
{
|
|
ServerLogger::Log(log_id, "Starting restore failed: "+ret, LL_ERROR);
|
|
|
|
finishFailedRestore(restore_identity, log_id, status_id, restore_id);
|
|
}
|
|
else
|
|
{
|
|
IScopedLock lock(restore_mutex.get());
|
|
|
|
running_restores.push_back(SRunningRestore(restore_identity, log_id, status_id, restore_id));
|
|
}
|
|
}
|
|
else if (next(msg, 0, "BACKUP DONE?"))
|
|
{
|
|
str_map params;
|
|
ParseParamStrHttp(msg.substr(12), ¶ms);
|
|
|
|
int64 status_id = watoi64(params["status_id"]);
|
|
updateLocalBackup(status_id, watoi(params["success"]) > 0,
|
|
watoi(params["should_backoff"]) > 0);
|
|
removeLocalBackup(status_id);
|
|
}
|
|
|
|
if(!msg.empty())
|
|
{
|
|
Server->Log("msg="+msg, LL_DEBUG);
|
|
}
|
|
}
|
|
|
|
Server->Log("Waiting for backup threads to finish (clientid="+convert(clientid)+", nthreads=" + convert(backup_queue.size())+")...", LL_DEBUG);
|
|
for(size_t i=0;i<backup_queue.size();++i)
|
|
{
|
|
if(backup_queue[i].ticket!=ILLEGAL_THREADPOOL_TICKET)
|
|
{
|
|
Server->getThreadPool()->waitFor(backup_queue[i].ticket);
|
|
ServerStatus::subRunningJob(clientmainname);
|
|
}
|
|
|
|
delete backup_queue[i].backup;
|
|
}
|
|
|
|
Server->wait(2000);
|
|
|
|
|
|
ServerStatus::setCommPipe(clientname, NULL);
|
|
|
|
//destroy channel
|
|
{
|
|
Server->Log("Stopping channel...", LL_DEBUG);
|
|
channel_thread.doExit();
|
|
Server->getThreadPool()->waitFor(channel_thread_id);
|
|
}
|
|
|
|
cleanupShares();
|
|
|
|
ServerLogger::reset(clientid);
|
|
|
|
|
|
server_settings.reset();
|
|
pipe->Write("ok");
|
|
Server->Log("client_main Thread for client "+clientname+" finished");
|
|
|
|
delete this;
|
|
}
|
|
|
|
void ClientMain::prepareSQL(void)
|
|
{
|
|
q_update_lastseen=db->Prepare("UPDATE clients SET lastseen=datetime(?, 'unixepoch') WHERE id=?", false);
|
|
q_update_setting=db->Prepare("UPDATE settings_db.settings SET value_client=?, use=?, use_last_modified=? WHERE key=? AND clientid=?", false);
|
|
q_insert_setting=db->Prepare("INSERT INTO settings_db.settings (key, value_client, clientid, use, use_last_modified) VALUES (?,?,?,?,?)", false);
|
|
q_get_setting = db->Prepare("SELECT value_client, use, use_last_modified FROM settings_db.settings WHERE clientid=? AND key=?", false);
|
|
q_get_unsent_logdata=db->Prepare("SELECT l.id AS id, strftime('%s', l.created) AS created, log_data.data AS logdata FROM (logs l INNER JOIN log_data ON l.id=log_data.logid) WHERE sent=0 AND clientid=?", false);
|
|
q_set_logdata_sent=db->Prepare("UPDATE logs SET sent=1 WHERE id=?", false);
|
|
}
|
|
|
|
int ClientMain::getClientID(IDatabase *db, const std::string &clientname, ServerSettings *server_settings,
|
|
bool *new_client, std::string* authkey, int* client_group, std::string* perm_uid)
|
|
{
|
|
if(new_client!=NULL)
|
|
*new_client=false;
|
|
|
|
IQuery *q=db->Prepare("SELECT id, perm_uid FROM clients WHERE name=?",false);
|
|
if(q==NULL) return -1;
|
|
|
|
q->Bind(clientname);
|
|
db_results res=q->Read();
|
|
db->destroyQuery(q);
|
|
|
|
if (!res.empty() &&
|
|
perm_uid!=nullptr)
|
|
{
|
|
*perm_uid = res[0]["perm_uid"];
|
|
}
|
|
|
|
if(res.size()>0)
|
|
return watoi(res[0]["id"]);
|
|
else
|
|
{
|
|
IQuery *q_get_num_clients=db->Prepare("SELECT count(*) AS c FROM clients WHERE lastseen > date('now', '-2 month')", false);
|
|
db_results res_r=q_get_num_clients->Read();
|
|
q_get_num_clients->Reset();
|
|
int c_clients=-1;
|
|
if(!res_r.empty()) c_clients=watoi(res_r[0]["c"]);
|
|
|
|
db->destroyQuery(q_get_num_clients);
|
|
|
|
if(server_settings==NULL || c_clients<server_settings->getSettings()->max_active_clients)
|
|
{
|
|
std::vector<char> uid(16);
|
|
Server->secureRandomFill(uid.data(), uid.size());
|
|
|
|
if (perm_uid != nullptr)
|
|
{
|
|
perm_uid->assign(uid.data(), uid.size());
|
|
}
|
|
|
|
DBScopedWriteTransaction trans(db);
|
|
IQuery *q_insert_newclient=db->Prepare("INSERT INTO clients "
|
|
"(name, lastseen,bytes_used_files,bytes_used_images,created,perm_uid) "
|
|
"VALUES (?, CURRENT_TIMESTAMP, 0, 0, strftime('%s', 'now'), ?)", false);
|
|
q_insert_newclient->Bind(clientname);
|
|
q_insert_newclient->Bind(uid.data(), uid.size());
|
|
q_insert_newclient->Write();
|
|
int rid=(int)db->getLastInsertID();
|
|
q_insert_newclient->Reset();
|
|
db->destroyQuery(q_insert_newclient);
|
|
|
|
IQuery *q_insert_setting=db->Prepare("INSERT INTO settings_db.settings (key,value, clientid) VALUES (?,?,?)", false);
|
|
std::string new_authkey = ServerSettings::generateRandomAuthKey();
|
|
q_insert_setting->Bind("internet_authkey");
|
|
q_insert_setting->Bind(new_authkey);
|
|
q_insert_setting->Bind(rid);
|
|
q_insert_setting->Write();
|
|
q_insert_setting->Reset();
|
|
|
|
if (client_group != NULL)
|
|
{
|
|
q_insert_setting->Bind("group_id");
|
|
q_insert_setting->Bind(*client_group);
|
|
q_insert_setting->Bind(rid);
|
|
q_insert_setting->Write();
|
|
q_insert_setting->Reset();
|
|
|
|
IQuery* q_set_group = db->Prepare("UPDATE clients SET groupid=? WHERE id=?", false);
|
|
q_set_group->Bind(*client_group);
|
|
q_set_group->Bind(rid);
|
|
q_set_group->Write();
|
|
q_set_group->Reset();
|
|
db->destroyQuery(q_set_group);
|
|
}
|
|
|
|
db->destroyQuery(q_insert_setting);
|
|
|
|
if (authkey != NULL)
|
|
{
|
|
*authkey = new_authkey;
|
|
}
|
|
|
|
if(new_client!=NULL)
|
|
*new_client=true;
|
|
|
|
|
|
|
|
return rid;
|
|
}
|
|
else
|
|
{
|
|
Server->Log("Too many clients. Didn't accept client '"+clientname+"'", LL_INFO);
|
|
return -1;
|
|
}
|
|
}
|
|
}
|
|
|
|
void ClientMain::updateLastseen(int64 lastseen)
|
|
{
|
|
q_update_lastseen->Bind(lastseen);
|
|
q_update_lastseen->Bind(clientid);
|
|
q_update_lastseen->Write();
|
|
q_update_lastseen->Reset();
|
|
}
|
|
|
|
bool ClientMain::isUpdateFull(int tgroup)
|
|
{
|
|
int update_freq_full = server_settings->getUpdateFreqFileFull();
|
|
if( update_freq_full<0 )
|
|
return false;
|
|
|
|
int update_freq_incr = server_settings->getUpdateFreqFileIncr();
|
|
if(update_freq_incr<0)
|
|
update_freq_incr=0;
|
|
|
|
return !backup_dao->hasRecentFullOrIncrFileBackup(convert(-1*update_freq_full)+" seconds",
|
|
clientid, convert(-1*update_freq_incr)+" seconds", tgroup).exists;
|
|
}
|
|
|
|
bool ClientMain::isUpdateIncr(int tgroup)
|
|
{
|
|
int update_freq = server_settings->getUpdateFreqFileIncr();
|
|
if( update_freq<0 )
|
|
return false;
|
|
|
|
return !backup_dao->hasRecentIncrFileBackup(convert(-1*update_freq)+" seconds",
|
|
clientid, tgroup).exists;
|
|
}
|
|
|
|
bool ClientMain::isUpdateFullImage(const std::string &letter)
|
|
{
|
|
int update_freq_full = server_settings->getUpdateFreqImageFull();
|
|
if( update_freq_full<0 )
|
|
return false;
|
|
|
|
int update_freq_incr = server_settings->getUpdateFreqImageIncr();
|
|
if(update_freq_incr<0)
|
|
update_freq_incr=0;
|
|
|
|
return !backup_dao->hasRecentFullOrIncrImageBackup(convert(-1*update_freq_full)+" seconds",
|
|
clientid, convert(-1*update_freq_incr)+" seconds", curr_image_version, normalizeVolumeUpper(letter)).exists;
|
|
}
|
|
|
|
bool ClientMain::isUpdateFullImage(void)
|
|
{
|
|
std::vector<std::string> vols=server_settings->getBackupVolumes(all_volumes, all_nonusb_volumes);
|
|
for(size_t i=0;i<vols.size();++i)
|
|
{
|
|
if( isUpdateFullImage(vols[i]) )
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool ClientMain::isUpdateIncrImage(void)
|
|
{
|
|
std::vector<std::string> vols=server_settings->getBackupVolumes(all_volumes, all_nonusb_volumes);
|
|
for(size_t i=0;i<vols.size();++i)
|
|
{
|
|
if( isUpdateIncrImage(vols[i]) )
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
bool ClientMain::isUpdateIncrImage(const std::string &letter)
|
|
{
|
|
int update_freq = server_settings->getUpdateFreqImageIncr();
|
|
if( update_freq<0 )
|
|
return false;
|
|
|
|
return !backup_dao->hasRecentIncrImageBackup(convert(-1*update_freq)+" seconds",
|
|
clientid, curr_image_version, normalizeVolumeUpper(letter)).exists;
|
|
}
|
|
|
|
std::string ClientMain::sendClientMessageRetry(const std::string &msg, const std::string &errmsg, unsigned int timeout,
|
|
size_t retry, bool logerr, int max_loglevel, unsigned int timeout_after_first, bool do_encrypt)
|
|
{
|
|
std::string res;
|
|
do
|
|
{
|
|
int64 starttime=Server->getTimeMS();
|
|
res = sendClientMessage(msg, errmsg, timeout, logerr, retry>0 ? LL_DEBUG : max_loglevel, NULL, do_encrypt);
|
|
|
|
if(res.empty())
|
|
{
|
|
if(retry>0)
|
|
{
|
|
if (timeout_after_first > 0)
|
|
{
|
|
timeout = timeout_after_first;
|
|
}
|
|
--retry;
|
|
|
|
int64 passed_time=Server->getTimeMS()-starttime;
|
|
if(passed_time<timeout)
|
|
{
|
|
Server->wait(static_cast<unsigned int>(timeout-passed_time));
|
|
}
|
|
}
|
|
else
|
|
{
|
|
return res;
|
|
}
|
|
}
|
|
}
|
|
while(res.empty());
|
|
|
|
return res;
|
|
}
|
|
|
|
std::string ClientMain::sendClientMessage(const std::string &msg, const std::string &errmsg,
|
|
unsigned int timeout, bool logerr, int max_loglevel, SConnection* conn, bool do_encrypt)
|
|
{
|
|
CTCPStack tcpstack(internet_connection);
|
|
|
|
std::unique_ptr<IPipe> cc;
|
|
if (conn != NULL
|
|
&& conn->conn.get() != NULL
|
|
&& conn->internet_connection == internet_connection)
|
|
{
|
|
cc.reset(conn->conn.release());
|
|
}
|
|
else
|
|
{
|
|
cc.reset(getClientCommandConnection(NULL, 10000, NULL, do_encrypt));
|
|
if (cc.get() == NULL)
|
|
{
|
|
if (logerr)
|
|
ServerLogger::Log(logid, "Connecting to ClientService of \"" + clientname + "\" failed: " + errmsg, max_loglevel);
|
|
else
|
|
Server->Log("Connecting to ClientService of \"" + clientname + "\" failed: " + errmsg, max_loglevel);
|
|
return "";
|
|
}
|
|
}
|
|
if (conn != NULL)
|
|
{
|
|
conn->conn.reset();
|
|
}
|
|
|
|
std::string identity = getIdentity();
|
|
|
|
tcpstack.Send(cc.get(), identity+msg);
|
|
|
|
std::string ret;
|
|
int64 starttime=Server->getTimeMS();
|
|
bool ok=false;
|
|
bool herr=false;
|
|
while(Server->getTimeMS()-starttime<=timeout)
|
|
{
|
|
size_t rc=cc->Read(&ret, timeout);
|
|
if(rc==0)
|
|
{
|
|
if(logerr)
|
|
ServerLogger::Log(logid, errmsg, max_loglevel);
|
|
else
|
|
Server->Log(errmsg, max_loglevel);
|
|
|
|
break;
|
|
}
|
|
tcpstack.AddData((char*)ret.c_str(), ret.size());
|
|
|
|
if(tcpstack.getPacket(ret))
|
|
{
|
|
if (conn != NULL)
|
|
{
|
|
conn->conn.reset(cc.release());
|
|
conn->internet_connection = internet_connection;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
}
|
|
|
|
if(logerr)
|
|
ServerLogger::Log(logid, "Timeout: "+errmsg, max_loglevel);
|
|
else
|
|
Server->Log("Timeout: "+errmsg, max_loglevel);
|
|
|
|
return "";
|
|
}
|
|
|
|
bool ClientMain::sendClientMessageRetry(const std::string &msg, const std::string &retok, const std::string &errmsg, unsigned int timeout,
|
|
size_t retry, bool logerr, int max_loglevel, bool *retok_err, std::string* retok_str, bool do_encrypt)
|
|
{
|
|
bool res;
|
|
do
|
|
{
|
|
int64 starttime=Server->getTimeMS();
|
|
res = sendClientMessage(msg, retok, errmsg, timeout, logerr, retry>0 ? LL_DEBUG : max_loglevel, retok_err, retok_str, NULL, do_encrypt);
|
|
|
|
if(!res)
|
|
{
|
|
if(retry>0)
|
|
{
|
|
--retry;
|
|
int64 passed_time=Server->getTimeMS()-starttime;
|
|
|
|
if(passed_time<timeout)
|
|
{
|
|
Server->wait(static_cast<unsigned int>(timeout-passed_time));
|
|
}
|
|
}
|
|
else
|
|
{
|
|
return res;
|
|
}
|
|
}
|
|
}
|
|
while(!res);
|
|
|
|
return res;
|
|
}
|
|
|
|
bool ClientMain::sendClientMessage(const std::string &msg, const std::string &retok,
|
|
const std::string &errmsg, unsigned int timeout, bool logerr, int max_loglevel, bool *retok_err,
|
|
std::string* retok_str, SConnection* conn, bool do_encrypt)
|
|
{
|
|
CTCPStack tcpstack(internet_connection);
|
|
|
|
std::unique_ptr<IPipe> cc;
|
|
if (conn != NULL
|
|
&& conn->conn.get() != NULL
|
|
&& conn->internet_connection == internet_connection)
|
|
{
|
|
cc.reset(conn->conn.release());
|
|
}
|
|
else
|
|
{
|
|
cc.reset(getClientCommandConnection(NULL, 10000, NULL, do_encrypt));
|
|
if (cc.get() == NULL)
|
|
{
|
|
if (logerr)
|
|
ServerLogger::Log(logid, "Connecting to ClientService of \"" + clientname + "\" failed: " + errmsg, max_loglevel);
|
|
else
|
|
Server->Log("Connecting to ClientService of \"" + clientname + "\" failed: " + errmsg, max_loglevel);
|
|
|
|
return false;
|
|
}
|
|
}
|
|
|
|
if (conn != NULL)
|
|
{
|
|
conn->conn.reset();
|
|
}
|
|
|
|
std::string identity = getIdentity();
|
|
|
|
tcpstack.Send(cc.get(), identity+msg);
|
|
|
|
std::string ret;
|
|
int64 starttime=Server->getTimeMS();
|
|
bool ok=false;
|
|
bool herr=false;
|
|
bool broken = false;
|
|
while(Server->getTimeMS()-starttime<=timeout)
|
|
{
|
|
size_t rc=cc->Read(&ret, timeout);
|
|
if(rc==0)
|
|
{
|
|
broken = true;
|
|
break;
|
|
}
|
|
tcpstack.AddData((char*)ret.c_str(), ret.size());
|
|
|
|
size_t packetsize;
|
|
char *pck=tcpstack.getPacket(&packetsize);
|
|
if(pck!=NULL && packetsize>0)
|
|
{
|
|
ret=pck;
|
|
delete [] pck;
|
|
if(retok_str!=NULL)
|
|
{
|
|
*retok_str=ret;
|
|
}
|
|
if(ret!=retok)
|
|
{
|
|
herr=true;
|
|
if (logerr)
|
|
ServerLogger::Log(logid, errmsg, max_loglevel);
|
|
else
|
|
Server->Log(errmsg, max_loglevel);
|
|
|
|
if(retok_err!=NULL)
|
|
*retok_err=true;
|
|
|
|
break;
|
|
}
|
|
else
|
|
{
|
|
ok=true;
|
|
break;
|
|
}
|
|
}
|
|
else if(pck!=NULL)
|
|
{
|
|
delete []pck;
|
|
}
|
|
}
|
|
if(!ok && !herr)
|
|
{
|
|
std::string reason = (broken ? "Connection broken: " : "Timeout: ");
|
|
|
|
if(logerr)
|
|
ServerLogger::Log(logid, reason+errmsg, max_loglevel);
|
|
else
|
|
Server->Log(reason +errmsg, max_loglevel);
|
|
}
|
|
|
|
if (conn!=NULL && (ok || herr) )
|
|
{
|
|
conn->conn.reset(cc.release());
|
|
conn->internet_connection = internet_connection;
|
|
}
|
|
|
|
return ok;
|
|
}
|
|
|
|
void ClientMain::sendClientBackupIncrIntervall(void)
|
|
{
|
|
SSettings* settings = server_settings->getSettings();
|
|
int incr_freq=INT_MAX;
|
|
if(server_settings->getUpdateFreqFileIncr()>0)
|
|
{
|
|
incr_freq = (std::min)(incr_freq, static_cast<int>(server_settings->getUpdateFreqFileIncr()*settings->backup_ok_mod_file));
|
|
}
|
|
if(server_settings->getUpdateFreqImageIncr()>0)
|
|
{
|
|
incr_freq = (std::min)(incr_freq, static_cast<int>(server_settings->getUpdateFreqImageIncr()*settings->backup_ok_mod_image));
|
|
}
|
|
if(server_settings->getUpdateFreqFileFull()>0)
|
|
{
|
|
incr_freq = (std::min)(incr_freq, static_cast<int>(server_settings->getUpdateFreqFileFull()*settings->backup_ok_mod_file));
|
|
}
|
|
if(server_settings->getUpdateFreqImageFull()>0)
|
|
{
|
|
incr_freq = (std::min)(incr_freq, static_cast<int>(server_settings->getUpdateFreqImageFull()*settings->backup_ok_mod_image));
|
|
}
|
|
int curr_startup_backup_delay = settings->startup_backup_delay;
|
|
if(incr_freq!=INT_MAX && (incr_freq!=last_incr_freq || curr_startup_backup_delay!=last_startup_backup_delay) )
|
|
{
|
|
std::string delay_str;
|
|
if(curr_startup_backup_delay>0)
|
|
{
|
|
delay_str="?startup_delay="+convert(curr_startup_backup_delay);
|
|
}
|
|
|
|
if(sendClientMessage("INCRINTERVALL \""+convert(incr_freq)+delay_str+"\"", "OK", "Sending backup interval to client failed", 10000))
|
|
{
|
|
last_incr_freq = incr_freq;
|
|
last_startup_backup_delay = curr_startup_backup_delay;
|
|
}
|
|
}
|
|
}
|
|
|
|
bool ClientMain::updateCapabilities(bool* needs_restart)
|
|
{
|
|
std::string capa_cmd = "CAPA";
|
|
if (!clientsubname.empty())
|
|
{
|
|
capa_cmd += " clientsubname=" + EscapeParamString(clientsubname);
|
|
}
|
|
|
|
std::string cap=sendClientMessageRetry(capa_cmd, "Querying client capabilities failed", 10000, 10, false);
|
|
if(cap!="ERR" && !cap.empty())
|
|
{
|
|
int new_capa = 0;
|
|
str_map params;
|
|
ParseParamStrHttp(cap, ¶ms);
|
|
if(params["IMAGE"]!="1")
|
|
{
|
|
Server->Log("Client doesn't have IMAGE capability", LL_DEBUG);
|
|
new_capa |= CAPA_NO_IMAGE_BACKUPS;
|
|
can_backup_images=false;
|
|
}
|
|
str_map::iterator it=params.find("FILESRV");
|
|
if(it!=params.end())
|
|
{
|
|
protocol_versions.filesrv_protocol_version=watoi(it->second);
|
|
}
|
|
|
|
it=params.find("FILE");
|
|
if(it!=params.end())
|
|
{
|
|
protocol_versions.file_protocol_version=watoi(it->second);
|
|
}
|
|
it=params.find("FILE2");
|
|
if(it!=params.end())
|
|
{
|
|
protocol_versions.file_protocol_version_v2=watoi(it->second);
|
|
}
|
|
it=params.find("SET_SETTINGS");
|
|
if(it!=params.end())
|
|
{
|
|
protocol_versions.set_settings_version=watoi(it->second);
|
|
}
|
|
it=params.find("IMAGE_VER");
|
|
if(it!=params.end())
|
|
{
|
|
protocol_versions.image_protocol_version=watoi(it->second);
|
|
}
|
|
it=params.find("CLIENTUPDATE");
|
|
if(it!=params.end())
|
|
{
|
|
update_version=watoi(it->second);
|
|
}
|
|
std::string client_version_str;
|
|
it=params.find("CLIENT_VERSION_STR");
|
|
if(it!=params.end())
|
|
{
|
|
client_version_str = it->second;
|
|
ServerStatus::setClientVersionString(clientname, it->second);
|
|
}
|
|
std::string os_version_str;
|
|
it=params.find("OS_VERSION_STR");
|
|
if(it!=params.end())
|
|
{
|
|
os_version_str = it->second;
|
|
ServerStatus::setOSVersionString(clientname, it->second);
|
|
}
|
|
it=params.find("ALL_VOLUMES");
|
|
if(it!=params.end())
|
|
{
|
|
all_volumes=(it->second);
|
|
}
|
|
it=params.find("ALL_NONUSB_VOLUMES");
|
|
if(it!=params.end())
|
|
{
|
|
all_nonusb_volumes=(it->second);
|
|
}
|
|
it=params.find("ETA");
|
|
if(it!=params.end())
|
|
{
|
|
protocol_versions.eta_version=watoi(it->second);
|
|
}
|
|
it=params.find("CDP");
|
|
if(it!=params.end())
|
|
{
|
|
protocol_versions.cdp_version=watoi(it->second);
|
|
}
|
|
it=params.find("EFI");
|
|
if(it!=params.end())
|
|
{
|
|
protocol_versions.efi_version=watoi(it->second);
|
|
}
|
|
it=params.find("FILE_META");
|
|
if(it!=params.end())
|
|
{
|
|
protocol_versions.file_meta=watoi(it->second);
|
|
}
|
|
it=params.find("SELECT_SHA");
|
|
if(it!=params.end())
|
|
{
|
|
protocol_versions.select_sha_version=watoi(it->second);
|
|
}
|
|
it = params.find("CLIENT_BITMAP");
|
|
if (it != params.end())
|
|
{
|
|
protocol_versions.client_bitmap_version = watoi(it->second);
|
|
}
|
|
it = params.find("REQ_PREV_CBITMAP");
|
|
if (it != params.end())
|
|
{
|
|
protocol_versions.require_previous_cbitmap = watoi(it->second);
|
|
}
|
|
it = params.find("CMD");
|
|
if (it != params.end())
|
|
{
|
|
protocol_versions.cmd_version = watoi(it->second);
|
|
}
|
|
it = params.find("OS_SIMPLE");
|
|
if (it != params.end())
|
|
{
|
|
protocol_versions.os_simple = it->second;
|
|
}
|
|
it = params.find("ASYNC_INDEX");
|
|
if (it != params.end())
|
|
{
|
|
protocol_versions.async_index_version = watoi(it->second);
|
|
}
|
|
it = params.find("SYMBIT");
|
|
if (it != params.end())
|
|
{
|
|
protocol_versions.symbit_version = watoi(it->second);
|
|
}
|
|
it = params.find("METERED");
|
|
if (it != params.end()
|
|
&& it->second=="1")
|
|
{
|
|
connection_metered = true;
|
|
}
|
|
it = params.find("LOCKED");
|
|
if (it != params.end()
|
|
&& it->second == "1")
|
|
{
|
|
windows_locked = true;
|
|
}
|
|
it = params.find("PHASH");
|
|
if (it != params.end())
|
|
{
|
|
protocol_versions.phash_version = watoi(it->second);
|
|
}
|
|
it = params.find("WTOKENS");
|
|
if (it != params.end())
|
|
{
|
|
protocol_versions.wtokens_version = watoi(it->second);
|
|
}
|
|
it = params.find("UPDATE_VOLS");
|
|
if (it != params.end())
|
|
{
|
|
protocol_versions.update_vols = watoi(it->second);
|
|
}
|
|
it=params.find("RESTORE");
|
|
if(it!=params.end())
|
|
{
|
|
if(it->second=="client-confirms")
|
|
{
|
|
ServerStatus::setRestore(clientname, ERestore_client_confirms);
|
|
}
|
|
else if(it->second=="server-confirms")
|
|
{
|
|
ServerStatus::setRestore(clientname, ERestore_server_confirms);
|
|
}
|
|
else
|
|
{
|
|
ServerStatus::setRestore(clientname, ERestore_disabled);
|
|
}
|
|
}
|
|
it = params.find("RESTORE_VER");
|
|
if (it != params.end())
|
|
{
|
|
protocol_versions.restore_version = watoi(it->second);
|
|
}
|
|
it = params.find("clientuid");
|
|
if (it != params.end())
|
|
{
|
|
/*
|
|
Malicious client can impersonate a good client for a while and change its
|
|
settings (including disabling encryption). So prevent new clients until
|
|
confirmed.
|
|
*/
|
|
ServerBackupDao::CondString curr_uid = backup_dao->getClientUid(clientid);
|
|
if (curr_uid.exists &&
|
|
!curr_uid.value.empty() &&
|
|
it->second!=curr_uid.value &&
|
|
server_settings->getSettings()->local_encrypt &&
|
|
!internet_connection)
|
|
{
|
|
ServerLogger::Log(logid, "Client UID changed from \"" + curr_uid.value + "\" to \"" + it->second + "\". "
|
|
"Disallowing client because connection to local/passive client is encrypted.", LL_WARNING);
|
|
ServerStatus::setStatusError(clientname, se_uid_changed);
|
|
return false;
|
|
}
|
|
|
|
if (needs_restart!=NULL
|
|
&& renameClient(it->second))
|
|
{
|
|
*needs_restart = true;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
ServerBackupDao::CondString curr_uid = backup_dao->getClientUid(clientid);
|
|
if (curr_uid.exists &&
|
|
!curr_uid.value.empty() &&
|
|
server_settings->getSettings()->local_encrypt &&
|
|
!internet_connection)
|
|
{
|
|
ServerLogger::Log(logid, "Client UID not received from client. Expecting \"" + curr_uid.value + "\". Disallowing client because connection to client is encrypted.", LL_WARNING);
|
|
ServerStatus::setStatusError(clientname, se_uid_changed);
|
|
return false;
|
|
}
|
|
}
|
|
it = params.find("UPDATE_CAPA_INTERVAL");
|
|
if(it != params.end())
|
|
{
|
|
protocol_versions.update_capa_interval = (std::max)(60000, watoi(it->second));
|
|
}
|
|
it = params.find("FILESRVTUNNEL");
|
|
if (it != params.end())
|
|
{
|
|
protocol_versions.filesrvtunnel = watoi(it->second);
|
|
}
|
|
it = params.find("BACKUP");
|
|
if (it != params.end())
|
|
{
|
|
if (it->second == "incr-file")
|
|
do_incr_backup_now = true;
|
|
else if (it->second == "full-file")
|
|
do_full_backup_now = true;
|
|
else if (it->second == "incr-image")
|
|
do_incr_image_now = true;
|
|
else if (it->second == "full-image")
|
|
do_full_image_now = true;
|
|
else
|
|
Server->Log("Unknown immediate backup type \"" + it->second + "\"", LL_WARNING);
|
|
}
|
|
|
|
bool update_settings = false;
|
|
size_t idx = 0;
|
|
while ((it = params.find("def_key_" + convert(idx))) != params.end())
|
|
{
|
|
std::string val = params["def_val_" + convert(idx)];
|
|
ServerBackupDao::CondString setting = backup_dao->getSetting(clientid, it->second);
|
|
if (!setting.exists)
|
|
{
|
|
backup_dao->insertSetting(it->second, val, clientid);
|
|
update_settings = true;
|
|
}
|
|
else if ( (it->second=="virtual_clients_add" || it->second =="image_snapshot_groups_def")
|
|
&& setting.value != val)
|
|
{
|
|
backup_dao->updateSetting(val, it->second, clientid);
|
|
update_settings = true;
|
|
}
|
|
++idx;
|
|
}
|
|
|
|
if (update_settings)
|
|
{
|
|
ServerSettings::updateClient(clientid);
|
|
}
|
|
|
|
backup_dao->updateClientOsAndClientVersion(protocol_versions.os_simple,
|
|
os_version_str, client_version_str, new_capa, clientid);
|
|
|
|
return true;
|
|
}
|
|
else
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
|
|
void ClientMain::sendSettings(void)
|
|
{
|
|
std::string s_settings;
|
|
|
|
if(!clientsubname.empty())
|
|
{
|
|
s_settings+="clientsubname="+clientsubname+"\n";
|
|
s_settings+="filebackup_group_offset="+convert(filebackup_group_offset)+"\n";
|
|
}
|
|
|
|
std::vector<std::string> settings_names=getSettingsList();
|
|
std::vector<std::string> global_settings_names=getGlobalizedSettingsList();
|
|
|
|
std::unique_ptr<ISettingsReader> settings_client, settings_default, settings_global;
|
|
int settings_default_id;
|
|
server_settings->createSettingsReaders(db, clientid, settings_default, settings_client, settings_global, settings_default_id);
|
|
|
|
ISettingsReader* settings_global_ptr = settings_global.get() != NULL ? settings_global.get() : settings_default.get();
|
|
|
|
SSettings* settings = server_settings->getSettings();
|
|
|
|
bool allow_overwrite = settings->allow_overwrite;
|
|
|
|
for (size_t i = 0; i < settings_names.size(); ++i)
|
|
{
|
|
std::string& key = settings_names[i];
|
|
std::string value;
|
|
|
|
bool globalized = std::find(global_settings_names.begin(), global_settings_names.end(), key) != global_settings_names.end();
|
|
|
|
if (globalized)
|
|
{
|
|
if (settings_global_ptr->getValue(key, &value))
|
|
{
|
|
s_settings += key + "=" + value + "\n";
|
|
s_settings += key + "_def=" + value + "\n";
|
|
}
|
|
}
|
|
else
|
|
{
|
|
std::string value_default;
|
|
if (settings_default_id != 0)
|
|
{
|
|
ServerBackupDao::SSetting setting_group = backup_dao->getServerSetting(key, 0);
|
|
if (setting_group.exists)
|
|
{
|
|
value_default = setting_group.value;
|
|
}
|
|
}
|
|
|
|
ServerBackupDao::SSetting setting_group = backup_dao->getServerSetting(key, settings_default_id);
|
|
if (setting_group.exists
|
|
&& (settings_default_id == 0 || setting_group.use == c_use_value))
|
|
{
|
|
value_default = setting_group.value;
|
|
}
|
|
|
|
s_settings += key + ".group=" + value_default + "\n";
|
|
|
|
ServerBackupDao::SSetting setting = backup_dao->getServerSetting(key, clientid);
|
|
|
|
if (!setting.exists)
|
|
{
|
|
s_settings += key + "=" + value_default + "\n";
|
|
}
|
|
else
|
|
{
|
|
if (key == "computername"
|
|
|| key == "internet_authkey")
|
|
{
|
|
setting.use = c_use_value;
|
|
}
|
|
else
|
|
{
|
|
if (key == "update_freq_incr")
|
|
int abct = 5;
|
|
|
|
s_settings += key + ".home=" + setting.value + "\n";
|
|
s_settings += key + ".client=" + setting.value_client + "\n";
|
|
s_settings += key + ".use=" + convert(setting.use) + "\n";
|
|
s_settings += key + ".use_lm=" + convert(setting.use_last_modified) + "\n";
|
|
}
|
|
|
|
if (setting.use == c_use_group)
|
|
value = value_default;
|
|
else if (setting.use == c_use_value)
|
|
value = setting.value;
|
|
else if (setting.use == c_use_value_client)
|
|
value = setting.value_client;
|
|
else
|
|
value = setting.value;
|
|
|
|
s_settings += key + "=" + value + "\n";
|
|
}
|
|
}
|
|
}
|
|
s_settings += "server_token=" + server_token + "\n";
|
|
s_settings += "perm_uid=" + bytesToHex(perm_uid) + "\n";
|
|
escapeClientMessage(s_settings);
|
|
if(!sendClientMessage("SETTINGS "+s_settings, "OK", "Sending settings to client failed", 10000))
|
|
{
|
|
do_update_settings = true;
|
|
}
|
|
}
|
|
|
|
bool ClientMain::getClientSettings(bool& doesnt_exist)
|
|
{
|
|
doesnt_exist=false;
|
|
std::string identity = getIdentity();
|
|
|
|
FileClient fc(false, identity, protocol_versions.filesrv_protocol_version, internet_connection, this, use_tmpfiles?NULL:this);
|
|
_u32 rc=getClientFilesrvConnection(&fc, server_settings.get());
|
|
if(rc!=ERR_CONNECTED)
|
|
{
|
|
ServerLogger::Log(logid, "Getting Client settings of "+clientname+" failed - CONNECT error", LL_ERROR);
|
|
return false;
|
|
}
|
|
|
|
IFsFile *tmp=getTemporaryFileRetry(true, std::string(), logid);
|
|
if(tmp==NULL)
|
|
{
|
|
ServerLogger::Log(logid, "Error creating temporary file in BackupServerGet::getClientSettings", LL_ERROR);
|
|
return false;
|
|
}
|
|
ScopedDeleteFile delete_tmp(tmp);
|
|
|
|
std::string settings_fn = "urbackup/settings.cfg";
|
|
|
|
if(!clientsubname.empty())
|
|
{
|
|
settings_fn = "urbackup/settings_"+(clientsubname)+".cfg";
|
|
}
|
|
|
|
rc=fc.GetFile(settings_fn, tmp, true, false, 0, false, 0);
|
|
if(rc!=ERR_SUCCESS)
|
|
{
|
|
ServerLogger::Log(logid, "Error getting Client settings of "+clientname+". Errorcode: "+fc.getErrorString(rc)+" ("+convert(rc)+")", LL_ERROR);
|
|
|
|
if(rc== ERR_CANNOT_OPEN_FILE)
|
|
{
|
|
doesnt_exist=true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
std::string settings_data = readToString(tmp);
|
|
std::string tmp_fn = tmp->getFilename();
|
|
delete_tmp.release();
|
|
ScopedDeleteFn delete_fn(tmp_fn);
|
|
Server->destroy(tmp);
|
|
|
|
std::unique_ptr<ISettingsReader> sr(Server->createFileSettingsReader(tmp_fn));
|
|
|
|
std::vector<std::string> setting_names=getSettingsList();
|
|
|
|
bool mod=false;
|
|
|
|
std::vector<std::string> only_server_settings = getOnlyServerClientSettingsList();
|
|
|
|
bool has_use = false;
|
|
|
|
for (size_t i = 0; i < setting_names.size(); ++i)
|
|
{
|
|
std::string value;
|
|
if (sr->getValue(setting_names[i] + ".use", &value))
|
|
{
|
|
has_use = true;
|
|
break;
|
|
}
|
|
}
|
|
|
|
for(size_t i=0;i<setting_names.size();++i)
|
|
{
|
|
std::string &key=setting_names[i];
|
|
std::string value;
|
|
|
|
if(std::find(only_server_settings.begin(), only_server_settings.end(),
|
|
key)!=only_server_settings.end())
|
|
{
|
|
continue;
|
|
}
|
|
|
|
int use = c_use_value;
|
|
|
|
if (sr->getValue(key + ".use", &value))
|
|
{
|
|
use = watoi(value);
|
|
std::string use_lm_str;
|
|
int64 use_lm = 0;
|
|
if (sr->getValue(key + ".use_lm", &use_lm_str))
|
|
use_lm = watoi64(use_lm_str);
|
|
|
|
if ( (use & c_use_value_client)>0
|
|
&& sr->getValue(key + ".client", &value))
|
|
{
|
|
if (internet_connection && key == "computername")
|
|
{
|
|
continue;
|
|
}
|
|
|
|
bool b = updateClientSetting(key, value, use, use_lm);
|
|
if (b)
|
|
mod = true;
|
|
}
|
|
}
|
|
else if (!has_use
|
|
&& sr->getValue(key, &value))
|
|
{
|
|
if (internet_connection && key == "computername")
|
|
{
|
|
continue;
|
|
}
|
|
|
|
bool b = updateClientSetting(key, value, c_use_undefined, 0);
|
|
if (b)
|
|
mod = true;
|
|
}
|
|
}
|
|
|
|
if(mod)
|
|
{
|
|
server_settings->update(true);
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool ClientMain::updateClientSetting(const std::string &key, const std::string &value, int use, int64 use_last_modified)
|
|
{
|
|
if (key == "update_freq_incr")
|
|
int abct = 45;
|
|
|
|
q_get_setting->Bind(clientid);
|
|
q_get_setting->Bind(key);
|
|
db_results res = q_get_setting->Read();
|
|
q_get_setting->Reset();
|
|
|
|
bool use_mod = false;
|
|
|
|
if (!res.empty())
|
|
{
|
|
int64 old_use_last_mod = watoi64(res[0]["use_last_modified"]);
|
|
int old_use = watoi(res[0]["use"]);
|
|
|
|
if (old_use_last_mod > use_last_modified)
|
|
{
|
|
use = old_use;
|
|
use_last_modified = old_use_last_mod;
|
|
}
|
|
|
|
if (use != c_use_undefined &&
|
|
old_use != use)
|
|
{
|
|
use_mod = true;
|
|
}
|
|
else if (use == c_use_undefined)
|
|
{
|
|
use = old_use;
|
|
}
|
|
}
|
|
|
|
|
|
if(res.empty())
|
|
{
|
|
q_insert_setting->Bind(key);
|
|
q_insert_setting->Bind(value);
|
|
q_insert_setting->Bind(clientid);
|
|
q_insert_setting->Bind(use);
|
|
q_insert_setting->Bind(use_last_modified);
|
|
q_insert_setting->Write();
|
|
q_insert_setting->Reset();
|
|
return true;
|
|
}
|
|
else if(res[0]["value_client"]!=value
|
|
|| use_mod )
|
|
{
|
|
q_update_setting->Bind(value);
|
|
q_update_setting->Bind(use);
|
|
q_update_setting->Bind(use_last_modified);
|
|
q_update_setting->Bind(key);
|
|
q_update_setting->Bind(clientid);
|
|
q_update_setting->Write();
|
|
q_update_setting->Reset();
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
void ClientMain::sendToPipe(const std::string &msg)
|
|
{
|
|
pipe->Write(msg);
|
|
}
|
|
|
|
void ClientMain::sendClientLogdata(void)
|
|
{
|
|
q_get_unsent_logdata->Bind(clientid);
|
|
db_results res=q_get_unsent_logdata->Read();
|
|
q_get_unsent_logdata->Reset();
|
|
|
|
for(size_t i=0;i<res.size();++i)
|
|
{
|
|
std::string logdata=(res[i]["logdata"]);
|
|
escapeClientMessage(logdata);
|
|
if(sendClientMessage("2LOGDATA "+res[i]["created"]+" "+logdata, "OK", "Sending logdata to client failed", 10000, false, LL_WARNING))
|
|
{
|
|
q_set_logdata_sent->Bind(res[i]["id"]);
|
|
q_set_logdata_sent->Write();
|
|
q_set_logdata_sent->Reset();
|
|
}
|
|
}
|
|
}
|
|
|
|
MailServer ClientMain::getMailServerSettings(void)
|
|
{
|
|
ISettingsReader *settings=Server->createDBSettingsReader(Server->getDatabase(Server->getThreadID(), URBACKUPDB_SERVER), "settings_db.settings", "SELECT value FROM settings_db.settings WHERE key=? AND clientid=0");
|
|
|
|
MailServer ms;
|
|
ms.servername=settings->getValue("mail_servername", "");
|
|
ms.port=(unsigned short)watoi(settings->getValue("mail_serverport", "587"));
|
|
ms.username=settings->getValue("mail_username", "");
|
|
ms.password=settings->getValue("mail_password", "");
|
|
ms.mailfrom=settings->getValue("mail_from", "");
|
|
if(ms.mailfrom.empty())
|
|
ms.mailfrom="report@urbackup.example.com";
|
|
|
|
ms.ssl_only = settings->getValue("mail_ssl_only", "false")=="true";
|
|
ms.check_certificate = settings->getValue("mail_check_certificate", "false")=="true";
|
|
ms.use_smtps = settings->getValue("mail_use_smtps", "false") == "true";
|
|
|
|
Server->destroy(settings);
|
|
return ms;
|
|
}
|
|
|
|
bool ClientMain::sendMailToAdmins(const std::string& subj, const std::string& message)
|
|
{
|
|
MailServer mail_server=getMailServerSettings();
|
|
if(mail_server.servername.empty())
|
|
return false;
|
|
|
|
if(url_fak==NULL)
|
|
return false;
|
|
|
|
ISettingsReader *settings=Server->createDBSettingsReader(Server->getDatabase(Server->getThreadID(), URBACKUPDB_SERVER), "settings_db.settings", "SELECT value FROM settings_db.settings WHERE key=? AND clientid=0");
|
|
std::string admin_addrs_str=settings->getValue("mail_admin_addrs", "");
|
|
|
|
if(admin_addrs_str.empty())
|
|
return false;
|
|
|
|
std::vector<std::string> admin_addrs;
|
|
Tokenize(admin_addrs_str, admin_addrs, ";,");
|
|
|
|
std::string errmsg;
|
|
bool b=url_fak->sendMail(mail_server, admin_addrs, "[UrBackup] "+subj, message, &errmsg);
|
|
if(!b)
|
|
{
|
|
Server->Log("Sending mail failed. "+errmsg, LL_WARNING);
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
|
|
|
|
void ClientMain::checkClientVersion(void)
|
|
{
|
|
std::string version;
|
|
if (protocol_versions.os_simple == "osx")
|
|
{
|
|
version = getFile("urbackup/version_osx.txt");
|
|
}
|
|
else if (protocol_versions.os_simple == "linux")
|
|
{
|
|
version = getFile("urbackup/version_linux.txt");
|
|
}
|
|
else
|
|
{
|
|
version = getFile("urbackup/version.txt");
|
|
}
|
|
|
|
|
|
if(!version.empty())
|
|
{
|
|
std::string r=sendClientMessage("VERSION "+version, "Sending version to client failed", 10000);
|
|
if(r=="update")
|
|
{
|
|
ScopedProcess process(clientname, sa_update, version, logid, false);
|
|
|
|
std::string signature_file = "urbackup/UrBackupUpdate.sig2";
|
|
std::string installer_file = "urbackup/UrBackupUpdate.exe";
|
|
|
|
if (protocol_versions.os_simple == "osx")
|
|
{
|
|
signature_file = "urbackup/UrBackupUpdateMac.sig2";
|
|
installer_file = "urbackup/UrBackupUpdateMac.sh";
|
|
}
|
|
else if (protocol_versions.os_simple == "linux")
|
|
{
|
|
signature_file = "urbackup/UrBackupUpdateLinux.sig2";
|
|
installer_file = "urbackup/UrBackupUpdateLinux.sh";
|
|
}
|
|
else if (update_version < 2)
|
|
{
|
|
signature_file = "urbackup/UrBackupUpdate.sig";
|
|
}
|
|
|
|
|
|
std::unique_ptr<IFile> sigfile(Server->openFile(signature_file, MODE_READ));
|
|
if(sigfile.get()==NULL)
|
|
{
|
|
ServerLogger::Log(logid, "Error opening sigfile", LL_ERROR);
|
|
return;
|
|
}
|
|
std::unique_ptr<IFile> updatefile(Server->openFile(installer_file, MODE_READ));
|
|
if(updatefile.get()==NULL)
|
|
{
|
|
ServerLogger::Log(logid, "Error opening updatefile", LL_ERROR);
|
|
return;
|
|
}
|
|
size_t datasize=3*sizeof(_u32)+version.size()+(size_t)sigfile->Size()+(size_t)updatefile->Size();
|
|
|
|
CTCPStack tcpstack(internet_connection);
|
|
std::unique_ptr<IPipe> cc(getClientCommandConnection(server_settings.get(), 10000));
|
|
if(cc.get()==NULL)
|
|
{
|
|
ServerLogger::Log(logid, "Connecting to ClientService of \""+clientname+"\" failed - CONNECT error", LL_ERROR);
|
|
return;
|
|
}
|
|
|
|
std::string msg="1CLIENTUPDATE size="+convert(datasize)+"&silent_update="+convert(server_settings->getSettings()->silent_update);
|
|
|
|
std::string identity = getIdentity();
|
|
tcpstack.Send(cc.get(), identity+msg);
|
|
|
|
int timeout=5*60*1000;
|
|
|
|
_u32 c_size=little_endian((_u32)version.size());
|
|
if(!cc->Write((char*)&c_size, sizeof(c_size), timeout) )
|
|
{
|
|
return;
|
|
}
|
|
if(!cc->Write(version, timeout) )
|
|
{
|
|
return;
|
|
}
|
|
c_size= little_endian((_u32)sigfile->Size());
|
|
if(!cc->Write((char*)&c_size, sizeof(c_size), timeout) )
|
|
{
|
|
return;
|
|
}
|
|
if(!sendFile(cc.get(), sigfile.get(), timeout) )
|
|
{
|
|
return;
|
|
}
|
|
c_size= little_endian((_u32)updatefile->Size());
|
|
if(!cc->Write((char*)&c_size, sizeof(c_size), timeout) )
|
|
{
|
|
return;
|
|
}
|
|
if(!sendFile(cc.get(), updatefile.get(), timeout) )
|
|
{
|
|
return;
|
|
}
|
|
|
|
sigfile.reset();
|
|
updatefile.reset();
|
|
|
|
std::string ret;
|
|
int64 starttime=Server->getTimeMS();
|
|
bool ok=false;
|
|
while(Server->getTimeMS()-starttime<=5*60*1000)
|
|
{
|
|
size_t rc=cc->Read(&ret, timeout);
|
|
if(rc==0)
|
|
{
|
|
ServerLogger::Log(logid, "Reading from client failed in update", LL_ERROR);
|
|
break;
|
|
}
|
|
tcpstack.AddData((char*)ret.c_str(), ret.size());
|
|
|
|
std::string msg;
|
|
if(tcpstack.getPacket(msg))
|
|
{
|
|
if(msg =="ok")
|
|
{
|
|
ok=true;
|
|
break;
|
|
}
|
|
else
|
|
{
|
|
ok=false;
|
|
ServerLogger::Log(logid, "Error in update: "+ msg, LL_ERROR);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
if(!ok)
|
|
{
|
|
ServerLogger::Log(logid, "Timeout: In client update of \"" + clientname + "\"", LL_ERROR);
|
|
}
|
|
|
|
cc.reset();
|
|
|
|
client_updated_time = Server->getTimeMS();
|
|
|
|
if(ok)
|
|
{
|
|
ServerLogger::Log(logid, "Updated client \""+clientname+"\" successfully", LL_INFO);
|
|
}
|
|
|
|
updateCapabilities(NULL);
|
|
}
|
|
}
|
|
}
|
|
|
|
bool ClientMain::sendFile(IPipe *cc, IFile *f, int timeout)
|
|
{
|
|
char buf[4096];
|
|
_u32 r;
|
|
while((r=f->Read(buf, 4096))>0)
|
|
{
|
|
if(!cc->Write(buf, r, timeout))
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
FileClient::SAddrHint ClientMain::getClientaddr(void)
|
|
{
|
|
IScopedLock lock(clientaddr_mutex);
|
|
return clientaddr;
|
|
}
|
|
|
|
|
|
bool ClientMain::isBackupsRunningOkay(bool file, bool incr)
|
|
{
|
|
IScopedLock lock(running_backup_mutex);
|
|
if(running_backups<server_settings->getSettings()->max_sim_backups)
|
|
{
|
|
if (running_backups_allowed
|
|
&& incr)
|
|
{
|
|
++running_backups;
|
|
if (file)
|
|
{
|
|
++running_file_backups;
|
|
}
|
|
}
|
|
|
|
return running_backups_allowed;
|
|
}
|
|
else
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
|
|
void ClientMain::stopBackupRunning(bool file)
|
|
{
|
|
IScopedLock lock(running_backup_mutex);
|
|
if (running_backups == 0)
|
|
{
|
|
Server->Log("running_backups is zero", LL_ERROR);
|
|
assert(false);
|
|
return;
|
|
}
|
|
--running_backups;
|
|
if(file)
|
|
{
|
|
if (running_file_backups == 0)
|
|
{
|
|
Server->Log("running_file_backups is zero", LL_ERROR);
|
|
assert(false);
|
|
return;
|
|
}
|
|
--running_file_backups;
|
|
}
|
|
}
|
|
|
|
int ClientMain::getNumberOfRunningBackups(void)
|
|
{
|
|
IScopedLock lock(running_backup_mutex);
|
|
return running_backups;
|
|
}
|
|
|
|
int ClientMain::getNumberOfRunningFileBackups(void)
|
|
{
|
|
IScopedLock lock(running_backup_mutex);
|
|
return running_file_backups;
|
|
}
|
|
|
|
bool ClientMain::tooManyClients(IDatabase * db, const std::string & clientname, ServerSettings * server_settings)
|
|
{
|
|
IQuery *q = db->Prepare("SELECT id FROM clients WHERE name=?", false);
|
|
if (q == NULL)
|
|
return false;
|
|
|
|
q->Bind(clientname);
|
|
db_results res = q->Read();
|
|
db->destroyQuery(q);
|
|
|
|
if(res.empty())
|
|
{
|
|
IQuery *q_get_num_clients = db->Prepare("SELECT count(*) AS c FROM clients WHERE lastseen > date('now', '-2 month')", false);
|
|
if (q_get_num_clients == NULL)
|
|
return false;
|
|
db_results res_r = q_get_num_clients->Read();
|
|
q_get_num_clients->Reset();
|
|
int c_clients = -1;
|
|
if (!res_r.empty()) c_clients = watoi(res_r[0]["c"]);
|
|
|
|
db->destroyQuery(q_get_num_clients);
|
|
|
|
if (server_settings!=NULL
|
|
&& c_clients >= server_settings->getSettings()->max_active_clients)
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
IPipeThrottler *ClientMain::getThrottler(int speed_bps)
|
|
{
|
|
IScopedLock lock(throttle_mutex);
|
|
|
|
if(client_throttler==NULL)
|
|
{
|
|
client_throttler=Server->createPipeThrottler(new ThrottleUpdater(clientid,
|
|
internet_connection?ThrottleScope_Internet:ThrottleScope_Local));
|
|
}
|
|
else
|
|
{
|
|
bool percent_max;
|
|
size_t bps = BackupServer::throttleSpeedToBps(speed_bps, percent_max);
|
|
client_throttler->changeThrottleLimit(bps,
|
|
percent_max);
|
|
}
|
|
|
|
return client_throttler;
|
|
}
|
|
|
|
void ClientMain::updateClientAccessKey()
|
|
{
|
|
std::string access_key = ServerSettings::generateRandomAuthKey(32);
|
|
backup_dao->updateOrInsertSetting(clientid, "client_access_key", access_key);
|
|
ServerSettings::updateClient(clientid);
|
|
|
|
sendClientMessageRetry("CLIENT ACCESS KEY key=" + access_key + "&token=" + server_token, "OK",
|
|
"Error sending client access key", 10000, 10, false);
|
|
}
|
|
|
|
bool ClientMain::isDataplanOkay(bool file)
|
|
{
|
|
return isDataplanOkay(server_settings.get(), file);
|
|
}
|
|
|
|
bool ClientMain::isOnline(ServerChannelThread& channel_thread)
|
|
{
|
|
return ServerStatus::getStatus(clientname).r_online || channel_thread.isOnline();
|
|
}
|
|
|
|
bool ClientMain::isDataplanOkay(ServerSettings* local_settings, bool file)
|
|
{
|
|
if (!internet_connection)
|
|
{
|
|
return true;
|
|
}
|
|
|
|
int64 limit;
|
|
std::string pattern;
|
|
std::string hostname;
|
|
if (!connection_metered)
|
|
{
|
|
hostname = Server->LookupHostname(getClientIpStr(clientname));
|
|
if (hostname.empty())
|
|
{
|
|
return true;
|
|
}
|
|
|
|
if (!DataplanDb::getInstance()->getLimit(hostname, pattern, limit))
|
|
{
|
|
limit = 1LL * 1024 * 1024 * 1024 * 1024;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
limit = 1LL * 1024 * 1024 * 1024;
|
|
pattern = "*info from client*";
|
|
hostname = "*not retrieved*";
|
|
}
|
|
|
|
if (local_settings == NULL)
|
|
{
|
|
return true;
|
|
}
|
|
|
|
int64 appl_limit;
|
|
if (file)
|
|
{
|
|
appl_limit = local_settings->getSettings()->internet_file_dataplan_limit;
|
|
}
|
|
else
|
|
{
|
|
appl_limit = local_settings->getSettings()->internet_image_dataplan_limit;
|
|
}
|
|
|
|
if (limit<0 || limit >= appl_limit)
|
|
{
|
|
return true;
|
|
}
|
|
|
|
ServerLogger::Log(logid, std::string("Not running ") + (file ? "file" : "image")
|
|
+ " backup because currently estimated data plan of "
|
|
+ PrettyPrintBytes(limit) + " per month "
|
|
"with pattern \""+pattern+"\" for hostname \""+hostname
|
|
+"\" is smaller than configured limit of "
|
|
+ PrettyPrintBytes(appl_limit) + " per month", LL_DEBUG);
|
|
|
|
return false;
|
|
}
|
|
|
|
std::string ClientMain::getClientIpStr(const std::string & clientname)
|
|
{
|
|
FileClient::SAddrHint addr_hint;
|
|
std::string ip_addr_binary = ServerStatus::getStatus(clientname).ip_addr_binary;
|
|
if (ip_addr_binary.size() == sizeof(addr_hint.addr_ipv6))
|
|
{
|
|
addr_hint.is_ipv6 = true;
|
|
memcpy(addr_hint.addr_ipv6, ip_addr_binary.data(), ip_addr_binary.size());
|
|
}
|
|
else if (ip_addr_binary.size() == sizeof(addr_hint.addr_ipv4))
|
|
{
|
|
addr_hint.is_ipv6 = false;
|
|
memcpy(&addr_hint.addr_ipv4, ip_addr_binary.data(), ip_addr_binary.size());
|
|
}
|
|
else
|
|
{
|
|
assert(false);
|
|
}
|
|
|
|
return addr_hint.toString();
|
|
}
|
|
|
|
void ClientMain::setConnectionMetered(bool b)
|
|
{
|
|
connection_metered = b;
|
|
}
|
|
|
|
void ClientMain::setWindowsLocked(bool b)
|
|
{
|
|
windows_locked = b;
|
|
}
|
|
|
|
void ClientMain::forceReauthenticate()
|
|
{
|
|
do_reauthenticate = true;
|
|
}
|
|
|
|
bool ClientMain::inBackupWindow(Backup * backup)
|
|
{
|
|
if (backup->isFileBackup())
|
|
{
|
|
if (backup->isIncrementalBackup())
|
|
{
|
|
return ServerSettings::isInTimeSpan(server_settings->getBackupWindowIncrFile());
|
|
}
|
|
else
|
|
{
|
|
return ServerSettings::isInTimeSpan(server_settings->getBackupWindowFullFile());
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (backup->isIncrementalBackup())
|
|
{
|
|
return ServerSettings::isInTimeSpan(server_settings->getBackupWindowIncrImage());
|
|
}
|
|
else
|
|
{
|
|
return ServerSettings::isInTimeSpan(server_settings->getBackupWindowFullImage());
|
|
}
|
|
}
|
|
}
|
|
|
|
IPipe *ClientMain::getClientCommandConnection(ServerSettings* server_settings, int timeoutms, std::string* clientaddr, bool do_encrypt)
|
|
{
|
|
std::string curr_clientname = (clientname);
|
|
if(!clientsubname.empty())
|
|
{
|
|
curr_clientname = (clientmainname);
|
|
}
|
|
|
|
if(clientaddr!=NULL)
|
|
{
|
|
*clientaddr = getClientIpStr(curr_clientname);
|
|
}
|
|
if(internet_connection)
|
|
{
|
|
IPipe *ret=InternetServiceConnector::getConnection(curr_clientname, SERVICE_COMMANDS, timeoutms);
|
|
if(server_settings!=NULL && ret!=NULL)
|
|
{
|
|
int internet_speed=server_settings->getInternetSpeed();
|
|
if(internet_speed!=0
|
|
&& internet_speed!=-1)
|
|
{
|
|
ret->addThrottler(getThrottler(internet_speed));
|
|
}
|
|
int global_internet_speed=server_settings->getGlobalInternetSpeed();
|
|
if(global_internet_speed!=0
|
|
&& global_internet_speed!=-1)
|
|
{
|
|
ret->addThrottler(BackupServer::getGlobalInternetThrottler(global_internet_speed));
|
|
}
|
|
}
|
|
return ret;
|
|
}
|
|
else
|
|
{
|
|
IPipe *ret=Server->ConnectStream(getClientaddr().toString(), serviceport, timeoutms);
|
|
if (ret == NULL)
|
|
return NULL;
|
|
|
|
if(server_settings!=NULL)
|
|
{
|
|
int local_speed=server_settings->getLocalSpeed();
|
|
if(local_speed!=0
|
|
&& local_speed!=-1)
|
|
{
|
|
ret->addThrottler(getThrottler(local_speed));
|
|
}
|
|
int global_local_speed=server_settings->getGlobalLocalSpeed();
|
|
if(global_local_speed!=0
|
|
&& global_local_speed!=-1)
|
|
{
|
|
ret->addThrottler(BackupServer::getGlobalLocalThrottler(global_local_speed));
|
|
}
|
|
}
|
|
|
|
std::string secret_session_key = getSecretSessionKey();
|
|
int compression_level;
|
|
std::string compression = getSessionCompression(compression_level);
|
|
if (do_encrypt && (!secret_session_key.empty() || !compression.empty()))
|
|
{
|
|
CTCPStack tcpstack(internet_connection);
|
|
std::string identity = getIdentity();
|
|
std::string server_keyadd;
|
|
if (!secret_session_key.empty())
|
|
{
|
|
server_keyadd.resize(16);
|
|
Server->randomFill(&server_keyadd[0], server_keyadd.size());
|
|
}
|
|
std::string tosend = identity + "ENC?keyadd="+ base64_encode_dash(server_keyadd)+"&compress="+EscapeParamString(compression)+"&compress_level="+convert(compression_level);
|
|
size_t rc = tcpstack.Send(ret, tosend);
|
|
if (rc != tosend.size())
|
|
{
|
|
Server->destroy(ret);
|
|
return NULL;
|
|
}
|
|
|
|
std::string msg;
|
|
int64 starttime = Server->getTimeMS();
|
|
bool ok = false;
|
|
bool herr = false;
|
|
str_map enc_response;
|
|
while (Server->getTimeMS() - starttime <= timeoutms)
|
|
{
|
|
size_t rc = ret->Read(&msg, timeoutms);
|
|
if (rc == 0)
|
|
{
|
|
Server->destroy(ret);
|
|
return NULL;
|
|
}
|
|
tcpstack.AddData(msg.data(), msg.size());
|
|
|
|
if (tcpstack.getPacket(msg))
|
|
{
|
|
ParseParamStrHttp(msg, &enc_response);
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (enc_response["ok"] != "1")
|
|
{
|
|
Server->destroy(ret);
|
|
return NULL;
|
|
}
|
|
|
|
if (!secret_session_key.empty())
|
|
{
|
|
std::string client_keyadd = base64_decode_dash(enc_response["keyadd"]);
|
|
if (client_keyadd.empty())
|
|
{
|
|
Server->destroy(ret);
|
|
return NULL;
|
|
}
|
|
Server->Log("Encrypting with key " + base64_encode_dash(secret_session_key + server_keyadd + client_keyadd) + " (server)");
|
|
InternetServicePipe2* isc = new InternetServicePipe2(ret, secret_session_key + server_keyadd + client_keyadd);
|
|
isc->destroyBackendPipeOnDelete(true);
|
|
ret = isc;
|
|
}
|
|
|
|
if (compression == "zlib")
|
|
{
|
|
CompressedPipe2* comp = new CompressedPipe2(ret, compression_level);
|
|
comp->destroyBackendPipeOnDelete(true);
|
|
ret = comp;
|
|
}
|
|
#ifndef NO_ZSTD_COMPRESSION
|
|
else if (compression == "zstd")
|
|
{
|
|
CompressedPipeZstd* comp = new CompressedPipeZstd(ret, compression_level, -1);
|
|
comp->destroyBackendPipeOnDelete(true);
|
|
ret = comp;
|
|
}
|
|
#endif
|
|
else if(!compression.empty())
|
|
{
|
|
ServerLogger::Log(logid, "Unknown compression method requested by client \"" + compression + "\"", LL_ERROR);
|
|
Server->destroy(ret);
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
}
|
|
|
|
_u32 ClientMain::getClientFilesrvConnection(FileClient *fc, ServerSettings* server_settings, int timeoutms)
|
|
{
|
|
std::string curr_clientname = (clientname);
|
|
if(!clientsubname.empty())
|
|
{
|
|
curr_clientname = (clientmainname);
|
|
}
|
|
|
|
fc->setProgressLogCallback(this);
|
|
if(internet_connection)
|
|
{
|
|
IPipe *cp=InternetServiceConnector::getConnection(curr_clientname, SERVICE_FILESRV, timeoutms);
|
|
|
|
_u32 ret=fc->Connect(cp);
|
|
|
|
if(server_settings!=NULL)
|
|
{
|
|
int internet_speed=server_settings->getInternetSpeed();
|
|
if(internet_speed!=0
|
|
&& internet_speed!=-1)
|
|
{
|
|
fc->addThrottler(getThrottler(internet_speed));
|
|
}
|
|
int global_internet_speed=server_settings->getGlobalInternetSpeed();
|
|
if(global_internet_speed!=0
|
|
&& global_internet_speed!=-1)
|
|
{
|
|
fc->addThrottler(BackupServer::getGlobalInternetThrottler(global_internet_speed));
|
|
}
|
|
}
|
|
|
|
fc->setReconnectionTimeout(c_internet_fileclient_timeout);
|
|
|
|
return ret;
|
|
}
|
|
else
|
|
{
|
|
_u32 ret;
|
|
if (protocol_versions.filesrvtunnel > 0)
|
|
{
|
|
IPipe* pipe = new_fileclient_connection();
|
|
if (pipe == NULL)
|
|
return ERR_ERROR;
|
|
|
|
fc->setAddChecksum(true);
|
|
ret = fc->Connect(pipe);
|
|
}
|
|
else
|
|
{
|
|
ret = fc->Connect(getClientaddr());
|
|
}
|
|
|
|
if(server_settings!=NULL)
|
|
{
|
|
int local_speed=server_settings->getLocalSpeed();
|
|
if(local_speed!=0
|
|
&& local_speed!=-1)
|
|
{
|
|
fc->addThrottler(getThrottler(local_speed));
|
|
}
|
|
int global_local_speed=server_settings->getGlobalLocalSpeed();
|
|
if(global_local_speed!=0
|
|
&& global_local_speed!=-1)
|
|
{
|
|
fc->addThrottler(BackupServer::getGlobalLocalThrottler(global_local_speed));
|
|
}
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
}
|
|
|
|
bool ClientMain::getClientChunkedFilesrvConnection(std::unique_ptr<FileClientChunked>& fc_chunked,
|
|
ServerSettings* server_settings, FileClientChunked::NoFreeSpaceCallback* no_free_space_callback, int timeoutms)
|
|
{
|
|
if (no_free_space_callback == NULL)
|
|
no_free_space_callback = this;
|
|
|
|
std::string curr_clientname = (clientname);
|
|
if(!clientsubname.empty())
|
|
{
|
|
curr_clientname = (clientmainname);
|
|
}
|
|
|
|
std::string identity = getIdentity();
|
|
if(internet_connection)
|
|
{
|
|
IPipe *cp=InternetServiceConnector::getConnection(curr_clientname, SERVICE_FILESRV, timeoutms);
|
|
if(cp!=NULL)
|
|
{
|
|
fc_chunked.reset(new FileClientChunked(cp, false, &tcpstack, this, use_tmpfiles?NULL: no_free_space_callback, identity, NULL));
|
|
fc_chunked->setReconnectionTimeout(c_internet_fileclient_timeout);
|
|
}
|
|
else
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
IPipe* pipe;
|
|
if (protocol_versions.filesrvtunnel > 0)
|
|
{
|
|
pipe = new_fileclient_connection();
|
|
if (pipe == NULL)
|
|
return false;
|
|
}
|
|
else
|
|
{
|
|
pipe = Server->ConnectStream(getClientaddr().toString(), TCP_PORT, timeoutms);
|
|
if (pipe == NULL)
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
fc_chunked.reset(new FileClientChunked(pipe, false, protocol_versions.filesrvtunnel > 0 ? &tcpstack_checksum : &tcpstack,
|
|
this, use_tmpfiles ? NULL : no_free_space_callback, identity, NULL));
|
|
}
|
|
|
|
fc_chunked->setProgressLogCallback(this);
|
|
|
|
if(fc_chunked->getPipe()!=NULL && server_settings!=NULL)
|
|
{
|
|
int speed;
|
|
if(internet_connection)
|
|
{
|
|
speed=server_settings->getInternetSpeed();
|
|
}
|
|
else
|
|
{
|
|
speed=server_settings->getLocalSpeed();
|
|
}
|
|
if(speed!=0
|
|
&& speed!=-1)
|
|
{
|
|
fc_chunked->addThrottler(getThrottler(speed));
|
|
}
|
|
|
|
if(internet_connection)
|
|
{
|
|
int global_speed=server_settings->getGlobalInternetSpeed();
|
|
if(global_speed!=0
|
|
&& global_speed!=-1)
|
|
{
|
|
fc_chunked->addThrottler(BackupServer::getGlobalInternetThrottler(global_speed));
|
|
}
|
|
}
|
|
else
|
|
{
|
|
int global_speed=server_settings->getGlobalLocalSpeed();
|
|
if(global_speed!=0
|
|
&& global_speed!=-1)
|
|
{
|
|
fc_chunked->addThrottler(BackupServer::getGlobalLocalThrottler(global_speed));
|
|
}
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
IFsFile *ClientMain::getTemporaryFileRetry(bool use_tmpfiles, const std::string& tmpfile_path, logid_t logid)
|
|
{
|
|
int tries=50;
|
|
IFsFile *pfd=NULL;
|
|
while(pfd==NULL)
|
|
{
|
|
if(use_tmpfiles)
|
|
{
|
|
pfd=Server->openTemporaryFile();
|
|
}
|
|
else
|
|
{
|
|
size_t num;
|
|
{
|
|
IScopedLock lock(tmpfile_mutex);
|
|
num=tmpfile_num++;
|
|
}
|
|
std::string fn = tmpfile_path + os_file_sep() + convert(num);
|
|
Server->deleteFile(os_file_prefix(fn));
|
|
pfd=Server->openFile(os_file_prefix(fn), MODE_RW_CREATE);
|
|
}
|
|
|
|
if(pfd==NULL)
|
|
{
|
|
if(!os_directory_exists(os_file_prefix(tmpfile_path)))
|
|
{
|
|
ServerLogger::Log(logid, "Temporary file path did not exist. Creating it.", LL_INFO);
|
|
|
|
if(!os_create_dir_recursive(os_file_prefix(tmpfile_path)))
|
|
{
|
|
ServerLogger::Log(logid, "Error creating temporary file path. "+os_last_error_str(), LL_WARNING);
|
|
}
|
|
}
|
|
ServerLogger::Log(logid, "Error opening temporary file. Retrying...", LL_WARNING);
|
|
--tries;
|
|
if(tries<0)
|
|
{
|
|
return NULL;
|
|
}
|
|
Server->wait(1000);
|
|
}
|
|
}
|
|
return pfd;
|
|
}
|
|
|
|
void ClientMain::destroyTemporaryFile(IFile *tmp)
|
|
{
|
|
std::string fn=tmp->getFilename();
|
|
Server->destroy(tmp);
|
|
Server->deleteFile(fn);
|
|
}
|
|
|
|
IPipe * ClientMain::new_fileclient_connection(void)
|
|
{
|
|
std::string curr_clientname = (clientname);
|
|
if(!clientsubname.empty())
|
|
{
|
|
curr_clientname = (clientmainname);
|
|
}
|
|
|
|
IPipe *rp=NULL;
|
|
if(internet_connection)
|
|
{
|
|
rp=InternetServiceConnector::getConnection(curr_clientname, SERVICE_FILESRV, c_filesrv_connect_timeout);
|
|
}
|
|
else
|
|
{
|
|
if (protocol_versions.filesrvtunnel > 0)
|
|
{
|
|
rp = getClientCommandConnection(server_settings.get(), c_filesrv_connect_timeout);
|
|
if (rp == NULL)
|
|
{
|
|
return NULL;
|
|
}
|
|
|
|
CTCPStack tcpstack(internet_connection);
|
|
std::string identity = getIdentity();
|
|
|
|
std::string tosend = identity + "FILESRV";
|
|
|
|
if (tcpstack.Send(rp, tosend) != tosend.size())
|
|
{
|
|
Server->destroy(rp);
|
|
return NULL;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
rp = Server->ConnectStream(getClientaddr().toString(), TCP_PORT, c_filesrv_connect_timeout);
|
|
}
|
|
}
|
|
return rp;
|
|
}
|
|
|
|
bool ClientMain::handle_not_enough_space(const std::string &path, logid_t logid)
|
|
{
|
|
int64 free_space=-1;
|
|
if(!path.empty())
|
|
{
|
|
free_space = os_free_space(os_file_prefix(path));
|
|
if(free_space==-1)
|
|
{
|
|
free_space = os_free_space(os_file_prefix(ExtractFilePath(path)));
|
|
}
|
|
}
|
|
if(free_space==-1)
|
|
{
|
|
free_space=os_free_space(os_file_prefix(server_settings->getSettings()->backupfolder));
|
|
}
|
|
if(free_space!=-1 && free_space<minfreespace_min)
|
|
{
|
|
Server->Log("No free space in backup folder. Free space="+PrettyPrintBytes(free_space)+" MinFreeSpace="+PrettyPrintBytes(minfreespace_min), LL_WARNING);
|
|
|
|
if(!ServerCleanupThread::cleanupSpace(minfreespace_min) )
|
|
{
|
|
ServerLogger::Log(logid, "FATAL: Could not free space. NOT ENOUGH FREE SPACE.", LL_ERROR);
|
|
sendMailToAdmins("Fatal error occurred during backup", ServerLogger::getWarningLevelTextLogdata(logid));
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool ClientMain::handle_not_enough_space(const std::string & path)
|
|
{
|
|
return handle_not_enough_space(path, logid);
|
|
}
|
|
|
|
unsigned int ClientMain::exponentialBackoffTime( size_t count, unsigned int sleeptime, unsigned div )
|
|
{
|
|
return static_cast<unsigned int>((std::max)(static_cast<double>(sleeptime), static_cast<double>(sleeptime)*pow(static_cast<double>(div), static_cast<double>(count))));
|
|
}
|
|
|
|
bool ClientMain::exponentialBackoff(size_t count, int64 lasttime, unsigned int sleeptime, unsigned div)
|
|
{
|
|
if(count>0)
|
|
{
|
|
int64 passed_time=Server->getTimeMS()-lasttime;
|
|
int64 sleeptime_exp = 1LL*exponentialBackoffTime(count, sleeptime, div)*1000;
|
|
|
|
return passed_time>=sleeptime_exp;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
|
|
unsigned int ClientMain::exponentialBackoffTimeImage()
|
|
{
|
|
return exponentialBackoffTime(count_image_backup_try, c_sleeptime_failed_imagebackup, c_exponential_backoff_div);
|
|
}
|
|
|
|
unsigned int ClientMain::exponentialBackoffTimeFile()
|
|
{
|
|
return exponentialBackoffTime(count_file_backup_try, c_sleeptime_failed_filebackup, c_exponential_backoff_div);
|
|
}
|
|
|
|
bool ClientMain::exponentialBackoffImage()
|
|
{
|
|
return exponentialBackoff(count_image_backup_try, last_image_backup_try, c_sleeptime_failed_imagebackup, c_exponential_backoff_div);
|
|
}
|
|
|
|
bool ClientMain::exponentialBackoffFile()
|
|
{
|
|
return exponentialBackoff(count_file_backup_try, last_file_backup_try, c_sleeptime_failed_filebackup, c_exponential_backoff_div);
|
|
}
|
|
|
|
bool ClientMain::exponentialBackoffCdp()
|
|
{
|
|
return exponentialBackoff(count_cdp_backup_try, last_cdp_backup_try, c_sleeptime_failed_filebackup, c_exponential_backoff_div);
|
|
}
|
|
|
|
bool ClientMain::pauseRetryBackup()
|
|
{
|
|
return Server->getTimeMS() - last_backup_try >= 5 * 60 * 1000;
|
|
}
|
|
|
|
bool ClientMain::sendServerIdentity(bool retry_exit)
|
|
{
|
|
{
|
|
IScopedLock lock(clientaddr_mutex);
|
|
session_identity.clear();
|
|
}
|
|
bool c = true;
|
|
while (c)
|
|
{
|
|
ServerStatus::setStatusError(clientname, se_authenticating);
|
|
|
|
c = false;
|
|
bool retok_err = false;
|
|
std::string ret_str;
|
|
bool b = sendClientMessage("ADD IDENTITY", "OK", "Sending Identity to client \"" + clientname + "\" failed. Retrying soon...",
|
|
10000, false, LL_DEBUG, &retok_err, &ret_str, NULL, false);
|
|
if (!b)
|
|
{
|
|
if (retok_err)
|
|
{
|
|
if (ret_str != "needs certificate")
|
|
{
|
|
ServerStatus::setStatusError(clientname, se_ident_error);
|
|
}
|
|
else
|
|
{
|
|
ServerStatus::setStatusError(clientname, se_authenticating);
|
|
needs_authentification = true;
|
|
return true;
|
|
}
|
|
}
|
|
|
|
if (!retry_exit)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
unsigned int retry_time = ident_err_retry_time;
|
|
|
|
if (retok_err)
|
|
{
|
|
retry_time = ident_err_retry_time_retok;
|
|
}
|
|
|
|
if (retok_err
|
|
&& !ret_str.empty())
|
|
{
|
|
Server->Log("Sending Identity to client \"" + clientname + "\" failed. Retrying in "+PrettyPrintTime(retry_time)+"... (returned \"" + ret_str + "\")", LL_INFO);
|
|
}
|
|
else
|
|
{
|
|
Server->Log("Sending Identity to client \"" + clientname + "\" failed. Retrying in " + PrettyPrintTime(retry_time) + "...", LL_INFO);
|
|
}
|
|
|
|
c = true;
|
|
std::string msg;
|
|
pipe->Read(&msg, retry_time);
|
|
if (msg == "exit" || msg == "exitnow")
|
|
{
|
|
pipe->Write(msg);
|
|
return false;
|
|
}
|
|
else if (next(msg, 0, "address"))
|
|
{
|
|
updateClientAddress(msg.substr(7));
|
|
|
|
tcpstack.setAddChecksum(internet_connection);
|
|
}
|
|
else
|
|
{
|
|
pipe->Write(msg);
|
|
Server->wait(retry_time);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
ServerStatus::setStatusError(clientname, se_none);
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool ClientMain::authenticatePubKey()
|
|
{
|
|
if (crypto_fak == NULL)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
IECDHKeyExchange* ecdh_key_exchange = NULL;
|
|
int64 ecdh_key_exchange_age;
|
|
{
|
|
IScopedLock lock(ecdh_key_exchange_mutex);
|
|
if (!ecdh_key_exchange_buffer.empty())
|
|
{
|
|
ecdh_key_exchange = ecdh_key_exchange_buffer[ecdh_key_exchange_buffer.size() - 1].first;
|
|
ecdh_key_exchange_age = ecdh_key_exchange_buffer[ecdh_key_exchange_buffer.size() - 1].second;
|
|
ecdh_key_exchange_buffer.pop_back();
|
|
}
|
|
}
|
|
if (ecdh_key_exchange == NULL)
|
|
{
|
|
ecdh_key_exchange = crypto_fak->createECDHKeyExchange();
|
|
ecdh_key_exchange_age = Server->getTimeMS();
|
|
}
|
|
|
|
bool ret = authenticatePubKeyInt(ecdh_key_exchange);
|
|
|
|
if (ecdh_key_exchange != NULL
|
|
&& Server->getTimeMS() - ecdh_key_exchange_age < max_ecdh_key_age)
|
|
{
|
|
IScopedLock lock(ecdh_key_exchange_mutex);
|
|
ecdh_key_exchange_buffer.push_back(std::make_pair(ecdh_key_exchange, ecdh_key_exchange_age));
|
|
}
|
|
else
|
|
{
|
|
Server->destroy(ecdh_key_exchange);
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
bool ClientMain::authenticatePubKeyInt(IECDHKeyExchange* ecdh_key_exchange)
|
|
{
|
|
std::string params;
|
|
|
|
if (!internet_connection)
|
|
{
|
|
params = " with_enc=1";
|
|
}
|
|
|
|
if (!clientsubname.empty())
|
|
{
|
|
params = "&clientsubname=" + EscapeParamString(clientsubname);
|
|
params[0] = ' ';
|
|
}
|
|
|
|
std::string challenge = sendClientMessageRetry("GET CHALLENGE" + params, "Failed to get challenge from client", 10000, 10,
|
|
false, LL_INFO, 0, false);
|
|
|
|
if (challenge == "ERR"
|
|
&& clientsubname.empty())
|
|
{
|
|
challenge = sendClientMessageRetry("GET CHALLENGE", "Failed to get challenge from client", 10000, 10, false, LL_INFO);
|
|
}
|
|
|
|
if(challenge=="ERR")
|
|
{
|
|
return false;
|
|
}
|
|
|
|
if(!challenge.empty())
|
|
{
|
|
str_map challenge_params;
|
|
if (challenge.find("?") != std::string::npos)
|
|
{
|
|
ParseParamStrHttp(getafter("?", challenge), &challenge_params);
|
|
challenge = getuntil("?", challenge);
|
|
}
|
|
|
|
std::string signature;
|
|
std::string signature_ecdsa409k1;
|
|
std::string privkey = getFile("urbackup/server_ident.priv");
|
|
|
|
if(privkey.empty())
|
|
{
|
|
Server->Log("Cannot read private key urbackup/server_ident.priv", LL_ERROR);
|
|
return false;
|
|
}
|
|
|
|
bool rc = crypto_fak->signDataDSA(privkey, challenge, signature);
|
|
|
|
if(!rc)
|
|
{
|
|
Server->Log("Signing challenge failed", LL_ERROR);
|
|
return false;
|
|
}
|
|
|
|
std::string privkey_ecdsa409k1 = getFile("urbackup/server_ident_ecdsa409k1.priv");
|
|
|
|
if (privkey_ecdsa409k1.empty())
|
|
{
|
|
Server->Log("Cannot read private key urbackup/server_ident_ecdsa409k1.priv", LL_ERROR);
|
|
return false;
|
|
}
|
|
|
|
rc = crypto_fak->signData(privkey_ecdsa409k1, challenge, signature_ecdsa409k1);
|
|
|
|
if(!rc)
|
|
{
|
|
Server->Log("Signing challenge failed -2", LL_ERROR);
|
|
return false;
|
|
}
|
|
|
|
std::string pubkey = getFile("urbackup/server_ident.pub");
|
|
|
|
if(pubkey.empty())
|
|
{
|
|
Server->Log("Reading public key from urbackup/server_ident.pub failed", LL_ERROR);
|
|
return false;
|
|
}
|
|
|
|
std::string pubkey_ecdsa = getFile("urbackup/server_ident_ecdsa409k1.pub");
|
|
|
|
if (pubkey_ecdsa.empty())
|
|
{
|
|
Server->Log("Reading public key from urbackup/server_ident_ecdsa409k1.pub failed", LL_ERROR);
|
|
return false;
|
|
}
|
|
|
|
std::string identity = ServerSettings::generateRandomAuthKey(20);
|
|
|
|
std::string client_ecdh_pubkey = base64_decode_dash(challenge_params["pubkey_ecdh233k1"]);
|
|
|
|
std::string session_key;
|
|
std::string l_secret_session_key;
|
|
if (!client_ecdh_pubkey.empty())
|
|
{
|
|
l_secret_session_key = ServerSettings::generateRandomBinaryKey();
|
|
|
|
std::string ecdh_shared_key = ecdh_key_exchange->getSharedKey(client_ecdh_pubkey);
|
|
if (ecdh_shared_key.empty())
|
|
{
|
|
Server->Log("Getting ECDH shared key failed", LL_ERROR);
|
|
return false;
|
|
}
|
|
|
|
session_key = "&secret_session_key="+ base64_encode_dash(crypto_fak->encryptAuthenticatedAES(l_secret_session_key,
|
|
ecdh_shared_key, 1));
|
|
|
|
std::string ecdh_pubkey = ecdh_key_exchange->getPublicKey();
|
|
session_key += "&pubkey_ecdh233k1=" + base64_encode_dash(ecdh_pubkey);
|
|
|
|
std::string signature_ecdh233k1;
|
|
rc = crypto_fak->signData(privkey_ecdsa409k1, challenge + ecdh_pubkey, signature_ecdh233k1);
|
|
|
|
if (!rc)
|
|
{
|
|
Server->Log("Signing ecdh pubkey failed -2", LL_ERROR);
|
|
return false;
|
|
}
|
|
|
|
session_key += "&signature_ecdh233k1=" + base64_encode_dash(signature_ecdh233k1);
|
|
}
|
|
|
|
std::string l_session_compressed = challenge_params["compress"];
|
|
int l_session_compression_level = server_settings->getSettings()->internet_compression_level;
|
|
|
|
#ifdef NO_ZSTD_COMPRESSION
|
|
if (l_session_compressed == "zstd")
|
|
{
|
|
l_session_compressed = "zlib";
|
|
}
|
|
#endif
|
|
|
|
bool ret = sendClientMessageRetry("SIGNATURE#pubkey="+base64_encode_dash(pubkey)+
|
|
"&pubkey_ecdsa409k1="+base64_encode_dash(pubkey_ecdsa)+
|
|
"&signature="+base64_encode_dash(signature)+
|
|
"&signature_ecdsa409k1="+base64_encode_dash(signature_ecdsa409k1)+
|
|
"&session_identity="+identity +
|
|
session_key +
|
|
(clientsubname.empty() ? "" : "&clientsubname="+ EscapeParamString(clientsubname)), "ok", "Error sending server signature to client",
|
|
10000, 10, true, LL_ERROR, NULL, NULL, false);
|
|
|
|
if(ret)
|
|
{
|
|
IScopedLock lock(clientaddr_mutex);
|
|
session_identity = "#I"+identity+"#";
|
|
session_identity_refreshtime = Server->getTimeMS();
|
|
secret_session_key = l_secret_session_key;
|
|
session_compressed = l_session_compressed;
|
|
session_compression_level = l_session_compression_level;
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
else
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
|
|
void ClientMain::timeoutRestores()
|
|
{
|
|
IScopedLock lock(restore_mutex.get());
|
|
|
|
for (size_t i = 0; i < running_restores.size();)
|
|
{
|
|
if (Server->getTimeMS() - running_restores[i].last_active > 5 * 60 * 1000) //5min
|
|
{
|
|
ServerLogger::Log(running_restores[i].log_id, "Restore was inactive for 5min. Timeout. Stopping restore...", LL_ERROR);
|
|
|
|
finishFailedRestore(running_restores[i].restore_identity, running_restores[i].log_id, running_restores[i].status_id, running_restores[i].restore_id);
|
|
|
|
running_restores.erase(running_restores.begin() + i);
|
|
}
|
|
else
|
|
{
|
|
++i;
|
|
}
|
|
}
|
|
}
|
|
|
|
void ClientMain::timeoutLocalBackups()
|
|
{
|
|
std::lock_guard<std::mutex> lock(local_backup_mutex);
|
|
|
|
for (size_t i = 0; i < running_local_backups.size();)
|
|
{
|
|
if (Server->getTimeMS() - running_local_backups[i].last_active > 5 * 60 * 1000) //5min
|
|
{
|
|
LocalBackup* lb = nullptr;
|
|
for (SRunningBackup& rb : backup_queue)
|
|
{
|
|
if (rb.backup->getStatusId() == running_local_backups[i].status_id)
|
|
lb = dynamic_cast<LocalBackup*>(rb.backup);
|
|
if (lb != nullptr)
|
|
break;
|
|
}
|
|
|
|
bool finished;
|
|
if (!lb->queryBackupFinished(60000, finished))
|
|
{
|
|
ServerLogger::Log(running_local_backups[i].log_id, "Local backup was inactive for 5min. Timeout. Stopping local backup...", LL_ERROR);
|
|
|
|
finishLocalBackup(false, running_local_backups[i].log_id, running_local_backups[i].status_id, running_local_backups[i].backupid);
|
|
|
|
running_local_backups.erase(running_local_backups.begin() + i);
|
|
}
|
|
else
|
|
{
|
|
running_local_backups[i].last_active = Server->getTimeMS();
|
|
|
|
if (finished)
|
|
{
|
|
finishLocalBackup(lb->getResult(), running_local_backups[i].log_id, running_local_backups[i].status_id,
|
|
running_local_backups[i].backupid);
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
++i;
|
|
}
|
|
}
|
|
}
|
|
|
|
std::string ClientMain::getIdentity()
|
|
{
|
|
IScopedLock lock(clientaddr_mutex);
|
|
return session_identity.empty() ? server_identity : session_identity;
|
|
}
|
|
|
|
std::string ClientMain::getSecretSessionKey()
|
|
{
|
|
IScopedLock lock(clientaddr_mutex);
|
|
return secret_session_key;
|
|
}
|
|
|
|
std::string ClientMain::getSessionCompression(int& level)
|
|
{
|
|
IScopedLock lock(clientaddr_mutex);
|
|
level = session_compression_level;
|
|
return session_compressed;
|
|
}
|
|
|
|
bool ClientMain::run_script( std::string name, const std::string& params, logid_t logid)
|
|
{
|
|
#ifdef _WIN32
|
|
name = name + ".bat";
|
|
#endif
|
|
|
|
if(!FileExists(name))
|
|
{
|
|
ServerLogger::Log(logid, "Script does not exist "+name, LL_DEBUG);
|
|
return true;
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
//I ... waaa ... even ... :(
|
|
std::string quoted_script_name = greplace(" ", "\" \"", name);
|
|
#else
|
|
std::string quoted_script_name = "\"" + greplace("\"", "\\\"", name) + "\"";
|
|
#endif
|
|
|
|
quoted_script_name +=" "+params;
|
|
|
|
quoted_script_name +=" 2>&1";
|
|
|
|
#ifdef _WIN32
|
|
FILE* fp = _wpopen(Server->ConvertToWchar(quoted_script_name).c_str(), L"rb");
|
|
#else
|
|
#ifdef __linux__
|
|
FILE* fp = popen(quoted_script_name.c_str(), "re");
|
|
if(!fp) fp = popen(quoted_script_name.c_str(), "r");
|
|
#else
|
|
FILE* fp = popen(quoted_script_name.c_str(), "r");
|
|
#endif
|
|
#endif
|
|
|
|
if(!fp)
|
|
{
|
|
ServerLogger::Log(logid, "Could not open pipe for command "+name, LL_DEBUG);
|
|
return false;
|
|
}
|
|
|
|
std::string output;
|
|
while(!feof(fp) && !ferror(fp))
|
|
{
|
|
char buf[4097];
|
|
size_t r = fread(buf, 1, 4096, fp);
|
|
buf[r]=0;
|
|
output+=buf;
|
|
}
|
|
|
|
#ifdef _WIN32
|
|
int rc = _pclose(fp);
|
|
#else
|
|
int rc = pclose(fp);
|
|
#endif
|
|
|
|
if(rc!=0)
|
|
{
|
|
ServerLogger::Log(logid, "Script "+name+" had error (code "+convert(rc)+")", LL_ERROR);
|
|
}
|
|
|
|
std::vector<std::string> toks;
|
|
Tokenize(output, toks, "\n");
|
|
|
|
for(size_t i=0;i<toks.size();++i)
|
|
{
|
|
ServerLogger::Log(logid, "Script output Line("+convert(i+1)+"): " + toks[i], rc!=0?LL_ERROR:LL_INFO);
|
|
}
|
|
|
|
return rc == 0;
|
|
}
|
|
|
|
void ClientMain::log_progress( const std::string& fn, int64 total, int64 downloaded, int64 speed_bps )
|
|
{
|
|
std::string fn_wo_token = fn;
|
|
std::string share = getuntil("/", fn);
|
|
if (!share.empty())
|
|
{
|
|
if (share.find("|") != std::string::npos)
|
|
{
|
|
fn_wo_token = getafter("|", fn);
|
|
}
|
|
}
|
|
|
|
if(total>0 && total!=LLONG_MAX)
|
|
{
|
|
int pc_complete = 0;
|
|
if(total>0)
|
|
{
|
|
pc_complete = static_cast<int>((static_cast<float>(downloaded)/total)*100.f);
|
|
}
|
|
ServerLogger::Log(logid, "Loading \""+ fn_wo_token +"\". "+convert(pc_complete)+"% finished "+PrettyPrintBytes(downloaded)+"/"+PrettyPrintBytes(total)+" at "+PrettyPrintSpeed(static_cast<size_t>(speed_bps)), LL_DEBUG);
|
|
}
|
|
else
|
|
{
|
|
ServerLogger::Log(logid, "Loading \""+ fn_wo_token +"\". Loaded "+PrettyPrintBytes(downloaded)+" at "+PrettyPrintSpeed(static_cast<size_t>(speed_bps)), LL_DEBUG);
|
|
}
|
|
}
|
|
|
|
|
|
|
|
void ClientMain::updateClientAddress(const std::string& address_data)
|
|
{
|
|
IScopedLock lock(clientaddr_mutex);
|
|
assert(address_data.size() == sizeof(clientaddr) + 1);
|
|
memcpy(&clientaddr, address_data.data(), sizeof(clientaddr));
|
|
bool prev_internet_connection = internet_connection;
|
|
internet_connection = (address_data[sizeof(clientaddr)] == 0) ? false : true;
|
|
|
|
if (prev_internet_connection != internet_connection)
|
|
{
|
|
IScopedLock lock(throttle_mutex);
|
|
|
|
if (client_throttler != NULL)
|
|
{
|
|
client_throttler->changeThrottleUpdater(new
|
|
ThrottleUpdater(clientid, internet_connection ?
|
|
ThrottleScope_Internet : ThrottleScope_Local));
|
|
}
|
|
}
|
|
}
|
|
|
|
bool ClientMain::createDirectoryForClient()
|
|
{
|
|
std::string backupfolder=server_settings->getSettings()->backupfolder;
|
|
if(!os_create_dir(os_file_prefix(backupfolder+os_file_sep()+clientname)) && !os_directory_exists(os_file_prefix(backupfolder+os_file_sep()+clientname)) )
|
|
{
|
|
Server->Log("Could not create or read directory for client \""+clientname+"\". "+os_last_error_str(), LL_ERROR);
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
bool ClientMain::isRunningImageBackup(const std::string& letter)
|
|
{
|
|
for(size_t i=0;i<backup_queue.size();++i)
|
|
{
|
|
if(!backup_queue[i].backup->isFileBackup()
|
|
&& backup_queue[i].letter== normalizeVolumeUpper(letter) )
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
std::string ClientMain::normalizeVolumeUpper(std::string volume)
|
|
{
|
|
if (volume.size() == 1)
|
|
{
|
|
strupper(&volume);
|
|
return volume + ":";
|
|
}
|
|
|
|
if (volume.size() == 2
|
|
&& volume[1] == ':')
|
|
{
|
|
strupper(&volume);
|
|
return volume;
|
|
}
|
|
|
|
if (volume.size() == 3
|
|
&& volume[1] == ':'
|
|
&& (volume[2] == '\\' || volume[2] == '/'))
|
|
{
|
|
strupper(&volume);
|
|
return volume.substr(0, 2);
|
|
}
|
|
|
|
return volume;
|
|
}
|
|
|
|
bool ClientMain::isImageGroupQueued(const std::string & letter, bool full)
|
|
{
|
|
std::string image_snapshot_groups = server_settings->getSettings()->image_snapshot_groups;
|
|
std::vector<std::string> groups;
|
|
Tokenize(image_snapshot_groups, groups, "|");
|
|
|
|
image_snapshot_groups = strlower(image_snapshot_groups);
|
|
|
|
std::string vol = normalizeVolumeUpper(letter);
|
|
|
|
for (size_t i = 0; i < groups.size(); ++i)
|
|
{
|
|
std::vector<std::string> vols;
|
|
Tokenize(groups[i], vols, ";,");
|
|
|
|
for (size_t j = 0; j < vols.size(); ++j)
|
|
{
|
|
vols[j] = normalizeVolumeUpper(vols[j]);
|
|
}
|
|
|
|
if (image_snapshot_groups=="all"
|
|
|| image_snapshot_groups == "all_nonusb"
|
|
|| std::find(vols.begin(), vols.end(), vol) != vols.end())
|
|
{
|
|
for (size_t k = 0; k < backup_queue.size(); ++k)
|
|
{
|
|
if ( (!backup_queue[k].backup->isIncrementalBackup()) != full
|
|
|| backup_queue[k].backup->isFileBackup())
|
|
{
|
|
continue;
|
|
}
|
|
|
|
if (image_snapshot_groups == "all"
|
|
|| image_snapshot_groups == "all_nonusb"
|
|
|| std::find(vols.begin(), vols.end(), normalizeVolumeUpper(backup_queue[k].letter))
|
|
!= vols.end())
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
bool ClientMain::isRunningFileBackup(int group, bool queue_only)
|
|
{
|
|
for(size_t i=0;i<backup_queue.size();++i)
|
|
{
|
|
if(backup_queue[i].backup->isFileBackup() && backup_queue[i].group==group)
|
|
{
|
|
if (queue_only)
|
|
{
|
|
return true;
|
|
}
|
|
else if (backup_queue[i].ticket != ILLEGAL_THREADPOOL_TICKET)
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
void ClientMain::addContinuousChanges( const std::string& changes )
|
|
{
|
|
IScopedLock lock(continuous_mutex);
|
|
if(continuous_backup!=NULL)
|
|
{
|
|
continuous_backup->addChanges(changes);
|
|
}
|
|
}
|
|
|
|
void ClientMain::setContinuousBackup( BackupServerContinuous* cb )
|
|
{
|
|
IScopedLock lock(continuous_mutex);
|
|
|
|
if(continuous_backup!=NULL)
|
|
{
|
|
continuous_backup->doStop();
|
|
}
|
|
|
|
continuous_backup = cb;
|
|
}
|
|
|
|
void ClientMain::addShareToCleanup( int clientid, const SShareCleanup& cleanupData )
|
|
{
|
|
if(fileserv==NULL) return;
|
|
|
|
IScopedLock lock(cleanup_mutex);
|
|
std::vector<SShareCleanup>& tocleanup = cleanup_shares[clientid];
|
|
tocleanup.push_back(cleanupData);
|
|
}
|
|
|
|
void ClientMain::cleanupRestoreShare(int clientid, std::string restore_identity)
|
|
{
|
|
IScopedLock lock(cleanup_mutex);
|
|
|
|
std::vector<SShareCleanup>& tocleanup = cleanup_shares[clientid];
|
|
|
|
for (size_t i = 0; i < tocleanup.size();)
|
|
{
|
|
if (tocleanup[i].identity == restore_identity)
|
|
{
|
|
cleanupShare(tocleanup[i]);
|
|
tocleanup.erase(tocleanup.begin() + i);
|
|
}
|
|
else
|
|
{
|
|
++i;
|
|
}
|
|
}
|
|
}
|
|
|
|
void ClientMain::updateLocalBackup(int64 status_id, bool success, bool should_backoff)
|
|
{
|
|
for (SRunningBackup& rb : backup_queue)
|
|
{
|
|
if (rb.backup->getStatusId() == status_id)
|
|
{
|
|
rb.backup->setResult(success, should_backoff);
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
bool ClientMain::removeRestore(int64 restore_id)
|
|
{
|
|
IScopedLock lock(restore_mutex.get());
|
|
|
|
for (size_t i = 0; i < running_restores.size(); ++i)
|
|
{
|
|
if (running_restores[i].restore_id == restore_id)
|
|
{
|
|
running_restores.erase(running_restores.begin() + i);
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
bool ClientMain::removeLocalBackup(int64 status_id)
|
|
{
|
|
std::lock_guard<std::mutex> lock(local_backup_mutex);
|
|
|
|
for (size_t i = 0; i < running_local_backups.size(); ++i)
|
|
{
|
|
if (running_local_backups[i].status_id == status_id)
|
|
{
|
|
running_local_backups.erase(running_local_backups.begin() + i);
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
bool ClientMain::updateRestoreRunning(int64 restore_id)
|
|
{
|
|
IScopedLock lock(restore_mutex.get());
|
|
|
|
for (size_t i = 0; i < running_restores.size(); ++i)
|
|
{
|
|
if (running_restores[i].restore_id == restore_id)
|
|
{
|
|
running_restores[i].last_active = Server->getTimeMS();
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
bool ClientMain::updateLocalBackupRunning(int backupid)
|
|
{
|
|
std::lock_guard<std::mutex> lock(local_backup_mutex);
|
|
|
|
for (size_t i = 0; i < running_local_backups.size(); ++i)
|
|
{
|
|
if (running_local_backups[i].backupid == backupid)
|
|
{
|
|
running_local_backups[i].last_active = Server->getTimeMS();
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
void ClientMain::cleanupShares()
|
|
{
|
|
if(fileserv!=NULL)
|
|
{
|
|
IScopedLock lock(cleanup_mutex);
|
|
std::vector<SShareCleanup>& tocleanup = cleanup_shares[clientid];
|
|
|
|
for(size_t i=0;i<tocleanup.size();++i)
|
|
{
|
|
cleanupShare(tocleanup[i]);
|
|
}
|
|
|
|
tocleanup.clear();
|
|
}
|
|
}
|
|
|
|
void ClientMain::cleanupShare(SShareCleanup & tocleanup)
|
|
{
|
|
if (tocleanup.cleanup_file)
|
|
{
|
|
std::string cleanupfile = fileserv->getShareDir(tocleanup.name, tocleanup.identity);
|
|
if (!cleanupfile.empty())
|
|
{
|
|
Server->deleteFile(cleanupfile);
|
|
}
|
|
}
|
|
|
|
if (tocleanup.remove_callback)
|
|
{
|
|
fileserv->removeMetadataCallback(tocleanup.name, tocleanup.identity);
|
|
}
|
|
|
|
fileserv->removeDir(tocleanup.name, tocleanup.identity);
|
|
|
|
if (!tocleanup.identity.empty())
|
|
{
|
|
fileserv->removeIdentity(tocleanup.identity);
|
|
}
|
|
}
|
|
|
|
void ClientMain::finishFailedRestore(std::string restore_identity, logid_t log_id, int64 status_id, int64 restore_id)
|
|
{
|
|
ServerStatus::stopProcess(clientname, status_id);
|
|
|
|
int errors = 0;
|
|
int warnings = 0;
|
|
int infos = 0;
|
|
std::string logdata = ServerLogger::getLogdata(log_id, errors, warnings, infos);
|
|
|
|
backup_dao->saveBackupLog(clientid, errors, warnings, infos, 0,
|
|
0, 0, 1);
|
|
|
|
backup_dao->saveBackupLogData(db->getLastInsertID(), logdata);
|
|
|
|
backup_dao->setRestoreDone(0, restore_id);
|
|
|
|
IScopedLock lock(cleanup_mutex);
|
|
|
|
std::vector<SShareCleanup>& tocleanup = cleanup_shares[clientid];
|
|
|
|
for (size_t i = 0; i < tocleanup.size();)
|
|
{
|
|
if (tocleanup[i].identity == restore_identity)
|
|
{
|
|
cleanupShare(tocleanup[i]);
|
|
tocleanup.erase(tocleanup.begin() + i);
|
|
}
|
|
else
|
|
{
|
|
++i;
|
|
}
|
|
}
|
|
|
|
ServerLogger::reset(log_id);
|
|
}
|
|
|
|
void ClientMain::finishLocalBackup(bool success, logid_t log_id, int64 status_id, int backupid)
|
|
{
|
|
SProcess proc = ServerStatus::getProcess(clientname, status_id);
|
|
|
|
if (proc.id == 0)
|
|
return;
|
|
|
|
ServerStatus::stopProcess(clientname, status_id);
|
|
|
|
int errors = 0;
|
|
int warnings = 0;
|
|
int infos = 0;
|
|
std::string logdata = ServerLogger::getLogdata(log_id, errors, warnings, infos);
|
|
|
|
int num_issues = 0;
|
|
bool image = proc.action == sa_full_image || proc.action == sa_incr_image;
|
|
bool incr = proc.action == sa_incr_image || proc.action == sa_incr_file || proc.action == sa_resume_incr_file;
|
|
bool resumed = proc.action == sa_resume_incr_file || proc.action == sa_resume_full_file;
|
|
|
|
backup_dao->saveBackupLog(clientid, errors, warnings, infos, image,
|
|
incr, resumed, 0);
|
|
|
|
backup_dao->saveBackupLogData(db->getLastInsertID(), logdata);
|
|
|
|
if (!success)
|
|
{
|
|
ServerCleanupDao cleanupdao(db);
|
|
cleanupdao.removeFileBackup(backupid);
|
|
}
|
|
else
|
|
{
|
|
ServerBackupDao backup_dao(db);
|
|
backup_dao.updateClientLastFileBackup(backupid, num_issues, clientid);
|
|
backup_dao.updateFileBackupSetComplete(backupid);
|
|
}
|
|
|
|
ServerLogger::reset(log_id);
|
|
}
|
|
|
|
bool ClientMain::isLocalBackupRunning(int64 status_id)
|
|
{
|
|
std::lock_guard<std::mutex> lock(local_backup_mutex);
|
|
|
|
for (size_t i = 0; i < running_local_backups.size();)
|
|
{
|
|
if (running_local_backups[i].status_id == status_id)
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
bool ClientMain::isBackupFinished(const SRunningBackup& rb)
|
|
{
|
|
if (rb.running)
|
|
{
|
|
LocalBackup* lb = dynamic_cast<LocalBackup*>(rb.backup);
|
|
if (lb != nullptr)
|
|
return !isLocalBackupRunning(lb->getStatusId());
|
|
}
|
|
else if (rb.ticket != ILLEGAL_THREADPOOL_TICKET)
|
|
{
|
|
return Server->getThreadPool()->waitFor(rb.ticket, 0);
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
void ClientMain::updateVirtualClients()
|
|
{
|
|
std::string virtual_clients = server_settings->getVirtualClients();
|
|
{
|
|
IScopedLock lock(clientaddr_mutex);
|
|
|
|
allow_restore_clients.clear();
|
|
allow_restore_clients.push_back(clientname);
|
|
if (!virtual_clients.empty())
|
|
{
|
|
std::vector<std::string> toks;
|
|
Tokenize(virtual_clients, toks, "|");
|
|
for (size_t i = 0; i < toks.size(); ++i)
|
|
{
|
|
allow_restore_clients.push_back(clientname + "[" + toks[i] + "]");
|
|
}
|
|
}
|
|
}
|
|
|
|
BackupServer::setVirtualClients(clientname, virtual_clients);
|
|
}
|
|
|
|
bool ClientMain::checkClientName(bool& continue_start_backups)
|
|
{
|
|
if (internet_connection
|
|
|| !clientsubname.empty())
|
|
return true;
|
|
|
|
std::string msg = sendClientMessageRetry("GET CLIENTNAME", "Error getting name of client", 10000, 10, false);
|
|
|
|
if (msg == "ERR")
|
|
{
|
|
return true;
|
|
}
|
|
else if (msg.empty())
|
|
{
|
|
continue_start_backups = false;
|
|
return true;
|
|
}
|
|
|
|
str_map msg_params;
|
|
ParseParamStrHttp(msg, &msg_params);
|
|
|
|
std::string curr_clientname = conv_filename(msg_params["name"]);
|
|
|
|
if (curr_clientname == clientname)
|
|
{
|
|
return true;
|
|
}
|
|
else
|
|
{
|
|
ServerLogger::Log(logid, "Client name check failed. Expected name is \"" + clientname + "\" got \"" + msg_params["name"] + "\"", LL_WARNING);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
bool ClientMain::renameClient(const std::string & clientuid)
|
|
{
|
|
std::vector<int> uids = backup_dao->getClientsByUid(clientuid);
|
|
|
|
if (std::find(uids.begin(), uids.end(), clientid)
|
|
== uids.end())
|
|
{
|
|
backup_dao->updateClientUid(clientuid, clientid);
|
|
}
|
|
|
|
if (uids.empty())
|
|
return false;
|
|
|
|
int rename_from = -1;
|
|
for (size_t i = 0; i < uids.size(); ++i)
|
|
{
|
|
if (uids[i] != clientid)
|
|
{
|
|
rename_from = static_cast<int>(uids[i]);
|
|
}
|
|
}
|
|
|
|
if (rename_from <= 0)
|
|
return false;
|
|
|
|
ServerBackupDao::SClientName old_name = backup_dao->getVirtualMainClientname(rename_from);
|
|
|
|
if (!old_name.exists)
|
|
return false;
|
|
|
|
if (old_name.virtualmain == clientname)
|
|
return false;
|
|
|
|
std::string backupfolder = server_settings->getSettings()->backupfolder;
|
|
std::vector<std::string> old_backupfolders = backup_dao->getOldBackupfolders();
|
|
|
|
if (os_directory_exists(backupfolder + os_file_sep() + old_name.name + os_file_sep() + ".directory_pool"))
|
|
{
|
|
return false;
|
|
}
|
|
|
|
for (size_t i = 0; i < old_backupfolders.size(); ++i)
|
|
{
|
|
if (os_directory_exists(old_backupfolders[i] + os_file_sep() + old_name.name + os_file_sep() + ".directory_pool"))
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
|
|
if (ServerStatus::getStatus(old_name.name).r_online)
|
|
{
|
|
//retry later once the old client is offline
|
|
return true;
|
|
}
|
|
|
|
DBScopedSynchronous synchronous(db);
|
|
DBScopedWriteTransaction trans(db);
|
|
|
|
ServerCleanupDao cleanup_dao(db);
|
|
|
|
std::vector<ServerCleanupDao::SImageBackupInfo> images = cleanup_dao.getClientImages(rename_from);
|
|
|
|
for (size_t i = 0; i < images.size(); ++i)
|
|
{
|
|
if (!ImageMount::unmount_images(images[i].id))
|
|
{
|
|
//retry later
|
|
return true;
|
|
}
|
|
}
|
|
|
|
if (!os_remove_dir(backupfolder + os_file_sep() + clientname))
|
|
{
|
|
return true;
|
|
}
|
|
|
|
if (!os_rename_file(backupfolder + os_file_sep() + old_name.name,
|
|
backupfolder + os_file_sep() + clientname))
|
|
{
|
|
os_create_dir(backupfolder + os_file_sep() + clientname);
|
|
return true;
|
|
}
|
|
|
|
os_sync(backupfolder);
|
|
|
|
for (size_t i = 0; i < images.size(); ++i)
|
|
{
|
|
ClientMain::SPathComponents components = extractBackupComponents(images[i].path, backupfolder, old_backupfolders);
|
|
|
|
std::string new_path = backupfolder + os_file_sep() + clientname + os_file_sep() + components.path;
|
|
|
|
cleanup_dao.changeImagePath(new_path, images[i].id);
|
|
}
|
|
|
|
|
|
if (backup_dao->hasFileBackups(rename_from) > 0)
|
|
{
|
|
backup_dao->addClientMoved(old_name.name, clientname);
|
|
|
|
std::vector<std::string> moved_to = backup_dao->getClientMovedLimit5(old_name.name);
|
|
for (size_t i = 0; i < moved_to.size(); ++i)
|
|
{
|
|
backup_dao->addClientMoved(moved_to[i], clientname);
|
|
}
|
|
}
|
|
|
|
ServerBackupDao::CondString internet_authkey = backup_dao->getSetting(clientid, "internet_authkey");
|
|
if (internet_authkey.exists)
|
|
{
|
|
backup_dao->updateSetting(internet_authkey.value, "internet_authkey", rename_from);
|
|
}
|
|
|
|
ServerBackupDao::CondString computername = backup_dao->getSetting(clientid, "computername");
|
|
if (computername.exists)
|
|
{
|
|
backup_dao->updateSetting(computername.value, "computername", rename_from);
|
|
}
|
|
|
|
ServerCleanupThread::deleteClientSQL(db, clientid);
|
|
if (clientsubname.empty())
|
|
{
|
|
backup_dao->changeClientName(clientname, rename_from);
|
|
}
|
|
else
|
|
{
|
|
backup_dao->changeClientNameWithVirtualmain(clientname, clientmainname, rename_from);
|
|
}
|
|
|
|
clientid = rename_from;
|
|
|
|
return true;
|
|
}
|
|
|
|
ClientMain::SPathComponents ClientMain::extractBackupComponents(const std::string & path, const std::string& backupfolder, const std::vector<std::string>& old_backupfolders)
|
|
{
|
|
ClientMain::SPathComponents ret;
|
|
|
|
size_t off = std::string::npos;
|
|
if (next(path, 0, backupfolder + os_file_sep()))
|
|
{
|
|
ret.backupfolder = backupfolder;
|
|
off = backupfolder.size() + 1;
|
|
}
|
|
else
|
|
{
|
|
for (size_t i = 0; i < old_backupfolders.size(); ++i)
|
|
{
|
|
if (next(path, 0, old_backupfolders[i] + os_file_sep()))
|
|
{
|
|
ret.backupfolder = old_backupfolders[i];
|
|
off = old_backupfolders[i].size() + 1;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (off == std::string::npos)
|
|
{
|
|
size_t cpos = path.find(clientname);
|
|
if (cpos == std::string::npos)
|
|
return ret;
|
|
|
|
if (cpos == 0)
|
|
return ret;
|
|
|
|
ret.backupfolder = path.substr(0, cpos - 1);
|
|
off = cpos;
|
|
}
|
|
|
|
size_t noff = path.find(os_file_sep(), off);
|
|
|
|
if (noff == std::string::npos)
|
|
return ret;
|
|
|
|
ret.clientname = path.substr(off, noff - off);
|
|
ret.path = path.substr(noff + 1);
|
|
|
|
return ret;
|
|
}
|
|
|
|
bool ClientMain::startBackupBarrier( int64 timeout_seconds )
|
|
{
|
|
IScopedLock lock(running_backup_mutex);
|
|
|
|
running_backups_allowed=false;
|
|
|
|
int64 starttime = Server->getTimeMS();
|
|
|
|
while(running_backups>0 && Server->getTimeMS()-starttime<timeout_seconds*1000)
|
|
{
|
|
lock.relock(NULL);
|
|
Server->wait(60000);
|
|
lock.relock(running_backup_mutex);
|
|
}
|
|
|
|
if(running_backups==0)
|
|
{
|
|
return true;
|
|
}
|
|
|
|
running_backups_allowed=true;
|
|
return false;
|
|
}
|
|
|
|
void ClientMain::stopBackupBarrier()
|
|
{
|
|
IScopedLock lock(running_backup_mutex);
|
|
running_backups_allowed=true;
|
|
}
|
|
|
|
void ClientMain::updateCapa()
|
|
{
|
|
update_capa = true;
|
|
}
|
|
|
|
std::vector<SLogEntry> ClientMain::parseLogData(int64 max_duration, const std::string & data)
|
|
{
|
|
std::vector<SLogEntry> ret;
|
|
std::vector<std::string> lines;
|
|
Tokenize(data, lines, "\n");
|
|
int64 initial_time = Server->getTimeSeconds();
|
|
for (size_t i = 0; i<lines.size(); ++i)
|
|
{
|
|
size_t s1 = lines[i].find("-");
|
|
size_t s2 = lines[i].find("-", s1 + 1);
|
|
|
|
if (s1 == std::string::npos)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
SLogEntry entry;
|
|
|
|
entry.loglevel = atoi(lines[i].substr(0, s1).c_str());
|
|
|
|
entry.time = 0;
|
|
|
|
if (s2 != std::string::npos)
|
|
{
|
|
entry.time = os_atoi64(lines[i].substr(s1 + 1, s2 - s1));
|
|
|
|
if (i == 0)
|
|
{
|
|
initial_time = entry.time;
|
|
continue;
|
|
}
|
|
|
|
entry.time = initial_time - entry.time;
|
|
|
|
if (entry.time>max_duration + 60)
|
|
{
|
|
entry.time = 0;
|
|
}
|
|
|
|
entry.data = lines[i].substr(s2 + 1);
|
|
}
|
|
else
|
|
{
|
|
entry.data = lines[i].substr(s1 + 1);
|
|
}
|
|
|
|
ret.push_back(entry);
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
bool ClientMain::authenticateIfNeeded(bool retry_exit, bool force)
|
|
{
|
|
{
|
|
IScopedLock lock(clientaddr_mutex);
|
|
|
|
if (session_identity_refreshtime!=0
|
|
&& Server->getTimeMS() - session_identity_refreshtime<30 * 60 * 1000
|
|
&& !force)
|
|
{
|
|
return true;
|
|
}
|
|
|
|
session_identity.clear();
|
|
}
|
|
|
|
ServerStatus::setStatusError(clientname, se_authenticating);
|
|
bool c = false;
|
|
do
|
|
{
|
|
c=false;
|
|
|
|
bool b = authenticatePubKey();
|
|
if(!b && needs_authentification)
|
|
{
|
|
ServerStatus::setStatusError(clientname, se_authentication_error);
|
|
|
|
if (!retry_exit)
|
|
{
|
|
Server->Log("Authentification failed for client \"" + clientname + "\"", LL_INFO);
|
|
return false;
|
|
}
|
|
|
|
Server->wait(5*60*1000); //5min
|
|
|
|
std::string msg;
|
|
pipe->Read(&msg, ident_err_retry_time);
|
|
if(msg=="exit" || msg=="exitnow")
|
|
{
|
|
Server->Log("client_main Thread for client \""+clientname+"\" finished and the authentification failed", LL_INFO);
|
|
pipe->Write(msg);
|
|
return false;
|
|
}
|
|
else if (next(msg, 0, "address"))
|
|
{
|
|
updateClientAddress(msg.substr(7));
|
|
|
|
tcpstack.setAddChecksum(internet_connection);
|
|
}
|
|
else
|
|
{
|
|
pipe->Write(msg);
|
|
Server->wait(ident_err_retry_time);
|
|
sendServerIdentity(false);
|
|
}
|
|
|
|
c=true;
|
|
}
|
|
else
|
|
{
|
|
ServerStatus::setStatusError(clientname, se_none);
|
|
}
|
|
}
|
|
while(c);
|
|
|
|
return true;
|
|
}
|