Commit f71cac22a3dc7ce41c180776124ec52bac46caa4
1 parent
c0b33ec7
fixes and additions for threaded code
Showing
21 changed files
with
207 additions
and
102 deletions
| @@ -64,7 +64,8 @@ TR_CLASSVARS_DECL(TR_CommEndPoint) { | @@ -64,7 +64,8 @@ TR_CLASSVARS_DECL(TR_CommEndPoint) { | ||
| 64 | #define TR_CEP_EVENT_SHUT_READ 10 // CommManager | 64 | #define TR_CEP_EVENT_SHUT_READ 10 // CommManager |
| 65 | #define TR_CEP_EVENT_SHUT_WRITE 11 // CommManager | 65 | #define TR_CEP_EVENT_SHUT_WRITE 11 // CommManager |
| 66 | #define TR_CEP_EVENT_CLOSE 12 // CommManager | 66 | #define TR_CEP_EVENT_CLOSE 12 // CommManager |
| 67 | -#define TR_CEP_EVENT_MAX ((size_t)TR_CEP_EVENT_CLOSE) | 67 | +#define TR_CEP_EVENT_IO_DONE 13 // CommManager |
| 68 | +#define TR_CEP_EVENT_MAX ((size_t)TR_CEP_EVENT_IO_DONE) | ||
| 68 | 69 | ||
| 69 | #define TR_cepSetClose(ep) ((ep)->do_close = 1) | 70 | #define TR_cepSetClose(ep) ((ep)->do_close = 1) |
| 70 | #define TR_cepHasProto(ep, proto) (TR_INSTANCE_OF(proto, TR_cepGetProto(ep))) | 71 | #define TR_cepHasProto(ep, proto) (TR_INSTANCE_OF(proto, TR_cepGetProto(ep))) |
| @@ -24,6 +24,7 @@ | @@ -24,6 +24,7 @@ | ||
| 24 | #define __TR_COMM_MANAGER_H__ | 24 | #define __TR_COMM_MANAGER_H__ |
| 25 | 25 | ||
| 26 | #include <sys/types.h> | 26 | #include <sys/types.h> |
| 27 | +#include <pthread.h> | ||
| 27 | 28 | ||
| 28 | #include "trbase.h" | 29 | #include "trbase.h" |
| 29 | #include "trdata.h" | 30 | #include "trdata.h" |
| @@ -40,6 +41,8 @@ TR_CLASS(TR_CommManager) { | @@ -40,6 +41,8 @@ TR_CLASS(TR_CommManager) { | ||
| 40 | TR_Hash read; | 41 | TR_Hash read; |
| 41 | size_t n_endpoints; | 42 | size_t n_endpoints; |
| 42 | size_t max_handle; | 43 | size_t max_handle; |
| 44 | + unsigned long io_triggered; | ||
| 45 | + pthread_mutex_t io_triggered_lock; | ||
| 43 | }; | 46 | }; |
| 44 | TR_INSTANCE_INIT(TR_CommManager); | 47 | TR_INSTANCE_INIT(TR_CommManager); |
| 45 | TR_CLASSVARS_DECL(TR_CommManager) { | 48 | TR_CLASSVARS_DECL(TR_CommManager) { |
| @@ -30,13 +30,13 @@ | @@ -30,13 +30,13 @@ | ||
| 30 | 30 | ||
| 31 | #include "tr/comm_end_point.h" | 31 | #include "tr/comm_end_point.h" |
| 32 | 32 | ||
| 33 | -typedef TR_EventDone (* fptr_TR_commManagerAddEndpoint)(void *, TR_CommEndPoint); | ||
| 34 | -typedef TR_EventDone (* fptr_TR_commManagerSelect)(void *, TR_Event, unsigned long); | ||
| 35 | -typedef TR_EventDone (* fptr_TR_commManagerPollWrite)(void *, TR_Event); | ||
| 36 | -typedef TR_EventDone (* fptr_TR_commManagerPollRead)(void *, TR_Event); | ||
| 37 | -typedef TR_EventDone (* fptr_TR_commManagerDisableWrite)(void *, TR_Event); | ||
| 38 | -typedef TR_EventDone (* fptr_TR_commManagerDisableRead)(void *, TR_Event); | ||
| 39 | -typedef TR_EventDone (* fptr_TR_commManagerClose)(void *, TR_Event); | 33 | +typedef void (* fptr_TR_commManagerAddEndpoint)(void *, TR_CommEndPoint); |
| 34 | +typedef size_t (* fptr_TR_commManagerSelect)(void *, TR_Event, unsigned long); | ||
| 35 | +typedef void (* fptr_TR_commManagerPollWrite)(void *, TR_Event); | ||
| 36 | +typedef void (* fptr_TR_commManagerPollRead)(void *, TR_Event); | ||
| 37 | +typedef void (* fptr_TR_commManagerDisableWrite)(void *, TR_Event); | ||
| 38 | +typedef void (* fptr_TR_commManagerDisableRead)(void *, TR_Event); | ||
| 39 | +typedef void (* fptr_TR_commManagerClose)(void *, TR_Event); | ||
| 40 | 40 | ||
| 41 | TR_INTERFACE(TR_CommManager) { | 41 | TR_INTERFACE(TR_CommManager) { |
| 42 | TR_IFID; | 42 | TR_IFID; |
| @@ -29,7 +29,7 @@ int | @@ -29,7 +29,7 @@ int | ||
| 29 | TR_cepWriteBuffered(TR_CommEndPoint this, size_t * size) | 29 | TR_cepWriteBuffered(TR_CommEndPoint this, size_t * size) |
| 30 | { | 30 | { |
| 31 | TR_RemoteData data; | 31 | TR_RemoteData data; |
| 32 | - int send; | 32 | + size_t send; |
| 33 | 33 | ||
| 34 | *size = 0; | 34 | *size = 0; |
| 35 | 35 | ||
| @@ -55,6 +55,9 @@ TR_cepWriteBuffered(TR_CommEndPoint this, size_t * size) | @@ -55,6 +55,9 @@ TR_cepWriteBuffered(TR_CommEndPoint this, size_t * size) | ||
| 55 | { | 55 | { |
| 56 | TR_RemoteData new_data = NULL; | 56 | TR_RemoteData new_data = NULL; |
| 57 | 57 | ||
| 58 | + printf("[~DEBUG~] wrote %zd bytes\n", send); | ||
| 59 | + fflush(stdout); | ||
| 60 | + | ||
| 58 | if (send != ((TR_SizedData)data)->size) { | 61 | if (send != ((TR_SizedData)data)->size) { |
| 59 | new_data = TR_new( | 62 | new_data = TR_new( |
| 60 | TR_RemoteData, | 63 | TR_RemoteData, |
| @@ -41,7 +41,7 @@ commEndPointCtor(void * _this, va_list * params) | @@ -41,7 +41,7 @@ commEndPointCtor(void * _this, va_list * params) | ||
| 41 | this->transport = va_arg(*params, TR_Socket); | 41 | this->transport = va_arg(*params, TR_Socket); |
| 42 | this->protocol = va_arg(*params, TR_Protocol); | 42 | this->protocol = va_arg(*params, TR_Protocol); |
| 43 | this->read_chunk_size = va_arg(*params, int); | 43 | this->read_chunk_size = va_arg(*params, int); |
| 44 | - this->do_close = 0; | 44 | + this->do_close = FALSE; |
| 45 | this->write_buffer = TR_new(TR_Queue); | 45 | this->write_buffer = TR_new(TR_Queue); |
| 46 | 46 | ||
| 47 | return 0; | 47 | return 0; |
| @@ -101,6 +101,7 @@ commEndPointCvInit(TR_class_ptr cls) | @@ -101,6 +101,7 @@ commEndPointCvInit(TR_class_ptr cls) | ||
| 101 | TR_EVENT_CREATE(cls, TR_CEP_EVENT_SHUT_READ); | 101 | TR_EVENT_CREATE(cls, TR_CEP_EVENT_SHUT_READ); |
| 102 | TR_EVENT_CREATE(cls, TR_CEP_EVENT_SHUT_WRITE); | 102 | TR_EVENT_CREATE(cls, TR_CEP_EVENT_SHUT_WRITE); |
| 103 | TR_EVENT_CREATE(cls, TR_CEP_EVENT_CLOSE); | 103 | TR_EVENT_CREATE(cls, TR_CEP_EVENT_CLOSE); |
| 104 | + TR_EVENT_CREATE(cls, TR_CEP_EVENT_IO_DONE); | ||
| 104 | } | 105 | } |
| 105 | 106 | ||
| 106 | const char * TR_cepEventStrings[] = { | 107 | const char * TR_cepEventStrings[] = { |
| @@ -117,6 +118,7 @@ const char * TR_cepEventStrings[] = { | @@ -117,6 +118,7 @@ const char * TR_cepEventStrings[] = { | ||
| 117 | "TR_CEP_EVENT_SHUT_READ", | 118 | "TR_CEP_EVENT_SHUT_READ", |
| 118 | "TR_CEP_EVENT_SHUT_WRITE", | 119 | "TR_CEP_EVENT_SHUT_WRITE", |
| 119 | "TR_CEP_EVENT_CLOSE", | 120 | "TR_CEP_EVENT_CLOSE", |
| 121 | + "TR_CEP_EVENT_IO_DONE", | ||
| 120 | }; | 122 | }; |
| 121 | 123 | ||
| 122 | intptr_t comm_end_point_events[TR_CEP_EVENT_MAX + 1]; | 124 | intptr_t comm_end_point_events[TR_CEP_EVENT_MAX + 1]; |
| @@ -30,6 +30,9 @@ TR_commEndPointRead(TR_CommEndPoint this, TR_RemoteData * data_ptr) | @@ -30,6 +30,9 @@ TR_commEndPointRead(TR_CommEndPoint this, TR_RemoteData * data_ptr) | ||
| 30 | { | 30 | { |
| 31 | *data_ptr = TR_socketRecv(this->transport, this->read_chunk_size); | 31 | *data_ptr = TR_socketRecv(this->transport, this->read_chunk_size); |
| 32 | 32 | ||
| 33 | + printf("[~DEBUG~] read %zd bytes\n", ((TR_SizedData)*data_ptr)->size); | ||
| 34 | + fflush(stdout); | ||
| 35 | + | ||
| 33 | if (! *data_ptr) return -1; // ment to trigger a close | 36 | if (! *data_ptr) return -1; // ment to trigger a close |
| 34 | if (*data_ptr == (void*)-1) return -2; // remote close... shutdown | 37 | if (*data_ptr == (void*)-1) return -2; // remote close... shutdown |
| 35 | if (*data_ptr == TR_emptyRemoteData) return FALSE; // read blocked | 38 | if (*data_ptr == TR_emptyRemoteData) return FALSE; // read blocked |
| @@ -22,6 +22,7 @@ | @@ -22,6 +22,7 @@ | ||
| 22 | 22 | ||
| 23 | #include <unistd.h> | 23 | #include <unistd.h> |
| 24 | #include <poll.h> | 24 | #include <poll.h> |
| 25 | +#include <pthread.h> | ||
| 25 | 26 | ||
| 26 | #include "trbase.h" | 27 | #include "trbase.h" |
| 27 | #include "trdata.h" | 28 | #include "trdata.h" |
| @@ -51,6 +52,8 @@ commManagerCtor(void * _this, va_list * params) | @@ -51,6 +52,8 @@ commManagerCtor(void * _this, va_list * params) | ||
| 51 | this->n_endpoints = sysconf(_SC_OPEN_MAX); | 52 | this->n_endpoints = sysconf(_SC_OPEN_MAX); |
| 52 | this->endpoints = TR_calloc(sizeof(TR_CommEndPoint), this->n_endpoints); | 53 | this->endpoints = TR_calloc(sizeof(TR_CommEndPoint), this->n_endpoints); |
| 53 | 54 | ||
| 55 | + pthread_mutex_init(&this->io_triggered_lock, NULL); | ||
| 56 | + | ||
| 54 | return 0; | 57 | return 0; |
| 55 | } | 58 | } |
| 56 | 59 | ||
| @@ -61,6 +64,8 @@ commManagerDtor(void * _this) | @@ -61,6 +64,8 @@ commManagerDtor(void * _this) | ||
| 61 | TR_CommManager this = _this; | 64 | TR_CommManager this = _this; |
| 62 | nfds_t i; | 65 | nfds_t i; |
| 63 | 66 | ||
| 67 | + pthread_mutex_destroy(&this->io_triggered_lock); | ||
| 68 | + | ||
| 64 | for (i=0; i<this->n_endpoints; i++) { | 69 | for (i=0; i<this->n_endpoints; i++) { |
| 65 | TR_delete(this->endpoints[i]); | 70 | TR_delete(this->endpoints[i]); |
| 66 | } | 71 | } |
| @@ -72,11 +77,22 @@ commManagerDtor(void * _this) | @@ -72,11 +77,22 @@ commManagerDtor(void * _this) | ||
| 72 | 77 | ||
| 73 | static | 78 | static |
| 74 | TR_EventDone | 79 | TR_EventDone |
| 75 | -TR_commManagerEnableWrite(void * _this, TR_Event event) | 80 | +TR_commManagerWriteIsBlocked(void * _this, TR_Event event) |
| 76 | { | 81 | { |
| 77 | TR_CommManager this = _this; | 82 | TR_CommManager this = _this; |
| 78 | 83 | ||
| 79 | - TR_hashAdd(this->write, event->subject); | 84 | + TR_hashDeleteByVal(this->write, TR_hashableGetHash(event->subject)); |
| 85 | + | ||
| 86 | + return TR_EVENT_DONE; | ||
| 87 | +} | ||
| 88 | + | ||
| 89 | +static | ||
| 90 | +TR_EventDone | ||
| 91 | +TR_commManagerDecrementIoTriggerd(TR_CommManager this, TR_Event event) | ||
| 92 | +{ | ||
| 93 | + pthread_mutex_lock(&this->io_triggered_lock); | ||
| 94 | + this->io_triggered--; | ||
| 95 | + pthread_mutex_unlock(&this->io_triggered_lock); | ||
| 80 | 96 | ||
| 81 | return TR_EVENT_DONE; | 97 | return TR_EVENT_DONE; |
| 82 | } | 98 | } |
| @@ -103,8 +119,6 @@ static | @@ -103,8 +119,6 @@ static | ||
| 103 | void | 119 | void |
| 104 | commManagerCvInit(TR_class_ptr cls) | 120 | commManagerCvInit(TR_class_ptr cls) |
| 105 | { | 121 | { |
| 106 | - TR_CLASSVARS(TR_EventHandler, cls)->event_methods->tree = TR_new(TR_Tree); | ||
| 107 | - | ||
| 108 | TR_EVENT_HANDLER_SET_METHOD( | 122 | TR_EVENT_HANDLER_SET_METHOD( |
| 109 | cls, TR_EventDispatcher, | 123 | cls, TR_EventDispatcher, |
| 110 | TR_DISPATCHER_EVENT_DATA_WAIT, | 124 | TR_DISPATCHER_EVENT_DATA_WAIT, |
| @@ -120,7 +134,7 @@ commManagerCvInit(TR_class_ptr cls) | @@ -120,7 +134,7 @@ commManagerCvInit(TR_class_ptr cls) | ||
| 120 | TR_EVENT_HANDLER_SET_METHOD( | 134 | TR_EVENT_HANDLER_SET_METHOD( |
| 121 | cls, TR_CommEndPoint, | 135 | cls, TR_CommEndPoint, |
| 122 | TR_CEP_EVENT_WRITE_BLOCK, | 136 | TR_CEP_EVENT_WRITE_BLOCK, |
| 123 | - TR_commManagerPollWrite); | 137 | + TR_commManagerWriteIsBlocked); |
| 124 | TR_EVENT_HANDLER_SET_METHOD( | 138 | TR_EVENT_HANDLER_SET_METHOD( |
| 125 | cls, TR_CommEndPoint, | 139 | cls, TR_CommEndPoint, |
| 126 | TR_CEP_EVENT_READ_BLOCK, | 140 | TR_CEP_EVENT_READ_BLOCK, |
| @@ -144,14 +158,18 @@ commManagerCvInit(TR_class_ptr cls) | @@ -144,14 +158,18 @@ commManagerCvInit(TR_class_ptr cls) | ||
| 144 | TR_EVENT_HANDLER_SET_METHOD( | 158 | TR_EVENT_HANDLER_SET_METHOD( |
| 145 | cls, TR_CommEndPoint, | 159 | cls, TR_CommEndPoint, |
| 146 | TR_CEP_EVENT_DATA_READY, | 160 | TR_CEP_EVENT_DATA_READY, |
| 147 | - TR_commManagerEnableWrite); | 161 | + TR_commManagerPollWrite); |
| 148 | TR_EVENT_HANDLER_SET_METHOD( | 162 | TR_EVENT_HANDLER_SET_METHOD( |
| 149 | cls, TR_CommEndPoint, | 163 | cls, TR_CommEndPoint, |
| 150 | TR_CEP_EVENT_DATA_END, | 164 | TR_CEP_EVENT_DATA_END, |
| 151 | TR_commManagerDisableWrite); | 165 | TR_commManagerDisableWrite); |
| 166 | + TR_EVENT_HANDLER_SET_METHOD( | ||
| 167 | + cls, TR_CommEndPoint, | ||
| 168 | + TR_CEP_EVENT_IO_DONE, | ||
| 169 | + TR_commManagerDecrementIoTriggerd); | ||
| 152 | } | 170 | } |
| 153 | 171 | ||
| 154 | -TR_INSTANCE(TR_Hash, commManagerEventMethods); | 172 | +TR_INIT_HANDLER(TR_CommManager); |
| 155 | TR_INIT_IFACE(TR_Class, commManagerCtor, commManagerDtor, NULL); | 173 | TR_INIT_IFACE(TR_Class, commManagerCtor, commManagerDtor, NULL); |
| 156 | TR_INIT_IFACE(TR_CommManager, NULL, NULL, NULL, NULL, NULL, NULL, NULL); | 174 | TR_INIT_IFACE(TR_CommManager, NULL, NULL, NULL, NULL, NULL, NULL, NULL); |
| 157 | TR_CREATE_CLASS( | 175 | TR_CREATE_CLASS( |
| @@ -160,7 +178,7 @@ TR_CREATE_CLASS( | @@ -160,7 +178,7 @@ TR_CREATE_CLASS( | ||
| 160 | commManagerCvInit, | 178 | commManagerCvInit, |
| 161 | TR_IF(TR_Class), | 179 | TR_IF(TR_Class), |
| 162 | TR_IF(TR_CommManager)) = { | 180 | TR_IF(TR_CommManager)) = { |
| 163 | - { &(_commManagerEventMethods.data) } | 181 | + { TR_HANDLER_CVARS(TR_CommManager) } |
| 164 | }; | 182 | }; |
| 165 | 183 | ||
| 166 | // vim: set ts=4 sw=4: | 184 | // vim: set ts=4 sw=4: |
| @@ -73,8 +73,8 @@ TR_commManagerEpollAddEndpoint(void * _this, TR_CommEndPoint endpoint) | @@ -73,8 +73,8 @@ TR_commManagerEpollAddEndpoint(void * _this, TR_CommEndPoint endpoint) | ||
| 73 | int handle = endpoint->transport->handle; | 73 | int handle = endpoint->transport->handle; |
| 74 | struct epoll_event event; | 74 | struct epoll_event event; |
| 75 | 75 | ||
| 76 | - //this->events[handle] = EPOLLET; | ||
| 77 | - this->events[handle] = 0; | 76 | + //this->events[handle] = EPOLLIN | EPOLLET; |
| 77 | + this->events[handle] = EPOLLIN; | ||
| 78 | event.data.ptr = endpoint; | 78 | event.data.ptr = endpoint; |
| 79 | event.events = this->events[handle]; | 79 | event.events = this->events[handle]; |
| 80 | 80 | ||
| @@ -82,19 +82,19 @@ TR_commManagerEpollAddEndpoint(void * _this, TR_CommEndPoint endpoint) | @@ -82,19 +82,19 @@ TR_commManagerEpollAddEndpoint(void * _this, TR_CommEndPoint endpoint) | ||
| 82 | } | 82 | } |
| 83 | 83 | ||
| 84 | static | 84 | static |
| 85 | -void | 85 | +size_t |
| 86 | TR_commManagerEpollSelect(void * _this, TR_Event event, unsigned long timeout) | 86 | TR_commManagerEpollSelect(void * _this, TR_Event event, unsigned long timeout) |
| 87 | { | 87 | { |
| 88 | TR_CommManagerEpoll this = _this; | 88 | TR_CommManagerEpoll this = _this; |
| 89 | TR_CommManager cmgr = _this; | 89 | TR_CommManager cmgr = _this; |
| 90 | int i, nevents; | 90 | int i, nevents; |
| 91 | - struct epoll_event _event; | 91 | + //struct epoll_event _event; |
| 92 | 92 | ||
| 93 | nevents = epoll_wait(this->handle, events, MAXEVENTS, timeout); | 93 | nevents = epoll_wait(this->handle, events, MAXEVENTS, timeout); |
| 94 | 94 | ||
| 95 | for (i=0; i<nevents; i++) { | 95 | for (i=0; i<nevents; i++) { |
| 96 | TR_CommEndPoint endpoint = (TR_CommEndPoint)events[i].data.ptr; | 96 | TR_CommEndPoint endpoint = (TR_CommEndPoint)events[i].data.ptr; |
| 97 | - int handle = endpoint->transport->handle; | 97 | + //int handle = endpoint->transport->handle; |
| 98 | 98 | ||
| 99 | if ((events[i].events & EPOLLIN) == EPOLLIN) { | 99 | if ((events[i].events & EPOLLIN) == EPOLLIN) { |
| 100 | if (TR_INSTANCE_OF(TR_TcpSocket, endpoint->transport) | 100 | if (TR_INSTANCE_OF(TR_TcpSocket, endpoint->transport) |
| @@ -106,20 +106,20 @@ TR_commManagerEpollSelect(void * _this, TR_Event event, unsigned long timeout) | @@ -106,20 +106,20 @@ TR_commManagerEpollSelect(void * _this, TR_Event event, unsigned long timeout) | ||
| 106 | } | 106 | } |
| 107 | } | 107 | } |
| 108 | 108 | ||
| 109 | - this->events[handle] &= ~EPOLLIN; | ||
| 110 | - _event.data.ptr = endpoint; | ||
| 111 | - _event.events = this->events[handle]; | ||
| 112 | - epoll_ctl(this->handle, EPOLL_CTL_MOD, handle, &_event); | 109 | + //this->events[handle] &= ~EPOLLIN; |
| 110 | + //_event.data.ptr = endpoint; | ||
| 111 | + //_event.events = this->events[handle]; | ||
| 112 | + //epoll_ctl(this->handle, EPOLL_CTL_MOD, handle, &_event); | ||
| 113 | } | 113 | } |
| 114 | 114 | ||
| 115 | if ((events[i].events & EPOLLOUT) == EPOLLOUT) { | 115 | if ((events[i].events & EPOLLOUT) == EPOLLOUT) { |
| 116 | if (! event->subject->fin) { | 116 | if (! event->subject->fin) { |
| 117 | TR_hashAdd(cmgr->write, endpoint); | 117 | TR_hashAdd(cmgr->write, endpoint); |
| 118 | } | 118 | } |
| 119 | - this->events[handle] &= ~EPOLLOUT; | ||
| 120 | - _event.data.ptr = endpoint; | ||
| 121 | - _event.events = this->events[handle]; | ||
| 122 | - epoll_ctl(this->handle, EPOLL_CTL_MOD, handle, &_event); | 119 | + //this->events[handle] &= ~EPOLLOUT; |
| 120 | + //_event.data.ptr = endpoint; | ||
| 121 | + //_event.events = this->events[handle]; | ||
| 122 | + //epoll_ctl(this->handle, EPOLL_CTL_MOD, handle, &_event); | ||
| 123 | } | 123 | } |
| 124 | 124 | ||
| 125 | if ((events[i].events & EPOLLHUP) == EPOLLHUP) { | 125 | if ((events[i].events & EPOLLHUP) == EPOLLHUP) { |
| @@ -131,6 +131,15 @@ TR_commManagerEpollSelect(void * _this, TR_Event event, unsigned long timeout) | @@ -131,6 +131,15 @@ TR_commManagerEpollSelect(void * _this, TR_Event event, unsigned long timeout) | ||
| 131 | NULL)); | 131 | NULL)); |
| 132 | } | 132 | } |
| 133 | } | 133 | } |
| 134 | + | ||
| 135 | + if (nevents >= 0) { | ||
| 136 | + return nevents; | ||
| 137 | + } else { | ||
| 138 | + perror("epoll"); | ||
| 139 | + fflush(stderr); | ||
| 140 | + fflush(stdout); | ||
| 141 | + return 0; | ||
| 142 | + } | ||
| 134 | } | 143 | } |
| 135 | 144 | ||
| 136 | static | 145 | static |
| @@ -180,9 +189,9 @@ static | @@ -180,9 +189,9 @@ static | ||
| 180 | void | 189 | void |
| 181 | TR_commManagerEpollEnableRead(void * _this, TR_Event event) | 190 | TR_commManagerEpollEnableRead(void * _this, TR_Event event) |
| 182 | { | 191 | { |
| 183 | - if (! TR_socketFinRd(((TR_CommEndPoint)event->subject)->transport)) { | ||
| 184 | - TR_commManagerEpollEnable(_this, EPOLLIN, event); | ||
| 185 | - } | 192 | +// if (! TR_socketFinRd(((TR_CommEndPoint)event->subject)->transport)) { |
| 193 | +// TR_commManagerEpollEnable(_this, EPOLLIN, event); | ||
| 194 | +// } | ||
| 186 | } | 195 | } |
| 187 | 196 | ||
| 188 | static | 197 | static |
| @@ -196,7 +205,7 @@ static | @@ -196,7 +205,7 @@ static | ||
| 196 | void | 205 | void |
| 197 | TR_commManagerEpollDisableRead(void * _this, TR_Event event) | 206 | TR_commManagerEpollDisableRead(void * _this, TR_Event event) |
| 198 | { | 207 | { |
| 199 | - TR_commManagerEpollDisable(_this, EPOLLIN, event); | 208 | +// TR_commManagerEpollDisable(_this, EPOLLIN, event); |
| 200 | } | 209 | } |
| 201 | 210 | ||
| 202 | static | 211 | static |
| @@ -71,45 +71,57 @@ TR_commManagerPollAddEndpoint(void * _this, TR_CommEndPoint endpoint) | @@ -71,45 +71,57 @@ TR_commManagerPollAddEndpoint(void * _this, TR_CommEndPoint endpoint) | ||
| 71 | TR_CommManagerPoll this = _this; | 71 | TR_CommManagerPoll this = _this; |
| 72 | 72 | ||
| 73 | this->fds[endpoint->transport->handle].fd = endpoint->transport->handle; | 73 | this->fds[endpoint->transport->handle].fd = endpoint->transport->handle; |
| 74 | - this->fds[endpoint->transport->handle].events = 0; | 74 | + this->fds[endpoint->transport->handle].events = POLLIN; |
| 75 | } | 75 | } |
| 76 | 76 | ||
| 77 | static | 77 | static |
| 78 | -void | 78 | +size_t |
| 79 | TR_commManagerPollSelect(void * _this, TR_Event event, unsigned long timeout) | 79 | TR_commManagerPollSelect(void * _this, TR_Event event, unsigned long timeout) |
| 80 | { | 80 | { |
| 81 | TR_CommManagerPoll this = _this; | 81 | TR_CommManagerPoll this = _this; |
| 82 | TR_CommManager cmgr = _this; | 82 | TR_CommManager cmgr = _this; |
| 83 | nfds_t i; | 83 | nfds_t i; |
| 84 | - int nevents; | 84 | + int nevents, doevents; |
| 85 | + | ||
| 86 | + for (i = 0; i < cmgr->max_handle+1; i++) { | ||
| 87 | + printf("[=DEBUG=] handle %ld POLLIN? %s\n", i, | ||
| 88 | + (this->fds[i].events & POLLIN) == POLLIN ? "YES" : "NO"); | ||
| 89 | + fflush(stdout); | ||
| 90 | + } | ||
| 85 | 91 | ||
| 86 | - nevents = poll(this->fds, cmgr->max_handle+1, timeout); | 92 | + nevents = doevents = poll(this->fds, cmgr->max_handle+1, timeout); |
| 87 | 93 | ||
| 88 | - if (nevents) { | 94 | + if (doevents) { |
| 89 | for (i = 0; i < cmgr->max_handle+1; i++) { | 95 | for (i = 0; i < cmgr->max_handle+1; i++) { |
| 90 | if (this->fds[i].revents != 0) { | 96 | if (this->fds[i].revents != 0) { |
| 91 | TR_CommEndPoint endpoint = cmgr->endpoints[i]; | 97 | TR_CommEndPoint endpoint = cmgr->endpoints[i]; |
| 92 | 98 | ||
| 93 | - nevents--; | 99 | + doevents--; |
| 94 | 100 | ||
| 95 | if ((this->fds[i].revents & POLLIN) == POLLIN) { | 101 | if ((this->fds[i].revents & POLLIN) == POLLIN) { |
| 96 | if (TR_INSTANCE_OF(TR_TcpSocket, endpoint->transport) | 102 | if (TR_INSTANCE_OF(TR_TcpSocket, endpoint->transport) |
| 97 | && ((TR_TcpSocket)endpoint->transport)->listen) { | 103 | && ((TR_TcpSocket)endpoint->transport)->listen) { |
| 104 | + pthread_mutex_lock(&cmgr->io_triggered_lock); | ||
| 98 | TR_hashAdd(cmgr->accept, endpoint); | 105 | TR_hashAdd(cmgr->accept, endpoint); |
| 106 | + pthread_mutex_unlock(&cmgr->io_triggered_lock); | ||
| 99 | } else { | 107 | } else { |
| 100 | if (! event->subject->fin) { | 108 | if (! event->subject->fin) { |
| 109 | + pthread_mutex_lock(&cmgr->io_triggered_lock); | ||
| 101 | TR_hashAdd(cmgr->read, endpoint); | 110 | TR_hashAdd(cmgr->read, endpoint); |
| 111 | + pthread_mutex_unlock(&cmgr->io_triggered_lock); | ||
| 102 | } | 112 | } |
| 103 | } | 113 | } |
| 104 | - this->fds[endpoint->transport->handle].events &= ~POLLIN; | 114 | + //this->fds[endpoint->transport->handle].events &= ~POLLIN; |
| 105 | } | 115 | } |
| 106 | 116 | ||
| 107 | if ((this->fds[i].revents & POLLOUT) == POLLOUT) { | 117 | if ((this->fds[i].revents & POLLOUT) == POLLOUT) { |
| 108 | if (! event->subject->fin) { | 118 | if (! event->subject->fin) { |
| 119 | + pthread_mutex_lock(&cmgr->io_triggered_lock); | ||
| 109 | TR_hashAdd(cmgr->write, endpoint); | 120 | TR_hashAdd(cmgr->write, endpoint); |
| 121 | + pthread_mutex_unlock(&cmgr->io_triggered_lock); | ||
| 110 | } | 122 | } |
| 111 | - this->fds[endpoint->transport->handle].events &= | ||
| 112 | - ~(POLLOUT|POLLHUP); | 123 | + //this->fds[endpoint->transport->handle].events &= |
| 124 | + // ~(POLLOUT|POLLHUP); | ||
| 113 | } | 125 | } |
| 114 | 126 | ||
| 115 | if ((this->fds[i].revents & POLLHUP) == POLLHUP) { | 127 | if ((this->fds[i].revents & POLLHUP) == POLLHUP) { |
| @@ -122,10 +134,12 @@ TR_commManagerPollSelect(void * _this, TR_Event event, unsigned long timeout) | @@ -122,10 +134,12 @@ TR_commManagerPollSelect(void * _this, TR_Event event, unsigned long timeout) | ||
| 122 | } | 134 | } |
| 123 | 135 | ||
| 124 | this->fds[i].revents = 0; | 136 | this->fds[i].revents = 0; |
| 125 | - if (nevents <= 0) break; | 137 | + if (doevents <= 0) break; |
| 126 | } | 138 | } |
| 127 | } | 139 | } |
| 128 | } | 140 | } |
| 141 | + | ||
| 142 | + return nevents; | ||
| 129 | } | 143 | } |
| 130 | 144 | ||
| 131 | static | 145 | static |
| @@ -144,12 +158,12 @@ static | @@ -144,12 +158,12 @@ static | ||
| 144 | void | 158 | void |
| 145 | TR_commManagerPollEnableRead(void * _this, TR_Event event) | 159 | TR_commManagerPollEnableRead(void * _this, TR_Event event) |
| 146 | { | 160 | { |
| 147 | - TR_CommManagerPoll this = _this; | ||
| 148 | - TR_CommEndPoint endpoint = (TR_CommEndPoint)event->subject; | ||
| 149 | - | ||
| 150 | - if (! TR_socketFinRd(endpoint->transport)) { | ||
| 151 | - this->fds[endpoint->transport->handle].events |= POLLIN; | ||
| 152 | - } | 161 | +// TR_CommManagerPoll this = _this; |
| 162 | +// TR_CommEndPoint endpoint = (TR_CommEndPoint)event->subject; | ||
| 163 | +// | ||
| 164 | +// if (! TR_socketFinRd(endpoint->transport)) { | ||
| 165 | +// this->fds[endpoint->transport->handle].events |= POLLIN; | ||
| 166 | +// } | ||
| 153 | } | 167 | } |
| 154 | 168 | ||
| 155 | static | 169 | static |
| @@ -166,10 +180,10 @@ static | @@ -166,10 +180,10 @@ static | ||
| 166 | void | 180 | void |
| 167 | TR_commManagerPollDisableRead(void * _this, TR_Event event) | 181 | TR_commManagerPollDisableRead(void * _this, TR_Event event) |
| 168 | { | 182 | { |
| 169 | - TR_CommManagerPoll this = _this; | ||
| 170 | - TR_CommEndPoint endpoint = (TR_CommEndPoint)event->subject; | ||
| 171 | - | ||
| 172 | - this->fds[endpoint->transport->handle].events &= ~POLLIN; | 183 | +// TR_CommManagerPoll this = _this; |
| 184 | +// TR_CommEndPoint endpoint = (TR_CommEndPoint)event->subject; | ||
| 185 | +// | ||
| 186 | +// this->fds[endpoint->transport->handle].events &= ~POLLIN; | ||
| 173 | } | 187 | } |
| 174 | 188 | ||
| 175 | static | 189 | static |
| @@ -37,10 +37,10 @@ static | @@ -37,10 +37,10 @@ static | ||
| 37 | int | 37 | int |
| 38 | connectionCtor(void * _this, va_list * params) | 38 | connectionCtor(void * _this, va_list * params) |
| 39 | { | 39 | { |
| 40 | - TR_Connection this = _this; | 40 | + //TR_Connection this = _this; |
| 41 | 41 | ||
| 42 | TR_PARENTCALL(TR_Connection, _this, TR_Class, ctor, params); | 42 | TR_PARENTCALL(TR_Connection, _this, TR_Class, ctor, params); |
| 43 | - this->current_message = NULL; | 43 | + //this->current_message = NULL; |
| 44 | 44 | ||
| 45 | return 0; | 45 | return 0; |
| 46 | } | 46 | } |
| @@ -76,6 +76,13 @@ connectorAccept(void * _this, TR_Event event) | @@ -76,6 +76,13 @@ connectorAccept(void * _this, TR_Event event) | ||
| 76 | socket = TR_socketAccept((TR_TcpSocket)connection->transport); | 76 | socket = TR_socketAccept((TR_TcpSocket)connection->transport); |
| 77 | } | 77 | } |
| 78 | 78 | ||
| 79 | + TR_eventHandlerIssueEvent( | ||
| 80 | + (TR_EventHandler)this, | ||
| 81 | + TR_eventSubjectEmit( | ||
| 82 | + (TR_EventSubject)connection, | ||
| 83 | + TR_CEP_EVENT_IO_DONE, | ||
| 84 | + NULL)); | ||
| 85 | + | ||
| 79 | if (! socket) { | 86 | if (! socket) { |
| 80 | TR_eventHandlerIssueEvent( | 87 | TR_eventHandlerIssueEvent( |
| 81 | (TR_EventHandler)this, | 88 | (TR_EventHandler)this, |
| @@ -92,8 +99,6 @@ static | @@ -92,8 +99,6 @@ static | ||
| 92 | void | 99 | void |
| 93 | connectorCvInit(TR_class_ptr cls) | 100 | connectorCvInit(TR_class_ptr cls) |
| 94 | { | 101 | { |
| 95 | - TR_CLASSVARS(TR_EventHandler, cls)->event_methods->tree = TR_new(TR_Tree); | ||
| 96 | - | ||
| 97 | TR_EVENT_HANDLER_SET_METHOD( | 102 | TR_EVENT_HANDLER_SET_METHOD( |
| 98 | cls, | 103 | cls, |
| 99 | TR_ConnEntryPoint, | 104 | TR_ConnEntryPoint, |
| @@ -101,6 +106,7 @@ connectorCvInit(TR_class_ptr cls) | @@ -101,6 +106,7 @@ connectorCvInit(TR_class_ptr cls) | ||
| 101 | connectorAccept); | 106 | connectorAccept); |
| 102 | } | 107 | } |
| 103 | 108 | ||
| 109 | +TR_INIT_HANDLER(TR_Connector); | ||
| 104 | TR_INSTANCE(TR_Hash, connectorEventMethods); | 110 | TR_INSTANCE(TR_Hash, connectorEventMethods); |
| 105 | TR_INIT_IFACE(TR_Class, connectorCtor, connectorDtor, NULL); | 111 | TR_INIT_IFACE(TR_Class, connectorCtor, connectorDtor, NULL); |
| 106 | TR_CREATE_CLASS( | 112 | TR_CREATE_CLASS( |
| @@ -108,7 +114,7 @@ TR_CREATE_CLASS( | @@ -108,7 +114,7 @@ TR_CREATE_CLASS( | ||
| 108 | TR_EventHandler, | 114 | TR_EventHandler, |
| 109 | connectorCvInit, | 115 | connectorCvInit, |
| 110 | TR_IF(TR_Class)) = { | 116 | TR_IF(TR_Class)) = { |
| 111 | - { &(_connectorEventMethods.data) } | 117 | + { TR_HANDLER_CVARS(TR_Connector) } |
| 112 | }; | 118 | }; |
| 113 | 119 | ||
| 114 | // vim: set ts=4 sw=4: | 120 | // vim: set ts=4 sw=4: |
| @@ -20,8 +20,11 @@ | @@ -20,8 +20,11 @@ | ||
| 20 | * along with this program. If not, see <http://www.gnu.org/licenses/>. | 20 | * along with this program. If not, see <http://www.gnu.org/licenses/>. |
| 21 | */ | 21 | */ |
| 22 | 22 | ||
| 23 | +#define _GNU_SOURCE | ||
| 24 | + | ||
| 23 | #include <errno.h> | 25 | #include <errno.h> |
| 24 | #include <poll.h> | 26 | #include <poll.h> |
| 27 | +#include <pthread.h> | ||
| 25 | 28 | ||
| 26 | #include "trbase.h" | 29 | #include "trbase.h" |
| 27 | #include "trevent.h" | 30 | #include "trevent.h" |
| @@ -59,6 +62,8 @@ TR_commManagerAddEndpoint(void * _this, TR_CommEndPoint endpoint) | @@ -59,6 +62,8 @@ TR_commManagerAddEndpoint(void * _this, TR_CommEndPoint endpoint) | ||
| 59 | TR_ISSUE_IO_READ_EVENT(this, endpoint); | 62 | TR_ISSUE_IO_READ_EVENT(this, endpoint); |
| 60 | } | 63 | } |
| 61 | 64 | ||
| 65 | + this->io_triggered++; | ||
| 66 | + | ||
| 62 | TR_CALL(_this, TR_CommManager, addEndpoint, endpoint); | 67 | TR_CALL(_this, TR_CommManager, addEndpoint, endpoint); |
| 63 | } | 68 | } |
| 64 | 69 | ||
| @@ -90,22 +95,35 @@ TR_commManagerSelect(void * _this, TR_Event event) | @@ -90,22 +95,35 @@ TR_commManagerSelect(void * _this, TR_Event event) | ||
| 90 | TR_Timer timer = (TR_Timer)event->data; | 95 | TR_Timer timer = (TR_Timer)event->data; |
| 91 | TR_EventDispatcher dispatcher = (TR_EventDispatcher)event->subject; | 96 | TR_EventDispatcher dispatcher = (TR_EventDispatcher)event->subject; |
| 92 | unsigned long timeout; // milliseconds | 97 | unsigned long timeout; // milliseconds |
| 93 | - | ||
| 94 | - if (! (TR_hashEmpty(this->read) | ||
| 95 | - && TR_hashEmpty(this->write) | ||
| 96 | - && TR_hashEmpty(this->accept))) { | ||
| 97 | - timeout = 0; | ||
| 98 | - } else if (NULL == timer) { | ||
| 99 | - timeout = TR_eventDispatcherGetDataWaitTime(dispatcher); | ||
| 100 | - } else { | ||
| 101 | - timeout = TR_timerGet(timer, NULL); | 98 | + char buffer[17]; |
| 99 | + | ||
| 100 | + pthread_getname_np(pthread_self(), buffer, 17); | ||
| 101 | + | ||
| 102 | + if (! this->io_triggered) { | ||
| 103 | + printf("[DEBUG] [%s] io triggerd was empty\n", buffer); | ||
| 104 | + fflush(stdout); | ||
| 105 | + pthread_mutex_lock(&this->io_triggered_lock); | ||
| 106 | + this->io_triggered = TR_hashEach(this->write, this, commManagerIssueWriteEvents); | ||
| 107 | + this->io_triggered += TR_hashEach(this->accept, this, commManagerIssueAcceptEvents); | ||
| 108 | + this->io_triggered += TR_hashEach(this->read, this, commManagerIssueReadEvents); | ||
| 109 | + pthread_mutex_unlock(&this->io_triggered_lock); | ||
| 102 | } | 110 | } |
| 103 | 111 | ||
| 104 | - TR_CALL(_this, TR_CommManager, select, event, timeout); | 112 | + printf("[DEBUG] [%s] io triggerd: %lu\n", buffer, this->io_triggered); |
| 113 | + fflush(stdout); | ||
| 114 | + | ||
| 115 | + if (! this->io_triggered) { | ||
| 116 | + if (NULL == timer) { | ||
| 117 | + timeout = TR_eventDispatcherGetDataWaitTime(dispatcher); | ||
| 118 | + } else { | ||
| 119 | + timeout = TR_timerGet(timer, NULL); | ||
| 120 | + } | ||
| 105 | 121 | ||
| 106 | - TR_hashEach(this->write, this, commManagerIssueWriteEvents); | ||
| 107 | - TR_hashEach(this->accept, this, commManagerIssueAcceptEvents); | ||
| 108 | - TR_hashEach(this->read, this, commManagerIssueReadEvents); | 122 | + printf("[DEBUG] [%s] select timeout: %lu\n", buffer, timeout); |
| 123 | + fflush(stdout); | ||
| 124 | + | ||
| 125 | + TR_CALL(_this, TR_CommManager, select, event, timeout); | ||
| 126 | + } | ||
| 109 | 127 | ||
| 110 | return TR_EVENT_DONE; | 128 | return TR_EVENT_DONE; |
| 111 | } | 129 | } |
| @@ -115,9 +133,13 @@ TR_commManagerPollWrite(void * _this, TR_Event event) | @@ -115,9 +133,13 @@ TR_commManagerPollWrite(void * _this, TR_Event event) | ||
| 115 | { | 133 | { |
| 116 | TR_CommManager this = _this; | 134 | TR_CommManager this = _this; |
| 117 | 135 | ||
| 118 | - TR_hashDeleteByVal(this->write, TR_hashableGetHash(event->subject)); | ||
| 119 | if (! TR_socketFinWr(((TR_CommEndPoint)event->subject)->transport)) { | 136 | if (! TR_socketFinWr(((TR_CommEndPoint)event->subject)->transport)) { |
| 137 | + pthread_mutex_lock(&this->io_triggered_lock); | ||
| 138 | + TR_hashAdd(this->write, event->subject); | ||
| 120 | TR_CALL(_this, TR_CommManager, pollWrite, event); | 139 | TR_CALL(_this, TR_CommManager, pollWrite, event); |
| 140 | + pthread_mutex_unlock(&this->io_triggered_lock); | ||
| 141 | + printf("[!DEBUG!] socket added to write hash\n"); | ||
| 142 | + fflush(stdout); | ||
| 121 | } | 143 | } |
| 122 | 144 | ||
| 123 | return TR_EVENT_DONE; | 145 | return TR_EVENT_DONE; |
| @@ -129,6 +151,10 @@ TR_commManagerPollRead(void * _this, TR_Event event) | @@ -129,6 +151,10 @@ TR_commManagerPollRead(void * _this, TR_Event event) | ||
| 129 | TR_CommManager this = _this; | 151 | TR_CommManager this = _this; |
| 130 | TR_CommEndPoint endpoint = (TR_CommEndPoint)event->subject; | 152 | TR_CommEndPoint endpoint = (TR_CommEndPoint)event->subject; |
| 131 | 153 | ||
| 154 | + if (! TR_socketFinRd(endpoint->transport)) { | ||
| 155 | + TR_CALL(_this, TR_CommManager, pollRead, event); | ||
| 156 | + } | ||
| 157 | + | ||
| 132 | if (TR_INSTANCE_OF(TR_TcpSocket, endpoint->transport) | 158 | if (TR_INSTANCE_OF(TR_TcpSocket, endpoint->transport) |
| 133 | && ((TR_TcpSocket)endpoint->transport)->listen) { | 159 | && ((TR_TcpSocket)endpoint->transport)->listen) { |
| 134 | TR_hashDeleteByVal(this->accept, TR_hashableGetHash(event->subject)); | 160 | TR_hashDeleteByVal(this->accept, TR_hashableGetHash(event->subject)); |
| @@ -136,10 +162,6 @@ TR_commManagerPollRead(void * _this, TR_Event event) | @@ -136,10 +162,6 @@ TR_commManagerPollRead(void * _this, TR_Event event) | ||
| 136 | TR_hashDeleteByVal(this->read, TR_hashableGetHash(event->subject)); | 162 | TR_hashDeleteByVal(this->read, TR_hashableGetHash(event->subject)); |
| 137 | } | 163 | } |
| 138 | 164 | ||
| 139 | - if (! TR_socketFinRd(endpoint->transport)) { | ||
| 140 | - TR_CALL(_this, TR_CommManager, pollRead, event); | ||
| 141 | - } | ||
| 142 | - | ||
| 143 | return TR_EVENT_DONE; | 165 | return TR_EVENT_DONE; |
| 144 | } | 166 | } |
| 145 | 167 | ||
| @@ -156,13 +178,17 @@ TR_commManagerDisableRead(void * _this, TR_Event event) | @@ -156,13 +178,17 @@ TR_commManagerDisableRead(void * _this, TR_Event event) | ||
| 156 | TR_EventDone | 178 | TR_EventDone |
| 157 | TR_commManagerDisableWrite(void * _this, TR_Event event) | 179 | TR_commManagerDisableWrite(void * _this, TR_Event event) |
| 158 | { | 180 | { |
| 159 | - TR_CommManager this = _this; | 181 | + TR_CommManager this = _this; |
| 182 | + TR_CommEndPoint endpoint = (TR_CommEndPoint)event->subject; | ||
| 160 | 183 | ||
| 161 | - TR_hashDeleteByVal(this->write, TR_hashableGetHash(event->subject)); | ||
| 162 | - if (! event->subject->fin) { | ||
| 163 | - TR_hashAdd(this->read, event->subject); | 184 | + if (! endpoint->write_buffer->nmsg) { |
| 185 | + // TODO think about a better way... | ||
| 186 | + TR_hashDeleteByVal(this->write, TR_hashableGetHash(event->subject)); | ||
| 187 | + //if (! event->subject->fin) { | ||
| 188 | + // TR_hashAdd(this->read, event->subject); | ||
| 189 | + //} | ||
| 190 | + TR_CALL(_this, TR_CommManager, disableWrite, event); | ||
| 164 | } | 191 | } |
| 165 | - TR_CALL(_this, TR_CommManager, disableWrite, event); | ||
| 166 | 192 | ||
| 167 | return TR_EVENT_DONE; | 193 | return TR_EVENT_DONE; |
| 168 | } | 194 | } |
| @@ -50,6 +50,7 @@ ioHandlerRead(void * _this, TR_Event event) | @@ -50,6 +50,7 @@ ioHandlerRead(void * _this, TR_Event event) | ||
| 50 | TR_CommEndPoint endpoint = (TR_CommEndPoint)event->subject; | 50 | TR_CommEndPoint endpoint = (TR_CommEndPoint)event->subject; |
| 51 | TR_Event revent; | 51 | TR_Event revent; |
| 52 | TR_RemoteData data; | 52 | TR_RemoteData data; |
| 53 | + char ip[16]; | ||
| 53 | 54 | ||
| 54 | switch (TR_commEndPointRead(endpoint, &data)) { | 55 | switch (TR_commEndPointRead(endpoint, &data)) { |
| 55 | case FALSE: // EAGAIN | 56 | case FALSE: // EAGAIN |
| @@ -83,6 +84,13 @@ ioHandlerRead(void * _this, TR_Event event) | @@ -83,6 +84,13 @@ ioHandlerRead(void * _this, TR_Event event) | ||
| 83 | return TR_EVENT_DONE; | 84 | return TR_EVENT_DONE; |
| 84 | } | 85 | } |
| 85 | 86 | ||
| 87 | + TR_socketAddrIpStr(data->remote, ip, 16); | ||
| 88 | + printf( | ||
| 89 | + "DEBUG: remote ip: %s / port: %d\n", | ||
| 90 | + ip, | ||
| 91 | + TR_socketAddrPort(data->remote)); | ||
| 92 | + fflush(stdout); | ||
| 93 | + | ||
| 86 | revent = TR_eventSubjectEmit( | 94 | revent = TR_eventSubjectEmit( |
| 87 | event->subject, | 95 | event->subject, |
| 88 | TR_CEP_EVENT_NEW_DATA, | 96 | TR_CEP_EVENT_NEW_DATA, |
| @@ -90,6 +98,13 @@ ioHandlerRead(void * _this, TR_Event event) | @@ -90,6 +98,13 @@ ioHandlerRead(void * _this, TR_Event event) | ||
| 90 | break; | 98 | break; |
| 91 | } | 99 | } |
| 92 | 100 | ||
| 101 | + TR_eventHandlerIssueEvent( | ||
| 102 | + (TR_EventHandler)_this, | ||
| 103 | + TR_eventSubjectEmit( | ||
| 104 | + event->subject, | ||
| 105 | + TR_CEP_EVENT_IO_DONE, | ||
| 106 | + NULL)); | ||
| 107 | + | ||
| 93 | TR_eventHandlerIssueEvent((TR_EventHandler)_this, revent); | 108 | TR_eventHandlerIssueEvent((TR_EventHandler)_this, revent); |
| 94 | 109 | ||
| 95 | return TR_EVENT_DONE; | 110 | return TR_EVENT_DONE; |
| @@ -147,6 +162,13 @@ ioHandlerWrite(void * _this, TR_Event event) | @@ -147,6 +162,13 @@ ioHandlerWrite(void * _this, TR_Event event) | ||
| 147 | 162 | ||
| 148 | endpoint->write_buffer_size -= written; | 163 | endpoint->write_buffer_size -= written; |
| 149 | 164 | ||
| 165 | + TR_eventHandlerIssueEvent( | ||
| 166 | + (TR_EventHandler)_this, | ||
| 167 | + TR_eventSubjectEmit( | ||
| 168 | + event->subject, | ||
| 169 | + TR_CEP_EVENT_IO_DONE, | ||
| 170 | + NULL)); | ||
| 171 | + | ||
| 150 | if (revent) { | 172 | if (revent) { |
| 151 | TR_eventHandlerIssueEvent((TR_EventHandler)_this, revent); | 173 | TR_eventHandlerIssueEvent((TR_EventHandler)_this, revent); |
| 152 | } | 174 | } |
| @@ -158,8 +180,6 @@ static | @@ -158,8 +180,6 @@ static | ||
| 158 | void | 180 | void |
| 159 | ioHandlerCvInit(TR_class_ptr cls) | 181 | ioHandlerCvInit(TR_class_ptr cls) |
| 160 | { | 182 | { |
| 161 | - TR_CLASSVARS(TR_EventHandler, cls)->event_methods->tree = TR_new(TR_Tree); | ||
| 162 | - | ||
| 163 | TR_EVENT_HANDLER_SET_METHOD( | 183 | TR_EVENT_HANDLER_SET_METHOD( |
| 164 | cls, | 184 | cls, |
| 165 | TR_CommEndPoint, | 185 | TR_CommEndPoint, |
| @@ -172,14 +192,14 @@ ioHandlerCvInit(TR_class_ptr cls) | @@ -172,14 +192,14 @@ ioHandlerCvInit(TR_class_ptr cls) | ||
| 172 | ioHandlerWrite); | 192 | ioHandlerWrite); |
| 173 | } | 193 | } |
| 174 | 194 | ||
| 175 | -TR_INSTANCE(TR_Hash, ioHandlerEventMethods); | 195 | +TR_INIT_HANDLER(TR_IoHandler); |
| 176 | TR_INIT_IFACE(TR_Class, ioHandlerCtor, ioHandlerDtor, NULL); | 196 | TR_INIT_IFACE(TR_Class, ioHandlerCtor, ioHandlerDtor, NULL); |
| 177 | TR_CREATE_CLASS( | 197 | TR_CREATE_CLASS( |
| 178 | TR_IoHandler, | 198 | TR_IoHandler, |
| 179 | TR_EventHandler, | 199 | TR_EventHandler, |
| 180 | ioHandlerCvInit, | 200 | ioHandlerCvInit, |
| 181 | TR_IF(TR_Class)) = { | 201 | TR_IF(TR_Class)) = { |
| 182 | - { &(_ioHandlerEventMethods.data) } | 202 | + { TR_HANDLER_CVARS(TR_IoHandler) } |
| 183 | }; | 203 | }; |
| 184 | 204 | ||
| 185 | // vim: set ts=4 sw=4: | 205 | // vim: set ts=4 sw=4: |
| @@ -125,8 +125,6 @@ static | @@ -125,8 +125,6 @@ static | ||
| 125 | void | 125 | void |
| 126 | protocolHandlerCvInit(TR_class_ptr cls) | 126 | protocolHandlerCvInit(TR_class_ptr cls) |
| 127 | { | 127 | { |
| 128 | - TR_CLASSVARS(TR_EventHandler, cls)->event_methods->tree = TR_new(TR_Tree); | ||
| 129 | - | ||
| 130 | TR_EVENT_HANDLER_SET_METHOD( | 128 | TR_EVENT_HANDLER_SET_METHOD( |
| 131 | cls, | 129 | cls, |
| 132 | TR_CommEndPoint, | 130 | TR_CommEndPoint, |
| @@ -144,14 +142,14 @@ protocolHandlerCvInit(TR_class_ptr cls) | @@ -144,14 +142,14 @@ protocolHandlerCvInit(TR_class_ptr cls) | ||
| 144 | // protocolHandlerUpgrade); | 142 | // protocolHandlerUpgrade); |
| 145 | } | 143 | } |
| 146 | 144 | ||
| 147 | -TR_INSTANCE(TR_Hash, protocolHandlerEventMethods); | 145 | +TR_INIT_HANDLER(TR_ProtocolHandler); |
| 148 | TR_INIT_IFACE(TR_Class, protocolHandlerCtor, protocolHandlerDtor, NULL); | 146 | TR_INIT_IFACE(TR_Class, protocolHandlerCtor, protocolHandlerDtor, NULL); |
| 149 | TR_CREATE_CLASS( | 147 | TR_CREATE_CLASS( |
| 150 | TR_ProtocolHandler, | 148 | TR_ProtocolHandler, |
| 151 | TR_EventHandler, | 149 | TR_EventHandler, |
| 152 | protocolHandlerCvInit, | 150 | protocolHandlerCvInit, |
| 153 | TR_IF(TR_Class)) = { | 151 | TR_IF(TR_Class)) = { |
| 154 | - { &(_protocolHandlerEventMethods.data) } | 152 | + { TR_HANDLER_CVARS(TR_ProtocolHandler) } |
| 155 | }; | 153 | }; |
| 156 | 154 | ||
| 157 | // vim: set ts=4 sw=4: | 155 | // vim: set ts=4 sw=4: |
| @@ -43,7 +43,7 @@ serverCtor(void * _this, va_list * params) | @@ -43,7 +43,7 @@ serverCtor(void * _this, va_list * params) | ||
| 43 | { | 43 | { |
| 44 | TR_Server this = _this; | 44 | TR_Server this = _this; |
| 45 | 45 | ||
| 46 | -#if 1 | 46 | +#if 0 |
| 47 | this->comm_manager = (TR_CommManager)TR_new(TR_CommManagerEpoll); | 47 | this->comm_manager = (TR_CommManager)TR_new(TR_CommManagerEpoll); |
| 48 | #else | 48 | #else |
| 49 | this->comm_manager = (TR_CommManager)TR_new(TR_CommManagerPoll); | 49 | this->comm_manager = (TR_CommManager)TR_new(TR_CommManagerPoll); |
| @@ -135,8 +135,6 @@ static | @@ -135,8 +135,6 @@ static | ||
| 135 | void | 135 | void |
| 136 | simpleClientCvInit(TR_class_ptr cls) | 136 | simpleClientCvInit(TR_class_ptr cls) |
| 137 | { | 137 | { |
| 138 | - TR_CLASSVARS(TR_EventHandler, cls)->event_methods->tree = TR_new(TR_Tree); | ||
| 139 | - | ||
| 140 | TR_EVENT_HANDLER_SET_METHOD( | 138 | TR_EVENT_HANDLER_SET_METHOD( |
| 141 | cls, | 139 | cls, |
| 142 | TR_EventDispatcher, | 140 | TR_EventDispatcher, |
| @@ -149,14 +147,14 @@ simpleClientCvInit(TR_class_ptr cls) | @@ -149,14 +147,14 @@ simpleClientCvInit(TR_class_ptr cls) | ||
| 149 | simpleClientHandleData); | 147 | simpleClientHandleData); |
| 150 | } | 148 | } |
| 151 | 149 | ||
| 152 | -TR_INSTANCE(TR_Hash, simpleClientEventMethods); | 150 | +TR_INIT_HANDLER(TR_SimpleClient); |
| 153 | TR_INIT_IFACE(TR_Class, simpleClientCtor, simpleClientDtor, NULL); | 151 | TR_INIT_IFACE(TR_Class, simpleClientCtor, simpleClientDtor, NULL); |
| 154 | TR_CREATE_CLASS( | 152 | TR_CREATE_CLASS( |
| 155 | TR_SimpleClient, | 153 | TR_SimpleClient, |
| 156 | TR_EventHandler, | 154 | TR_EventHandler, |
| 157 | simpleClientCvInit, | 155 | simpleClientCvInit, |
| 158 | TR_IF(TR_Class)) = { | 156 | TR_IF(TR_Class)) = { |
| 159 | - { &(_simpleClientEventMethods.data) } | 157 | + { TR_HANDLER_CVARS(TR_SimpleClient) } |
| 160 | }; | 158 | }; |
| 161 | 159 | ||
| 162 | // vim: set ts=4 sw=4: | 160 | // vim: set ts=4 sw=4: |
| @@ -36,6 +36,7 @@ threadedServerCtor(void * _this, va_list * params) | @@ -36,6 +36,7 @@ threadedServerCtor(void * _this, va_list * params) | ||
| 36 | { | 36 | { |
| 37 | TR_ThreadedServer this = _this; | 37 | TR_ThreadedServer this = _this; |
| 38 | int i; | 38 | int i; |
| 39 | + char buffer[16]; | ||
| 39 | 40 | ||
| 40 | TR_PARENTCALL(TR_ThreadedServer, _this, TR_Class, ctor, params); | 41 | TR_PARENTCALL(TR_ThreadedServer, _this, TR_Class, ctor, params); |
| 41 | 42 | ||
| @@ -43,9 +44,11 @@ threadedServerCtor(void * _this, va_list * params) | @@ -43,9 +44,11 @@ threadedServerCtor(void * _this, va_list * params) | ||
| 43 | this->threads = TR_malloc(sizeof(TR_EventThread) * this->n_threads); | 44 | this->threads = TR_malloc(sizeof(TR_EventThread) * this->n_threads); |
| 44 | 45 | ||
| 45 | for (i=0; i<this->n_threads; i++) { | 46 | for (i=0; i<this->n_threads; i++) { |
| 47 | + sprintf(buffer, "test%03d", i); | ||
| 46 | this->threads[i] = TR_new( | 48 | this->threads[i] = TR_new( |
| 47 | TR_EventThread, | 49 | TR_EventThread, |
| 48 | - ((TR_Server)this)->dispatcher); | 50 | + ((TR_Server)this)->dispatcher, |
| 51 | + buffer); | ||
| 49 | } | 52 | } |
| 50 | 53 | ||
| 51 | return 0; | 54 | return 0; |
| 1 | #!/bin/bash | 1 | #!/bin/bash |
| 2 | #TRLIBS="-ltrbase -ltrhashing -ltrio -ltrdata -ltrevent -ltrcomm" | 2 | #TRLIBS="-ltrbase -ltrhashing -ltrio -ltrdata -ltrevent -ltrcomm" |
| 3 | 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" | 3 | 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" |
| 4 | -LIBS="-lcrypto -lssl -lrt -luuid" | 4 | +LIBS="-lcrypto -lssl -lrt -luuid -lpthread" |
| 5 | gcc ${CFLAGS} -c -o test_handler.o test_handler.c | 5 | gcc ${CFLAGS} -c -o test_handler.o test_handler.c |
| 6 | gcc ${CFLAGS} -I/usr/local/include -L/usr/local/lib ${LIBS} -o testserver testserver.c test_handler.o ${TRLIBS} | 6 | gcc ${CFLAGS} -I/usr/local/include -L/usr/local/lib ${LIBS} -o testserver testserver.c test_handler.o ${TRLIBS} |
| 7 | gcc ${CFLAGS} -I/usr/local/include -L/usr/local/lib ${LIBS} -o testserver2 testserver2.c test_handler.o ${TRLIBS} | 7 | gcc ${CFLAGS} -I/usr/local/include -L/usr/local/lib ${LIBS} -o testserver2 testserver2.c test_handler.o ${TRLIBS} |
| 8 | gcc ${CFLAGS} -I/usr/local/include -L/usr/local/lib ${LIBS} -o testtcp testclient.c ${TRLIBS} | 8 | gcc ${CFLAGS} -I/usr/local/include -L/usr/local/lib ${LIBS} -o testtcp testclient.c ${TRLIBS} |
| 9 | gcc ${CFLAGS} -I/usr/local/include -L/usr/local/lib ${LIBS} -DUDP=1 -o testudp testclient.c ${TRLIBS} | 9 | gcc ${CFLAGS} -I/usr/local/include -L/usr/local/lib ${LIBS} -DUDP=1 -o testudp testclient.c ${TRLIBS} |
| 10 | -gcc ${CFLAGS} -I/usr/local/include -L/usr/local/lib ${LIBS} -lpthread -o testserver_thread testserver_thread.c test_handler.o ${TRLIBS} | 10 | +gcc ${CFLAGS} -I/usr/local/include -L/usr/local/lib ${LIBS} -o testserver_thread testserver_thread.c test_handler.o ${TRLIBS} |
| @@ -15,7 +15,7 @@ testHandlerHeartbeat(TR_EventHandler this, TR_Event event) | @@ -15,7 +15,7 @@ testHandlerHeartbeat(TR_EventHandler this, TR_Event event) | ||
| 15 | double size_msg = ((TestHandler)this)->size | 15 | double size_msg = ((TestHandler)this)->size |
| 16 | ? size / ((TestHandler)this)->handled | 16 | ? size / ((TestHandler)this)->handled |
| 17 | : 0.0; | 17 | : 0.0; |
| 18 | - int div_count = 0; | 18 | + int div_count = ' '; |
| 19 | 19 | ||
| 20 | while (size > 1024. && div_count != 'G') { | 20 | while (size > 1024. && div_count != 'G') { |
| 21 | size /= 1024.; | 21 | size /= 1024.; |
| @@ -31,6 +31,7 @@ testHandlerHeartbeat(TR_EventHandler this, TR_Event event) | @@ -31,6 +31,7 @@ testHandlerHeartbeat(TR_EventHandler this, TR_Event event) | ||
| 31 | ((TR_EventDispatcher)event->subject)->n_beats, | 31 | ((TR_EventDispatcher)event->subject)->n_beats, |
| 32 | ((TestHandler)this)->handled, | 32 | ((TestHandler)this)->handled, |
| 33 | size, div_count, size_msg); | 33 | size, div_count, size_msg); |
| 34 | + fflush(stdout); | ||
| 34 | ((TestHandler)this)->handled = 0; | 35 | ((TestHandler)this)->handled = 0; |
| 35 | ((TestHandler)this)->size = 0; | 36 | ((TestHandler)this)->size = 0; |
| 36 | 37 |
| @@ -30,7 +30,7 @@ main (int argc, char * argv[]) | @@ -30,7 +30,7 @@ main (int argc, char * argv[]) | ||
| 30 | TR_ProtoMessageRaw message; | 30 | TR_ProtoMessageRaw message; |
| 31 | int i, j=0; | 31 | int i, j=0; |
| 32 | 32 | ||
| 33 | - TR_logger = TR_INSTANCE_CAST(TR_Logger, mylogger2); | 33 | + //TR_logger = TR_INSTANCE_CAST(TR_Logger, mylogger2); |
| 34 | 34 | ||
| 35 | protocol = TR_new(TR_ProtocolRaw); | 35 | protocol = TR_new(TR_ProtocolRaw); |
| 36 | #if UDP | 36 | #if UDP |
| @@ -19,7 +19,7 @@ main (int argc, char * argv[]) | @@ -19,7 +19,7 @@ main (int argc, char * argv[]) | ||
| 19 | TR_Protocol protocol = TR_new(TR_ProtocolRaw); | 19 | TR_Protocol protocol = TR_new(TR_ProtocolRaw); |
| 20 | TestHandler test_handler = TR_new(TestHandler); | 20 | TestHandler test_handler = TR_new(TestHandler); |
| 21 | 21 | ||
| 22 | - TR_logger = TR_INSTANCE_CAST(TR_Logger, mylogger2); | 22 | + //TR_logger = TR_INSTANCE_CAST(TR_Logger, mylogger2); |
| 23 | 23 | ||
| 24 | TR_serverAddHandler(server, (TR_EventHandler)test_handler); | 24 | TR_serverAddHandler(server, (TR_EventHandler)test_handler); |
| 25 | TR_serverBindTcp(server, "0.0.0.0", 5678, protocol); | 25 | TR_serverBindTcp(server, "0.0.0.0", 5678, protocol); |
Please
register
or
login
to post a comment