mirror of
https://github.com/monero-project/monero.git
synced 2024-12-29 13:16:13 -05:00
Remove http_server_thread_per_connect and its dependencies
This commit is contained in:
parent
bd0a511995
commit
f9d6504594
@ -1,318 +0,0 @@
|
|||||||
// Copyright (c) 2006-2013, Andrey N. Sabelnikov, www.sabelnikov.net
|
|
||||||
// All rights reserved.
|
|
||||||
//
|
|
||||||
// Redistribution and use in source and binary forms, with or without
|
|
||||||
// modification, are permitted provided that the following conditions are met:
|
|
||||||
// * 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.
|
|
||||||
// * Neither the name of the Andrey N. Sabelnikov nor the
|
|
||||||
// names of its contributors may be used to endorse or promote products
|
|
||||||
// derived from this software without specific prior written permission.
|
|
||||||
//
|
|
||||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
|
|
||||||
// ANY EXPRESS 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 COPYRIGHT OWNER 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 _ABSTRACT_TCP_SERVER_H_
|
|
||||||
#define _ABSTRACT_TCP_SERVER_H_
|
|
||||||
|
|
||||||
#include <process.h>
|
|
||||||
#include <list>
|
|
||||||
#include <winsock2.h>
|
|
||||||
#include "winobj.h"
|
|
||||||
//#include "threads_helper.h"
|
|
||||||
#include "net_utils_base.h"
|
|
||||||
|
|
||||||
#pragma comment(lib, "Ws2_32.lib")
|
|
||||||
|
|
||||||
#undef MONERO_DEFAULT_LOG_CATEGORY
|
|
||||||
#define MONERO_DEFAULT_LOG_CATEGORY "net"
|
|
||||||
|
|
||||||
namespace epee
|
|
||||||
{
|
|
||||||
namespace net_utils
|
|
||||||
{
|
|
||||||
/************************************************************************/
|
|
||||||
/* */
|
|
||||||
/************************************************************************/
|
|
||||||
class soket_sender: public i_service_endpoint
|
|
||||||
{
|
|
||||||
public:
|
|
||||||
soket_sender(SOCKET sock):m_sock(sock){}
|
|
||||||
private:
|
|
||||||
virtual bool handle_send(const void* ptr, size_t cb)
|
|
||||||
{
|
|
||||||
if(cb != send(m_sock, (char*)ptr, (int)cb, 0))
|
|
||||||
{
|
|
||||||
int sock_err = WSAGetLastError();
|
|
||||||
LOG_ERROR("soket_sender: Failed to send " << cb << " bytes, Error=" << sock_err);
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
return true;
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
SOCKET m_sock;
|
|
||||||
};
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
/************************************************************************/
|
|
||||||
/* */
|
|
||||||
/************************************************************************/
|
|
||||||
template<class THandler>
|
|
||||||
class abstract_tcp_server
|
|
||||||
{
|
|
||||||
public:
|
|
||||||
abstract_tcp_server();
|
|
||||||
|
|
||||||
bool init_server(int port_no);
|
|
||||||
bool deinit_server();
|
|
||||||
bool run_server();
|
|
||||||
bool send_stop_signal();
|
|
||||||
|
|
||||||
typename THandler::config_type& get_config_object(){return m_config;}
|
|
||||||
|
|
||||||
private:
|
|
||||||
bool invoke_connection(SOCKET hnew_sock, long ip_from, int post_from);
|
|
||||||
static unsigned __stdcall ConnectionHandlerProc(void* lpParameter);
|
|
||||||
|
|
||||||
class thread_context;
|
|
||||||
typedef std::list<thread_context> connections_container;
|
|
||||||
typedef typename connections_container::iterator connections_iterator;
|
|
||||||
|
|
||||||
struct thread_context
|
|
||||||
{
|
|
||||||
HANDLE m_htread;
|
|
||||||
SOCKET m_socket;
|
|
||||||
abstract_tcp_server* powner;
|
|
||||||
connection_context m_context;
|
|
||||||
typename connections_iterator m_self_it;
|
|
||||||
};
|
|
||||||
|
|
||||||
SOCKET m_listen_socket;
|
|
||||||
int m_port;
|
|
||||||
bool m_initialized;
|
|
||||||
volatile LONG m_stop_server;
|
|
||||||
volatile LONG m_threads_count;
|
|
||||||
typename THandler::config_type m_config;
|
|
||||||
connections_container m_connections;
|
|
||||||
critical_section m_connections_lock;
|
|
||||||
};
|
|
||||||
|
|
||||||
template<class THandler>
|
|
||||||
unsigned __stdcall abstract_tcp_server<THandler>::ConnectionHandlerProc(void* lpParameter)
|
|
||||||
{
|
|
||||||
|
|
||||||
thread_context* pthread_context = (thread_context*)lpParameter;
|
|
||||||
if(!pthread_context)
|
|
||||||
return 0;
|
|
||||||
abstract_tcp_server<THandler>* pthis = pthread_context->powner;
|
|
||||||
|
|
||||||
::InterlockedIncrement(&pthis->m_threads_count);
|
|
||||||
|
|
||||||
::CoInitialize(NULL);
|
|
||||||
|
|
||||||
|
|
||||||
LOG_PRINT("Handler thread STARTED with socket=" << pthread_context->m_socket, LOG_LEVEL_2);
|
|
||||||
int res = 0;
|
|
||||||
|
|
||||||
soket_sender sndr(pthread_context->m_socket);
|
|
||||||
THandler srv(&sndr, pthread_context->powner->m_config, pthread_context->m_context);
|
|
||||||
|
|
||||||
|
|
||||||
srv.after_init_connection();
|
|
||||||
|
|
||||||
char buff[1000] = {0};
|
|
||||||
std::string ansver;
|
|
||||||
while ( (res = recv(pthread_context->m_socket, (char*)buff, 1000, 0)) > 0)
|
|
||||||
{
|
|
||||||
LOG_PRINT("Data in, " << res << " bytes", LOG_LEVEL_3);
|
|
||||||
if(!srv.handle_recv(buff, res))
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
shutdown(pthread_context->m_socket, SD_BOTH);
|
|
||||||
closesocket(pthread_context->m_socket);
|
|
||||||
|
|
||||||
abstract_tcp_server* powner = pthread_context->powner;
|
|
||||||
LOG_PRINT("Handler thread with socket=" << pthread_context->m_socket << " STOPPED", LOG_LEVEL_2);
|
|
||||||
powner->m_connections_lock.lock();
|
|
||||||
::CloseHandle(pthread_context->m_htread);
|
|
||||||
pthread_context->powner->m_connections.erase(pthread_context->m_self_it);
|
|
||||||
powner->m_connections_lock.unlock();
|
|
||||||
CoUninitialize();
|
|
||||||
::InterlockedDecrement(&pthis->m_threads_count);
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
//----------------------------------------------------------------------------------------
|
|
||||||
template<class THandler>
|
|
||||||
abstract_tcp_server<THandler>::abstract_tcp_server():m_listen_socket(INVALID_SOCKET),
|
|
||||||
m_initialized(false),
|
|
||||||
m_stop_server(0), m_port(0), m_threads_count(0)
|
|
||||||
{
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
//----------------------------------------------------------------------------------------
|
|
||||||
template<class THandler>
|
|
||||||
bool abstract_tcp_server<THandler>::init_server(int port_no)
|
|
||||||
{
|
|
||||||
m_port = port_no;
|
|
||||||
WSADATA wsad = {0};
|
|
||||||
int err = ::WSAStartup(MAKEWORD(2,2), &wsad);
|
|
||||||
if ( err != 0 || LOBYTE( wsad.wVersion ) != 2 || HIBYTE( wsad.wVersion ) != 2 )
|
|
||||||
{
|
|
||||||
LOG_ERROR("Could not find a usable WinSock DLL, err = " << err << " \"" << socket_errors::get_socket_error_text(err) <<"\"");
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
m_initialized = true;
|
|
||||||
|
|
||||||
m_listen_socket = ::WSASocket(AF_INET, SOCK_STREAM, IPPROTO_TCP, NULL, 0, 0);
|
|
||||||
if(INVALID_SOCKET == m_listen_socket)
|
|
||||||
{
|
|
||||||
err = ::WSAGetLastError();
|
|
||||||
LOG_ERROR("Failed to create socket, err = " << err << " \"" << socket_errors::get_socket_error_text(err) <<"\"");
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
int opt = 1;
|
|
||||||
setsockopt (m_listen_socket, SOL_SOCKET,SO_REUSEADDR, reinterpret_cast<char*>(&opt), sizeof(int));
|
|
||||||
|
|
||||||
sockaddr_in adr = {0};
|
|
||||||
adr.sin_family = AF_INET;
|
|
||||||
adr.sin_addr.s_addr = htonl(INADDR_ANY);
|
|
||||||
adr.sin_port = (u_short)htons(port_no);
|
|
||||||
|
|
||||||
err = bind(m_listen_socket, (const sockaddr*)&adr, sizeof(adr ));
|
|
||||||
if(SOCKET_ERROR == err )
|
|
||||||
{
|
|
||||||
err = ::WSAGetLastError();
|
|
||||||
LOG_PRINT("Failed to Bind, err = " << err << " \"" << socket_errors::get_socket_error_text(err) <<"\"", LOG_LEVEL_2);
|
|
||||||
deinit_server();
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
::InterlockedExchange(&m_stop_server, 0);
|
|
||||||
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
//----------------------------------------------------------------------------------------
|
|
||||||
template<class THandler>
|
|
||||||
bool abstract_tcp_server<THandler>::deinit_server()
|
|
||||||
{
|
|
||||||
|
|
||||||
if(!m_initialized)
|
|
||||||
return true;
|
|
||||||
|
|
||||||
if(INVALID_SOCKET != m_listen_socket)
|
|
||||||
{
|
|
||||||
shutdown(m_listen_socket, SD_BOTH);
|
|
||||||
int res = closesocket(m_listen_socket);
|
|
||||||
if(SOCKET_ERROR == res)
|
|
||||||
{
|
|
||||||
int err = ::WSAGetLastError();
|
|
||||||
LOG_ERROR("Failed to closesocket(), err = " << err << " \"" << socket_errors::get_socket_error_text(err) <<"\"");
|
|
||||||
}
|
|
||||||
m_listen_socket = INVALID_SOCKET;
|
|
||||||
}
|
|
||||||
|
|
||||||
int res = ::WSACleanup();
|
|
||||||
if(SOCKET_ERROR == res)
|
|
||||||
{
|
|
||||||
int err = ::WSAGetLastError();
|
|
||||||
LOG_ERROR("Failed to WSACleanup(), err = " << err << " \"" << socket_errors::get_socket_error_text(err) <<"\"");
|
|
||||||
}
|
|
||||||
m_initialized = false;
|
|
||||||
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
//----------------------------------------------------------------------------------------
|
|
||||||
template<class THandler>
|
|
||||||
bool abstract_tcp_server<THandler>::send_stop_signal()
|
|
||||||
{
|
|
||||||
InterlockedExchange(&m_stop_server, 1);
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
//----------------------------------------------------------------------------------------
|
|
||||||
template<class THandler>
|
|
||||||
bool abstract_tcp_server<THandler>::run_server()
|
|
||||||
{
|
|
||||||
int err = listen(m_listen_socket, 10000);
|
|
||||||
if(SOCKET_ERROR == err )
|
|
||||||
{
|
|
||||||
err = ::WSAGetLastError();
|
|
||||||
LOG_ERROR("Failed to listen, err = " << err << " \"" << socket_errors::get_socket_error_text(err) <<"\"");
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
LOG_PRINT("Listening port "<< m_port << "...." , LOG_LEVEL_2);
|
|
||||||
|
|
||||||
while(!m_stop_server)
|
|
||||||
{
|
|
||||||
sockaddr_in adr_from = {0};
|
|
||||||
int adr_len = sizeof(adr_from);
|
|
||||||
fd_set read_fs = {0};
|
|
||||||
read_fs.fd_count = 1;
|
|
||||||
read_fs.fd_array[0] = m_listen_socket;
|
|
||||||
TIMEVAL tv = {0};
|
|
||||||
tv.tv_usec = 100;
|
|
||||||
int select_res = select(0, &read_fs, NULL, NULL, &tv);
|
|
||||||
if(!select_res)
|
|
||||||
continue;
|
|
||||||
SOCKET new_sock = WSAAccept(m_listen_socket, (sockaddr *)&adr_from, &adr_len, NULL, NULL);
|
|
||||||
LOG_PRINT("Accepted connection on socket=" << new_sock, LOG_LEVEL_2);
|
|
||||||
invoke_connection(new_sock, adr_from.sin_addr.s_addr, adr_from.sin_port);
|
|
||||||
}
|
|
||||||
|
|
||||||
deinit_server();
|
|
||||||
|
|
||||||
#define ABSTR_TCP_SRV_WAIT_COUNT_MAX 5000
|
|
||||||
#define ABSTR_TCP_SRV_WAIT_COUNT_INTERVAL 1000
|
|
||||||
|
|
||||||
int wait_count = 0;
|
|
||||||
|
|
||||||
while(m_threads_count && wait_count*1000 < ABSTR_TCP_SRV_WAIT_COUNT_MAX)
|
|
||||||
{
|
|
||||||
::Sleep(ABSTR_TCP_SRV_WAIT_COUNT_INTERVAL);
|
|
||||||
wait_count++;
|
|
||||||
}
|
|
||||||
LOG_PRINT("abstract_tcp_server exit with wait count=" << wait_count*ABSTR_TCP_SRV_WAIT_COUNT_INTERVAL << "(max=" << ABSTR_TCP_SRV_WAIT_COUNT_MAX <<")", LOG_LEVEL_0);
|
|
||||||
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
//----------------------------------------------------------------------------------------
|
|
||||||
template<class THandler>
|
|
||||||
bool abstract_tcp_server<THandler>::invoke_connection(SOCKET hnew_sock, const network_address &remote_address)
|
|
||||||
{
|
|
||||||
m_connections_lock.lock();
|
|
||||||
m_connections.push_back(thread_context());
|
|
||||||
m_connections_lock.unlock();
|
|
||||||
m_connections.back().m_socket = hnew_sock;
|
|
||||||
m_connections.back().powner = this;
|
|
||||||
m_connections.back().m_self_it = --m_connections.end();
|
|
||||||
m_connections.back().m_context.m_remote_address = remote_address;
|
|
||||||
m_connections.back().m_htread = threads_helper::create_thread(ConnectionHandlerProc, &m_connections.back()); // ugh, seems very risky
|
|
||||||
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
//----------------------------------------------------------------------------------------
|
|
||||||
|
|
||||||
//----------------------------------------------------------------------------------------
|
|
||||||
//----------------------------------------------------------------------------------------
|
|
||||||
}
|
|
||||||
}
|
|
||||||
#endif //_ABSTRACT_TCP_SERVER_H_
|
|
@ -1,48 +0,0 @@
|
|||||||
// Copyright (c) 2006-2013, Andrey N. Sabelnikov, www.sabelnikov.net
|
|
||||||
// All rights reserved.
|
|
||||||
//
|
|
||||||
// Redistribution and use in source and binary forms, with or without
|
|
||||||
// modification, are permitted provided that the following conditions are met:
|
|
||||||
// * 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.
|
|
||||||
// * Neither the name of the Andrey N. Sabelnikov nor the
|
|
||||||
// names of its contributors may be used to endorse or promote products
|
|
||||||
// derived from this software without specific prior written permission.
|
|
||||||
//
|
|
||||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
|
|
||||||
// ANY EXPRESS 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 COPYRIGHT OWNER 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 _HTTP_SERVER_CP_H_
|
|
||||||
#define _HTTP_SERVER_CP_H_
|
|
||||||
|
|
||||||
#include "abstract_tcp_server.h"
|
|
||||||
#include "http_server.h"
|
|
||||||
|
|
||||||
namespace epee
|
|
||||||
{
|
|
||||||
namespace net_utils
|
|
||||||
{
|
|
||||||
typedef abstract_tcp_server<http::simple_http_connection_handler> mt_http_server_file_system;
|
|
||||||
typedef abstract_tcp_server<http::http_custom_handler> mt_http_server_custum_handling;
|
|
||||||
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
#endif
|
|
||||||
|
|
||||||
|
|
@ -1,227 +0,0 @@
|
|||||||
// Copyright (c) 2006-2013, Andrey N. Sabelnikov, www.sabelnikov.net
|
|
||||||
// All rights reserved.
|
|
||||||
//
|
|
||||||
// Redistribution and use in source and binary forms, with or without
|
|
||||||
// modification, are permitted provided that the following conditions are met:
|
|
||||||
// * 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.
|
|
||||||
// * Neither the name of the Andrey N. Sabelnikov nor the
|
|
||||||
// names of its contributors may be used to endorse or promote products
|
|
||||||
// derived from this software without specific prior written permission.
|
|
||||||
//
|
|
||||||
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
|
|
||||||
// ANY EXPRESS 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 COPYRIGHT OWNER 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 __WINH_OBJ_H__
|
|
||||||
#define __WINH_OBJ_H__
|
|
||||||
|
|
||||||
#include <boost/thread/locks.hpp>
|
|
||||||
|
|
||||||
namespace epee
|
|
||||||
{
|
|
||||||
class critical_region;
|
|
||||||
|
|
||||||
class critical_section {
|
|
||||||
|
|
||||||
boost::mutex m_section;
|
|
||||||
|
|
||||||
public:
|
|
||||||
|
|
||||||
critical_section( const critical_section& section ) {
|
|
||||||
InitializeCriticalSection( &m_section );
|
|
||||||
}
|
|
||||||
|
|
||||||
critical_section() {
|
|
||||||
InitializeCriticalSection( &m_section );
|
|
||||||
}
|
|
||||||
|
|
||||||
~critical_section() {
|
|
||||||
DeleteCriticalSection( &m_section );
|
|
||||||
}
|
|
||||||
|
|
||||||
void lock() {
|
|
||||||
EnterCriticalSection( &m_section );
|
|
||||||
}
|
|
||||||
|
|
||||||
void unlock() {
|
|
||||||
LeaveCriticalSection( &m_section );
|
|
||||||
}
|
|
||||||
|
|
||||||
bool tryLock() {
|
|
||||||
return TryEnterCriticalSection( &m_section )? true:false;
|
|
||||||
}
|
|
||||||
|
|
||||||
critical_section& operator=( const critical_section& section )
|
|
||||||
{
|
|
||||||
return *this;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
};
|
|
||||||
|
|
||||||
class critical_region {
|
|
||||||
|
|
||||||
::critical_section *m_locker;
|
|
||||||
|
|
||||||
critical_region( const critical_region& ){}
|
|
||||||
|
|
||||||
public:
|
|
||||||
|
|
||||||
critical_region(critical_section &cs ) {
|
|
||||||
m_locker = &cs;
|
|
||||||
cs.lock();
|
|
||||||
}
|
|
||||||
|
|
||||||
~critical_region()
|
|
||||||
{
|
|
||||||
m_locker->unlock();
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
|
|
||||||
class shared_critical_section
|
|
||||||
{
|
|
||||||
public:
|
|
||||||
shared_critical_section()
|
|
||||||
{
|
|
||||||
::InitializeSRWLock(&m_srw_lock);
|
|
||||||
}
|
|
||||||
~shared_critical_section()
|
|
||||||
{}
|
|
||||||
|
|
||||||
bool lock_shared()
|
|
||||||
{
|
|
||||||
AcquireSRWLockShared(&m_srw_lock);
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
bool unlock_shared()
|
|
||||||
{
|
|
||||||
ReleaseSRWLockShared(&m_srw_lock);
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
bool lock_exclusive()
|
|
||||||
{
|
|
||||||
::AcquireSRWLockExclusive(&m_srw_lock);
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
bool unlock_exclusive()
|
|
||||||
{
|
|
||||||
::ReleaseSRWLockExclusive(&m_srw_lock);
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
private:
|
|
||||||
SRWLOCK m_srw_lock;
|
|
||||||
|
|
||||||
};
|
|
||||||
|
|
||||||
|
|
||||||
class shared_guard
|
|
||||||
{
|
|
||||||
public:
|
|
||||||
shared_guard(shared_critical_section& ref_sec):m_ref_sec(ref_sec)
|
|
||||||
{
|
|
||||||
m_ref_sec.lock_shared();
|
|
||||||
}
|
|
||||||
|
|
||||||
~shared_guard()
|
|
||||||
{
|
|
||||||
m_ref_sec.unlock_shared();
|
|
||||||
}
|
|
||||||
|
|
||||||
private:
|
|
||||||
shared_critical_section& m_ref_sec;
|
|
||||||
};
|
|
||||||
|
|
||||||
|
|
||||||
class exclusive_guard
|
|
||||||
{
|
|
||||||
public:
|
|
||||||
exclusive_guard(shared_critical_section& ref_sec):m_ref_sec(ref_sec)
|
|
||||||
{
|
|
||||||
m_ref_sec.lock_exclusive();
|
|
||||||
}
|
|
||||||
|
|
||||||
~exclusive_guard()
|
|
||||||
{
|
|
||||||
m_ref_sec.unlock_exclusive();
|
|
||||||
}
|
|
||||||
|
|
||||||
private:
|
|
||||||
shared_critical_section& m_ref_sec;
|
|
||||||
};
|
|
||||||
|
|
||||||
|
|
||||||
class event
|
|
||||||
{
|
|
||||||
public:
|
|
||||||
event()
|
|
||||||
{
|
|
||||||
m_hevent = ::CreateEvent(NULL, FALSE, FALSE, NULL);
|
|
||||||
}
|
|
||||||
~event()
|
|
||||||
{
|
|
||||||
::CloseHandle(m_hevent);
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
bool set()
|
|
||||||
{
|
|
||||||
return ::SetEvent(m_hevent) ? true:false;
|
|
||||||
}
|
|
||||||
|
|
||||||
bool reset()
|
|
||||||
{
|
|
||||||
return ::ResetEvent(m_hevent) ? true:false;
|
|
||||||
}
|
|
||||||
|
|
||||||
HANDLE get_handle()
|
|
||||||
{
|
|
||||||
return m_hevent;
|
|
||||||
}
|
|
||||||
private:
|
|
||||||
HANDLE m_hevent;
|
|
||||||
|
|
||||||
};
|
|
||||||
|
|
||||||
|
|
||||||
#define SHARED_CRITICAL_REGION_BEGIN(x) { shared_guard critical_region_var(x)
|
|
||||||
#define EXCLUSIVE_CRITICAL_REGION_BEGIN(x) { exclusive_guard critical_region_var(x)
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
#define CRITICAL_REGION_LOCAL(x) critical_region critical_region_var(x)
|
|
||||||
#define CRITICAL_REGION_BEGIN(x) { critical_region critical_region_var(x)
|
|
||||||
#define CRITICAL_REGION_END() }
|
|
||||||
|
|
||||||
|
|
||||||
inline const char* get_wait_for_result_as_text(DWORD res)
|
|
||||||
{
|
|
||||||
switch(res)
|
|
||||||
{
|
|
||||||
case WAIT_ABANDONED: return "WAIT_ABANDONED";
|
|
||||||
case WAIT_TIMEOUT: return "WAIT_TIMEOUT";
|
|
||||||
case WAIT_OBJECT_0: return "WAIT_OBJECT_0";
|
|
||||||
case WAIT_OBJECT_0+1: return "WAIT_OBJECT_1";
|
|
||||||
case WAIT_OBJECT_0+2: return "WAIT_OBJECT_2";
|
|
||||||
default:
|
|
||||||
return "UNKNOWN CODE";
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
}// namespace epee
|
|
||||||
|
|
||||||
#endif
|
|
Loading…
Reference in New Issue
Block a user