MockLinkFileServer.h 5.3 KB
Newer Older
1 2
/****************************************************************************
 *
Gus Grubba's avatar
Gus Grubba committed
3
 * (c) 2009-2020 QGROUNDCONTROL PROJECT <http://www.qgroundcontrol.org>
4 5 6 7 8 9
 *
 * QGroundControl is licensed according to the terms in the file
 * COPYING.md in the root of the source code directory.
 *
 ****************************************************************************/

10

11 12
/// @file
///     @author Don Gagne <don@thegagnes.com>
Don Gagne's avatar
Don Gagne committed
13

14
#pragma once
15 16 17

#include "FileManager.h"

Don Gagne's avatar
Don Gagne committed
18
#include <QStringList>
19

20 21 22 23
class MockLink;

/// Mock implementation of Mavlink FTP server.
class MockLinkFileServer : public QObject
24 25 26 27
{
    Q_OBJECT
    
public:
28
    MockLinkFileServer(uint8_t systemIdServer, uint8_t componentIdServer, MockLink* mockLink);
29
    
30 31
    /// @brief Sets the list of files returned by the List command. Prepend names with F or D
    /// to indicate (F)ile or (D)irectory.
Don Gagne's avatar
Don Gagne committed
32
    void setFileList(QStringList& fileList) { _fileList = fileList; }
33
    
34 35 36 37 38 39 40
    /// @brief By calling setErrorMode with one of these modes you can cause the server to simulate an error.
    typedef enum {
        errModeNone,                ///< No error, respond correctly
        errModeNoResponse,          ///< No response to any request, client should eventually time out with no Ack
        errModeNakResponse,         ///< Nak all requests
        errModeNoSecondResponse,    ///< No response to subsequent request to initial command
        errModeNakSecondResponse,   ///< Nak subsequent request to initial command
41
        errModeBadSequence          ///< Return response with bad sequence number
42 43 44 45 46 47 48 49 50 51 52 53 54
    } ErrorMode_t;
    
    /// @brief Sets the error mode for command responses. This allows you to simulate various server errors.
    void setErrorMode(ErrorMode_t errMode) { _errMode = errMode; };
    
    /// @brief Array of failure modes you can cycle through for testing. By looping through this array you can avoid
    /// hardcoding the specific error modes in your unit test. This way when new error modes are added your unit test
    /// code may not need to be modified.
    static const ErrorMode_t rgFailureModes[];
    
    /// @brief The number of ErrorModes in the rgFailureModes array.
    static const size_t cFailureModes;
    
55 56
    /// Called to handle an FTP message
    void handleFTPMessage(const mavlink_message_t& message);
57
    
58
    /// @brief Used to represent a single test case for download testing.
59
    struct FileTestCase {
60 61
        const char* filename;               ///< Filename to download
        uint8_t     length;                 ///< Length of file in bytes
62 63
		int			packetCount;			///< Number of packets required for data
        bool        exactFit;				///< true: last packet is exact fit, false: last packet is partially filled
64 65
    };
    
66
    /// @brief The numbers of test cases in the rgFileTestCases array.
67
    static const size_t cFileTestCases = 3;
68 69
    
    /// @brief The set of files supported by the mock server for testing purposes. Each one represents a different edge case for testing.
70 71
    static const FileTestCase rgFileTestCases[cFileTestCases];
    
72 73
    void enableRandromDrops(bool enable) { _randomDropsEnabled = enable; }

74
signals:
75
    /// You can connect to this signal to be notified when the server receives a Terminate command.
76
    void terminateCommandReceived(void);
77
    
78 79 80
    /// You can connect to this signal to be notified when the server receives a Reset command.
    void resetCommandReceived(void);
    
Don Gagne's avatar
Don Gagne committed
81
private:
82 83 84 85 86 87 88 89 90
	void _sendAck(uint8_t targetSystemId, uint8_t targetComponentId, uint16_t seqNumber, FileManager::Opcode reqOpcode);
    void _sendNak(uint8_t targetSystemId, uint8_t targetComponentId, FileManager::ErrorCode error, uint16_t seqNumber, FileManager::Opcode reqOpcode);
    void _sendResponse(uint8_t targetSystemId, uint8_t targetComponentId, FileManager::Request* request, uint16_t seqNumber);
    void _listCommand(uint8_t senderSystemId, uint8_t senderComponentId, FileManager::Request* request, uint16_t seqNumber);
    void _openCommand(uint8_t senderSystemId, uint8_t senderComponentId, FileManager::Request* request, uint16_t seqNumber);
    void _readCommand(uint8_t senderSystemId, uint8_t senderComponentId, FileManager::Request* request, uint16_t seqNumber);
	void _streamCommand(uint8_t senderSystemId, uint8_t senderComponentId, FileManager::Request* request, uint16_t seqNumber);
    void _terminateCommand(uint8_t senderSystemId, uint8_t senderComponentId, FileManager::Request* request, uint16_t seqNumber);
    void _resetCommand(uint8_t senderSystemId, uint8_t senderComponentId, uint16_t seqNumber);
91
    uint16_t _nextSeqNumber(uint16_t seqNumber);
92
    
93 94 95
    /// if request is a string, this ensures it's null-terminated
    static void ensureNullTemination(FileManager::Request* request);

96
    QStringList _fileList;  ///< List of files returned by List command
97
    
98
    static const uint8_t    _sessionId;
99 100
    uint8_t                 _readFileLength;    ///< Length of active file being read
    ErrorMode_t             _errMode;           ///< Currently set error mode, as specified by setErrorMode
101
    const uint8_t           _systemIdServer;    ///< System ID for server
102 103
    const uint8_t           _componentIdServer; ///< Component ID for server
    MockLink*               _mockLink;          ///< MockLink to communicate through
104 105 106 107 108 109

    bool _lastReplyValid;
    uint16_t _lastReplySequence;
    mavlink_message_t _lastReply;

    bool _randomDropsEnabled;
110 111
};