aboutsummaryrefslogtreecommitdiff
path: root/src/game/WorldSocketMgr.cpp
diff options
context:
space:
mode:
authormegamage <none@none>2008-12-08 16:34:03 -0600
committermegamage <none@none>2008-12-08 16:34:03 -0600
commitce643187f06cd0b3829ab90deff335f5c9ec5a84 (patch)
treedc09eece1fc5dce2a942bdc27a4a045518ef9daa /src/game/WorldSocketMgr.cpp
parent5321be4c0aa48daaf84ffd602d47e650e53403ef (diff)
*Update to Mangos 6888. Source: Mangos.
*Major change: Make corpse erase async, Avoid DB access on deleting guild member. *Skip 6885 because of performance concern. --HG-- branch : trunk
Diffstat (limited to 'src/game/WorldSocketMgr.cpp')
-rw-r--r--src/game/WorldSocketMgr.cpp462
1 files changed, 235 insertions, 227 deletions
diff --git a/src/game/WorldSocketMgr.cpp b/src/game/WorldSocketMgr.cpp
index a11960f33f5..baf8896f3f2 100644
--- a/src/game/WorldSocketMgr.cpp
+++ b/src/game/WorldSocketMgr.cpp
@@ -1,5 +1,5 @@
/*
-* Copyright (C) 2005-2008 MaNGOS <http://www.mangosproject.org/>
+* Copyright (C) 2005-2008,2007 MaNGOS <http://getmangos.com/>
*
* Copyright (C) 2008 Trinity <http://www.trinitycore.org/>
*
@@ -10,12 +10,12 @@
*
* This program 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
+* 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 this program; if not, write to the Free Software
-* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
*/
/** \file WorldSocketMgr.cpp
@@ -46,315 +46,323 @@
#include "Database/DatabaseEnv.h"
#include "WorldSocket.h"
-/**
-* This is a helper class to WorldSocketMgr ,that manages
+/**
+* This is a helper class to WorldSocketMgr ,that manages
* network threads, and assigning connections from acceptor thread
* to other network threads
*/
class ReactorRunnable : protected ACE_Task_Base
{
-public:
+ public:
- ReactorRunnable () :
- m_ThreadId (-1),
- m_Connections (0),
- m_Reactor (0)
- {
- ACE_Reactor_Impl* imp = 0;
+ ReactorRunnable () :
+ m_ThreadId (-1),
+ m_Connections (0),
+ m_Reactor (0)
+ {
+ ACE_Reactor_Impl* imp = 0;
-#if defined (ACE_HAS_EVENT_POLL) || defined (ACE_HAS_DEV_POLL)
+ #if defined (ACE_HAS_EVENT_POLL) || defined (ACE_HAS_DEV_POLL)
- imp = new ACE_Dev_Poll_Reactor ();
+ imp = new ACE_Dev_Poll_Reactor ();
- imp->max_notify_iterations (128);
- imp->restart (1);
+ imp->max_notify_iterations (128);
+ imp->restart (1);
-#else
+ #else
- imp = new ACE_TP_Reactor ();
- imp->max_notify_iterations (128);
+ imp = new ACE_TP_Reactor ();
+ imp->max_notify_iterations (128);
-#endif
+ #endif
- m_Reactor = new ACE_Reactor (imp, 1);
- }
+ m_Reactor = new ACE_Reactor (imp, 1);
+ }
- virtual ~ReactorRunnable ()
- {
- this->Stop ();
- this->Wait ();
+ virtual ~ReactorRunnable ()
+ {
+ Stop ();
+ Wait ();
- if (m_Reactor)
- delete m_Reactor;
- }
+ if (m_Reactor)
+ delete m_Reactor;
+ }
- void Stop ()
- {
- m_Reactor->end_reactor_event_loop ();
- }
+ void Stop ()
+ {
+ m_Reactor->end_reactor_event_loop ();
+ }
- int Start ()
- {
- if (m_ThreadId != -1)
- return -1;
+ int Start ()
+ {
+ if (m_ThreadId != -1)
+ return -1;
- return (m_ThreadId = this->activate ());
- }
+ return (m_ThreadId = activate ());
+ }
- void Wait ()
- {
- ACE_Task_Base::wait ();
- }
+ void Wait () { ACE_Task_Base::wait (); }
- long Connections ()
- {
- return static_cast<long> (m_Connections.value ());
- }
+ long Connections ()
+ {
+ return static_cast<long> (m_Connections.value ());
+ }
- int AddSocket (WorldSocket* sock)
- {
- ACE_GUARD_RETURN (ACE_Thread_Mutex, Guard, m_NewSockets_Lock, -1);
+ int AddSocket (WorldSocket* sock)
+ {
+ ACE_GUARD_RETURN (ACE_Thread_Mutex, Guard, m_NewSockets_Lock, -1);
- ++m_Connections;
- sock->AddReference();
- sock->reactor (m_Reactor);
- m_NewSockets.insert (sock);
+ ++m_Connections;
+ sock->AddReference();
+ sock->reactor (m_Reactor);
+ m_NewSockets.insert (sock);
- return 0;
- }
+ return 0;
+ }
- ACE_Reactor* GetReactor ()
- {
- return m_Reactor;
- }
+ ACE_Reactor* GetReactor ()
+ {
+ return m_Reactor;
+ }
-protected:
+ protected:
- void AddNewSockets ()
- {
- ACE_GUARD (ACE_Thread_Mutex, Guard, m_NewSockets_Lock);
+ void AddNewSockets ()
+ {
+ ACE_GUARD (ACE_Thread_Mutex, Guard, m_NewSockets_Lock);
- if (m_NewSockets.empty ())
- return;
+ if (m_NewSockets.empty ())
+ return;
- for (SocketSet::iterator i = m_NewSockets.begin (); i != m_NewSockets.end (); ++i)
- {
- WorldSocket* sock = (*i);
+ for (SocketSet::iterator i = m_NewSockets.begin (); i != m_NewSockets.end (); ++i)
+ {
+ WorldSocket* sock = (*i);
- if (sock->IsClosed ())
- {
- sock->RemoveReference ();
- --m_Connections;
- }
- else
- m_Sockets.insert (sock);
- }
+ if (sock->IsClosed ())
+ {
+ sock->RemoveReference ();
+ --m_Connections;
+ }
+ else
+ m_Sockets.insert (sock);
+ }
- m_NewSockets.clear ();
- }
+ m_NewSockets.clear ();
+ }
- virtual int svc (void)
- {
- DEBUG_LOG ("Network Thread Starting");
+ virtual int svc ()
+ {
+ DEBUG_LOG ("Network Thread Starting");
- WorldDatabase.ThreadStart ();
+ WorldDatabase.ThreadStart ();
- ACE_ASSERT (m_Reactor);
+ ACE_ASSERT (m_Reactor);
- SocketSet::iterator i, t;
+ SocketSet::iterator i, t;
- while (!m_Reactor->reactor_event_loop_done ())
- {
- // Don't be too smart to move this outside the loop
- // the run_reactor_event_loop will modify interval
- ACE_Time_Value interval (0, 10000);
+ while (!m_Reactor->reactor_event_loop_done ())
+ {
+ // dont be too smart to move this outside the loop
+ // the run_reactor_event_loop will modify interval
+ ACE_Time_Value interval (0, 10000);
- if (m_Reactor->run_reactor_event_loop (interval) == -1)
- break;
+ if (m_Reactor->run_reactor_event_loop (interval) == -1)
+ break;
- AddNewSockets ();
+ AddNewSockets ();
- for (i = m_Sockets.begin (); i != m_Sockets.end ();)
- {
- if ((*i)->Update () == -1)
- {
- t = i;
- i++;
- (*t)->CloseSocket ();
- (*t)->RemoveReference ();
- --m_Connections;
- m_Sockets.erase (t);
- }
- else
- i++;
- }
- }
-
- WorldDatabase.ThreadEnd ();
+ for (i = m_Sockets.begin (); i != m_Sockets.end ();)
+ {
+ if ((*i)->Update () == -1)
+ {
+ t = i;
+ i++;
+ (*t)->CloseSocket ();
+ (*t)->RemoveReference ();
+ --m_Connections;
+ m_Sockets.erase (t);
+ }
+ else
+ i++;
+ }
+ }
- DEBUG_LOG ("Network Thread Exitting");
-
- return 0;
- }
+ WorldDatabase.ThreadEnd ();
-private:
- typedef ACE_Atomic_Op<ACE_SYNCH_MUTEX, long> AtomicInt;
- typedef std::set<WorldSocket*> SocketSet;
+ DEBUG_LOG ("Network Thread Exitting");
- ACE_Reactor* m_Reactor;
- AtomicInt m_Connections;
- int m_ThreadId;
+ return 0;
+ }
- SocketSet m_Sockets;
+ private:
+ typedef ACE_Atomic_Op<ACE_SYNCH_MUTEX, long> AtomicInt;
+ typedef std::set<WorldSocket*> SocketSet;
- SocketSet m_NewSockets;
- ACE_Thread_Mutex m_NewSockets_Lock;
-};
+ ACE_Reactor* m_Reactor;
+ AtomicInt m_Connections;
+ int m_ThreadId;
+ SocketSet m_Sockets;
+ SocketSet m_NewSockets;
+ ACE_Thread_Mutex m_NewSockets_Lock;
+};
WorldSocketMgr::WorldSocketMgr () :
-m_NetThreadsCount (0),
-m_NetThreads (0),
-m_SockOutKBuff (-1),
-m_SockOutUBuff (65536),
-m_UseNoDelay (true),
-m_Acceptor (0) {}
+ m_NetThreadsCount (0),
+ m_NetThreads (0),
+ m_SockOutKBuff (-1),
+ m_SockOutUBuff (65536),
+ m_UseNoDelay (true),
+ m_Acceptor (0)
+{
+}
WorldSocketMgr::~WorldSocketMgr ()
{
- if (m_NetThreads)
- delete [] m_NetThreads;
+ if (m_NetThreads)
+ delete [] m_NetThreads;
- if(m_Acceptor)
- delete m_Acceptor;
+ if(m_Acceptor)
+ delete m_Acceptor;
}
-int WorldSocketMgr::StartReactiveIO (ACE_UINT16 port, const char* address)
+int
+WorldSocketMgr::StartReactiveIO (ACE_UINT16 port, const char* address)
{
- m_UseNoDelay = sConfig.GetBoolDefault ("Network.TcpNodelay", true);
+ m_UseNoDelay = sConfig.GetBoolDefault ("Network.TcpNodelay", true);
- int num_threads = sConfig.GetIntDefault ("Network.Threads", 1);
+ int num_threads = sConfig.GetIntDefault ("Network.Threads", 1);
- if (num_threads <= 0)
- {
- sLog.outError ("Network.Threads is wrong in your config file");
- return -1;
- }
+ if (num_threads <= 0)
+ {
+ sLog.outError ("Network.Threads is wrong in your config file");
+ return -1;
+ }
- m_NetThreadsCount = static_cast<size_t> (num_threads + 1);
+ m_NetThreadsCount = static_cast<size_t> (num_threads + 1);
- m_NetThreads = new ReactorRunnable[m_NetThreadsCount];
+ m_NetThreads = new ReactorRunnable[m_NetThreadsCount];
- sLog.outBasic ("Max alowed socket connections %d",ACE::max_handles ());
+ sLog.outBasic ("Max allowed socket connections %d",ACE::max_handles ());
- m_SockOutKBuff = sConfig.GetIntDefault ("Network.OutKBuff", -1); // -1 means use default
+ // -1 means use default
+ m_SockOutKBuff = sConfig.GetIntDefault ("Network.OutKBuff", -1);
- m_SockOutUBuff = sConfig.GetIntDefault ("Network.OutUBuff", 65536);
+ m_SockOutUBuff = sConfig.GetIntDefault ("Network.OutUBuff", 65536);
- if ( m_SockOutUBuff <= 0 )
- {
- sLog.outError ("Network.OutUBuff is wrong in your config file");
- return -1;
- }
+ if ( m_SockOutUBuff <= 0 )
+ {
+ sLog.outError ("Network.OutUBuff is wrong in your config file");
+ return -1;
+ }
- WorldSocket::Acceptor *acc = new WorldSocket::Acceptor;
- m_Acceptor = acc;
+ WorldSocket::Acceptor *acc = new WorldSocket::Acceptor;
+ m_Acceptor = acc;
- ACE_INET_Addr listen_addr (port, address);
+ ACE_INET_Addr listen_addr (port, address);
- if (acc->open (listen_addr, m_NetThreads[0].GetReactor (), ACE_NONBLOCK) == -1)
- {
- sLog.outError ("Failed to open acceptor ,check if the port is free");
- return -1;
- }
+ if (acc->open (listen_addr, m_NetThreads[0].GetReactor (), ACE_NONBLOCK) == -1)
+ {
+ sLog.outError ("Failed to open acceptor ,check if the port is free");
+ return -1;
+ }
- for (size_t i = 0; i < m_NetThreadsCount; ++i)
- m_NetThreads[i].Start ();
+ for (size_t i = 0; i < m_NetThreadsCount; ++i)
+ m_NetThreads[i].Start ();
- return 0;
+ return 0;
}
-int WorldSocketMgr::StartNetwork (ACE_UINT16 port, const char* address)
+int
+WorldSocketMgr::StartNetwork (ACE_UINT16 port, const char* address)
{
- if (!sLog.IsOutDebug ())
- ACE_Log_Msg::instance ()->priority_mask (LM_ERROR, ACE_Log_Msg::PROCESS);
+ if (!sLog.IsOutDebug ())
+ ACE_Log_Msg::instance ()->priority_mask (LM_ERROR, ACE_Log_Msg::PROCESS);
- if (this->StartReactiveIO (port, address) == -1)
- return -1;
+ if (StartReactiveIO (port, address) == -1)
+ return -1;
- return 0;
+ return 0;
}
-void WorldSocketMgr::StopNetwork ()
+void
+WorldSocketMgr::StopNetwork ()
{
- if (m_Acceptor)
- {
- WorldSocket::Acceptor* acc = dynamic_cast<WorldSocket::Acceptor*> (m_Acceptor);
+ if (m_Acceptor)
+ {
+ WorldSocket::Acceptor* acc = dynamic_cast<WorldSocket::Acceptor*> (m_Acceptor);
- if (acc)
- acc->close ();
- }
+ if (acc)
+ acc->close ();
+ }
- if (m_NetThreadsCount != 0)
- {
- for (size_t i = 0; i < m_NetThreadsCount; ++i)
- m_NetThreads[i].Stop ();
- }
+ if (m_NetThreadsCount != 0)
+ {
+ for (size_t i = 0; i < m_NetThreadsCount; ++i)
+ m_NetThreads[i].Stop ();
+ }
- this->Wait ();
+ Wait ();
}
-void WorldSocketMgr::Wait ()
+void
+WorldSocketMgr::Wait ()
{
- if (m_NetThreadsCount != 0)
- {
- for (size_t i = 0; i < m_NetThreadsCount; ++i)
- m_NetThreads[i].Wait ();
- }
+ if (m_NetThreadsCount != 0)
+ {
+ for (size_t i = 0; i < m_NetThreadsCount; ++i)
+ m_NetThreads[i].Wait ();
+ }
}
-int WorldSocketMgr::OnSocketOpen (WorldSocket* sock)
+int
+WorldSocketMgr::OnSocketOpen (WorldSocket* sock)
{
- // set some options here
- if (m_SockOutKBuff >= 0)
- if (sock->peer ().set_option (SOL_SOCKET,
- SO_SNDBUF,
- (void*) & m_SockOutKBuff,
- sizeof (int)) == -1 && errno != ENOTSUP)
- {
- sLog.outError ("WorldSocketMgr::OnSocketOpen set_option SO_SNDBUF");
- return -1;
- }
-
- static const int ndoption = 1;
-
- // Set TCP_NODELAY.
- if (m_UseNoDelay)
- if (sock->peer ().set_option (ACE_IPPROTO_TCP,
- TCP_NODELAY,
- (void*) & ndoption,
- sizeof (int)) == -1)
- {
- sLog.outError ("WorldSocketMgr::OnSocketOpen: peer ().set_option TCP_NODELAY errno = %s", ACE_OS::strerror (errno));
- return -1;
- }
-
- sock->m_OutBufferSize = static_cast<size_t> (m_SockOutUBuff);
-
- // we skip the Acceptor Thread
- size_t min = 1;
-
- ACE_ASSERT (m_NetThreadsCount >= 1);
-
- for (size_t i = 1; i < m_NetThreadsCount; ++i)
- if (m_NetThreads[i].Connections () < m_NetThreads[min].Connections ())
- min = i;
-
- return m_NetThreads[min].AddSocket (sock);
+ // set some options here
+ if (m_SockOutKBuff >= 0)
+ {
+ if (sock->peer ().set_option (SOL_SOCKET,
+ SO_SNDBUF,
+ (void*) & m_SockOutKBuff,
+ sizeof (int)) == -1 && errno != ENOTSUP)
+ {
+ sLog.outError ("WorldSocketMgr::OnSocketOpen set_option SO_SNDBUF");
+ return -1;
+ }
+ }
+
+ static const int ndoption = 1;
+
+ // Set TCP_NODELAY.
+ if (m_UseNoDelay)
+ {
+ if (sock->peer ().set_option (ACE_IPPROTO_TCP,
+ TCP_NODELAY,
+ (void*)&ndoption,
+ sizeof (int)) == -1)
+ {
+ sLog.outError ("WorldSocketMgr::OnSocketOpen: peer ().set_option TCP_NODELAY errno = %s", ACE_OS::strerror (errno));
+ return -1;
+ }
+ }
+
+ sock->m_OutBufferSize = static_cast<size_t> (m_SockOutUBuff);
+
+ // we skip the Acceptor Thread
+ size_t min = 1;
+
+ ACE_ASSERT (m_NetThreadsCount >= 1);
+
+ for (size_t i = 1; i < m_NetThreadsCount; ++i)
+ if (m_NetThreads[i].Connections () < m_NetThreads[min].Connections ())
+ min = i;
+
+ return m_NetThreads[min].AddSocket (sock);
}
-WorldSocketMgr* WorldSocketMgr::Instance ()
+WorldSocketMgr*
+WorldSocketMgr::Instance ()
{
- return ACE_Singleton<WorldSocketMgr,ACE_Thread_Mutex>::instance();
+ return ACE_Singleton<WorldSocketMgr,ACE_Thread_Mutex>::instance();
}