MockLinkFTP.h

Go to the documentation of this file.

#pragma once
 
#include <QtCore/QByteArray>
#include <QtCore/QFile>
#include <QtCore/QHash>
#include <QtCore/QObject>
#include <QtCore/QStringList>
 
#include "MAVLinkFTP.h"
 
class MockLink;
 
class MockLinkFTP : public QObject
{
    Q_OBJECT
 
public:
    MockLinkFTP(uint8_t systemIdServer, uint8_t componentIdServer, MockLink *mockLink);
    ~MockLinkFTP();
 
    void setFileList(const QStringList &fileList) { _fileList = fileList; }
 
    void mavlinkMessageReceived(const mavlink_message_t &message);
 
    void enableRandomDrops(bool enable) { _randomDropsEnabled = enable; }
 
    QStringList uploadedFiles() const { return _uploadedFiles.keys(); }
 
    QByteArray uploadedFileContents(const QString& remotePath) const { return _uploadedFiles.value(remotePath); }
 
    void clearUploadedFiles() { _uploadedFiles.clear(); }
 
    enum ErrorMode_t {
        errModeNone,                        
        errModeNoResponse,                  
        errModeNakResponse,                 
        errModeNoSecondResponse,            
        errModeNoSecondResponseAllowRetry,  
        errModeNakSecondResponse,           
        errModeBadSequence                  
    };
 
    void setErrorMode(ErrorMode_t errMode) { _errMode = errMode; };
 
    static constexpr const ErrorMode_t rgFailureModes[] = {
        errModeNoResponse,
        errModeNakResponse,
        errModeNoSecondResponse,
        errModeNakSecondResponse,
        errModeBadSequence,
    };
 
    static constexpr const size_t cFailureModes = std::size(MockLinkFTP::rgFailureModes);
 
    static constexpr const char *sizeFilenamePrefix = "mocklink-size-";
 
signals:
    void terminateCommandReceived();
 
    void resetCommandReceived();
 
private:
    void _sendAck(uint8_t targetSystemId, uint8_t targetComponentId, uint16_t seqNumber, MavlinkFTP::OpCode_t reqOpCode);
    void _sendNak(uint8_t targetSystemId, uint8_t targetComponentId, MavlinkFTP::ErrorCode_t error, uint16_t seqNumber, MavlinkFTP::OpCode_t reqOpCode);
    void _sendNakErrno(uint8_t targetSystemId, uint8_t targetComponentId, uint8_t nakErrno, uint16_t seqNumber, MavlinkFTP::OpCode_t reqOpCode);
    void _sendResponse(uint8_t targetSystemId, uint8_t targetComponentId, MavlinkFTP::Request *request, uint16_t seqNumber);
    void _listCommand(uint8_t senderSystemId, uint8_t senderComponentId, MavlinkFTP::Request *request, uint16_t seqNumber);
    void _openCommand(uint8_t senderSystemId, uint8_t senderComponentId, MavlinkFTP::Request *request, uint16_t seqNumber);
    void _createFileCommand(uint8_t senderSystemId, uint8_t senderComponentId, MavlinkFTP::Request *request, uint16_t seqNumber);
    void _openFileWOCommand(uint8_t senderSystemId, uint8_t senderComponentId, MavlinkFTP::Request *request, uint16_t seqNumber);
    void _readCommand(uint8_t senderSystemId, uint8_t senderComponentId, MavlinkFTP::Request *request, uint16_t seqNumber);
    void _burstReadCommand(uint8_t senderSystemId, uint8_t senderComponentId, MavlinkFTP::Request *request, uint16_t seqNumber);
    void _terminateCommand(uint8_t senderSystemId, uint8_t senderComponentId, MavlinkFTP::Request *request, uint16_t seqNumber);
    void _resetCommand(uint8_t senderSystemId, uint8_t senderComponentId, uint16_t seqNumber);
    void _writeCommand(uint8_t senderSystemId, uint8_t senderComponentId, MavlinkFTP::Request *request, uint16_t seqNumber);
    void _finalizeActiveUpload();
    uint16_t _nextSeqNumber(uint16_t seqNumber) const;
    static QString _createTestTempFile(int size);
    QString _generateParamPck(bool withDefaults);
 
    static void ensureNullTemination(MavlinkFTP::Request *request);
 
    const uint8_t _systemIdServer;              
    const uint8_t _componentIdServer;           
    MockLink *_mockLink;                        
 
    bool _lastReplyValid = false;
    bool _randomDropsEnabled = false;
    ErrorMode_t _errMode = errModeNone;         
    mavlink_message_t _lastReply{};
    QFile _currentFile;
    QString _paramPckTempFile;
    struct UploadSession {
        bool active = false;
        QString remotePath;
        QByteArray buffer;
 
        void reset() {
            active = false;
            remotePath.clear();
            buffer.clear();
        }
    };
    UploadSession _uploadSession;
    QHash<QString, QByteArray> _uploadedFiles;
    QStringList _fileList;                      
    uint16_t _lastReplySequence = 0;
 
    static constexpr uint8_t _sessionId = 1;    
};