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


11
/*!
pixhawk's avatar
pixhawk committed
12 13 14 15 16 17 18 19 20 21 22 23 24 25
 * @file
 *   @brief UDP connection (server) for unmanned vehicles
 *   @author Lorenz Meier <mavteam@student.ethz.ch>
 *
 */

#ifndef UDPLINK_H
#define UDPLINK_H

#include <QString>
#include <QList>
#include <QMap>
#include <QMutex>
#include <QUdpSocket>
26 27 28
#include <QMutexLocker>
#include <QQueue>
#include <QByteArray>
29

30 31 32 33
#if defined(QGC_ZEROCONF_ENABLED)
#include <dns_sd.h>
#endif

34
#include "QGCConfig.h"
35 36
#include "LinkManager.h"

Gus Grubba's avatar
Gus Grubba committed
37 38 39 40 41 42 43 44 45 46 47 48 49 50
class UDPCLient {
public:
    UDPCLient(const QHostAddress& address_, quint16 port_)
        : address(address_)
        , port(port_)
    {}
    UDPCLient(const UDPCLient* other)
        : address(other->address)
        , port(other->port)
    {}
    QHostAddress    address;
    quint16         port;
};

51 52
class UDPConfiguration : public LinkConfiguration
{
Don Gagne's avatar
Don Gagne committed
53
    Q_OBJECT
54 55
public:

56 57 58
    Q_PROPERTY(quint16      localPort   READ localPort  WRITE setLocalPort  NOTIFY localPortChanged)
    Q_PROPERTY(QStringList  hostList    READ hostList                       NOTIFY  hostListChanged)

59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76
    /*!
     * @brief Regular constructor
     *
     * @param[in] name Configuration (user friendly) name
     */
    UDPConfiguration(const QString& name);

    /*!
     * @brief Copy contructor
     *
     * When manipulating data, you create a copy of the configuration, edit it
     * and then transfer its content to the original (using copyFrom() below). Use this
     * contructor to create an editable copy.
     *
     * @param[in] source Original configuration
     */
    UDPConfiguration(UDPConfiguration* source);

Gus Grubba's avatar
Gus Grubba committed
77
    ~UDPConfiguration();
78 79 80 81 82 83 84 85 86 87 88 89 90

    /*!
     * @brief The UDP port we bind to
     *
     * @return Port number
     */
    quint16 localPort   () { return _localPort; }

    /*!
     * @brief Add a target host
     *
     * @param[in] host Host name in standard formatt, e.g. localhost:14551 or 192.168.1.1:14551
     */
91
    Q_INVOKABLE void addHost (const QString host);
92 93 94 95 96 97 98

    /*!
     * @brief Add a target host
     *
     * @param[in] host Host name, e.g. localhost or 192.168.1.1
     * @param[in] port Port number
     */
Gus Grubba's avatar
Gus Grubba committed
99
    void addHost        (const QString& host, quint16 port);
100 101 102 103 104 105

    /*!
     * @brief Remove a target host from the list
     *
     * @param[in] host Host name, e.g. localhost or 192.168.1.1
     */
106
    Q_INVOKABLE void removeHost  (const QString host);
107 108 109 110 111 112 113 114

    /*!
     * @brief Set the UDP port we bind to
     *
     * @param[in] port Port number
     */
    void setLocalPort   (quint16 port);

115 116 117 118 119
    /*!
     * @brief QML Interface
     */
    QStringList hostList    () { return _hostList; }

Gus Grubba's avatar
Gus Grubba committed
120 121
    const QList<UDPCLient*> targetHosts() { return _targetHosts; }

122
    /// From LinkConfiguration
123 124 125 126 127 128
    LinkType    type                 () { return LinkConfiguration::TypeUdp; }
    void        copyFrom             (LinkConfiguration* source);
    void        loadSettings         (QSettings& settings, const QString& root);
    void        saveSettings         (QSettings& settings, const QString& root);
    void        updateSettings       ();
    bool        isAutoConnectAllowed () { return true; }
129
    bool        isHighLatencyAllowed () { return true; }
130
    QString     settingsURL          () { return "UdpSettings.qml"; }
131

132 133 134 135 136 137
signals:
    void localPortChanged   ();
    void hostListChanged    ();

private:
    void _updateHostList    ();
Gus Grubba's avatar
Gus Grubba committed
138 139
    void _clearTargetHosts  ();
    void _copyFrom          (LinkConfiguration *source);
140

141
private:
Gus Grubba's avatar
Gus Grubba committed
142 143 144
    QList<UDPCLient*>   _targetHosts;
    QStringList         _hostList;      ///< Exposed to QML
    quint16             _localPort;
145
};
pixhawk's avatar
pixhawk committed
146 147 148 149

class UDPLink : public LinkInterface
{
    Q_OBJECT
150

151
    friend class UDPConfiguration;
152
    friend class LinkManager;
153

pixhawk's avatar
pixhawk committed
154
public:
Gus Grubba's avatar
Gus Grubba committed
155 156 157
    void    requestReset            () override { }
    bool    isConnected             () const override;
    QString getName                 () const override;
pixhawk's avatar
pixhawk committed
158

159
    // Extensive statistics for scientific purposes
Gus Grubba's avatar
Gus Grubba committed
160 161 162
    qint64  getConnectionSpeed      () const override;
    qint64  getCurrentInDataRate    () const;
    qint64  getCurrentOutDataRate   () const;
pixhawk's avatar
pixhawk committed
163

Gus Grubba's avatar
Gus Grubba committed
164 165
    // Thread
    void    run                     () override;
166

167 168
    // These are left unimplemented in order to cause linker errors which indicate incorrect usage of
    // connect/disconnect on link directly. All connect/disconnect calls should be made through LinkManager.
Gus Grubba's avatar
Gus Grubba committed
169 170
    bool    connect                 (void);
    bool    disconnect              (void);
pixhawk's avatar
pixhawk committed
171 172

public slots:
Gus Grubba's avatar
Gus Grubba committed
173
    void    readBytes               ();
174

175
private slots:
Gus Grubba's avatar
Gus Grubba committed
176
    void    _writeBytes             (const QByteArray data) override;
pixhawk's avatar
pixhawk committed
177

oberion's avatar
oberion committed
178
private:
179
    // Links are only created/destroyed by LinkManager so constructor/destructor is not public
180
    UDPLink(SharedLinkConfigurationPointer& config);
181
    ~UDPLink();
182

183
    // From LinkInterface
Gus Grubba's avatar
Gus Grubba committed
184 185
    bool    _connect                (void) override;
    void    _disconnect             (void) override;
186

187
    bool    _isIpLocal              (const QHostAddress& add);
Gus Grubba's avatar
Gus Grubba committed
188 189 190 191 192
    bool    _hardwareConnect        ();
    void    _restartConnection      ();
    void    _registerZeroconf       (uint16_t port, const std::string& regType);
    void    _deregisterZeroconf     ();
    void    _writeDataGram          (const QByteArray data, const UDPCLient* target);
Gus Grubba's avatar
Gus Grubba committed
193 194

#if defined(QGC_ZEROCONF_ENABLED)
195 196 197
    DNSServiceRef  _dnssServiceRef;
#endif

198 199 200 201 202 203
    bool                    _running;
    QUdpSocket*             _socket;
    UDPConfiguration*       _udpConfig;
    bool                    _connectState;
    QList<UDPCLient*>       _sessionTargets;
    QList<QHostAddress>     _localAddress;
Gus Grubba's avatar
Gus Grubba committed
204

pixhawk's avatar
pixhawk committed
205 206 207
};

#endif // UDPLINK_H