io_asio_socket.h

Go to the documentation of this file.

// Copyright (C) 2010-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 IO_ASIO_SOCKET_H
#define IO_ASIO_SOCKET_H 1
 
// IMPORTANT NOTE: only very few ASIO headers files can be included in
// this file.  In particular, asio.hpp should never be included here.
// See the description of the namespace below.
#include <config.h>
 
 
#include <exceptions/exceptions.h>
#include <asiolink/io_error.h>
#include <asiolink/io_socket.h>
#include <util/buffer.h>
 
#include <functional>
#include <string>
 
#include <unistd.h>             // for some network system calls
 
// We want to use coroutine.hpp from the system's boost headers if possible.
// However, very old Boost versions (provided by RHEL 7 or CentOS 7) didn't have
// this header. So we can resort to our bundled version, but only if necessary.
#ifndef HAVE_BOOST_ASIO_COROUTINE_HPP
#include <ext/coroutine/coroutine.hpp>
#else
#include <boost/asio/coroutine.hpp>
#endif
 
namespace isc {
namespace asiolink {
 
class SocketNotOpen : public IOError {
public:
    SocketNotOpen(const char* file, size_t line, const char* what) :
        IOError(file, line, what) {}
};
 
class SocketSetError : public IOError {
public:
    SocketSetError(const char* file, size_t line, const char* what) :
        IOError(file, line, what) {}
};
 
class BufferOverflow : public IOError {
public:
    BufferOverflow(const char* file, size_t line, const char* what) :
        IOError(file, line, what) {}
};
 
class IOEndpoint;
 
 
 
template <typename C>
class IOAsioSocket : public IOSocket {
 
 
private:
    IOAsioSocket(const IOAsioSocket<C>& source);
    IOAsioSocket& operator=(const IOAsioSocket<C>& source);
protected:
    IOAsioSocket() {}
public:
    virtual ~IOAsioSocket() = default;
 
    virtual int getNative() const = 0;
 
    virtual int getProtocol() const = 0;
 
    virtual bool isOpenSynchronous() const = 0;
 
    virtual void open(const IOEndpoint* endpoint, C& callback) = 0;
 
    virtual void asyncSend(const void* data, size_t length,
                           const IOEndpoint* endpoint, C& callback) = 0;
 
    virtual void asyncReceive(void* data, size_t length, size_t offset,
                              IOEndpoint* endpoint, C& callback) = 0;
 
    virtual bool processReceivedData(const void* staging, size_t length,
                                     size_t& cumulative,
                                     size_t& offset, size_t& expected,
                                     isc::util::OutputBufferPtr& buff) = 0;
 
    virtual void cancel() = 0;
 
    virtual void close() = 0;
};
 
 
 
template <typename C>
class DummyAsioSocket : public IOAsioSocket<C> {
private:
    DummyAsioSocket(const DummyAsioSocket<C>& source);
    DummyAsioSocket& operator=(const DummyAsioSocket<C>& source);
public:
    DummyAsioSocket(const int protocol) : protocol_(protocol) {}
 
    virtual int getNative() const { return (-1); }
 
    virtual int getProtocol() const { return (protocol_); }
 
 
    bool isOpenSynchronous() const {
        return true;
    }
 
    virtual bool open(const IOEndpoint*, C&) {
        return (false);
    }
 
    virtual void asyncSend(const void*, size_t, const IOEndpoint*, C&) {
    }
 
    virtual void asyncReceive(void*, size_t, size_t, IOEndpoint*, C&) {
    }
 
    virtual bool receiveComplete(const void*, size_t, size_t&, size_t&,
                                 size_t&, isc::util::OutputBufferPtr&)
    {
        return (true);
    }
 
 
    virtual void cancel() {
    }
 
    virtual void close() {
    }
 
private:
    const int protocol_;
};
 
}  // namespace asiolink
}  // namespace isc
 
#endif // IO_ASIO_SOCKET_H