mirror of
https://github.com/uroni/urbackup_backend.git
synced 2025-10-26 11:36:50 +00:00
150 lines
3.9 KiB
C++
150 lines
3.9 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_ping.h"
|
|
#include "ClientMain.h"
|
|
#include "../Interface/Server.h"
|
|
#include "../stringtools.h"
|
|
#include "server_status.h"
|
|
|
|
const int64 ping_intervall=10000;
|
|
|
|
ServerPingThread::ServerPingThread(ClientMain *client_main, const std::string& clientname,
|
|
size_t status_id, bool with_eta, std::string server_token)
|
|
: client_main(client_main), clientname(clientname), status_id(status_id),
|
|
with_eta(with_eta), server_token(server_token)
|
|
{
|
|
stop=false;
|
|
is_timeout=false;
|
|
}
|
|
|
|
void ServerPingThread::operator()(void)
|
|
{
|
|
ClientMain::SConnection connection;
|
|
int64 last_ping_ok=Server->getTimeMS();
|
|
while(!stop)
|
|
{
|
|
//Server->Log("Sending ping running...", LL_DEBUG);
|
|
std::string pcdone;
|
|
SProcess proc = ServerStatus::getProcess(clientname, status_id);
|
|
|
|
if(proc.action==sa_none)
|
|
{
|
|
break;
|
|
}
|
|
|
|
int i_pcdone = proc.pcdone;
|
|
if(i_pcdone>=0)
|
|
{
|
|
pcdone=convert(i_pcdone);
|
|
}
|
|
|
|
int64 add_time = Server->getTimeMS() - proc.eta_set_time;
|
|
int64 etams = proc.eta_ms - add_time;
|
|
|
|
if(!with_eta)
|
|
{
|
|
if(client_main->sendClientMessage("PING RUNNING -"+pcdone+"-#token="+server_token,
|
|
"OK", "Error sending 'running' ping to client", 30000, false, LL_DEBUG,
|
|
NULL, NULL, &connection))
|
|
{
|
|
last_ping_ok=Server->getTimeMS();
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if(etams>0 && etams<60*1000)
|
|
{
|
|
etams=61*1000;
|
|
}
|
|
std::string ret = client_main->sendClientMessage("2PING RUNNING pc_done="+pcdone+(etams>0?"&eta_ms="+convert(etams):"")
|
|
+"&status_id="+convert(status_id)
|
|
+"&speed_bpms="+convert(proc.speed_bpms)
|
|
+"&total_bytes="+convert(proc.total_bytes)
|
|
+"&done_bytes="+convert(proc.done_bytes)
|
|
+"&paused_fb=1"
|
|
+"#token="+server_token, "Error sending 'running' (2) ping to client",
|
|
30000, false, LL_DEBUG, &connection);
|
|
if(ret=="OK"
|
|
|| ret=="PAUSED")
|
|
{
|
|
if (ret == "PAUSED")
|
|
{
|
|
setPaused(&proc, true);
|
|
}
|
|
else
|
|
{
|
|
setPaused(&proc, false);
|
|
}
|
|
last_ping_ok=Server->getTimeMS();
|
|
client_main->refreshSessionIdentity();
|
|
}
|
|
else
|
|
{
|
|
setPaused(&proc, false);
|
|
|
|
if (!ret.empty())
|
|
{
|
|
ret = ". Received: \"" + ret+"\"";
|
|
}
|
|
Server->Log("Error sending 'running' (3) ping to client" + ret, LL_DEBUG);
|
|
}
|
|
}
|
|
|
|
//Server->Log("Done sending ping running.", LL_DEBUG);
|
|
|
|
if(Server->getTimeMS()-last_ping_ok>ping_intervall*6)
|
|
{
|
|
is_timeout=true;
|
|
}
|
|
else
|
|
{
|
|
is_timeout=false;
|
|
}
|
|
|
|
Server->wait(ping_intervall);
|
|
}
|
|
while(!stop)
|
|
{
|
|
Server->wait(1000);
|
|
}
|
|
setPaused(NULL, false);
|
|
delete this;
|
|
}
|
|
|
|
void ServerPingThread::setStop(bool b)
|
|
{
|
|
stop=b;
|
|
}
|
|
|
|
bool ServerPingThread::isTimeout(void)
|
|
{
|
|
return is_timeout;
|
|
}
|
|
|
|
void ServerPingThread::setPaused(SProcess* proc, bool b)
|
|
{
|
|
if (proc==NULL || b != proc->paused)
|
|
{
|
|
ServerStatus::setProcessPaused(clientname, status_id, b);
|
|
}
|
|
}
|
|
|
|
#endif //CLIENT_ONLY
|