diff --git a/core/backends/lan/compositeuploadjob.cpp b/core/backends/lan/compositeuploadjob.cpp index 619901c1..b9d34669 100644 --- a/core/backends/lan/compositeuploadjob.cpp +++ b/core/backends/lan/compositeuploadjob.cpp @@ -1,292 +1,292 @@ /** * Copyright 2018 Erik Duisters * * This program is free software; you can redistribute it and/or * modify it under the terms of the GNU General Public License as * published by the Free Software Foundation; either version 2 of * the License or (at your option) version 3 or any later version * accepted by the membership of KDE e.V. (or its successor approved * by the membership of KDE e.V.), which shall act as a proxy * defined in Section 14 of version 3 of the license. * * 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 General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program. If not, see . */ #include "compositeuploadjob.h" #include #include #include #include #include "lanlinkprovider.h" #include #include "plugins/share/shareplugin.h" CompositeUploadJob::CompositeUploadJob(const QString& deviceId, bool displayNotification) : KCompositeJob() , m_server(new Server(this)) , m_socket(nullptr) , m_port(0) , m_deviceId(deviceId) , m_running(false) , m_currentJobNum(1) , m_totalJobs(0) , m_currentJobSendPayloadSize(0) , m_totalSendPayloadSize(0) , m_totalPayloadSize(0) , m_currentJob(nullptr) , m_prevElapsedTime(0) , m_updatePacketPending(false) { setCapabilities(Killable); if (displayNotification) { KIO::getJobTracker()->registerJob(this); } } bool CompositeUploadJob::isRunning() { return m_running; } void CompositeUploadJob::start() { if (m_running) { qCWarning(KDECONNECT_CORE) << "CompositeUploadJob::start() - already running"; return; } if (!hasSubjobs()) { qCWarning(KDECONNECT_CORE) << "CompositeUploadJob::start() - there are no subjobs to start"; emitResult(); return; } if (!startListening()) { return; } connect(m_server, &QTcpServer::newConnection, this, &CompositeUploadJob::newConnection); m_running = true; //Give SharePlugin some time to add subjobs QMetaObject::invokeMethod(this, "startNextSubJob", Qt::QueuedConnection); } bool CompositeUploadJob::startListening() { m_port = MIN_PORT; while (!m_server->listen(QHostAddress::Any, m_port)) { m_port++; if (m_port > MAX_PORT) { //No ports available? qCWarning(KDECONNECT_CORE) << "CompositeUploadJob::startListening() - Error opening a port in range" << MIN_PORT << "-" << MAX_PORT; m_port = 0; setError(NoPortAvailable); setErrorText(i18n("Couldn't find an available port")); emitResult(); return false; } } qCDebug(KDECONNECT_CORE) << "CompositeUploadJob::startListening() - listening on port: " << m_port; return true; } void CompositeUploadJob::startNextSubJob() { m_currentJob = qobject_cast(subjobs().at(0)); m_currentJobSendPayloadSize = 0; emitDescription(m_currentJob->getNetworkPacket().get(QStringLiteral("filename"))); connect(m_currentJob, SIGNAL(processedAmount(KJob*,KJob::Unit,qulonglong)), this, SLOT(slotProcessedAmount(KJob*,KJob::Unit,qulonglong))); //Already done by KCompositeJob //connect(m_currentJob, &KJob::result, this, &CompositeUploadJob::slotResult); //TODO: Create a copy of the networkpacket that can be re-injected if sending via lan fails? NetworkPacket np = m_currentJob->getNetworkPacket(); np.setPayload(nullptr, np.payloadSize()); np.setPayloadTransferInfo({{"port", m_port}}); np.set(QStringLiteral("numberOfFiles"), m_totalJobs); np.set(QStringLiteral("totalPayloadSize"), m_totalPayloadSize); if (Daemon::instance()->getDevice(m_deviceId)->sendPacket(np)) { m_server->resumeAccepting(); } else { setError(SendingNetworkPacketFailed); setErrorText(i18n("Failed to send packet to %1", Daemon::instance()->getDevice(m_deviceId)->name())); emitResult(); } } void CompositeUploadJob::newConnection() { m_server->pauseAccepting(); m_socket = m_server->nextPendingConnection(); if (!m_socket) { qCDebug(KDECONNECT_CORE) << "CompositeUploadJob::newConnection() - m_server->nextPendingConnection() returned a nullptr"; return; } m_currentJob->setSocket(m_socket); connect(m_socket, &QSslSocket::disconnected, this, &CompositeUploadJob::socketDisconnected); connect(m_socket, QOverload::of(&QAbstractSocket::error), this, &CompositeUploadJob::socketError); connect(m_socket, QOverload &>::of(&QSslSocket::sslErrors), this, &CompositeUploadJob::sslError); connect(m_socket, &QSslSocket::encrypted, this, &CompositeUploadJob::encrypted); LanLinkProvider::configureSslSocket(m_socket, m_deviceId, true); m_socket->startServerEncryption(); } void CompositeUploadJob::socketDisconnected() { m_socket->close(); } void CompositeUploadJob::socketError(QAbstractSocket::SocketError error) { Q_UNUSED(error); //Do not close the socket because when android closes the socket (share is cancelled) closing the socket leads to a cyclic socketError and eventually a segv setError(SocketError); emitResult(); m_running = false; } void CompositeUploadJob::sslError(const QList& errors) { Q_UNUSED(errors); m_socket->close(); setError(SslError); emitResult(); m_running = false; } void CompositeUploadJob::encrypted() { if (!m_timer.isValid()) { m_timer.start(); } m_currentJob->start(); } bool CompositeUploadJob::addSubjob(KJob* job) { if (UploadJob *uploadJob = qobject_cast(job)) { NetworkPacket np = uploadJob->getNetworkPacket(); m_totalJobs++; if (np.payloadSize() >= 0 ) { m_totalPayloadSize += np.payloadSize(); setTotalAmount(Bytes, m_totalPayloadSize); } QString filename; QString filenameArg = QStringLiteral("filename"); if (m_currentJob) { filename = m_currentJob->getNetworkPacket().get(filenameArg); } else { filename = np.get(filenameArg); } emitDescription(filename); - - if (m_running && !m_updatePacketPending) { + + if (m_running && m_currentJob && !m_updatePacketPending) { m_updatePacketPending = true; QMetaObject::invokeMethod(this, "sendUpdatePacket", Qt::QueuedConnection); } return KCompositeJob::addSubjob(job); } else { qCDebug(KDECONNECT_CORE) << "CompositeUploadJob::addSubjob() - you can only add UploadJob's, ignoring"; return false; } } void CompositeUploadJob::sendUpdatePacket() { NetworkPacket np(PACKET_TYPE_SHARE_REQUEST_UPDATE); np.set(QStringLiteral("numberOfFiles"), m_totalJobs); np.set(QStringLiteral("totalPayloadSize"), m_totalPayloadSize); Daemon::instance()->getDevice(m_deviceId)->sendPacket(np); m_updatePacketPending = false; } bool CompositeUploadJob::doKill() { if (m_running) { m_running = false; return m_currentJob->stop(); } return true; } void CompositeUploadJob::slotProcessedAmount(KJob *job, KJob::Unit unit, qulonglong amount) { Q_UNUSED(job); m_currentJobSendPayloadSize = amount; quint64 uploaded = m_totalSendPayloadSize + m_currentJobSendPayloadSize; if (uploaded == m_totalPayloadSize || m_prevElapsedTime == 0 || m_timer.elapsed() - m_prevElapsedTime >= 100) { m_prevElapsedTime = m_timer.elapsed(); setProcessedAmount(unit, uploaded); const auto elapsed = m_timer.elapsed(); if (elapsed > 0) { emitSpeed((1000 * uploaded) / elapsed); } } } void CompositeUploadJob::slotResult(KJob *job) { //Copies job error and errorText and emits result if job is in error otherwise removes job from subjob list KCompositeJob::slotResult(job); if (error() || !m_running) { return; } m_totalSendPayloadSize += m_currentJobSendPayloadSize; if (hasSubjobs()) { m_currentJobNum++; startNextSubJob(); } else { Q_EMIT description(this, i18n("Finished sending to %1", Daemon::instance()->getDevice(this->m_deviceId)->name()), { QStringLiteral(""), i18np("Sent 1 file", "Sent %1 files", m_totalJobs) } ); emitResult(); } } void CompositeUploadJob::emitDescription(const QString& currentFileName) { QPair field2; if (m_totalJobs > 1) { field2.first = i18n("Progress"); field2.second = i18n("Sending file %1 of %2", m_currentJobNum, m_totalJobs); } Q_EMIT description(this, i18n("Sending to %1", Daemon::instance()->getDevice(this->m_deviceId)->name()), { i18n("File"), currentFileName }, field2 ); }