Scale Stack 0.3 Developer Documentation

scalestack/network/ip/tcp/connection_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 <cstring>
#include <exception>
#include <memory>
#include <netdb.h>

#include <scalestack/kernel/module.h>
#include <scalestack/kernel/option.h>
#include <scalestack/network/ip/address_service.h>
#include <scalestack/network/ip/tcp/connection.h>
#include <scalestack/network/ip/tcp/connection_service.h>

using namespace std;

namespace scalestack
{
namespace network
{
namespace ip
{
namespace tcp
{

class connection_address_service: public ip::address_service
{
public:

  connection_address_service(kernel::module& module,
                             event::handler_service& handler_service,
                             stream_service& stream_service);

  bool add_addresses(const struct addrinfo* addresses);

private:

  event::handler_service& _handler_service;
  stream_service& _stream_service;
};

/*
 * Public methods.
 */

void connection_service::options(kernel::module& module, const char* addresses)
{
  module.add_option("addresses", _("List of host:port pairs to connect to."),
                    "HOST[:PORT][,HOST[:PORT]]...", addresses);
}

connection_service::connection_service(kernel::module& module,
  std::auto_ptr<stream_service> stream_service,
  const std::string& default_port):
  stream_service_provider(module, stream_service),
  _default_port(default_port)
{
}

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

void connection_service::add_accepted(int file_descriptor,
                                      struct sockaddr& peer,
                                      socklen_t peer_size)
{
  char host_name[NI_MAXHOST];
  char port_name[NI_MAXSERV];

  int return_code = getnameinfo(&peer,
                                peer_size,
                                host_name, NI_MAXHOST,
                                port_name, NI_MAXSERV,
                                NI_NUMERICHOST | NI_NUMERICSERV);
  if (return_code != 0)
    _module.log_fatal(_("Failed getnameinfo: %s"), gai_strerror(return_code));

  _module.log_notice(_("TCP connection accepted from %s:%s"), host_name,
                     port_name);

  auto_ptr<connection> new_connection(
    new connection(_handler_service, *_stream_service, file_descriptor));
  new_connection->start();
  new_connection.release();
}

void connection_service::add_connections(void)
{
  add_connections(_module.get_option("addresses").get_value());
}

void connection_service::add_connections(const std::string& addresses)
{
  struct addrinfo address_options;

  memset(&address_options, 0, sizeof(struct addrinfo));
  address_options.ai_flags  = AI_PASSIVE;
  address_options.ai_family = AF_UNSPEC;
  address_options.ai_socktype = SOCK_STREAM;
  address_options.ai_protocol = IPPROTO_TCP;

  connection_address_service address_service(_module, _handler_service,
                                             *_stream_service);
  address_service.parse_addresses(addresses, false, _default_port,
                                  address_options);
}

/*
 * Public methods for address_service.
 */

connection_address_service::connection_address_service(
  kernel::module& module,
  event::handler_service& handler_service,
  stream_service& stream_service):
  ip::address_service(module),
  _handler_service(handler_service),
  _stream_service(stream_service)
{
}

bool connection_address_service::add_addresses(const struct addrinfo* addresses)
{
  auto_ptr<connection> new_connection(
    new connection(_handler_service, _stream_service, addresses));
  new_connection->start();
  new_connection.release();
  return false;
}

} /* namespace tcp */
} /* namespace ip */
} /* namespace network */
} /* namespace scalestack */

---