comm_manager.c 3.27 KB
/**
 * \file
 *
 * \author	Georg Hopp
 *
 * \copyright
 * Copyright © 2014 Georg Hopp
 *
 * 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 3 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, see <http://www.gnu.org/licenses/>.
 */

#include <unistd.h>
#include <poll.h>

#include "trbase.h"
#include "trevent.h"

#include "tr/comm_end_point.h"
#include "tr/connection.h"
#include "tr/comm_manager.h"
#include "tr/interface/comm_manager.h"

static
int
commManagerCtor(void * _this, va_list * params)
{
	TR_CommManager this = _this;

	TR_PARENTCALL(TR_CommManager, _this, TR_Class, ctor, params);

	this->n_endpoints = sysconf(_SC_OPEN_MAX);
	this->endpoints   = TR_calloc(sizeof(TR_CommEndPoint), this->n_endpoints);

	return 0;
}

static
void
commManagerDtor(void * _this)
{
	TR_CommManager this = _this;
	nfds_t         i;

	for (i = 0; i < this->n_endpoints; i++) {
		TR_delete(this->endpoints[i]);
	}
	TR_MEM_FREE(this->endpoints);
}

static
TR_EventDone
TR__commManagerAddEndpoint(void * _this, TR_Event event)
{
	TR_commManagerAddEndpoint(
			(TR_CommManager)_this,
			(TR_CommEndPoint)event->subject);

	return TR_EVENT_DONE;
}

TR_EventDone TR_commManagerSelect(void *, TR_Event, int);
TR_EventDone TR_commManagerEnableWrite(void *, TR_Event);
TR_EventDone TR_commManagerDisableWrite(void *, TR_Event);
TR_EventDone TR_commManagerClose(void *, TR_Event);
TR_EventDone TR_commManagerShutdownRead(void *, TR_Event);
TR_EventDone TR_commManagerShutdownWrite(void *, TR_Event);

static
void
commManagerCvInit(TR_class_ptr cls)
{
	TR_EVENT_HANDLER_SET_METHOD(
			cls,
			TR_EventDispatcher,
			TR_DISPATCHER_EVENT_DATA_WAIT,
			TR_commManagerSelect);

	TR_EVENT_HANDLER_SET_METHOD(
			cls,
			TR_EventDispatcher,
			TR_DISPATCHER_EVENT_SHUTDOWN,
			TR_commManagerShutdown);

	TR_EVENT_HANDLER_SET_METHOD(
			cls,
			TR_Connection,
			TR_CON_EVENT_NEW_CON,
			TR__commManagerAddEndpoint);

	TR_EVENT_HANDLER_SET_METHOD(
			cls,
			TR_Connection,
			TR_CEP_EVENT_PENDING_DATA,
			TR_commManagerEnableWrite);

	TR_EVENT_HANDLER_SET_METHOD(
			cls,
			TR_Connection,
			TR_CEP_EVENT_END_DATA,
			TR_commManagerDisableWrite);

	TR_EVENT_HANDLER_SET_METHOD(
			cls,
			TR_CommEndPoint,
			TR_CEP_EVENT_CLOSE,
			TR_commManagerClose);

	TR_EVENT_HANDLER_SET_METHOD(
			cls,
			TR_CommEndPoint,
			TR_CEP_EVENT_SHUT_READ,
			TR_commManagerShutdownRead);

	TR_EVENT_HANDLER_SET_METHOD(
			cls,
			TR_CommEndPoint,
			TR_CEP_EVENT_SHUT_READ,
			TR_commManagerShutdownWrite);
}

TR_INSTANCE(TR_Hash, commManagerEventMethods);
TR_INIT_IFACE(TR_Class, commManagerCtor, commManagerDtor, NULL);
TR_INIT_IFACE(TR_CommManager, NULL, NULL, NULL, NULL, NULL, NULL, NULL);
TR_CREATE_CLASS(
		TR_CommManager,
		TR_EventHandler,
		commManagerCvInit,
		TR_IF(TR_Class),
		TR_IF(TR_CommManager)) = {
	{ &(_commManagerEventMethods.data) }
};

// vim: set ts=4 sw=4: