urbackup_backend/urbackupserver/server_cleanup.cpp

3156 lines
90 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/>.
**************************************************************************/
#ifndef CLIENT_ONLY
#include "server_cleanup.h"
#include "../Interface/Server.h"
#include "../Interface/SettingsReader.h"
#include "../Interface/ThreadPool.h"
#include "../Interface/DatabaseCursor.h"
#include "database.h"
#include "../stringtools.h"
#include "server_settings.h"
#include "../urbackupcommon/os_functions.h"
#include "server_update_stats.h"
#include "server_update.h"
#include "server_status.h"
#include "ClientMain.h"
#include "server.h"
#include "snapshot_helper.h"
#include "apps/cleanup_cmd.h"
#include "dao/ServerCleanupDao.h"
#include "dao/ServerLinkDao.h"
#include "dao/ServerFilesDao.h"
#include "server_dir_links.h"
#include <stdio.h>
#include <algorithm>
#include "create_files_index.h"
#include "../urbackupcommon/WalCheckpointThread.h"
#include "../clouddrive/ObjectCollector.h"
#include "../clouddrive/IKvStoreFrontend.h"
#include "../clouddrive/IClouddriveFactory.h"
#include "../urbackupcommon/backup_url_parser.h"
#include "copy_storage.h"
#include <assert.h>
#include <set>
IMutex *ServerCleanupThread::mutex=NULL;
ICondition *ServerCleanupThread::cond=NULL;
bool ServerCleanupThread::update_stats=false;
IMutex *ServerCleanupThread::a_mutex=NULL;
bool ServerCleanupThread::update_stats_interruptible=false;
volatile bool ServerCleanupThread::do_quit=false;
bool ServerCleanupThread::update_stats_disabled = false;
std::map<int, size_t> ServerCleanupThread::locked_images;
IMutex* ServerCleanupThread::cleanup_lock_mutex = NULL;
bool ServerCleanupThread::allow_clientlist_deletion = true;
void cleanupLastActs();
const unsigned int min_cleanup_interval=12*60*60;
extern IClouddriveFactory* clouddrive_fak;
void ServerCleanupThread::initMutex(void)
{
mutex=Server->createMutex();
cond=Server->createCondition();
a_mutex=Server->createMutex();
cleanup_lock_mutex = Server->createMutex();
}
void ServerCleanupThread::destroyMutex(void)
{
Server->destroy(mutex);
Server->destroy(cond);
Server->destroy(a_mutex);
Server->destroy(cleanup_lock_mutex);
}
ServerCleanupThread::ServerCleanupThread(CleanupAction cleanup_action)
: cleanup_action(cleanup_action)
{
logid = ServerLogger::getLogId(LOG_CATEGORY_CLEANUP);
}
ServerCleanupThread::~ServerCleanupThread(void)
{
}
void ServerCleanupThread::operator()(void)
{
db=Server->getDatabase(Server->getThreadID(), URBACKUPDB_SERVER);
if(cleanup_action.action!=ECleanupAction_None)
{
cleanupdao.reset(new ServerCleanupDao(db));
backupdao.reset(new ServerBackupDao(db));
filesdao.reset(new ServerFilesDao(Server->getDatabase(Server->getThreadID(), URBACKUPDB_SERVER_FILES)));
fileindex.reset(create_lmdb_files_index());
switch(cleanup_action.action)
{
case ECleanupAction_DeleteFilebackup:
{
bool b = deleteFileBackup(cleanup_action.backupfolder, cleanup_action.clientid, cleanup_action.backupid, cleanup_action.force_remove,
true);
if (cleanup_action.result != NULL)
{
*cleanup_action.result = b;
}
} break;
case ECleanupAction_DeleteImagebackup:
{
ServerSettings settings(db);
bool b =removeImage(cleanup_action.backupid, &settings, true, cleanup_action.force_remove, true, true);
if (cleanup_action.result != NULL)
{
*cleanup_action.result = b;
}
} break;
case ECleanupAction_FreeMinspace:
{
ScopedProcess nightly_cleanup(std::string(), sa_emergency_cleanup, std::string(), logid, false, LOG_CATEGORY_CLEANUP);
deletePendingClients();
bool b = do_cleanup(cleanup_action.minspace, cleanup_action.cleanup_other);
if(cleanup_action.result!=NULL)
{
*(cleanup_action.result)=b;
}
} break;
case ECleanupAction_RemoveUnknown:
do_remove_unknown();
break;
}
cleanupdao.reset();
backupdao.reset();
filesdao.reset();
fileindex.reset();
Server->destroyDatabases(Server->getThreadID());
delete this;
return;
}
int64 last_cleanup=0;
Server->waitForStartupComplete();
#ifndef _DEBUG
{
IScopedLock lock(mutex);
cond->wait(&lock, 1000);
if(do_quit)
{
delete this;
return;
}
}
#endif
if (FileExists("urbackup/migrate_storage_to"))
{
std::string migrate_storage_to = trim(getFile("urbackup/migrate_storage_to"));
if (!migrate_storage_to.empty())
{
while (true)
{
ScopedActiveThread sat;
IScopedLock lock(a_mutex);
bool ignore_copy_errors = FileExists("urbackup/migrate_storage_to.ignore_copy_errors");
if (copy_storage(migrate_storage_to, ignore_copy_errors) == 0)
{
writestring("done", "urbackup/migrate_storage_to.done");
}
Server->wait(10*60000);
}
}
}
else
{
ServerSettings settings(db);
if (os_directory_exists(settings.getSettings()->backupfolder + os_file_sep() + "inode_db"))
{
Server->deleteFile(settings.getSettings()->backupfolder + os_file_sep() + "inode_db" + os_file_sep() + "inode_db.lmdb");
Server->deleteFile(settings.getSettings()->backupfolder + os_file_sep() + "inode_db" + os_file_sep() + "inode_db.lmdb-lock");
os_remove_dir(settings.getSettings()->backupfolder + os_file_sep() + "inode_db");
}
}
{
ScopedActiveThread sat;
ISettingsReader *settings=Server->createDBSettingsReader(db, "settings_db.settings",
"SELECT value FROM settings_db.settings WHERE key=? AND clientid=0");
if( settings->getValue("autoshutdown", "false")=="true" )
{
IScopedLock lock(a_mutex);
cleanupdao.reset(new ServerCleanupDao(db));
backupdao.reset(new ServerBackupDao(db));
filesdao.reset(new ServerFilesDao(Server->getDatabase(Server->getThreadID(), URBACKUPDB_SERVER_FILES)));
fileindex.reset(create_lmdb_files_index());
{
logid = ServerLogger::getLogId(LOG_CATEGORY_CLEANUP);
ScopedProcess nightly_cleanup(std::string(), sa_nightly_cleanup, std::string(), logid, false, LOG_CATEGORY_CLEANUP);
deletePendingClients();
do_cleanup();
}
cleanupdao.reset();
backupdao.reset();
filesdao.reset();
fileindex.reset();
{
ServerSettings settings(db);
if (settings.getSettings()->backupfolder != Server->getServerWorkingDir())
{
setClientlistDeletionAllowed(false);
if (backup_database() && backup_clientlists() && backup_ident())
{
ren_files_backupfolder();
}
setClientlistDeletionAllowed(true);
}
else
{
Server->Log("Not running database backup because backupfolder=working dir", LL_ERROR);
}
}
}
if( settings->getValue("download_client", "true")=="true" )
{
IScopedLock lock(a_mutex);
ServerUpdate upd;
upd.update_client();
}
if( settings->getValue("show_server_updates", "true")=="true" )
{
IScopedLock lock(a_mutex);
ServerUpdate upd;
upd.update_server_version_info();
}
if (settings->getValue("update_dataplan_db", "true") == "true")
{
IScopedLock lock(a_mutex);
ServerUpdate upd;
upd.update_dataplan_db();
}
Server->destroy(settings);
}
while(true)
{
Server->clearDatabases(Server->getThreadID());
{
IScopedLock lock(mutex);
if(!update_stats)
{
cond->wait(&lock, 3600000);
}
if(do_quit)
{
delete this;
return;
}
if(update_stats)
{
if (!update_stats_disabled)
{
lock.relock(NULL);
Server->Log("Updating statistics...", LL_INFO);
ScopedActiveThread sat;
{
IScopedLock lock(a_mutex);
ServerUpdateStats sus(false, update_stats_interruptible);
sus();
}
Server->Log("Done updating statistics.", LL_INFO);
}
update_stats = false;
}
}
db=Server->getDatabase(Server->getThreadID(), URBACKUPDB_SERVER);
db_results res=db->Read("SELECT strftime('%H','now', 'localtime') AS time");
if(res.empty())
Server->Log("Reading time failed!", LL_ERROR);
else
{
int chour=watoi(res[0]["time"]);
ServerSettings settings(db);
std::vector<STimeSpan> tw=settings.getCleanupWindow();
if( ( (!tw.empty() && ServerSettings::isInTimeSpan(tw)) || ( tw.empty() && (chour==3 || chour==4) ) )
&& Server->getTimeSeconds()-last_cleanup>min_cleanup_interval
&& os_directory_exists(settings.getSettings()->backupfolder) )
{
IScopedLock lock(a_mutex);
ISettingsReader *settings=Server->createDBSettingsReader(db, "settings_db.settings",
"SELECT value FROM settings_db.settings WHERE key=? AND clientid=0");
ScopedActiveThread sat;
if( settings->getValue("download_client", "true")=="true" )
{
ServerUpdate upd;
upd.update_client();
}
if( settings->getValue("show_server_updates", "true")=="true" )
{
IScopedLock lock(a_mutex);
ServerUpdate upd;
upd.update_server_version_info();
}
if (settings->getValue("update_dataplan_db", "true") == "true")
{
IScopedLock lock(a_mutex);
ServerUpdate upd;
upd.update_dataplan_db();
}
cleanupdao.reset(new ServerCleanupDao(db));
backupdao.reset(new ServerBackupDao(db));
filesdao.reset(new ServerFilesDao(Server->getDatabase(Server->getThreadID(), URBACKUPDB_SERVER_FILES)));
fileindex.reset(create_lmdb_files_index());
{
logid = ServerLogger::getLogId(LOG_CATEGORY_CLEANUP);
ScopedProcess nightly_cleanup(std::string(), sa_nightly_cleanup, std::string(), logid, false, LOG_CATEGORY_CLEANUP);
deletePendingClients();
do_cleanup();
enforce_quotas();
}
cleanupdao.reset();
backupdao.reset();
filesdao.reset();
fileindex.reset();
{
ServerSettings settings(db);
if (settings.getSettings()->backupfolder != Server->getServerWorkingDir())
{
setClientlistDeletionAllowed(false);
if (backup_database() && backup_clientlists() && backup_ident())
{
ren_files_backupfolder();
}
setClientlistDeletionAllowed(true);
}
else
{
Server->Log("Not running database backup because backupfolder=working dir", LL_ERROR);
}
}
Server->destroy(settings);
Server->clearDatabases(Server->getThreadID());
last_cleanup=Server->getTimeSeconds();
}
}
}
delete this;
}
void ServerCleanupThread::updateStats(bool interruptible)
{
IScopedLock lock(mutex);
update_stats=true;
update_stats_interruptible=interruptible;
cond->notify_all();
}
bool ServerCleanupThread::isUpdateingStats()
{
IScopedLock lock(mutex);
return update_stats;
}
void ServerCleanupThread::disableUpdateStats()
{
IScopedLock lock(mutex);
update_stats_disabled = true;
}
void ServerCleanupThread::enableUpdateStats()
{
IScopedLock lock(mutex);
update_stats_disabled = false;
}
void ServerCleanupThread::do_cleanup(void)
{
db_results cache_res;
if(db->getEngineName()=="sqlite")
{
cache_res=db->Read("PRAGMA cache_size");
ServerSettings server_settings(db);
db->Write("PRAGMA cache_size = -"+convert(server_settings.getSettings()->update_stats_cachesize));
}
removeerr.clear();
cleanup_images();
cleanup_files();
{
ServerSettings server_settings(db);
int64 total_space=os_total_space(server_settings.getSettings()->backupfolder);
if(total_space!=-1)
{
int64 amount=cleanup_amount(server_settings.getSettings()->global_soft_fs_quota, db);
if(amount<total_space)
{
ServerLogger::Log(logid, "Space to free: "+PrettyPrintBytes(total_space-amount), LL_INFO);
cleanup_images(total_space-amount);
cleanup_files(total_space-amount);
}
}
else
{
ServerLogger::Log(logid, "Error getting total used space of backup folder", LL_ERROR);
}
}
ServerLogger::Log(logid, "Updating statistics...", LL_INFO);
ServerUpdateStats sus;
sus();
ServerLogger::Log(logid, "Done updating statistics.", LL_INFO);
cleanup_other();
if(!cache_res.empty())
{
db->Write("PRAGMA cache_size = "+cache_res[0]["cache_size"]);
db->Write("PRAGMA shrink_memory");
}
}
bool ServerCleanupThread::do_cleanup(int64 minspace, bool do_cleanup_other)
{
ServerStatus::incrementServerNospcStalled(1);
IScopedLock lock(a_mutex);
db_results cache_res;
if(db->getEngineName()=="sqlite")
{
cache_res=db->Read("PRAGMA cache_size");
ServerSettings server_settings(db);
db->Write("PRAGMA cache_size = -"+convert(server_settings.getSettings()->update_stats_cachesize));
}
if(minspace>0)
{
ServerLogger::Log(logid, "Space to free: "+PrettyPrintBytes(minspace), LL_INFO);
}
removeerr.clear();
cleanup_images(minspace);
cleanup_files(minspace);
cleanup_images();
cleanup_files();
if(do_cleanup_other)
{
cleanup_other();
}
ServerLogger::Log(logid, "Updating statistics...", LL_INFO);
ServerUpdateStats sus;
sus();
ServerLogger::Log(logid, "Done updating statistics.", LL_INFO);
db->destroyAllQueries();
ServerSettings settings(db);
int r=hasEnoughFreeSpace(minspace, &settings);
ServerStatus::incrementServerNospcStalled(-1);
bool success=(r==1);
if(!success)
{
ServerStatus::setServerNospcFatal(true);
}
if(!cache_res.empty())
{
db->Write("PRAGMA cache_size = "+cache_res[0]["cache_size"]);
db->Write("PRAGMA shrink_memory");
}
FileIndex::flush();
return success;
}
void ServerCleanupThread::do_remove_unknown(void)
{
ServerSettings settings(db);
replay_directory_link_journal();
ServerLinkDao link_dao(Server->getDatabase(Server->getThreadID(), URBACKUPDB_SERVER_LINKS));
std::string backupfolder=settings.getSettings()->backupfolder;
std::vector<ServerCleanupDao::SClientInfo> res_clients=cleanupdao->getClients();
for(size_t i=0;i<res_clients.size();++i)
{
int clientid=res_clients[i].id;
const std::string& clientname=res_clients[i].name;
Server->Log("Removing unknown for client \""+clientname+"\"");
std::vector<ServerCleanupDao::SFileBackupInfo> res_file_backups=cleanupdao->getFileBackupsOfClient(clientid);
for(size_t j=0;j<res_file_backups.size();++j)
{
std::string backup_path=backupfolder+os_file_sep()+clientname+os_file_sep()+res_file_backups[j].path;
int backupid=res_file_backups[j].id;
if(!os_directory_exists(backup_path))
{
Server->Log("Path for file backup [id="+convert(res_file_backups[j].id)+" path="+res_file_backups[j].path+" clientname="+clientname+"] does not exist. Deleting it from the database.", LL_WARNING);
removeFileBackupSql(backupid);
}
}
std::vector<ServerCleanupDao::SImageBackupInfo> res_image_backups=cleanupdao->getImageBackupsOfClient(clientid);
for(size_t j=0;j<res_image_backups.size();++j)
{
std::string backup_path=res_image_backups[j].path;
IFile *tf=Server->openFile(os_file_prefix(backup_path), MODE_READ);
if(tf==NULL)
{
Server->Log("Image backup [id="+convert(res_image_backups[j].id)+" path="+res_image_backups[j].path+" clientname="+clientname+"] does not exist. Deleting it from the database.", LL_WARNING);
cleanupdao->removeImage(res_image_backups[j].id);
}
else
{
Server->destroy(tf);
}
}
cleanup_system_images(clientid, clientname, settings);
std::vector<SFile> files=getFiles(backupfolder+os_file_sep()+clientname, NULL);
std::vector<ServerCleanupDao::SImageBackupInfo> res_images=cleanupdao->getClientImages(clientid);
for(size_t j=0;j<files.size();++j)
{
SFile cf=files[j];
if(cf.name=="current")
continue;
if(cf.name==".directory_pool")
continue;
if(cf.isdir
|| ( !cf.isdir
&& cf.name.find(".")==std::string::npos ) )
{
if (cf.name.find("Image") == std::string::npos)
{
ServerCleanupDao::CondInt res_id = cleanupdao->findFileBackup(clientid, cf.name);
if (!res_id.exists)
{
Server->Log("File backup \"" + cf.name + "\" of client \"" + clientname + "\" not found in database. Deleting it.", LL_WARNING);
bool remove_folder = false;
if (!cf.isdir)
{
SnapshotHelper::removeFilesystem(false, clientname, cf.name);
Server->deleteFile(os_file_prefix(backupfolder
+ os_file_sep() + clientname + os_file_sep() + cf.name));
}
else if (BackupServer::isFileSnapshotsEnabled())
{
if (!SnapshotHelper::removeFilesystem(false, clientname, cf.name))
{
remove_folder = true;
}
}
else
{
remove_folder = true;
}
if (remove_folder)
{
std::string rm_dir = backupfolder + os_file_sep() + clientname + os_file_sep() + cf.name;
if (!remove_directory_link_dir(rm_dir, link_dao, clientid))
{
Server->Log("Could not delete directory \"" + rm_dir + "\"", LL_ERROR);
}
}
}
}
else
{
std::vector<SFile> image_files = getFiles(backupfolder + os_file_sep() + clientname + os_file_sep() + cf.name);
bool found_image = false;
for (size_t l = 0; l < image_files.size(); ++l)
{
std::string extension = findextension(image_files[l].name);
if (extension != "vhd" && extension != "vhdz" && extension != "raw")
continue;
found_image = true;
bool found = false;
for (size_t k = 0; k<res_images.size(); ++k)
{
if (ExtractFileName(res_images[k].path) == image_files[l].name)
{
found = true;
break;
}
}
if (!found)
{
Server->Log("Image backup \"" + cf.name + "\" of client \"" + clientname + "\" not found in database. Deleting it.", LL_WARNING);
if (extension == "raw")
{
SnapshotHelper::removeFilesystem(true, clientname, cf.name);
}
else
{
os_remove_nonempty_dir(os_file_prefix(backupfolder + os_file_sep() + clientname + os_file_sep() + cf.name));
}
}
}
if (!found_image)
{
if (!cf.isdir)
{
SnapshotHelper::removeFilesystem(true, clientname, cf.name);
Server->deleteFile(os_file_prefix(backupfolder + os_file_sep() + clientname + os_file_sep() + cf.name));
}
else
{
os_remove_nonempty_dir(os_file_prefix(backupfolder + os_file_sep() + clientname + os_file_sep() + cf.name));
if(os_directory_exists(os_file_prefix(backupfolder + os_file_sep() + clientname + os_file_sep() + cf.name))
&& BackupServer::isImageSnapshotsEnabled() )
{
SnapshotHelper::removeFilesystem(true, clientname, cf.name);
}
}
}
}
}
else
{
std::string extension=findextension(cf.name);
if(extension!="vhd" && extension!="vhdz" && extension!="raw")
continue;
bool found=false;
for(size_t k=0;k<res_images.size();++k)
{
if(ExtractFileName(res_images[k].path)==cf.name)
{
found=true;
break;
}
}
if(!found)
{
Server->Log("Image backup \""+cf.name+"\" of client \""+clientname+"\" not found in database. Deleting it.", LL_WARNING);
std::string rm_file=backupfolder+os_file_sep()+clientname+os_file_sep()+cf.name;
if(!Server->deleteFile(rm_file))
{
Server->Log("Could not delete file \""+rm_file+"\"", LL_ERROR);
}
if(!Server->deleteFile(rm_file+".mbr"))
{
Server->Log("Could not delete file \""+rm_file+".mbr\"", LL_ERROR);
}
if(!Server->deleteFile(rm_file+".hash"))
{
Server->Log("Could not delete file \""+rm_file+".hash\"", LL_ERROR);
}
Server->deleteFile(rm_file+".bitmap");
Server->deleteFile(rm_file+".cbitmap");
Server->deleteFile(rm_file + ".sync");
}
}
}
check_symlinks(res_clients[i], backupfolder, false);
}
Server->Log("Removing dangling file entries...", LL_INFO);
IQuery* q_backup_ids = db->Prepare("SELECT id FROM backups", false);
IDatabaseCursor* cur = q_backup_ids->Cursor();
db_single_result res;
IDatabase* files_db = Server->getDatabase(Server->getThreadID(), URBACKUPDB_SERVER_FILES);
files_db->Write("CREATE TEMPORARY TABLE backups (id INTEGER PRIMARY KEY)");
IQuery* q_insert = files_db->Prepare("INSERT INTO backups (id) VALUES (?)", false);
bool ok = true;
while (cur->next(res))
{
q_insert->Bind(res["id"]);
ok &= q_insert->Write();
q_insert->Reset();
}
db->destroyQuery(q_backup_ids);
files_db->destroyQuery(q_insert);
if (ok)
{
filesdao->removeDanglingFiles();
Server->Log("Deleted " + convert(files_db->getLastChanges()) + " file entries", LL_INFO);
}
files_db->Write("DROP TABLE backups");
FileIndex::flush();
}
int ServerCleanupThread::hasEnoughFreeSpace(int64 minspace, ServerSettings *settings)
{
if(minspace!=-1)
{
std::string path=settings->getSettings()->backupfolder;
int64 available_space=os_free_space(os_file_prefix(path));
if(available_space==-1)
{
ServerLogger::Log(logid, "Error getting free space for path \""+path+"\"", LL_ERROR);
return -1;
}
else
{
if(available_space>minspace)
{
ServerLogger::Log(logid, "Enough free space now.", LL_DEBUG);
return 1;
}
}
ServerLogger::Log(logid, "Free space: "+PrettyPrintBytes(available_space), LL_DEBUG);
}
return 0;
}
bool ServerCleanupThread::deleteAndTruncateFile(logid_t logid, std::string path)
{
if(!Server->deleteFile(os_file_prefix(path)))
{
std::string errmsg = os_last_error_str();
if (os_get_file_type(os_file_prefix(path)) & EFileType_File)
{
ServerLogger::Log(logid, "Deleting " + path + " failed. " + errmsg + " . Truncating it instead.", LL_WARNING);
os_file_truncate(os_file_prefix(path), 0);
}
else
{
ServerLogger::Log(logid, "Deleting " + path + " failed. " + errmsg, LL_WARNING);
}
return false;
}
return true;
}
bool ServerCleanupThread::deleteImage(logid_t logid, std::string clientname, std::string path)
{
std::string image_extension = findextension(path);
if (image_extension != "raw"
|| !BackupServer::isImageSnapshotsEnabled())
{
bool b = true;
if (!deleteAndTruncateFile(logid, path))
{
b = false;
}
if (!deleteAndTruncateFile(logid, path + ".hash"))
{
b = false;
}
if (!deleteAndTruncateFile(logid, path + ".mbr"))
{
if (os_get_file_type(os_file_prefix(path + ".mbr")) & EFileType_File)
{
b = false;
}
}
deleteAndTruncateFile(logid, path + ".cbitmap");
deleteAndTruncateFile(logid, path + ".sync");
deleteAndTruncateFile(logid, path + ".bitmap");
if (b && ExtractFileName(ExtractFilePath(path)) != clientname)
{
b = os_remove_dir(os_file_prefix(ExtractFilePath(path)));
}
return b;
}
else
{
bool b = SnapshotHelper::removeFilesystem(true, clientname, ExtractFileName(ExtractFilePath(path)));
if (b
&& BackupServer::getSnapshotMethod(true) == BackupServer::ESnapshotMethod_Zfs)
{
Server->deleteFile(ExtractFilePath(path));
}
return b;
}
}
int ServerCleanupThread::max_removable_incr_images(ServerSettings& settings, int backupid,
int del_in_stack)
{
int incr_image_num=cleanupdao->getIncrNumImagesForBackup(backupid);
incr_image_num -= del_in_stack;
if (incr_image_num < 0)
incr_image_num = 0;
int min_incr_image_num = static_cast<int>(settings.getSettings()->min_image_incr);
int max_allowed_del_refs = 0;
if(min_incr_image_num<incr_image_num)
{
max_allowed_del_refs = incr_image_num-min_incr_image_num;
}
return max_allowed_del_refs;
}
int ServerCleanupThread::max_removable_incr_file_backups(ServerSettings& settings, int backupid, int del_in_stack)
{
int incr_file_num = cleanupdao->getIncrNumFileBackupsForBackup(backupid);
incr_file_num -= del_in_stack;
if (incr_file_num < 0)
incr_file_num = 0;
int min_incr_file_num = static_cast<int>(settings.getSettings()->min_file_incr);
int max_allowed_del_refs = 0;
if (min_incr_file_num < incr_file_num)
{
max_allowed_del_refs = incr_file_num - min_incr_file_num;
}
return max_allowed_del_refs;
}
bool ServerCleanupThread::cleanup_one_imagebackup_client(int clientid, int64 minspace, int& imagebid)
{
ServerSettings settings(db, clientid);
std::string val_info = "max_image_full";
int max_image_full=settings.getSettings()->max_image_full;
if(minspace!=-1)
{
val_info = "min_image_full";
max_image_full=settings.getSettings()->min_image_full;
}
std::vector<int> notit;
int backupid;
int full_image_num=(int)getImagesFullNum(clientid, backupid, notit);
ServerLogger::Log(logid, "Client with id="+convert(clientid)+" has "+convert(full_image_num)+" full image backups "+ val_info +"="+convert(max_image_full), LL_DEBUG);
while(full_image_num>max_image_full
&& full_image_num>0)
{
ServerCleanupDao::SImageBackupInfo res_info=cleanupdao->getImageBackupInfo(backupid);
ServerCleanupDao::CondString clientname=cleanupdao->getClientName(clientid);
if(clientname.exists && res_info.exists)
{
ServerLogger::Log(logid, "Deleting full image backup ( id="+convert(res_info.id)+", backuptime="+res_info.backuptime+", path="+res_info.path+", letter="+res_info.letter+" ) from client \""+clientname.value+"\" ( id="+convert(clientid)+" ) ...", LL_INFO);
}
if (isImageLockedFromCleanup(backupid))
{
ServerLogger::Log(logid, "Backup image is locked from cleanup");
notit.push_back(backupid);
}
else if(findUncompleteImageRef(cleanupdao.get(), backupid) )
{
ServerLogger::Log(logid, "Backup image has dependent image which is not complete");
notit.push_back(backupid);
}
else if (findLockedImageRef(cleanupdao.get(), backupid))
{
ServerLogger::Log(logid, "Backup image has dependent image which is currently locked from cleanup");
notit.push_back(backupid);
}
else if (findArchivedImageRef(cleanupdao.get(), backupid))
{
ServerLogger::Log(logid, "Backup image has dependent image which is currently archived");
notit.push_back(backupid);
}
else
{
if (!removeImage(backupid, &settings, true, false, true, true))
{
notit.push_back(backupid);
}
else
{
imagebid = backupid;
return true;
}
}
full_image_num=(int)getImagesFullNum(clientid, backupid, notit);
}
notit.clear();
val_info = "max_image_incr";
int max_image_incr=settings.getSettings()->max_image_incr;
if (minspace != -1)
{
val_info = "min_image_incr";
max_image_incr = settings.getSettings()->min_image_incr;
}
int incr_image_num=(int)getImagesIncrNum(clientid, backupid, notit);
ServerLogger::Log(logid, "Client with id="+convert(clientid)+" has "+convert(incr_image_num)+" incremental image backups "+val_info +"="+convert(max_image_incr), LL_DEBUG);
while(incr_image_num>max_image_incr
&& incr_image_num>0)
{
ServerCleanupDao::SImageBackupInfo res_info=cleanupdao->getImageBackupInfo(backupid);
ServerCleanupDao::CondString clientname=cleanupdao->getClientName(clientid);
if(clientname.exists && res_info.exists)
{
ServerLogger::Log(logid, "Deleting incremental image backup ( id="+convert(res_info.id)+", backuptime="+res_info.backuptime+", path="+res_info.path+", letter="+res_info.letter+" ) from client \""+clientname.value+"\" ( id="+convert(clientid)+" ) ...", LL_INFO);
}
if (isImageLockedFromCleanup(backupid))
{
ServerLogger::Log(logid, "Backup image is locked from cleanup");
notit.push_back(backupid);
}
else if (findUncompleteImageRef(cleanupdao.get(), backupid))
{
ServerLogger::Log(logid, "Backup image has dependent image which is not complete");
notit.push_back(backupid);
}
else if (findLockedImageRef(cleanupdao.get(), backupid))
{
ServerLogger::Log(logid, "Backup image has dependent image which is currently locked from cleanup");
notit.push_back(backupid);
}
else if (findArchivedImageRef(cleanupdao.get(), backupid))
{
ServerLogger::Log(logid, "Backup image has dependent image which is currently archived");
notit.push_back(backupid);
}
else
{
if(!removeImage(backupid, &settings, true, false, true, true, 1))
{
notit.push_back(backupid);
}
else
{
imagebid = backupid;
return true;
}
}
incr_image_num=(int)getImagesIncrNum(clientid, backupid, notit);
}
return false;
}
void ServerCleanupThread::cleanup_images(int64 minspace)
{
std::vector<ServerCleanupDao::SIncompleteImages> incomplete_images=cleanupdao->getIncompleteImages();
for(size_t i=0;i<incomplete_images.size();++i)
{
if (!isImageLockedFromCleanup(incomplete_images[i].id)
&& cleanupdao->getIncompleteImage(incomplete_images[i].id).exists )
{
ServerLogger::Log(logid, "Deleting incomplete image file \"" + incomplete_images[i].path + "\"...", LL_INFO);
if (!deleteImage(logid, incomplete_images[i].clientname, incomplete_images[i].path))
{
ServerLogger::Log(logid, "Deleting incomplete image \"" + incomplete_images[i].path + "\" failed.", LL_WARNING);
}
cleanupdao->removeImage(incomplete_images[i].id);
}
}
std::vector<ServerCleanupDao::SIncompleteImages> delete_pending_images = cleanupdao->getDeletePendingImages();
if (!delete_pending_images.empty())
{
ServerSettings settings(db);
for (size_t i = 0; i < delete_pending_images.size(); ++i)
{
if ( !isImageLockedFromCleanup(delete_pending_images[i].id)
&& !findLockedImageRef(cleanupdao.get(), delete_pending_images[i].id)
&& !findUncompleteImageRef(cleanupdao.get(), delete_pending_images[i].id)
&& !findArchivedImageRef(cleanupdao.get(), delete_pending_images[i].id)
&& cleanupdao->getImageArchived(delete_pending_images[i].id).value==0 )
{
ServerLogger::Log(logid, "Deleting image file \"" + delete_pending_images[i].path + "\" because it was manually set to be deleted...", LL_INFO);
if (!removeImage(delete_pending_images[i].id, &settings, true, false, true, true))
{
ServerLogger::Log(logid, "Deleting image \"" + delete_pending_images[i].path + "\" (manually set to be deleted) failed.", LL_WARNING);
}
}
}
}
ServerSettings settings(db);
cleanup_all_system_images(settings);
int r=hasEnoughFreeSpace(minspace, &settings);
if( r==-1 || r==1)
return;
bool deleted_something = true;
while (deleted_something)
{
deleted_something = false;
std::vector<int> res = cleanupdao->getClientsSortImagebackups();
for (size_t i = 0; i<res.size(); ++i)
{
int clientid = res[i];
int imagebid;
if (cleanup_one_imagebackup_client(clientid, minspace, imagebid))
{
int r = hasEnoughFreeSpace(minspace, &settings);
if (r == -1 || r == 1)
return;
deleted_something = true;
}
}
int r = hasEnoughFreeSpace(minspace, &settings);
if (r == -1 || r == 1)
return;
}
}
bool ServerCleanupThread::removeImage(int backupid, ServerSettings* settings,
bool update_stat, bool force_remove, bool remove_associated, bool remove_references,
int del_incr_in_stack)
{
int64 deleted_size_bytes = 0;
if(update_stat)
{
deleted_size_bytes=getImageSize(backupid);
}
if(update_stat)
{
deleted_size_bytes=getImageSize(backupid);
}
bool ret = true;
if(remove_references)
{
assert(settings!=NULL);
std::vector<ServerCleanupDao::SImageRef> refs=cleanupdao->getImageRefs(backupid);
for(size_t i=0;i<refs.size();++i)
{
if(!force_remove &&
max_removable_incr_images(*settings, refs[i].id, del_incr_in_stack)<=0)
{
ServerLogger::Log(logid, "Cannot delete image because incremental image backups referencing this image are not allowed to be deleted", LL_INFO);
return false;
}
bool b=removeImage(refs[i].id, settings, true, force_remove,
remove_associated, remove_references,
del_incr_in_stack+1);
if(!b)
{
ServerLogger::Log(logid, "Cannot delete image because incremental image backup referencing this image cannot be deleted", LL_ERROR);
return false;
}
}
}
else if(!remove_references && !force_remove)
{
std::vector<ServerCleanupDao::SImageRef> refs=cleanupdao->getImageRefs(backupid);
if(!refs.empty())
{
ServerLogger::Log(logid, "Cannot delete image because incremental image backups referencing this image exist", LL_INFO);
return false;
}
}
if(remove_associated)
{
std::vector<int> assoc=cleanupdao->getAssocImageBackups(backupid);
for(size_t i=0;i<assoc.size();++i)
{
int64 is=getImageSize(assoc[i]);
if(is>0) deleted_size_bytes+=is;
removeImage(assoc[i], settings, false, force_remove, remove_associated,
remove_references, del_incr_in_stack);
}
}
ServerStatus::updateActive();
ServerCleanupDao::CondString res=cleanupdao->getImagePath(backupid);
ServerCleanupDao::CondString res_clientname = cleanupdao->getImageClientname(backupid);
if(res.exists && res_clientname.exists)
{
ServerLogger::Log(logid, "Deleting image backup ( id="+convert(backupid)+", path="+res.value+" ) ...", LL_INFO);
_i64 stat_id;
if(update_stat)
{
cleanupdao->addToImageStats(deleted_size_bytes, backupid);
stat_id=db->getLastInsertID();
}
if( deleteImage(logid, res_clientname.value, res.value) || force_remove )
{
db->BeginWriteTransaction();
cleanupdao->removeImage(backupid);
cleanupdao->removeImageSize(backupid);
db->EndTransaction();
}
else
{
ServerLogger::Log(logid, "Deleting image backup failed.", LL_INFO);
ret=false;
}
if(update_stat)
{
cleanupdao->updateDelImageStats(stat_id);
}
}
else
{
ret=false;
}
ServerStatus::updateActive();
return ret;
}
bool ServerCleanupThread::findUncompleteImageRef(ServerCleanupDao* cleanupdao, int backupid)
{
std::vector<ServerCleanupDao::SImageRef> refs=cleanupdao->getImageRefs(backupid);
for(size_t i=0;i<refs.size();++i)
{
if( refs[i].complete!=1
|| findUncompleteImageRef(cleanupdao, refs[i].id) )
return true;
}
return false;
}
bool ServerCleanupThread::findIncompleteFileBackupRef(ServerCleanupDao* cleanupdao, int backupid)
{
std::vector<ServerCleanupDao::SFileBackupRef> refs = cleanupdao->getFileBackupRefs(backupid);
for (size_t i = 0; i < refs.size(); ++i)
{
if (refs[i].complete != 1
|| findIncompleteFileBackupRef(cleanupdao, refs[i].id))
return true;
}
return false;
}
bool ServerCleanupThread::findLockedImageRef(ServerCleanupDao* cleanupdao, int backupid)
{
std::vector<ServerCleanupDao::SImageRef> refs = cleanupdao->getImageRefs(backupid);
for (size_t i = 0; i<refs.size(); ++i)
{
if (ServerCleanupThread::isImageLockedFromCleanup(refs[i].id)
|| findLockedImageRef(cleanupdao, refs[i].id))
return true;
}
return false;
}
bool ServerCleanupThread::findArchivedImageRef(ServerCleanupDao* cleanupdao, int backupid)
{
std::vector<int> assoc = cleanupdao->getAssocImageBackups(backupid);
for (size_t i = 0; i < assoc.size(); ++i)
{
if (cleanupdao->getImageArchived(assoc[i]).value == 1)
{
return true;
}
}
std::vector<ServerCleanupDao::SImageRef> refs = cleanupdao->getImageRefs(backupid);
for (size_t i = 0; i<refs.size(); ++i)
{
if (refs[i].archived==1
|| findArchivedImageRef(cleanupdao, refs[i].id))
return true;
}
return false;
}
bool ServerCleanupThread::findArchivedFileBackupRef(ServerCleanupDao* cleanupdao, int backupid)
{
std::vector<ServerCleanupDao::SFileBackupRef> refs = cleanupdao->getFileBackupRefs(backupid);
for (size_t i = 0; i < refs.size(); ++i)
{
if (refs[i].archived == 1
|| findArchivedFileBackupRef(cleanupdao, refs[i].id))
return true;
}
return false;
}
size_t ServerCleanupThread::getImagesFullNum(int clientid, int &backupid_top, const std::vector<int> &notit)
{
std::vector<ServerCleanupDao::SImageLetter> res=cleanupdao->getFullNumImages(clientid);
std::map<std::string, std::vector<int> > images_ids;
for(size_t i=0;i<res.size();++i)
{
std::string letter=res[i].letter;
int cid=res[i].id;
if(std::find(notit.begin(), notit.end(), cid)==notit.end())
{
images_ids[letter].push_back(cid);
}
}
size_t max_nimages=0;
for(std::map<std::string, std::vector<int> >::iterator iter=images_ids.begin();iter!=images_ids.end();++iter)
{
if(iter->second.size()>max_nimages)
{
backupid_top=iter->second[0];
max_nimages=iter->second.size();
}
}
return max_nimages;
}
size_t ServerCleanupThread::getImagesIncrNum(int clientid, int &backupid_top, const std::vector<int> &notit)
{
std::vector<ServerCleanupDao::SImageLetter> res=cleanupdao->getIncrNumImages(clientid);
std::map<std::string, std::vector<int> > images_ids;
for(size_t i=0;i<res.size();++i)
{
std::string letter=res[i].letter;
int cid=res[i].id;
if(std::find(notit.begin(), notit.end(), cid)==notit.end())
{
images_ids[letter].push_back(cid);
}
}
size_t max_nimages=0;
for(std::map<std::string, std::vector<int> >::iterator iter=images_ids.begin();iter!=images_ids.end();++iter)
{
if(iter->second.size()>max_nimages)
{
backupid_top=iter->second[0];
max_nimages=iter->second.size();
}
}
return max_nimages;
}
bool ServerCleanupThread::cleanup_one_filebackup_client(int clientid, int64 minspace, int& filebid)
{
ServerSettings settings(db, clientid);
int max_file_full=settings.getSettings()->max_file_full;
int max_file_incr=settings.getSettings()->max_file_incr;
std::string full_val_info = "max_file_full";
std::string incr_val_info = "max_file_incr";
if(minspace!=-1)
{
max_file_full=settings.getSettings()->min_file_full;
max_file_incr=settings.getSettings()->min_file_incr;
full_val_info = "min_file_full";
incr_val_info = "min_file_incr";
}
int backupid;
int full_file_num=(int)getFilesFullNum(clientid, backupid);
ServerLogger::Log(logid, "Client with id="+convert(clientid)+" has "+convert(full_file_num)+" full file backups "+ full_val_info +"="+convert(max_file_full), LL_DEBUG);
while(full_file_num>max_file_full
&& full_file_num>0
&& !(full_file_num==1
&& settings.getSettings()->max_file_incr>0
&& settings.getUpdateFreqFileIncr()>=0
&& getFilesIncrNum(clientid, backupid)==0) )
{
ServerCleanupDao::SFileBackupInfo res_info=cleanupdao->getFileBackupInfo(backupid);
ServerCleanupDao::CondString clientname=cleanupdao->getClientName(clientid);
if(clientname.exists && res_info.exists)
{
ServerLogger::Log(logid, "Deleting full file backup ( id="+convert(res_info.id)+", backuptime="+res_info.backuptime+", path="+res_info.path+" ) from client \""+clientname.value+"\" ( id="+convert(clientid)+" ) ...", LL_INFO);
}
bool b=deleteFileBackup(settings.getSettings()->backupfolder, clientid, backupid, false, false);
filebid=backupid;
ServerLogger::Log(logid, "Done.", LL_INFO);
if(b)
{
return true;
}
full_file_num=(int)getFilesFullNum(clientid, backupid);
}
int incr_file_num=(int)getFilesIncrNum(clientid, backupid);
ServerLogger::Log(logid, "Client with id="+convert(clientid)+" has "+convert(incr_file_num)+" incremental file backups "+ incr_val_info +"="+convert(max_file_incr), LL_DEBUG);
while(incr_file_num>max_file_incr
&& incr_file_num>0)
{
ServerCleanupDao::SFileBackupInfo res_info=cleanupdao->getFileBackupInfo(backupid);
ServerCleanupDao::CondString clientname=cleanupdao->getClientName(clientid);
if(clientname.exists && res_info.exists)
{
ServerLogger::Log(logid, "Deleting incremental file backup ( id="+convert(res_info.id)+", backuptime="+res_info.backuptime+", path="+res_info.path+" ) from client \""+clientname.value+"\" ( id="+convert(clientid)+" ) ...", LL_INFO);
}
bool b=deleteFileBackup(settings.getSettings()->backupfolder, clientid, backupid, false, false);
filebid=backupid;
ServerLogger::Log(logid, "Done.", LL_INFO);
if(b)
{
return true;
}
incr_file_num=(int)getFilesIncrNum(clientid, backupid);
}
return false;
}
void ServerCleanupThread::cleanup_files(int64 minspace)
{
ServerSettings settings(db);
delete_incomplete_file_backups();
delete_pending_file_backups();
bool deleted_something=true;
while(deleted_something)
{
deleted_something=false;
{
int r=hasEnoughFreeSpace(minspace, &settings);
if( r==-1 || r==1 )
return;
}
std::vector<int> res=cleanupdao->getClientsSortFilebackups();
for(size_t i=0;i<res.size();++i)
{
int clientid=res[i];
int filebid;
if(cleanup_one_filebackup_client(clientid, minspace, filebid))
{
ServerSettings settings(db);
int r=hasEnoughFreeSpace(minspace, &settings);
if( r==-1 || r==1 )
return;
deleted_something=true;
}
}
}
}
size_t ServerCleanupThread::getFilesFullNum(int clientid, int &backupid_top)
{
std::vector<int> res=cleanupdao->getFullNumFiles(clientid);
std::vector<int> no_err_res;
if(!removeerr.empty())
{
for(size_t i=0;i<res.size();++i)
{
int bid=res[i];
if(std::find(removeerr.begin(), removeerr.end(), bid)==removeerr.end())
{
no_err_res.push_back(res[i]);
}
}
}
else
{
no_err_res=res;
}
if(!no_err_res.empty())
{
backupid_top=no_err_res[0];
}
return no_err_res.size();
}
size_t ServerCleanupThread::getFilesIncrNum(int clientid, int &backupid_top)
{
std::vector<int> res=cleanupdao->getIncrNumFiles(clientid);
std::vector<int> no_err_res;
if(!removeerr.empty())
{
for(size_t i=0;i<res.size();++i)
{
int bid=res[i];
if(std::find(removeerr.begin(), removeerr.end(), bid)==removeerr.end())
{
no_err_res.push_back(res[i]);
}
}
}
else
{
no_err_res=res;
}
if(!no_err_res.empty())
{
backupid_top=no_err_res[0];
}
return no_err_res.size();
}
bool ServerCleanupThread::deleteFileBackup(const std::string &backupfolder, int clientid, int backupid,
bool force_remove, bool remove_references, int del_incr_in_stack)
{
ServerStatus::updateActive();
ServerCleanupDao::CondString cond_clientname=cleanupdao->getClientName(clientid);
if(!cond_clientname.exists)
{
ServerLogger::Log(logid, "Error getting clientname in ServerCleanupThread::deleteFileBackup", LL_ERROR);
return false;
}
std::string &clientname=cond_clientname.value;
if (cleanupdao->getFileBackupClientId(backupid).value != clientid)
{
ServerLogger::Log(logid, "Clientid does not match clientid in backup in ServerCleanupThread::deleteFileBackup", LL_ERROR);
return false;
}
ServerCleanupDao::CondString cond_backuppath=cleanupdao->getFileBackupPath(backupid);
if(!cond_backuppath.exists)
{
ServerLogger::Log(logid, "Error getting backuppath in ServerCleanupThread::deleteFileBackup", LL_ERROR);
return false;
}
std::string backuppath=cond_backuppath.value;
if(backuppath.empty())
{
ServerLogger::Log(logid, "Error backuppath empty in ServerCleanupThread::deleteFileBackup", LL_ERROR);
return false;
}
if(backupfolder.empty())
{
ServerLogger::Log(logid, "Error backupfolder empty in ServerCleanupThread::deleteFileBackup", LL_ERROR);
return false;
}
if(clientname.empty())
{
ServerLogger::Log(logid, "Error clientname empty in ServerCleanupThread::deleteFileBackup", LL_ERROR);
return false;
}
if (remove_references)
{
std::vector<ServerCleanupDao::SFileBackupRef> refs = cleanupdao->getFileBackupRefs(backupid);
ServerSettings settings(db, clientid);
for (ServerCleanupDao::SFileBackupRef& ref : refs)
{
if (!force_remove &&
max_removable_incr_file_backups(settings, ref.id, del_incr_in_stack) <= 0)
{
ServerLogger::Log(logid, "Cannot delete file backup because incremental file backups referencing this file backup are not allowed to be deleted", LL_INFO);
return false;
}
bool b = deleteFileBackup(backupfolder, clientid, ref.id, force_remove,
remove_references, del_incr_in_stack + 1);
if (!b)
{
ServerLogger::Log(logid, "Cannot delete file backup because incremental file backup backup referencing this file backup cannot be deleted", LL_ERROR);
return false;
}
}
}
else if (!remove_references && !force_remove)
{
std::vector<ServerCleanupDao::SFileBackupRef> refs = cleanupdao->getFileBackupRefs(backupid);
if (!refs.empty())
{
ServerLogger::Log(logid, "Cannot delete file backup because incremental file backups referencing this file backup exist", LL_INFO);
return false;
}
}
std::string path=backupfolder+os_file_sep()+clientname+os_file_sep()+backuppath;
if (!os_directory_exists(os_file_prefix(path))
&& os_directory_exists(os_file_prefix(path + ".startup-del")))
{
backuppath += ".startup-del";
path += ".startup-del";
}
if (os_get_file_type(os_file_prefix(path + os_file_sep() + ".hashes"
+ os_file_sep() + ".local_backup_7267f182-4341-4cda-ada1-4c0c554322d8"))
& EFileType_File)
{
return delete_local_backup(backupfolder, clientid, backupid, force_remove);
}
bool b=false;
if( BackupServer::isFileSnapshotsEnabled())
{
b=SnapshotHelper::removeFilesystem(false, clientname, backuppath);
if(!b)
{
ServerLinkDao link_dao(Server->getDatabase(Server->getThreadID(), URBACKUPDB_SERVER_LINKS));
b=remove_directory_link_dir(path, link_dao, clientid);
if(!b && SnapshotHelper::isSubvolume(false, clientname, backuppath) )
{
ServerLogger::Log(logid, "Deleting directory failed. Trying to truncate all files in subvolume to zero...", LL_ERROR);
b=truncate_files_recurisve(os_file_prefix(path));
if(b)
{
b=remove_directory_link_dir(path, link_dao, clientid);
}
}
}
else if (BackupServer::getSnapshotMethod(false) == BackupServer::ESnapshotMethod_ZfsFile)
{
Server->deleteFile(path);
}
}
else
{
ServerLinkDao link_dao(Server->getDatabase(Server->getThreadID(), URBACKUPDB_SERVER_LINKS));
b=remove_directory_link_dir(path, link_dao, clientid);
}
bool del=true;
bool err=false;
if(!b)
{
if(!os_directory_exists(os_file_prefix(path)))
{
if(os_directory_exists(os_file_prefix(backupfolder)))
{
del=true;
}
ServerLogger::Log(logid, "Warning: Directory doesn't exist: \""+path+"\"", LL_WARNING);
}
else
{
del=false;
removeerr.push_back(backupid);
ServerLogger::Log(logid, "Error removing directory \""+path+"\"", LL_ERROR);
err=true;
}
}
if(os_directory_exists(os_file_prefix(path)) )
{
del=false;
ServerLogger::Log(logid, "Directory still exists. Deleting backup failed. Path: \""+path+"\"", LL_ERROR);
err=true;
removeerr.push_back(backupid);
}
if(del || force_remove)
{
removeFileBackupSql(backupid);
}
ServerStatus::updateActive();
return !err;
}
void ServerCleanupThread::removeClient(int clientid)
{
std::string clientname=cleanupdao->getClientName(clientid).value;
ServerLogger::Log(logid, "Deleting client with id \""+convert(clientid)+"\" name \""+clientname+"\"", LL_INFO);
std::vector<ServerCleanupDao::SImageBackupInfo> res_images;
//remove image backups
do
{
res_images=cleanupdao->getClientImages(clientid);
if(!res_images.empty())
{
int backupid=res_images[0].id;
ServerLogger::Log(logid, "Removing image with id \""+convert(backupid)+"\"", LL_INFO);
removeImage(backupid, NULL, true, true, false, false);
}
}while(!res_images.empty());
//remove file backups
ServerSettings settings(db);
std::vector<int> res_filebackups;
do
{
res_filebackups=cleanupdao->getClientFileBackups(clientid);
if(!res_filebackups.empty())
{
int backupid=res_filebackups[0];
ServerLogger::Log(logid, "Removing file backup with id \""+convert(backupid)+"\"", LL_INFO);
bool b=deleteFileBackup(settings.getSettings()->backupfolder, clientid, backupid, true, true);
if(b)
ServerLogger::Log(logid, "Removing file backup with id \""+convert(backupid)+"\" successful.", LL_INFO);
else
ServerLogger::Log(logid, "Removing file backup with id \""+convert(backupid)+"\" failed.", LL_ERROR);
}
}while(!res_filebackups.empty());
//Update stats
{
ServerLogger::Log(logid, "Updating statistics...", LL_INFO);
ServerUpdateStats sus;
sus();
}
ServerLogger::Log(logid, "Deleting database table entries of client..", LL_INFO);
deleteClientSQL(db, clientid);
//delete dirs
os_remove_nonempty_dir(settings.getSettings()->backupfolder+os_file_sep()+clientname);
Server->deleteFile(settings.getSettings()->backupfolder+os_file_sep()+"clients"+os_file_sep()+clientname);
}
void ServerCleanupThread::deletePendingClients(void)
{
db_results res=db->Read("SELECT id, name FROM clients WHERE delete_pending=1");
for(size_t i=0;i<res.size();++i)
{
SStatus status=ServerStatus::getStatus(res[i]["name"]);
if(status.has_status==true)
{
ServerLogger::Log(logid, "Cannot remove client \""+res[i]["name"]+"\" ( with id "+res[i]["id"]+"): Client is online or backup is in progress", LL_WARNING);
continue;
}
removeClient(watoi(res[i]["id"]));
}
}
int64 ServerCleanupThread::getImageSize(int backupid)
{
ServerCleanupDao::CondInt64 cond_res=cleanupdao->getImageSize(backupid);
if(cond_res.exists)
{
return cond_res.value;
}
return -1;
}
namespace
{
class BackupProgress : public IDatabase::IBackupProgress
{
public:
BackupProgress(size_t status_id)
: status_id(status_id),
last_update(Server->getTimeMS()),
last_pos(0)
{
}
virtual void backupProgress(int64 pos, int64 total)
{
int64 ctime = Server->getTimeMS();
int64 passed = ctime - last_update;
if (passed > 0)
{
int64 new_bytes = pos - last_pos;
last_pos = pos;
last_update = ctime;
if (total > 0)
{
int done_pc = static_cast<int>((pos*100) / total);
ServerStatus::setProcessPcDone(std::string(), status_id, done_pc);
}
ServerStatus::setProcessDoneBytes(std::string(), status_id, pos, total);
ServerStatus::setProcessSpeed(std::string(), status_id, static_cast<double>(new_bytes) / passed);
}
}
private:
size_t status_id;
int64 last_update;
int64 last_pos;
};
bool copy_db_file(std::string src, std::string dst, IDatabase::IBackupProgress* progress, std::string& errmsg, bool close_src)
{
static std::map<std::string, IFile*> db_src_files;
IFile* src_file;
ObjectScope src_file_destroy(NULL);
if (!close_src)
{
src_file = db_src_files[src];
if (src_file == NULL)
{
src_file = Server->openFile(os_file_prefix(src), MODE_READ_DEVICE);
db_src_files[src] = src_file;
}
if (!src_file->Seek(0))
{
return false;
}
}
else
{
src_file = Server->openFile(os_file_prefix(src), MODE_READ_DEVICE);
src_file_destroy.reset(src_file);
}
std::unique_ptr<IFile> dst_file(Server->openFile(os_file_prefix(dst), MODE_WRITE));
bool copy_ok = true;
if (src_file != NULL
&& dst_file.get() != NULL)
{
std::vector<char> buf;
buf.resize(32768);
size_t cnt = 0;
int64 done_bytes = 0;
int64 total_bytes = src_file->Size();
size_t rc;
bool has_error = false;
while ((rc = (_u32)src_file->Read(buf.data(), static_cast<_u32>(buf.size()), &has_error))>0)
{
if (has_error)
{
errmsg = os_last_error_str();
break;
}
if (rc>0)
{
done_bytes += rc;
dst_file->Write(buf.data(), (_u32)rc, &has_error);
if (has_error)
{
errmsg = os_last_error_str();
break;
}
if (cnt % 32 == 0
&& progress!=NULL)
{
progress->backupProgress(done_bytes, total_bytes);
}
++cnt;
}
}
if (!has_error)
{
copy_ok = dst_file->Sync();
if (!copy_ok)
{
errmsg = os_last_error_str();
}
}
else
{
copy_ok = false;
}
}
else
{
errmsg = os_last_error_str();
copy_ok = false;
}
return copy_ok;
}
}
bool ServerCleanupThread::backup_database(void)
{
ServerSettings settings(db);
if(settings.getSettings()->backup_database)
{
std::vector<DATABASE_ID> copy_backup_ids;
copy_backup_ids.push_back(URBACKUPDB_SERVER);
copy_backup_ids.push_back(URBACKUPDB_SERVER_SETTINGS);
copy_backup_ids.push_back(URBACKUPDB_SERVER_FILES);
copy_backup_ids.push_back(URBACKUPDB_SERVER_LINKS);
copy_backup_ids.push_back(URBACKUPDB_SERVER_LINK_JOURNAL);
std::vector<std::string> copy_backup;
copy_backup.push_back("backup_server.db");
copy_backup.push_back("backup_server_settings.db");
copy_backup.push_back("backup_server_files.db");
copy_backup.push_back("backup_server_links.db");
copy_backup.push_back("backup_server_link_journal.db");
copy_backup.push_back("backup_server.db-wal");
copy_backup.push_back("backup_server_settings.db-wal");
copy_backup.push_back("backup_server_files.db-wal");
copy_backup.push_back("backup_server_links.db-wal");
copy_backup.push_back("backup_server_link_journal.db-wal");
bool integrity_ok = true;
{
logid = ServerLogger::getLogId(LOG_CATEGORY_CLEANUP);
ScopedProcess check_integrity(std::string(), sa_check_integrity, std::string(), logid, false, LOG_CATEGORY_CLEANUP);
for (size_t i = 0; i < copy_backup_ids.size(); ++i)
{
if (integrity_ok)
{
IDatabase* copy_db = Server->getDatabase(Server->getThreadID(), copy_backup_ids[i]);
ServerLogger::Log(logid, "Checking integrity of " + copy_backup[i], LL_INFO);
db_results res = copy_db->Read("PRAGMA quick_check");
integrity_ok = !res.empty() && (res[0]["quick_check"] == "ok"
|| res[0]["integrity_check"] == "ok");
if (!integrity_ok)
{
ServerLogger::Log(logid, "Integrity check failed", LL_ERROR);
}
}
}
}
if(integrity_ok)
{
std::string bfolder=settings.getSettings()->backupfolder+os_file_sep()+"urbackup";
if(!os_directory_exists(bfolder) )
{
os_create_dir(bfolder);
}
bool total_copy_ok = true;
for (size_t i = 0; i < copy_backup_ids.size(); ++i)
{
IDatabase* copy_db = Server->getDatabase(Server->getThreadID(), copy_backup_ids[i]);
logid = ServerLogger::getLogId(LOG_CATEGORY_CLEANUP);
ScopedProcess database_backup(std::string(), sa_backup_database, copy_backup[i], logid, false, LOG_CATEGORY_CLEANUP);
ServerLogger::Log(logid, "Starting database backup of " + copy_backup[i] + "...", LL_INFO);
ServerLogger::Log(logid, "Stop checkpointing of " + copy_backup[i] + "...", LL_INFO);
WalCheckpointThread::lockForBackup("urbackup" + os_file_sep() + copy_backup[i]);
ServerLogger::Log(logid, "Stop writes to " + copy_backup[i]+"...", LL_INFO);
DBScopedWriteTransaction copy_db_transaction(copy_db);
BackupProgress backup_progress(database_backup.getStatusId());
ServerLogger::Log(logid, "Copying " + copy_backup[i] + " to "+ bfolder+"...", LL_INFO);
std::string errmsg1;
bool copy_ok = copy_db_file(Server->getServerWorkingDir()+ os_file_sep() + "urbackup" + os_file_sep() + copy_backup[i],
bfolder + os_file_sep() + copy_backup[i] + "~", &backup_progress, errmsg1, false);
if (copy_ok)
{
ServerStatus::setProcessDetails(std::string(), database_backup.getStatusId(), copy_backup[i] + "-wal", -1);
BackupProgress backup_progress_wal(database_backup.getStatusId());
std::string errmsg2;
copy_ok = copy_db_file(Server->getServerWorkingDir() + os_file_sep() + "urbackup" + os_file_sep() + copy_backup[i]+"-wal",
bfolder + os_file_sep() + copy_backup[i]+"-wal~", &backup_progress_wal, errmsg2, true);
if (!copy_ok)
{
ServerLogger::Log(logid, "Backing up database failed. Copying urbackup" + os_file_sep() + copy_backup[i] + "-wal to " + bfolder + os_file_sep() + copy_backup[i] + "-wal~ failed. "+ errmsg2, LL_ERROR);
total_copy_ok = false;
}
else
{
ServerLogger::Log(logid, "Backup of " + copy_backup[i] + " done.", LL_INFO);
}
}
else
{
ServerLogger::Log(logid, "Backing up database failed. Copying urbackup" + os_file_sep() + copy_backup[i] + " to " + bfolder + os_file_sep() + copy_backup[i] + "~ failed. "+errmsg1, LL_ERROR);
total_copy_ok = false;
}
WalCheckpointThread::unlockForBackup("urbackup" + os_file_sep() + copy_backup[i]);
}
return total_copy_ok;
}
else
{
ServerLogger::Log(logid, "Database integrity check failed. Skipping Database backup.", LL_ERROR);
Server->setFailBit(IServer::FAIL_DATABASE_CORRUPTED);
ClientMain::sendMailToAdmins("Database integrity check failed", "Database integrity check failed before database backup. You should restore the UrBackup database from a backup or try to repair it.");
return false;
}
}
return true;
}
void ServerCleanupThread::doQuit(void)
{
do_quit=true;
cond->notify_all();
}
void ServerCleanupThread::lockImageFromCleanup(int backupid)
{
IScopedLock lock(cleanup_lock_mutex);
++locked_images[backupid];
}
void ServerCleanupThread::unlockImageFromCleanup(int backupid)
{
IScopedLock lock(cleanup_lock_mutex);
std::map<int, size_t>::iterator it = locked_images.find(backupid);
if (it != locked_images.end())
{
assert(it->second > 0);
--it->second;
if (it->second == 0)
{
locked_images.erase(it);
}
}
}
bool ServerCleanupThread::isImageLockedFromCleanup(int backupid)
{
IScopedLock lock(cleanup_lock_mutex);
return locked_images.find(backupid) != locked_images.end();
}
bool ServerCleanupThread::isClientlistDeletionAllowed()
{
IScopedLock lock(mutex);
return allow_clientlist_deletion;
}
bool ServerCleanupThread::truncate_files_recurisve(std::string path)
{
std::vector<SFile> files=getFiles(path, NULL);
for(size_t i=0;i<files.size();++i)
{
if(files[i].isdir)
{
bool b=truncate_files_recurisve(path+os_file_sep()+files[i].name);
if(!b)
{
return false;
}
}
else
{
bool b=os_file_truncate(path+os_file_sep()+files[i].name, 0);
if(!b)
{
ServerLogger::Log(logid, "Truncating file \""+path+os_file_sep()+files[i].name+"\" failed. Stopping truncating.", LL_ERROR);
return false;
}
}
}
return true;
}
bool ServerCleanupThread::cleanupSpace(int64 minspace, bool do_cleanup_other)
{
bool result;
Server->getThreadPool()->executeWait(new ServerCleanupThread(CleanupAction(minspace, &result, do_cleanup_other)),
"free space");
return result;
}
void ServerCleanupThread::removeUnknown(void)
{
Server->getThreadPool()->executeWait(new ServerCleanupThread(CleanupAction(ECleanupAction_RemoveUnknown)),
"remove unknown");
}
void ServerCleanupThread::enforce_quotas(void)
{
db_results cache_res;
if(db->getEngineName()=="sqlite")
{
cache_res=db->Read("PRAGMA cache_size");
ServerSettings server_settings(db);
db->Write("PRAGMA cache_size = -"+convert(server_settings.getSettings()->update_stats_cachesize));
}
std::vector<ServerCleanupDao::SClientInfo> clients=cleanupdao->getClients();
for(size_t i=0;i<clients.size();++i)
{
ServerLogger::Log(logid, "Enforcing quota for client \"" + (clients[i].name)+ "\" (id="+convert(clients[i].id)+")", LL_INFO);
std::ostringstream log;
log << "Quota enforcement report for client \"" << (clients[i].name) << "\" (id=" << clients[i].id << ")" << std::endl;
if(!enforce_quota(clients[i].id, log))
{
ClientMain::sendMailToAdmins("Quota enforcement failed", log.str());
ServerLogger::Log(logid, log.str(), LL_ERROR);
}
else
{
ServerLogger::Log(logid, log.str(), LL_DEBUG);
}
}
if(!cache_res.empty())
{
db->Write("PRAGMA cache_size = "+cache_res[0]["cache_size"]);
db->Write("PRAGMA shrink_memory");
}
}
bool ServerCleanupThread::enforce_quota(int clientid, std::ostringstream& log)
{
ServerSettings client_settings(db, clientid);
std::string client_quota = trim(client_settings.getSettings()->client_quota);
if(client_quota.empty() || client_quota=="100%" || client_quota=="-")
{
log << "Client does not have a quota or quota is 100%" << std::endl;
return true;
}
bool did_remove_something=false;
do
{
ServerCleanupDao::CondInt64 used_storage=cleanupdao->getUsedStorage(clientid);
if(!used_storage.exists || used_storage.value<0)
{
log << "Error getting used storage of client" << std::endl;
return false;
}
int64 client_quota=cleanup_amount(client_settings.getSettings()->client_quota, db);
log << "Client uses " << PrettyPrintBytes(used_storage.value) << " and has a quota of " << PrettyPrintBytes(client_quota) << std::endl;
if(used_storage.value<=client_quota)
{
log << "Client within assigned quota." << std::endl;
return true;
}
else
{
log << "This requires enforcement of the quota." << std::endl;
}
did_remove_something=false;
int state=0;
int nopc=0;
while(used_storage.value>client_quota && nopc<2)
{
std::string path=client_settings.getSettings()->backupfolder;
int64 available_space=os_free_space(os_file_prefix(path));
if(available_space==-1)
{
log << "Error getting free space -5" << std::endl;
return false;
}
int64 space_to_free=used_storage.value-client_quota;
int64 target_minspace=available_space+space_to_free;
if(target_minspace<0)
{
log << "Error. Target space is negative" << std::endl;
return false;
}
if(state==0)
{
int imagebid;
bool enough_space = false;
while (true)
{
if (cleanup_one_imagebackup_client(clientid, target_minspace, imagebid))
{
log << "Removed image backupd with id " << imagebid << std::endl;
did_remove_something = true;
//TODO: wait here for btrfs subvol remove to finish
if (hasEnoughFreeSpace(target_minspace, &client_settings))
{
enough_space = true;
break;
}
}
else
{
++nopc;
break;
}
}
if (enough_space)
{
break;
}
}
else
{
int filebid;
if(cleanup_one_filebackup_client(clientid, target_minspace, filebid))
{
log << "Removed file backup with id " << filebid << std::endl;
did_remove_something=true;
nopc=0;
//TODO: wait here for btrfs subvol remove to finish
if(hasEnoughFreeSpace(target_minspace, &client_settings))
{
break;
}
}
else
{
++nopc;
}
}
state=(state+1)%2;
}
if(did_remove_something)
{
ServerUpdateStats sus(false, false);
sus();
}
}
while(did_remove_something);
return false;
}
namespace
{
std::vector<bool> check_backupfolders_exist(std::vector<std::string>& old_backupfolders)
{
std::vector<bool> ret;
ret.resize(old_backupfolders.size());
for(size_t i=0;i<old_backupfolders.size();++i)
{
ret[i] = os_directory_exists(os_file_prefix(old_backupfolders[i]));
}
return ret;
}
}
bool ServerCleanupThread::correct_poolname( const std::string& backupfolder, const std::string& clientname, const std::string& pool_name, std::string& pool_path )
{
const std::string pool_root = backupfolder + os_file_sep() + clientname + os_file_sep() + ".directory_pool";
pool_path = pool_root + os_file_sep() + pool_name.substr(0, 2) + os_file_sep() + pool_name;
if(os_directory_exists(pool_path))
{
return true;
}
static std::vector<std::string> old_backupfolders = backupdao->getOldBackupfolders();
static std::vector<bool> backupfolders_exist = check_backupfolders_exist(old_backupfolders);
for(size_t i=0;i<old_backupfolders.size();++i)
{
if(backupfolders_exist[i])
{
pool_path = old_backupfolders[i] + os_file_sep() + clientname + os_file_sep() + ".directory_pool"
+ os_file_sep() + pool_name.substr(0, 2) + os_file_sep() + pool_name;
if(os_directory_exists(pool_path))
{
return true;
}
}
}
return false;
}
void ServerCleanupThread::check_symlinks( const ServerCleanupDao::SClientInfo& client_info, const std::string& backupfolder,
bool deep_check)
{
const int clientid=client_info.id;
const std::string& clientname=client_info.name;
const std::string pool_root = backupfolder + os_file_sep() + clientname + os_file_sep() + ".directory_pool";
IDatabase* db_links = Server->getDatabase(Server->getThreadID(), URBACKUPDB_SERVER_LINKS);
ServerLinkDao link_dao(db_links);
std::vector<int64> del_ids;
std::vector<std::pair<int64, std::string> > target_adjustments;
IQuery* q_directory_links = db_links->Prepare("SELECT id, name, target FROM directory_links WHERE clientid=?", false);
q_directory_links->Bind(clientid);
IDatabaseCursor* cursor = q_directory_links->Cursor();
std::vector<ServerLinkDao::DirectoryLinkEntry> deep_check_links;
db_single_result res;
while(cursor->next(res))
{
const std::string& pool_name = res["name"];
const std::string& target = res["target"];
std::string new_target = target;
std::string pool_path;
bool exists=true;
if(!correct_poolname(backupfolder, clientname, pool_name, pool_path) || !correct_target(backupfolder, new_target))
{
if(!correct_poolname(backupfolder, clientname, pool_name, pool_path))
{
Server->Log("Pool directory for pool entry \""+pool_name+"\" not found.");
}
if(!correct_target(backupfolder, new_target))
{
Server->Log("Pool directory source symbolic link \""+new_target+"\" not found.");
}
Server->Log("Deleting pool reference entry");
del_ids.push_back(watoi(res["id"]));
exists=false;
}
else if(target!=new_target)
{
Server->Log("Adjusting pool directory target from \""+target+"\" to \""+new_target+"\"");
target_adjustments.push_back(std::make_pair(watoi(res["id"]), new_target));
}
if(exists)
{
std::string symlink_pool_path;
if(os_get_symlink_target(os_file_prefix(new_target), symlink_pool_path))
{
if(symlink_pool_path!=pool_path)
{
Server->Log("Correcting target of symlink \""+new_target+"\" to \""+pool_path+"\"");
if(os_remove_symlink_dir(os_file_prefix(new_target)))
{
if(!os_link_symbolic(pool_path, os_file_prefix(new_target)))
{
Server->Log("Could not create symlink at \""+new_target+"\" to \""+pool_path+"\"", LL_ERROR);
}
}
else
{
Server->Log("Error deleting symlink \""+new_target+"\"", LL_ERROR);
}
}
if (deep_check)
{
ServerLinkDao::DirectoryLinkEntry e;
e.target = symlink_pool_path;
e.name = pool_name;
deep_check_links.push_back(e);
}
}
}
}
db_links->destroyQuery(q_directory_links);
for(size_t i=0;i<del_ids.size();++i)
{
link_dao.deleteLinkReferenceEntry(del_ids[i]);
}
for(size_t i=0;i<target_adjustments.size();++i)
{
link_dao.updateLinkReferenceTarget(target_adjustments[i].second, target_adjustments[i].first);
}
for (size_t i = 0; i < deep_check_links.size(); ++i)
{
reference_contained_directory_links(link_dao, clientid, deep_check_links[i].name,
std::string(), deep_check_links[i].target);
}
if(os_directory_exists(pool_root))
{
std::vector<SFile> first_files = getFiles(pool_root, NULL);
for(size_t i=0;i<first_files.size();++i)
{
if(!first_files[i].isdir)
continue;
if(first_files[i].issym)
continue;
std::string curr_path = pool_root + os_file_sep() + first_files[i].name;
std::vector<SFile> pool_files = getFiles(curr_path, NULL);
if (pool_files.empty())
{
os_remove_dir(curr_path);
continue;
}
bool has_dir = false;
for(size_t j=0;j<pool_files.size();++j)
{
if(!pool_files[j].isdir)
continue;
if(pool_files[j].issym)
continue;
std::string pool_path = curr_path + os_file_sep() + pool_files[j].name;
if(link_dao.getDirectoryRefcount(clientid, pool_files[j].name)==0)
{
Server->Log("Refcount of \""+pool_path+"\" is zero. Deleting pool folder.", LL_WARNING);
if (!deep_check)
{
Server->Log("Restarting symlink check. Starting deep directory link check...", LL_WARNING);
check_symlinks(client_info, backupfolder, true);
return;
}
if(!remove_directory_link_dir(pool_path, link_dao, clientid))
{
Server->Log("Could not remove pool folder \""+pool_path+"\"", LL_ERROR);
}
}
else
{
has_dir = true;
}
}
if (!has_dir)
{
os_remove_dir(curr_path);
}
}
}
}
bool ServerCleanupThread::correct_target(const std::string& backupfolder, std::string& target)
{
if(os_is_symlink(os_file_prefix(target)))
{
return true;
}
static std::vector<std::string> old_backupfolders = backupdao->getOldBackupfolders();
for(size_t i=0;i<old_backupfolders.size();++i)
{
size_t erase_size = old_backupfolders[i].size() + os_file_sep().size();
if(target.size()>erase_size &&
next(target, 0, old_backupfolders[i]))
{
std::string new_target = backupfolder + os_file_sep() + target.substr(erase_size);
if(os_is_symlink(os_file_prefix(new_target)))
{
target = new_target;
return true;
}
}
}
return false;
}
void ServerCleanupThread::delete_incomplete_file_backups( void )
{
std::vector<ServerCleanupDao::SIncompleteFileBackup> incomplete_file_backups =
cleanupdao->getIncompleteFileBackups();
ServerSettings settings(db);
for(size_t i=0;i<incomplete_file_backups.size();++i)
{
const ServerCleanupDao::SIncompleteFileBackup& backup = incomplete_file_backups[i];
ServerLogger::Log(logid, "Deleting incomplete file backup ( id="+convert(backup.id)+", backuptime="+backup.backuptime+", path="+backup.path+" ) from client \""+backup.clientname+"\" ( id="+convert(backup.clientid)+" ) ...", LL_INFO);
if(!deleteFileBackup(settings.getSettings()->backupfolder, backup.clientid, backup.id, true, true))
{
ServerLogger::Log(logid, "Error deleting file backup", LL_WARNING);
}
else
{
ServerLogger::Log(logid, "done.");
}
}
}
void ServerCleanupThread::delete_pending_file_backups()
{
std::vector<ServerCleanupDao::SIncompleteFileBackup> delete_pending_file_backups =
cleanupdao->getDeletePendingFileBackups();
ServerSettings settings(db);
for (size_t i = 0; i<delete_pending_file_backups.size(); ++i)
{
const ServerCleanupDao::SIncompleteFileBackup& backup = delete_pending_file_backups[i];
ServerLogger::Log(logid, "Deleting manually marked to be deleted file backup ( id=" + convert(backup.id) + ", backuptime=" + backup.backuptime + ", path=" + backup.path + " ) from client \"" + backup.clientname + "\" ( id=" + convert(backup.clientid) + " ) ...", LL_INFO);
if (!deleteFileBackup(settings.getSettings()->backupfolder, backup.clientid, backup.id, false, true))
{
ServerLogger::Log(logid, "Error deleting file backup", LL_WARNING);
}
else
{
ServerLogger::Log(logid, "done.");
}
}
}
void ServerCleanupThread::rewrite_history(const std::string& back_start, const std::string& back_stop, const std::string& date_grouping)
{
ServerLogger::Log(logid, "Reading history...", LL_DEBUG);
std::vector<ServerCleanupDao::SHistItem> daily_history = cleanupdao->getClientHistory(back_start, back_stop, date_grouping);
ServerLogger::Log(logid, convert(daily_history.size()) + " history items read", LL_DEBUG);
db_results foreign_keys;
if(db->getEngineName()=="sqlite")
{
foreign_keys=db->Read("PRAGMA foreign_keys");
db->Write("PRAGMA foreign_keys = 0");
}
{
DBScopedDetach detachDbs(db);
DBScopedWriteTransaction transaction(db);
ServerLogger::Log(logid, "Deleting history...", LL_DEBUG);
cleanupdao->deleteClientHistoryIds(back_start, back_stop);
cleanupdao->deleteClientHistoryItems(back_start, back_stop);
ServerLogger::Log(logid, "Writing history...", LL_DEBUG);
for(size_t i=0;i<daily_history.size();++i)
{
const ServerCleanupDao::SHistItem& hist_item = daily_history[i];
cleanupdao->insertClientHistoryId(hist_item.max_created);
int64 hist_id = db->getLastInsertID();
cleanupdao->insertClientHistoryItem(hist_item.id,
hist_item.name, hist_item.lastbackup, hist_item.lastseen,
hist_item.lastbackup_image,
hist_item.bytes_used_files, hist_item.bytes_used_images,
hist_item.max_created,
hist_id);
}
}
if(db->getEngineName()=="sqlite" &&
!foreign_keys.empty() )
{
db->Write("PRAGMA foreign_keys = " + foreign_keys[0]["foreign_keys"]);
}
}
void ServerCleanupThread::cleanup_client_hist()
{
ServerLogger::Log(logid, "Rewriting daily history...", LL_INFO);
rewrite_history("-2 days", "-2 month", "%Y-%m-%d");
ServerLogger::Log(logid, "Rewriting monthly history...", LL_INFO);
rewrite_history("-2 month", "-4 years", "%Y-%m");
ServerLogger::Log(logid, "Rewriting yearly history...", LL_INFO);
rewrite_history("-2 years", "-1000 years", "%Y");
}
void ServerCleanupThread::cleanup_all_system_images(ServerSettings & settings)
{
std::vector<ServerCleanupDao::SClientInfo> res_clients = cleanupdao->getClients();
for (size_t i = 0; i < res_clients.size(); ++i)
{
int clientid = res_clients[i].id;
const std::string& clientname = res_clients[i].name;
cleanup_system_images(clientid, clientname, settings);
}
}
void ServerCleanupThread::cleanup_system_images(int clientid, std::string clientname, ServerSettings& settings)
{
std::vector<ServerCleanupDao::SImageBackupInfo>
res_image_backups = cleanupdao->getOldImageBackupsOfClient(clientid);
for (size_t j = 0; j<res_image_backups.size(); ++j)
{
if (res_image_backups[j].letter == "SYSVOL"
|| res_image_backups[j].letter == "ESP")
{
if (!cleanupdao->getParentImageBackup(res_image_backups[j].id).exists
&& !isImageLockedFromCleanup(res_image_backups[j].id) )
{
ServerLogger::Log(logid, "Image backup [id=" + convert(res_image_backups[j].id) + " path="
+ res_image_backups[j].path + " clientname=" + clientname
+ "] is a system reserved or EFI system partition image older than 24h and has no parent image. Deleting it.", LL_INFO);
if (!removeImage(res_image_backups[j].id, &settings, false, false, true, false))
{
ServerLogger::Log(logid, "Could not remove image backup [id=" + convert(res_image_backups[j].id) + " path=" + res_image_backups[j].path + " clientname=" + clientname + "]", LL_ERROR);
}
}
}
}
}
void ServerCleanupThread::cleanup_other()
{
ServerLogger::Log(logid, "Deleting old logs...", LL_INFO);
cleanupdao->cleanupBackupLogs();
cleanupdao->cleanupAuthLog();
ServerLogger::Log(logid, "Done deleting old logs", LL_INFO);
ServerLogger::Log(logid, "Cleaning history...", LL_INFO);
cleanup_client_hist();
ServerLogger::Log(logid, "Done cleaning history", LL_INFO);
ServerLogger::Log(logid, "Cleaning deleted backups history...", LL_INFO);
cleanupLastActs();
ServerLogger::Log(logid, "Done cleaning deleted backups history.", LL_INFO);
ServerLogger::Log(logid, "Cleaning up client lists...", LL_INFO);
cleanup_clientlists();
ServerLogger::Log(logid, "Done cleaning up client lists.", LL_INFO);
ServerLogger::Log(logid, "Cleaning up old client access tokens...", LL_INFO);
cleanupdao->getDatabase()->Write("DELETE FROM user_tokens WHERE id IN ("
"SELECT a.id FROM user_tokens a WHERE EXISTS ("
"SELECT id FROM user_tokens b WHERE a.username=b.username "
"AND a.clientid=b.clientid AND a.tgroup=b.tgroup AND b.created>a.created "
"AND (strftime('%s', b.created)-strftime('%s', a.created)) > 15552000));");
ServerLogger::Log(logid, "Done cleaning up old client access tokens.", LL_INFO);
}
void ServerCleanupThread::removeFileBackupSql( int backupid )
{
DBScopedSynchronous synchronous_files(filesdao->getDatabase());
filesdao->BeginWriteTransaction();
BackupServerHash::SInMemCorrection correction;
ServerFilesDao::SBackupIdMinMax minmax = filesdao->getBackupIdMinMax(backupid);
correction.max_correct = minmax.tmax;
correction.min_correct = minmax.tmin;
IQuery* q_iterate = filesdao->getDatabase()->Prepare("SELECT id, shahash, filesize, rsize, clientid, backupid, incremental, next_entry, prev_entry, pointed_to FROM files WHERE backupid=?", false);
q_iterate->Bind(backupid);
IDatabaseCursor* cursor = q_iterate->Cursor();
bool modified_file_entry_index = false;
db_single_result res;
while(cursor->next(res))
{
int64 id = watoi64(res["id"]);
int64 filesize = watoi64(res["filesize"]);
int64 rsize = watoi64(res["rsize"]);
int clientid = watoi(res["clientid"]);
int backupid = watoi(res["backupid"]);
int incremental = watoi(res["incremental"]);
int64 next_entry = watoi64(res["next_entry"]);
int64 prev_entry = watoi64(res["prev_entry"]);
int pointed_to = watoi(res["pointed_to"]);
std::map<int64, int64>::iterator it_next = correction.next_entries.find(id);
if (it_next != correction.next_entries.end())
{
if (it_next->second != next_entry)
{
int abc = 5;
}
next_entry = it_next->second;
correction.next_entries.erase(it_next);
}
std::map<int64, int64>::iterator it_prev= correction.prev_entries.find(id);
if (it_prev != correction.prev_entries.end())
{
if (it_prev->second != prev_entry)
{
int abc = 5;
}
prev_entry = it_prev->second;
correction.prev_entries.erase(it_prev);
}
std::map<int64, int>::iterator it_pointed_to = correction.pointed_to.find(id);
if (it_pointed_to != correction.pointed_to.end())
{
if (it_pointed_to->second != pointed_to)
{
int abc = 5;
}
pointed_to = it_pointed_to->second;
correction.pointed_to.erase(it_pointed_to);
}
if (pointed_to)
{
modified_file_entry_index = true;
}
BackupServerHash::deleteFileSQL(*filesdao, *fileindex.get(), res["shahash"].c_str(),
filesize, rsize, clientid, backupid, incremental, id, prev_entry, next_entry, pointed_to, false, false, false, true, &correction);
}
filesdao->getDatabase()->destroyQuery(q_iterate);
for (std::map<int64, int64>::iterator it_next = correction.next_entries.begin();
it_next != correction.next_entries.end(); ++it_next)
{
filesdao->setNextEntry(it_next->second, it_next->first);
}
for (std::map<int64, int64>::iterator it_prev = correction.prev_entries.begin();
it_prev != correction.prev_entries.end(); ++it_prev)
{
filesdao->setPrevEntry(it_prev->second, it_prev->first);
}
for (std::map<int64, int>::iterator it_pointed_to = correction.pointed_to.begin();
it_pointed_to != correction.pointed_to.end(); ++it_pointed_to)
{
filesdao->setPointedTo(it_pointed_to->second, it_pointed_to->first);
}
filesdao->deleteFiles(backupid);
if (modified_file_entry_index)
{
FileIndex::flush();
}
filesdao->endTransaction();
cleanupdao->removeFileBackup(backupid);
}
bool ServerCleanupThread::backup_clientlists()
{
ServerSettings settings(db);
if(settings.getSettings()->backup_database)
{
std::string bfolder=settings.getSettings()->backupfolder+os_file_sep()+"urbackup";
std::string srcfolder = Server->getServerWorkingDir()+os_file_sep()+"urbackup";
std::vector<SFile> files = getFiles(srcfolder);
bool has_error=false;
for(size_t i=0;i<files.size();++i)
{
if(files[i].name.find("clientlist_b_")==0)
{
std::string error_str;
if(!copy_file(os_file_prefix(srcfolder + os_file_sep() + files[i].name),
os_file_prefix(bfolder + os_file_sep() + files[i].name+"~"), true, &error_str))
{
ServerLogger::Log(logid, "Error backing up "+files[i].name+". "+error_str, LL_ERROR);
has_error=true;
}
}
}
return !has_error;
}
else
{
return true;
}
}
void ServerCleanupThread::ren_files_backupfolder()
{
ServerSettings settings(db);
std::string bfolder=settings.getSettings()->backupfolder+os_file_sep()+"urbackup";
if(!os_directory_exists(bfolder) )
{
os_create_dir(bfolder);
return;
}
std::vector<SFile> files = getFiles(bfolder);
for (size_t i = 0; i < files.size(); ++i)
{
if (!files[i].isdir &&
!files[i].name.empty() &&
files[i].name[files[i].name.size() - 1] != '~')
{
Server->deleteFile(os_file_prefix(bfolder + os_file_sep() + files[i].name));
}
}
for(size_t i=0;i<files.size();++i)
{
if(!files[i].isdir &&
!files[i].name.empty() &&
files[i].name[files[i].name.size()-1]=='~')
{
os_rename_file(os_file_prefix(bfolder+os_file_sep()+files[i].name), os_file_prefix(bfolder+os_file_sep()+files[i].name.substr(0, files[i].name.size()-1)));
}
}
}
void ServerCleanupThread::setClientlistDeletionAllowed(bool b)
{
IScopedLock lock(mutex);
allow_clientlist_deletion = b;
}
namespace
{
class CleanupFrontend : public IKvStoreFrontend
{
public:
virtual void incr_total_del_ops() {}
virtual std::string prefixKey(const std::string& key) {
std::string md5sum = Server->GenerateHexMD5(key);
return md5sum.substr(0, 3) + "/" + md5sum.substr(3, 2) + "/" + key;
}
virtual std::string encodeKey(int64 cd_id, const std::string& key, int64 transid) {
if (cd_id == 0)
return std::to_string(transid) + "_" +
bytesToHex(reinterpret_cast<const unsigned char*>(key.c_str()), key.size());
return std::to_string(cd_id) + "_" + std::to_string(transid) + "_" +
bytesToHex(reinterpret_cast<const unsigned char*>(key.c_str()), key.size());
}
virtual bool log_del_mirror(const std::string& fn) {
return false;
}
};
}
bool ServerCleanupThread::delete_local_backup(const std::string& backupfolder, int clientid, int backupid, bool force_remove)
{
ServerCleanupDao::CondString cond_clientname = cleanupdao->getClientName(clientid);
if (!cond_clientname.exists)
{
ServerLogger::Log(logid, "Error getting clientname in ServerCleanupThread::delete_local_backup", LL_ERROR);
return false;
}
std::string& clientname = cond_clientname.value;
ServerCleanupDao::CondString cond_perm_uid = cleanupdao->getClientPermUid(clientid);
if (!cond_perm_uid.exists)
{
ServerLogger::Log(logid, "Error getting perm_uid in ServerCleanupThread::delete_local_backup", LL_ERROR);
return false;
}
std::string perm_uid = bytesToHex(cond_perm_uid.value);
if (cleanupdao->getFileBackupClientId(backupid).value != clientid)
{
ServerLogger::Log(logid, "Clientid does not match clientid in backup in ServerCleanupThread::delete_local_backup", LL_ERROR);
return false;
}
ServerCleanupDao::CondString cond_backuppath = cleanupdao->getFileBackupPath(backupid);
if (!cond_backuppath.exists)
{
ServerLogger::Log(logid, "Error getting backuppath in ServerCleanupThread::delete_local_backup", LL_ERROR);
return false;
}
std::string backuppath = cond_backuppath.value;
if (backuppath.empty())
{
ServerLogger::Log(logid, "Error backuppath empty in ServerCleanupThread::delete_local_backup", LL_ERROR);
return false;
}
if (backupfolder.empty())
{
ServerLogger::Log(logid, "Error backupfolder empty in ServerCleanupThread::delete_local_backup", LL_ERROR);
return false;
}
if (clientname.empty())
{
ServerLogger::Log(logid, "Error clientname empty in ServerCleanupThread::delete_local_backup", LL_ERROR);
return false;
}
std::string path = backupfolder + os_file_sep() + clientname + os_file_sep() + backuppath;
if (!os_directory_exists(os_file_prefix(path))
&& os_directory_exists(os_file_prefix(path + ".startup-del")))
{
backuppath += ".startup-del";
path += ".startup-del";
}
if (getFile(path + os_file_sep() + ".hashes" + os_file_sep() + ".local_backup_7267f182-4341-4cda-ada1-4c0c554322d8") != "1")
{
ServerLogger::Log(logid, "No a local backup in ServerCleanupThread::delete_local_backup", LL_ERROR);
return false;
}
bool deletion_protected = cleanupdao->getFileBackupDeletionProtected(backupid).value != 0;
if (deletion_protected)
{
std::vector<SFile> files = getFiles(path);
ServerSettings settings(db, clientid);
std::string dest_url = settings.getSettings()->backup_dest_url;
dest_url = greplace("$CLIENTUID$", perm_uid, dest_url);
dest_url = greplace("$CLIENTNAME$", clientname, dest_url);
for (SFile cleanup_file : files)
{
if (!cleanup_file.isdir &&
next(cleanup_file.name, 0, "del_objs_"))
{
CleanupFrontend cleanup_frontend;
ObjectCollector object_collector(&cleanup_frontend, path + os_file_sep() + cleanup_file.name);
if (object_collector.get_has_error())
{
ServerLogger::Log(logid, "Error reading " + path + os_file_sep() + cleanup_file.name, LL_ERROR);
return false;
}
object_collector.finalize();
str_map secret_params;
ParseParamStrHttp(settings.getSettings()->backup_dest_secret_params, &secret_params);
IClouddriveFactory::CloudSettings cloud_settings;
if (!parse_backup_url(dest_url,
settings.getSettings()->backup_dest_params, secret_params,
cloud_settings))
{
ServerLogger::Log(logid, "Error parsing backup dest url to delete objects: " + settings.getSettings()->backup_dest_url, LL_ERROR);
return false;
}
std::unique_ptr<IKvStoreBackend> backend(clouddrive_fak->createBackend(nullptr, cloud_settings));
if (!backend)
{
ServerLogger::Log(logid, "Error creating backend to delete objects", LL_ERROR);
return false;
}
for (size_t i = 0; i < object_collector.key_next_funs.size(); ++i)
{
if (object_collector.locinfo_next_funs[i] != nullptr)
{
if (!backend->del(object_collector.key_next_funs[i],
object_collector.locinfo_next_funs[i], true))
{
ServerLogger::Log(logid, "Error deleting objects of backup (with locinfo)", LL_ERROR);
return false;
}
object_collector.locinfo_next_funs[i](IKvStoreBackend::key_next_action_t::clear, nullptr);
}
else
{
if (!backend->del(object_collector.key_next_funs[i], true))
{
ServerLogger::Log(logid, "Error deleting objects of backup", LL_ERROR);
return false;
}
}
object_collector.key_next_funs[i](IKvStoreBackend::key_next_action_t::clear, nullptr);
}
if (object_collector.task_id == 2) //remove transaction
{
if (object_collector.trans_ids.size() != 1)
{
ServerLogger::Log(logid, "Remove transaction task without transaction info", LL_ERROR);
return false;
}
int64 trans_id = object_collector.trans_ids[0];
std::vector<std::string> add_backend_keys;
if (object_collector.completed != 0)
{
if (object_collector.cd_id == 0)
add_backend_keys.push_back(cleanup_frontend.prefixKey(convert(trans_id) + "_complete"));
else
add_backend_keys.push_back(cleanup_frontend.prefixKey(convert(object_collector.cd_id) + "_" + convert(trans_id) + "_complete"));
}
if (object_collector.active == 0)
{
if (object_collector.cd_id == 0)
add_backend_keys.push_back(cleanup_frontend.prefixKey(convert(trans_id) + "_inactive"));
else
add_backend_keys.push_back(cleanup_frontend.prefixKey(convert(object_collector.cd_id) + "_" + convert(trans_id) + "_inactive"));
}
if (!add_backend_keys.empty())
{
size_t idx = 0;
backend->del([&add_backend_keys, &idx](IKvStoreBackend::key_next_action_t action, std::string* key) {
if (action == IKvStoreBackend::key_next_action_t::clear) {
add_backend_keys.clear();
assert(key == nullptr);
return true;
}
if (action == IKvStoreBackend::key_next_action_t::reset) {
idx = 0;
assert(key == nullptr);
return true;
}
assert(action == IKvStoreBackend::key_next_action_t::next);
if (idx >= add_backend_keys.size())
return false;
*key = add_backend_keys[idx];
++idx;
return true;
}, true);
}
}
}
}
cleanupdao->resetFileBackupDeletionProtected(backupid);
cleanupdao->resetFileBackupIncrementalRef(backupid);
}
cleanupdao->setFileBackupDeleteClientPending(backupid);
ServerStatus::updateActive();
return true;
}
bool ServerCleanupThread::backup_ident()
{
ServerSettings settings(db);
std::string bfolder=settings.getSettings()->backupfolder+os_file_sep()+"urbackup";
if (!os_directory_exists(bfolder))
{
os_create_dir(bfolder);
}
std::string srcfolder = Server->getServerWorkingDir()+os_file_sep()+"urbackup";
std::vector<std::string> filelist;
filelist.push_back("server_ident.key");
filelist.push_back("server_ident.priv");
filelist.push_back("server_ident_ecdsa409k1.priv");
filelist.push_back("server_ident.pub");
filelist.push_back("server_ident_ecdsa409k1.pub");
bool has_error=false;
for(size_t i=0;i<filelist.size();++i)
{
if(!copy_file(os_file_prefix(srcfolder + os_file_sep() + filelist[i]), os_file_prefix(bfolder + os_file_sep() + filelist[i]+"~"), true))
{
ServerLogger::Log(logid, "Error backing up "+filelist[i], LL_ERROR);
has_error=true;
}
}
return !has_error;
}
bool ServerCleanupThread::cleanup_clientlists()
{
std::string srcfolder = Server->getServerWorkingDir()+os_file_sep()+"urbackup";
std::vector<SFile> files = getFiles(srcfolder);
bool has_error=false;
for(size_t i=0;i<files.size();++i)
{
if(next(files[i].name, 0, "clientlist_b_"))
{
int backupid = watoi(getbetween("clientlist_b_", ".ub", files[i].name));
if(cleanupdao->hasMoreRecentFileBackup(backupid).exists)
{
if(!Server->deleteFile(os_file_prefix(srcfolder+os_file_sep()+files[i].name)))
{
has_error=true;
}
}
}
}
return !has_error;
}
void ServerCleanupThread::deleteClientSQL(IDatabase * db, int clientid)
{
//Remove logentries
IQuery *q = db->Prepare("DELETE FROM logs WHERE clientid=?", false);
q->Bind(clientid); q->Write(); q->Reset();
db->destroyQuery(q);
//history data
q = db->Prepare("DELETE FROM clients_hist WHERE id=?", false);
q->Bind(clientid); q->Write(); q->Reset();
db->destroyQuery(q);
//settings
q = db->Prepare("DELETE FROM settings_db.settings WHERE clientid=?", false);
q->Bind(clientid); q->Write(); q->Reset();
db->destroyQuery(q);
//stats
q = db->Prepare("DELETE FROM del_stats WHERE clientid=?", false);
q->Bind(clientid); q->Write(); q->Reset();
db->destroyQuery(q);
//client
q = db->Prepare("DELETE FROM clients WHERE id=?", false);
q->Bind(clientid); q->Write(); q->Reset();
db->destroyQuery(q);
}
#endif //CLIENT_ONLY