socket.h

Go to the documentation of this file.

// Copyright (c) Microsoft Corporation. All rights reserved.
// Licensed under the Apache 2.0 License.
#pragma once
 
#include "ds/internal_logger.h"
 
#include <uv.h>
 
namespace asynchost
{
  template <class ConnType>
  class SocketBehaviour
  {
  protected:
    const char* name;
    const char* conn_name;
 
  public:
    SocketBehaviour(const char* name, const char* conn_name) :
      name(name),
      conn_name(conn_name)
    {}
    virtual ~SocketBehaviour() = default;
 
    virtual bool on_read(
      size_t /*unused*/, uint8_t*& /*unused*/, sockaddr /*unused*/)
    {
      return true;
    }
 
    virtual void on_accept(ConnType& /*unused*/) {}
 
    virtual void on_start(int64_t /*unused*/) {}
 
    virtual void on_listening(
      const std::string& host, const std::string& service)
    {
      LOG_INFO_FMT("{} {} listening on {}:{}", conn_name, name, host, service);
    }
    virtual void on_connect()
    {
      LOG_INFO_FMT("{} {} connected", conn_name, name);
    }
    virtual void on_disconnect()
    {
      LOG_TRACE_FMT("{} {} disconnected", conn_name, name);
    }
 
    virtual void on_bind_failed()
    {
      LOG_INFO_FMT("{} {} bind failed", conn_name, name);
    }
    virtual void on_connect_failed()
    {
      LOG_INFO_FMT("{} {} connect failed", conn_name, name);
    }
 
    virtual void on_resolve_failed()
    {
      LOG_FATAL_FMT("{} {} resolve failed", conn_name, name);
      abort();
    }
    virtual void on_listen_failed()
    {
      LOG_FATAL_FMT("{} {} listen failed", conn_name, name);
      abort();
    }
  };
 
  inline std::pair<std::string, std::string> addr_to_str(
    const sockaddr* addr, int address_family = AF_INET)
  {
    constexpr auto buf_len = INET6_ADDRSTRLEN;
    char buf[buf_len] = {};
    int rc = 0;
 
    if (address_family == AF_INET6)
    {
      const auto* const in6 = reinterpret_cast<const sockaddr_in6*>(addr);
      if ((rc = uv_ip6_name(in6, buf, buf_len)) != 0)
      {
        LOG_FAIL_FMT("uv_ip6_name failed: {}", uv_strerror(rc));
      }
 
      return {
        fmt::format("[{}]", buf), fmt::format("{}", ntohs(in6->sin6_port))};
    }
 
    assert(address_family == AF_INET);
    const auto* const in4 = reinterpret_cast<const sockaddr_in*>(addr);
    if ((rc = uv_ip4_name(in4, buf, buf_len)) != 0)
    {
      LOG_FAIL_FMT("uv_ip4_name failed: {}", uv_strerror(rc));
    }
 
    return {buf, fmt::format("{}", ntohs(in4->sin_port))};
  }
}