94 lines
2.2 KiB
C++
94 lines
2.2 KiB
C++
/*
|
|
* SPDX-FileCopyrightText: 2013 Albert Vaca <albertvaka@gmail.com>
|
|
*
|
|
* SPDX-License-Identifier: GPL-2.0-only OR GPL-3.0-only OR LicenseRef-KDE-Accepted-GPL
|
|
*/
|
|
|
|
#include "uploadjob.h"
|
|
|
|
#include <KLocalizedString>
|
|
|
|
#include "lanlinkprovider.h"
|
|
#include "kdeconnectconfig.h"
|
|
#include "core_debug.h"
|
|
#include <daemon.h>
|
|
|
|
UploadJob::UploadJob(const NetworkPacket& networkPacket)
|
|
: KJob()
|
|
, m_networkPacket(networkPacket)
|
|
, m_input(networkPacket.payload())
|
|
, m_socket(nullptr)
|
|
{
|
|
}
|
|
|
|
void UploadJob::setSocket(QSslSocket* socket)
|
|
{
|
|
m_socket = socket;
|
|
m_socket->setParent(this);
|
|
}
|
|
|
|
void UploadJob::start()
|
|
{
|
|
if (!m_input->open(QIODevice::ReadOnly)) {
|
|
qCWarning(KDECONNECT_CORE) << "error when opening the input to upload";
|
|
return; //TODO: Handle error, clean up...
|
|
}
|
|
|
|
if (!m_socket) {
|
|
qCWarning(KDECONNECT_CORE) << "you must call setSocket() before calling start()";
|
|
return;
|
|
}
|
|
|
|
connect(m_input.data(), &QIODevice::aboutToClose, this, &UploadJob::aboutToClose);
|
|
|
|
bytesUploaded = 0;
|
|
setProcessedAmount(Bytes, bytesUploaded);
|
|
|
|
connect(m_socket, &QSslSocket::encryptedBytesWritten, this, &UploadJob::encryptedBytesWritten);
|
|
|
|
uploadNextPacket();
|
|
}
|
|
|
|
void UploadJob::uploadNextPacket()
|
|
{
|
|
qint64 bytesAvailable = m_input->bytesAvailable();
|
|
|
|
if ( bytesAvailable > 0) {
|
|
qint64 bytesToSend = qMin(m_input->bytesAvailable(), (qint64)4096);
|
|
bytesUploading = m_socket->write(m_input->read(bytesToSend));
|
|
}
|
|
|
|
if (bytesAvailable <= 0 || bytesUploading < 0) {
|
|
m_input->close();
|
|
disconnect(m_socket, &QSslSocket::encryptedBytesWritten, this, &UploadJob::encryptedBytesWritten);
|
|
}
|
|
}
|
|
|
|
void UploadJob::encryptedBytesWritten(qint64 bytes)
|
|
{
|
|
Q_UNUSED(bytes);
|
|
|
|
if (m_socket->encryptedBytesToWrite() == 0) {
|
|
bytesUploaded += bytesUploading;
|
|
setProcessedAmount(Bytes, bytesUploaded);
|
|
|
|
uploadNextPacket();
|
|
}
|
|
}
|
|
|
|
void UploadJob::aboutToClose()
|
|
{
|
|
m_socket->disconnectFromHost();
|
|
emitResult();
|
|
}
|
|
|
|
bool UploadJob::stop() {
|
|
m_input->close();
|
|
|
|
return true;
|
|
}
|
|
|
|
const NetworkPacket UploadJob::getNetworkPacket()
|
|
{
|
|
return m_networkPacket;
|
|
}
|