mirror of
https://github.com/araxiaonline/TrinityCore.git
synced 2026-06-20 15:01:38 -04:00
Shauren
114 lines
3.3 KiB
C++
114 lines
3.3 KiB
C++
/*
|
|
* Copyright (C) 2008-2014 TrinityCore <http://www.trinitycore.org/>
|
|
*
|
|
* This program 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 2 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 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 _BATTLENETSOCKET_H
|
|
#define _BATTLENETSOCKET_H
|
|
|
|
#include "RealmSocket.h"
|
|
#include "BattlenetPackets.h"
|
|
#include "BattlenetPacketCrypt.h"
|
|
#include "BigNumber.h"
|
|
|
|
class ACE_INET_Addr;
|
|
|
|
namespace Battlenet
|
|
{
|
|
struct PacketHeader;
|
|
class BitStream;
|
|
|
|
enum ModuleType
|
|
{
|
|
MODULE_PASSWORD,
|
|
MODULE_TOKEN,
|
|
MODULE_THUMBPRINT,
|
|
MODULE_SELECT_GAME_ACCOUNT,
|
|
MODULE_RISK_FINGERPRINT,
|
|
|
|
MODULE_COUNT
|
|
};
|
|
|
|
class Socket : public RealmSocket::Session
|
|
{
|
|
public:
|
|
static uint32 const SRP6_V_Size;
|
|
static uint32 const SRP6_S_Size;
|
|
|
|
Socket(RealmSocket& socket);
|
|
|
|
typedef bool(Socket::*PacketHandler)(PacketHeader& socket, BitStream& packet);
|
|
|
|
// Auth
|
|
bool HandleAuthChallenge(PacketHeader& header, BitStream& packet);
|
|
bool HandleAuthProofResponse(PacketHeader& header, BitStream& packet);
|
|
|
|
// Creep
|
|
bool HandlePing(PacketHeader& header, BitStream& packet);
|
|
bool HandleEnableEncryption(PacketHeader& header, BitStream& packet);
|
|
|
|
// WoW
|
|
bool HandleRealmUpdate(PacketHeader& header, BitStream& packet);
|
|
bool HandleRealmJoinRequest(PacketHeader& header, BitStream& packet);
|
|
|
|
void OnRead() override;
|
|
void OnAccept() override;
|
|
void OnClose() override;
|
|
|
|
void Send(ServerPacket& packet);
|
|
|
|
private:
|
|
void _SetVSFields(std::string const& rI);
|
|
static ACE_INET_Addr const& GetAddressForClient(Realm const& realm, ACE_INET_Addr const& clientAddr);
|
|
|
|
typedef bool(Socket::*ModuleHandler)(BitStream* dataStream, ServerPacket** response);
|
|
static ModuleHandler const ModuleHandlers[MODULE_COUNT];
|
|
|
|
bool HandlePasswordModule(BitStream* dataStream, ServerPacket** response);
|
|
bool HandleSelectGameAccountModule(BitStream* dataStream, ServerPacket** response);
|
|
bool HandleRiskFingerprintModule(BitStream* dataStream, ServerPacket** response);
|
|
bool UnhandledModule(BitStream* dataStream, ServerPacket** response);
|
|
|
|
RealmSocket& _socket;
|
|
|
|
uint32 _accountId;
|
|
std::string _accountName;
|
|
std::string _locale;
|
|
std::string _os;
|
|
uint32 _build;
|
|
uint32 _gameAccountId;
|
|
AccountTypes _accountSecurityLevel;
|
|
|
|
BigNumber N;
|
|
BigNumber g;
|
|
BigNumber k;
|
|
|
|
BigNumber I;
|
|
BigNumber s;
|
|
BigNumber v;
|
|
|
|
BigNumber b;
|
|
BigNumber B;
|
|
BigNumber K; // session key
|
|
|
|
std::queue<ModuleType> _modulesWaitingForData;
|
|
|
|
PacketCrypt _crypt;
|
|
};
|
|
|
|
}
|
|
|
|
#endif // _BATTLENETSOCKET_H
|