8 changed files with 504 additions and 1 deletions
-
3include/Makefile.am
-
52include/tr/comm_manager_dispatcher.h
-
52include/tr/comm_worker.h
-
46include/tr/connection_getter.h
-
5src/Makefile.am
-
128src/comm_manager_dispatcher.c
-
90src/comm_worker.c
-
129src/connection_getter.c
@ -0,0 +1,52 @@ |
|||
/** |
|||
* \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/>. |
|||
*/ |
|||
|
|||
#ifndef __TR_COMM_MANAGER_DISPATCHER_H__ |
|||
#define __TR_COMM_MANAGER_DISPATCHER_H__ |
|||
|
|||
#include <sys/types.h> |
|||
#include <sys/epoll.h> |
|||
|
|||
#include "trbase.h" |
|||
#include "trdata.h" |
|||
#include "trevent.h" |
|||
|
|||
#include "tr/comm_worker.h" |
|||
|
|||
TR_CLASS(TR_CommManagerDispatcher) { |
|||
TR_EXTENDS(TR_CommManager); |
|||
|
|||
TR_Set workers; |
|||
}; |
|||
|
|||
#define TR_commManDispAddWorker(this, worker) \ |
|||
(TR_setAdd((this)->workers, (worker)) |
|||
|
|||
TR_INSTANCE_INIT(TR_CommManagerDispatcher); |
|||
TR_CLASSVARS_DECL(TR_CommManagerDispatcher) { |
|||
TR_CV_EXTENDS(TR_EventHandler); |
|||
}; |
|||
|
|||
#endif // __TR_COMM_MANAGER_DISPATCHER_H__ |
|||
|
|||
// vim: set ts=4 sw=4: |
|||
|
|||
@ -0,0 +1,52 @@ |
|||
/** |
|||
* \file |
|||
* This exists only as a base type for all protocols. |
|||
* |
|||
* \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/>. |
|||
*/ |
|||
|
|||
#ifndef __TR_COMM_WORKER_H__ |
|||
#define __TR_COMM_WORKER_H__ |
|||
|
|||
#include <sys/types.h> |
|||
#include <semaphore.h> |
|||
|
|||
#include "trbase.h" |
|||
#include "trio.h" |
|||
|
|||
typedef struct s_TR_WorkerShm { |
|||
struct c_TR_Socket socket; |
|||
sem_t semaphore; |
|||
} s_TR_WorkerShm; |
|||
typedef s_TR_WorkerShm * sptr_TR_WorkerShm; |
|||
|
|||
TR_CLASS(TR_CommWorker) { |
|||
const char * const name; |
|||
pid_t pid; |
|||
TR_Socket socket; |
|||
sptr_TR_WorkerShm shm; |
|||
}; |
|||
|
|||
TR_INSTANCE_INIT(TR_CommWorker); |
|||
TR_CLASSVARS_DECL(TR_CommWorker) {}; |
|||
|
|||
#endif // __TR_COMM_WORKER_H__ |
|||
|
|||
// vim: set ts=4 sw=4: |
|||
|
|||
@ -0,0 +1,46 @@ |
|||
/** |
|||
* \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/>. |
|||
*/ |
|||
|
|||
#ifndef __TR_CONNECTION_GETTER_H__ |
|||
#define __TR_CONNECTION_GETTER_H__ |
|||
|
|||
#include <sys/types.h> |
|||
|
|||
#include "trbase.h" |
|||
#include "trevent.h" |
|||
|
|||
#include "tr/comm_worker.h" |
|||
|
|||
TR_CLASS(TR_ConnectionGetter) { |
|||
TR_EXTENDS(TR_EventHandler); |
|||
|
|||
TR_CommWorker worker; |
|||
}; |
|||
TR_INSTANCE_INIT(TR_ConnectionGetter); |
|||
TR_CLASSVARS_DECL(TR_ConnectionGetter) { |
|||
TR_CV_EXTENDS(TR_EventHandler); |
|||
}; |
|||
|
|||
#endif // __TR_CONNECTION_GETTER_H__ |
|||
|
|||
// vim: set ts=4 sw=4: |
|||
|
|||
@ -0,0 +1,128 @@ |
|||
/** |
|||
* \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 <sys/types.h> |
|||
#include <sys/wait.h> |
|||
#include <semaphore.h> |
|||
#include <errno.h> |
|||
|
|||
#include "trbase.h" |
|||
#include "trdata.h" |
|||
#include "trevent.h" |
|||
|
|||
#include "tr/comm_manager.h" |
|||
#include "tr/comm_manager_dispatcher.h" |
|||
#include "tr/interface/comm_manager.h" |
|||
#include "tr/comm_end_point.h" |
|||
#include "tr/connection.h" |
|||
#include "tr/connect_entry_point.h" |
|||
#include "tr/_comm_manager.h" // ?? may we don't need this one |
|||
|
|||
|
|||
static |
|||
int |
|||
commManagerDispatcherCtor(void * _this, va_list * params) |
|||
{ |
|||
TR_CommManagerDispatcher this = _this; |
|||
TR_PARENTCALL(TR_CommManagerDispatcher, _this, TR_Class, ctor, params); |
|||
this->workers = va_arg(*params, TR_Set); |
|||
|
|||
return 0; |
|||
} |
|||
|
|||
static |
|||
void |
|||
commManagerDispatcherDtor(void * _this) |
|||
{ |
|||
TR_PARENTCALL(TR_CommManagerDispatcher, _this, TR_Class, dtor); |
|||
} |
|||
|
|||
static |
|||
TR_EventDone |
|||
TR_commManagerDispatcherSendEndpoint(void * _this, TR_CommEndPoint endpoint) |
|||
{ |
|||
TR_CommManagerDispatcher this = _this; |
|||
TR_CommWorker worker = NULL; |
|||
int pid_stat; |
|||
|
|||
/* code to dispatch stuff.... */ |
|||
while (! TR_setEmpty(this->workers) && ! worker) { |
|||
if (! TR_iterableValid(this->workers)) { |
|||
TR_iterableRewind(this->workers); |
|||
} |
|||
worker = (TR_CommWorker)TR_iterableCurrent(this->workers); |
|||
if (0 != waitpid(worker->pid, &pid_stat, WNOHANG)) { |
|||
TR_setDelete(this->workers, worker); |
|||
TR_delete(worker); |
|||
continue; |
|||
} |
|||
} |
|||
|
|||
if (! worker) { |
|||
TR_delete(endpoint); // no ones there to handle you, sorry... |
|||
return TR_EVENT_DONE; |
|||
} |
|||
|
|||
TR_iterableNext(this->workers); |
|||
|
|||
if (-1 == sem_trywait(&(worker->shm->semaphore))) { |
|||
switch(errno) { |
|||
case EAGAIN: |
|||
case EINTR: |
|||
break; |
|||
|
|||
default: |
|||
TR_setDelete(this->workers, worker); |
|||
TR_delete(worker); |
|||
break; |
|||
} |
|||
} |
|||
|
|||
memcpy( |
|||
&(worker->shm->socket), |
|||
endpoint->transport, |
|||
sizeof(struct c_TR_Socket)); |
|||
TR_socketSendFd(worker->socket, TR_socketHandle(endpoint->transport)); |
|||
TR_delete(endpoint); |
|||
|
|||
return TR_EVENT_DONE; |
|||
} |
|||
|
|||
|
|||
static |
|||
void |
|||
TR_commManagerDispatcherCvInit(TR_class_ptr cls) |
|||
{ |
|||
TR_INHERIT_CLASSVARS(TR_CommManagerDispatcher, TR_CommManager); |
|||
} |
|||
|
|||
TR_INIT_IFACE(TR_Class, commManagerDispatcherCtor, |
|||
commManagerDispatcherDtor, NULL); |
|||
TR_INIT_IFACE(TR_CommManager, TR_commManagerDispatcherSendEndpoint, |
|||
NULL, NULL, NULL, NULL, NULL, NULL); |
|||
TR_CREATE_CLASS( |
|||
TR_CommManagerDispatcher, |
|||
TR_CommManager, |
|||
TR_commManagerDispatcherCvInit, |
|||
TR_IF(TR_Class), |
|||
TR_IF(TR_CommManager)); |
|||
|
|||
// vim: set ts=4 sw=4: |
|||
@ -0,0 +1,90 @@ |
|||
/** |
|||
* \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/>. |
|||
*/ |
|||
|
|||
#define _GNU_SOURCE |
|||
|
|||
#include <sys/types.h> |
|||
#include <sys/stat.h> |
|||
#include <sys/mman.h> |
|||
#include <sys/wait.h> |
|||
#include <stdarg.h> |
|||
#include <semaphore.h> |
|||
#include <unistd.h> |
|||
#include <fcntl.h> |
|||
|
|||
#include "trbase.h" |
|||
#include "trio.h" |
|||
|
|||
#include "tr/comm_worker.h" |
|||
|
|||
static |
|||
int |
|||
commWorkerCtor(void * _this, va_list * params) |
|||
{ |
|||
TR_CommWorker this = _this; |
|||
TR_Socket socket[2]; |
|||
|
|||
this->shm = mmap(NULL, sizeof(s_TR_WorkerShm), |
|||
PROT_READ|PROT_WRITE, |
|||
MAP_ANONYMOUS|MAP_SHARED, |
|||
0, 0); |
|||
|
|||
sem_init(&(this->shm->semaphore), 1, 1); |
|||
TR_socketPair(socket, SOCK_DGRAM); |
|||
|
|||
/* TODO error handling... */ |
|||
switch (this->pid = fork()) { |
|||
case -1: |
|||
TR_delete(socket[0]); |
|||
TR_delete(socket[1]); |
|||
break; |
|||
|
|||
case 0: |
|||
TR_delete(socket[0]); |
|||
this->socket = socket[1]; |
|||
break; |
|||
|
|||
default: |
|||
TR_delete(socket[1]); |
|||
this->socket = socket[0]; |
|||
break; |
|||
} |
|||
|
|||
return this->pid; |
|||
} |
|||
|
|||
static void commWorkerDtor(void * _this) { |
|||
TR_CommWorker this = _this; |
|||
|
|||
TR_delete(this->socket); |
|||
sem_close(&(this->shm->semaphore)); |
|||
munmap(this->shm, sizeof(s_TR_WorkerShm)); |
|||
if (this->pid != 0) { |
|||
int state; |
|||
waitpid(this->pid, &state, 0); |
|||
} |
|||
} |
|||
|
|||
TR_INIT_IFACE(TR_Class, commWorkerCtor, commWorkerDtor, NULL); |
|||
TR_CREATE_CLASS(TR_CommWorker, NULL, NULL, TR_IF(TR_Class)); |
|||
|
|||
// vim: set ts=4 sw=4: |
|||
@ -0,0 +1,129 @@ |
|||
/** |
|||
* \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 <stdarg.h> |
|||
#include <unistd.h> |
|||
|
|||
#include <sys/types.h> |
|||
#include <sys/stat.h> |
|||
#include <sys/mman.h> |
|||
#include <fcntl.h> |
|||
#include <semaphore.h> |
|||
|
|||
#include "trbase.h" |
|||
#include "trio.h" |
|||
#include "trevent.h" |
|||
|
|||
#include "tr/connection_getter.h" |
|||
#include "tr/connection.h" |
|||
#include "tr/protocol.h" |
|||
#include "tr/connect_entry_point.h" |
|||
|
|||
static |
|||
int |
|||
connectionGetterCtor(void * _this, va_list * params) |
|||
{ |
|||
TR_ConnectionGetter this = _this; |
|||
|
|||
TR_PARENTCALL(TR_ConnectionGetter, _this, TR_Class, ctor, params); |
|||
this->worker = va_arg(*params, TR_CommWorker); |
|||
|
|||
return 0; |
|||
} |
|||
|
|||
static |
|||
void |
|||
connectionGetterDtor(void * _this) |
|||
{ |
|||
TR_PARENTCALL(TR_ConnectionGetter, _this, TR_Class, dtor); |
|||
} |
|||
|
|||
static |
|||
TR_EventDone |
|||
connectionGetterAccept(void * _this, TR_Event event) |
|||
{ |
|||
int count = 0; |
|||
TR_ConnectionGetter this = _this; |
|||
TR_CommEndPoint connection = (TR_CommEndPoint)event->subject; |
|||
|
|||
int handle = TR_socketGetFd((TR_Socket)connection->transport); |
|||
|
|||
while (handle != -1) { |
|||
TR_Socket socket = TR_new(TR_Socket); |
|||
|
|||
memcpy( |
|||
socket, |
|||
&(this->worker->shm->socket), |
|||
sizeof(struct c_TR_Socket)); |
|||
sem_post(&(this->worker->shm->semaphore)); |
|||
TR_socketHandle(socket) = handle; |
|||
|
|||
TR_socketNonblock(socket); |
|||
TR_Connection new_con = TR_new( |
|||
TR_Connection, |
|||
socket, |
|||
connection->protocol, |
|||
CEP_DEFAULT_READ_SIZE); |
|||
TR_eventHandlerIssueEvent( |
|||
(TR_EventHandler)this, |
|||
TR_eventSubjectEmit( |
|||
(TR_EventSubject)new_con, |
|||
TR_CON_EVENT_NEW_CON, |
|||
NULL)); |
|||
if (++count > 100) break; |
|||
handle = TR_socketGetFd((TR_Socket)connection->transport); |
|||
} |
|||
|
|||
if (! handle) { |
|||
TR_eventHandlerIssueEvent( |
|||
(TR_EventHandler)this, |
|||
TR_eventSubjectEmit( |
|||
(TR_EventSubject)connection, |
|||
TR_CEP_EVENT_READ_BLOCK, |
|||
NULL)); |
|||
} |
|||
|
|||
return TR_EVENT_DONE; |
|||
} |
|||
|
|||
static |
|||
void |
|||
connectorCvInit(TR_class_ptr cls) |
|||
{ |
|||
TR_EVENT_HANDLER_SET_METHOD( |
|||
cls, |
|||
TR_ConnEntryPoint, |
|||
TR_CET_EVENT_ACC_READY, |
|||
connectionGetterAccept); |
|||
} |
|||
|
|||
TR_INIT_HANDLER(TR_ConnectionGetter); |
|||
TR_INIT_IFACE(TR_Class, connectionGetterCtor, connectionGetterDtor, NULL); |
|||
TR_CREATE_CLASS( |
|||
TR_ConnectionGetter, |
|||
TR_EventHandler, |
|||
connectorCvInit, |
|||
TR_IF(TR_Class)) = { |
|||
{ TR_HANDLER_CVARS(TR_ConnectionGetter) } |
|||
}; |
|||
|
|||
// vim: set ts=4 sw=4: |
|||
Write
Preview
Loading…
Cancel
Save
Reference in new issue