ghopp
/
libtrcomm
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
Browse Source

fixes and additions for threaded code

1.0.0
Georg Hopp 11 years ago
parent
c0b33ec777
commit
f71cac22a3
21 changed files with 207 additions and 102 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Split View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 3
      include/tr/comm_end_point.h
  2. 3
      include/tr/comm_manager.h
  3. 14
      include/tr/interface/comm_manager.h
  4. 5
      src/cep_write_buffered.c
  5. 4
      src/comm_end_point.c
  6. 3
      src/comm_end_point_read.c
  7. 34
      src/comm_manager.c
  8. 43
      src/comm_manager_epoll.c
  9. 54
      src/comm_manager_poll.c
  10. 4
      src/connection.c
  11. 12
      src/connector.c
  12. 60
      src/i_comm_manager.c
  13. 28
      src/io_handler.c
  14. 6
      src/protocol_handler.c
  15. 2
      src/server.c
  16. 6
      src/simple_client.c
  17. 5
      src/threaded_server.c
  18. 4
      testers/build.sh
  19. 3
      testers/test_handler.c
  20. 2
      testers/testclient.c
  21. 2
      testers/testserver2.c

3
include/tr/comm_end_point.h
View File

@ -64,7 +64,8 @@ TR_CLASSVARS_DECL(TR_CommEndPoint) {
#define TR_CEP_EVENT_SHUT_READ 10 // CommManager
#define TR_CEP_EVENT_SHUT_WRITE 11 // CommManager
#define TR_CEP_EVENT_CLOSE 12 // CommManager
#define TR_CEP_EVENT_MAX ((size_t)TR_CEP_EVENT_CLOSE)
#define TR_CEP_EVENT_IO_DONE 13 // CommManager
#define TR_CEP_EVENT_MAX ((size_t)TR_CEP_EVENT_IO_DONE)
#define TR_cepSetClose(ep) ((ep)->do_close = 1)
#define TR_cepHasProto(ep, proto) (TR_INSTANCE_OF(proto, TR_cepGetProto(ep)))

3
include/tr/comm_manager.h
View File

@ -24,6 +24,7 @@
#define __TR_COMM_MANAGER_H__
#include <sys/types.h>
#include <pthread.h>
#include "trbase.h"
#include "trdata.h"
@ -40,6 +41,8 @@ TR_CLASS(TR_CommManager) {
TR_Hash read;
size_t n_endpoints;
size_t max_handle;
unsigned long io_triggered;
pthread_mutex_t io_triggered_lock;
};
TR_INSTANCE_INIT(TR_CommManager);
TR_CLASSVARS_DECL(TR_CommManager) {

14
include/tr/interface/comm_manager.h
View File

@ -30,13 +30,13 @@
#include "tr/comm_end_point.h"
typedef TR_EventDone (* fptr_TR_commManagerAddEndpoint)(void *, TR_CommEndPoint);
typedef TR_EventDone (* fptr_TR_commManagerSelect)(void *, TR_Event, unsigned long);
typedef TR_EventDone (* fptr_TR_commManagerPollWrite)(void *, TR_Event);
typedef TR_EventDone (* fptr_TR_commManagerPollRead)(void *, TR_Event);
typedef TR_EventDone (* fptr_TR_commManagerDisableWrite)(void *, TR_Event);
typedef TR_EventDone (* fptr_TR_commManagerDisableRead)(void *, TR_Event);
typedef TR_EventDone (* fptr_TR_commManagerClose)(void *, TR_Event);
typedef void (* fptr_TR_commManagerAddEndpoint)(void *, TR_CommEndPoint);
typedef size_t (* fptr_TR_commManagerSelect)(void *, TR_Event, unsigned long);
typedef void (* fptr_TR_commManagerPollWrite)(void *, TR_Event);
typedef void (* fptr_TR_commManagerPollRead)(void *, TR_Event);
typedef void (* fptr_TR_commManagerDisableWrite)(void *, TR_Event);
typedef void (* fptr_TR_commManagerDisableRead)(void *, TR_Event);
typedef void (* fptr_TR_commManagerClose)(void *, TR_Event);
TR_INTERFACE(TR_CommManager) {
TR_IFID;

5
src/cep_write_buffered.c
View File

@ -29,7 +29,7 @@ int
TR_cepWriteBuffered(TR_CommEndPoint this, size_t * size)
{
TR_RemoteData data;
int send;
size_t send;
*size = 0;
@ -55,6 +55,9 @@ TR_cepWriteBuffered(TR_CommEndPoint this, size_t * size)
{
TR_RemoteData new_data = NULL;
printf("[~DEBUG~] wrote %zd bytes\n", send);
fflush(stdout);
if (send != ((TR_SizedData)data)->size) {
new_data = TR_new(
TR_RemoteData,

4
src/comm_end_point.c
View File

@ -41,7 +41,7 @@ commEndPointCtor(void * _this, va_list * params)
this->transport = va_arg(*params, TR_Socket);
this->protocol = va_arg(*params, TR_Protocol);
this->read_chunk_size = va_arg(*params, int);
this->do_close = 0;
this->do_close = FALSE;
this->write_buffer = TR_new(TR_Queue);
return 0;
@ -101,6 +101,7 @@ commEndPointCvInit(TR_class_ptr cls)
TR_EVENT_CREATE(cls, TR_CEP_EVENT_SHUT_READ);
TR_EVENT_CREATE(cls, TR_CEP_EVENT_SHUT_WRITE);
TR_EVENT_CREATE(cls, TR_CEP_EVENT_CLOSE);
TR_EVENT_CREATE(cls, TR_CEP_EVENT_IO_DONE);
}
const char * TR_cepEventStrings[] = {
@ -117,6 +118,7 @@ const char * TR_cepEventStrings[] = {
"TR_CEP_EVENT_SHUT_READ",
"TR_CEP_EVENT_SHUT_WRITE",
"TR_CEP_EVENT_CLOSE",
"TR_CEP_EVENT_IO_DONE",
};
intptr_t comm_end_point_events[TR_CEP_EVENT_MAX + 1];

3
src/comm_end_point_read.c
View File

@ -30,6 +30,9 @@ TR_commEndPointRead(TR_CommEndPoint this, TR_RemoteData * data_ptr)
{
*data_ptr = TR_socketRecv(this->transport, this->read_chunk_size);
printf("[~DEBUG~] read %zd bytes\n", ((TR_SizedData)*data_ptr)->size);
fflush(stdout);
if (! *data_ptr) return -1; // ment to trigger a close
if (*data_ptr == (void*)-1) return -2; // remote close... shutdown
if (*data_ptr == TR_emptyRemoteData) return FALSE; // read blocked

34
src/comm_manager.c
View File

@ -22,6 +22,7 @@
#include <unistd.h>
#include <poll.h>
#include <pthread.h>
#include "trbase.h"
#include "trdata.h"
@ -51,6 +52,8 @@ commManagerCtor(void * _this, va_list * params)
this->n_endpoints = sysconf(_SC_OPEN_MAX);
this->endpoints = TR_calloc(sizeof(TR_CommEndPoint), this->n_endpoints);
pthread_mutex_init(&this->io_triggered_lock, NULL);
return 0;
}
@ -61,6 +64,8 @@ commManagerDtor(void * _this)
TR_CommManager this = _this;
nfds_t i;
pthread_mutex_destroy(&this->io_triggered_lock);
for (i=0; i<this->n_endpoints; i++) {
TR_delete(this->endpoints[i]);
}
@ -72,11 +77,22 @@ commManagerDtor(void * _this)
static
TR_EventDone
TR_commManagerEnableWrite(void * _this, TR_Event event)
TR_commManagerWriteIsBlocked(void * _this, TR_Event event)
{
TR_CommManager this = _this;
TR_hashAdd(this->write, event->subject);
TR_hashDeleteByVal(this->write, TR_hashableGetHash(event->subject));
return TR_EVENT_DONE;
}
static
TR_EventDone
TR_commManagerDecrementIoTriggerd(TR_CommManager this, TR_Event event)
{
pthread_mutex_lock(&this->io_triggered_lock);
this->io_triggered--;
pthread_mutex_unlock(&this->io_triggered_lock);
return TR_EVENT_DONE;
}
@ -103,8 +119,6 @@ static
void
commManagerCvInit(TR_class_ptr cls)
{
TR_CLASSVARS(TR_EventHandler, cls)->event_methods->tree = TR_new(TR_Tree);
TR_EVENT_HANDLER_SET_METHOD(
cls, TR_EventDispatcher,
TR_DISPATCHER_EVENT_DATA_WAIT,
@ -120,7 +134,7 @@ commManagerCvInit(TR_class_ptr cls)
TR_EVENT_HANDLER_SET_METHOD(
cls, TR_CommEndPoint,
TR_CEP_EVENT_WRITE_BLOCK,
TR_commManagerPollWrite);
TR_commManagerWriteIsBlocked);
TR_EVENT_HANDLER_SET_METHOD(
cls, TR_CommEndPoint,
TR_CEP_EVENT_READ_BLOCK,
@ -144,14 +158,18 @@ commManagerCvInit(TR_class_ptr cls)
TR_EVENT_HANDLER_SET_METHOD(
cls, TR_CommEndPoint,
TR_CEP_EVENT_DATA_READY,
TR_commManagerEnableWrite);
TR_commManagerPollWrite);
TR_EVENT_HANDLER_SET_METHOD(
cls, TR_CommEndPoint,
TR_CEP_EVENT_DATA_END,
TR_commManagerDisableWrite);
TR_EVENT_HANDLER_SET_METHOD(
cls, TR_CommEndPoint,
TR_CEP_EVENT_IO_DONE,
TR_commManagerDecrementIoTriggerd);
}
TR_INSTANCE(TR_Hash, commManagerEventMethods);
TR_INIT_HANDLER(TR_CommManager);
TR_INIT_IFACE(TR_Class, commManagerCtor, commManagerDtor, NULL);
TR_INIT_IFACE(TR_CommManager, NULL, NULL, NULL, NULL, NULL, NULL, NULL);
TR_CREATE_CLASS(
@ -160,7 +178,7 @@ TR_CREATE_CLASS(
commManagerCvInit,
TR_IF(TR_Class),
TR_IF(TR_CommManager)) = {
{ &(_commManagerEventMethods.data) }
{ TR_HANDLER_CVARS(TR_CommManager) }
};
// vim: set ts=4 sw=4:

43
src/comm_manager_epoll.c
View File

@ -73,8 +73,8 @@ TR_commManagerEpollAddEndpoint(void * _this, TR_CommEndPoint endpoint)
int handle = endpoint->transport->handle;
struct epoll_event event;
//this->events[handle] = EPOLLET;
this->events[handle] = 0;
//this->events[handle] = EPOLLIN | EPOLLET;
this->events[handle] = EPOLLIN;
event.data.ptr = endpoint;
event.events = this->events[handle];
@ -82,19 +82,19 @@ TR_commManagerEpollAddEndpoint(void * _this, TR_CommEndPoint endpoint)
}
static
void
size_t
TR_commManagerEpollSelect(void * _this, TR_Event event, unsigned long timeout)
{
TR_CommManagerEpoll this = _this;
TR_CommManager cmgr = _this;
int i, nevents;
struct epoll_event _event;
//struct epoll_event _event;
nevents = epoll_wait(this->handle, events, MAXEVENTS, timeout);
for (i=0; i<nevents; i++) {
TR_CommEndPoint endpoint = (TR_CommEndPoint)events[i].data.ptr;
int handle = endpoint->transport->handle;
//int handle = endpoint->transport->handle;
if ((events[i].events & EPOLLIN) == EPOLLIN) {
if (TR_INSTANCE_OF(TR_TcpSocket, endpoint->transport)
@ -106,20 +106,20 @@ TR_commManagerEpollSelect(void * _this, TR_Event event, unsigned long timeout)
}
}
this->events[handle] &= ~EPOLLIN;
_event.data.ptr = endpoint;
_event.events = this->events[handle];
epoll_ctl(this->handle, EPOLL_CTL_MOD, handle, &_event);
//this->events[handle] &= ~EPOLLIN;
//_event.data.ptr = endpoint;
//_event.events = this->events[handle];
//epoll_ctl(this->handle, EPOLL_CTL_MOD, handle, &_event);
}
if ((events[i].events & EPOLLOUT) == EPOLLOUT) {
if (! event->subject->fin) {
TR_hashAdd(cmgr->write, endpoint);
}
this->events[handle] &= ~EPOLLOUT;
_event.data.ptr = endpoint;
_event.events = this->events[handle];
epoll_ctl(this->handle, EPOLL_CTL_MOD, handle, &_event);
//this->events[handle] &= ~EPOLLOUT;
//_event.data.ptr = endpoint;
//_event.events = this->events[handle];
//epoll_ctl(this->handle, EPOLL_CTL_MOD, handle, &_event);
}
if ((events[i].events & EPOLLHUP) == EPOLLHUP) {
@ -131,6 +131,15 @@ TR_commManagerEpollSelect(void * _this, TR_Event event, unsigned long timeout)
NULL));
}
}
if (nevents >= 0) {
return nevents;
} else {
perror("epoll");
fflush(stderr);
fflush(stdout);
return 0;
}
}
static
@ -180,9 +189,9 @@ static
void
TR_commManagerEpollEnableRead(void * _this, TR_Event event)
{
if (! TR_socketFinRd(((TR_CommEndPoint)event->subject)->transport)) {
TR_commManagerEpollEnable(_this, EPOLLIN, event);
}
// if (! TR_socketFinRd(((TR_CommEndPoint)event->subject)->transport)) {
// TR_commManagerEpollEnable(_this, EPOLLIN, event);
// }
}
static
@ -196,7 +205,7 @@ static
void
TR_commManagerEpollDisableRead(void * _this, TR_Event event)
{
TR_commManagerEpollDisable(_this, EPOLLIN, event);
// TR_commManagerEpollDisable(_this, EPOLLIN, event);
}
static

54
src/comm_manager_poll.c
View File

@ -71,45 +71,57 @@ TR_commManagerPollAddEndpoint(void * _this, TR_CommEndPoint endpoint)
TR_CommManagerPoll this = _this;
this->fds[endpoint->transport->handle].fd = endpoint->transport->handle;
this->fds[endpoint->transport->handle].events = 0;
this->fds[endpoint->transport->handle].events = POLLIN;
}
static
void
size_t
TR_commManagerPollSelect(void * _this, TR_Event event, unsigned long timeout)
{
TR_CommManagerPoll this = _this;
TR_CommManager cmgr = _this;
nfds_t i;
int nevents;
int nevents, doevents;
nevents = poll(this->fds, cmgr->max_handle+1, timeout);
for (i = 0; i < cmgr->max_handle+1; i++) {
printf("[=DEBUG=] handle %ld POLLIN? %s\n", i,
(this->fds[i].events & POLLIN) == POLLIN ? "YES" : "NO");
fflush(stdout);
}
if (nevents) {
nevents = doevents = poll(this->fds, cmgr->max_handle+1, timeout);
if (doevents) {
for (i = 0; i < cmgr->max_handle+1; i++) {
if (this->fds[i].revents != 0) {
TR_CommEndPoint endpoint = cmgr->endpoints[i];
nevents--;
doevents--;
if ((this->fds[i].revents & POLLIN) == POLLIN) {
if (TR_INSTANCE_OF(TR_TcpSocket, endpoint->transport)
&& ((TR_TcpSocket)endpoint->transport)->listen) {
pthread_mutex_lock(&cmgr->io_triggered_lock);
TR_hashAdd(cmgr->accept, endpoint);
pthread_mutex_unlock(&cmgr->io_triggered_lock);
} else {
if (! event->subject->fin) {
pthread_mutex_lock(&cmgr->io_triggered_lock);
TR_hashAdd(cmgr->read, endpoint);
pthread_mutex_unlock(&cmgr->io_triggered_lock);
}
}
this->fds[endpoint->transport->handle].events &= ~POLLIN;
//this->fds[endpoint->transport->handle].events &= ~POLLIN;
}
if ((this->fds[i].revents & POLLOUT) == POLLOUT) {
if (! event->subject->fin) {
pthread_mutex_lock(&cmgr->io_triggered_lock);
TR_hashAdd(cmgr->write, endpoint);
pthread_mutex_unlock(&cmgr->io_triggered_lock);
}
this->fds[endpoint->transport->handle].events &=
~(POLLOUT|POLLHUP);
//this->fds[endpoint->transport->handle].events &=
// ~(POLLOUT|POLLHUP);
}
if ((this->fds[i].revents & POLLHUP) == POLLHUP) {
@ -122,10 +134,12 @@ TR_commManagerPollSelect(void * _this, TR_Event event, unsigned long timeout)
}
this->fds[i].revents = 0;
if (nevents <= 0) break;
if (doevents <= 0) break;
}
}
}
return nevents;
}
static
@ -144,12 +158,12 @@ static
void
TR_commManagerPollEnableRead(void * _this, TR_Event event)
{
TR_CommManagerPoll this = _this;
TR_CommEndPoint endpoint = (TR_CommEndPoint)event->subject;
if (! TR_socketFinRd(endpoint->transport)) {
this->fds[endpoint->transport->handle].events |= POLLIN;
}
// TR_CommManagerPoll this = _this;
// TR_CommEndPoint endpoint = (TR_CommEndPoint)event->subject;
//
// if (! TR_socketFinRd(endpoint->transport)) {
// this->fds[endpoint->transport->handle].events |= POLLIN;
// }
}
static
@ -166,10 +180,10 @@ static
void
TR_commManagerPollDisableRead(void * _this, TR_Event event)
{
TR_CommManagerPoll this = _this;
TR_CommEndPoint endpoint = (TR_CommEndPoint)event->subject;
this->fds[endpoint->transport->handle].events &= ~POLLIN;
// TR_CommManagerPoll this = _this;
// TR_CommEndPoint endpoint = (TR_CommEndPoint)event->subject;
//
// this->fds[endpoint->transport->handle].events &= ~POLLIN;
}
static

4
src/connection.c
View File

@ -37,10 +37,10 @@ static
int
connectionCtor(void * _this, va_list * params)
{
TR_Connection this = _this;
//TR_Connection this = _this;
TR_PARENTCALL(TR_Connection, _this, TR_Class, ctor, params);
this->current_message = NULL;
//this->current_message = NULL;
return 0;
}

12
src/connector.c
View File

@ -76,6 +76,13 @@ connectorAccept(void * _this, TR_Event event)
socket = TR_socketAccept((TR_TcpSocket)connection->transport);
}
TR_eventHandlerIssueEvent(
(TR_EventHandler)this,
TR_eventSubjectEmit(
(TR_EventSubject)connection,
TR_CEP_EVENT_IO_DONE,
NULL));
if (! socket) {
TR_eventHandlerIssueEvent(
(TR_EventHandler)this,
@ -92,8 +99,6 @@ static
void
connectorCvInit(TR_class_ptr cls)
{
TR_CLASSVARS(TR_EventHandler, cls)->event_methods->tree = TR_new(TR_Tree);
TR_EVENT_HANDLER_SET_METHOD(
cls,
TR_ConnEntryPoint,
@ -101,6 +106,7 @@ connectorCvInit(TR_class_ptr cls)
connectorAccept);
}
TR_INIT_HANDLER(TR_Connector);
TR_INSTANCE(TR_Hash, connectorEventMethods);
TR_INIT_IFACE(TR_Class, connectorCtor, connectorDtor, NULL);
TR_CREATE_CLASS(
@ -108,7 +114,7 @@ TR_CREATE_CLASS(
TR_EventHandler,
connectorCvInit,
TR_IF(TR_Class)) = {
{ &(_connectorEventMethods.data) }
{ TR_HANDLER_CVARS(TR_Connector) }
};
// vim: set ts=4 sw=4:

60
src/i_comm_manager.c
View File

@ -20,8 +20,11 @@
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
#define _GNU_SOURCE
#include <errno.h>
#include <poll.h>
#include <pthread.h>
#include "trbase.h"
#include "trevent.h"
@ -59,6 +62,8 @@ TR_commManagerAddEndpoint(void * _this, TR_CommEndPoint endpoint)
TR_ISSUE_IO_READ_EVENT(this, endpoint);
}
this->io_triggered++;
TR_CALL(_this, TR_CommManager, addEndpoint, endpoint);
}
@ -90,22 +95,35 @@ TR_commManagerSelect(void * _this, TR_Event event)
TR_Timer timer = (TR_Timer)event->data;
TR_EventDispatcher dispatcher = (TR_EventDispatcher)event->subject;
unsigned long timeout; // milliseconds
char buffer[17];
pthread_getname_np(pthread_self(), buffer, 17);
if (! this->io_triggered) {
printf("[DEBUG] [%s] io triggerd was empty\n", buffer);
fflush(stdout);
pthread_mutex_lock(&this->io_triggered_lock);
this->io_triggered = TR_hashEach(this->write, this, commManagerIssueWriteEvents);
this->io_triggered += TR_hashEach(this->accept, this, commManagerIssueAcceptEvents);
this->io_triggered += TR_hashEach(this->read, this, commManagerIssueReadEvents);
pthread_mutex_unlock(&this->io_triggered_lock);
}
printf("[DEBUG] [%s] io triggerd: %lu\n", buffer, this->io_triggered);
fflush(stdout);
if (! (TR_hashEmpty(this->read)
&& TR_hashEmpty(this->write)
&& TR_hashEmpty(this->accept))) {
timeout = 0;
} else if (NULL == timer) {
if (! this->io_triggered) {
if (NULL == timer) {
timeout = TR_eventDispatcherGetDataWaitTime(dispatcher);
} else {
timeout = TR_timerGet(timer, NULL);
}
TR_CALL(_this, TR_CommManager, select, event, timeout);
printf("[DEBUG] [%s] select timeout: %lu\n", buffer, timeout);
fflush(stdout);
TR_hashEach(this->write, this, commManagerIssueWriteEvents);
TR_hashEach(this->accept, this, commManagerIssueAcceptEvents);
TR_hashEach(this->read, this, commManagerIssueReadEvents);
TR_CALL(_this, TR_CommManager, select, event, timeout);
}
return TR_EVENT_DONE;
}
@ -115,9 +133,13 @@ TR_commManagerPollWrite(void * _this, TR_Event event)
{
TR_CommManager this = _this;
TR_hashDeleteByVal(this->write, TR_hashableGetHash(event->subject));
if (! TR_socketFinWr(((TR_CommEndPoint)event->subject)->transport)) {
pthread_mutex_lock(&this->io_triggered_lock);
TR_hashAdd(this->write, event->subject);
TR_CALL(_this, TR_CommManager, pollWrite, event);
pthread_mutex_unlock(&this->io_triggered_lock);
printf("[!DEBUG!] socket added to write hash\n");
fflush(stdout);
}
return TR_EVENT_DONE;
@ -129,6 +151,10 @@ TR_commManagerPollRead(void * _this, TR_Event event)
TR_CommManager this = _this;
TR_CommEndPoint endpoint = (TR_CommEndPoint)event->subject;
if (! TR_socketFinRd(endpoint->transport)) {
TR_CALL(_this, TR_CommManager, pollRead, event);
}
if (TR_INSTANCE_OF(TR_TcpSocket, endpoint->transport)
&& ((TR_TcpSocket)endpoint->transport)->listen) {
TR_hashDeleteByVal(this->accept, TR_hashableGetHash(event->subject));
@ -136,10 +162,6 @@ TR_commManagerPollRead(void * _this, TR_Event event)
TR_hashDeleteByVal(this->read, TR_hashableGetHash(event->subject));
}
if (! TR_socketFinRd(endpoint->transport)) {
TR_CALL(_this, TR_CommManager, pollRead, event);
}
return TR_EVENT_DONE;
}
@ -157,12 +179,16 @@ TR_EventDone
TR_commManagerDisableWrite(void * _this, TR_Event event)
{
TR_CommManager this = _this;
TR_CommEndPoint endpoint = (TR_CommEndPoint)event->subject;
if (! endpoint->write_buffer->nmsg) {
// TODO think about a better way...
TR_hashDeleteByVal(this->write, TR_hashableGetHash(event->subject));
if (! event->subject->fin) {
TR_hashAdd(this->read, event->subject);
}
//if (! event->subject->fin) {
// TR_hashAdd(this->read, event->subject);
//}
TR_CALL(_this, TR_CommManager, disableWrite, event);
}
return TR_EVENT_DONE;
}

28
src/io_handler.c
View File

@ -50,6 +50,7 @@ ioHandlerRead(void * _this, TR_Event event)
TR_CommEndPoint endpoint = (TR_CommEndPoint)event->subject;
TR_Event revent;
TR_RemoteData data;
char ip[16];
switch (TR_commEndPointRead(endpoint, &data)) {
case FALSE: // EAGAIN
@ -83,6 +84,13 @@ ioHandlerRead(void * _this, TR_Event event)
return TR_EVENT_DONE;
}
TR_socketAddrIpStr(data->remote, ip, 16);
printf(
"DEBUG: remote ip: %s / port: %d\n",
ip,
TR_socketAddrPort(data->remote));
fflush(stdout);
revent = TR_eventSubjectEmit(
event->subject,
TR_CEP_EVENT_NEW_DATA,
@ -90,6 +98,13 @@ ioHandlerRead(void * _this, TR_Event event)
break;
}
TR_eventHandlerIssueEvent(
(TR_EventHandler)_this,
TR_eventSubjectEmit(
event->subject,
TR_CEP_EVENT_IO_DONE,
NULL));
TR_eventHandlerIssueEvent((TR_EventHandler)_this, revent);
return TR_EVENT_DONE;
@ -147,6 +162,13 @@ ioHandlerWrite(void * _this, TR_Event event)
endpoint->write_buffer_size -= written;
TR_eventHandlerIssueEvent(
(TR_EventHandler)_this,
TR_eventSubjectEmit(
event->subject,
TR_CEP_EVENT_IO_DONE,
NULL));
if (revent) {
TR_eventHandlerIssueEvent((TR_EventHandler)_this, revent);
}
@ -158,8 +180,6 @@ static
void
ioHandlerCvInit(TR_class_ptr cls)
{
TR_CLASSVARS(TR_EventHandler, cls)->event_methods->tree = TR_new(TR_Tree);
TR_EVENT_HANDLER_SET_METHOD(
cls,
TR_CommEndPoint,
@ -172,14 +192,14 @@ ioHandlerCvInit(TR_class_ptr cls)
ioHandlerWrite);
}
TR_INSTANCE(TR_Hash, ioHandlerEventMethods);
TR_INIT_HANDLER(TR_IoHandler);
TR_INIT_IFACE(TR_Class, ioHandlerCtor, ioHandlerDtor, NULL);
TR_CREATE_CLASS(
TR_IoHandler,
TR_EventHandler,
ioHandlerCvInit,
TR_IF(TR_Class)) = {
{ &(_ioHandlerEventMethods.data) }
{ TR_HANDLER_CVARS(TR_IoHandler) }
};
// vim: set ts=4 sw=4:

6
src/protocol_handler.c
View File

@ -125,8 +125,6 @@ static
void
protocolHandlerCvInit(TR_class_ptr cls)
{
TR_CLASSVARS(TR_EventHandler, cls)->event_methods->tree = TR_new(TR_Tree);
TR_EVENT_HANDLER_SET_METHOD(
cls,
TR_CommEndPoint,
@ -144,14 +142,14 @@ protocolHandlerCvInit(TR_class_ptr cls)
// protocolHandlerUpgrade);
}
TR_INSTANCE(TR_Hash, protocolHandlerEventMethods);
TR_INIT_HANDLER(TR_ProtocolHandler);
TR_INIT_IFACE(TR_Class, protocolHandlerCtor, protocolHandlerDtor, NULL);
TR_CREATE_CLASS(
TR_ProtocolHandler,
TR_EventHandler,
protocolHandlerCvInit,
TR_IF(TR_Class)) = {
{ &(_protocolHandlerEventMethods.data) }
{ TR_HANDLER_CVARS(TR_ProtocolHandler) }
};
// vim: set ts=4 sw=4:

2
src/server.c
View File

@ -43,7 +43,7 @@ serverCtor(void * _this, va_list * params)
{
TR_Server this = _this;
#if 1
#if 0
this->comm_manager = (TR_CommManager)TR_new(TR_CommManagerEpoll);
#else
this->comm_manager = (TR_CommManager)TR_new(TR_CommManagerPoll);

6
src/simple_client.c
View File

@ -135,8 +135,6 @@ static
void
simpleClientCvInit(TR_class_ptr cls)
{
TR_CLASSVARS(TR_EventHandler, cls)->event_methods->tree = TR_new(TR_Tree);
TR_EVENT_HANDLER_SET_METHOD(
cls,
TR_EventDispatcher,
@ -149,14 +147,14 @@ simpleClientCvInit(TR_class_ptr cls)
simpleClientHandleData);
}
TR_INSTANCE(TR_Hash, simpleClientEventMethods);
TR_INIT_HANDLER(TR_SimpleClient);
TR_INIT_IFACE(TR_Class, simpleClientCtor, simpleClientDtor, NULL);
TR_CREATE_CLASS(
TR_SimpleClient,
TR_EventHandler,
simpleClientCvInit,
TR_IF(TR_Class)) = {
{ &(_simpleClientEventMethods.data) }
{ TR_HANDLER_CVARS(TR_SimpleClient) }
};
// vim: set ts=4 sw=4:

5
src/threaded_server.c
View File

@ -36,6 +36,7 @@ threadedServerCtor(void * _this, va_list * params)
{
TR_ThreadedServer this = _this;
int i;
char buffer[16];
TR_PARENTCALL(TR_ThreadedServer, _this, TR_Class, ctor, params);
@ -43,9 +44,11 @@ threadedServerCtor(void * _this, va_list * params)
this->threads = TR_malloc(sizeof(TR_EventThread) * this->n_threads);
for (i=0; i<this->n_threads; i++) {
sprintf(buffer, "test%03d", i);
this->threads[i] = TR_new(
TR_EventThread,
((TR_Server)this)->dispatcher);
((TR_Server)this)->dispatcher,
buffer);
}
return 0;

4
testers/build.sh
View File

@ -1,10 +1,10 @@
#!/bin/bash
#TRLIBS="-ltrbase -ltrhashing -ltrio -ltrdata -ltrevent -ltrcomm"
TRLIBS="/usr/local/lib/libtrcomm.a /usr/local/lib/libtrevent.a /usr/local/lib/libtrdata.a /usr/local/lib/libtrio.a /usr/local/lib/libtrhashing.a /usr/local/lib/libtrbase.a"
LIBS="-lcrypto -lssl -lrt -luuid"
LIBS="-lcrypto -lssl -lrt -luuid -lpthread"
gcc ${CFLAGS} -c -o test_handler.o test_handler.c
gcc ${CFLAGS} -I/usr/local/include -L/usr/local/lib ${LIBS} -o testserver testserver.c test_handler.o ${TRLIBS}
gcc ${CFLAGS} -I/usr/local/include -L/usr/local/lib ${LIBS} -o testserver2 testserver2.c test_handler.o ${TRLIBS}
gcc ${CFLAGS} -I/usr/local/include -L/usr/local/lib ${LIBS} -o testtcp testclient.c ${TRLIBS}
gcc ${CFLAGS} -I/usr/local/include -L/usr/local/lib ${LIBS} -DUDP=1 -o testudp testclient.c ${TRLIBS}
gcc ${CFLAGS} -I/usr/local/include -L/usr/local/lib ${LIBS} -lpthread -o testserver_thread testserver_thread.c test_handler.o ${TRLIBS}
gcc ${CFLAGS} -I/usr/local/include -L/usr/local/lib ${LIBS} -o testserver_thread testserver_thread.c test_handler.o ${TRLIBS}

3
testers/test_handler.c
View File

@ -15,7 +15,7 @@ testHandlerHeartbeat(TR_EventHandler this, TR_Event event)
double size_msg = ((TestHandler)this)->size
? size / ((TestHandler)this)->handled
: 0.0;
int div_count = 0;
int div_count = ' ';
while (size > 1024. && div_count != 'G') {
size /= 1024.;
@ -31,6 +31,7 @@ testHandlerHeartbeat(TR_EventHandler this, TR_Event event)
((TR_EventDispatcher)event->subject)->n_beats,
((TestHandler)this)->handled,
size, div_count, size_msg);
fflush(stdout);
((TestHandler)this)->handled = 0;
((TestHandler)this)->size = 0;

2
testers/testclient.c
View File

@ -30,7 +30,7 @@ main (int argc, char * argv[])
TR_ProtoMessageRaw message;
int i, j=0;
TR_logger = TR_INSTANCE_CAST(TR_Logger, mylogger2);
//TR_logger = TR_INSTANCE_CAST(TR_Logger, mylogger2);
protocol = TR_new(TR_ProtocolRaw);
#if UDP

2
testers/testserver2.c
View File

@ -19,7 +19,7 @@ main (int argc, char * argv[])
TR_Protocol protocol = TR_new(TR_ProtocolRaw);
TestHandler test_handler = TR_new(TestHandler);
TR_logger = TR_INSTANCE_CAST(TR_Logger, mylogger2);
//TR_logger = TR_INSTANCE_CAST(TR_Logger, mylogger2);
TR_serverAddHandler(server, (TR_EventHandler)test_handler);
TR_serverBindTcp(server, "0.0.0.0", 5678, protocol);

Write Preview
Loading…
Cancel
Save