Thread portable class Gate portable class Timer class Pool class Terimber 2.0 About C++

Downloads Products & Services Support Clients Open Source About Home / Open source / Terimber 2.0 aiocomport.h Go to the documentation of this file. /* * The Software License * ================================================================================= * Copyright (c) 2003-.The Terimber Corporation. All rights reserved. * ================================================================================= * Redistributions of source code must retain the above copyright notice, * this list of conditions and the following disclaimer. * Redistributions in binary form must reproduce the above copyright notice, * this list of conditions and the following disclaimer in the documentation * and/or other materials provided with the distribution. * The end-user documentation included with the redistribution, if any, * must include the following acknowledgment: * "This product includes software developed by the Terimber Corporation." * ================================================================================= * THIS SOFTWARE IS PROVIDED "AS IS" AND ANY EXPRESSED OR IMPLIED WARRANTIES, * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY * AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. * IN NO EVENT SHALL THE TERIMBER CORPORATION OR ITS CONTRIBUTORS BE LIABLE FOR * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. * ================================================================================ */ #ifndef _terimber_aiocomport_h_ #define _terimber_aiocomport_h_ #include "base/map.h" #include "base/list.h" #include "base/stack.h" #include "base/primitives.h" #include "threadpool/threadpool.h" #if OS_TYPE != OS_WIN32 #ifndef HANDLE typedef int HANDLE; #endif #ifndef SOCKET typedef int SOCKET; #endif #ifndef INVALID_SOCKET #define INVALID_SOCKET ((SOCKET)~0) #endif BEGIN_TERIMBER_NAMESPACE #pragma pack(4) typedef struct _OVERLAPPED { SOCKET hAccept; sockaddr_in remoteAddress; unsigned long offset; } OVERLAPPED; typedef OVERLAPPED* LPOVERLAPPED; enum en_type { TYPE_UNKNOWN = 0, TYPE_TCP, TYPE_UDP, TYPE_FILE }; HANDLE CreateIoCompletionPort ( HANDLE sock_fd, HANDLE port_fd, size_t completion_key, en_type type ); bool AcceptEx ( HANDLE sock_listen, LPOVERLAPPED overlapped ); bool ConnectEx ( HANDLE sock_fd, const struct sockaddr_in* name, LPOVERLAPPED overlapped ); bool WSARecv ( HANDLE sock_fd, void* buf, size_t len, LPOVERLAPPED overlapped ); bool WSARecvFrom ( HANDLE sock_fd, void* buf, size_t len, struct sockaddr_in* name, LPOVERLAPPED overlapped ); bool ReadFile ( HANDLE sock_fd, void* buf, size_t len, LPOVERLAPPED overlapped ); bool WSASend ( HANDLE sock_fd, const void* buf, size_t len, LPOVERLAPPED overlapped ); bool WSASendTo ( HANDLE sock_fd, const void* buf, size_t len, const struct sockaddr_in* name, LPOVERLAPPED overlapped ); bool WriteFile ( HANDLE sock_fd, const void* buf, size_t len, LPOVERLAPPED overlapped ); int GetQueuedCompletionStatus ( HANDLE port_fd, size_t* number_bytes, size_t* completion_key, LPOVERLAPPED* overlapped, size_t timeout_milliseconds ); bool PostQueuedCompletionStatus ( HANDLE port_fd, size_t bytes_transferred, size_t completion_key, LPOVERLAPPED overlapped ); bool CloseHandle ( HANDLE port_fd ); bool CancelIo ( HANDLE sock_fd, LPOVERLAPPED overlapped ); void SetLog ( terimber_log* log ); void DoXRay ( ); class aiocomport : public terimber_thread_employer, public terimber_log_helper { public: enum en_action { ACTION_UNKNOWN = 0, ACTION_USERDATA, ACTION_CONNECT, ACTION_ACCEPT, ACTION_RECV, ACTION_SEND, ACTION_READ, ACTION_WRITE }; private: class queue_item : public aiocb { public: queue_item() { memset(this, 0, sizeof(queue_item)); aio_fildes = (int)INVALID_SOCKET; } en_action _action; size_t _error; size_t _processed; size_t _key; LPOVERLAPPED _overlapped; }; typedef node_allocator< _list< queue_item >::_node > queue_allocator_t; typedef _list< queue_item, queue_allocator_t > queue_container_t; typedef queue_container_t::iterator queue_container_iterator_t; class port_attributes { port_attributes& operator=(const port_attributes& x); public: port_attributes() : _go_to_exit(false), _loop_lock(false) { } port_attributes(const port_attributes& x) : _go_to_exit(x._go_to_exit), _loop_lock(x._loop_lock) { } bool _go_to_exit; size_t _loop_lock; event _queue_event; event _exit_event; queue_container_t _completion_container; }; typedef map< HANDLE, port_attributes > port_attr_map_t; typedef port_attr_map_t::iterator port_attr_map_iterator_t; class fd_item { public: fd_item(port_attr_map_iterator_t port_iter, size_t completion_key, unsigned long int type) : _port_iter(port_iter), _completion_key(completion_key), _type((en_type)type) { } queue_container_t _initial_container; port_attr_map_iterator_t _port_iter; size_t _completion_key; en_type _type; }; typedef map< HANDLE, fd_item > fd_port_map_t; public: aiocomport(); ~aiocomport(); HANDLE CreateIoCompletionPort ( HANDLE sock_fd, HANDLE port_fd, size_t completion_key, en_type type ); bool AcceptEx ( HANDLE sock_listen, LPOVERLAPPED overlapped ); bool ConnectEx ( HANDLE sock_fd, const struct sockaddr_in* name, LPOVERLAPPED overlapped ); bool WSARecv ( HANDLE sock_fd, void* buf, size_t len, LPOVERLAPPED overlapped ); bool WSARecvFrom ( HANDLE sock_fd, void* buf, size_t len, struct sockaddr_in* name, LPOVERLAPPED overlapped ); bool ReadFile ( HANDLE sock_fd, void* buf, size_t len, LPOVERLAPPED overlapped ); bool WSASend ( HANDLE sock_fd, const void* buf, size_t len, LPOVERLAPPED overlapped ); bool WSASendTo ( HANDLE sock_fd, const void* buf, size_t len, const struct sockaddr_in* name, LPOVERLAPPED overlapped ); bool WriteFile ( HANDLE sock_fd, const void* buf, size_t len, LPOVERLAPPED overlapped ); int GetQueuedCompletionStatus ( HANDLE port_fd, size_t* number_bytes, size_t* completion_key, LPOVERLAPPED* overlapped, size_t timeout_milliseconds ); bool PostQueuedCompletionStatus ( HANDLE port_fd, size_t bytes_transferred, size_t completion_key, LPOVERLAPPED overlapped ); bool CloseHandle ( HANDLE port_fd ); bool CancelIo ( HANDLE sock_fd, LPOVERLAPPED overlapped ); void SetLog ( terimber_log* log ); void DoXRay ( ); // -- MAC OS does not have file descriptor in siginfo_t structure - send it explicitly void func_sock_signal(int fd, int si_code); void func_file_signal(int fd, void* ptr); protected: virtual bool v_has_job( size_t ident, void* data ); virtual void v_do_job( size_t ident, void* data ); private: bool initiate_action( en_action action, HANDLE sock_id, LPOVERLAPPED overlapped, const void* buf, size_t len, const sockaddr_in* name ); private: mutex _port_mtx; port_attr_map_t _port_attr_map; fd_port_map_t _fd_port_map; unique_key_generator _generator; size_t _capacity; queue_allocator_t _queue_allocator; thread _event_thread; volatile pthread_t _event_thread_id; event _ev_activate; event _ev_deactivate; #ifdef NO_NPTL event _ev_wakeup; #endif }; #pragma pack() END_TERIMBER_NAMESPACE #endif #endif // _terimber_completionport_h_

© Copyright Terimber 2003-.