117 lines
5.0 KiB
C++
117 lines
5.0 KiB
C++
/*****************************************************************************
|
|
* PokerTH - The open source texas holdem engine *
|
|
* Copyright (C) 2006-2012 Felix Hammer, Florian Thauer, Lothar May *
|
|
* *
|
|
* 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 Affero General Public License *
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>. *
|
|
* *
|
|
* *
|
|
* Additional permission under GNU AGPL version 3 section 7 *
|
|
* *
|
|
* If you modify this program, or any covered work, by linking or *
|
|
* combining it with the OpenSSL project's OpenSSL library (or a *
|
|
* modified version of that library), containing parts covered by the *
|
|
* terms of the OpenSSL or SSLeay licenses, the authors of PokerTH *
|
|
* (Felix Hammer, Florian Thauer, Lothar May) grant you additional *
|
|
* permission to convey the resulting work. *
|
|
* Corresponding Source for a non-source form of such a combination *
|
|
* shall include the source code for the parts of OpenSSL used as well *
|
|
* as that of the covered work. *
|
|
*****************************************************************************/
|
|
|
|
#include <net/validation/pokerthmessagevalidator.h>
|
|
#include <net/validation/authmessagevalidator.h>
|
|
#include <net/validation/lobbymessagevalidator.h>
|
|
#include <net/validation/gamemessagevalidator.h>
|
|
#include <net/validation/validationhelper.h>
|
|
#include <boost/mem_fn.hpp>
|
|
#include <boost/bind.hpp>
|
|
|
|
using namespace std;
|
|
|
|
|
|
PokerTHMessageValidator::PokerTHMessageValidator()
|
|
{
|
|
m_validationMap.insert(make_pair(PokerTHMessage_PokerTHMessageType_Type_AnnounceMessage, ValidateAnnounceMessage));
|
|
m_validationMap.insert(make_pair(PokerTHMessage_PokerTHMessageType_Type_AuthMessage, boost::bind(boost::mem_fn(&PokerTHMessageValidator::ValidateAuthMessage), this, _1)));
|
|
m_validationMap.insert(make_pair(PokerTHMessage_PokerTHMessageType_Type_LobbyMessage, boost::bind(boost::mem_fn(&PokerTHMessageValidator::ValidateLobbyMessage), this, _1)));
|
|
m_validationMap.insert(make_pair(PokerTHMessage_PokerTHMessageType_Type_GameMessage, boost::bind(boost::mem_fn(&PokerTHMessageValidator::ValidateGameMessage), this, _1)));
|
|
|
|
m_authValidator.reset(new AuthMessageValidator);
|
|
m_lobbyValidator.reset(new LobbyMessageValidator);
|
|
m_gameValidator.reset(new GameMessageValidator);
|
|
}
|
|
|
|
bool
|
|
PokerTHMessageValidator::IsValidMessage(const PokerTHMessage &msg) const
|
|
{
|
|
// Default: Invalid packet.
|
|
bool retVal = false;
|
|
ValidateFunctorMap::const_iterator pos = m_validationMap.find(msg.messagetype());
|
|
if (pos != m_validationMap.end()) {
|
|
// Call validation functor.
|
|
retVal = pos->second(msg);
|
|
}
|
|
return retVal;
|
|
}
|
|
|
|
bool
|
|
PokerTHMessageValidator::ValidateAnnounceMessage(const PokerTHMessage &msg)
|
|
{
|
|
bool retVal = false;
|
|
if (msg.has_announcemessage()) {
|
|
const AnnounceMessage &announce = msg.announcemessage();
|
|
if (VALIDATE_IS_UINT16(announce.protocolversion().majorversion())
|
|
&& VALIDATE_IS_UINT16(announce.protocolversion().minorversion())
|
|
&& VALIDATE_IS_UINT16(announce.latestgameversion().majorversion())
|
|
&& VALIDATE_IS_UINT16(announce.latestgameversion().minorversion())
|
|
&& VALIDATE_IS_UINT16(announce.latestbetarevision())
|
|
&& VALIDATE_IS_UINT16(announce.numplayersonserver())) {
|
|
|
|
retVal = true;
|
|
}
|
|
}
|
|
return retVal;
|
|
}
|
|
|
|
bool
|
|
PokerTHMessageValidator::ValidateAuthMessage(const PokerTHMessage &msg)
|
|
{
|
|
bool retVal = false;
|
|
if (msg.has_authmessage()) {
|
|
retVal = m_authValidator->IsValidMessage(msg.authmessage());
|
|
}
|
|
return retVal;
|
|
}
|
|
|
|
bool
|
|
PokerTHMessageValidator::ValidateLobbyMessage(const PokerTHMessage &msg)
|
|
{
|
|
bool retVal = false;
|
|
if (msg.has_lobbymessage()) {
|
|
retVal = m_lobbyValidator->IsValidMessage(msg.lobbymessage());
|
|
}
|
|
return retVal;
|
|
}
|
|
|
|
bool
|
|
PokerTHMessageValidator::ValidateGameMessage(const PokerTHMessage &msg)
|
|
{
|
|
bool retVal = false;
|
|
if (msg.has_gamemessage()) {
|
|
retVal = m_gameValidator->IsValidMessage(msg.gamemessage());
|
|
}
|
|
return retVal;
|
|
}
|
|
|