diff options
Diffstat (limited to 'branches/1.0/tinyNET/src/tnet_transport_win32.c')
| -rw-r--r-- | branches/1.0/tinyNET/src/tnet_transport_win32.c | 786 |
1 files changed, 0 insertions, 786 deletions
diff --git a/branches/1.0/tinyNET/src/tnet_transport_win32.c b/branches/1.0/tinyNET/src/tnet_transport_win32.c deleted file mode 100644 index baa745d..0000000 --- a/branches/1.0/tinyNET/src/tnet_transport_win32.c +++ /dev/null @@ -1,786 +0,0 @@ -/* -* Copyright (C) 2009-2010 Mamadou Diop. -* -* Contact: Mamadou Diop <diopmamadou(at)doubango.org> -* -* This file is part of Open Source Doubango Framework. -* -* DOUBANGO 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. -* -* DOUBANGO 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 DOUBANGO. -* -*/ - -/**@file tnet_transport_win32.c - * @brief Network transport layer for WIN32(XP/Vista/7) and WINCE(5.0 or higher) systems. - * - * @author Mamadou Diop <diopmamadou(at)doubango.org> - * - * @date Created: Sat Nov 8 16:54:58 2009 mdiop - */ -#include "tnet_transport.h" -#include "tsk_memory.h" -#include "tsk_string.h" -#include "tsk_debug.h" -#include "tsk_thread.h" -#include "tsk_buffer.h" -#include "tsk_safeobj.h" - -#if TNET_UNDER_WINDOWS && !TNET_USE_POLL - -/*== Socket description ==*/ -typedef struct transport_socket_s -{ - tnet_fd_t fd; - unsigned owner:1; - unsigned connected:1; - unsigned paused:1; - - tnet_socket_type_t type; - tnet_tls_socket_handle_t* tlshandle; -} -transport_socket_t; - -/*== Transport context structure definition ==*/ -typedef struct transport_context_s -{ - TSK_DECLARE_OBJECT; - - tsk_size_t count; - WSAEVENT events[WSA_MAXIMUM_WAIT_EVENTS]; - transport_socket_t* sockets[WSA_MAXIMUM_WAIT_EVENTS]; - - TSK_DECLARE_SAFEOBJ; -} -transport_context_t; - -static transport_socket_t* getSocket(transport_context_t *context, tnet_fd_t fd); -static int addSocket(tnet_fd_t fd, tnet_socket_type_t type, tnet_transport_t *transport, int take_ownership, int is_client); -static int removeSocket(int index, transport_context_t *context); - -/* Checks if socket is connected */ -int tnet_transport_isconnected(const tnet_transport_handle_t *handle, tnet_fd_t fd) -{ - tnet_transport_t *transport = (tnet_transport_t*)handle; - transport_context_t *context; - tsk_size_t i; - - if(!transport) - { - TSK_DEBUG_ERROR("Invalid server handle."); - return 0; - } - - context = (transport_context_t*)transport->context; - for(i=0; i<context->count; i++) - { - const transport_socket_t* socket = context->sockets[i]; - if(socket->fd == fd){ - return socket->connected; - } - } - - return 0; -} - -int tnet_transport_have_socket(const tnet_transport_handle_t *handle, tnet_fd_t fd) -{ - tnet_transport_t *transport = (tnet_transport_t*)handle; - - if(!transport){ - TSK_DEBUG_ERROR("Invalid server handle."); - return 0; - } - - return (getSocket((transport_context_t*)transport->context, fd) != 0); -} - -const tnet_tls_socket_handle_t* tnet_transport_get_tlshandle(const tnet_transport_handle_t *handle, tnet_fd_t fd) -{ - tnet_transport_t *transport = (tnet_transport_t*)handle; - transport_socket_t *socket; - - if(!transport){ - TSK_DEBUG_ERROR("Invalid server handle."); - return 0; - } - - if((socket = getSocket((transport_context_t*)transport->context, fd))){ - return socket->tlshandle; - } - return 0; -} - -/* -* Add new socket to the watcher. -*/ -int tnet_transport_add_socket(const tnet_transport_handle_t *handle, tnet_fd_t fd, tnet_socket_type_t type, tsk_bool_t take_ownership, tsk_bool_t isClient) -{ - tnet_transport_t *transport = (tnet_transport_t*)handle; - transport_context_t* context; - int ret = -1; - - if(!transport){ - TSK_DEBUG_ERROR("Invalid server handle."); - return ret; - } - - if(!(context = (transport_context_t*)transport->context)){ - TSK_DEBUG_ERROR("Invalid context."); - return -2; - } - - if(TNET_SOCKET_TYPE_IS_TLS(type)){ - transport->tls.have_tls = 1; - } - - addSocket(fd, type, transport, take_ownership, isClient); - if(WSAEventSelect(fd, context->events[context->count - 1], FD_ALL_EVENTS) == SOCKET_ERROR){ - removeSocket((context->count - 1), context); - TNET_PRINT_LAST_ERROR("WSAEventSelect have failed."); - return -1; - } - - /* Signal */ - if(WSASetEvent(context->events[0])){ - TSK_DEBUG_INFO("New socket added to the network transport."); - return 0; - } - - // ... - - return -1; -} - -int tnet_transport_pause_socket(const tnet_transport_handle_t *handle, tnet_fd_t fd, tsk_bool_t pause) -{ - tnet_transport_t *transport = (tnet_transport_t*)handle; - transport_context_t *context; - transport_socket_t* socket; - - if(!transport || !(context = (transport_context_t*)transport->context)){ - TSK_DEBUG_ERROR("Invalid parameter"); - return -1; - } - - if((socket = getSocket(context, fd))){ - socket->paused = pause; - } - else{ - TSK_DEBUG_WARN("Socket does not exist in this context"); - } - return 0; -} - -/* Remove socket -*/ -int tnet_transport_remove_socket(const tnet_transport_handle_t *handle, tnet_fd_t* fd) -{ - tnet_transport_t *transport = (tnet_transport_t*)handle; - transport_context_t *context; - int ret = -1; - tsk_size_t i; - tsk_bool_t found = tsk_false; - - if(!transport || !(context = (transport_context_t*)transport->context)){ - TSK_DEBUG_ERROR("Invalid parameter"); - return -1; - } - - for(i=0; i<context->count; i++){ - if(context->sockets[i]->fd == *fd){ - removeSocket(i, context); - found = tsk_true; - *fd = TNET_INVALID_FD; - break; - } - } - - if(found){ - /* Signal */ - if(WSASetEvent(context->events[0])){ - TSK_DEBUG_INFO("Old socket removed from the network transport."); - return 0; - } - } - - // ... - - return -1; -} - -/* -* Sends stream/dgram data to the remote peer (shall be previously connected using @tnet_transport_connectto). -*/ -tsk_size_t tnet_transport_send(const tnet_transport_handle_t *handle, tnet_fd_t from, const void* buf, tsk_size_t size) -{ - tnet_transport_t *transport = (tnet_transport_t*)handle; - int ret = -1; - DWORD numberOfBytesSent = 0; - - if(!transport){ - TSK_DEBUG_ERROR("Invalid transport handle."); - goto bail; - } - - if(transport->tls.have_tls){ - transport_socket_t* socket = getSocket(transport->context, from); - if(socket && socket->tlshandle){ - if(!tnet_tls_socket_send(socket->tlshandle, buf, size)){ - numberOfBytesSent = size; - } - else{ - numberOfBytesSent = 0; - } - goto bail; - } - } - else{ - WSABUF wsaBuffer; - wsaBuffer.buf = (CHAR*)buf; - wsaBuffer.len = size; - - if((ret = WSASend(from, &wsaBuffer, 1, &numberOfBytesSent, 0, NULL, NULL)) == SOCKET_ERROR){ - if((ret = WSAGetLastError()) == WSA_IO_PENDING){ - TSK_DEBUG_INFO("WSA_IO_PENDING error for WSASend SSESSION"); - ret = 0; - } - else{ - TNET_PRINT_LAST_ERROR("WSASend have failed."); - - //tnet_sockfd_close(&from); - goto bail; - } - } - else{ - ret = 0; - } - } - -bail: - return numberOfBytesSent; -} - -/* -* Sends dgarm to the specified destionation. -*/ -tsk_size_t tnet_transport_sendto(const tnet_transport_handle_t *handle, tnet_fd_t from, const struct sockaddr *to, const void* buf, tsk_size_t size) -{ - tnet_transport_t *transport = (tnet_transport_t*)handle; - WSABUF wsaBuffer; - DWORD numberOfBytesSent = 0; - int ret = -1; - - if(!transport){ - TSK_DEBUG_ERROR("Invalid server handle."); - return ret; - } - - if(!TNET_SOCKET_TYPE_IS_DGRAM(transport->master->type)){ - TSK_DEBUG_ERROR("In order to use WSASendTo you must use an udp transport."); - return ret; - } - - wsaBuffer.buf = (CHAR*)buf; - wsaBuffer.len = size; - - if((ret = WSASendTo(from, &wsaBuffer, 1, &numberOfBytesSent, 0, to, tnet_get_sockaddr_size(to), 0, 0)) == SOCKET_ERROR){ - if((ret = WSAGetLastError()) == WSA_IO_PENDING){ - TSK_DEBUG_INFO("WSA_IO_PENDING error for WSASendTo SSESSION"); - ret = 0; - } - else{ - TNET_PRINT_LAST_ERROR("WSASendTo have failed."); - return ret; - } - } else ret = 0; - - return numberOfBytesSent; -} - - - - - - - - -/*== CAllback function to check if we should accept the new socket or not == */ -int CALLBACK AcceptCondFunc(LPWSABUF lpCallerId, LPWSABUF lpCallerData, LPQOS lpSQos, LPQOS lpGQos, LPWSABUF lpCalleeId, LPWSABUF lpCalleeData, GROUP FAR *Group, DWORD CallbackData) -{ - transport_context_t *context = (transport_context_t*)CallbackData; - return context->count < WSA_MAXIMUM_WAIT_EVENTS ? CF_ACCEPT : CF_REJECT; -} - -/*== Get socket ==*/ -static transport_socket_t* getSocket(transport_context_t *context, tnet_fd_t fd) -{ - tsk_size_t i; - transport_socket_t* ret = 0; - - if(context){ - tsk_safeobj_lock(context); - for(i=0; i<context->count; i++){ - if(context->sockets[i]->fd == fd){ - ret = context->sockets[i]; - break; - } - } - tsk_safeobj_unlock(context); - } - - return ret; -} - -/*== Add new socket ==*/ -static int addSocket(tnet_fd_t fd, tnet_socket_type_t type, tnet_transport_t *transport, int take_ownership, int is_client) -{ - transport_context_t *context = transport?transport->context:0; - - if(context){ - transport_socket_t *sock = tsk_calloc(1, sizeof(transport_socket_t)); - sock->fd = fd; - sock->type = type; - sock->owner = take_ownership ? 1 : 0; - - if(TNET_SOCKET_TYPE_IS_TLS(sock->type)){ - sock->tlshandle = tnet_sockfd_set_tlsfiles(sock->fd, is_client, transport->tls.ca, transport->tls.pvk, transport->tls.pbk); - } - - tsk_safeobj_lock(context); - context->events[context->count] = WSACreateEvent(); - context->sockets[context->count] = sock; - - context->count++; - tsk_safeobj_unlock(context); - - return 0; - } - else{ - TSK_DEBUG_ERROR("Context is Null."); - return -1; - } -} - -/*== Remove socket ==*/ -static int removeSocket(int index, transport_context_t *context) -{ - tsk_size_t i; - - tsk_safeobj_lock(context); - - if(index < (int)context->count){ - - /* Close the socket if we are the owner. */ - if(context->sockets[index]->owner){ - tnet_sockfd_close(&(context->sockets[index]->fd)); - } - - /* Free tls context */ - if(context->sockets[index]->tlshandle){ - TSK_OBJECT_SAFE_FREE(context->sockets[index]->tlshandle); - } - // Free socket - TSK_FREE(context->sockets[index]); - - // Close event - WSACloseEvent(context->events[index]); - - for(i=index ; i<context->count-1; i++){ - context->sockets[i] = context->sockets[i+1]; - context->events[i] = context->events[i+1]; - } - - context->sockets[context->count-1] = 0; - context->events[context->count-1] = 0; - - context->count--; - } - - tsk_safeobj_unlock(context); - - return 0; -} - -/*=== stop all threads */ -int tnet_transport_stop(tnet_transport_t *transport) -{ - int ret; - - if((ret = tsk_runnable_stop(TSK_RUNNABLE(transport)))){ - return ret; - } - - if(transport->mainThreadId[0]){ - WSASetEvent(((transport_context_t*)(transport->context))->events[0]); - return tsk_thread_join(transport->mainThreadId); - } - else{ - /* already stoppped */ - return 0; - } -} - - -int tnet_transport_prepare(tnet_transport_t *transport) -{ - int ret = -1; - transport_context_t *context; - - if(!transport || !transport->context){ - TSK_DEBUG_ERROR("Invalid parameter."); - return -1; - } - else{ - context = transport->context; - } - - if(transport->prepared){ - TSK_DEBUG_ERROR("Transport already prepared."); - return -2; - } - - /* Start listening */ - if(TNET_SOCKET_TYPE_IS_STREAM(transport->master->type)){ - if((ret = tnet_sockfd_listen(transport->master->fd, WSA_MAXIMUM_WAIT_EVENTS))){ - TNET_PRINT_LAST_ERROR("listen have failed."); - goto bail; - } - } - - /* Add the master socket to the context. */ - if((ret = addSocket(transport->master->fd, transport->master->type, transport, tsk_true, tsk_false))){ - TSK_DEBUG_ERROR("Failed to add master socket"); - goto bail; - } - - /* set events on master socket */ - if((ret = WSAEventSelect(transport->master->fd, context->events[context->count - 1], TNET_SOCKET_TYPE_IS_DGRAM(transport->master->type) ? FD_READ : FD_ALL_EVENTS/*FD_ACCEPT | FD_READ | FD_CONNECT | FD_CLOSE*/) == SOCKET_ERROR)){ - TNET_PRINT_LAST_ERROR("WSAEventSelect have failed."); - goto bail; - } - - transport->prepared = tsk_true; - -bail: - return ret; -} - -int tnet_transport_unprepare(tnet_transport_t *transport) -{ - int ret = -1; - transport_context_t *context; - - if(!transport || !transport->context){ - TSK_DEBUG_ERROR("Invalid parameter."); - return -1; - } - else{ - context = transport->context; - } - - if(!transport->prepared){ - return 0; - } - - transport->prepared = tsk_false; - - while(context->count){ - removeSocket(0, context); // safe - } - - return 0; -} - -/*=== Main thread */ -void *tnet_transport_mainthread(void *param) -{ - tnet_transport_t *transport = param; - transport_context_t *context = transport->context; - DWORD evt; - WSANETWORKEVENTS networkEvents; - DWORD flags = 0; - int ret; - - struct sockaddr_storage remote_addr = {0}; - WSAEVENT active_event; - transport_socket_t* active_socket; - int index; - - TSK_DEBUG_INFO("Starting [%s] server with IP {%s} on port {%d}...", transport->description, transport->master->ip, transport->master->port); - - while(TSK_RUNNABLE(transport)->running || TSK_RUNNABLE(transport)->started) - { - /* Wait for multiple events */ - if((evt = WSAWaitForMultipleEvents(context->count, context->events, FALSE, WSA_INFINITE, FALSE)) == WSA_WAIT_FAILED){ - TNET_PRINT_LAST_ERROR("WSAWaitForMultipleEvents have failed."); - goto bail; - } - - if(!TSK_RUNNABLE(transport)->running && !TSK_RUNNABLE(transport)->started){ - goto bail; - } - - /* lock context */ - tsk_safeobj_lock(context); - - /* Get active event and socket */ - index = (evt - WSA_WAIT_EVENT_0); - active_event = context->events[index]; - if(!(active_socket = context->sockets[index])){ - goto done; - } - - /* Get the network events flags */ - if (WSAEnumNetworkEvents(active_socket->fd, active_event, &networkEvents) == SOCKET_ERROR){ - TSK_RUNNABLE_ENQUEUE(transport, event_error, transport->callback_data, active_socket->fd); - TNET_PRINT_LAST_ERROR("WSAEnumNetworkEvents have failed."); - - tsk_safeobj_unlock(context); - goto bail; - } - - /*================== FD_ACCEPT ==================*/ - if(networkEvents.lNetworkEvents & FD_ACCEPT) - { - tnet_fd_t fd; - - TSK_DEBUG_INFO("NETWORK EVENT FOR SERVER [%s] -- FD_ACCEPT", transport->description); - - if(networkEvents.iErrorCode[FD_ACCEPT_BIT]){ - TSK_RUNNABLE_ENQUEUE(transport, event_error, transport->callback_data, active_socket->fd); - TNET_PRINT_LAST_ERROR("ACCEPT FAILED."); - goto done; - } - - /* Accept the connection */ - if((fd = WSAAccept(active_socket->fd, NULL, NULL, AcceptCondFunc, (DWORD_PTR)context)) != INVALID_SOCKET){ - /* Add the new fd to the server context */ - addSocket(fd, transport->master->type, transport, tsk_true, tsk_false); - if(WSAEventSelect(fd, context->events[context->count - 1], FD_READ | FD_WRITE | FD_CLOSE) == SOCKET_ERROR){ - removeSocket((context->count - 1), context); - TNET_PRINT_LAST_ERROR("WSAEventSelect have failed."); - goto done; - } - TSK_RUNNABLE_ENQUEUE(transport, event_accepted, transport->callback_data, fd); - } - else{ - TNET_PRINT_LAST_ERROR("ACCEPT FAILED."); - goto done; - } - - - - - } - - /*================== FD_CONNECT ==================*/ - if(networkEvents.lNetworkEvents & FD_CONNECT) - { - //tnet_fd_t fd; - - TSK_DEBUG_INFO("NETWORK EVENT FOR SERVER [%s] -- FD_CONNECT", transport->description); - - if(networkEvents.iErrorCode[FD_CONNECT_BIT]){ - TSK_RUNNABLE_ENQUEUE(transport, event_error, transport->callback_data, active_socket->fd); - TNET_PRINT_LAST_ERROR("CONNECT FAILED."); - goto done; - } - else{ - TSK_RUNNABLE_ENQUEUE(transport, event_connected, transport->callback_data, active_socket->fd); - active_socket->connected = 1; - } - } - - - /*================== FD_READ ==================*/ - if(networkEvents.lNetworkEvents & FD_READ) - { - DWORD readCount = 0; - WSABUF wsaBuffer; - - /* TSK_DEBUG_INFO("NETWORK EVENT FOR SERVER [%s] -- FD_READ", transport->description); */ - - /* check whether the socket is paused or not */ - if(active_socket->paused){ - TSK_DEBUG_INFO("Socket is paused"); - goto FD_READ_DONE; - } - - if(networkEvents.iErrorCode[FD_READ_BIT]){ - TSK_RUNNABLE_ENQUEUE(transport, event_error, transport->callback_data, active_socket->fd); - TNET_PRINT_LAST_ERROR("READ FAILED."); - goto done; - } - - /* Retrieve the amount of pending data */ - if(tnet_ioctlt(active_socket->fd, FIONREAD, &(wsaBuffer.len)) < 0){ - TNET_PRINT_LAST_ERROR("IOCTLT FAILED."); - goto done; - } - - if(!wsaBuffer.len){ - goto done; - } - - /* Alloc data */ - if(!(wsaBuffer.buf = tsk_calloc(wsaBuffer.len, sizeof(uint8_t)))){ - goto done; - } - - /* Receive the waiting data. */ - if(active_socket->tlshandle){ - int isEncrypted; - tsk_size_t len = wsaBuffer.len; - if(!(ret = tnet_tls_socket_recv(active_socket->tlshandle, &wsaBuffer.buf, &len, &isEncrypted))){ - if(isEncrypted){ - TSK_FREE(wsaBuffer.buf); - goto done; - } - wsaBuffer.len = len; - } - } - else{ - if(TNET_SOCKET_TYPE_IS_STREAM(transport->master->type)){ - ret = WSARecv(active_socket->fd, &wsaBuffer, 1, &readCount, &flags, 0, 0); - } - else{ - int len = sizeof(remote_addr); - ret = WSARecvFrom(active_socket->fd, &wsaBuffer, 1, &readCount, &flags, - (struct sockaddr*)&remote_addr, &len, 0, 0); - } - if(readCount < wsaBuffer.len){ - wsaBuffer.len = readCount; - /* wsaBuffer.buf = tsk_realloc(wsaBuffer.buf, readCount); */ - } - } - - if(ret){ - ret = WSAGetLastError(); - if(ret == WSAEWOULDBLOCK){ - TSK_DEBUG_WARN("WSAEWOULDBLOCK error for READ SSESSION"); - } - else if(ret == WSAECONNRESET && TNET_SOCKET_TYPE_IS_DGRAM(transport->master->type)){ - /* For DGRAM ==> The sent packet gernerated "ICMP Destination/Port unreachable" result. */ - TSK_FREE(wsaBuffer.buf); - goto done; // ignore and retry. - } - else{ - TSK_FREE(wsaBuffer.buf); - - removeSocket(index, context); - TNET_PRINT_LAST_ERROR("WSARecv have failed."); - goto done; - } - } - else - { - tnet_transport_event_t* e = tnet_transport_event_create(event_data, transport->callback_data, active_socket->fd); - e->data = wsaBuffer.buf; - e->size = wsaBuffer.len; - e->remote_addr = remote_addr; - - TSK_RUNNABLE_ENQUEUE_OBJECT_SAFE(TSK_RUNNABLE(transport), e); - } -FD_READ_DONE:; - } - - - - - /*================== FD_WRITE ==================*/ - if(networkEvents.lNetworkEvents & FD_WRITE) - { - TSK_DEBUG_INFO("NETWORK EVENT FOR SERVER [%s] -- FD_WRITE", transport->description); - - if(networkEvents.iErrorCode[FD_WRITE_BIT]){ - TSK_RUNNABLE_ENQUEUE(transport, event_error, transport->callback_data, active_socket->fd); - TNET_PRINT_LAST_ERROR("WRITE FAILED."); - goto done; - } - } - - - - /*================== FD_CLOSE ==================*/ - if(networkEvents.lNetworkEvents & FD_CLOSE){ - TSK_DEBUG_INFO("NETWORK EVENT FOR SERVER [%s] -- FD_CLOSE", transport->description); - - TSK_RUNNABLE_ENQUEUE(transport, event_closed, transport->callback_data, active_socket->fd); - removeSocket(index, context); - } - - /* http://msdn.microsoft.com/en-us/library/ms741690(VS.85).aspx - - The proper way to reset the state of an event object used with the WSAEventSelect function - is to pass the handle of the event object to the WSAEnumNetworkEvents function in the hEventObject parameter. - This will reset the event object and adjust the status of active FD events on the socket in an atomic fashion. - */ - /* WSAResetEvent(active_event); <== DO NOT USE (see above) */ - -done: - /* unlock context */ - tsk_safeobj_unlock(context); - - } /* while(transport->running) */ - - -bail: - - - TSK_DEBUG_INFO("Stopped [%s] server with IP {%s} on port {%d}...", transport->description, transport->master->ip, transport->master->port); - return 0; -} - - - - -tsk_object_t* tnet_transport_context_create() -{ - return tsk_object_new(tnet_transport_context_def_t); -} - - -//================================================================================================= -// Transport context object definition -// -static tsk_object_t* transport_context_ctor(tsk_object_t * self, va_list * app) -{ - transport_context_t *context = self; - if(context){ - tsk_safeobj_init(context); - } - return self; -} - -static tsk_object_t* transport_context_dtor(tsk_object_t * self) -{ - transport_context_t *context = self; - if(context){ - while(context->count){ - removeSocket(0, context); - } - tsk_safeobj_deinit(context); - } - return self; -} - -static const tsk_object_def_t tnet_transport_context_def_s = -{ -sizeof(transport_context_t), -transport_context_ctor, -transport_context_dtor, -tsk_null, -}; -const tsk_object_def_t *tnet_transport_context_def_t = &tnet_transport_context_def_s; -#endif /* TNET_UNDER_WINDOWS */ - |
