mirror of
https://github.com/mod-playerbots/azerothcore-wotlk.git
synced 2026-01-22 13:16:23 +00:00
9815025341
* feat(Core/Network): Add Proxy Protocol v2 support. * Fix codestyle and build. * Another codestyle fix. * One more missing include.
66 lines
2.0 KiB
C++
66 lines
2.0 KiB
C++
/*
|
|
* This file is part of the AzerothCore Project. See AUTHORS file for Copyright information
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify it
|
|
* under the terms of the GNU Affero General Public License as published by the
|
|
* Free Software Foundation; either version 3 of the License, or (at your
|
|
* option) any later version.
|
|
*
|
|
* 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 GNU Affero General Public License for
|
|
* more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along
|
|
* with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#ifndef AuthSocketMgr_h__
|
|
#define AuthSocketMgr_h__
|
|
|
|
#include "AuthSession.h"
|
|
#include "SocketMgr.h"
|
|
#include "Config.h"
|
|
|
|
class AuthSocketMgr : public SocketMgr<AuthSession>
|
|
{
|
|
typedef SocketMgr<AuthSession> BaseSocketMgr;
|
|
|
|
public:
|
|
static AuthSocketMgr& Instance()
|
|
{
|
|
static AuthSocketMgr instance;
|
|
return instance;
|
|
}
|
|
|
|
bool StartNetwork(Acore::Asio::IoContext& ioContext, std::string const& bindIp, uint16 port, int threadCount = 1) override
|
|
{
|
|
if (!BaseSocketMgr::StartNetwork(ioContext, bindIp, port, threadCount))
|
|
return false;
|
|
|
|
_acceptor->AsyncAcceptWithCallback<&AuthSocketMgr::OnSocketAccept>();
|
|
return true;
|
|
}
|
|
|
|
protected:
|
|
NetworkThread<AuthSession>* CreateThreads() const override
|
|
{
|
|
NetworkThread<AuthSession>* threads = new NetworkThread<AuthSession>[1];
|
|
|
|
bool proxyProtocolEnabled = sConfigMgr->GetOption<bool>("EnableProxyProtocol", false, true);
|
|
if (proxyProtocolEnabled)
|
|
threads[0].EnableProxyProtocol();
|
|
|
|
return threads;
|
|
}
|
|
|
|
static void OnSocketAccept(tcp::socket&& sock, uint32 threadIndex)
|
|
{
|
|
Instance().OnSocketOpen(std::forward<tcp::socket>(sock), threadIndex);
|
|
}
|
|
};
|
|
|
|
#define sAuthSocketMgr AuthSocketMgr::Instance()
|
|
|
|
#endif // AuthSocketMgr_h__
|