Scale Stack 0.3 Developer Documentation

scalestack/proxy/proxy.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 <exception>

#include <scalestack/kernel/core.h>
#include <scalestack/kernel/module.h>
#include <scalestack/network/ip/tcp/connection_service.h>
#include <scalestack/network/ip/tcp/listener_service.h>
#include <scalestack/proxy/client_service.h>
#include <scalestack/proxy/server_service.h>

using namespace std;
using namespace scalestack;

static void options(kernel::module& module)
{
  network::ip::tcp::listener_service::options(module, "12345");
  module.add_option("server_address", _("The server host:port to proxy to."),
                    "HOST:PORT", "localhost:7");
  module.add_option("server_timeout",
                    _("How long to wait for the server to connect."),
                    "SECONDS", "15");
}

static void start(kernel::module& module)
{
  auto_ptr<proxy::server_service> server_service(
    new proxy::server_service(module));
  auto_ptr<network::stream_service> client_service(
    new proxy::client_service(module, *server_service));
  auto_ptr<network::ip::tcp::connection_service> tcp_server_service(
    new network::ip::tcp::connection_service(module,
    static_cast<auto_ptr<network::stream_service> >(server_service), "12345"));
  auto_ptr<network::stream_service_provider> tcp_client_service(
    new network::ip::tcp::connection_service(module, client_service, "12345"));
  auto_ptr<network::ip::tcp::listener_service> listener_service(
    new network::ip::tcp::listener_service(module, tcp_client_service, ""));

  listener_service->add_listeners();
  module.get_core().add_service("proxy::listener", listener_service.get());
  listener_service.release();

  try
  {
    module.get_core().add_service("proxy::server", tcp_server_service.get());
    tcp_server_service.release();
  }
  catch (exception&)
  {
    delete module.get_core().remove_service("proxy::listener");
    throw;
  }
}

static void stop(kernel::module& module)
{
  delete module.get_core().remove_service("proxy::listener");
  delete module.get_core().remove_service("proxy::server");
}

SCALESTACK_KERNEL_MODULE(options, start, stop, NULL, NULL);

---