construct/ircd/client.cc

/*
 *  charybdis: an advanced ircd.
 *  client.c: Controls clients.
 *
 *  Copyright (C) 1990 Jarkko Oikarinen and University of Oulu, Co Center
 *  Copyright (C) 1996-2002 Hybrid Development Team
 *  Copyright (C) 2002-2005 ircd-ratbox development team
 *  Copyright (C) 2007 William Pitcock
 *  Copyright (C) 2016 Charybdis Development Team
 *  Copyright (C) 2016 Jason Volk <jason@zemos.net>
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307
 *  USA
 */

#include <boost/asio.hpp>
#include <ircd/sock.h>

namespace ircd
{
	struct client
	:std::enable_shared_from_this<client>
	{
		clist::const_iterator clit;
		std::unique_ptr<struct sock> sock;

		client();
		client(const client &) = delete;
		client &operator=(const client &) = delete;
		~client() noexcept;
	};

	clist clients_list;

	bool handle_error(client &, const error_code &);
	void handle_recv(client &, const error_code &, const size_t);
	void set_recv(client &);
}

using namespace ircd;

client::client()
{
}

client::~client()
noexcept
{
}

const clist &
ircd::clients()
{
	return clients_list;
}

std::shared_ptr<client>
ircd::add_client(std::unique_ptr<struct sock> sock)
{
	auto client(add_client());
	client->sock = std::move(sock);
	log::info("New client[%s] on local[%s]",
	          string(remote_address(*client)).c_str(),
	          string(local_address(*client)).c_str());

	set_recv(*client);
	return client;
}

std::shared_ptr<client>
ircd::add_client()
{
	auto client(std::make_shared<client>());
	client->clit = clients_list.emplace(end(clients_list), client);
	return client;
}

std::weak_ptr<client>
ircd::weak_from(client &client)
{
	return shared_from(client);
}

std::shared_ptr<client>
ircd::shared_from(client &client)
{
	return client.shared_from_this();
}

void
ircd::set_recv(client &client)
{
	using boost::asio::async_read;

	auto &sock(*client.sock);
	sock.checked = 0;
	sock.length = 0;

	async_read(sock.sd, mutable_buffers_1(sock.rbuf.data(), sock.rbuf.size()),
	           std::bind(&sock::handle_pck, &sock, ph::_1, ph::_2),
	           std::bind(&ircd::handle_recv, std::ref(client), ph::_1, ph::_2));
}

void
ircd::handle_recv(client &client,
                  const error_code &ec,
                  const size_t bytes)
try
{
	if(!handle_error(client, ec))
		return;

	auto &reel(client.sock->reel);
	for(const auto &line : reel)
		std::cout << line << std::endl;

	reel.clear();
	set_recv(client);
}
catch(const rfc1459::syntax_error &e)
{
	std::cerr << e.what() << std::endl;
}
catch(const std::exception &e)
{
	std::cerr << "errored: " << e.what() << std::endl;
}

bool
ircd::handle_error(client &client,
                   const error_code &ec)
{
	using namespace boost::system::errc;
	using boost::asio::error::eof;

	if(client.sock && client.sock->eptr)
		std::rethrow_exception(client.sock->eptr);

	switch(ec.value())
	{
		case success:    return true;
		default:         throw boost::system::system_error(ec);
	}
}

std::string
ircd::string(const ip_port &pair)
{
	std::string ret(64, '\0');
	ret.resize(snprintf(&ret.front(), ret.size(), "%s:%u",
	                    pair.first.c_str(),
	                    pair.second));
	return ret;
}

ircd::ip_port
ircd::local_address(const client &client)
{
	if(!client.sock)
		return { "0.0.0.0"s, 0 };

	return { local_ip(*client.sock), local_port(*client.sock) };
}

ircd::ip_port
ircd::remote_address(const client &client)
{
	if(!client.sock)
		return { "0.0.0.0"s, 0 };

	return { remote_ip(*client.sock), remote_port(*client.sock) };
}


///////////////////////////////////////////////////////////////////////////////
//
// client_sock.h
//

ircd::sock::sock(boost::asio::io_service *const &ios)
:sd{*ios}
,timer{*ios}
,checked{0}
,length{0}
{
}

size_t
ircd::sock::handle_pck(const error_code &ec,
                       const size_t bytes)
noexcept try
{
	if(ec)
		return 0;

	length += bytes;
	if(reel.append(rbuf.data(), length))
		return 0;

	if(terminated())
		throw rfc1459::syntax_error("invalid syntax"); //TODO: eps + ERR_

	checked = length;
	return remaining()?: throw rfc1459::syntax_error("message length exceeded"); //TODO: ERR_
}
catch(...)
{
	eptr = std::current_exception();
	return 0;
}

uint16_t
ircd::sock::remaining()
const
{
	return sizeof(rbuf) - length;
}

bool
ircd::sock::terminated()
const
{
	const auto b(std::next(rbuf.rbegin(), remaining()));
	const auto e(std::next(rbuf.rbegin(), sizeof(rbuf) - checked));
	return std::find(b, e, '\n') != e;
}