mirror of
https://github.com/nextcloud/desktop.git
synced 2024-10-27 23:17:13 +03:00
The GET jobs were redirected, but the custom incremental handling in readyRead didn't propagate to the follow-up job.
208 lines
6.9 KiB
C++
208 lines
6.9 KiB
C++
/*
|
|
* Copyright (C) by Olivier Goffart <ogoffart@owncloud.com>
|
|
*
|
|
* 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) 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 General Public License
|
|
* for more details.
|
|
*/
|
|
#pragma once
|
|
|
|
#include "owncloudpropagator.h"
|
|
#include "networkjobs.h"
|
|
|
|
#include <QBuffer>
|
|
#include <QFile>
|
|
|
|
namespace OCC {
|
|
|
|
/**
|
|
* @brief The GETFileJob class
|
|
* @ingroup libsync
|
|
*/
|
|
class GETFileJob : public AbstractNetworkJob
|
|
{
|
|
Q_OBJECT
|
|
QFile *_device;
|
|
QMap<QByteArray, QByteArray> _headers;
|
|
QString _errorString;
|
|
QByteArray _expectedEtagForResume;
|
|
quint64 _resumeStart;
|
|
SyncFileItem::Status _errorStatus;
|
|
QUrl _directDownloadUrl;
|
|
QByteArray _etag;
|
|
bool _bandwidthLimited; // if _bandwidthQuota will be used
|
|
bool _bandwidthChoked; // if download is paused (won't read on readyRead())
|
|
qint64 _bandwidthQuota;
|
|
QPointer<BandwidthManager> _bandwidthManager;
|
|
bool _hasEmittedFinishedSignal;
|
|
time_t _lastModified;
|
|
|
|
/// Will be set to true once we've seen a 2xx response header
|
|
bool _saveBodyToFile = false;
|
|
|
|
public:
|
|
// DOES NOT take ownership of the device.
|
|
explicit GETFileJob(AccountPtr account, const QString &path, QFile *device,
|
|
const QMap<QByteArray, QByteArray> &headers, const QByteArray &expectedEtagForResume,
|
|
quint64 resumeStart, QObject *parent = 0);
|
|
// For directDownloadUrl:
|
|
explicit GETFileJob(AccountPtr account, const QUrl &url, QFile *device,
|
|
const QMap<QByteArray, QByteArray> &headers, const QByteArray &expectedEtagForResume,
|
|
quint64 resumeStart, QObject *parent = 0);
|
|
virtual ~GETFileJob()
|
|
{
|
|
if (_bandwidthManager) {
|
|
_bandwidthManager->unregisterDownloadJob(this);
|
|
}
|
|
}
|
|
|
|
virtual void start() Q_DECL_OVERRIDE;
|
|
virtual bool finished() Q_DECL_OVERRIDE
|
|
{
|
|
if (reply()->bytesAvailable()) {
|
|
return false;
|
|
} else {
|
|
if (_bandwidthManager) {
|
|
_bandwidthManager->unregisterDownloadJob(this);
|
|
}
|
|
if (!_hasEmittedFinishedSignal) {
|
|
emit finishedSignal();
|
|
}
|
|
_hasEmittedFinishedSignal = true;
|
|
return true; // discard
|
|
}
|
|
}
|
|
|
|
void newReplyHook(QNetworkReply *reply) override;
|
|
|
|
void setBandwidthManager(BandwidthManager *bwm);
|
|
void setChoked(bool c);
|
|
void setBandwidthLimited(bool b);
|
|
void giveBandwidthQuota(qint64 q);
|
|
qint64 currentDownloadPosition();
|
|
|
|
QString errorString() const;
|
|
void setErrorString(const QString &s) { _errorString = s; }
|
|
|
|
SyncFileItem::Status errorStatus() { return _errorStatus; }
|
|
void setErrorStatus(const SyncFileItem::Status &s) { _errorStatus = s; }
|
|
|
|
void onTimedOut() Q_DECL_OVERRIDE;
|
|
|
|
QByteArray &etag() { return _etag; }
|
|
quint64 resumeStart() { return _resumeStart; }
|
|
time_t lastModified() { return _lastModified; }
|
|
|
|
|
|
signals:
|
|
void finishedSignal();
|
|
void downloadProgress(qint64, qint64);
|
|
private slots:
|
|
void slotReadyRead();
|
|
void slotMetaDataChanged();
|
|
};
|
|
|
|
/**
|
|
* @brief The PropagateDownloadFile class
|
|
* @ingroup libsync
|
|
*
|
|
* This is the flow:
|
|
|
|
\code{.unparsed}
|
|
start()
|
|
|
|
|
| deleteExistingFolder() if enabled
|
|
|
|
|
+--> mtime and size identical?
|
|
| then compute the local checksum
|
|
| done?-> conflictChecksumComputed()
|
|
| |
|
|
| checksum differs? |
|
|
+-> startDownload() <--------------------------+
|
|
| |
|
|
+-> run a GETFileJob | checksum identical?
|
|
|
|
|
done?-> slotGetFinished() |
|
|
| |
|
|
+-> validate checksum header |
|
|
|
|
|
done?-> transmissionChecksumValidated() |
|
|
| |
|
|
+-> compute the content checksum |
|
|
|
|
|
done?-> contentChecksumComputed() |
|
|
| |
|
|
+-> downloadFinished() |
|
|
| |
|
|
+------------------+ |
|
|
| |
|
|
+-> updateMetadata() <-------------------------+
|
|
|
|
\endcode
|
|
*/
|
|
class PropagateDownloadFile : public PropagateItemJob
|
|
{
|
|
Q_OBJECT
|
|
public:
|
|
PropagateDownloadFile(OwncloudPropagator *propagator, const SyncFileItemPtr &item)
|
|
: PropagateItemJob(propagator, item)
|
|
, _resumeStart(0)
|
|
, _downloadProgress(0)
|
|
, _deleteExisting(false)
|
|
{
|
|
}
|
|
void start() Q_DECL_OVERRIDE;
|
|
qint64 committedDiskSpace() const Q_DECL_OVERRIDE;
|
|
|
|
// We think it might finish quickly because it is a small file.
|
|
bool isLikelyFinishedQuickly() Q_DECL_OVERRIDE { return _item->_size < propagator()->smallFileSize(); }
|
|
|
|
/**
|
|
* Whether an existing folder with the same name may be deleted before
|
|
* the download.
|
|
*
|
|
* If it's a non-empty folder, it'll be renamed to a conflict-style name
|
|
* to preserve any non-synced content that may be inside.
|
|
*
|
|
* Default: false.
|
|
*/
|
|
void setDeleteExistingFolder(bool enabled);
|
|
|
|
private slots:
|
|
/// Called when ComputeChecksum on the local file finishes,
|
|
/// maybe the local and remote checksums are identical?
|
|
void conflictChecksumComputed(const QByteArray &checksumType, const QByteArray &checksum);
|
|
/// Called to start downloading the remote file
|
|
void startDownload();
|
|
/// Called when the GETFileJob finishes
|
|
void slotGetFinished();
|
|
/// Called when the download's checksum header was validated
|
|
void transmissionChecksumValidated(const QByteArray &checksumType, const QByteArray &checksum);
|
|
/// Called when the download's checksum computation is done
|
|
void contentChecksumComputed(const QByteArray &checksumType, const QByteArray &checksum);
|
|
void downloadFinished();
|
|
/// Called when it's time to update the db metadata
|
|
void updateMetadata(bool isConflict);
|
|
|
|
void abort(PropagatorJob::AbortType abortType) Q_DECL_OVERRIDE;
|
|
void slotDownloadProgress(qint64, qint64);
|
|
void slotChecksumFail(const QString &errMsg);
|
|
|
|
private:
|
|
void deleteExistingFolder();
|
|
|
|
quint64 _resumeStart;
|
|
qint64 _downloadProgress;
|
|
QPointer<GETFileJob> _job;
|
|
QFile _tmpFile;
|
|
bool _deleteExisting;
|
|
|
|
QElapsedTimer _stopwatch;
|
|
};
|
|
}
|