Scale Stack 0.3 Developer Documentation

scalestack/network/local/datagram/socket_service.cc

Go to the documentation of this file.

/*
 * Scale Stack
 *
 * Copyright 2010 Eric Day
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "config.h"

#include <cstdio>
#include <memory>
#include <netdb.h>
#include <stdint.h>
#include <sys/socket.h>
#include <sys/time.h>
#include <sys/types.h>
#include <sys/un.h>
#include <unistd.h>

#include <scalestack/kernel/module.h>
#include <scalestack/kernel/option.h>
#include <scalestack/network/datagram_service.h>
#include <scalestack/network/local/local.h>
#include <scalestack/network/local/datagram/socket.h>
#include <scalestack/network/local/datagram/socket_service.h>

using namespace std;

namespace scalestack
{
namespace network
{
namespace local
{
namespace datagram
{

/*
 * Public methods.
 */

void socket_service::local_options(kernel::module& module,
                                   const std::string& addresses)
{
  module.add_option("addresses", _("List of local sockets to bind to."),
                    "SOCKET[,SOCKET]...", addresses);
}

void socket_service::peer_options(kernel::module& module,
                                  const std::string& addresses)
{
  module.add_option("addresses", _("List of local sockets to send to."),
                    "SOCKET[,SOCKET]...", addresses);
}

socket_service::socket_service(kernel::module& module,
  std::auto_ptr<datagram_service> datagram_service):
  kernel::service(module),
  _datagram_service(datagram_service),
  _handler_service(module)
{
}

socket_service::~socket_service()
{
  _handler_service.stop();
}

void socket_service::add_socket(const struct addrinfo* local,
                                const struct addrinfo* peer)
{
  string local_address;

  if (local != NULL)
  {
    local_address =
      reinterpret_cast<struct sockaddr_un*>(local->ai_addr)->sun_path;
    _module.log_info(_("Local datagram socket trying to bind to %s"),
                     local_address.c_str());
  }

  if (peer != NULL)
  {
    _module.log_info(_("Local datagram socket with peer %s"),
      reinterpret_cast<struct sockaddr_un*>(peer->ai_addr)->sun_path);
  }

  auto_ptr<socket> new_socket(
    new socket(_handler_service, *_datagram_service, local, local_address,
               peer));
  new_socket->start();
  new_socket.release();

  if (local != NULL)
  {
    _module.log_notice(_("Local datagram socket bound to %s"),
      reinterpret_cast<struct sockaddr_un*>(local->ai_addr)->sun_path);
  }
}

void socket_service::add_local_sockets(void)
{
  add_local_sockets(_module.get_option("addresses").get_value());
}

void socket_service::add_local_sockets(const std::string& addresses)
{
  string::size_type current = 0;

  do
  {
    string::size_type next = addresses.find_first_of(",", current);
    string address = addresses.substr(current, next - current);

    struct addrinfo addrinfo;
    struct sockaddr_un sockaddr;
    set_address(addrinfo, sockaddr, SOCK_DGRAM, address);

    add_socket(&addrinfo, NULL);

    current = addresses.find_first_not_of(",", next);
  }
  while (current != string::npos);
}

void socket_service::add_peer_sockets(void)
{
  add_peer_sockets(_module.get_option("addresses").get_value());
}

void socket_service::add_peer_sockets(const std::string& addresses)
{
  string::size_type current = 0;

  do
  {
    string::size_type next = addresses.find_first_of(",", current);
    string address = addresses.substr(current, next - current);

    struct addrinfo addrinfo;
    struct sockaddr_un sockaddr;
    set_address(addrinfo, sockaddr, SOCK_DGRAM, address);

    /* Set a default local address for return packets. */
    struct timeval tv;
    gettimeofday(&tv, 0);
    char local_address[64];
    snprintf(local_address, 64, "/tmp/local.%u.%u",
             static_cast<uint32_t>(tv.tv_sec),
             static_cast<uint32_t>(tv.tv_usec));
    struct addrinfo local_addrinfo;
    struct sockaddr_un local_sockaddr;
    unlink(local_address);
    set_address(local_addrinfo, local_sockaddr, SOCK_DGRAM, local_address);
    add_socket(&local_addrinfo, &addrinfo);

    current = addresses.find_first_not_of(",", next);
  }
  while (current != string::npos);
}

} /* namespace datagram */
} /* namespace local */
} /* namespace network */
} /* namespace scalestack */

---