urbackup_backend/urbackupclient/ClientService.cpp
2021-10-23 16:34:51 +02:00

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), &params, 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), &params);
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;
}