| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268 |
- /*
- * This software is in the public domain, furnished "as is", without technical
- * support, and with no warranty, express or implied, as to its usefulness for
- * any purpose.
- *
- */
- #pragma once
- #include "account.h"
- #include "creds/abstractcredentials.h"
- #include "logger.h"
- #include "filesystem.h"
- #include "syncengine.h"
- #include "common/syncjournaldb.h"
- #include "common/syncjournalfilerecord.h"
- #include "common/vfs.h"
- #include "csync_exclude.h"
- #include <QDir>
- #include <QNetworkReply>
- #include <QMap>
- #include <QtTest>
- #include <memory>
- #include <cookiejar.h>
- /*
- * TODO: In theory we should use QVERIFY instead of Q_ASSERT for testing, but this
- * only works when directly called from a QTest :-(
- */
- static const QUrl sRootUrl("owncloud://somehost/owncloud/remote.php/webdav/");
- static const QUrl sRootUrl2("owncloud://somehost/owncloud/remote.php/dav/files/admin/");
- static const QUrl sUploadUrl("owncloud://somehost/owncloud/remote.php/dav/uploads/admin/");
- inline QString getFilePathFromUrl(const QUrl &url) {
- QString path = url.path();
- if (path.startsWith(sRootUrl.path()))
- return path.mid(sRootUrl.path().length());
- if (path.startsWith(sRootUrl2.path()))
- return path.mid(sRootUrl2.path().length());
- if (path.startsWith(sUploadUrl.path()))
- return path.mid(sUploadUrl.path().length());
- return {};
- }
- inline QString generateEtag() {
- return QString::number(QDateTime::currentDateTimeUtc().toMSecsSinceEpoch(), 16) + QByteArray::number(qrand(), 16);
- }
- inline QByteArray generateFileId() {
- return QByteArray::number(qrand(), 16);
- }
- class PathComponents : public QStringList {
- public:
- PathComponents(const char *path) : PathComponents{QString::fromUtf8(path)} {}
- PathComponents(const QString &path) : QStringList{path.split('/', QString::SkipEmptyParts)} { }
- PathComponents(const QStringList &pathComponents) : QStringList{pathComponents} { }
- PathComponents parentDirComponents() const {
- return PathComponents{mid(0, size() - 1)};
- }
- PathComponents subComponents() const& { return PathComponents{mid(1)}; }
- PathComponents subComponents() && { removeFirst(); return std::move(*this); }
- QString pathRoot() const { return first(); }
- QString fileName() const { return last(); }
- };
- class FileModifier
- {
- public:
- virtual ~FileModifier() = default;
- virtual void remove(const QString &relativePath) = 0;
- virtual void insert(const QString &relativePath, qint64 size = 64, char contentChar = 'W') = 0;
- virtual void setContents(const QString &relativePath, char contentChar) = 0;
- virtual void appendByte(const QString &relativePath) = 0;
- virtual void mkdir(const QString &relativePath) = 0;
- virtual void rename(const QString &relativePath, const QString &relativeDestinationDirectory) = 0;
- virtual void setModTime(const QString &relativePath, const QDateTime &modTime) = 0;
- };
- class DiskFileModifier : public FileModifier
- {
- QDir _rootDir;
- public:
- DiskFileModifier(const QString &rootDirPath) : _rootDir(rootDirPath) { }
- void remove(const QString &relativePath) override {
- QFileInfo fi{_rootDir.filePath(relativePath)};
- if (fi.isFile())
- QVERIFY(_rootDir.remove(relativePath));
- else
- QVERIFY(QDir{fi.filePath()}.removeRecursively());
- }
- void insert(const QString &relativePath, qint64 size = 64, char contentChar = 'W') override {
- QFile file{_rootDir.filePath(relativePath)};
- QVERIFY(!file.exists());
- file.open(QFile::WriteOnly);
- QByteArray buf(1024, contentChar);
- for (int x = 0; x < size/buf.size(); ++x) {
- file.write(buf);
- }
- file.write(buf.data(), size % buf.size());
- file.close();
- // Set the mtime 30 seconds in the past, for some tests that need to make sure that the mtime differs.
- OCC::FileSystem::setModTime(file.fileName(), OCC::Utility::qDateTimeToTime_t(QDateTime::currentDateTimeUtc().addSecs(-30)));
- QCOMPARE(file.size(), size);
- }
- void setContents(const QString &relativePath, char contentChar) override {
- QFile file{_rootDir.filePath(relativePath)};
- QVERIFY(file.exists());
- qint64 size = file.size();
- file.open(QFile::WriteOnly);
- file.write(QByteArray{}.fill(contentChar, size));
- }
- void appendByte(const QString &relativePath) override {
- QFile file{_rootDir.filePath(relativePath)};
- QVERIFY(file.exists());
- file.open(QFile::ReadWrite);
- QByteArray contents = file.read(1);
- file.seek(file.size());
- file.write(contents);
- }
- void mkdir(const QString &relativePath) override {
- _rootDir.mkpath(relativePath);
- }
- void rename(const QString &from, const QString &to) override {
- QVERIFY(_rootDir.exists(from));
- QVERIFY(_rootDir.rename(from, to));
- }
- void setModTime(const QString &relativePath, const QDateTime &modTime) override {
- OCC::FileSystem::setModTime(_rootDir.filePath(relativePath), OCC::Utility::qDateTimeToTime_t(modTime));
- }
- };
- class FileInfo : public FileModifier
- {
- public:
- static FileInfo A12_B12_C12_S12() {
- FileInfo fi{QString{}, {
- {QStringLiteral("A"), {
- {QStringLiteral("a1"), 4},
- {QStringLiteral("a2"), 4}
- }},
- {QStringLiteral("B"), {
- {QStringLiteral("b1"), 16},
- {QStringLiteral("b2"), 16}
- }},
- {QStringLiteral("C"), {
- {QStringLiteral("c1"), 24},
- {QStringLiteral("c2"), 24}
- }},
- }};
- FileInfo sharedFolder{QStringLiteral("S"), {
- {QStringLiteral("s1"), 32},
- {QStringLiteral("s2"), 32}
- }};
- sharedFolder.isShared = true;
- sharedFolder.children[QStringLiteral("s1")].isShared = true;
- sharedFolder.children[QStringLiteral("s2")].isShared = true;
- fi.children.insert(sharedFolder.name, std::move(sharedFolder));
- return fi;
- }
- FileInfo() = default;
- FileInfo(const QString &name) : name{name} { }
- FileInfo(const QString &name, qint64 size) : name{name}, isDir{false}, size{size} { }
- FileInfo(const QString &name, qint64 size, char contentChar) : name{name}, isDir{false}, size{size}, contentChar{contentChar} { }
- FileInfo(const QString &name, const std::initializer_list<FileInfo> &children) : name{name} {
- for (const auto &source : children)
- addChild(source);
- }
- void addChild(const FileInfo &info)
- {
- auto &dest = this->children[info.name] = info;
- dest.parentPath = path();
- dest.fixupParentPathRecursively();
- }
- void remove(const QString &relativePath) override {
- const PathComponents pathComponents{relativePath};
- FileInfo *parent = findInvalidatingEtags(pathComponents.parentDirComponents());
- Q_ASSERT(parent);
- parent->children.erase(std::find_if(parent->children.begin(), parent->children.end(),
- [&pathComponents](const FileInfo &fi){ return fi.name == pathComponents.fileName(); }));
- }
- void insert(const QString &relativePath, qint64 size = 64, char contentChar = 'W') override {
- create(relativePath, size, contentChar);
- }
- void setContents(const QString &relativePath, char contentChar) override {
- FileInfo *file = findInvalidatingEtags(relativePath);
- Q_ASSERT(file);
- file->contentChar = contentChar;
- }
- void appendByte(const QString &relativePath) override {
- FileInfo *file = findInvalidatingEtags(relativePath);
- Q_ASSERT(file);
- file->size += 1;
- }
- void mkdir(const QString &relativePath) override {
- createDir(relativePath);
- }
- void rename(const QString &oldPath, const QString &newPath) override {
- const PathComponents newPathComponents{newPath};
- FileInfo *dir = findInvalidatingEtags(newPathComponents.parentDirComponents());
- Q_ASSERT(dir);
- Q_ASSERT(dir->isDir);
- const PathComponents pathComponents{oldPath};
- FileInfo *parent = findInvalidatingEtags(pathComponents.parentDirComponents());
- Q_ASSERT(parent);
- FileInfo fi = parent->children.take(pathComponents.fileName());
- fi.parentPath = dir->path();
- fi.name = newPathComponents.fileName();
- fi.fixupParentPathRecursively();
- dir->children.insert(newPathComponents.fileName(), std::move(fi));
- }
- void setModTime(const QString &relativePath, const QDateTime &modTime) override {
- FileInfo *file = findInvalidatingEtags(relativePath);
- Q_ASSERT(file);
- file->lastModified = modTime;
- }
- FileInfo *find(PathComponents pathComponents, const bool invalidateEtags = false) {
- if (pathComponents.isEmpty()) {
- if (invalidateEtags) {
- etag = generateEtag();
- }
- return this;
- }
- QString childName = pathComponents.pathRoot();
- auto it = children.find(childName);
- if (it != children.end()) {
- auto file = it->find(std::move(pathComponents).subComponents(), invalidateEtags);
- if (file && invalidateEtags) {
- // Update parents on the way back
- etag = generateEtag();
- }
- return file;
- }
- return nullptr;
- }
- FileInfo *createDir(const QString &relativePath) {
- const PathComponents pathComponents{relativePath};
- FileInfo *parent = findInvalidatingEtags(pathComponents.parentDirComponents());
- Q_ASSERT(parent);
- FileInfo &child = parent->children[pathComponents.fileName()] = FileInfo{pathComponents.fileName()};
- child.parentPath = parent->path();
- child.etag = generateEtag();
- return &child;
- }
- FileInfo *create(const QString &relativePath, qint64 size, char contentChar) {
- const PathComponents pathComponents{relativePath};
- FileInfo *parent = findInvalidatingEtags(pathComponents.parentDirComponents());
- Q_ASSERT(parent);
- FileInfo &child = parent->children[pathComponents.fileName()] = FileInfo{pathComponents.fileName(), size};
- child.parentPath = parent->path();
- child.contentChar = contentChar;
- child.etag = generateEtag();
- return &child;
- }
- bool operator<(const FileInfo &other) const {
- return name < other.name;
- }
- bool operator==(const FileInfo &other) const {
- // Consider files to be equal between local<->remote as a user would.
- return name == other.name
- && isDir == other.isDir
- && size == other.size
- && contentChar == other.contentChar
- && children == other.children;
- }
- bool operator!=(const FileInfo &other) const {
- return !operator==(other);
- }
- QString path() const {
- return (parentPath.isEmpty() ? QString() : (parentPath + '/')) + name;
- }
- void fixupParentPathRecursively() {
- auto p = path();
- for (auto it = children.begin(); it != children.end(); ++it) {
- Q_ASSERT(it.key() == it->name);
- it->parentPath = p;
- it->fixupParentPathRecursively();
- }
- }
- QString name;
- bool isDir = true;
- bool isShared = false;
- OCC::RemotePermissions permissions; // When uset, defaults to everything
- QDateTime lastModified = QDateTime::currentDateTimeUtc().addDays(-7);
- QString etag = generateEtag();
- QByteArray fileId = generateFileId();
- QByteArray checksums;
- QByteArray extraDavProperties;
- qint64 size = 0;
- char contentChar = 'W';
- // Sorted by name to be able to compare trees
- QMap<QString, FileInfo> children;
- QString parentPath;
- FileInfo *findInvalidatingEtags(PathComponents pathComponents) {
- return find(std::move(pathComponents), true);
- }
- friend inline QDebug operator<<(QDebug dbg, const FileInfo& fi) {
- return dbg << "{ " << fi.path() << ": " << fi.children;
- }
- };
- class FakePropfindReply : public QNetworkReply
- {
- Q_OBJECT
- public:
- QByteArray payload;
- FakePropfindReply(FileInfo &remoteRootFileInfo, QNetworkAccessManager::Operation op, const QNetworkRequest &request, QObject *parent)
- : QNetworkReply{parent} {
- setRequest(request);
- setUrl(request.url());
- setOperation(op);
- open(QIODevice::ReadOnly);
- QString fileName = getFilePathFromUrl(request.url());
- Q_ASSERT(!fileName.isNull()); // for root, it should be empty
- const FileInfo *fileInfo = remoteRootFileInfo.find(fileName);
- if (!fileInfo) {
- QMetaObject::invokeMethod(this, "respond404", Qt::QueuedConnection);
- return;
- }
- QString prefix = request.url().path().left(request.url().path().size() - fileName.size());
- // Don't care about the request and just return a full propfind
- const QString davUri{QStringLiteral("DAV:")};
- const QString ocUri{QStringLiteral("http://owncloud.org/ns")};
- QBuffer buffer{&payload};
- buffer.open(QIODevice::WriteOnly);
- QXmlStreamWriter xml( &buffer );
- xml.writeNamespace(davUri, "d");
- xml.writeNamespace(ocUri, "oc");
- xml.writeStartDocument();
- xml.writeStartElement(davUri, QStringLiteral("multistatus"));
- auto writeFileResponse = [&](const FileInfo &fileInfo) {
- xml.writeStartElement(davUri, QStringLiteral("response"));
- xml.writeTextElement(davUri, QStringLiteral("href"), prefix + QUrl::toPercentEncoding(fileInfo.path(), "/"));
- xml.writeStartElement(davUri, QStringLiteral("propstat"));
- xml.writeStartElement(davUri, QStringLiteral("prop"));
- if (fileInfo.isDir) {
- xml.writeStartElement(davUri, QStringLiteral("resourcetype"));
- xml.writeEmptyElement(davUri, QStringLiteral("collection"));
- xml.writeEndElement(); // resourcetype
- } else
- xml.writeEmptyElement(davUri, QStringLiteral("resourcetype"));
- auto gmtDate = fileInfo.lastModified.toUTC();
- auto stringDate = QLocale::c().toString(gmtDate, "ddd, dd MMM yyyy HH:mm:ss 'GMT'");
- xml.writeTextElement(davUri, QStringLiteral("getlastmodified"), stringDate);
- xml.writeTextElement(davUri, QStringLiteral("getcontentlength"), QString::number(fileInfo.size));
- xml.writeTextElement(davUri, QStringLiteral("getetag"), QStringLiteral("\"%1\"").arg(fileInfo.etag));
- xml.writeTextElement(ocUri, QStringLiteral("permissions"), !fileInfo.permissions.isNull()
- ? QString(fileInfo.permissions.toString())
- : fileInfo.isShared ? QStringLiteral("SRDNVCKW") : QStringLiteral("RDNVCKW"));
- xml.writeTextElement(ocUri, QStringLiteral("id"), fileInfo.fileId);
- xml.writeTextElement(ocUri, QStringLiteral("checksums"), fileInfo.checksums);
- buffer.write(fileInfo.extraDavProperties);
- xml.writeEndElement(); // prop
- xml.writeTextElement(davUri, QStringLiteral("status"), "HTTP/1.1 200 OK");
- xml.writeEndElement(); // propstat
- xml.writeEndElement(); // response
- };
- writeFileResponse(*fileInfo);
- foreach(const FileInfo &childFileInfo, fileInfo->children)
- writeFileResponse(childFileInfo);
- xml.writeEndElement(); // multistatus
- xml.writeEndDocument();
- QMetaObject::invokeMethod(this, "respond", Qt::QueuedConnection);
- }
- Q_INVOKABLE void respond() {
- setHeader(QNetworkRequest::ContentLengthHeader, payload.size());
- setHeader(QNetworkRequest::ContentTypeHeader, "application/xml; charset=utf-8");
- setAttribute(QNetworkRequest::HttpStatusCodeAttribute, 207);
- setFinished(true);
- emit metaDataChanged();
- if (bytesAvailable())
- emit readyRead();
- emit finished();
- }
- Q_INVOKABLE void respond404() {
- setAttribute(QNetworkRequest::HttpStatusCodeAttribute, 404);
- setError(InternalServerError, "Not Found");
- emit metaDataChanged();
- emit finished();
- }
- void abort() override { }
- qint64 bytesAvailable() const override { return payload.size() + QIODevice::bytesAvailable(); }
- qint64 readData(char *data, qint64 maxlen) override {
- qint64 len = std::min(qint64{payload.size()}, maxlen);
- strncpy(data, payload.constData(), len);
- payload.remove(0, len);
- return len;
- }
- };
- class FakePutReply : public QNetworkReply
- {
- Q_OBJECT
- FileInfo *fileInfo;
- public:
- FakePutReply(FileInfo &remoteRootFileInfo, QNetworkAccessManager::Operation op, const QNetworkRequest &request, const QByteArray &putPayload, QObject *parent)
- : QNetworkReply{parent} {
- setRequest(request);
- setUrl(request.url());
- setOperation(op);
- open(QIODevice::ReadOnly);
- fileInfo = perform(remoteRootFileInfo, request, putPayload);
- QMetaObject::invokeMethod(this, "respond", Qt::QueuedConnection);
- }
- static FileInfo *perform(FileInfo &remoteRootFileInfo, const QNetworkRequest &request, const QByteArray &putPayload)
- {
- QString fileName = getFilePathFromUrl(request.url());
- Q_ASSERT(!fileName.isEmpty());
- FileInfo *fileInfo = remoteRootFileInfo.find(fileName);
- if (fileInfo) {
- fileInfo->size = putPayload.size();
- fileInfo->contentChar = putPayload.at(0);
- } else {
- // Assume that the file is filled with the same character
- fileInfo = remoteRootFileInfo.create(fileName, putPayload.size(), putPayload.at(0));
- }
- fileInfo->lastModified = OCC::Utility::qDateTimeFromTime_t(request.rawHeader("X-OC-Mtime").toLongLong());
- remoteRootFileInfo.find(fileName, /*invalidateEtags=*/true);
- return fileInfo;
- }
- Q_INVOKABLE virtual void respond()
- {
- emit uploadProgress(fileInfo->size, fileInfo->size);
- setRawHeader("OC-ETag", fileInfo->etag.toLatin1());
- setRawHeader("ETag", fileInfo->etag.toLatin1());
- setRawHeader("OC-FileID", fileInfo->fileId);
- setRawHeader("X-OC-MTime", "accepted"); // Prevents Q_ASSERT(!_runningNow) since we'll call PropagateItemJob::done twice in that case.
- setAttribute(QNetworkRequest::HttpStatusCodeAttribute, 200);
- emit metaDataChanged();
- emit finished();
- }
- void abort() override
- {
- setError(OperationCanceledError, "abort");
- emit finished();
- }
- qint64 readData(char *, qint64) override { return 0; }
- };
- class FakeMkcolReply : public QNetworkReply
- {
- Q_OBJECT
- FileInfo *fileInfo;
- public:
- FakeMkcolReply(FileInfo &remoteRootFileInfo, QNetworkAccessManager::Operation op, const QNetworkRequest &request, QObject *parent)
- : QNetworkReply{parent} {
- setRequest(request);
- setUrl(request.url());
- setOperation(op);
- open(QIODevice::ReadOnly);
- QString fileName = getFilePathFromUrl(request.url());
- Q_ASSERT(!fileName.isEmpty());
- fileInfo = remoteRootFileInfo.createDir(fileName);
- if (!fileInfo) {
- abort();
- return;
- }
- QMetaObject::invokeMethod(this, "respond", Qt::QueuedConnection);
- }
- Q_INVOKABLE void respond() {
- setRawHeader("OC-FileId", fileInfo->fileId);
- setAttribute(QNetworkRequest::HttpStatusCodeAttribute, 201);
- emit metaDataChanged();
- emit finished();
- }
- void abort() override { }
- qint64 readData(char *, qint64) override { return 0; }
- };
- class FakeDeleteReply : public QNetworkReply
- {
- Q_OBJECT
- public:
- FakeDeleteReply(FileInfo &remoteRootFileInfo, QNetworkAccessManager::Operation op, const QNetworkRequest &request, QObject *parent)
- : QNetworkReply{parent} {
- setRequest(request);
- setUrl(request.url());
- setOperation(op);
- open(QIODevice::ReadOnly);
- QString fileName = getFilePathFromUrl(request.url());
- Q_ASSERT(!fileName.isEmpty());
- remoteRootFileInfo.remove(fileName);
- QMetaObject::invokeMethod(this, "respond", Qt::QueuedConnection);
- }
- Q_INVOKABLE void respond() {
- setAttribute(QNetworkRequest::HttpStatusCodeAttribute, 204);
- emit metaDataChanged();
- emit finished();
- }
- void abort() override { }
- qint64 readData(char *, qint64) override { return 0; }
- };
- class FakeMoveReply : public QNetworkReply
- {
- Q_OBJECT
- public:
- FakeMoveReply(FileInfo &remoteRootFileInfo, QNetworkAccessManager::Operation op, const QNetworkRequest &request, QObject *parent)
- : QNetworkReply{parent} {
- setRequest(request);
- setUrl(request.url());
- setOperation(op);
- open(QIODevice::ReadOnly);
- QString fileName = getFilePathFromUrl(request.url());
- Q_ASSERT(!fileName.isEmpty());
- QString dest = getFilePathFromUrl(QUrl::fromEncoded(request.rawHeader("Destination")));
- Q_ASSERT(!dest.isEmpty());
- remoteRootFileInfo.rename(fileName, dest);
- QMetaObject::invokeMethod(this, "respond", Qt::QueuedConnection);
- }
- Q_INVOKABLE void respond() {
- setAttribute(QNetworkRequest::HttpStatusCodeAttribute, 201);
- emit metaDataChanged();
- emit finished();
- }
- void abort() override { }
- qint64 readData(char *, qint64) override { return 0; }
- };
- class FakeGetReply : public QNetworkReply
- {
- Q_OBJECT
- public:
- const FileInfo *fileInfo;
- char payload;
- int size;
- bool aborted = false;
- FakeGetReply(FileInfo &remoteRootFileInfo, QNetworkAccessManager::Operation op, const QNetworkRequest &request, QObject *parent)
- : QNetworkReply{parent} {
- setRequest(request);
- setUrl(request.url());
- setOperation(op);
- open(QIODevice::ReadOnly);
- QString fileName = getFilePathFromUrl(request.url());
- Q_ASSERT(!fileName.isEmpty());
- fileInfo = remoteRootFileInfo.find(fileName);
- if (!fileInfo)
- qWarning() << "Could not find file" << fileName << "on the remote";
- QMetaObject::invokeMethod(this, "respond", Qt::QueuedConnection);
- }
- Q_INVOKABLE void respond() {
- if (aborted) {
- setError(OperationCanceledError, "Operation Canceled");
- emit metaDataChanged();
- emit finished();
- return;
- }
- payload = fileInfo->contentChar;
- size = fileInfo->size;
- setHeader(QNetworkRequest::ContentLengthHeader, size);
- setAttribute(QNetworkRequest::HttpStatusCodeAttribute, 200);
- setRawHeader("OC-ETag", fileInfo->etag.toLatin1());
- setRawHeader("ETag", fileInfo->etag.toLatin1());
- setRawHeader("OC-FileId", fileInfo->fileId);
- emit metaDataChanged();
- if (bytesAvailable())
- emit readyRead();
- emit finished();
- }
- void abort() override {
- aborted = true;
- }
- qint64 bytesAvailable() const override {
- if (aborted)
- return 0;
- return size + QIODevice::bytesAvailable();
- }
- qint64 readData(char *data, qint64 maxlen) override {
- qint64 len = std::min(qint64{size}, maxlen);
- std::fill_n(data, len, payload);
- size -= len;
- return len;
- }
- // useful to be public for testing
- using QNetworkReply::setRawHeader;
- };
- class FakeChunkMoveReply : public QNetworkReply
- {
- Q_OBJECT
- FileInfo *fileInfo;
- public:
- FakeChunkMoveReply(FileInfo &uploadsFileInfo, FileInfo &remoteRootFileInfo,
- QNetworkAccessManager::Operation op, const QNetworkRequest &request,
- QObject *parent)
- : QNetworkReply{ parent }
- {
- setRequest(request);
- setUrl(request.url());
- setOperation(op);
- open(QIODevice::ReadOnly);
- fileInfo = perform(uploadsFileInfo, remoteRootFileInfo, request);
- if (!fileInfo) {
- QTimer::singleShot(0, this, &FakeChunkMoveReply::respondPreconditionFailed);
- } else {
- QTimer::singleShot(0, this, &FakeChunkMoveReply::respond);
- }
- }
- static FileInfo *perform(FileInfo &uploadsFileInfo, FileInfo &remoteRootFileInfo, const QNetworkRequest &request)
- {
- QString source = getFilePathFromUrl(request.url());
- Q_ASSERT(!source.isEmpty());
- Q_ASSERT(source.endsWith("/.file"));
- source = source.left(source.length() - static_cast<int>(qstrlen("/.file")));
- auto sourceFolder = uploadsFileInfo.find(source);
- Q_ASSERT(sourceFolder);
- Q_ASSERT(sourceFolder->isDir);
- qint64 count = 0;
- qint64 size = 0;
- char payload = '\0';
- do {
- QString chunkName = QString::number(count).rightJustified(16, '0');
- if (!sourceFolder->children.contains(chunkName))
- break;
- auto &x = sourceFolder->children[chunkName];
- Q_ASSERT(!x.isDir);
- Q_ASSERT(x.size > 0); // There should not be empty chunks
- size += x.size;
- Q_ASSERT(!payload || payload == x.contentChar);
- payload = x.contentChar;
- ++count;
- } while(true);
- Q_ASSERT(count > 1); // There should be at least two chunks, otherwise why would we use chunking?
- Q_ASSERT(sourceFolder->children.count() == count); // There should not be holes or extra files
- QString fileName = getFilePathFromUrl(QUrl::fromEncoded(request.rawHeader("Destination")));
- Q_ASSERT(!fileName.isEmpty());
- FileInfo *fileInfo = remoteRootFileInfo.find(fileName);
- if (fileInfo) {
- // The client should put this header
- Q_ASSERT(request.hasRawHeader("If"));
- // And it should condition on the destination file
- auto start = QByteArray("<" + request.rawHeader("Destination") + ">");
- Q_ASSERT(request.rawHeader("If").startsWith(start));
- if (request.rawHeader("If") != start + " ([\"" + fileInfo->etag.toLatin1() + "\"])") {
- return nullptr;
- }
- fileInfo->size = size;
- fileInfo->contentChar = payload;
- } else {
- Q_ASSERT(!request.hasRawHeader("If"));
- // Assume that the file is filled with the same character
- fileInfo = remoteRootFileInfo.create(fileName, size, payload);
- }
- fileInfo->lastModified = OCC::Utility::qDateTimeFromTime_t(request.rawHeader("X-OC-Mtime").toLongLong());
- remoteRootFileInfo.find(fileName, /*invalidateEtags=*/true);
- return fileInfo;
- }
- Q_INVOKABLE virtual void respond()
- {
- setAttribute(QNetworkRequest::HttpStatusCodeAttribute, 201);
- setRawHeader("OC-ETag", fileInfo->etag.toLatin1());
- setRawHeader("ETag", fileInfo->etag.toLatin1());
- setRawHeader("OC-FileId", fileInfo->fileId);
- emit metaDataChanged();
- emit finished();
- }
- Q_INVOKABLE void respondPreconditionFailed() {
- setAttribute(QNetworkRequest::HttpStatusCodeAttribute, 412);
- setError(InternalServerError, "Precondition Failed");
- emit metaDataChanged();
- emit finished();
- }
- void abort() override
- {
- setError(OperationCanceledError, "abort");
- emit finished();
- }
- qint64 readData(char *, qint64) override { return 0; }
- };
- class FakePayloadReply : public QNetworkReply
- {
- Q_OBJECT
- public:
- FakePayloadReply(QNetworkAccessManager::Operation op, const QNetworkRequest &request,
- const QByteArray &body, QObject *parent)
- : QNetworkReply{ parent }
- , _body(body)
- {
- setRequest(request);
- setUrl(request.url());
- setOperation(op);
- open(QIODevice::ReadOnly);
- QTimer::singleShot(10, this, &FakePayloadReply::respond);
- }
- void respond()
- {
- setAttribute(QNetworkRequest::HttpStatusCodeAttribute, 200);
- setHeader(QNetworkRequest::ContentLengthHeader, _body.size());
- emit metaDataChanged();
- emit readyRead();
- setFinished(true);
- emit finished();
- }
- void abort() override {}
- qint64 readData(char *buf, qint64 max) override
- {
- max = qMin<qint64>(max, _body.size());
- memcpy(buf, _body.constData(), max);
- _body = _body.mid(max);
- return max;
- }
- qint64 bytesAvailable() const override
- {
- return _body.size();
- }
- QByteArray _body;
- };
- class FakeErrorReply : public QNetworkReply
- {
- Q_OBJECT
- public:
- FakeErrorReply(QNetworkAccessManager::Operation op, const QNetworkRequest &request,
- QObject *parent, int httpErrorCode, const QByteArray &body = QByteArray())
- : QNetworkReply{parent}, _body(body) {
- setRequest(request);
- setUrl(request.url());
- setOperation(op);
- open(QIODevice::ReadOnly);
- setAttribute(QNetworkRequest::HttpStatusCodeAttribute, httpErrorCode);
- setError(InternalServerError, "Internal Server Fake Error");
- QMetaObject::invokeMethod(this, "respond", Qt::QueuedConnection);
- }
- Q_INVOKABLE virtual void respond() {
- emit metaDataChanged();
- emit readyRead();
- // finishing can come strictly after readyRead was called
- QTimer::singleShot(5, this, &FakeErrorReply::slotSetFinished);
- }
- // make public to give tests easy interface
- using QNetworkReply::setError;
- using QNetworkReply::setAttribute;
- public slots:
- void slotSetFinished() {
- setFinished(true);
- emit finished();
- }
- public:
- void abort() override { }
- qint64 readData(char *buf, qint64 max) override {
- max = qMin<qint64>(max, _body.size());
- memcpy(buf, _body.constData(), max);
- _body = _body.mid(max);
- return max;
- }
- qint64 bytesAvailable() const override {
- return _body.size();
- }
- QByteArray _body;
- };
- // A reply that never responds
- class FakeHangingReply : public QNetworkReply
- {
- Q_OBJECT
- public:
- FakeHangingReply(QNetworkAccessManager::Operation op, const QNetworkRequest &request, QObject *parent)
- : QNetworkReply(parent)
- {
- setRequest(request);
- setUrl(request.url());
- setOperation(op);
- open(QIODevice::ReadOnly);
- }
- void abort() override {
- // Follow more or less the implementation of QNetworkReplyImpl::abort
- close();
- setError(OperationCanceledError, tr("Operation canceled"));
- emit error(OperationCanceledError);
- setFinished(true);
- emit finished();
- }
- qint64 readData(char *, qint64) override { return 0; }
- };
- // A delayed reply
- template <class OriginalReply>
- class DelayedReply : public OriginalReply
- {
- public:
- template <typename... Args>
- explicit DelayedReply(quint64 delayMS, Args &&... args)
- : OriginalReply(std::forward<Args>(args)...)
- , _delayMs(delayMS)
- {
- }
- quint64 _delayMs;
- void respond() override
- {
- QTimer::singleShot(_delayMs, static_cast<OriginalReply *>(this), [this] {
- // Explicit call to bases's respond();
- this->OriginalReply::respond();
- });
- }
- };
- class FakeQNAM : public QNetworkAccessManager
- {
- public:
- using Override = std::function<QNetworkReply *(Operation, const QNetworkRequest &, QIODevice *)>;
- private:
- FileInfo _remoteRootFileInfo;
- FileInfo _uploadFileInfo;
- // maps a path to an HTTP error
- QHash<QString, int> _errorPaths;
- // monitor requests and optionally provide custom replies
- Override _override;
- public:
- FakeQNAM(FileInfo initialRoot)
- : _remoteRootFileInfo{std::move(initialRoot)}
- {
- setCookieJar(new OCC::CookieJar);
- }
- FileInfo ¤tRemoteState() { return _remoteRootFileInfo; }
- FileInfo &uploadState() { return _uploadFileInfo; }
- QHash<QString, int> &errorPaths() { return _errorPaths; }
- void setOverride(const Override &override) { _override = override; }
- protected:
- QNetworkReply *createRequest(Operation op, const QNetworkRequest &request,
- QIODevice *outgoingData = nullptr) {
- if (_override) {
- if (auto reply = _override(op, request, outgoingData))
- return reply;
- }
- const QString fileName = getFilePathFromUrl(request.url());
- Q_ASSERT(!fileName.isNull());
- if (_errorPaths.contains(fileName))
- return new FakeErrorReply{op, request, this, _errorPaths[fileName]};
- bool isUpload = request.url().path().startsWith(sUploadUrl.path());
- FileInfo &info = isUpload ? _uploadFileInfo : _remoteRootFileInfo;
- auto verb = request.attribute(QNetworkRequest::CustomVerbAttribute);
- if (verb == "PROPFIND")
- // Ignore outgoingData always returning somethign good enough, works for now.
- return new FakePropfindReply{info, op, request, this};
- else if (verb == QLatin1String("GET") || op == QNetworkAccessManager::GetOperation)
- return new FakeGetReply{info, op, request, this};
- else if (verb == QLatin1String("PUT") || op == QNetworkAccessManager::PutOperation)
- return new FakePutReply{info, op, request, outgoingData->readAll(), this};
- else if (verb == QLatin1String("MKCOL"))
- return new FakeMkcolReply{info, op, request, this};
- else if (verb == QLatin1String("DELETE") || op == QNetworkAccessManager::DeleteOperation)
- return new FakeDeleteReply{info, op, request, this};
- else if (verb == QLatin1String("MOVE") && !isUpload)
- return new FakeMoveReply{info, op, request, this};
- else if (verb == QLatin1String("MOVE") && isUpload)
- return new FakeChunkMoveReply{ info, _remoteRootFileInfo, op, request, this };
- else {
- qDebug() << verb << outgoingData;
- Q_UNREACHABLE();
- }
- }
- };
- class FakeCredentials : public OCC::AbstractCredentials
- {
- QNetworkAccessManager *_qnam;
- public:
- FakeCredentials(QNetworkAccessManager *qnam) : _qnam{qnam} { }
- virtual QString authType() const { return "test"; }
- virtual QString user() const { return "admin"; }
- virtual QNetworkAccessManager *createQNAM() const { return _qnam; }
- virtual bool ready() const { return true; }
- virtual void fetchFromKeychain() { }
- virtual void askFromUser() { }
- virtual bool stillValid(QNetworkReply *) { return true; }
- virtual void persist() { }
- virtual void invalidateToken() { }
- virtual void forgetSensitiveData() { }
- };
- class FakeFolder
- {
- QTemporaryDir _tempDir;
- DiskFileModifier _localModifier;
- // FIXME: Clarify ownership, double delete
- FakeQNAM *_fakeQnam;
- OCC::AccountPtr _account;
- std::unique_ptr<OCC::SyncJournalDb> _journalDb;
- std::unique_ptr<OCC::SyncEngine> _syncEngine;
- public:
- FakeFolder(const FileInfo &fileTemplate)
- : _localModifier(_tempDir.path())
- {
- // Needs to be done once
- OCC::SyncEngine::minimumFileAgeForUpload = std::chrono::milliseconds(0);
- OCC::Logger::instance()->setLogFile("-");
- QDir rootDir{_tempDir.path()};
- qDebug() << "FakeFolder operating on" << rootDir;
- toDisk(rootDir, fileTemplate);
- _fakeQnam = new FakeQNAM(fileTemplate);
- _account = OCC::Account::create();
- _account->setUrl(QUrl(QStringLiteral("http://admin:admin@localhost/owncloud")));
- _account->setCredentials(new FakeCredentials{_fakeQnam});
- _account->setDavDisplayName("fakename");
- _account->setServerVersion("10.0.0");
- _journalDb = std::make_unique<OCC::SyncJournalDb>(localPath() + ".sync_test.db");
- _syncEngine = std::make_unique<OCC::SyncEngine>(_account, localPath(), "", _journalDb.get());
- // Ignore temporary files from the download. (This is in the default exclude list, but we don't load it)
- _syncEngine->excludedFiles().addManualExclude("]*.~*");
- // Ensure we have a valid VfsOff instance "running"
- switchToVfs(_syncEngine->syncOptions()._vfs);
- // A new folder will update the local file state database on first sync.
- // To have a state matching what users will encounter, we have to a sync
- // using an identical local/remote file tree first.
- ENFORCE(syncOnce());
- }
- void switchToVfs(QSharedPointer<OCC::Vfs> vfs)
- {
- auto opts = _syncEngine->syncOptions();
- opts._vfs->stop();
- QObject::disconnect(_syncEngine.get(), 0, opts._vfs.data(), 0);
- opts._vfs = vfs;
- _syncEngine->setSyncOptions(opts);
- OCC::VfsSetupParams vfsParams;
- vfsParams.filesystemPath = localPath();
- vfsParams.remotePath = "/";
- vfsParams.account = _account;
- vfsParams.journal = _journalDb.get();
- vfsParams.providerName = "OC-TEST";
- vfsParams.providerVersion = "0.1";
- vfsParams.enableShellIntegration = false;
- QObject::connect(_syncEngine.get(), &QObject::destroyed, vfs.data(), [vfs]() {
- vfs->stop();
- vfs->unregisterFolder();
- });
- vfs->start(vfsParams);
- }
- OCC::AccountPtr account() const { return _account; }
- OCC::SyncEngine &syncEngine() const { return *_syncEngine; }
- OCC::SyncJournalDb &syncJournal() const { return *_journalDb; }
- FileModifier &localModifier() { return _localModifier; }
- FileInfo &remoteModifier() { return _fakeQnam->currentRemoteState(); }
- FileInfo currentLocalState() {
- QDir rootDir{_tempDir.path()};
- FileInfo rootTemplate;
- fromDisk(rootDir, rootTemplate);
- rootTemplate.fixupParentPathRecursively();
- return rootTemplate;
- }
- FileInfo currentRemoteState() { return _fakeQnam->currentRemoteState(); }
- FileInfo &uploadState() { return _fakeQnam->uploadState(); }
- FileInfo dbState() const;
- struct ErrorList {
- FakeQNAM *_qnam;
- void append(const QString &path, int error = 500)
- { _qnam->errorPaths().insert(path, error); }
- void clear() { _qnam->errorPaths().clear(); }
- };
- ErrorList serverErrorPaths() { return {_fakeQnam}; }
- void setServerOverride(const FakeQNAM::Override &override) { _fakeQnam->setOverride(override); }
- QString localPath() const {
- // SyncEngine wants a trailing slash
- if (_tempDir.path().endsWith('/'))
- return _tempDir.path();
- return _tempDir.path() + '/';
- }
- void scheduleSync() {
- // Have to be done async, else, an error before exec() does not terminate the event loop.
- QMetaObject::invokeMethod(_syncEngine.get(), "startSync", Qt::QueuedConnection);
- }
- void execUntilBeforePropagation() {
- QSignalSpy spy(_syncEngine.get(), SIGNAL(aboutToPropagate(SyncFileItemVector&)));
- QVERIFY(spy.wait());
- }
- void execUntilItemCompleted(const QString &relativePath) {
- QSignalSpy spy(_syncEngine.get(), SIGNAL(itemCompleted(const SyncFileItemPtr &)));
- QElapsedTimer t;
- t.start();
- while (t.elapsed() < 5000) {
- spy.clear();
- QVERIFY(spy.wait());
- for(const QList<QVariant> &args : spy) {
- auto item = args[0].value<OCC::SyncFileItemPtr>();
- if (item->destination() == relativePath)
- return;
- }
- }
- QVERIFY(false);
- }
- bool execUntilFinished() {
- QSignalSpy spy(_syncEngine.get(), SIGNAL(finished(bool)));
- bool ok = spy.wait(3600000);
- Q_ASSERT(ok && "Sync timed out");
- return spy[0][0].toBool();
- }
- bool syncOnce() {
- scheduleSync();
- return execUntilFinished();
- }
- private:
- static void toDisk(QDir &dir, const FileInfo &templateFi) {
- foreach (const FileInfo &child, templateFi.children) {
- if (child.isDir) {
- QDir subDir(dir);
- dir.mkdir(child.name);
- subDir.cd(child.name);
- toDisk(subDir, child);
- } else {
- QFile file{dir.filePath(child.name)};
- file.open(QFile::WriteOnly);
- file.write(QByteArray{}.fill(child.contentChar, child.size));
- file.close();
- OCC::FileSystem::setModTime(file.fileName(), OCC::Utility::qDateTimeToTime_t(child.lastModified));
- }
- }
- }
- static void fromDisk(QDir &dir, FileInfo &templateFi) {
- foreach (const QFileInfo &diskChild, dir.entryInfoList(QDir::AllEntries | QDir::NoDotAndDotDot)) {
- if (diskChild.isDir()) {
- QDir subDir = dir;
- subDir.cd(diskChild.fileName());
- FileInfo &subFi = templateFi.children[diskChild.fileName()] = FileInfo{diskChild.fileName()};
- fromDisk(subDir, subFi);
- } else {
- QFile f{diskChild.filePath()};
- f.open(QFile::ReadOnly);
- auto content = f.read(1);
- if (content.size() == 0) {
- qWarning() << "Empty file at:" << diskChild.filePath();
- continue;
- }
- char contentChar = content.at(0);
- templateFi.children.insert(diskChild.fileName(), FileInfo{diskChild.fileName(), diskChild.size(), contentChar});
- }
- }
- }
- };
- static FileInfo &findOrCreateDirs(FileInfo &base, PathComponents components)
- {
- if (components.isEmpty())
- return base;
- auto childName = components.pathRoot();
- auto it = base.children.find(childName);
- if (it != base.children.end()) {
- return findOrCreateDirs(*it, components.subComponents());
- }
- auto &newDir = base.children[childName] = FileInfo{childName};
- newDir.parentPath = base.path();
- return findOrCreateDirs(newDir, components.subComponents());
- }
- inline FileInfo FakeFolder::dbState() const
- {
- FileInfo result;
- _journalDb->getFilesBelowPath("", [&](const OCC::SyncJournalFileRecord &record) {
- auto components = PathComponents(QString::fromUtf8(record._path));
- auto &parentDir = findOrCreateDirs(result, components.parentDirComponents());
- auto name = components.fileName();
- auto &item = parentDir.children[name];
- item.name = name;
- item.parentPath = parentDir.path();
- item.size = record._fileSize;
- item.isDir = record._type == ItemTypeDirectory;
- item.permissions = record._remotePerm;
- item.etag = record._etag;
- item.lastModified = OCC::Utility::qDateTimeFromTime_t(record._modtime);
- item.fileId = record._fileId;
- item.checksums = record._checksumHeader;
- // item.contentChar can't be set from the db
- });
- return result;
- }
- /* Return the FileInfo for a conflict file for the specified relative filename */
- inline const FileInfo *findConflict(FileInfo &dir, const QString &filename)
- {
- QFileInfo info(filename);
- const FileInfo *parentDir = dir.find(info.path());
- if (!parentDir)
- return nullptr;
- QString start = info.baseName() + " (conflicted copy";
- for (const auto &item : parentDir->children) {
- if (item.name.startsWith(start)) {
- return &item;
- }
- }
- return nullptr;
- }
- struct ItemCompletedSpy : QSignalSpy {
- explicit ItemCompletedSpy(FakeFolder &folder)
- : QSignalSpy(&folder.syncEngine(), &OCC::SyncEngine::itemCompleted)
- {}
- OCC::SyncFileItemPtr findItem(const QString &path) const
- {
- for (const QList<QVariant> &args : *this) {
- auto item = args[0].value<OCC::SyncFileItemPtr>();
- if (item->destination() == path)
- return item;
- }
- return OCC::SyncFileItemPtr::create();
- }
- };
- // QTest::toString overloads
- namespace OCC {
- inline char *toString(const SyncFileStatus &s) {
- return QTest::toString(QString("SyncFileStatus(" + s.toSocketAPIString() + ")"));
- }
- }
- inline void addFiles(QStringList &dest, const FileInfo &fi)
- {
- if (fi.isDir) {
- dest += QString("%1 - dir").arg(fi.path());
- foreach (const FileInfo &fi, fi.children)
- addFiles(dest, fi);
- } else {
- dest += QString("%1 - %2 %3-bytes").arg(fi.path()).arg(fi.size).arg(fi.contentChar);
- }
- }
- inline QString toStringNoElide(const FileInfo &fi)
- {
- QStringList files;
- foreach (const FileInfo &fi, fi.children)
- addFiles(files, fi);
- files.sort();
- return QString("FileInfo with %1 files(\n\t%2\n)").arg(files.size()).arg(files.join("\n\t"));
- }
- inline char *toString(const FileInfo &fi)
- {
- return QTest::toString(toStringNoElide(fi));
- }
- inline void addFilesDbData(QStringList &dest, const FileInfo &fi)
- {
- // could include etag, permissions etc, but would need extra work
- if (fi.isDir) {
- dest += QString("%1 - %2 %3 %4").arg(
- fi.name,
- fi.isDir ? "dir" : "file",
- QString::number(fi.lastModified.toSecsSinceEpoch()),
- fi.fileId);
- foreach (const FileInfo &fi, fi.children)
- addFilesDbData(dest, fi);
- } else {
- dest += QString("%1 - %2 %3 %4 %5").arg(
- fi.name,
- fi.isDir ? "dir" : "file",
- QString::number(fi.size),
- QString::number(fi.lastModified.toSecsSinceEpoch()),
- fi.fileId);
- }
- }
- inline char *printDbData(const FileInfo &fi)
- {
- QStringList files;
- foreach (const FileInfo &fi, fi.children)
- addFilesDbData(files, fi);
- return QTest::toString(QString("FileInfo with %1 files(%2)").arg(files.size()).arg(files.join(", ")));
- }
|