| /**************************************************************************** |
| ** |
| ** Copyright (C) 2016 The Qt Company Ltd. |
| ** Contact: https://www.qt.io/licensing/ |
| ** |
| ** This file is part of the examples of the Qt Toolkit. |
| ** |
| ** $QT_BEGIN_LICENSE:BSD$ |
| ** Commercial License Usage |
| ** Licensees holding valid commercial Qt licenses may use this file in |
| ** accordance with the commercial license agreement provided with the |
| ** Software or, alternatively, in accordance with the terms contained in |
| ** a written agreement between you and The Qt Company. For licensing terms |
| ** and conditions see https://www.qt.io/terms-conditions. For further |
| ** information use the contact form at https://www.qt.io/contact-us. |
| ** |
| ** BSD License Usage |
| ** Alternatively, you may use this file under the terms of the BSD license |
| ** as follows: |
| ** |
| ** "Redistribution and use in source and binary forms, with or without |
| ** modification, are permitted provided that the following conditions are |
| ** met: |
| ** * Redistributions of source code must retain the above copyright |
| ** notice, this list of conditions and the following disclaimer. |
| ** * Redistributions in binary form must reproduce the above copyright |
| ** notice, this list of conditions and the following disclaimer in |
| ** the documentation and/or other materials provided with the |
| ** distribution. |
| ** * Neither the name of The Qt Company Ltd nor the names of its |
| ** contributors may be used to endorse or promote products derived |
| ** from this software without specific prior written permission. |
| ** |
| ** |
| ** THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS |
| ** "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT |
| ** LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR |
| ** A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT |
| ** OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
| ** SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
| ** LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, |
| ** DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY |
| ** THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT |
| ** (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE |
| ** OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE." |
| ** |
| ** $QT_END_LICENSE$ |
| ** |
| ****************************************************************************/ |
| |
| #ifndef FILEMANAGER_H |
| #define FILEMANAGER_H |
| |
| #include <QBitArray> |
| #include <QList> |
| #include <QMutex> |
| #include <QThread> |
| #include <QWaitCondition> |
| |
| #include "metainfo.h" |
| |
| QT_BEGIN_NAMESPACE |
| class QByteArray; |
| class QFile; |
| class QTimerEvent; |
| QT_END_NAMESPACE |
| |
| class FileManager : public QThread |
| { |
| Q_OBJECT |
| |
| public: |
| FileManager(QObject *parent = nullptr); |
| virtual ~FileManager(); |
| |
| inline void setMetaInfo(const MetaInfo &info) { metaInfo = info; } |
| inline void setDestinationFolder(const QString &directory) { destinationPath = directory; } |
| |
| int read(int pieceIndex, int offset, int length); |
| void write(int pieceIndex, int offset, const QByteArray &data); |
| void verifyPiece(int pieceIndex); |
| inline qint64 totalSize() const { return totalLength; } |
| |
| inline int pieceCount() const { return numPieces; } |
| int pieceLengthAt(int pieceIndex) const; |
| |
| QBitArray completedPieces() const; |
| void setCompletedPieces(const QBitArray &pieces); |
| |
| QString errorString() const; |
| |
| public slots: |
| void startDataVerification(); |
| |
| signals: |
| void dataRead(int id, int pieceIndex, int offset, const QByteArray &data); |
| void error(); |
| void verificationProgress(int percent); |
| void verificationDone(); |
| void pieceVerified(int pieceIndex, bool verified); |
| |
| protected: |
| void run() override; |
| |
| private slots: |
| bool verifySinglePiece(int pieceIndex); |
| void wakeUp(); |
| |
| private: |
| bool generateFiles(); |
| QByteArray readBlock(int pieceIndex, int offset, int length); |
| bool writeBlock(int pieceIndex, int offset, const QByteArray &data); |
| void verifyFileContents(); |
| |
| struct WriteRequest { |
| int pieceIndex; |
| int offset; |
| QByteArray data; |
| }; |
| struct ReadRequest { |
| int pieceIndex; |
| int offset; |
| int length; |
| int id; |
| }; |
| |
| QString errString; |
| QString destinationPath; |
| MetaInfo metaInfo; |
| QList<QFile *> files; |
| QList<QByteArray> sha1s; |
| QBitArray verifiedPieces; |
| |
| bool newFile; |
| int pieceLength; |
| qint64 totalLength; |
| int numPieces; |
| int readId; |
| bool startVerification; |
| bool quit; |
| bool wokeUp; |
| |
| QList<WriteRequest> writeRequests; |
| QList<ReadRequest> readRequests; |
| QList<int> pendingVerificationRequests; |
| QList<int> newPendingVerificationRequests; |
| QList<qint64> fileSizes; |
| |
| mutable QMutex mutex; |
| mutable QWaitCondition cond; |
| }; |
| |
| #endif |