mirror of
https://github.com/uroni/urbackup_backend.git
synced 2025-10-26 11:36:50 +00:00
4793 lines
120 KiB
C++
4793 lines
120 KiB
C++
/*************************************************************************
|
|
* UrBackup - Client/Server backup system
|
|
* Copyright (C) 2011-2016 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 "ClientService.h"
|
|
#include "RestoreFiles.h"
|
|
#include "client.h"
|
|
#include "../Interface/Server.h"
|
|
#include "../Interface/Database.h"
|
|
#include "../Interface/Query.h"
|
|
#include "../Interface/SettingsReader.h"
|
|
#include "../Interface/File.h"
|
|
#include "../Interface/ThreadPool.h"
|
|
#include "../stringtools.h"
|
|
#include "database.h"
|
|
#include "../common/data.h"
|
|
#include "../fsimageplugin/IFilesystem.h"
|
|
#include "../fsimageplugin/IFSImageFactory.h"
|
|
#include "../cryptoplugin/ICryptoFactory.h"
|
|
#include "../urbackupcommon/sha2/sha2.h"
|
|
#include "ServerIdentityMgr.h"
|
|
#include "../urbackupcommon/settings.h"
|
|
#include "ImageThread.h"
|
|
#include "InternetClient.h"
|
|
#include "../urbackupcommon/settingslist.h"
|
|
#include "../urbackupcommon/capa_bits.h"
|
|
#include "../urbackupcommon/os_functions.h"
|
|
#include "../urbackupcommon/CompressedPipe2.h"
|
|
#include "../urbackupcommon/CompressedPipeZstd.h"
|
|
#include "../urbackupcommon/InternetServicePipe2.h"
|
|
#include "RansomwareCanary.h"
|
|
#include "FilesystemManager.h"
|
|
#include "LocalFullFileBackup.h"
|
|
#include "LocalIncrFileBackup.h"
|
|
#include "file_permissions.h"
|
|
|
|
#include <memory.h>
|
|
#include <stdlib.h>
|
|
#include <limits.h>
|
|
#include <memory>
|
|
#include <algorithm>
|
|
#include <assert.h>
|
|
|
|
|
|
#ifndef _WIN32
|
|
#define _atoi64 atoll
|
|
#endif
|
|
|
|
#ifdef _WIN32
|
|
#define UPDATE_FILE_PREFIX ""
|
|
#define UPDATE_SIGNATURE_PREFIX ""
|
|
#else
|
|
#define UPDATE_FILE_PREFIX "urbackup/"
|
|
#include "../config.h"
|
|
#define UPDATE_SIGNATURE_PREFIX DATADIR "/urbackup/"
|
|
#endif
|
|
|
|
extern ICryptoFactory *crypto_fak;
|
|
extern IFSImageFactory* image_fak;
|
|
extern std::string time_format_str;
|
|
|
|
ICustomClient* ClientService::createClient()
|
|
{
|
|
return new ClientConnector();
|
|
}
|
|
|
|
void ClientService::destroyClient( ICustomClient * pClient)
|
|
{
|
|
delete ((ClientConnector*)pClient);
|
|
}
|
|
|
|
std::vector<SRunningProcess> ClientConnector::running_processes;
|
|
std::vector<SFinishedProcess> ClientConnector::finished_processes;
|
|
int64 ClientConnector::curr_backup_running_id = 0;
|
|
IMutex *ClientConnector::backup_mutex=nullptr;
|
|
IMutex *ClientConnector::process_mutex = nullptr;
|
|
int ClientConnector::backup_interval=6*60*60;
|
|
int ClientConnector::backup_alert_delay=5*60;
|
|
std::vector<SChannel> ClientConnector::channel_pipes;
|
|
db_results ClientConnector::cached_status;
|
|
std::map<std::string, int64> ClientConnector::last_token_times;
|
|
int ClientConnector::last_capa=0;
|
|
IMutex *ClientConnector::ident_mutex=nullptr;
|
|
std::vector<std::string> ClientConnector::new_server_idents;
|
|
bool ClientConnector::end_to_end_file_backup_verification_enabled=false;
|
|
std::map<std::pair<std::string, std::string>, ClientConnector::SChallenge> ClientConnector::challenges;
|
|
bool ClientConnector::has_file_changes = false;
|
|
std::vector < ClientConnector::SFilesrvConnection > ClientConnector::fileserv_connections;
|
|
RestoreOkStatus ClientConnector::restore_ok_status = RestoreOk_None;
|
|
bool ClientConnector::status_updated= false;
|
|
RestoreFiles* ClientConnector::restore_files = nullptr;
|
|
size_t ClientConnector::needs_restore_restart = 0;
|
|
size_t ClientConnector::ask_restore_ok = 0;
|
|
int64 ClientConnector::service_starttime = 0;
|
|
SRestoreToken ClientConnector::restore_token;
|
|
std::map<std::string, SAsyncFileList> ClientConnector::async_file_index;
|
|
std::deque<std::pair<std::string, std::string> > ClientConnector::finished_async_file_index;
|
|
bool ClientConnector::last_metered = false;
|
|
bool ClientConnector::last_locked = false;
|
|
int64 ClientConnector::startup_timestamp = 0;
|
|
|
|
|
|
#ifdef _WIN32
|
|
SVolumesCache* ClientConnector::volumes_cache;
|
|
#endif
|
|
|
|
|
|
#ifdef _WIN32
|
|
const std::string pw_file="pw.txt";
|
|
const std::string pw_change_file="pw_change.txt";
|
|
#else
|
|
const std::string pw_file="urbackup/pw.txt";
|
|
const std::string pw_change_file="urbackup/pw_change.txt";
|
|
#endif
|
|
|
|
namespace
|
|
{
|
|
class UpdateSilentThread : public IThread
|
|
{
|
|
public:
|
|
void operator()(void)
|
|
{
|
|
Server->wait(2 * 60 * 1000); //2min
|
|
|
|
#ifdef _WIN32
|
|
STARTUPINFOW si;
|
|
PROCESS_INFORMATION pi;
|
|
memset(&si, 0, sizeof(STARTUPINFO));
|
|
memset(&pi, 0, sizeof(PROCESS_INFORMATION));
|
|
si.cb = sizeof(STARTUPINFO);
|
|
if (!CreateProcessW(L"UrBackupUpdate.exe", L"UrBackupUpdate.exe /S", NULL, NULL, false, NORMAL_PRIORITY_CLASS | CREATE_NO_WINDOW, NULL, NULL, &si, &pi))
|
|
{
|
|
Server->Log("Executing silent update failed: " + convert((int)GetLastError()), LL_ERROR);
|
|
}
|
|
else
|
|
{
|
|
CloseHandle(pi.hProcess);
|
|
CloseHandle(pi.hThread);
|
|
}
|
|
#else
|
|
os_system("/bin/sh urbackup/UrBackupUpdate.sh -- silent");
|
|
#endif
|
|
|
|
delete this;
|
|
}
|
|
};
|
|
|
|
class TimeoutFilesrvThread : public IThread
|
|
{
|
|
public:
|
|
void operator()()
|
|
{
|
|
while (true)
|
|
{
|
|
Server->wait(600000);
|
|
ClientConnector::timeoutFilesrvConnections();
|
|
}
|
|
}
|
|
};
|
|
|
|
class TimeoutBackupImmediate : public IThread
|
|
{
|
|
void operator()()
|
|
{
|
|
std::string backup_type_str = Server->getServerParameter("backup_immediate");
|
|
|
|
RunningAction ra;
|
|
if (backup_type_str == "incr-file")
|
|
ra = RUNNING_INCR_FILE;
|
|
else if (backup_type_str == "full-file")
|
|
ra = RUNNING_FULL_FILE;
|
|
else if (backup_type_str == "incr-image")
|
|
ra = RUNNING_INCR_IMAGE;
|
|
else
|
|
ra = RUNNING_FULL_IMAGE;
|
|
|
|
int64 start_timeout = watoi64(Server->getServerParameter("backup_immediate_start_timeout"))*1000;
|
|
int64 resume_timeout = watoi64(Server->getServerParameter("backup_immediate_timeout"))*1000;
|
|
bool has_backup = false;
|
|
int64 starttime = Server->getTimeMS();
|
|
while (true)
|
|
{
|
|
Server->wait(1000);
|
|
ClientConnector::timeoutBackupImmediate(start_timeout, resume_timeout, ra, has_backup, starttime);
|
|
}
|
|
}
|
|
};
|
|
|
|
class FileservClientThread : public IThread
|
|
{
|
|
public:
|
|
FileservClientThread(IPipe* pipe, const char* p_extra_buffer, size_t extra_buffer_size)
|
|
: pipe(pipe)
|
|
{
|
|
if (extra_buffer_size > 0)
|
|
{
|
|
extra_buffer.assign(p_extra_buffer, p_extra_buffer + extra_buffer_size);
|
|
}
|
|
}
|
|
|
|
void operator()()
|
|
{
|
|
IndexThread::getFileSrv()->runClient(pipe, &extra_buffer);
|
|
delete pipe;
|
|
delete this;
|
|
}
|
|
|
|
private:
|
|
IPipe* pipe;
|
|
std::vector<char> extra_buffer;
|
|
};
|
|
}
|
|
|
|
void ClientConnector::init_mutex(void)
|
|
{
|
|
if(backup_mutex==nullptr)
|
|
{
|
|
backup_mutex=Server->createMutex();
|
|
ident_mutex=Server->createMutex();
|
|
process_mutex = Server->createMutex();
|
|
|
|
Server->createThread(new TimeoutFilesrvThread, "filesrv timeout");
|
|
|
|
if (!Server->getServerParameter("backup_immediate").empty())
|
|
{
|
|
Server->createThread(new TimeoutBackupImmediate, "backup timeout");
|
|
}
|
|
}
|
|
IDatabase *db=Server->getDatabase(Server->getThreadID(), URBACKUPDB_CLIENT);
|
|
db_results res=db->Read("SELECT tvalue FROM misc WHERE tkey='last_capa'");
|
|
if(!res.empty())
|
|
{
|
|
last_capa=watoi(res[0]["last_capa"]);
|
|
}
|
|
else
|
|
{
|
|
db->Write("INSERT INTO misc (tkey, tvalue) VALUES ('last_capa', '0');");
|
|
last_capa=0;
|
|
}
|
|
|
|
ClientDAO clientdao(db);
|
|
|
|
std::string tmp = clientdao.getMiscValue("backup_alert_delay");
|
|
if(!tmp.empty())
|
|
{
|
|
backup_alert_delay = watoi(tmp);
|
|
}
|
|
tmp = clientdao.getMiscValue("backup_interval");
|
|
if(!tmp.empty())
|
|
{
|
|
backup_interval = watoi(tmp);
|
|
}
|
|
|
|
service_starttime = Server->getTimeMS();
|
|
|
|
startup_timestamp = Server->getTimeSeconds();
|
|
}
|
|
|
|
void ClientConnector::destroy_mutex(void)
|
|
{
|
|
Server->destroy(backup_mutex);
|
|
Server->destroy(ident_mutex);
|
|
Server->destroy(process_mutex);
|
|
}
|
|
|
|
ClientConnector::ClientConnector(void)
|
|
{
|
|
}
|
|
|
|
bool ClientConnector::wantReceive(void)
|
|
{
|
|
return want_receive;
|
|
}
|
|
|
|
void ClientConnector::Init(THREAD_ID pTID, IPipe *pPipe, const std::string& pEndpointName)
|
|
{
|
|
tid=pTID;
|
|
pipe=pPipe;
|
|
state=CCSTATE_NORMAL;
|
|
image_inf.thread_action=TA_NONE;
|
|
image_inf.image_thread=nullptr;
|
|
image_inf.clientsubname.clear();
|
|
lasttime=Server->getTimeMS();
|
|
do_quit=false;
|
|
is_channel=false;
|
|
want_receive=true;
|
|
last_channel_ping=0;
|
|
file_version=1;
|
|
internet_conn=false;
|
|
tcpstack.setAddChecksum(false);
|
|
last_update_time=lasttime;
|
|
endpoint_name = pEndpointName;
|
|
make_fileserv=false;
|
|
local_backup_running_id = 0;
|
|
run_other = nullptr;
|
|
idle_timeout = 10000;
|
|
bitmapfile = nullptr;
|
|
retrieved_has_components=false;
|
|
status_has_components=false;
|
|
curr_result_id = 0;
|
|
is_encrypted = false;
|
|
}
|
|
|
|
ClientConnector::~ClientConnector(void)
|
|
{
|
|
if (curr_result_id != 0)
|
|
{
|
|
IndexThread::removeResult(curr_result_id);
|
|
}
|
|
}
|
|
|
|
bool ClientConnector::Run(IRunOtherCallback* p_run_other)
|
|
{
|
|
run_other = p_run_other;
|
|
|
|
if(do_quit)
|
|
{
|
|
if(is_channel)
|
|
{
|
|
IScopedLock lock(backup_mutex);
|
|
for(size_t i=0;i<channel_pipes.size();++i)
|
|
{
|
|
if(channel_pipes[i].pipe==pipe)
|
|
{
|
|
channel_pipes.erase(channel_pipes.begin()+i);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
if(waitForThread())
|
|
{
|
|
want_receive = false;
|
|
return true;
|
|
}
|
|
IndexThread::unrefResult(curr_result_id);
|
|
curr_result_id = 0;
|
|
delete image_inf.image_thread;
|
|
image_inf.image_thread=nullptr;
|
|
return false;
|
|
}
|
|
|
|
switch(state)
|
|
{
|
|
case CCSTATE_NORMAL:
|
|
if(Server->getTimeMS()-lasttime>idle_timeout)
|
|
{
|
|
Server->Log("Client timeout in ClientConnector::Run", LL_DEBUG);
|
|
if(waitForThread())
|
|
{
|
|
do_quit=true;
|
|
return true;
|
|
}
|
|
IndexThread::unrefResult(curr_result_id);
|
|
curr_result_id = 0;
|
|
return false;
|
|
}
|
|
return true;
|
|
case CCSTATE_START_FILEBACKUP_ASYNC:
|
|
case CCSTATE_START_FILEBACKUP:
|
|
{
|
|
std::string msg;
|
|
bool has_result = false;
|
|
|
|
if (curr_result_id != 0)
|
|
{
|
|
has_result = IndexThread::getResult(curr_result_id, 0, msg);
|
|
}
|
|
else if (curr_backup_tt != ILLEGAL_THREADPOOL_TICKET)
|
|
{
|
|
has_result = true;
|
|
if (Server->getThreadPool()->waitFor(curr_backup_tt, 0))
|
|
{
|
|
msg = "done";
|
|
}
|
|
}
|
|
|
|
if (state == CCSTATE_START_FILEBACKUP_ASYNC)
|
|
{
|
|
IScopedLock lock(backup_mutex);
|
|
std::map<std::string, SAsyncFileList>::iterator it = async_file_index.find(async_file_list_id);
|
|
if (it != async_file_index.end())
|
|
{
|
|
if (!msg.empty()
|
|
&& msg!="phash")
|
|
{
|
|
Server->Log("Async index " + bytesToHex(it->first) + " finished with \"" + msg + "\"", LL_DEBUG);
|
|
if (it->second.refcount > 0) --it->second.refcount;
|
|
finished_async_file_index.push_back(std::make_pair(it->first, msg));
|
|
while (finished_async_file_index.size() > 10)
|
|
{
|
|
finished_async_file_index.pop_front();
|
|
}
|
|
IndexThread::removeResult(it->second.result_id);
|
|
async_file_index.erase(it);
|
|
}
|
|
else
|
|
{
|
|
it->second.last_update = Server->getTimeMS();
|
|
}
|
|
}
|
|
}
|
|
|
|
if(msg=="exit" || !has_result)
|
|
{
|
|
if(waitForThread())
|
|
{
|
|
do_quit=true;
|
|
return true;
|
|
}
|
|
IndexThread::unrefResult(curr_result_id);
|
|
curr_result_id = 0;
|
|
IScopedLock lock(backup_mutex);
|
|
removeRunningProcess(local_backup_running_id, false);
|
|
return false;
|
|
}
|
|
else if(msg=="done")
|
|
{
|
|
tcpstack.Send(pipe, "DONE");
|
|
lasttime=Server->getTimeMS();
|
|
idle_timeout = 5*60*1000;
|
|
state=CCSTATE_NORMAL;
|
|
}
|
|
else if (msg == "phash")
|
|
{
|
|
tcpstack.Send(pipe, "PHASH");
|
|
}
|
|
else if(!msg.empty())
|
|
{
|
|
tcpstack.Send(pipe, msg);
|
|
lasttime=Server->getTimeMS();
|
|
state=CCSTATE_NORMAL;
|
|
IScopedLock lock(backup_mutex);
|
|
removeRunningProcess(local_backup_running_id, false);
|
|
}
|
|
else if( ( file_version>1 || state==CCSTATE_START_FILEBACKUP_ASYNC)
|
|
&& Server->getTimeMS()-last_update_time>30000 )
|
|
{
|
|
last_update_time=Server->getTimeMS();
|
|
tcpstack.Send(pipe, "BUSY");
|
|
}
|
|
}break;
|
|
case CCSTATE_SHADOWCOPY:
|
|
{
|
|
std::string msg;
|
|
bool has_result = IndexThread::getResult(curr_result_id, 0, msg);
|
|
if (!msg.empty() || !has_result)
|
|
{
|
|
if (msg == "exit" || !has_result)
|
|
{
|
|
assert(false);
|
|
if (waitForThread())
|
|
{
|
|
do_quit = true;
|
|
return true;
|
|
}
|
|
IndexThread::unrefResult(curr_result_id);
|
|
curr_result_id = 0;
|
|
return false;
|
|
}
|
|
else if (msg.find("done") == 0)
|
|
{
|
|
tcpstack.Send(pipe, "DONE");
|
|
}
|
|
else if (msg.find("failed") == 0)
|
|
{
|
|
tcpstack.Send(pipe, "FAILED");
|
|
}
|
|
else if (msg.find("in use") == 0)
|
|
{
|
|
tcpstack.Send(pipe, "IN USE");
|
|
}
|
|
else
|
|
{
|
|
Server->Log("Unknown msg " + msg + " in CCSTATE_SHADOWCOPY", LL_ERROR);
|
|
assert(false);
|
|
}
|
|
lasttime = Server->getTimeMS();
|
|
state = CCSTATE_NORMAL;
|
|
}
|
|
}break;
|
|
case CCSTATE_CHANNEL: //Channel
|
|
{
|
|
IScopedLock lock(backup_mutex);
|
|
|
|
SChannel* chan = getCurrChannel();
|
|
|
|
if (chan->state == SChannel::EChannelState_Used)
|
|
{
|
|
return true;
|
|
}
|
|
else
|
|
{
|
|
want_receive = true;
|
|
}
|
|
|
|
bool has_ping = chan != nullptr && chan->state == SChannel::EChannelState_Pinging;
|
|
|
|
int64 timeout_interval = 180000;
|
|
|
|
int64 ctime = Server->getTimeMS();
|
|
if(ctime-lasttime>timeout_interval
|
|
|| (has_ping && ctime-last_channel_ping>10000) )
|
|
{
|
|
std::string extra;
|
|
if (has_ping && ctime - last_channel_ping > 10000)
|
|
{
|
|
extra = " (ping timeout)";
|
|
}
|
|
else
|
|
{
|
|
extra = " (total timeout)";
|
|
}
|
|
Server->Log("Client timeout in ClientConnector::Run - Channel"+extra, LL_DEBUG);
|
|
for(size_t i=0;i<channel_pipes.size();++i)
|
|
{
|
|
if(channel_pipes[i].pipe==pipe)
|
|
{
|
|
channel_pipes.erase(channel_pipes.begin()+i);
|
|
break;
|
|
}
|
|
}
|
|
if(waitForThread())
|
|
{
|
|
do_quit=true;
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
if(chan!=nullptr && chan->state==SChannel::EChannelState_Exit)
|
|
{
|
|
do_quit=true;
|
|
break;
|
|
}
|
|
if(Server->getTimeMS()-last_channel_ping>60000
|
|
&& chan->state==SChannel::EChannelState_Idle)
|
|
{
|
|
tcpstack.Send(pipe, "PING");
|
|
last_channel_ping=Server->getTimeMS();
|
|
chan->state = SChannel::EChannelState_Pinging;
|
|
}
|
|
if(make_fileserv
|
|
&& chan->state == SChannel::EChannelState_Idle)
|
|
{
|
|
size_t idx=std::string::npos;
|
|
for(size_t i=0;i<channel_pipes.size();++i)
|
|
{
|
|
if(channel_pipes[i].pipe==pipe)
|
|
{
|
|
idx=i;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if(idx!=std::string::npos)
|
|
{
|
|
tcpstack.Send(pipe, "FILESERV");
|
|
state=CCSTATE_FILESERV;
|
|
fileserv_connections.push_back(SFilesrvConnection(channel_pipes[idx].token, pipe));
|
|
|
|
channel_pipes.erase(channel_pipes.begin()+idx);
|
|
}
|
|
|
|
return false;
|
|
}
|
|
}break;
|
|
case CCSTATE_IMAGE:
|
|
case CCSTATE_IMAGE_HASHDATA:
|
|
case CCSTATE_IMAGE_BITMAP:
|
|
{
|
|
if(!Server->getThreadPool()->isRunning(image_inf.thread_ticket))
|
|
{
|
|
delete image_inf.image_thread;
|
|
image_inf.image_thread=nullptr;
|
|
IndexThread::unrefResult(curr_result_id);
|
|
curr_result_id = 0;
|
|
return false;
|
|
}
|
|
}break;
|
|
case CCSTATE_UPDATE_DATA:
|
|
case CCSTATE_UPDATE_FINISH:
|
|
{
|
|
if(Server->getTimeMS()-lasttime>10000)
|
|
{
|
|
Server->Log("Client timeout in ClientConnector::Run-update(state6|7)", LL_DEBUG);
|
|
if(waitForThread())
|
|
{
|
|
do_quit=true;
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
if(state==CCSTATE_UPDATE_FINISH)
|
|
{
|
|
if(hashdataok)
|
|
{
|
|
hashdatafile->Seek(0);
|
|
writeUpdateFile(hashdatafile, UPDATE_FILE_PREFIX "version_new.txt");
|
|
writeUpdateFile(hashdatafile, UPDATE_FILE_PREFIX "UrBackupUpdate.sig2");
|
|
writeUpdateFile(hashdatafile, UPDATE_FILE_PREFIX "UrBackupUpdate_untested.dat");
|
|
|
|
std::string hashdatafile_fn=hashdatafile->getFilename();
|
|
Server->destroy(hashdatafile);
|
|
Server->deleteFile(hashdatafile_fn);
|
|
|
|
if(crypto_fak!=nullptr)
|
|
{
|
|
#if defined(__APPLE__)
|
|
// ./UrBackup\ Client.app/Contents/MacOS/sbin/../share/urbackup/urbackup_ecdsa409k1.pub
|
|
std::string pubkey = ExtractFilePath(Server->getServerWorkingDir()) + "/share/urbackup/urbackup_ecdsa409k1.pub";
|
|
#else
|
|
std::string pubkey = UPDATE_SIGNATURE_PREFIX "urbackup_ecdsa409k1.pub";
|
|
#endif
|
|
if (crypto_fak->verifyFile(pubkey,
|
|
UPDATE_FILE_PREFIX "UrBackupUpdate_untested.dat", UPDATE_FILE_PREFIX "UrBackupUpdate.sig2"))
|
|
{
|
|
std::unique_ptr<IFile> updatefile(Server->openFile(UPDATE_FILE_PREFIX "UrBackupUpdate_untested.dat"));
|
|
if(updatefile.get()!=nullptr)
|
|
{
|
|
if(checkHash(getSha512Hash(updatefile.get()))
|
|
&& checkVersion(updatefile.get()) )
|
|
{
|
|
updatefile.reset();
|
|
#ifdef _WIN32
|
|
Server->deleteFile(UPDATE_FILE_PREFIX "UrBackupUpdate.exe");
|
|
moveFile(UPDATE_FILE_PREFIX "UrBackupUpdate_untested.dat", UPDATE_FILE_PREFIX "UrBackupUpdate.exe");
|
|
#else
|
|
Server->deleteFile(UPDATE_FILE_PREFIX "UrBackupUpdate.sh");
|
|
moveFile(UPDATE_FILE_PREFIX "UrBackupUpdate_untested.dat", UPDATE_FILE_PREFIX "UrBackupUpdate.sh");
|
|
#endif
|
|
|
|
tcpstack.Send(pipe, "ok");
|
|
|
|
if(silent_update)
|
|
{
|
|
update_silent();
|
|
}
|
|
else
|
|
{
|
|
Server->deleteFile(UPDATE_FILE_PREFIX "version.txt");
|
|
moveFile(UPDATE_FILE_PREFIX "version_new.txt", UPDATE_FILE_PREFIX "version.txt");
|
|
}
|
|
}
|
|
else
|
|
{
|
|
updatefile.reset();
|
|
Server->Log("Verifying update file failed. Update was installed previously", LL_ERROR);
|
|
Server->deleteFile(UPDATE_FILE_PREFIX "UrBackupUpdate_untested.dat");
|
|
tcpstack.Send(pipe, "verify_sig_already_used_err");
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
Server->Log("Verifying update file failed. Signature did not match", LL_ERROR);
|
|
Server->deleteFile(UPDATE_FILE_PREFIX "UrBackupUpdate_untested.dat");
|
|
tcpstack.Send(pipe, "verify_sig_err");
|
|
}
|
|
}
|
|
else
|
|
{
|
|
Server->Log("Verifying update file failed. Cryptomodule not present", LL_ERROR);
|
|
tcpstack.Send(pipe, "verify_cryptmodule_err");
|
|
}
|
|
|
|
state=CCSTATE_NORMAL;
|
|
}
|
|
else
|
|
{
|
|
do_quit=true;
|
|
}
|
|
}
|
|
return true;
|
|
}break;
|
|
case CCSTATE_STATUS:
|
|
{
|
|
IScopedLock lock(backup_mutex);
|
|
if(Server->getTimeMS()-lasttime>50000 || status_updated)
|
|
{
|
|
sendStatus();
|
|
|
|
lasttime = Server->getTimeMS();
|
|
state = CCSTATE_NORMAL;
|
|
status_updated=false;
|
|
}
|
|
return true;
|
|
} break;
|
|
|
|
}
|
|
return true;
|
|
}
|
|
|
|
std::string ClientConnector::getSha512Hash(IFile *fn)
|
|
{
|
|
sha512_ctx ctx;
|
|
char buf[4096];
|
|
_u32 r;
|
|
sha512_init(&ctx);
|
|
while((r=fn->Read(buf, 4096))!=0)
|
|
{
|
|
sha512_update(&ctx, (unsigned char*)buf, r);
|
|
}
|
|
unsigned char digest[64];
|
|
sha512_final(&ctx,digest);
|
|
return bytesToHex(digest, 64);
|
|
}
|
|
|
|
bool ClientConnector::checkHash(std::string shah)
|
|
{
|
|
std::string prev_h=getFile(UPDATE_FILE_PREFIX "updates_h.dat");
|
|
std::vector<std::string> lines;
|
|
Tokenize(prev_h, lines, "\n");
|
|
for(size_t i=0;i<lines.size();++i)
|
|
{
|
|
std::string l=strlower(trim(lines[i]));
|
|
if(!l.empty() && next(shah, 0, l))
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
bool ClientConnector::checkVersion(IFile* updatef)
|
|
{
|
|
//TODO: Enable when tested
|
|
return true;
|
|
|
|
const std::string version_start_guid = "#ab6b754c02624b348795c92da78b1e73";
|
|
std::string new_version = trim(getFile(UPDATE_FILE_PREFIX "version_new.txt"));
|
|
|
|
updatef->Seek(0);
|
|
std::string data;
|
|
size_t state = 0;
|
|
size_t pos = 0;
|
|
while (!(data = updatef->Read(4096)).empty())
|
|
{
|
|
for (size_t i = 0; i < data.size(); ++i)
|
|
{
|
|
if (data[i] == version_start_guid[state])
|
|
{
|
|
if (state == version_start_guid.size() - 1)
|
|
{
|
|
std::string curr_version;
|
|
if (i + 1 + new_version.size() < data.size())
|
|
curr_version = data.substr(i+1, new_version.size());
|
|
else
|
|
curr_version = updatef->Read(pos + i+ 1, static_cast<_u32>(new_version.size()));
|
|
|
|
if (curr_version == new_version)
|
|
{
|
|
std::string version_2 = trim(getFile(UPDATE_FILE_PREFIX "curr_version.txt"));
|
|
if (versionNeedsUpdate(version_2, curr_version))
|
|
{
|
|
return true;
|
|
}
|
|
else
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
++state;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
state = 0;
|
|
}
|
|
}
|
|
pos += data.size();
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
bool ClientConnector::writeUpdateFile(IFile *datafile, std::string outfn)
|
|
{
|
|
unsigned int size;
|
|
if(datafile->Read((char*)&size, sizeof(unsigned int))!=sizeof(unsigned int))
|
|
return false;
|
|
|
|
IFile *out=Server->openFile(outfn, MODE_WRITE);
|
|
if(out==nullptr)
|
|
return false;
|
|
|
|
size_t read=0;
|
|
char buf[4096];
|
|
while(true)
|
|
{
|
|
size_t tr=(std::min)((size_t)4096, size-read);
|
|
if(tr==0)
|
|
break;
|
|
|
|
if(datafile->Read(buf, (_u32)tr)!=tr)
|
|
{
|
|
Server->destroy(out);
|
|
return false;
|
|
}
|
|
|
|
if(out->Write(buf, (_u32)tr)!=tr)
|
|
{
|
|
Server->destroy(out);
|
|
return false;
|
|
}
|
|
|
|
read+=tr;
|
|
}
|
|
|
|
Server->destroy(out);
|
|
return true;
|
|
}
|
|
|
|
void ClientConnector::ReceivePackets(IRunOtherCallback* p_run_other)
|
|
{
|
|
run_other = p_run_other;
|
|
|
|
if(state==CCSTATE_UPDATE_FINISH)
|
|
{
|
|
return;
|
|
}
|
|
|
|
IMutex *l_mutex=nullptr;
|
|
if(is_channel)
|
|
{
|
|
l_mutex=backup_mutex;
|
|
}
|
|
IScopedLock g_lock(l_mutex);
|
|
|
|
if (is_channel)
|
|
{
|
|
SChannel* chan = getCurrChannel();
|
|
if (chan != nullptr
|
|
&& chan->state == SChannel::EChannelState_Used)
|
|
{
|
|
want_receive = false;
|
|
return;
|
|
}
|
|
}
|
|
|
|
std::string cmd;
|
|
size_t rc=pipe->Read(&cmd, is_channel?0:-1);
|
|
if(rc==0 )
|
|
{
|
|
if(!do_quit)
|
|
{
|
|
Server->Log("rc=0 hasError="+convert(pipe->hasError())+" state="+convert(state), LL_DEBUG);
|
|
#ifdef _WIN32
|
|
#ifdef _DEBUG
|
|
Server->Log("Err: "+convert((int)GetLastError()), LL_DEBUG);
|
|
#endif
|
|
#endif
|
|
}
|
|
|
|
if(is_channel && pipe->hasError())
|
|
{
|
|
do_quit=true;
|
|
}
|
|
else if(!is_channel)
|
|
{
|
|
do_quit=true;
|
|
}
|
|
else
|
|
{
|
|
lasttime=Server->getTimeMS();
|
|
}
|
|
return;
|
|
}
|
|
|
|
while(state==CCSTATE_IMAGE_HASHDATA
|
|
|| state==CCSTATE_UPDATE_DATA
|
|
|| state== CCSTATE_IMAGE_BITMAP)
|
|
{
|
|
lasttime=Server->getTimeMS();
|
|
|
|
IFile* datafile;
|
|
unsigned int* dataleft;
|
|
if (state == CCSTATE_IMAGE_HASHDATA
|
|
|| state == CCSTATE_UPDATE_DATA)
|
|
{
|
|
datafile = hashdatafile;
|
|
dataleft = &hashdataleft;
|
|
}
|
|
else
|
|
{
|
|
datafile = bitmapfile;
|
|
dataleft = &bitmapleft;
|
|
}
|
|
|
|
unsigned int towrite = (std::min)(*dataleft, static_cast<unsigned int>(cmd.size()));
|
|
|
|
if(datafile->Write(cmd.substr(0, towrite))!=towrite)
|
|
{
|
|
Server->Log("Error writing to data to temporary file", LL_ERROR);
|
|
do_quit=true;
|
|
return;
|
|
}
|
|
|
|
*dataleft -= towrite;
|
|
|
|
if(*dataleft==0)
|
|
{
|
|
if (state == CCSTATE_IMAGE_HASHDATA
|
|
|| state == CCSTATE_UPDATE_DATA)
|
|
{
|
|
if (bitmapfile == nullptr)
|
|
{
|
|
hashdataok = true;
|
|
if (state == CCSTATE_IMAGE_HASHDATA)
|
|
state = CCSTATE_IMAGE;
|
|
else if (state == CCSTATE_UPDATE_DATA)
|
|
state = CCSTATE_UPDATE_FINISH;
|
|
|
|
if (towrite != cmd.size())
|
|
{
|
|
Server->Log("Too much hash data (needed="+convert(towrite)+" got="+convert(cmd.size()), LL_ERROR);
|
|
}
|
|
|
|
return;
|
|
}
|
|
else
|
|
{
|
|
cmd.erase(0, towrite);
|
|
state = CCSTATE_IMAGE_BITMAP;
|
|
|
|
if (cmd.empty())
|
|
{
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (towrite != cmd.size())
|
|
{
|
|
Server->Log("Too much bitmap data (needed=" + convert(towrite) + " got=" + convert(cmd.size()), LL_ERROR);
|
|
}
|
|
|
|
hashdataok = true;
|
|
state = CCSTATE_IMAGE;
|
|
|
|
return;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
return;
|
|
}
|
|
}
|
|
|
|
tcpstack.AddData(cmd.data(), cmd.size());
|
|
|
|
while(true)
|
|
{
|
|
if (!tcpstack.getPacket(cmd) || cmd.empty())
|
|
{
|
|
size_t rc = pipe->Read(&cmd, 0);
|
|
if (rc > 0)
|
|
{
|
|
tcpstack.AddData(cmd.data(), cmd.size());
|
|
continue;
|
|
}
|
|
else
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
|
|
Server->Log("ClientService cmd: "+cmd, LL_DEBUG);
|
|
|
|
bool pw_ok=false;
|
|
bool pw_change_ok=false;
|
|
std::string identity;
|
|
bool ident_ok=false;
|
|
str_map params;
|
|
size_t hashpos;
|
|
if(cmd.size()>3 && cmd[0]=='#' && cmd[1]=='I' ) //From server
|
|
{
|
|
identity=getbetween("#I", "#", cmd);
|
|
replaceNonAlphaNumeric(identity, '_');
|
|
cmd.erase(0, identity.size()+3);
|
|
size_t tp=cmd.find("#token=");
|
|
if(tp!=std::string::npos)
|
|
{
|
|
server_token=cmd.substr(tp+7);
|
|
cmd.erase(tp, cmd.size()-tp);
|
|
}
|
|
}
|
|
else if((hashpos=cmd.find("#"))!=std::string::npos) //From front-end
|
|
{
|
|
ParseParamStrHttp(getafter("#", cmd), ¶ms, false);
|
|
|
|
cmd.erase(hashpos, cmd.size()-hashpos);
|
|
if(!checkPassword(params["pw"], pw_change_ok))
|
|
{
|
|
Server->Log("Password wrong!", LL_ERROR);
|
|
do_quit=true;
|
|
continue;
|
|
}
|
|
else
|
|
{
|
|
pw_ok=true;
|
|
}
|
|
}
|
|
|
|
std::string secret_session_key;
|
|
if(!identity.empty() && ServerIdentityMgr::checkServerSessionIdentity(identity, endpoint_name, secret_session_key))
|
|
{
|
|
ident_ok=true;
|
|
}
|
|
else if(!identity.empty() && ServerIdentityMgr::checkServerIdentity(identity))
|
|
{
|
|
if(!ServerIdentityMgr::hasPublicKey(identity, true) || crypto_fak==nullptr)
|
|
{
|
|
ident_ok=true;
|
|
}
|
|
}
|
|
|
|
if( (ident_ok || is_channel) && !internet_conn )
|
|
{
|
|
InternetClient::hasLANConnection();
|
|
}
|
|
|
|
if(cmd=="ADD IDENTITY" )
|
|
{
|
|
CMD_ADD_IDENTITY(identity, cmd, ident_ok); continue;
|
|
}
|
|
else if(next(cmd, 0, "GET CHALLENGE"))
|
|
{
|
|
CMD_GET_CHALLENGE(identity, cmd); continue;
|
|
}
|
|
else if(next(cmd, 0, "SIGNATURE"))
|
|
{
|
|
CMD_SIGNATURE(identity, cmd); continue;
|
|
}
|
|
else if(next(cmd, 0, "FULL IMAGE ") )
|
|
{
|
|
CMD_FULL_IMAGE(cmd, ident_ok); continue;
|
|
}
|
|
else if(next(cmd, 0, "INCR IMAGE ") )
|
|
{
|
|
CMD_INCR_IMAGE(cmd, ident_ok); continue;
|
|
}
|
|
|
|
if(ident_ok) //Commands from Server
|
|
{
|
|
if (next(cmd, 0, "ENC?"))
|
|
{
|
|
str_map params;
|
|
ParseParamStrHttp(cmd.substr(4), ¶ms);
|
|
|
|
std::string resp = "ok=1";
|
|
|
|
std::string client_keyadd;
|
|
std::string server_keyadd;
|
|
if (!secret_session_key.empty())
|
|
{
|
|
server_keyadd = base64_decode_dash(params["keyadd"]);
|
|
if (server_keyadd.empty())
|
|
{
|
|
Server->Log("Server keyadd is empty", LL_WARNING);
|
|
return;
|
|
}
|
|
client_keyadd.resize(16);
|
|
Server->randomFill(&client_keyadd[0], client_keyadd.size());
|
|
resp += "&keyadd=" + base64_encode_dash(client_keyadd);
|
|
}
|
|
|
|
tcpstack.Send(pipe, resp);
|
|
|
|
if (!secret_session_key.empty())
|
|
{
|
|
Server->Log("Encrypting with key " + base64_encode_dash(secret_session_key + server_keyadd + client_keyadd) + " (client)");
|
|
InternetServicePipe2* isp = new InternetServicePipe2(pipe, secret_session_key + server_keyadd + client_keyadd);
|
|
isp->destroyBackendPipeOnDelete(true);
|
|
pipe = isp;
|
|
is_encrypted = true;
|
|
}
|
|
|
|
std::string compress = params["compress"];
|
|
int compression_level = watoi(params["compress_level"]);
|
|
|
|
if (compress == "zlib")
|
|
{
|
|
CompressedPipe2* comp_pipe = new CompressedPipe2(pipe, compression_level);
|
|
comp_pipe->destroyBackendPipeOnDelete(true);
|
|
pipe = comp_pipe;
|
|
}
|
|
#ifndef NO_ZSTD_COMPRESSION
|
|
else if (compress == "zstd")
|
|
{
|
|
CompressedPipeZstd* comp_pipe = new CompressedPipeZstd(pipe, compression_level, -1);
|
|
comp_pipe->destroyBackendPipeOnDelete(true);
|
|
pipe = comp_pipe;
|
|
|
|
}
|
|
#endif
|
|
else
|
|
{
|
|
Server->Log("Unknown compression method from server: \"" + compress + "\"", LL_ERROR);
|
|
}
|
|
|
|
continue;
|
|
}
|
|
|
|
if (!secret_session_key.empty()
|
|
&& !is_encrypted)
|
|
{
|
|
tcpstack.Send(pipe, "ERR REQUIRE ENC");
|
|
return;
|
|
}
|
|
|
|
if( cmd=="START BACKUP" || cmd=="2START BACKUP" || next(cmd, 0, "3START BACKUP") )
|
|
{
|
|
CMD_START_INCR_FILEBACKUP(cmd, identity); continue;
|
|
}
|
|
else if( cmd=="START FULL BACKUP" || cmd=="2START FULL BACKUP" || next(cmd, 0, "3START FULL BACKUP") )
|
|
{
|
|
CMD_START_FULL_FILEBACKUP(cmd, identity); continue;
|
|
}
|
|
else if (cmd == "FILESRV")
|
|
{
|
|
Server->Log("Start FILESRV thread");
|
|
Server->getThreadPool()->execute(new FileservClientThread(pipe,
|
|
tcpstack.getBuffer(), tcpstack.getBuffersize()), "tfileserver");
|
|
state = CCSTATE_FILESERV;
|
|
do_quit = true;
|
|
want_receive = false;
|
|
return;
|
|
}
|
|
else if (next(cmd, 0, "WAIT FOR INDEX "))
|
|
{
|
|
CMD_WAIT_FOR_INDEX(cmd); continue;
|
|
}
|
|
else if(next(cmd, 0, "START SC \"") )
|
|
{
|
|
CMD_START_SHADOWCOPY(cmd); continue;
|
|
}
|
|
else if(next(cmd, 0, "STOP SC \"") )
|
|
{
|
|
CMD_STOP_SHADOWCOPY(cmd); continue;
|
|
}
|
|
else if(next(cmd, 0, "INCRINTERVALL \"") )
|
|
{
|
|
CMD_SET_INCRINTERVAL(cmd); continue;
|
|
}
|
|
else if(cmd=="DID BACKUP" )
|
|
{
|
|
CMD_DID_BACKUP(cmd); continue;
|
|
}
|
|
else if (next(cmd, 0, "2DID BACKUP "))
|
|
{
|
|
CMD_DID_BACKUP2(cmd); continue;
|
|
}
|
|
else if (next(cmd, 0, "BACKUP FAILED "))
|
|
{
|
|
CMD_BACKUP_FAILED(cmd); continue;
|
|
}
|
|
else if(next(cmd, 0, "SETTINGS ") )
|
|
{
|
|
CMD_UPDATE_SETTINGS(cmd, identity); continue;
|
|
}
|
|
else if(next(cmd, 0, "PING RUNNING") )
|
|
{
|
|
CMD_PING_RUNNING(cmd); continue;
|
|
}
|
|
else if(next(cmd, 0, "2PING RUNNING ") )
|
|
{
|
|
CMD_PING_RUNNING2(cmd); continue;
|
|
}
|
|
else if( next(cmd, 0, "1CHANNEL") )
|
|
{
|
|
CMD_CHANNEL(cmd, &g_lock, identity); continue;
|
|
}
|
|
else if(next(cmd, 0, "2LOGDATA ") )
|
|
{
|
|
CMD_LOGDATA(cmd); continue;
|
|
}
|
|
else if( next(cmd, 0, "MBR ") )
|
|
{
|
|
CMD_MBR(cmd); continue;
|
|
}
|
|
else if( next(cmd, 0, "VERSION ") )
|
|
{
|
|
CMD_VERSION_UPDATE(cmd); continue;
|
|
}
|
|
else if( next(cmd, 0, "1CLIENTUPDATE ") )
|
|
{
|
|
CMD_CLIENT_UPDATE(cmd); continue;
|
|
}
|
|
else if(next(cmd, 0, "CAPA") )
|
|
{
|
|
CMD_CAPA(cmd); continue;
|
|
}
|
|
else if( cmd=="ENABLE END TO END FILE BACKUP VERIFICATION")
|
|
{
|
|
CMD_ENABLE_END_TO_END_FILE_BACKUP_VERIFICATION(cmd); continue;
|
|
}
|
|
else if( cmd=="GET VSSLOG")
|
|
{
|
|
CMD_GET_VSSLOG(cmd); continue;
|
|
}
|
|
else if( cmd=="CONTINUOUS WATCH START")
|
|
{
|
|
CMD_CONTINUOUS_WATCH_START(); continue;
|
|
}
|
|
else if( next(cmd, 0, "SCRIPT STDERR ") )
|
|
{
|
|
CMD_SCRIPT_STDERR(cmd); continue;
|
|
}
|
|
else if( next(cmd, 0, "FILE RESTORE "))
|
|
{
|
|
CMD_FILE_RESTORE(cmd.substr(13), identity); continue;
|
|
}
|
|
else if (next(cmd, 0, "CLIENT ACCESS KEY "))
|
|
{
|
|
CMD_CLIENT_ACCESS_KEY(cmd.substr(18)); continue;
|
|
}
|
|
else if (next(cmd, 0, "WRITE TOKENS "))
|
|
{
|
|
CMD_WRITE_TOKENS(cmd.substr(13)); continue;
|
|
}
|
|
else if (next(cmd, 0, "ADD IDENTITY "))
|
|
{
|
|
CMD_ADD_IDENTITY(cmd.substr(13)); continue;
|
|
}
|
|
else if (cmd=="GET CLIENTNAME")
|
|
{
|
|
CMD_GET_CLIENTNAME(cmd); continue;
|
|
}
|
|
else if (next(cmd, 0, "FINISH LBACKUP?"))
|
|
{
|
|
CMD_FINISH_LBACKUP(cmd);
|
|
}
|
|
}
|
|
if(pw_ok) //Commands from client frontend
|
|
{
|
|
if(pw_change_ok) //Administrator commands
|
|
{
|
|
if(cmd=="SAVE BACKUP DIRS" )
|
|
{
|
|
CMD_SAVE_BACKUPDIRS(cmd, params); continue;
|
|
}
|
|
else if(next(cmd, 0, "UPDATE SETTINGS ") )
|
|
{
|
|
CMD_TOCHANNEL_UPDATE_SETTINGS(cmd); continue;
|
|
}
|
|
else if(cmd=="GET LOGPOINTS" )
|
|
{
|
|
CMD_GET_LOGPOINTS(cmd); continue;
|
|
}
|
|
else if(cmd=="GET LOGDATA" )
|
|
{
|
|
CMD_GET_LOGDATA(cmd, params); continue;
|
|
}
|
|
else if( cmd=="NEW SERVER" )
|
|
{
|
|
CMD_NEW_SERVER(params); continue;
|
|
}
|
|
else if (cmd == "RESET KEEP")
|
|
{
|
|
CMD_RESET_KEEP(params); continue;
|
|
}
|
|
}
|
|
|
|
if( cmd=="GET BACKUP DIRS" )
|
|
{
|
|
CMD_GET_BACKUPDIRS(cmd); continue;
|
|
}
|
|
else if(cmd=="STATUS" || cmd=="FSTATUS" )
|
|
{
|
|
CMD_STATUS(cmd); continue;
|
|
}
|
|
else if(cmd=="STATUS DETAIL")
|
|
{
|
|
CMD_STATUS_DETAIL(cmd); continue;
|
|
}
|
|
else if(cmd=="START BACKUP INCR" )
|
|
{
|
|
CMD_TOCHANNEL_START_INCR_FILEBACKUP(cmd, params); continue;
|
|
}
|
|
else if(cmd=="START BACKUP FULL" )
|
|
{
|
|
CMD_TOCHANNEL_START_FULL_FILEBACKUP(cmd, params); continue;
|
|
}
|
|
else if(cmd=="START IMAGE FULL" )
|
|
{
|
|
CMD_TOCHANNEL_START_FULL_IMAGEBACKUP(cmd, params); continue;
|
|
}
|
|
else if(cmd=="START IMAGE INCR" )
|
|
{
|
|
CMD_TOCHANNEL_START_INCR_IMAGEBACKUP(cmd, params); continue;
|
|
}
|
|
else if(next(cmd, 0, "PAUSE ") )
|
|
{
|
|
CMD_PAUSE(cmd); continue;
|
|
}
|
|
else if( cmd.find("GET BACKUPCLIENTS")==0 )
|
|
{
|
|
CMD_RESTORE_GET_BACKUPCLIENTS(cmd); continue;
|
|
}
|
|
else if( cmd.find("GET BACKUPIMAGES ")==0 )
|
|
{
|
|
CMD_RESTORE_GET_BACKUPIMAGES(cmd); continue;
|
|
}
|
|
else if( cmd=="GET FILE BACKUPS" )
|
|
{
|
|
CMD_RESTORE_GET_FILE_BACKUPS(cmd); continue;
|
|
}
|
|
else if( cmd=="GET FILE BACKUPS TOKENS")
|
|
{
|
|
CMD_RESTORE_GET_FILE_BACKUPS_TOKENS(cmd, params); continue;
|
|
}
|
|
else if(cmd=="GET FILE LIST TOKENS")
|
|
{
|
|
CMD_GET_FILE_LIST_TOKENS(cmd, params); continue;
|
|
}
|
|
else if(cmd=="DOWNLOAD FILES TOKENS")
|
|
{
|
|
CMD_DOWNLOAD_FILES_TOKENS(cmd, params); continue;
|
|
}
|
|
else if( cmd=="LOGIN FOR DOWNLOAD" )
|
|
{
|
|
CMD_RESTORE_LOGIN_FOR_DOWNLOAD(cmd, params); continue;
|
|
}
|
|
else if( cmd=="GET SALT" )
|
|
{
|
|
CMD_RESTORE_GET_SALT(cmd, params); continue;
|
|
}
|
|
else if( cmd=="DOWNLOAD IMAGE" )
|
|
{
|
|
CMD_RESTORE_DOWNLOAD_IMAGE(cmd, params); continue;
|
|
}
|
|
else if( cmd=="DOWNLOAD FILES" )
|
|
{
|
|
CMD_RESTORE_DOWNLOAD_FILES(cmd, params); continue;
|
|
}
|
|
else if( cmd=="GET DOWNLOADPROGRESS" )
|
|
{
|
|
CMD_RESTORE_DOWNLOADPROGRESS(cmd); continue;
|
|
}
|
|
else if( cmd=="GET ACCESS PARAMETERS")
|
|
{
|
|
CMD_GET_ACCESS_PARAMS(params); continue;
|
|
}
|
|
else if( cmd=="RESTORE OK")
|
|
{
|
|
CMD_RESTORE_OK(params); continue;
|
|
}
|
|
}
|
|
if(is_channel) //Channel commands from server
|
|
{
|
|
if(cmd=="PONG" )
|
|
{
|
|
CMD_CHANNEL_PONG(cmd, endpoint_name); continue;
|
|
}
|
|
else if(cmd=="PING" )
|
|
{
|
|
CMD_CHANNEL_PING(cmd, endpoint_name); continue;
|
|
}
|
|
}
|
|
|
|
tcpstack.Send(pipe, "ERR");
|
|
}
|
|
}
|
|
|
|
bool ClientConnector::checkPassword(const std::string &pw, bool& change_pw)
|
|
{
|
|
static std::string stored_pw=getFile(pw_file);
|
|
static std::string stored_pw_change=getFile(pw_change_file);
|
|
std::string utf8_pw = (pw);
|
|
if(stored_pw_change==utf8_pw)
|
|
{
|
|
change_pw=true;
|
|
return true;
|
|
}
|
|
if(stored_pw==utf8_pw)
|
|
{
|
|
change_pw=false;
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
std::string ClientConnector::removeIllegalCharsFromBackupName(std::string in)
|
|
{
|
|
char illegalchars[] = {'*', ':', '/' , '\\'};
|
|
std::string ret;
|
|
for(size_t i=0;i<in.size();++i)
|
|
{
|
|
bool found=false;
|
|
for(size_t j=0;j<sizeof(illegalchars)/sizeof(illegalchars[0]);++j)
|
|
{
|
|
if(illegalchars[j]==in[i])
|
|
{
|
|
found=true;
|
|
break;
|
|
}
|
|
}
|
|
if(!found)
|
|
{
|
|
ret+=in[i];
|
|
}
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
bool ClientConnector::saveBackupDirs(str_map &args, bool server_default, int group_offset, int facet_id)
|
|
{
|
|
IDatabase *db=Server->getDatabase(Server->getThreadID(), URBACKUPDB_CLIENT);
|
|
db->BeginWriteTransaction();
|
|
db_results backupdirs=db->Read("SELECT name, path FROM backupdirs WHERE facet="+convert(facet_id));
|
|
bool all_virtual_clients = false;
|
|
if (args.find("all_virtual_clients") != args.end())
|
|
{
|
|
all_virtual_clients = true;
|
|
if (!server_default)
|
|
{
|
|
db->Write("DELETE FROM backupdirs WHERE symlinked=0 AND server_default=0 AND facet="+convert(facet_id));
|
|
}
|
|
else
|
|
{
|
|
db->Write("DELETE FROM backupdirs WHERE symlinked=0 AND server_default!=2 AND facet="+convert(facet_id));
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (!server_default)
|
|
{
|
|
db->Write("DELETE FROM backupdirs WHERE symlinked=0 AND server_default=0 AND tgroup BETWEEN "
|
|
+ convert(group_offset) + " AND " + convert(group_offset + c_group_max)+ " AND facet="+convert(facet_id));
|
|
}
|
|
else
|
|
{
|
|
db->Write("DELETE FROM backupdirs WHERE symlinked=0 AND server_default!=2 AND tgroup BETWEEN "
|
|
+ convert(group_offset) + " AND " + convert(group_offset + c_group_max)+ " AND facet="+convert(facet_id));
|
|
}
|
|
}
|
|
IQuery *q_insert_dir=db->Prepare("INSERT INTO backupdirs (name, path, server_default, optional, tgroup, facet) VALUES (?, ? , ?, ?, ?, ?)");
|
|
/**
|
|
Use empty client settings
|
|
if(server_default==false)
|
|
{
|
|
q->Bind(L"*"); q->Bind(L"*"); q->Bind(1);
|
|
q->Write();
|
|
q->Reset();
|
|
}
|
|
*/
|
|
IQuery *q2=db->Prepare("SELECT id FROM backupdirs WHERE name=? AND facet="+convert(facet_id));
|
|
IQuery* q_get_virtual_client_offset = db->Prepare("SELECT group_offset FROM virtual_client_group_offsets WHERE virtual_client=? AND facet="+convert(facet_id));
|
|
std::string dir;
|
|
size_t i=0;
|
|
std::vector<SBackupDir> new_watchdirs;
|
|
std::vector<std::string> all_backupdirs;
|
|
do
|
|
{
|
|
dir=args["dir_"+convert(i)];
|
|
if(!dir.empty())
|
|
{
|
|
all_backupdirs.push_back(dir);
|
|
|
|
std::string name;
|
|
str_map::iterator name_arg=args.find("dir_"+convert(i)+"_name");
|
|
if(name_arg!=args.end() && !name_arg->second.empty())
|
|
name=name_arg->second;
|
|
else
|
|
name=ExtractFileName(dir);
|
|
|
|
int curr_offset = group_offset;
|
|
|
|
str_map::iterator virtual_client_arg = args.find("dir_" + convert(i) + "_virtual_client");
|
|
if (virtual_client_arg != args.end() && !virtual_client_arg->second.empty())
|
|
{
|
|
q_get_virtual_client_offset->Bind(virtual_client_arg->second);
|
|
db_results res_offset = q_get_virtual_client_offset->Read();
|
|
q_get_virtual_client_offset->Reset();
|
|
if (!res_offset.empty())
|
|
{
|
|
curr_offset = watoi(res_offset[0]["group_offset"]);
|
|
}
|
|
}
|
|
|
|
int group = curr_offset + c_group_default;
|
|
|
|
str_map::iterator group_arg=args.find("dir_"+convert(i)+"_group");
|
|
if(group_arg!=args.end() && !group_arg->second.empty())
|
|
group= curr_offset + watoi(group_arg->second);
|
|
|
|
int flags = EBackupDirFlag_FollowSymlinks | EBackupDirFlag_SymlinksOptional | EBackupDirFlag_ShareHashes; //default flags
|
|
size_t flags_off = name.find("/");
|
|
if(flags_off!=std::string::npos)
|
|
{
|
|
flags=0;
|
|
|
|
std::vector<std::string> str_flags;
|
|
Tokenize(getafter("/", name), str_flags, ",;");
|
|
|
|
for(size_t i=0;i<str_flags.size();++i)
|
|
{
|
|
std::string flag = strlower(trim(str_flags[i]));
|
|
if(flag=="optional")
|
|
{
|
|
flags |= EBackupDirFlag_Optional;
|
|
}
|
|
else if(flag=="follow_symlinks"
|
|
|| flag=="follow_symbolic_links")
|
|
{
|
|
flags |= EBackupDirFlag_FollowSymlinks;
|
|
}
|
|
else if(flag=="symlinks_optional"
|
|
|| flag=="symbolic_links_optional")
|
|
{
|
|
flags |= EBackupDirFlag_SymlinksOptional;
|
|
}
|
|
else if(flag=="one_filesystem" || flag=="one_fs")
|
|
{
|
|
flags |= EBackupDirFlag_OneFilesystem;
|
|
}
|
|
else if (flag == "require_snapshot" || flag=="require_shadowcopy" )
|
|
{
|
|
flags |= EBackupDirFlag_RequireSnapshot;
|
|
}
|
|
else if (flag == "share_hashes")
|
|
{
|
|
flags |= EBackupDirFlag_ShareHashes;
|
|
}
|
|
else if (flag == "keep" || flag == "keep_files")
|
|
{
|
|
flags |= EBackupDirFlag_KeepFiles;
|
|
}
|
|
else if (flag == "required" || flag == "require")
|
|
{
|
|
flags |= EBackupDirFlag_Required;
|
|
}
|
|
else if (flag == "include_dir_symlinks"
|
|
|| flag=="include_directory_symlinks"
|
|
|| flag=="include_directory_symbolic_links")
|
|
{
|
|
flags |= EBackupDirFlag_IncludeDirectorySymlinks;
|
|
}
|
|
}
|
|
name.resize(flags_off);
|
|
}
|
|
|
|
name=removeIllegalCharsFromBackupName(name);
|
|
|
|
if(dir[dir.size()-1]=='\\' || dir[dir.size()-1]=='/' )
|
|
{
|
|
dir.erase(dir.size()-1,1);
|
|
#ifndef _WIN32
|
|
if(dir.empty())
|
|
{
|
|
dir="/";
|
|
if(name.empty())
|
|
{
|
|
name="rootfs";
|
|
}
|
|
}
|
|
#endif
|
|
}
|
|
|
|
q2->Bind(name);
|
|
std::string orig_name = name;
|
|
if(q2->Read().empty()==false
|
|
|| name=="urbackup")
|
|
{
|
|
for(int k=0;k<100;++k)
|
|
{
|
|
q2->Reset();
|
|
q2->Bind(name+"_"+convert(k));
|
|
if(q2->Read().empty()==true)
|
|
{
|
|
name+="_"+convert(k);
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (orig_name == name)
|
|
{
|
|
name += "_" + Server->secureRandomString(16);
|
|
}
|
|
}
|
|
q2->Reset();
|
|
|
|
bool found=false;
|
|
for(size_t i=0;i<backupdirs.size();++i)
|
|
{
|
|
if(backupdirs[i]["path"]==dir)
|
|
{
|
|
backupdirs[i]["need"]="1";
|
|
backupdirs[i]["group"] = convert(group);
|
|
found=true;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if(!found)
|
|
{
|
|
//It's not already watched. Add it
|
|
SBackupDir new_dir = {
|
|
0, name, dir, flags, group };
|
|
|
|
new_watchdirs.push_back(new_dir);
|
|
}
|
|
|
|
int curr_server_default = 0;
|
|
|
|
if (server_default)
|
|
{
|
|
str_map::iterator server_default_arg = args.find("dir_" + convert(i) + "_server_default");
|
|
if (server_default_arg != args.end())
|
|
{
|
|
curr_server_default = watoi(server_default_arg->second);
|
|
}
|
|
}
|
|
|
|
q_insert_dir->Bind(name);
|
|
q_insert_dir->Bind(dir);
|
|
q_insert_dir->Bind(curr_server_default);
|
|
q_insert_dir->Bind(flags);
|
|
q_insert_dir->Bind(group);
|
|
q_insert_dir->Bind(facet_id);
|
|
q_insert_dir->Write();
|
|
q_insert_dir->Reset();
|
|
}
|
|
++i;
|
|
}
|
|
while(!dir.empty());
|
|
db->EndTransaction();
|
|
|
|
#ifdef _WIN32
|
|
for(size_t i=0;i<new_watchdirs.size();++i)
|
|
{
|
|
//Add watch
|
|
CWData data;
|
|
data.addChar(IndexThread::IndexThreadAction_AddWatchdir);
|
|
data.addUInt(0);
|
|
data.addString(new_watchdirs[i].path);
|
|
if(new_watchdirs[i].group==c_group_continuous)
|
|
{
|
|
data.addString(new_watchdirs[i].tname);
|
|
}
|
|
IndexThread::getMsgPipe()->Write(data.getDataPtr(), data.getDataSize());
|
|
}
|
|
|
|
IQuery* q_other_has_path = NULL;
|
|
if (!all_virtual_clients)
|
|
{
|
|
q_other_has_path = db->Prepare("SELECT id FROM backupdirs WHERE path=? AND tgroup NOT BETWEEN " + convert(group_offset) + " AND " + convert(group_offset + c_group_max)+" AND facet="+convert(facet_id));
|
|
}
|
|
|
|
IQuery* q_other_facet_has_path = db->Prepare("SELECT id FROM backupdirs WHERE path=? AND facet!=" + convert(facet_id));
|
|
|
|
for(size_t i=0;i<backupdirs.size();++i)
|
|
{
|
|
if(backupdirs[i]["need"]!="1" && backupdirs[i]["path"]!="*")
|
|
{
|
|
if (!all_virtual_clients)
|
|
{
|
|
q_other_has_path->Bind(backupdirs[i]["path"]);
|
|
db_results res = q_other_has_path->Read();
|
|
q_other_has_path->Reset();
|
|
|
|
if (!res.empty())
|
|
continue;
|
|
}
|
|
|
|
q_other_facet_has_path->Bind(backupdirs[i]["path"]);
|
|
db_results res_other_f = q_other_facet_has_path->Read();
|
|
q_other_facet_has_path->Reset();
|
|
if (!res_other_f.empty())
|
|
continue;
|
|
|
|
//Delete the watch
|
|
CWData data;
|
|
data.addChar(IndexThread::IndexThreadAction_RemoveWatchdir);
|
|
data.addUInt(0);
|
|
data.addString(backupdirs[i]["path"]);
|
|
if(watoi(backupdirs[i]["group"])%c_group_size==c_group_continuous)
|
|
{
|
|
data.addString(backupdirs[i]["name"]);
|
|
}
|
|
IndexThread::getMsgPipe()->Write(data.getDataPtr(), data.getDataSize());
|
|
}
|
|
}
|
|
|
|
if (!new_watchdirs.empty())
|
|
{
|
|
CWData data;
|
|
data.addChar(IndexThread::IndexThreadAction_UpdateCbt);
|
|
IndexThread::getMsgPipe()->Write(data.getDataPtr(), data.getDataSize());
|
|
}
|
|
|
|
if(Server->fileExists(Server->getServerWorkingDir()+"\\UrBackupClient.exe"))
|
|
{
|
|
{
|
|
bool deleted_key;
|
|
size_t i=0;
|
|
do
|
|
{
|
|
deleted_key=false;
|
|
if(RegDeleteKeyW(HKEY_CLASSES_ROOT, Server->ConvertToWchar("AllFilesystemObjects\\shell\\urbackup.access." + convert(i)+"\\Command").c_str())==ERROR_SUCCESS
|
|
&& RegDeleteKeyW(HKEY_CLASSES_ROOT, Server->ConvertToWchar("AllFilesystemObjects\\shell\\urbackup.access."+convert(i)).c_str())==ERROR_SUCCESS)
|
|
{
|
|
deleted_key=true;
|
|
}
|
|
++i;
|
|
} while (deleted_key);
|
|
}
|
|
|
|
|
|
std::wstring mui_text=L"&Access backups";
|
|
std::string read_mui_text=getFile("access_backups_shell_mui.txt");
|
|
if( !read_mui_text.empty() && read_mui_text.find("@")==std::string::npos)
|
|
{
|
|
mui_text=Server->ConvertToWchar(read_mui_text);
|
|
}
|
|
|
|
/**
|
|
* The Advanced Query Syntax (AQL) in appliesTo seems to have a length restriction.
|
|
* That's why each entry for each folder is added separately instead of using "OR".
|
|
*/
|
|
for(size_t i=0;i<all_backupdirs.size();++i)
|
|
{
|
|
HKEY urbackup_access;
|
|
if(RegCreateKeyExW(HKEY_CLASSES_ROOT, Server->ConvertToWchar("AllFilesystemObjects\\shell\\urbackup.access."+convert(i)).c_str(), 0, NULL, 0, KEY_ALL_ACCESS, NULL, &urbackup_access, NULL)==ERROR_SUCCESS)
|
|
{
|
|
HKEY urbackup_access_command;
|
|
if(RegCreateKeyExW(HKEY_CLASSES_ROOT, Server->ConvertToWchar("AllFilesystemObjects\\shell\\urbackup.access."+convert(i)+"\\Command").c_str(), 0, NULL, 0, KEY_ALL_ACCESS, NULL, &urbackup_access_command, NULL)!=ERROR_SUCCESS)
|
|
{
|
|
Server->Log("Error creating command registry sub-key", LL_ERROR);
|
|
}
|
|
else
|
|
{
|
|
std::wstring cmd=Server->ConvertToWchar("\""+Server->getServerWorkingDir()+"\\UrBackupClient.exe\" access \"%1\"");
|
|
//cmd = greplace(L"\\", L"\\\\", cmd);
|
|
if(RegSetValueExW(urbackup_access_command, NULL, 0, REG_SZ, reinterpret_cast<const BYTE*>(cmd.c_str()), static_cast<DWORD>(cmd.size()*sizeof(wchar_t)))!=ERROR_SUCCESS)
|
|
{
|
|
Server->Log("Error setting command in registry", LL_ERROR);
|
|
}
|
|
}
|
|
|
|
if(RegSetValueExW(urbackup_access, L"MUIVerb", 0, REG_SZ, reinterpret_cast<const BYTE*>(mui_text.c_str()), static_cast<DWORD>(mui_text.size()*sizeof(wchar_t)))!=ERROR_SUCCESS)
|
|
{
|
|
Server->Log("Error setting MUIVerb in registry", LL_ERROR);
|
|
}
|
|
|
|
std::wstring icon_path = Server->ConvertToWchar(Server->getServerWorkingDir()+"\\backup-ok.ico");
|
|
|
|
if(RegSetValueExW(urbackup_access, L"Icon", 0, REG_SZ, reinterpret_cast<const BYTE*>(icon_path.c_str()), static_cast<DWORD>(icon_path.size()*sizeof(wchar_t)))!=ERROR_SUCCESS)
|
|
{
|
|
Server->Log("Error setting Icon in registry", LL_ERROR);
|
|
}
|
|
|
|
std::string path = greplace("/", "\\", all_backupdirs[i]);
|
|
|
|
std::wstring applies_to=Server->ConvertToWchar("System.ParsingPath:~<\""+path+"\"");
|
|
|
|
if(RegSetValueExW(urbackup_access, L"AppliesTo", 0, REG_SZ, reinterpret_cast<const BYTE*>(applies_to.c_str()), static_cast<DWORD>(applies_to.size()*sizeof(wchar_t)))!=ERROR_SUCCESS)
|
|
{
|
|
Server->Log("Error setting AppliesTo in registry", LL_ERROR);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
#endif
|
|
|
|
if (updateDefaultDirsSetting(db, all_virtual_clients, group_offset, args.find("enable_client_paths_use")!=args.end(), facet_id))
|
|
{
|
|
IScopedLock lock(backup_mutex);
|
|
for (size_t o = 0; o<channel_pipes.size(); ++o)
|
|
{
|
|
CTCPStack tmpstack(channel_pipes[o].internet_connection);
|
|
tmpstack.Send(channel_pipes[o].pipe, "UPDATE SETTINGS");
|
|
}
|
|
}
|
|
|
|
db->destroyAllQueries();
|
|
return true;
|
|
}
|
|
|
|
std::string ClientConnector::replaceChars(std::string in)
|
|
{
|
|
char legalchars[] = {'_', '-'};
|
|
for(size_t i=0;i<in.size();++i)
|
|
{
|
|
bool found=false;
|
|
for(size_t j=0;j<sizeof(legalchars);++j)
|
|
{
|
|
if(legalchars[j]==in[i])
|
|
{
|
|
found=true;
|
|
break;
|
|
}
|
|
}
|
|
if( !isletter(in[i]) && !str_isnumber(in[i]) && !found )
|
|
{
|
|
in[i]='_';
|
|
}
|
|
}
|
|
return in;
|
|
}
|
|
|
|
void ClientConnector::updateLastBackup(void)
|
|
{
|
|
IDatabase *db=Server->getDatabase(Server->getThreadID(), URBACKUPDB_CLIENT);
|
|
IQuery *q=db->Prepare("UPDATE status SET last_backup=CURRENT_TIMESTAMP WHERE id=1");
|
|
IQuery *q2=db->Prepare("SELECT last_backup FROM status WHERE id=1");
|
|
if(q!=nullptr && q2!=nullptr)
|
|
{
|
|
if(q2->Read().size()>0)
|
|
{
|
|
q->Write();
|
|
}
|
|
else
|
|
{
|
|
q=db->Prepare("INSERT INTO status (last_backup, id) VALUES (CURRENT_TIMESTAMP, 1)");
|
|
if(q!=nullptr)
|
|
q->Write();
|
|
}
|
|
}
|
|
db->destroyAllQueries();
|
|
}
|
|
|
|
std::vector<std::string> getSettingsList(void);
|
|
|
|
void ClientConnector::updateSettings(const std::string &pData, const std::string& server_identity)
|
|
{
|
|
IDatabase *db = Server->getDatabase(Server->getThreadID(), URBACKUPDB_CLIENT);
|
|
std::unique_ptr<ISettingsReader> new_settings(Server->createMemorySettingsReader(pData));
|
|
|
|
int facet_id = getFacetId(ServerIdentityMgr::getIdentityFromSessionIdentity(server_identity));
|
|
std::string facet_dir = "data_" + convert(facet_id);
|
|
|
|
std::string settings_fn="urbackup/"+ facet_dir +"/settings.cfg";
|
|
std::string clientsubname;
|
|
std::string str_group_offset;
|
|
int group_offset=0;
|
|
if(new_settings->getValue("clientsubname", &clientsubname) && !clientsubname.empty()
|
|
&& new_settings->getValue("filebackup_group_offset", &str_group_offset))
|
|
{
|
|
settings_fn = "urbackup/"+ facet_dir +"/settings_"+conv_filename(clientsubname)+".cfg";
|
|
group_offset = atoi(str_group_offset.c_str());
|
|
|
|
db_results res_old_client = db->Read("SELECT virtual_client FROM virtual_client_group_offsets WHERE group_offset=" + convert(group_offset));
|
|
if (!res_old_client.empty()
|
|
&& res_old_client[0]["virtual_client"]!=clientsubname)
|
|
{
|
|
db->Write("DELETE FROM backupdirs WHERE tgroup=" + convert(group_offset));
|
|
db->Write("DELETE FROM virtual_client_group_offsets WHERE group_offset=" + convert(group_offset));
|
|
}
|
|
IQuery* q = db->Prepare("INSERT OR REPLACE INTO virtual_client_group_offsets (virtual_client, group_offset) VALUES (?,?)", false);
|
|
q->Bind(clientsubname);
|
|
q->Bind(group_offset);
|
|
q->Write();
|
|
q->Reset();
|
|
db->destroyQuery(q);
|
|
}
|
|
|
|
std::string ransomware_canary_paths;
|
|
std::string server_token;
|
|
if (new_settings->getValue("ransomware_canary_paths", &ransomware_canary_paths)
|
|
&& !ransomware_canary_paths.empty()
|
|
&& new_settings->getValue("server_token", &server_token)
|
|
&& !server_token.empty())
|
|
{
|
|
setupRansomwareCanaries(ransomware_canary_paths, server_token,
|
|
facet_id, group_offset);
|
|
}
|
|
|
|
std::unique_ptr<ISettingsReader> curr_settings(Server->createFileSettingsReader(settings_fn));
|
|
|
|
std::vector<std::string> critical_settings;
|
|
critical_settings.push_back("internet_mode_enabled");
|
|
critical_settings.push_back("internet_server");
|
|
critical_settings.push_back("internet_server_port");
|
|
critical_settings.push_back("internet_server_proxy");
|
|
critical_settings.push_back("internet_authkey");
|
|
critical_settings.push_back("computername");
|
|
|
|
bool mod = false;
|
|
|
|
std::string settings_add="\n";
|
|
for (size_t i = 0; i < critical_settings.size(); ++i)
|
|
{
|
|
std::string curr_v;
|
|
std::string new_v;
|
|
if (curr_settings->getValue(critical_settings[i], &curr_v)
|
|
&& !new_settings->getValue(critical_settings[i], &new_v) )
|
|
{
|
|
settings_add += critical_settings[i] + "=" + curr_v + "\n";
|
|
mod = true;
|
|
}
|
|
else if (curr_v != new_v && new_v.empty()
|
|
&& critical_settings[i] != "internet_server_proxy")
|
|
{
|
|
settings_add += critical_settings[i] + "=" + curr_v + "\n";
|
|
mod = true;
|
|
}
|
|
}
|
|
|
|
|
|
int default_dirs_use = new_settings->getValue("default_dirs.use", 0);
|
|
std::vector<std::string> default_dirs_toks;
|
|
size_t default_dirs_client_off = std::string::npos;
|
|
|
|
if (default_dirs_use & c_use_group)
|
|
{
|
|
std::string val;
|
|
Tokenize(new_settings->getValue("default_dirs.group", ""), default_dirs_toks, ";");
|
|
}
|
|
if (default_dirs_use & c_use_value)
|
|
{
|
|
std::string val;
|
|
std::vector<std::string> toks;
|
|
Tokenize(new_settings->getValue("default_dirs.home", ""), toks, ";");
|
|
default_dirs_toks.insert(default_dirs_toks.end(), toks.begin(), toks.end());
|
|
}
|
|
if (default_dirs_use & c_use_value_client)
|
|
{
|
|
std::string val;
|
|
std::vector<std::string> toks;
|
|
Tokenize(new_settings->getValue("default_dirs.client", ""), toks, ";");
|
|
default_dirs_client_off = default_dirs_toks.size();
|
|
default_dirs_toks.insert(default_dirs_toks.end(), toks.begin(), toks.end());
|
|
}
|
|
|
|
if(!default_dirs_toks.empty())
|
|
{
|
|
str_map args;
|
|
for(size_t i=0;i<default_dirs_toks.size();++i)
|
|
{
|
|
std::string path = UnescapeParamString(trim(default_dirs_toks[i]));
|
|
std::string name;
|
|
int group = c_group_default;
|
|
if(path.find("|")!=std::string::npos)
|
|
{
|
|
std::vector<std::string> toks;
|
|
Tokenize(path, toks, "|");
|
|
path = toks[0];
|
|
name = toks[1];
|
|
if(toks.size()>2)
|
|
{
|
|
group = (std::min)(c_group_max, (std::max)(0, watoi(toks[2])));
|
|
}
|
|
}
|
|
args["dir_"+convert(i)]=path;
|
|
if(!name.empty())
|
|
args["dir_"+convert(i)+"_name"]=name;
|
|
|
|
args["dir_"+convert(i)+"_group"]=convert(group);
|
|
args["dir_" + convert(i) + "_server_default"] = convert(i>= default_dirs_client_off ? 0 : 1);
|
|
}
|
|
|
|
saveBackupDirs(args, true, group_offset, facet_id);
|
|
}
|
|
|
|
if(mod
|
|
|| getFile(settings_fn)!= pData)
|
|
{
|
|
std::unique_ptr<IFile> newf(Server->openFile(settings_fn + ".new", MODE_WRITE));
|
|
if (newf.get() != nullptr
|
|
&& newf->Write(pData+ settings_add) == pData.size()+settings_add.size()
|
|
&& newf->Sync())
|
|
{
|
|
newf.reset();
|
|
os_rename_file(settings_fn + ".new", settings_fn);
|
|
InternetClient::updateSettings();
|
|
}
|
|
|
|
CWData data;
|
|
data.addChar(IndexThread::IndexThreadAction_UpdateCbt);
|
|
IndexThread::getMsgPipe()->Write(data.getDataPtr(), data.getDataSize());
|
|
}
|
|
}
|
|
|
|
void ClientConnector::replaceSettings(const std::string &pData)
|
|
{
|
|
std::unique_ptr<ISettingsReader> new_settings(Server->createMemorySettingsReader(pData));
|
|
|
|
std::string facet_name = new_settings->getValue("facet_name", "default");
|
|
|
|
int facet_id = getFacetIdByName(facet_name);
|
|
|
|
if (facet_id == 0)
|
|
{
|
|
Server->Log("Facet \"" + facet_name + "\" not found. Cannot update settings.", LL_ERROR);
|
|
return;
|
|
}
|
|
|
|
std::string ncname=new_settings->getValue("computername", "");
|
|
if(!ncname.empty() && ncname!=IndexThread::getFileSrv()->getServerName())
|
|
{
|
|
Server->Log("Restarting filesrv because of name change. Old name: " + IndexThread::getFileSrv()->getServerName() + " New name: " + ncname, LL_DEBUG);
|
|
|
|
CWData data;
|
|
data.addChar(IndexThread::IndexThreadAction_RestartFilesrv);
|
|
data.addUInt(0);
|
|
IndexThread::getMsgPipe()->Write(data.getDataPtr(), data.getDataSize());
|
|
}
|
|
|
|
std::string settings_fn="urbackup/data_"+convert(facet_id )+"/settings.cfg";
|
|
std::string clientsubname;
|
|
if(new_settings->getValue("clientsubname", &clientsubname) && !clientsubname.empty())
|
|
{
|
|
settings_fn = "urbackup/data_"+convert(facet_id)+"/settings_"+conv_filename(clientsubname)+".cfg";
|
|
}
|
|
|
|
std::unique_ptr<ISettingsReader> old_settings(Server->createFileSettingsReader(settings_fn));
|
|
|
|
std::vector<std::string> new_keys = new_settings->getKeys();
|
|
bool modified_settings=true;
|
|
if(old_settings.get()!=nullptr)
|
|
{
|
|
modified_settings=false;
|
|
std::vector<std::string> old_keys = old_settings->getKeys();
|
|
|
|
for(size_t i=0;i<old_keys.size();++i)
|
|
{
|
|
std::string old_val;
|
|
std::string new_val;
|
|
if( old_settings->getValue(old_keys[i], &old_val) &&
|
|
(!new_settings->getValue(old_keys[i], &new_val) ||
|
|
old_val!=new_val ) )
|
|
{
|
|
modified_settings=true;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if(!modified_settings)
|
|
{
|
|
for(size_t i=0;i<new_keys.size();++i)
|
|
{
|
|
std::string old_val;
|
|
if(!old_settings->getValue(new_keys[i], &old_val))
|
|
{
|
|
modified_settings=true;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if(modified_settings)
|
|
{
|
|
std::string new_data;
|
|
|
|
for(size_t i=0;i<new_keys.size();++i)
|
|
{
|
|
std::string val;
|
|
if(new_settings->getValue(new_keys[i], &val))
|
|
{
|
|
new_data+=new_keys[i]+"="+val+"\n";
|
|
}
|
|
}
|
|
|
|
if (old_settings.get() != nullptr)
|
|
{
|
|
std::vector<std::string> old_keys = old_settings->getKeys();
|
|
|
|
for (size_t i = 0; i < old_keys.size(); ++i)
|
|
{
|
|
if (std::find(new_keys.begin(), new_keys.end(), old_keys[i]) == new_keys.end())
|
|
{
|
|
std::string val;
|
|
if (old_settings->getValue(old_keys[i], &val))
|
|
{
|
|
new_data += old_keys[i] + "=" + val + "\n";
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
InternetClient::updateSettings();
|
|
|
|
writestring(new_data, settings_fn);
|
|
|
|
CWData data;
|
|
data.addChar(IndexThread::IndexThreadAction_UpdateCbt);
|
|
IndexThread::getMsgPipe()->Write(data.getDataPtr(), data.getDataSize());
|
|
}
|
|
}
|
|
|
|
void ClientConnector::saveLogdata(const std::string &created, const std::string &pData)
|
|
{
|
|
IDatabase *db=Server->getDatabase(Server->getThreadID(), URBACKUPDB_CLIENT);
|
|
IQuery *q_p=db->Prepare("INSERT INTO logs (ttime) VALUES (datetime(?, 'unixepoch'))");
|
|
q_p->Bind(created);
|
|
q_p->Write();
|
|
_i64 logid=db->getLastInsertID();
|
|
|
|
while(!db->BeginWriteTransaction())
|
|
Server->wait(500);
|
|
|
|
IQuery *q=db->Prepare("INSERT INTO logdata (logid, loglevel, message, idx, ltime) VALUES (?, ?, ?, ?, datetime(?, 'unixepoch'))");
|
|
|
|
std::vector<std::string> lines;
|
|
Tokenize(pData, lines, "\n");
|
|
for(size_t i=0,lc=lines.size();i<lc;++i)
|
|
{
|
|
std::string l=lines[i];
|
|
int loglevel=atoi(getuntil("-",l).c_str());
|
|
std::string u_msg=getafter("-", l);
|
|
unsigned int ltime=0;
|
|
if(u_msg.find("-")!=std::string::npos)
|
|
{
|
|
std::string s_ltime=getuntil("-", u_msg);
|
|
bool isnum=true;
|
|
for(size_t j=0;j<s_ltime.size();++j)
|
|
{
|
|
if(!str_isnumber(s_ltime[j]))
|
|
{
|
|
isnum=false;
|
|
break;
|
|
}
|
|
}
|
|
if(isnum)
|
|
{
|
|
ltime=atoi(s_ltime.c_str());
|
|
u_msg=getafter("-", u_msg);
|
|
}
|
|
}
|
|
std::string msg=(u_msg);
|
|
|
|
q->Bind(logid);
|
|
q->Bind(loglevel);
|
|
q->Bind(msg);
|
|
q->Bind(i);
|
|
q->Bind(ltime);
|
|
q->Write();
|
|
q->Reset();
|
|
}
|
|
db->EndTransaction();
|
|
db->destroyAllQueries();
|
|
}
|
|
|
|
std::string ClientConnector::getLogpoints(void)
|
|
{
|
|
IDatabase *db=Server->getDatabase(Server->getThreadID(), URBACKUPDB_CLIENT);
|
|
int timeoutms=300;
|
|
IQuery *q=db->Prepare("SELECT id, strftime('%s',ttime) AS ltime FROM logs ORDER BY ttime DESC LIMIT 100");
|
|
db_results res=q->Read(&timeoutms);
|
|
std::string ret;
|
|
for(size_t i=0;i<res.size();++i)
|
|
{
|
|
ret+=(res[i]["id"])+"-";
|
|
ret+=(res[i]["ltime"])+"\n";
|
|
}
|
|
db->destroyAllQueries();
|
|
return ret;
|
|
}
|
|
|
|
void ClientConnector::getLogLevel(int logid, int loglevel, std::string &data)
|
|
{
|
|
IDatabase *db=Server->getDatabase(Server->getThreadID(), URBACKUPDB_CLIENT);
|
|
IQuery *q=db->Prepare("SELECT loglevel, message FROM logdata WHERE logid=? AND loglevel>=? ORDER BY idx ASC");
|
|
q->Bind(logid);
|
|
q->Bind(loglevel);
|
|
int timeoutms=300;
|
|
db_results res=q->Read(&timeoutms);
|
|
for(size_t i=0;i<res.size();++i)
|
|
{
|
|
data+=(res[i]["loglevel"])+"-";
|
|
data+=(res[i]["message"])+"\n";
|
|
}
|
|
db->destroyAllQueries();
|
|
}
|
|
|
|
bool ClientConnector::sendFullImage(void)
|
|
{
|
|
image_inf.thread_action=TA_FULL_IMAGE;
|
|
image_inf.image_thread=new ImageThread(this, pipe, curr_result_id, &image_inf, server_token, hashdatafile, nullptr);
|
|
|
|
IScopedLock lock(backup_mutex);
|
|
|
|
{
|
|
IScopedLock process_lock(process_mutex);
|
|
SRunningProcess* cproc = getRunningBackupProcess(server_token, image_inf.server_status_id);
|
|
if (cproc != nullptr
|
|
&& cproc->action == RUNNING_FULL_IMAGE)
|
|
{
|
|
local_backup_running_id = cproc->id;
|
|
cproc->last_pingtime = Server->getTimeMS();
|
|
++cproc->refs;
|
|
}
|
|
else
|
|
{
|
|
SRunningProcess new_proc;
|
|
new_proc.action = RUNNING_FULL_IMAGE;
|
|
new_proc.server_token = server_token;
|
|
new_proc.pcdone = 0;
|
|
new_proc.details = image_inf.orig_image_letter;
|
|
new_proc.server_id = image_inf.server_status_id;
|
|
new_proc.refs = 1;
|
|
local_backup_running_id = addNewProcess(new_proc);
|
|
}
|
|
|
|
removeTimedOutProcesses(server_token, false);
|
|
}
|
|
|
|
status_updated = true;
|
|
image_inf.running_process_id = local_backup_running_id;
|
|
image_inf.thread_ticket=Server->getThreadPool()->execute(image_inf.image_thread, "full image upload");
|
|
state=CCSTATE_IMAGE;
|
|
|
|
return true;
|
|
}
|
|
|
|
bool ClientConnector::sendIncrImage(void)
|
|
{
|
|
image_inf.thread_action=TA_INCR_IMAGE;
|
|
image_inf.image_thread=new ImageThread(this, pipe, curr_result_id, &image_inf, server_token, hashdatafile, bitmapfile);
|
|
|
|
IScopedLock lock(backup_mutex);
|
|
|
|
{
|
|
IScopedLock process_lock(process_mutex);
|
|
|
|
SRunningProcess* cproc = getRunningBackupProcess(server_token, image_inf.server_status_id);
|
|
if (cproc != nullptr
|
|
&& cproc->action == RUNNING_INCR_IMAGE)
|
|
{
|
|
local_backup_running_id = cproc->id;
|
|
cproc->last_pingtime = Server->getTimeMS();
|
|
}
|
|
else
|
|
{
|
|
SRunningProcess new_proc;
|
|
new_proc.action = RUNNING_INCR_IMAGE;
|
|
new_proc.server_token = server_token;
|
|
new_proc.pcdone = 0;
|
|
new_proc.details = image_inf.orig_image_letter;
|
|
new_proc.server_id = image_inf.server_status_id;
|
|
|
|
local_backup_running_id = addNewProcess(new_proc);
|
|
}
|
|
|
|
removeTimedOutProcesses(server_token, false);
|
|
}
|
|
|
|
status_updated = true;
|
|
image_inf.running_process_id = local_backup_running_id;
|
|
image_inf.thread_ticket=Server->getThreadPool()->execute(image_inf.image_thread, "incr image upload");
|
|
|
|
return true;
|
|
}
|
|
|
|
bool ClientConnector::waitForThread(void)
|
|
{
|
|
if(image_inf.thread_action!=TA_NONE && Server->getThreadPool()->isRunning(image_inf.thread_ticket ) )
|
|
return true;
|
|
else
|
|
return false;
|
|
}
|
|
|
|
namespace
|
|
{
|
|
bool parseDevicePartNumber(const std::string& volfn, std::string& dev, int& DeviceNumber, int& PartNumber)
|
|
{
|
|
if (next(volfn, 0, "/dev/mapper/")
|
|
|| next(volfn, 0, "/dev/dm-") )
|
|
{
|
|
std::string dm_table;
|
|
//TODO: Use ioctl here
|
|
if (os_popen("dmsetup table \"" + greplace("\"", "_", volfn) + "\"", dm_table)==0)
|
|
{
|
|
std::vector<std::string> toks;
|
|
Tokenize(trim(dm_table), toks, " ");
|
|
std::string back_dev;
|
|
if (toks.size() > 2)
|
|
{
|
|
if (toks[2] == "linear"
|
|
&& toks.size()>3)
|
|
{
|
|
back_dev = toks[3];
|
|
}
|
|
else if (toks[2] == "era"
|
|
&& toks.size() > 4)
|
|
{
|
|
back_dev = toks[4];
|
|
}
|
|
else if (toks[2] == "snapshot-origin"
|
|
&& toks.size()>3)
|
|
{
|
|
back_dev = toks[3];
|
|
}
|
|
else
|
|
{
|
|
Server->Log("Cannot follow device mapping " + toks[2], LL_WARNING);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
Server->Log("Cannot parse device mapper table \"" + dm_table + "\"", LL_WARNING);
|
|
}
|
|
|
|
if (!back_dev.empty())
|
|
{
|
|
Server->Log("Following device mapping ("+toks[2]+") to device "+back_dev, LL_DEBUG);
|
|
|
|
std::string uevent_info = getFile("/sys/dev/block/" + back_dev + "/uevent");
|
|
std::string dev_name = getbetween("DEVNAME=", "\n", uevent_info);
|
|
|
|
if (FileExists("/dev/" + dev_name))
|
|
{
|
|
Server->Log("Device name /dev/"+dev_name, LL_DEBUG);
|
|
|
|
return parseDevicePartNumber("/dev/" + dev_name, dev, DeviceNumber, PartNumber);
|
|
}
|
|
else
|
|
{
|
|
Server->Log("Could not find device name of device "+back_dev, LL_WARNING);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
std::string dl_devnum;
|
|
const char* const devnames[] = { "sd", "xvd", "vd", "hd", "loop", "nvme", "nbd", nullptr };
|
|
|
|
for (const char* const * devname = devnames; *devname != nullptr; ++devname)
|
|
{
|
|
if (next(volfn, 0, std::string("/dev/") + *devname))
|
|
{
|
|
dl_devnum = getafter(std::string("/dev/") + *devname, volfn);
|
|
dev = std::string("/dev/") + *devname;
|
|
break;
|
|
}
|
|
}
|
|
|
|
if (dl_devnum.empty())
|
|
{
|
|
DeviceNumber = -1;
|
|
PartNumber = -1;
|
|
return false;
|
|
}
|
|
|
|
Server->Log("dl_devnum=" + dl_devnum, LL_DEBUG);
|
|
|
|
int state = 0;
|
|
int devnum = -1;
|
|
std::string data;
|
|
std::vector<int> nums;
|
|
for (size_t i = 0; i <= dl_devnum.size(); ++i)
|
|
{
|
|
char ch = 0;
|
|
if (i < dl_devnum.size())
|
|
ch = dl_devnum[i];
|
|
|
|
if (state == 0)
|
|
{
|
|
if (str_isnumber(ch))
|
|
{
|
|
state = 1;
|
|
data += ch;
|
|
dev += ch;
|
|
}
|
|
else
|
|
{
|
|
state = 2;
|
|
if (ch != 0)
|
|
{
|
|
dev += ch;
|
|
devnum = tolower(ch) - 'a';
|
|
}
|
|
}
|
|
}
|
|
else if (state == 1)
|
|
{
|
|
if (str_isnumber(ch))
|
|
{
|
|
data += ch;
|
|
}
|
|
else
|
|
{
|
|
if (ch != 'n')
|
|
{
|
|
nums.push_back(watoi(data));
|
|
}
|
|
else
|
|
{
|
|
dev += ch;
|
|
}
|
|
data.clear();
|
|
|
|
if (ch == 'p')
|
|
{
|
|
state = 1;
|
|
}
|
|
else
|
|
{
|
|
state = 0;
|
|
}
|
|
}
|
|
}
|
|
else if (state == 2)
|
|
{
|
|
if (str_isnumber(ch))
|
|
{
|
|
data += ch;
|
|
state = 1;
|
|
}
|
|
else if (ch != 0)
|
|
{
|
|
dev += ch;
|
|
devnum = (devnum+1) * ('z' - 'a') + tolower(ch) - 'a';
|
|
}
|
|
}
|
|
}
|
|
|
|
if (devnum != -1)
|
|
{
|
|
DeviceNumber = devnum;
|
|
PartNumber = -1;
|
|
|
|
if (!nums.empty())
|
|
{
|
|
PartNumber = nums[0];
|
|
}
|
|
|
|
if (nums.size() > 1)
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (nums.empty())
|
|
return false;
|
|
|
|
DeviceNumber = nums[0];
|
|
PartNumber = -1;
|
|
|
|
if (nums.size() > 1)
|
|
{
|
|
PartNumber = nums[1];
|
|
}
|
|
|
|
if (nums.size() > 2)
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool check_partition_used(IFile* dev, int64 offset, int64 length)
|
|
{
|
|
bool gpt_style2;
|
|
std::vector<IFSImageFactory::SPartition> parts = image_fak->readPartitions(dev, gpt_style2);
|
|
|
|
for (size_t i = 0; i < parts.size(); ++i)
|
|
{
|
|
if ((offset >= parts[i].offset
|
|
&& offset < parts[i].offset + parts[i].length)
|
|
|| (offset + length >= parts[i].offset
|
|
&& offset + length < parts[i].offset + parts[i].length))
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
}
|
|
|
|
void parse_devnum_test()
|
|
{
|
|
int deviceNumber, partNumber;
|
|
std::string dev;
|
|
assert(parseDevicePartNumber("/dev/sda1", dev, deviceNumber, partNumber));
|
|
assert(deviceNumber == 0);
|
|
assert(partNumber == 1);
|
|
assert(dev == "/dev/sda");
|
|
assert(parseDevicePartNumber("/dev/xvda1", dev, deviceNumber, partNumber));
|
|
assert(deviceNumber == 0);
|
|
assert(partNumber == 1);
|
|
assert(dev == "/dev/xvda");
|
|
assert(parseDevicePartNumber("/dev/xvdc1", dev, deviceNumber, partNumber));
|
|
assert(deviceNumber == 2);
|
|
assert(partNumber == 1);
|
|
assert(dev == "/dev/xvdc");
|
|
assert(parseDevicePartNumber("/dev/xvdc3", dev, deviceNumber, partNumber));
|
|
assert(deviceNumber == 2);
|
|
assert(partNumber == 3);
|
|
assert(dev == "/dev/xvdc");
|
|
assert(parseDevicePartNumber("/dev/nvme0n1p1", dev, deviceNumber, partNumber));
|
|
assert(deviceNumber == 1);
|
|
assert(partNumber == 1);
|
|
assert(dev == "/dev/nvme0n1");
|
|
assert(parseDevicePartNumber("/dev/nvme0n5p8", dev, deviceNumber, partNumber));
|
|
assert(deviceNumber == 5);
|
|
assert(partNumber == 8);
|
|
assert(dev == "/dev/nvme0n5");
|
|
assert(parseDevicePartNumber("/dev/loop0", dev, deviceNumber, partNumber));
|
|
assert(deviceNumber == 0);
|
|
assert(partNumber == -1);
|
|
assert(dev == "/dev/loop0");
|
|
assert(parseDevicePartNumber("/dev/sda", dev, deviceNumber, partNumber));
|
|
assert(deviceNumber == 0);
|
|
assert(partNumber == -1);
|
|
assert(dev == "/dev/sda");
|
|
assert(parseDevicePartNumber("/dev/sdc", dev, deviceNumber, partNumber));
|
|
assert(deviceNumber == 2);
|
|
assert(partNumber == -1);
|
|
assert(dev == "/dev/sdc");
|
|
assert(parseDevicePartNumber("/dev/sdaa", dev, deviceNumber, partNumber));
|
|
assert(deviceNumber == 25);
|
|
assert(partNumber == -1);
|
|
assert(dev == "/dev/sdaa");
|
|
assert(parseDevicePartNumber("/dev/loop2p3", dev, deviceNumber, partNumber));
|
|
assert(deviceNumber == 2);
|
|
assert(partNumber == 3);
|
|
assert(dev == "/dev/loop2");
|
|
}
|
|
|
|
bool ClientConnector::sendMBR(std::string dl, std::string &errmsg)
|
|
{
|
|
#pragma pack(push)
|
|
#pragma pack(1)
|
|
struct EfiHeader
|
|
{
|
|
uint64 signature;
|
|
_u32 revision;
|
|
_u32 header_size;
|
|
_u32 header_crc;
|
|
_u32 reserved;
|
|
int64 current_lba;
|
|
int64 backup_lba;
|
|
int64 first_lba;
|
|
int64 last_lba;
|
|
char disk_guid[16];
|
|
int64 partition_table_lba;
|
|
_u32 num_parition_entries;
|
|
_u32 partition_entry_size;
|
|
_u32 partition_table_crc;
|
|
};
|
|
#pragma pack(pop)
|
|
|
|
const uint64 gpt_magic = 0x5452415020494645ULL;
|
|
bool gpt_style = false;
|
|
unsigned int logical_sector_size = 512;
|
|
|
|
#ifdef _WIN32
|
|
std::string vpath=dl;
|
|
if(!vpath.empty() && vpath[0]!='\\')
|
|
{
|
|
dl+=":";
|
|
vpath="\\\\.\\"+dl;
|
|
}
|
|
|
|
HANDLE hVolume=CreateFileW(Server->ConvertToWchar(vpath).c_str(), GENERIC_READ, FILE_SHARE_READ|FILE_SHARE_WRITE, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
|
|
if(hVolume==INVALID_HANDLE_VALUE)
|
|
{
|
|
errmsg="CreateFile of volume '"+dl+"' failed. - sendMBR. Errorcode: "+convert((int)GetLastError());
|
|
Server->Log(errmsg, LL_ERROR);
|
|
return false;
|
|
}
|
|
|
|
STORAGE_DEVICE_NUMBER dev_num;
|
|
DWORD ret_bytes;
|
|
BOOL b=DeviceIoControl(hVolume, IOCTL_STORAGE_GET_DEVICE_NUMBER, NULL, 0, &dev_num, sizeof(STORAGE_DEVICE_NUMBER), &ret_bytes, NULL);
|
|
|
|
bool dynamic_volume=false;
|
|
if(b==FALSE)
|
|
{
|
|
errmsg="DeviceIoControl IOCTL_STORAGE_GET_DEVICE_NUMBER failed. Volume: '"+dl+"'";
|
|
Server->Log(errmsg, LL_WARNING);
|
|
dynamic_volume=true;
|
|
}
|
|
|
|
{
|
|
std::unique_ptr<VOLUME_DISK_EXTENTS> vde((VOLUME_DISK_EXTENTS*)new char[sizeof(VOLUME_DISK_EXTENTS)]);
|
|
b=DeviceIoControl(hVolume, IOCTL_VOLUME_GET_VOLUME_DISK_EXTENTS, NULL, 0, vde.get(), sizeof(VOLUME_DISK_EXTENTS), &ret_bytes, NULL);
|
|
if(b==0 && GetLastError()==ERROR_MORE_DATA)
|
|
{
|
|
DWORD ext_num=vde->NumberOfDiskExtents;
|
|
errmsg="DeviceIoControl IOCTL_VOLUME_GET_VOLUME_DISK_EXTENTS failed. Extends: "+convert((int)ext_num);
|
|
Server->Log(errmsg, LL_WARNING);
|
|
DWORD vde_size=sizeof(VOLUME_DISK_EXTENTS)+sizeof(DISK_EXTENT)*(ext_num-1);
|
|
vde.reset((VOLUME_DISK_EXTENTS*)new char[vde_size]);
|
|
b=DeviceIoControl(hVolume, IOCTL_VOLUME_GET_VOLUME_DISK_EXTENTS, NULL, 0, vde.get(), vde_size, &ret_bytes, NULL);
|
|
if(b==0)
|
|
{
|
|
errmsg="DeviceIoControl IOCTL_VOLUME_GET_VOLUME_DISK_EXTENTS failed twice. Volume: '"+dl+"'";
|
|
Server->Log(errmsg, LL_ERROR);
|
|
CloseHandle(hVolume);
|
|
return false;
|
|
}
|
|
}
|
|
else if(b==0)
|
|
{
|
|
errmsg="DeviceIoControl IOCTL_VOLUME_GET_VOLUME_DISK_EXTENTS failed. Volume: '"+dl+"' Error: "+convert((int)GetLastError());
|
|
Server->Log(errmsg, LL_ERROR);
|
|
CloseHandle(hVolume);
|
|
return false;
|
|
}
|
|
|
|
if(vde->NumberOfDiskExtents>0)
|
|
{
|
|
HANDLE hDevice=CreateFileW(Server->ConvertToWchar("\\\\.\\PhysicalDrive"+convert((int)vde->Extents[0].DiskNumber)).c_str(), GENERIC_READ, FILE_SHARE_READ|FILE_SHARE_WRITE, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
|
|
if(hDevice==INVALID_HANDLE_VALUE)
|
|
{
|
|
errmsg="CreateFile of device '"+dl+"' failed. - sendMBR";
|
|
Server->Log(errmsg, LL_ERROR);
|
|
CloseHandle(hVolume);
|
|
return false;
|
|
}
|
|
|
|
DWORD numPartitions=10;
|
|
DWORD inf_size=sizeof(DRIVE_LAYOUT_INFORMATION_EX)+sizeof(PARTITION_INFORMATION_EX)*(numPartitions-1);
|
|
|
|
std::unique_ptr<DRIVE_LAYOUT_INFORMATION_EX> inf((DRIVE_LAYOUT_INFORMATION_EX*)new char[sizeof(DRIVE_LAYOUT_INFORMATION_EX)+sizeof(PARTITION_INFORMATION_EX)*(numPartitions-1)]);
|
|
|
|
b=DeviceIoControl(hDevice, IOCTL_DISK_GET_DRIVE_LAYOUT_EX, NULL, 0, inf.get(), inf_size, &ret_bytes, NULL);
|
|
while(b==0 && GetLastError()==ERROR_INSUFFICIENT_BUFFER && numPartitions<1000)
|
|
{
|
|
numPartitions*=2;
|
|
inf_size=sizeof(DRIVE_LAYOUT_INFORMATION_EX)+sizeof(PARTITION_INFORMATION_EX)*(numPartitions-1);
|
|
inf.reset((DRIVE_LAYOUT_INFORMATION_EX*)new char[inf_size]);
|
|
b=DeviceIoControl(hDevice, IOCTL_DISK_GET_DRIVE_LAYOUT_EX, NULL, 0, inf.get(), inf_size, &ret_bytes, NULL);
|
|
}
|
|
if(b==0)
|
|
{
|
|
errmsg="DeviceIoControl IOCTL_DISK_GET_DRIVE_LAYOUT_EX failed. Volume: '"+dl+"' Error: "+convert((int)GetLastError());
|
|
Server->Log(errmsg, LL_ERROR);
|
|
CloseHandle(hDevice);
|
|
CloseHandle(hVolume);
|
|
return false;
|
|
}
|
|
|
|
if(inf->PartitionStyle==PARTITION_STYLE_GPT)
|
|
{
|
|
gpt_style=true;
|
|
}
|
|
else if(inf->PartitionStyle!=PARTITION_STYLE_MBR)
|
|
{
|
|
errmsg="Partition style "+convert((unsigned int)inf->PartitionStyle)+" not supported. Volume: '"+dl;
|
|
Server->Log(errmsg, LL_ERROR);
|
|
CloseHandle(hDevice);
|
|
CloseHandle(hVolume);
|
|
return false;
|
|
}
|
|
|
|
if(gpt_style)
|
|
{
|
|
STORAGE_PROPERTY_QUERY query = {};
|
|
query.PropertyId = StorageAccessAlignmentProperty;
|
|
query.QueryType = PropertyStandardQuery;
|
|
|
|
STORAGE_ACCESS_ALIGNMENT_DESCRIPTOR alignment_descriptor;
|
|
|
|
b=DeviceIoControl( hVolume, IOCTL_STORAGE_QUERY_PROPERTY, &query, sizeof(query),
|
|
&alignment_descriptor, sizeof(alignment_descriptor),
|
|
&ret_bytes, NULL);
|
|
|
|
if(b==FALSE)
|
|
{
|
|
Server->Log("Cannot get physical sector size of volume: '"+dl+". Assuming 512 bytes.", LL_WARNING);
|
|
}
|
|
else
|
|
{
|
|
unsigned int r_logical_sector_size = alignment_descriptor.BytesPerLogicalSector;
|
|
|
|
if( (r_logical_sector_size & (r_logical_sector_size-1))==0 &&
|
|
r_logical_sector_size!=0 )
|
|
{
|
|
logical_sector_size = r_logical_sector_size;
|
|
}
|
|
}
|
|
}
|
|
|
|
if(dynamic_volume)
|
|
{
|
|
bool found=false;
|
|
for(DWORD j=0;j<vde->NumberOfDiskExtents;++j)
|
|
{
|
|
for(DWORD i=0;i<inf->PartitionCount;++i)
|
|
{
|
|
if(inf->PartitionEntry[i].StartingOffset.QuadPart==vde->Extents[j].StartingOffset.QuadPart)
|
|
{
|
|
dev_num.PartitionNumber=inf->PartitionEntry[i].PartitionNumber;
|
|
dev_num.DeviceNumber=vde->Extents[j].DiskNumber;
|
|
found=true;
|
|
}
|
|
}
|
|
}
|
|
|
|
if(found)
|
|
{
|
|
errmsg="Dynamic volumes are not supported. It may work with mirrored whole disk volumes though. Volume: '"+dl+"'";
|
|
Server->Log(errmsg, LL_WARNING);
|
|
}
|
|
else
|
|
{
|
|
errmsg="Did not find PartitionNumber of dynamic volume. Volume: '"+dl+"'";
|
|
Server->Log(errmsg, LL_ERROR);
|
|
CloseHandle(hVolume);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
CloseHandle(hDevice);
|
|
}
|
|
else
|
|
{
|
|
errmsg="DeviceIoControl IOCTL_VOLUME_GET_VOLUME_DISK_EXTENTS returned no extends. Volume: '"+dl+"'";
|
|
Server->Log(errmsg, LL_ERROR);
|
|
CloseHandle(hVolume);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
CloseHandle(hVolume);
|
|
|
|
wchar_t voln[MAX_PATH+1];
|
|
DWORD voln_size=MAX_PATH+1;
|
|
DWORD voln_sern;
|
|
wchar_t fsn[MAX_PATH+1];
|
|
DWORD fsn_size=MAX_PATH+1;
|
|
b=GetVolumeInformationW(Server->ConvertToWchar(dl+"\\").c_str(), voln, voln_size, &voln_sern, NULL, NULL, fsn, fsn_size);
|
|
if(b==0)
|
|
{
|
|
errmsg="GetVolumeInformationW failed. Volume: '"+dl+"'. "+os_last_error_str();
|
|
Server->Log(errmsg, LL_ERROR);
|
|
return false;
|
|
}
|
|
|
|
std::string fi_serial = convert((_i64)voln_sern);
|
|
std::string volume_name = Server->ConvertFromWchar(voln);
|
|
std::string fi_name = Server->ConvertFromWchar(fsn);
|
|
#else //_WIN32
|
|
struct
|
|
{
|
|
int DeviceNumber;
|
|
int PartitionNumber;
|
|
} dev_num;
|
|
|
|
std::string dev_fn;
|
|
parseDevicePartNumber(dl, dev_fn, dev_num.DeviceNumber, dev_num.PartitionNumber);
|
|
|
|
IFile* dev = Server->openFile(dev_fn, MODE_READ_DEVICE);
|
|
if (dev == nullptr)
|
|
{
|
|
errmsg = "Error opening Device " + dev_fn + ". " + os_last_error_str();
|
|
Server->Log(errmsg, LL_ERROR);
|
|
return false;
|
|
}
|
|
|
|
std::string gpt_header = dev->Read(512LL, 512, nullptr);
|
|
|
|
if (next(gpt_header, 0, "EFI PART"))
|
|
{
|
|
Server->Log("gpt_style=true", LL_DEBUG);
|
|
gpt_style = true;
|
|
}
|
|
|
|
std::string blkid;
|
|
os_popen("blkid \"" + dl + "\" -o export", blkid);
|
|
std::string fi_serial = trim(getbetween("UUID=", "\n", blkid));
|
|
std::string volume_name = "LINUX:"+dl;
|
|
std::string fi_name = trim(getbetween("PARTLABEL=", "\n", blkid));
|
|
if(fi_name.empty())
|
|
fi_name= trim(getbetween("TYPE=", "\n", blkid));
|
|
#endif
|
|
|
|
CWData mbr;
|
|
mbr.addChar(1);
|
|
mbr.addChar(gpt_style?1:0);
|
|
mbr.addInt(dev_num.DeviceNumber);
|
|
mbr.addInt(dev_num.PartitionNumber);
|
|
mbr.addString(fi_serial);
|
|
mbr.addString(volume_name);
|
|
mbr.addString(fi_name);
|
|
|
|
#ifdef _WIN32
|
|
IFile *dev=Server->openFile("\\\\.\\PhysicalDrive"+convert((int)dev_num.DeviceNumber), MODE_READ_DEVICE);
|
|
|
|
if(dev==NULL)
|
|
{
|
|
errmsg="Error opening Device "+convert((int)dev_num.DeviceNumber)+". "+ os_last_error_str();
|
|
Server->Log(errmsg, LL_ERROR);
|
|
return false;
|
|
}
|
|
#endif
|
|
|
|
dev->Seek(0);
|
|
std::string mbr_bytes=dev->Read(512);
|
|
|
|
mbr.addString(mbr_bytes);
|
|
|
|
if(gpt_style)
|
|
{
|
|
Server->Log("GUID partition table found", LL_DEBUG);
|
|
|
|
if(!dev->Seek(logical_sector_size))
|
|
{
|
|
errmsg="Error seeking in device to GPT header "+convert((int)dev_num.DeviceNumber)+". "+ os_last_error_str();
|
|
Server->Log(errmsg, LL_ERROR);
|
|
return false;
|
|
}
|
|
|
|
std::string primary_gpt_header = dev->Read(logical_sector_size);
|
|
|
|
if(primary_gpt_header.size()!=logical_sector_size)
|
|
{
|
|
errmsg="Error reading GPT header "+convert((int)dev_num.DeviceNumber);
|
|
Server->Log(errmsg, LL_ERROR);
|
|
return false;
|
|
}
|
|
|
|
if(primary_gpt_header.size()<sizeof(EfiHeader))
|
|
{
|
|
errmsg="GPT header too small "+convert((int)dev_num.DeviceNumber);
|
|
Server->Log(errmsg, LL_ERROR);
|
|
return false;
|
|
}
|
|
|
|
const EfiHeader* gpt_header_s = reinterpret_cast<const EfiHeader*>(primary_gpt_header.data());
|
|
|
|
if(gpt_header_s->signature!=gpt_magic)
|
|
{
|
|
errmsg="GPT magic wrong "+convert((int)dev_num.DeviceNumber);
|
|
Server->Log(errmsg, LL_ERROR);
|
|
return false;
|
|
}
|
|
|
|
mbr.addInt64(logical_sector_size);
|
|
mbr.addString(primary_gpt_header);
|
|
|
|
int64 primary_paritition_table_pos = gpt_header_s->partition_table_lba*logical_sector_size;
|
|
if(!dev->Seek(primary_paritition_table_pos))
|
|
{
|
|
errmsg="Error seeking in device to GPT partition table "+convert((int)dev_num.DeviceNumber)+". "+ os_last_error_str();
|
|
Server->Log(errmsg, LL_ERROR);
|
|
return false;
|
|
}
|
|
|
|
_u32 toread = gpt_header_s->num_parition_entries*gpt_header_s->partition_entry_size;
|
|
std::string primary_gpt_table = dev->Read(toread);
|
|
|
|
Server->Log("GUID partition table size is "+PrettyPrintBytes(toread), LL_DEBUG);
|
|
|
|
if(primary_gpt_table.size()!=toread)
|
|
{
|
|
errmsg="Error reading GPT partition table "+convert((int)dev_num.DeviceNumber)+". "+ os_last_error_str();
|
|
Server->Log(errmsg, LL_ERROR);
|
|
return false;
|
|
}
|
|
|
|
mbr.addInt64(primary_paritition_table_pos);
|
|
mbr.addString(primary_gpt_table);
|
|
|
|
// BACKUP HEADER
|
|
int64 backup_gpt_location = gpt_header_s->backup_lba*logical_sector_size;
|
|
std::string backup_gpt_header;
|
|
if(!dev->Seek(backup_gpt_location))
|
|
{
|
|
errmsg="Error seeking in device to backup GPT header "+convert((int)dev_num.DeviceNumber)+" at "+convert(backup_gpt_location)+". "+ os_last_error_str();
|
|
Server->Log(errmsg, LL_WARNING);
|
|
|
|
backup_gpt_location = logical_sector_size;
|
|
backup_gpt_header = primary_gpt_header;
|
|
}
|
|
else
|
|
{
|
|
backup_gpt_header = dev->Read(logical_sector_size);
|
|
|
|
if (backup_gpt_header.size() != logical_sector_size)
|
|
{
|
|
errmsg = "Error reading backup GPT header " + convert((int)dev_num.DeviceNumber) + " at " + convert(backup_gpt_location);
|
|
Server->Log(errmsg, LL_WARNING);
|
|
|
|
backup_gpt_location = logical_sector_size;
|
|
backup_gpt_header = primary_gpt_header;
|
|
}
|
|
}
|
|
|
|
if(backup_gpt_header.size()<sizeof(EfiHeader))
|
|
{
|
|
errmsg="Backup GPT header too small "+convert((int)dev_num.DeviceNumber);
|
|
Server->Log(errmsg, LL_WARNING);
|
|
|
|
backup_gpt_location = logical_sector_size;
|
|
backup_gpt_header = primary_gpt_header;
|
|
}
|
|
|
|
const EfiHeader* backup_gpt_header_s = reinterpret_cast<const EfiHeader*>(backup_gpt_header.data());
|
|
|
|
if(backup_gpt_header_s->signature!=gpt_magic)
|
|
{
|
|
errmsg="Backup GPT magic wrong "+convert((int)dev_num.DeviceNumber);
|
|
Server->Log(errmsg, LL_WARNING);
|
|
|
|
backup_gpt_location = logical_sector_size;
|
|
backup_gpt_header = primary_gpt_header;
|
|
backup_gpt_header_s = reinterpret_cast<const EfiHeader*>(backup_gpt_header.data());
|
|
}
|
|
|
|
mbr.addInt64(backup_gpt_location);
|
|
mbr.addString(backup_gpt_header);
|
|
|
|
int64 backup_paritition_table_pos = backup_gpt_header_s->partition_table_lba*logical_sector_size;
|
|
if(!dev->Seek(backup_paritition_table_pos))
|
|
{
|
|
errmsg="Error seeking in device to GPT partition table "+convert((int)dev_num.DeviceNumber)+". "+ os_last_error_str();
|
|
Server->Log(errmsg, LL_WARNING);
|
|
|
|
backup_paritition_table_pos = primary_paritition_table_pos;
|
|
dev->Seek(backup_paritition_table_pos);
|
|
}
|
|
|
|
toread = backup_gpt_header_s->num_parition_entries*backup_gpt_header_s->partition_entry_size;
|
|
std::string backup_gpt_table = dev->Read(toread);
|
|
|
|
if(backup_gpt_table.size()!=toread)
|
|
{
|
|
errmsg="Error reading GPT partition table "+convert((int)dev_num.DeviceNumber)+". "+ os_last_error_str();
|
|
Server->Log(errmsg, LL_WARNING);
|
|
|
|
backup_paritition_table_pos = primary_paritition_table_pos;
|
|
backup_gpt_table = primary_gpt_table;
|
|
}
|
|
|
|
mbr.addInt64(backup_paritition_table_pos);
|
|
mbr.addString(backup_gpt_table);
|
|
}
|
|
|
|
mbr.addString(errmsg);
|
|
|
|
if (!gpt_style
|
|
&& mbr_bytes.find("VeraCrypt Boot Loader") != std::string::npos
|
|
&& !check_partition_used(dev, 512, 63*512) )
|
|
{
|
|
std::string veracrypt_data = dev->Read(512LL, 63 * 512);
|
|
mbr.addVarInt(512);
|
|
mbr.addString2(veracrypt_data);
|
|
}
|
|
else if (!gpt_style
|
|
&& mbr_bytes.find("GRUB") != std::string::npos
|
|
&& !check_partition_used(dev, 512, 63 * 512) )
|
|
{
|
|
std::string grub_stage_1_5 = dev->Read(512LL, 63 * 512);
|
|
if (grub_stage_1_5.find("Geom") != std::string::npos
|
|
&& grub_stage_1_5.find("Read") != std::string::npos
|
|
&& grub_stage_1_5.find("Error") != std::string::npos)
|
|
{
|
|
mbr.addVarInt(512);
|
|
mbr.addString2(grub_stage_1_5);
|
|
}
|
|
}
|
|
|
|
tcpstack.Send(pipe, mbr);
|
|
|
|
return true;
|
|
}
|
|
|
|
const int64 receive_timeouttime=60000;
|
|
std::string ClientConnector::receivePacket(const SChannel& channel, int64 timeoutms)
|
|
{
|
|
CTCPStack localstack(channel.internet_connection);
|
|
int64 starttime=Server->getTimeMS();
|
|
while(Server->getTimeMS()-starttime<= timeoutms)
|
|
{
|
|
std::string ret;
|
|
size_t rc=channel.pipe->Read(&ret, static_cast<int>(timeoutms));
|
|
if(rc==0)
|
|
{
|
|
return "";
|
|
}
|
|
localstack.AddData((char*)ret.c_str(), ret.size());
|
|
|
|
ret.clear();
|
|
if(localstack.getPacket(ret))
|
|
{
|
|
return ret;
|
|
}
|
|
}
|
|
return "";
|
|
}
|
|
|
|
void ClientConnector::removeChannelpipe(IPipe *cp)
|
|
{
|
|
for (size_t i = 0; i < channel_pipes.size(); ++i)
|
|
{
|
|
if (channel_pipes[i].pipe == cp)
|
|
{
|
|
channel_pipes[i].state = SChannel::EChannelState_Exit;
|
|
}
|
|
}
|
|
}
|
|
|
|
void ClientConnector::downloadImage(str_map params, IScopedLock& backup_mutex_lock)
|
|
{
|
|
_i64 imgsize=-1;
|
|
if(channel_pipes.size()==0)
|
|
{
|
|
imgsize=-2;
|
|
pipe->Write((char*)&imgsize, sizeof(_i64), (int)receive_timeouttime);
|
|
Server->Log("No channel found!", LL_DEBUG);
|
|
return;
|
|
}
|
|
|
|
{
|
|
IScopedLock lock(backup_mutex);
|
|
SRunningProcess new_proc;
|
|
new_proc.action = RUNNING_RESTORE_IMAGE;
|
|
new_proc.id = ++curr_backup_running_id;
|
|
new_proc.pcdone = 0;
|
|
|
|
local_backup_running_id = new_proc.id;
|
|
|
|
running_processes.push_back(new_proc);
|
|
status_updated = true;
|
|
}
|
|
|
|
ScopedRemoveRunningBackup remove_running_backup(local_backup_running_id);
|
|
|
|
for(size_t i=0;i<channel_pipes.size();++i)
|
|
{
|
|
SChannel::EChannelState orig_state = channel_pipes[i].state;
|
|
|
|
if(orig_state!=SChannel::EChannelState_Idle)
|
|
continue;
|
|
|
|
int restore_version = channel_pipes[i].restore_version;
|
|
|
|
channel_pipes[i].state = SChannel::EChannelState_Used;
|
|
|
|
_i64 received_bytes = 0;
|
|
std::string offset;
|
|
int64 start_offset = 0;
|
|
if(params.find("offset")!=params.end())
|
|
{
|
|
offset="&offset="+params["offset"];
|
|
start_offset = watoi64(params["offset"]);
|
|
if (params.find("received_bytes") != params.end())
|
|
{
|
|
received_bytes = watoi64(params["received_bytes"]);
|
|
}
|
|
}
|
|
|
|
std::string token_param;
|
|
if (params.find("token") != params.end())
|
|
{
|
|
token_param = "&token=" + EscapeParamString(params["token"]);
|
|
}
|
|
|
|
sendChannelPacket(channel_pipes[i], "DOWNLOAD IMAGE with_used_bytes=1&img_id="
|
|
+ params["img_id"] + "&time=" + params["time"] + "&mbr=" + params["mbr"] + offset + token_param);
|
|
|
|
Server->Log("Downloading from channel "+convert((int)i), LL_DEBUG);
|
|
|
|
_i64 imgsize=-1;
|
|
IPipe *c = channel_pipes[i].pipe;
|
|
c->Read((char*)&imgsize, sizeof(_i64), 60000);
|
|
Server->Log("Imagesize "+convert(imgsize), LL_DEBUG);
|
|
if(imgsize==-1)
|
|
{
|
|
Server->Log("Error reading size", LL_ERROR);
|
|
if(i+1<channel_pipes.size())
|
|
{
|
|
removeChannelpipe(c);
|
|
continue;
|
|
}
|
|
else
|
|
{
|
|
pipe->Write((char*)&imgsize, sizeof(_i64), (int)receive_timeouttime);
|
|
removeChannelpipe(c);
|
|
return;
|
|
}
|
|
}
|
|
|
|
backup_mutex_lock.relock(nullptr);
|
|
|
|
if(!pipe->Write((char*)&imgsize, sizeof(_i64), (int)receive_timeouttime))
|
|
{
|
|
Server->Log("Could not write to pipe! downloadImage-1", LL_ERROR);
|
|
backup_mutex_lock.relock(backup_mutex);
|
|
removeChannelpipe(c);
|
|
return;
|
|
}
|
|
|
|
const size_t c_buffer_size=32768;
|
|
const unsigned int c_blocksize=4096;
|
|
char buf[c_buffer_size];
|
|
_i64 read=0;
|
|
|
|
if(params["mbr"]=="true")
|
|
{
|
|
Server->Log("Downloading MBR...", LL_DEBUG);
|
|
while(read<imgsize)
|
|
{
|
|
size_t c_read=c->Read(buf, c_buffer_size, 60000);
|
|
if(c_read==0)
|
|
{
|
|
backup_mutex_lock.relock(backup_mutex);
|
|
Server->Log("Read Timeout", LL_ERROR);
|
|
removeChannelpipe(c);
|
|
return;
|
|
}
|
|
if(!pipe->Write(buf, (_u32)c_read, (int)receive_timeouttime*5))
|
|
{
|
|
backup_mutex_lock.relock(backup_mutex);
|
|
Server->Log("Could not write to pipe! downloadImage-5", LL_ERROR);
|
|
removeChannelpipe(c);
|
|
return;
|
|
}
|
|
read+=c_read;
|
|
}
|
|
Server->Log("Downloading MBR done");
|
|
backup_mutex_lock.relock(backup_mutex);
|
|
for(auto& cp: channel_pipes)
|
|
{
|
|
if(cp.pipe == c)
|
|
{
|
|
cp.state = orig_state;
|
|
}
|
|
}
|
|
return;
|
|
}
|
|
|
|
_i64 used_bytes = imgsize;
|
|
if (restore_version > 0)
|
|
{
|
|
if (!c->Read((char*)&used_bytes, sizeof(_i64), 60000))
|
|
{
|
|
Server->Log("Error getting used bytes", LL_ERROR);
|
|
backup_mutex_lock.relock(backup_mutex);
|
|
if (i + 1<channel_pipes.size())
|
|
{
|
|
removeChannelpipe(c);
|
|
waitForPings(&backup_mutex_lock);
|
|
continue;
|
|
}
|
|
else
|
|
{
|
|
removeChannelpipe(c);
|
|
return;
|
|
}
|
|
}
|
|
Server->Log("Used bytes " + convert(used_bytes), LL_INFO);
|
|
}
|
|
else
|
|
{
|
|
Server->Log("Restore version 0. No used bytes", LL_INFO);
|
|
received_bytes = start_offset;
|
|
}
|
|
|
|
{
|
|
int t_pcdone=(std::min)((int)100, (int)(((float)received_bytes/(float)used_bytes)*100.f+0.5f));
|
|
|
|
IScopedLock lock(process_mutex);
|
|
|
|
SRunningProcess* proc = getRunningProcess(local_backup_running_id);
|
|
if (proc != nullptr)
|
|
{
|
|
proc->pcdone = t_pcdone;
|
|
proc->total_bytes = used_bytes;
|
|
proc->done_bytes = received_bytes;
|
|
}
|
|
}
|
|
|
|
|
|
if (received_bytes >= used_bytes)
|
|
{
|
|
received_bytes = used_bytes;
|
|
}
|
|
|
|
unsigned int blockleft=0;
|
|
unsigned int off=0;
|
|
_i64 pos=0;
|
|
int64 lastupdatetime = Server->getTimeMS();
|
|
int64 last_received = received_bytes;
|
|
while(pos<imgsize)
|
|
{
|
|
size_t r=c->Read(&buf[off], c_buffer_size-off, 180000);
|
|
if( r==0 )
|
|
{
|
|
Server->Log("Read Timeout -2 CS", LL_ERROR);
|
|
backup_mutex_lock.relock(backup_mutex);
|
|
removeChannelpipe(c);
|
|
return;
|
|
}
|
|
if(!pipe->Write(&buf[off], r, (int)receive_timeouttime*5))
|
|
{
|
|
Server->Log("Could not write to pipe! downloadImage-3 size "+convert(r)+" off "+convert(off), LL_ERROR);
|
|
backup_mutex_lock.relock(backup_mutex);
|
|
removeChannelpipe(c);
|
|
return;
|
|
}
|
|
if(r!=0)
|
|
r+=off;
|
|
off=0;
|
|
while(true)
|
|
{
|
|
if( blockleft==0 )
|
|
{
|
|
if(r-off>=sizeof(_i64) )
|
|
{
|
|
blockleft=c_blocksize;
|
|
_i64 s;
|
|
memcpy((char*)&s, &buf[off], sizeof(_i64) );
|
|
if(s>imgsize
|
|
&& s!= 0x7fffffffffffffffLL)
|
|
{
|
|
Server->Log("invalid seek value: "+convert(s), LL_ERROR);
|
|
}
|
|
off+=sizeof(_i64);
|
|
pos=s;
|
|
}
|
|
else if(r-off>0)
|
|
{
|
|
memmove(buf, &buf[off], r-off);
|
|
off=(_u32)r-off;
|
|
break;
|
|
}
|
|
else
|
|
{
|
|
off=0;
|
|
break;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
unsigned int available=(std::min)((unsigned int)r-off, blockleft);
|
|
read+=available;
|
|
blockleft-=available;
|
|
off+=available;
|
|
received_bytes += available;
|
|
if(off>=r)
|
|
{
|
|
off=0;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
lasttime=Server->getTimeMS();
|
|
|
|
if(lasttime - lastupdatetime>1000)
|
|
{
|
|
int64 passed_time = lasttime - lastupdatetime;
|
|
lastupdatetime = lasttime;
|
|
int t_pcdone=(std::min)((int)100, (int)(((float)received_bytes/(float)used_bytes)*100.f+0.5f));
|
|
int64 new_bytes = received_bytes - last_received;
|
|
last_received = received_bytes;
|
|
double speed_bpms = static_cast<double>(new_bytes)/passed_time;
|
|
|
|
IScopedLock lock(process_mutex);
|
|
|
|
SRunningProcess* proc = getRunningProcess(local_backup_running_id);
|
|
if (proc != nullptr)
|
|
{
|
|
proc->pcdone = t_pcdone;
|
|
proc->done_bytes = received_bytes;
|
|
proc->speed_bpms = speed_bpms;
|
|
proc->last_pingtime = lasttime;
|
|
status_updated = true;
|
|
}
|
|
}
|
|
}
|
|
remove_running_backup.setSuccess(true);
|
|
Server->Log("Downloading image done", LL_DEBUG);
|
|
backup_mutex_lock.relock(backup_mutex);
|
|
for(auto& cp: channel_pipes)
|
|
{
|
|
if(cp.pipe == c)
|
|
{
|
|
cp.state = orig_state;
|
|
}
|
|
}
|
|
return;
|
|
}
|
|
imgsize=-2;
|
|
pipe->Write((char*)&imgsize, sizeof(_i64), (int)receive_timeouttime);
|
|
}
|
|
|
|
void ClientConnector::waitForPings(IScopedLock *lock)
|
|
{
|
|
Server->Log("Waiting for pings...", LL_DEBUG);
|
|
while(hasChannelPing())
|
|
{
|
|
lock->relock(nullptr);
|
|
Server->wait(10);
|
|
if (run_other != nullptr)
|
|
{
|
|
run_other->runOther();
|
|
}
|
|
lock->relock(backup_mutex);
|
|
}
|
|
Server->Log("done. (Waiting for pings)", LL_DEBUG);
|
|
}
|
|
|
|
bool ClientConnector::hasChannelPing()
|
|
{
|
|
for (size_t i = 0; i < channel_pipes.size(); ++i)
|
|
{
|
|
if (channel_pipes[i].state == SChannel::EChannelState_Pinging)
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
int64 ClientConnector::getLastTokenTime(const std::string & tok)
|
|
{
|
|
IScopedLock lock(backup_mutex);
|
|
|
|
std::map<std::string, int64>::iterator it=last_token_times.find(tok);
|
|
if(it!=last_token_times.end())
|
|
{
|
|
return it->second;
|
|
}
|
|
else
|
|
{
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
void ClientConnector::tochannelSendStartbackup(RunningAction backup_type, const std::string& virtual_client)
|
|
{
|
|
std::string ts;
|
|
if(backup_type==RUNNING_INCR_FILE)
|
|
ts="START BACKUP INCR";
|
|
else if(backup_type==RUNNING_FULL_FILE)
|
|
ts="START BACKUP FULL";
|
|
else if(backup_type==RUNNING_FULL_IMAGE)
|
|
ts="START IMAGE FULL";
|
|
else if(backup_type==RUNNING_INCR_IMAGE)
|
|
ts="START IMAGE INCR";
|
|
else
|
|
return;
|
|
|
|
IScopedLock lock(backup_mutex);
|
|
IScopedLock lock_process(process_mutex);
|
|
lasttime=Server->getTimeMS();
|
|
if (getActiveProcess(x_pingtimeout) != nullptr)
|
|
{
|
|
tcpstack.Send(pipe, "RUNNING");
|
|
}
|
|
else
|
|
{
|
|
lock_process.relock(nullptr);
|
|
|
|
size_t selidx = 0;
|
|
for (size_t i = 0; i < channel_pipes.size(); ++i)
|
|
{
|
|
if (channel_pipes[i].virtual_client == virtual_client)
|
|
{
|
|
selidx = i;
|
|
}
|
|
}
|
|
|
|
bool ok=false;
|
|
if(selidx<channel_pipes.size())
|
|
{
|
|
CTCPStack tmpstack(channel_pipes[selidx].internet_connection);
|
|
_u32 rc=(_u32)tmpstack.Send(channel_pipes[selidx].pipe, ts);
|
|
if(rc!=0)
|
|
ok=true;
|
|
|
|
if(!ok)
|
|
{
|
|
tcpstack.Send(pipe, "FAILED");
|
|
}
|
|
else
|
|
{
|
|
tcpstack.Send(pipe, "OK");
|
|
}
|
|
}
|
|
else
|
|
{
|
|
tcpstack.Send(pipe, "NO SERVER");
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
void ClientConnector::doQuitClient(void)
|
|
{
|
|
do_quit=true;
|
|
}
|
|
|
|
bool ClientConnector::isQuitting(void)
|
|
{
|
|
return do_quit;
|
|
}
|
|
|
|
bool ClientConnector::isHashdataOkay(void)
|
|
{
|
|
return hashdataok;
|
|
}
|
|
|
|
void ClientConnector::ImageErr(const std::string &msg)
|
|
{
|
|
Server->Log(msg, LL_ERROR);
|
|
#ifdef _WIN32
|
|
uint64 bs=0xFFFFFFFFFFFFFFFF;
|
|
#else
|
|
uint64 bs=0xFFFFFFFFFFFFFFFFLLU;
|
|
#endif
|
|
char *buffer=new char[sizeof(uint64)+msg.size()];
|
|
memcpy(buffer, &bs, sizeof(uint64) );
|
|
memcpy(&buffer[sizeof(uint64)], msg.c_str(), msg.size());
|
|
pipe->Write(buffer, sizeof(uint64)+msg.size());
|
|
delete [] buffer;
|
|
}
|
|
|
|
void ClientConnector::setIsInternetConnection(void)
|
|
{
|
|
internet_conn=true;
|
|
tcpstack.setAddChecksum(true);
|
|
}
|
|
|
|
void ClientConnector::update_silent(void)
|
|
{
|
|
Server->getThreadPool()->execute(new UpdateSilentThread(), "silent client update");
|
|
}
|
|
|
|
bool ClientConnector::calculateFilehashesOnClient(const std::string& clientsubname, int facet_id)
|
|
{
|
|
if(internet_conn)
|
|
{
|
|
std::string settings_fn = "urbackup/data_" + convert(facet_id) + "/settings.cfg";
|
|
if(!clientsubname.empty())
|
|
{
|
|
settings_fn = "urbackup/data_" + convert(facet_id) + "/settings_"+clientsubname+".cfg";
|
|
}
|
|
ISettingsReader *curr_settings=Server->createFileSettingsReader(settings_fn);
|
|
|
|
if (curr_settings == nullptr)
|
|
return false;
|
|
|
|
std::string val;
|
|
if(curr_settings->getValue("internet_calculate_filehashes_on_client", &val)
|
|
|| curr_settings->getValue("internet_calculate_filehashes_on_client_def", &val))
|
|
{
|
|
if(val=="true")
|
|
{
|
|
Server->destroy(curr_settings);
|
|
return true;
|
|
}
|
|
}
|
|
|
|
Server->destroy(curr_settings);
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
bool ClientConnector::getBackupDest(const std::string& clientsubname, int facet_id,
|
|
std::string& dest, std::string& dest_params, str_map& dest_secret_params,
|
|
std::string& computername, size_t& max_backups, std::string& perm_uid)
|
|
{
|
|
std::string settings_fn = "urbackup/data_"+convert(facet_id )+"/settings.cfg";
|
|
if (!clientsubname.empty())
|
|
{
|
|
settings_fn = "urbackup/data_"+convert(facet_id )+"/settings_" + clientsubname + ".cfg";
|
|
}
|
|
|
|
std::string secrets_fn = "urbackup/secrets_"+std::to_string(facet_id)+".cfg";
|
|
if (!clientsubname.empty())
|
|
{
|
|
secrets_fn = "urbackup/secrets_" + std::to_string(facet_id) + "_"+clientsubname+".cfg";
|
|
}
|
|
|
|
std::unique_ptr<ISettingsReader> curr_settings(Server->createFileSettingsReader(settings_fn));
|
|
|
|
if (!curr_settings)
|
|
return false;
|
|
|
|
std::string val;
|
|
if (curr_settings->getValue("backup_dest_url", &val)
|
|
|| curr_settings->getValue("backup_dest_url_def", &val))
|
|
{
|
|
dest = val;
|
|
}
|
|
|
|
if (curr_settings->getValue("backup_dest_params", &val)
|
|
|| curr_settings->getValue("backup_dest_params_def", &val))
|
|
{
|
|
dest_params = val;
|
|
}
|
|
|
|
if (curr_settings->getValue("computername", &val)
|
|
|| curr_settings->getValue("computername_def", &val))
|
|
{
|
|
computername = val;
|
|
}
|
|
|
|
if (curr_settings->getValue("min_file_incr", &val)
|
|
|| curr_settings->getValue("min_file_incr_def", &val))
|
|
{
|
|
max_backups = watoi(val);
|
|
}
|
|
|
|
if (curr_settings->getValue("perm_uid", &val))
|
|
{
|
|
perm_uid = watoi(val);
|
|
}
|
|
|
|
if (computername.empty())
|
|
{
|
|
computername = IndexThread::getFileSrv()->getServerName();
|
|
}
|
|
|
|
if (!FileExists(secrets_fn))
|
|
{
|
|
std::string encryption_key;
|
|
encryption_key.resize(16);
|
|
Server->secureRandomFill(&encryption_key[0], encryption_key.size());
|
|
|
|
write_file_only_admin("encryption_key=" + bytesToHex(encryption_key) + "\n", secrets_fn);
|
|
}
|
|
|
|
std::unique_ptr<ISettingsReader> curr_secrets(Server->createFileSettingsReader(secrets_fn));
|
|
|
|
if (curr_secrets)
|
|
{
|
|
for (auto key : curr_secrets->getKeys())
|
|
{
|
|
dest_secret_params[key] = curr_secrets->getValue(key);
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
bool ClientConnector::isBackupRunning()
|
|
{
|
|
IScopedLock lock(backup_mutex);
|
|
|
|
int pcdone;
|
|
std::string job = getCurrRunningJob(false, pcdone);
|
|
|
|
return job!="NOA";
|
|
}
|
|
|
|
bool ClientConnector::tochannelSendLocked(bool locked)
|
|
{
|
|
std::string msg = std::string("LOCKED locked=") + (locked ? "1" : "0");
|
|
bool rc = sendMessageToAllChannels(msg, 10000);
|
|
if (rc)
|
|
{
|
|
IScopedLock lock(ident_mutex);
|
|
last_locked = locked;
|
|
}
|
|
return rc;
|
|
}
|
|
|
|
bool ClientConnector::tochannelSendChanges( const char* changes, size_t changes_size)
|
|
{
|
|
IScopedLock lock(backup_mutex);
|
|
|
|
if(channel_pipes.empty())
|
|
{
|
|
has_file_changes = true;
|
|
return false;
|
|
}
|
|
|
|
std::string changes_str = "CHANGES "+std::string(changes, changes+changes_size);
|
|
|
|
CTCPStack tmpstack(channel_pipes.front().internet_connection);
|
|
if(tmpstack.Send(channel_pipes.front().pipe, changes_str)!=changes_str.size())
|
|
{
|
|
has_file_changes = true;
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
int ClientConnector::getCapabilities(IDatabase* db)
|
|
{
|
|
int capa=0;
|
|
if(channel_pipes.size()==0)
|
|
{
|
|
capa=last_capa;
|
|
capa|=DONT_ALLOW_STARTING_FILE_BACKUPS;
|
|
capa|=DONT_ALLOW_STARTING_IMAGE_BACKUPS;
|
|
}
|
|
else
|
|
{
|
|
capa=INT_MAX;
|
|
for(size_t i=0;i<channel_pipes.size();++i)
|
|
{
|
|
capa=capa & channel_pipes[i].capa;
|
|
}
|
|
|
|
if(capa!=last_capa)
|
|
{
|
|
IDatabase *db=Server->getDatabase(Server->getThreadID(), URBACKUPDB_CLIENT);
|
|
IQuery *cq=db->Prepare("UPDATE misc SET tvalue=? WHERE tkey='last_capa'", false);
|
|
if(cq!=nullptr)
|
|
{
|
|
cq->Bind(capa);
|
|
if(cq->Write(0))
|
|
{
|
|
last_capa=capa;
|
|
}
|
|
cq->Reset();
|
|
db->destroyQuery(cq);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (!retrieved_has_components)
|
|
{
|
|
retrieved_has_components = true;
|
|
|
|
db_results res_components = db->Read("SELECT tvalue FROM misc WHERE tkey='has_components'");
|
|
if (!res_components.empty()
|
|
&& res_components[0]["tvalue"] == "1")
|
|
{
|
|
status_has_components = true;
|
|
}
|
|
else
|
|
{
|
|
status_has_components = false;
|
|
}
|
|
}
|
|
|
|
if (!status_has_components)
|
|
{
|
|
capa |= STATUS_NO_COMPONENTS;
|
|
}
|
|
|
|
return capa;
|
|
}
|
|
|
|
bool ClientConnector::multipleChannelServers()
|
|
{
|
|
if (channel_pipes.size() <= 1)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
std::string min_token = channel_pipes[0].token;
|
|
for (size_t i = 1; i < channel_pipes.size(); ++i)
|
|
{
|
|
if (channel_pipes[i].token.size() < min_token.size())
|
|
{
|
|
min_token = channel_pipes[i].token;
|
|
}
|
|
}
|
|
|
|
for (size_t i = 0; i < channel_pipes.size(); ++i)
|
|
{
|
|
if (min_token != channel_pipes[i].token
|
|
&& !next(channel_pipes[i].token, 0, min_token) )
|
|
{
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
void ClientConnector::exit_backup_immediate(int rc)
|
|
{
|
|
std::string backup_imm = Server->getServerParameter("backup_immediate");
|
|
|
|
if (!backup_imm.empty()
|
|
&& (backup_imm == "incr-file"
|
|
|| backup_imm == "full-file"))
|
|
{
|
|
if (rc != 0)
|
|
{
|
|
Server->Log("File backup failed.", LL_ERROR);
|
|
}
|
|
|
|
exit(rc);
|
|
}
|
|
}
|
|
|
|
void ClientConnector::createFacet(const std::string& server_identity, const std::string& facet_name)
|
|
{
|
|
IDatabase* db = Server->getDatabase(Server->getThreadID(), URBACKUPDB_CLIENT);
|
|
ClientDAO clientdao(db);
|
|
|
|
ClientDAO::SClientFacet init_facet = clientdao.getClientFacet("");
|
|
if (init_facet.exists)
|
|
{
|
|
clientdao.updateClientFacet(server_identity, init_facet.id);
|
|
}
|
|
|
|
ClientDAO::SClientFacet facet = clientdao.getClientFacet(server_identity);
|
|
|
|
if (!facet.exists)
|
|
{
|
|
clientdao.addClientFacet(facet_name, server_identity);
|
|
facet.id = static_cast<int>(db->getLastInsertID());
|
|
}
|
|
|
|
std::string facet_dir = "urbackup" + os_file_sep() + "data_" + convert(facet.id);
|
|
if (!os_directory_exists(facet_dir))
|
|
{
|
|
os_create_dir(facet_dir);
|
|
#ifndef _DEBUG
|
|
change_file_permissions_admin_only(facet_dir);
|
|
#endif
|
|
}
|
|
}
|
|
|
|
int ClientConnector::getFacetId(const std::string& server_identity)
|
|
{
|
|
IDatabase* db = Server->getDatabase(Server->getThreadID(), URBACKUPDB_CLIENT);
|
|
ClientDAO clientdao(db);
|
|
|
|
ClientDAO::SClientFacet facet = clientdao.getClientFacet(server_identity);
|
|
return facet.id;
|
|
}
|
|
|
|
int ClientConnector::getFacetIdByName(const std::string& facet_name)
|
|
{
|
|
IDatabase* db = Server->getDatabase(Server->getThreadID(), URBACKUPDB_CLIENT);
|
|
ClientDAO clientdao(db);
|
|
|
|
ClientDAO::SClientFacet facet = clientdao.getClientFacetByName(facet_name);
|
|
return facet.id;
|
|
}
|
|
|
|
IPipe* ClientConnector::getFileServConnection(const std::string& server_token, unsigned int timeoutms)
|
|
{
|
|
IScopedLock lock(backup_mutex);
|
|
|
|
int64 starttime = Server->getTimeMS();
|
|
|
|
do
|
|
{
|
|
for(size_t i=0;i<channel_pipes.size();++i)
|
|
{
|
|
if(channel_pipes[i].make_fileserv!=nullptr &&
|
|
channel_pipes[i].token==server_token &&
|
|
!(*channel_pipes[i].make_fileserv))
|
|
{
|
|
*channel_pipes[i].make_fileserv=true;
|
|
}
|
|
}
|
|
|
|
lock.relock(nullptr);
|
|
Server->wait(100);
|
|
lock.relock(backup_mutex);
|
|
|
|
for(size_t i=0;i<fileserv_connections.size();++i)
|
|
{
|
|
if(fileserv_connections[i].token==server_token )
|
|
{
|
|
IPipe* ret = fileserv_connections[i].pipe;
|
|
fileserv_connections.erase(fileserv_connections.begin()+i);
|
|
return ret;
|
|
}
|
|
}
|
|
|
|
} while (Server->getTimeMS()-starttime<timeoutms);
|
|
|
|
Server->Log("Timeout while getting a fileserv connection");
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
bool ClientConnector::closeSocket( void )
|
|
{
|
|
if(state!=CCSTATE_FILESERV)
|
|
{
|
|
return true;
|
|
}
|
|
else
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
|
|
void ClientConnector::sendStatus()
|
|
{
|
|
IDatabase *db=Server->getDatabase(Server->getThreadID(), URBACKUPDB_CLIENT);
|
|
|
|
IScopedLock lock(backup_mutex);
|
|
|
|
state = CCSTATE_STATUS;
|
|
|
|
int64 last_backup_time = getLastBackupTime();
|
|
|
|
std::string ret;
|
|
if (last_backup_time > 0)
|
|
{
|
|
ret += convert(last_backup_time);
|
|
}
|
|
|
|
int pcdone = -1;
|
|
ret += "#" + getCurrRunningJob(true, pcdone);
|
|
|
|
ret+="#"+convert(pcdone);
|
|
|
|
if(IdleCheckerThread::getPause())
|
|
{
|
|
ret+="#P";
|
|
}
|
|
else
|
|
{
|
|
ret+="#NP";
|
|
}
|
|
|
|
ret+="#capa="+convert(getCapabilities(db));
|
|
|
|
{
|
|
IScopedLock lock(ident_mutex);
|
|
if(!new_server_idents.empty())
|
|
{
|
|
ret+="&new_ident="+new_server_idents[new_server_idents.size()-1];
|
|
new_server_idents.erase(new_server_idents.begin()+new_server_idents.size()-1);
|
|
}
|
|
}
|
|
|
|
ret+="&has_server=";
|
|
if(channel_pipes.empty())
|
|
{
|
|
ret+="false";
|
|
}
|
|
else
|
|
{
|
|
ret+="true";
|
|
}
|
|
|
|
if(restore_ok_status==RestoreOk_Wait)
|
|
{
|
|
ret+="&restore_ask="+convert(ask_restore_ok);
|
|
|
|
if (restore_files != nullptr)
|
|
{
|
|
if (restore_files->is_single_file())
|
|
{
|
|
ret += "&restore_file=true";
|
|
}
|
|
else
|
|
{
|
|
ret += "&restore_file=false";
|
|
}
|
|
|
|
ret += "&restore_path=" + EscapeParamString(restore_files->get_restore_path());
|
|
|
|
ret += "&process_id=" + convert(restore_files->get_local_process_id());
|
|
}
|
|
}
|
|
|
|
if(needs_restore_restart>0)
|
|
{
|
|
ret+="&needs_restore_restart="+convert(needs_restore_restart);
|
|
}
|
|
|
|
tcpstack.Send(pipe, ret);
|
|
|
|
db->destroyAllQueries();
|
|
}
|
|
|
|
bool ClientConnector::tochannelLog(int64 log_id, const std::string& msg, int loglevel,
|
|
const std::string& identity, int timeoutms)
|
|
{
|
|
return sendMessageToChannel("LOG "+convert(log_id)+"-"+convert(loglevel)+"-"+msg, timeoutms, identity);
|
|
}
|
|
|
|
void ClientConnector::updateRestorePc(int64 local_process_id, int64 restore_id, int64 status_id, int nv, const std::string& identity,
|
|
const std::string& fn, int fn_pc, int64 total_bytes, int64 done_bytes, double speed_bpms)
|
|
{
|
|
IScopedLock lock(backup_mutex);
|
|
|
|
{
|
|
IScopedLock lock_process(process_mutex);
|
|
|
|
SRunningProcess* proc = getRunningProcess(local_process_id);
|
|
|
|
if (proc != nullptr)
|
|
{
|
|
if (nv > 100)
|
|
{
|
|
removeRunningProcess(local_process_id, total_bytes==done_bytes);
|
|
}
|
|
else
|
|
{
|
|
proc->pcdone = nv;
|
|
proc->last_pingtime = Server->getTimeMS();
|
|
proc->details = fn;
|
|
proc->detail_pc = fn_pc;
|
|
proc->total_bytes = total_bytes;
|
|
proc->done_bytes = done_bytes;
|
|
proc->speed_bpms = speed_bpms;
|
|
}
|
|
}
|
|
}
|
|
|
|
sendMessageToChannel("RESTORE PERCENT pc="+convert(nv)+"&status_id="+convert(status_id)+"&id="+convert(restore_id)
|
|
+"&details="+EscapeParamString(fn)+"&detail_pc="+convert(fn_pc)+"&total_bytes="+convert(total_bytes)+"&done_bytes="+convert(done_bytes)
|
|
+"&speed_bpms="+convert(speed_bpms),
|
|
0, identity);
|
|
}
|
|
|
|
bool ClientConnector::restoreDone( int64 log_id, int64 status_id, int64 restore_id, bool success, const std::string& identity )
|
|
{
|
|
return sendMessageToChannel("RESTORE DONE status_id="+convert(status_id)+
|
|
"&log_id=" + convert(log_id) +
|
|
"&id=" + convert(restore_id) +
|
|
"&success=" + convert(success), 60000, identity);
|
|
}
|
|
|
|
void ClientConnector::updateLocalBackupPc(int64 local_process_id, int64 backup_id, int64 status_id, int nv,
|
|
const std::string& identity, const std::string& details, int64 total_bytes, int64 done_bytes, double speed_bpms,
|
|
int64 eta, int64 eta_set_time)
|
|
{
|
|
IScopedLock lock(backup_mutex);
|
|
|
|
{
|
|
IScopedLock lock_process(process_mutex);
|
|
|
|
SRunningProcess* proc = getRunningProcess(local_process_id);
|
|
|
|
if (proc != nullptr)
|
|
{
|
|
if (nv > 100)
|
|
{
|
|
removeRunningProcess(local_process_id, total_bytes == done_bytes);
|
|
}
|
|
else
|
|
{
|
|
proc->pcdone = nv;
|
|
proc->last_pingtime = Server->getTimeMS();
|
|
proc->details = details;
|
|
proc->total_bytes = total_bytes;
|
|
proc->done_bytes = done_bytes;
|
|
proc->speed_bpms = speed_bpms;
|
|
proc->eta_ms = eta;
|
|
}
|
|
}
|
|
}
|
|
|
|
sendMessageToChannel("BACKUP PERCENT pc=" + convert(nv) + "&status_id=" + convert(status_id) + "&id=" + convert(backup_id)
|
|
+ "&details=" + EscapeParamString(details) + "&total_bytes=" + convert(total_bytes) + "&done_bytes=" + convert(done_bytes)
|
|
+ "&speed_bpms=" + convert(speed_bpms)+"&eta="+std::to_string(eta) +"&eta_set_time="+std::to_string(Server->getTimeMS() - eta_set_time),
|
|
0, identity);
|
|
}
|
|
|
|
bool ClientConnector::localBackupDone(int64 log_id, int64 status_id, int64 backup_id, bool success, const std::string& identity)
|
|
{
|
|
return sendMessageToChannel("BACKUP DONE status_id=" + convert(status_id) +
|
|
"&log_id=" + convert(log_id) +
|
|
"&id=" + convert(backup_id) +
|
|
"&success=" + convert(success), 60000, identity);
|
|
}
|
|
|
|
bool ClientConnector::sendMessageToChannel( const std::string& msg, int timeoutms, const std::string& identity )
|
|
{
|
|
IScopedLock lock(backup_mutex);
|
|
|
|
int64 starttime = Server->getTimeMS();
|
|
|
|
do
|
|
{
|
|
for(size_t i=0;i<channel_pipes.size();++i)
|
|
{
|
|
if(channel_pipes[i].token == identity)
|
|
{
|
|
CTCPStack tmpstack(channel_pipes[i].internet_connection);
|
|
if(tmpstack.Send(channel_pipes[i].pipe, msg)!=msg.size())
|
|
{
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
}
|
|
|
|
if (timeoutms == 0)
|
|
return false;
|
|
|
|
lock.relock(nullptr);
|
|
Server->wait(100);
|
|
lock.relock(backup_mutex);
|
|
} while(Server->getTimeMS()-starttime<timeoutms);
|
|
|
|
return false;
|
|
}
|
|
|
|
bool ClientConnector::sendMessageToAllChannels(const std::string& msg, int timeoutms)
|
|
{
|
|
std::set<std::string> done_tokens;
|
|
IScopedLock lock(backup_mutex);
|
|
|
|
int64 starttime = Server->getTimeMS();
|
|
|
|
do
|
|
{
|
|
for (size_t i = 0; i < channel_pipes.size(); ++i)
|
|
{
|
|
if (done_tokens.find(channel_pipes[i].token) == done_tokens.end())
|
|
{
|
|
CTCPStack tmpstack(channel_pipes[i].internet_connection);
|
|
if (tmpstack.Send(channel_pipes[i].pipe, msg) == msg.size())
|
|
{
|
|
done_tokens.insert(channel_pipes[i].token);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (done_tokens.size() == channel_pipes.size())
|
|
return true;
|
|
|
|
if (timeoutms == 0)
|
|
return false;
|
|
|
|
lock.relock(nullptr);
|
|
Server->wait(100);
|
|
lock.relock(backup_mutex);
|
|
} while (Server->getTimeMS() - starttime < timeoutms);
|
|
|
|
return done_tokens.size()==channel_pipes.size();
|
|
}
|
|
|
|
std::string ClientConnector::getAccessTokensParams(const std::string& tokens, bool with_clientname, const std::string& virtual_client)
|
|
{
|
|
if(tokens.empty())
|
|
{
|
|
return std::string();
|
|
}
|
|
|
|
std::unique_ptr<ISettingsReader> access_keys(
|
|
Server->createFileSettingsReader("urbackup/access_keys.properties"));
|
|
|
|
std::vector<std::string> server_token_keys = access_keys->getKeys();
|
|
|
|
if(server_token_keys.empty())
|
|
{
|
|
Server->Log("No access key present", LL_ERROR);
|
|
return std::string();
|
|
}
|
|
|
|
std::string ret;
|
|
|
|
std::string session_key;
|
|
session_key.resize(32);
|
|
Server->secureRandomFill(&session_key[0], session_key.size());
|
|
|
|
bool has_token=false;
|
|
size_t tokenidx = 0;
|
|
for(size_t i=0;i<server_token_keys.size();++i)
|
|
{
|
|
if (!next(server_token_keys[i], 0, "key.")
|
|
&& !next(server_token_keys[i], 0, "last.key.") )
|
|
{
|
|
continue;
|
|
}
|
|
|
|
std::string server_key;
|
|
|
|
if(access_keys->getValue(server_token_keys[i],
|
|
&server_key) && !server_key.empty())
|
|
{
|
|
ret += "&tokens"+convert(tokenidx)+"="+base64_encode_dash(
|
|
crypto_fak->encryptAuthenticatedAES(session_key,
|
|
server_key, 1));
|
|
++tokenidx;
|
|
has_token=true;
|
|
}
|
|
}
|
|
|
|
if(has_token)
|
|
{
|
|
ret += "&token_data="+base64_encode_dash(
|
|
crypto_fak->encryptAuthenticatedAES(tokens,
|
|
session_key, 1) );
|
|
}
|
|
|
|
if(with_clientname)
|
|
{
|
|
std::unique_ptr<ISettingsReader> settings(
|
|
Server->createFileSettingsReader("urbackup/data/settings.cfg"));
|
|
|
|
std::string computername;
|
|
if( !settings->getValue("computername", &computername) )
|
|
{
|
|
settings->getValue("computername_def", &computername);
|
|
}
|
|
if(computername.empty())
|
|
{
|
|
computername = IndexThread::getFileSrv()->getServerName();
|
|
}
|
|
|
|
if(!computername.empty())
|
|
{
|
|
if (!virtual_client.empty())
|
|
{
|
|
computername += "[" + virtual_client + "]";
|
|
}
|
|
|
|
ret+="&clientname="+EscapeParamString(computername);
|
|
}
|
|
}
|
|
|
|
return ret;
|
|
}
|
|
|
|
bool ClientConnector::sendChannelPacket(const SChannel& channel, const std::string& msg)
|
|
{
|
|
CTCPStack localstack(channel.internet_connection);
|
|
return localstack.Send(channel.pipe, msg) == msg.size();
|
|
}
|
|
|
|
bool ClientConnector::versionNeedsUpdate(const std::string & local_version, const std::string & server_version)
|
|
{
|
|
std::vector<std::string> local_features;
|
|
int ilocal_version = parseVersion(local_version, local_features);
|
|
std::vector<std::string> server_features;
|
|
int iserver_version = parseVersion(server_version, server_features);
|
|
|
|
for (size_t i = 0; i < local_features.size(); ++i)
|
|
{
|
|
if (std::find(server_features.begin(), server_features.end(), local_features[i]) == server_features.end())
|
|
{
|
|
Server->Log("Server update does not have feature " + local_features[i] + ". Not updating.", LL_INFO);
|
|
return false;
|
|
}
|
|
}
|
|
|
|
if (iserver_version > ilocal_version)
|
|
{
|
|
Server->Log("Server has new version "+convert(iserver_version)+" (client version: "+convert(ilocal_version)+"). Updating...", LL_INFO);
|
|
return true;
|
|
}
|
|
|
|
if (iserver_version < ilocal_version)
|
|
{
|
|
Server->Log("Not down-grading from current client version " + convert(ilocal_version) + " to server version " + convert(iserver_version), LL_INFO);
|
|
return false;
|
|
}
|
|
|
|
for (size_t i = 0; i < server_features.size(); ++i)
|
|
{
|
|
if (std::find(local_features.begin(), local_features.end(), server_features[i]) == local_features.end())
|
|
{
|
|
Server->Log("Client currently does not have feature " + server_features[i] + ". Updating...", LL_INFO);
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
int ClientConnector::parseVersion(const std::string & version, std::vector<std::string>& features)
|
|
{
|
|
if (version.find("-") == std::string::npos)
|
|
{
|
|
return atoi(version.c_str());
|
|
}
|
|
else
|
|
{
|
|
Tokenize(getafter("-", version), features, ",");
|
|
|
|
return atoi(getuntil("-", version).c_str());
|
|
}
|
|
}
|
|
|
|
void ClientConnector::requestRestoreRestart()
|
|
{
|
|
IScopedLock lock(backup_mutex);
|
|
++needs_restore_restart;
|
|
status_updated=true;
|
|
}
|
|
|
|
std::string ClientConnector::getHasNoRecentBackup()
|
|
{
|
|
IScopedLock lock(backup_mutex);
|
|
|
|
static int64 last_monotonic_time = Server->getTimeMS();
|
|
static int64 last_nonmonotonic_time = Server->getTimeSeconds();
|
|
|
|
int64 cmontime = Server->getTimeMS();
|
|
int64 cnmontime = Server->getTimeSeconds();
|
|
|
|
int64 monotonic_passed_time = cmontime - last_monotonic_time;
|
|
int64 nonmonotonic_passed_time = cnmontime - last_nonmonotonic_time;
|
|
|
|
last_monotonic_time = cmontime;
|
|
last_nonmonotonic_time = cnmontime;
|
|
|
|
if (monotonic_passed_time+60000 < nonmonotonic_passed_time * 1000)
|
|
{
|
|
Server->Log("Detected forward time jump. Resetting system start time.");
|
|
service_starttime = cmontime;
|
|
}
|
|
|
|
int64 last_backup_time = getLastBackupTime();
|
|
|
|
if(last_backup_time==0)
|
|
{
|
|
return "NO_RECENT";
|
|
}
|
|
|
|
if(Server->getTimeMS()-service_starttime<backup_alert_delay)
|
|
{
|
|
return "NOA";
|
|
}
|
|
|
|
if(Server->getTimeSeconds()-last_backup_time<= backup_interval)
|
|
{
|
|
return "NOA";
|
|
}
|
|
|
|
return "NO_RECENT";
|
|
}
|
|
|
|
|
|
void ClientConnector::refreshSessionFromChannel(const std::string& endpoint_name)
|
|
{
|
|
IScopedLock lock(backup_mutex);
|
|
|
|
for (size_t i = 0; i<channel_pipes.size(); ++i)
|
|
{
|
|
if (channel_pipes[i].pipe == pipe)
|
|
{
|
|
if (!channel_pipes[i].server_identity.empty())
|
|
{
|
|
std::string secret_key;
|
|
ServerIdentityMgr::checkServerSessionIdentity(channel_pipes[i].server_identity, endpoint_name, secret_key);
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
bool ClientConnector::localBackup(std::string dest_url, const std::string& dest_params,
|
|
const str_map& dest_secret_params, const std::string& computername,
|
|
bool full, size_t max_backups, const std::string& server_identity, str_map& params,
|
|
const std::string& perm_uid)
|
|
{
|
|
if (dest_url.empty())
|
|
return false;
|
|
|
|
if (next(dest_url, 0, "urbackup://"))
|
|
return false;
|
|
|
|
dest_url = greplace("$CLIENTNAME$", computername, dest_url);
|
|
dest_url = greplace("$CLIENTUID$", perm_uid, dest_url);
|
|
|
|
int64 server_id = watoi64(params["status_id"]);
|
|
|
|
int group = c_group_default;
|
|
|
|
str_map::iterator it_group = params.find("group");
|
|
if (it_group != params.end())
|
|
{
|
|
group = watoi(it_group->second);
|
|
}
|
|
|
|
std::string clientsubname;
|
|
|
|
str_map::iterator it_clientsubname = params.find("clientsubname");
|
|
if (it_clientsubname != params.end())
|
|
{
|
|
clientsubname = conv_filename(it_clientsubname->second);
|
|
}
|
|
|
|
int facet_id = getFacetId(ServerIdentityMgr::getIdentityFromSessionIdentity(server_identity));
|
|
|
|
std::string async_id;
|
|
async_id.resize(16);
|
|
Server->randomFill(&async_id[0], async_id.size());
|
|
|
|
SRunningProcess new_proc;
|
|
|
|
new_proc.action = full ? RUNNING_FULL_FILE : RUNNING_INCR_FILE;
|
|
new_proc.server_id = server_id;
|
|
new_proc.id = ++curr_backup_running_id;
|
|
new_proc.server_token = server_token;
|
|
|
|
IScopedLock process_lock(process_mutex);
|
|
|
|
removeTimedOutProcesses(server_token, true);
|
|
|
|
running_processes.push_back(new_proc);
|
|
|
|
int64 log_id = watoi64(params["log_id"]);
|
|
|
|
THREADPOOL_TICKET backup_ticket;
|
|
if (full)
|
|
{
|
|
LocalFullFileBackup* fb = new LocalFullFileBackup(group,
|
|
clientsubname, new_proc.id, log_id, server_id, new_proc.id,
|
|
server_token, server_identity, facet_id, max_backups,
|
|
dest_url, dest_params, dest_secret_params);
|
|
|
|
backup_ticket = Server->getThreadPool()->execute(fb, "lfbackup full");
|
|
}
|
|
else
|
|
{
|
|
LocalIncrFileBackup* fb = new LocalIncrFileBackup(group,
|
|
clientsubname, new_proc.id, log_id, server_id, new_proc.id, server_token,
|
|
server_identity, facet_id, max_backups,
|
|
dest_url, dest_params, dest_secret_params);
|
|
|
|
backup_ticket = Server->getThreadPool()->execute(fb, "lfbackup incr");
|
|
}
|
|
|
|
SAsyncFileList new_async_backup = {
|
|
Server->getTimeMS(),
|
|
0,
|
|
0,
|
|
backup_ticket
|
|
};
|
|
|
|
async_file_index[async_id] = new_async_backup;
|
|
|
|
tcpstack.Send(pipe, "ASYNC-async_id=" + bytesToHex(async_id));
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
bool ClientConnector::updateDefaultDirsSetting(IDatabase* db, bool all_virtual_clients, int group_offset, bool update_use, int facet_id)
|
|
{
|
|
db_results res_virtual_clients;
|
|
if(all_virtual_clients)
|
|
res_virtual_clients = db->Read("SELECT virtual_client, group_offset FROM virtual_client_group_offsets WHERE facet="+convert(facet_id));
|
|
else
|
|
res_virtual_clients = db->Read("SELECT virtual_client, group_offset FROM virtual_client_group_offsets WHERE group_offset="+convert(group_offset)+" AND facet="+convert(facet_id));
|
|
|
|
if (all_virtual_clients || group_offset == 0)
|
|
{
|
|
db_single_result res_def_client;
|
|
res_def_client["group_offset"] = "0";
|
|
res_virtual_clients.push_back(res_def_client);
|
|
}
|
|
|
|
|
|
bool mod = false;
|
|
for (size_t i = 0; i < res_virtual_clients.size(); ++i)
|
|
{
|
|
db_single_result& res_virtual_client = res_virtual_clients[i];
|
|
|
|
int curr_group_offset = watoi(res_virtual_client["group_offset"]);
|
|
|
|
db_results res_paths = db->Read("SELECT path, name, optional, server_default FROM backupdirs WHERE symlinked=0 AND tgroup=" + convert(curr_group_offset)+" AND facet="+convert(facet_id));
|
|
|
|
int default_dirs_use = c_use_value_client;
|
|
bool has_client_path = false;
|
|
std::string default_dirs;
|
|
for (size_t j = 0; j < res_paths.size(); ++j)
|
|
{
|
|
db_single_result& res_path = res_paths[j];
|
|
|
|
if (res_path["server_default"] != "0")
|
|
{
|
|
default_dirs_use |= c_use_group | c_use_value;
|
|
continue;
|
|
}
|
|
else
|
|
{
|
|
has_client_path = true;
|
|
}
|
|
|
|
int optional = watoi(res_path["optional"]);
|
|
|
|
std::vector<std::pair<int, std::string> > flag_mapping = getFlagStrMapping();
|
|
|
|
std::string str_flags;
|
|
std::string str_flags_default;
|
|
for (size_t j = 0; j < flag_mapping.size(); ++j)
|
|
{
|
|
if (optional & flag_mapping[j].first)
|
|
{
|
|
if (!str_flags.empty()) str_flags += ",";
|
|
str_flags += flag_mapping[j].second;
|
|
}
|
|
|
|
if (EBackupDirFlags_Default&flag_mapping[j].first)
|
|
{
|
|
if (!str_flags_default.empty()) str_flags_default += ",";
|
|
str_flags_default += flag_mapping[j].second;
|
|
}
|
|
}
|
|
|
|
if (!default_dirs.empty())
|
|
default_dirs += ";";
|
|
|
|
if (!str_flags.empty()
|
|
&& str_flags != str_flags_default)
|
|
str_flags = "/" + str_flags;
|
|
else if (str_flags == str_flags_default)
|
|
str_flags.clear();
|
|
|
|
std::string path = greplace("%2F", "/", EscapePathParamString(res_path["path"]));
|
|
|
|
default_dirs += path +"|"+EscapePathParamString(res_path["name"]) + str_flags;
|
|
}
|
|
|
|
std::string settings_fn = "urbackup/data_"+convert(facet_id)+"/settings.cfg";
|
|
if (curr_group_offset !=0)
|
|
{
|
|
settings_fn = "urbackup/data_"+convert(facet_id)+"/settings_" + conv_filename(res_virtual_client["virtual_client"]) + ".cfg";
|
|
}
|
|
|
|
if (res_paths.empty() &&
|
|
!FileExists(settings_fn))
|
|
continue;
|
|
|
|
std::unique_ptr<ISettingsReader> curr_settings(Server->createFileSettingsReader(settings_fn));
|
|
|
|
if (curr_settings.get() != nullptr)
|
|
{
|
|
str_map settings_repl;
|
|
|
|
int64 default_dirs_use_lm_orig = curr_settings->getValue("default_dirs.use_lm", 0LL);
|
|
|
|
int64 ctime = Server->getTimeSeconds();
|
|
if (default_dirs_use_lm_orig > ctime)
|
|
ctime = default_dirs_use_lm_orig + 1;
|
|
|
|
if (curr_settings->getValue("default_dirs.use", 0) == 0)
|
|
{
|
|
settings_repl["default_dirs.use"] = convert(default_dirs_use);
|
|
settings_repl["default_dirs.use_lm"] = convert(ctime);
|
|
mod = true;
|
|
}
|
|
else if (update_use)
|
|
{
|
|
int curr_use = curr_settings->getValue("default_dirs.use", 0);
|
|
if (has_client_path && !(curr_use & c_use_value_client))
|
|
{
|
|
settings_repl["default_dirs.use"] = convert(curr_use|c_use_value_client);
|
|
settings_repl["default_dirs.use_lm"] = convert(ctime);
|
|
mod = true;
|
|
}
|
|
}
|
|
|
|
|
|
if (curr_settings->getValue("default_dirs.client", "") != default_dirs)
|
|
{
|
|
settings_repl["default_dirs.client"] = default_dirs;
|
|
mod = true;
|
|
}
|
|
|
|
if (!settings_repl.empty())
|
|
{
|
|
std::vector<std::string> keys = curr_settings->getKeys();
|
|
|
|
for (str_map::iterator it = settings_repl.begin(); it != settings_repl.end(); ++it)
|
|
{
|
|
if (std::find(keys.begin(), keys.end(), it->first) == keys.end())
|
|
keys.push_back(it->first);
|
|
}
|
|
|
|
std::string new_data;
|
|
|
|
for (size_t i = 0; i < keys.size(); ++i)
|
|
{
|
|
std::string val;
|
|
str_map::iterator it = settings_repl.find(keys[i]);
|
|
if (it != settings_repl.end())
|
|
val = it->second;
|
|
else
|
|
curr_settings->getValue(keys[i], &val);
|
|
|
|
new_data += keys[i] + "=" + val + "\n";
|
|
}
|
|
|
|
std::unique_ptr<IFile> settings_f(Server->openFile(settings_fn+".new_2", MODE_WRITE));
|
|
|
|
if (settings_f.get() != nullptr)
|
|
{
|
|
if (settings_f->Write(new_data) == new_data.size()
|
|
&& settings_f->Sync())
|
|
{
|
|
settings_f.reset();
|
|
if (!os_rename_file(settings_fn + ".new_2", settings_fn))
|
|
{
|
|
Server->Log("Error renaming " + settings_fn + ".new_2 to " + settings_fn + ". " + os_last_error_str(), LL_ERROR);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
Server->Log("Error writing new settings to " + settings_fn + ".new_2. " + os_last_error_str(), LL_ERROR);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return mod;
|
|
}
|
|
|
|
void ClientConnector::timeoutAsyncFileIndex()
|
|
{
|
|
int64 ctime = Server->getTimeMS();
|
|
IScopedLock lock(backup_mutex);
|
|
for (std::map<std::string, SAsyncFileList>::iterator it = async_file_index.begin();
|
|
it != async_file_index.end();)
|
|
{
|
|
if (ctime - it->second.last_update > async_index_timeout * 2)
|
|
{
|
|
Server->Log("Async index timeout: " + bytesToHex(it->first), LL_DEBUG);
|
|
IndexThread::removeResult(it->second.result_id);
|
|
std::map<std::string, SAsyncFileList>::iterator it_curr = it;
|
|
++it;
|
|
async_file_index.erase(it_curr);
|
|
}
|
|
else
|
|
{
|
|
++it;
|
|
}
|
|
}
|
|
}
|
|
|
|
SRunningProcess * ClientConnector::getRunningProcess(RunningAction action, std::string server_token)
|
|
{
|
|
for (size_t i = 0; i < running_processes.size(); ++i)
|
|
{
|
|
SRunningProcess& curr = running_processes[i];
|
|
if (curr.action == action
|
|
&& (curr.server_token == server_token || curr.server_token.empty() || server_token.empty())
|
|
&& curr.server_id == 0)
|
|
{
|
|
return &curr;
|
|
}
|
|
}
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
SRunningProcess * ClientConnector::getRunningFileBackupProcess(std::string server_token, int64 server_id)
|
|
{
|
|
for (size_t i = 0; i < running_processes.size(); ++i)
|
|
{
|
|
SRunningProcess& curr = running_processes[i];
|
|
if ((curr.action == RUNNING_FULL_FILE || curr.action == RUNNING_INCR_FILE || curr.action == RUNNING_RESUME_FULL_FILE || curr.action == RUNNING_RESUME_INCR_FILE)
|
|
&& (curr.server_token == server_token || curr.server_token.empty() || server_token.empty())
|
|
&& ((curr.server_id == 0 && server_id == 0) || curr.server_id == server_id))
|
|
{
|
|
return &curr;
|
|
}
|
|
}
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
SRunningProcess * ClientConnector::getRunningBackupProcess(std::string server_token, int64 server_id)
|
|
{
|
|
for (size_t i = running_processes.size(); i-- > 0;)
|
|
{
|
|
SRunningProcess& curr = running_processes[i];
|
|
if ( (curr.server_token == server_token || curr.server_token.empty() || server_token.empty())
|
|
&& ((curr.server_id == 0 && server_id == 0) || curr.server_id == server_id))
|
|
{
|
|
return &curr;
|
|
}
|
|
}
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
SRunningProcess * ClientConnector::getRunningProcess(int64 id)
|
|
{
|
|
for (size_t i = 0; i < running_processes.size(); ++i)
|
|
{
|
|
SRunningProcess& curr = running_processes[i];
|
|
if (curr.id == id)
|
|
{
|
|
return &curr;
|
|
}
|
|
}
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
SRunningProcess * ClientConnector::getActiveProcess(int64 timeout)
|
|
{
|
|
int64 ctime = Server->getTimeMS();
|
|
for (size_t i = 0; i < running_processes.size(); ++i)
|
|
{
|
|
SRunningProcess& curr = running_processes[i];
|
|
if (ctime - curr.last_pingtime<timeout)
|
|
{
|
|
return &curr;
|
|
}
|
|
}
|
|
|
|
return nullptr;
|
|
}
|
|
|
|
bool ClientConnector::removeRunningProcess(int64 id, bool success, bool consider_refs)
|
|
{
|
|
IScopedLock lock(process_mutex);
|
|
|
|
for (size_t i = 0; i < running_processes.size(); ++i)
|
|
{
|
|
SRunningProcess& curr = running_processes[i];
|
|
if (curr.id == id)
|
|
{
|
|
if (curr.refs > 0)
|
|
--curr.refs;
|
|
|
|
if (!consider_refs || curr.refs == 0)
|
|
{
|
|
Server->Log("Removing running process (1) id " + convert(curr.id) + " server_id " + convert(curr.server_id) + " token " + curr.server_token + " action " + convert((int)curr.action), LL_DEBUG);
|
|
finished_processes.push_back(SFinishedProcess(id, success));
|
|
while (finished_processes.size() > 20)
|
|
{
|
|
finished_processes.erase(finished_processes.begin());
|
|
}
|
|
running_processes.erase(running_processes.begin() + i);
|
|
return true;
|
|
}
|
|
else
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
void ClientConnector::timeoutFilesrvConnections()
|
|
{
|
|
IScopedLock lock(backup_mutex);
|
|
|
|
for (size_t i = 0; i < fileserv_connections.size();)
|
|
{
|
|
if (Server->getTimeMS() - fileserv_connections[i].starttime>60000)
|
|
{
|
|
Server->destroy(fileserv_connections[i].pipe);
|
|
fileserv_connections.erase(fileserv_connections.begin() + i);
|
|
}
|
|
else
|
|
{
|
|
++i;
|
|
}
|
|
}
|
|
}
|
|
|
|
void ClientConnector::timeoutBackupImmediate(int64 start_timeout, int64 resume_timeout, RunningAction ra, bool & has_backup, int64& starttime)
|
|
{
|
|
IScopedLock lock(backup_mutex);
|
|
|
|
bool found = false;
|
|
for (size_t i = 0; i < running_processes.size(); ++i)
|
|
{
|
|
if (Server->getTimeMS() - running_processes[i].last_pingtime > x_pingtimeout)
|
|
{
|
|
continue;
|
|
}
|
|
|
|
RunningAction cra = running_processes[i].action;
|
|
if ((ra == RUNNING_INCR_FILE || ra == RUNNING_FULL_FILE)
|
|
&& (cra == RUNNING_INCR_FILE || cra == RUNNING_FULL_FILE
|
|
|| cra == RUNNING_RESUME_INCR_FILE || cra == RUNNING_RESUME_FULL_FILE))
|
|
{
|
|
found = true;
|
|
break;
|
|
}
|
|
else if ((ra == RUNNING_INCR_IMAGE || ra == RUNNING_FULL_IMAGE)
|
|
&& (cra == RUNNING_INCR_IMAGE || cra == RUNNING_FULL_IMAGE))
|
|
{
|
|
found = true;
|
|
break;
|
|
}
|
|
}
|
|
|
|
lock.relock(nullptr);
|
|
|
|
if (found)
|
|
{
|
|
has_backup = true;
|
|
starttime = Server->getTimeMS();
|
|
}
|
|
else if(has_backup)
|
|
{
|
|
if (Server->getTimeMS() - starttime > resume_timeout)
|
|
{
|
|
Server->Log("Backup timeout after connection has been lost for " + PrettyPrintTime(resume_timeout), LL_ERROR);
|
|
exit(11);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (Server->getTimeMS() - starttime > start_timeout)
|
|
{
|
|
Server->Log("Backup timeout after backup failed to start for " + PrettyPrintTime(start_timeout), LL_ERROR);
|
|
exit(10);
|
|
}
|
|
}
|
|
}
|
|
|
|
bool ClientConnector::updateRunningPc(int64 id, int pcdone)
|
|
{
|
|
IScopedLock lock(process_mutex);
|
|
|
|
SRunningProcess* proc = getRunningProcess(id);
|
|
if (proc == nullptr)
|
|
{
|
|
return false;
|
|
}
|
|
|
|
if (proc->pcdone != pcdone)
|
|
{
|
|
status_updated = true;
|
|
}
|
|
proc->pcdone = pcdone;
|
|
|
|
return true;
|
|
}
|
|
|
|
std::string ClientConnector::actionToStr(RunningAction action)
|
|
{
|
|
switch (action)
|
|
{
|
|
case RUNNING_INCR_FILE:
|
|
return "INCR";
|
|
case RUNNING_FULL_FILE:
|
|
return "FULL";
|
|
case RUNNING_FULL_IMAGE:
|
|
return "FULLI";
|
|
case RUNNING_INCR_IMAGE:
|
|
return "INCRI";
|
|
case RUNNING_RESUME_INCR_FILE:
|
|
return "R_INCR";
|
|
case RUNNING_RESUME_FULL_FILE:
|
|
return "R_FULL";
|
|
case RUNNING_RESTORE_IMAGE:
|
|
return "RESTORE_IMAGE";
|
|
case RUNNING_RESTORE_FILE:
|
|
return "RESTORE_FILES";
|
|
default:
|
|
return "";
|
|
}
|
|
}
|
|
|
|
void ClientConnector::removeTimedOutProcesses(std::string server_token, bool file)
|
|
{
|
|
int64 ctime = Server->getTimeMS();
|
|
|
|
for (size_t i = 0; i < running_processes.size();)
|
|
{
|
|
SRunningProcess& curr = running_processes[i];
|
|
|
|
bool curr_file = (curr.action == RUNNING_FULL_FILE || curr.action == RUNNING_INCR_FILE || curr.action == RUNNING_RESUME_FULL_FILE || curr.action == RUNNING_RESUME_INCR_FILE);
|
|
|
|
if ( (curr_file || !file)
|
|
&& (curr.server_token == server_token || curr.server_token.empty() || server_token.empty())
|
|
&& ctime - curr.last_pingtime>x_pingtimeout )
|
|
{
|
|
Server->Log("Removing running process (timeout) id " + convert(curr.id) + " server_id "+convert(curr.server_id)+" token "+curr.server_token+" action " + convert((int)curr.action), LL_DEBUG);
|
|
running_processes.erase(running_processes.begin() + i);
|
|
continue;
|
|
}
|
|
|
|
++i;
|
|
}
|
|
}
|
|
|
|
int64 ClientConnector::addNewProcess(SRunningProcess proc)
|
|
{
|
|
IScopedLock lock(process_mutex);
|
|
if (proc.id == 0)
|
|
{
|
|
proc.id = ++curr_backup_running_id;
|
|
}
|
|
running_processes.push_back(proc);
|
|
return proc.id;
|
|
}
|
|
|
|
|