UDPInterface.hpp

Go to the documentation of this file.

// MAVLink router and firewall.
// Copyright (C) 2018  Michael R. Shannon <mrshannon.aerospace@gmail.com>
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.


#ifndef UDPINTERFACE_HPP_
#define UDPINTERFACE_HPP_


#include <chrono>
#include <map>
#include <memory>

#include "Connection.hpp"
#include "ConnectionFactory.hpp"
#include "ConnectionPool.hpp"
#include "Interface.hpp"
#include "IPAddress.hpp"
#include "PacketParser.hpp"
#include "UDPSocket.hpp"


/** A UDP (User Datagram Protocol) interface.
 *
 *  An interface (for sending and receiving packets) implementing the user
 *  datagram protocol.
 */
class UDPInterface : public Interface
{
    public:
        UDPInterface(
            std::unique_ptr<UDPSocket> socket,
            std::shared_ptr<ConnectionPool> connection_pool,
            std::unique_ptr<ConnectionFactory<>> connection_factory);
        // LCOV_EXCL_START
        ~UDPInterface() = default;
        // LCOV_EXCL_STOP
        void send_packet(const std::chrono::nanoseconds &timeout) final;
        void receive_packet(const std::chrono::nanoseconds &timeout) final;

    protected:
        std::ostream &print_(std::ostream &os) const final;

    private:
        // Variables.
        std::unique_ptr<UDPSocket> socket_;
        std::shared_ptr<ConnectionPool> connection_pool_;
        std::unique_ptr<ConnectionFactory<>> connection_factory_;
        IPAddress last_ip_address_;
        std::map<IPAddress, std::shared_ptr<Connection>> connections_;
        PacketParser parser_;
        // Methods
        void update_connections_(
            const MAVAddress &mav_address, const IPAddress &ip_address);
};


#endif // UDPINTERFACE_HPP_