ncr_udp.h

Go to the documentation of this file.

// Copyright (C) 2013-2024 Internet Systems Consortium, Inc. ("ISC")
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
 
#ifndef NCR_UDP_LISTENER_H
#define NCR_UDP_LISTENER_H
 
 
#include <asiolink/asio_wrapper.h>
#include <asiolink/io_address.h>
#include <asiolink/io_service.h>
#include <asiolink/udp_endpoint.h>
#include <asiolink/udp_socket.h>
#include <dhcp_ddns/ncr_io.h>
#include <util/buffer.h>
#include <util/watch_socket.h>
 
#include <boost/shared_array.hpp>
#include <boost/enable_shared_from_this.hpp>
 
namespace isc {
namespace dhcp_ddns {
 
class NcrUDPError : public isc::Exception {
public:
    NcrUDPError(const char* file, size_t line, const char* what) :
        isc::Exception(file, line, what) { };
};
 
class UDPCallback;
typedef std::function<void(const bool, const UDPCallback*)> UDPCompletionHandler;
 
typedef boost::shared_array<uint8_t> RawBufferPtr;
 
typedef boost::shared_ptr<asiolink::UDPEndpoint> UDPEndpointPtr;
 
class UDPCallback {
 
public:
    struct Data {
 
        Data(RawBufferPtr& buffer, const size_t buf_size, UDPEndpointPtr& data_source)
            : buffer_(buffer), buf_size_(buf_size), data_source_(data_source),
              put_len_(0), error_code_(), bytes_transferred_(0) {
        };
 
        RawBufferPtr buffer_;
 
        size_t buf_size_;
 
        UDPEndpointPtr data_source_;
 
        size_t put_len_;
 
        boost::system::error_code error_code_;
 
        size_t bytes_transferred_;
 
    };
 
    UDPCallback(RawBufferPtr& buffer, const size_t buf_size,
                UDPEndpointPtr& data_source,
                const UDPCompletionHandler& handler);
 
    void operator ()(const boost::system::error_code error_code,
                     const size_t bytes_transferred);
 
    size_t getBytesTransferred() const {
        return (data_->bytes_transferred_);
    }
 
    void setBytesTransferred(const size_t value) {
        data_->bytes_transferred_ = value;
    }
 
    boost::system::error_code getErrorCode() const {
        return (data_->error_code_);
    }
 
    void setErrorCode(const boost::system::error_code value) {
        data_->error_code_  = value;
    }
 
    RawBufferPtr getBuffer() const {
        return (data_->buffer_);
    }
 
    size_t getBufferSize() const {
        return (data_->buf_size_);
    }
 
    const uint8_t* getData() const {
        return (data_->buffer_.get());
    }
 
    void putData(const uint8_t* src, size_t len);
 
    size_t getPutLen() const {
        return (data_->put_len_);
    }
 
    void setDataSource(UDPEndpointPtr& endpoint) {
        data_->data_source_ = endpoint;
    }
 
    const UDPEndpointPtr& getDataSource() {
        return (data_->data_source_);
    }
 
  private:
    UDPCompletionHandler handler_;
 
    boost::shared_ptr<Data> data_;
};
 
typedef isc::asiolink::UDPSocket<UDPCallback> NameChangeUDPSocket;
 
class NameChangeUDPListener : public NameChangeListener {
public:
    static const size_t RECV_BUF_MAX = isc::asiolink::UDPSocket<UDPCallback>::MIN_SIZE;
 
    NameChangeUDPListener(const isc::asiolink::IOAddress& ip_address,
                          const uint32_t port,
                          const NameChangeFormat format,
                          RequestReceiveHandlerPtr ncr_recv_handler,
                          const bool reuse_address = false);
 
    virtual ~NameChangeUDPListener();
 
    virtual void open(const isc::asiolink::IOServicePtr& io_service);
 
    virtual void close();
 
    void doReceive();
 
    void receiveCompletionHandler(const bool successful,
                                  const UDPCallback* recv_callback);
 
private:
 
    isc::asiolink::IOServicePtr io_service_;
 
    isc::asiolink::IOAddress ip_address_;
 
    uint32_t port_;
 
    NameChangeFormat format_;
 
    boost::shared_ptr<boost::asio::ip::udp::socket> asio_socket_;
 
    boost::shared_ptr<NameChangeUDPSocket> socket_;
 
    boost::shared_ptr<UDPCallback> recv_callback_;
 
    bool reuse_address_;
 
private:
    NameChangeUDPListener(const NameChangeUDPListener& source);
    NameChangeUDPListener& operator=(const NameChangeUDPListener& source);
};
 
class NameChangeUDPSender : public NameChangeSender {
public:
 
    static const size_t SEND_BUF_MAX =  NameChangeUDPListener::RECV_BUF_MAX;
 
    NameChangeUDPSender(const isc::asiolink::IOAddress& ip_address,
        const uint32_t port, const isc::asiolink::IOAddress& server_address,
        const uint32_t server_port, const NameChangeFormat format,
        RequestSendHandlerPtr ncr_send_handler,
        const size_t send_que_max = NameChangeSender::MAX_QUEUE_DEFAULT,
        const bool reuse_address = false);
 
    virtual ~NameChangeUDPSender();
 
    virtual void open(const isc::asiolink::IOServicePtr& io_service);
 
    virtual void close();
 
    virtual void doSend(NameChangeRequestPtr& ncr);
 
    void sendCompletionHandler(const bool successful,
                               const UDPCallback* send_callback);
 
    virtual int getSelectFd();
 
    virtual bool ioReady();
 
private:
 
    void closeWatchSocket();
 
    isc::asiolink::IOAddress ip_address_;
 
    uint32_t port_;
 
    isc::asiolink::IOAddress server_address_;
 
    uint32_t server_port_;
 
    NameChangeFormat format_;
 
    boost::shared_ptr<boost::asio::ip::udp::socket> asio_socket_;
 
    boost::shared_ptr<NameChangeUDPSocket> socket_;
 
    boost::shared_ptr<isc::asiolink::UDPEndpoint> server_endpoint_;
 
    boost::shared_ptr<UDPCallback> send_callback_;
 
    bool reuse_address_;
 
    util::WatchSocketPtr watch_socket_;
};
 
} // namespace isc::dhcp_ddns
} // namespace isc
 
#endif