merge fix

This commit is contained in:
Andre Antunes 2011-04-09 22:27:52 -03:00
commit d5e2115599
11 changed files with 561 additions and 42 deletions

View File

@ -12,6 +12,7 @@ FIND_PACKAGE(Lua51 REQUIRED)
FIND_PACKAGE(YamlCpp REQUIRED) FIND_PACKAGE(YamlCpp REQUIRED)
FIND_PACKAGE(PhysFS REQUIRED) FIND_PACKAGE(PhysFS REQUIRED)
FIND_PACKAGE(PNG REQUIRED) FIND_PACKAGE(PNG REQUIRED)
FIND_PACKAGE(GMP REQUIRED)
# choose a default build type if not specified # choose a default build type if not specified
IF(NOT CMAKE_BUILD_TYPE) IF(NOT CMAKE_BUILD_TYPE)
@ -33,6 +34,7 @@ INCLUDE_DIRECTORIES(
${LUA_INCLUDE_DIRS} ${LUA_INCLUDE_DIRS}
${YAMLCPP_INCLUDE_DIRS} ${YAMLCPP_INCLUDE_DIRS}
${PHYSFS_INCLUDE_DIRS} ${PHYSFS_INCLUDE_DIRS}
${GMP_INCLUDE_DIR}
${PNG_INCLUDE_DIRS}) ${PNG_INCLUDE_DIRS})
LINK_DIRECTORIES( LINK_DIRECTORIES(
@ -55,6 +57,9 @@ SET(SOURCES
src/menustate.cpp src/menustate.cpp
src/teststate.cpp src/teststate.cpp
# game net
src/net/protocoltibia87.cpp
# framework sources # framework sources
src/framework/image.cpp src/framework/image.cpp
src/framework/borderedimage.cpp src/framework/borderedimage.cpp
@ -83,7 +88,12 @@ SET(SOURCES
# network # network
src/framework/net/connection.cpp src/framework/net/connection.cpp
src/framework/net/connections.cpp) src/framework/net/connections.cpp
src/framework/net/protocol.cpp
src/framework/net/networkmessage.cpp
# util
src/framework/util/rsa.cpp)
IF(WIN32) IF(WIN32)
SET(SOURCES ${SOURCES} src/framework/win32platform.cpp) SET(SOURCES ${SOURCES} src/framework/win32platform.cpp)
@ -105,4 +115,5 @@ TARGET_LINK_LIBRARIES(otclient
${YAMLCPP_LIBRARY} ${YAMLCPP_LIBRARY}
${PHYSFS_LIBRARY} ${PHYSFS_LIBRARY}
${PNG_LIBRARY} ${PNG_LIBRARY}
${GMP_LIBRARIES}
${ADDITIONAL_LIBRARIES}) ${ADDITIONAL_LIBRARIES})

View File

@ -43,49 +43,140 @@ void Connection::stop()
} }
} }
void Connection::connect(const std::string& ip, uint16 port) bool Connection::connect(const std::string& ip, uint16 port, ConnectionCallback onConnect)
{ {
if(m_connecting){ if(m_connecting){
logError("Already is connecting."); logError("Already is connecting.");
return; return false;
} }
if(m_connected){
logError("Already is connected.");
return false;
}
m_connectCallback = onConnect;
m_connecting = true; m_connecting = true;
m_ip = ip; m_ip = ip;
m_port = port; m_port = port;
logDebug("connecting...");
//first resolve dns //first resolve dns
boost::asio::ip::tcp::resolver::query query(ip, convertType<std::string, uint16>(port)); boost::asio::ip::tcp::resolver::query query(ip, convertType<std::string, uint16>(port));
m_resolver.async_resolve(query, boost::bind(&Connection::onResolveDns, this, boost::asio::placeholders::error, boost::asio::placeholders::iterator)); m_resolver.async_resolve(query, boost::bind(&Connection::onResolveDns, this, boost::asio::placeholders::error, boost::asio::placeholders::iterator));
return true;
} }
void Connection::onResolveDns(const boost::system::error_code& error, boost::asio::ip::tcp::resolver::iterator endpoint_iterator) void Connection::onResolveDns(const boost::system::error_code& error, boost::asio::ip::tcp::resolver::iterator endpointIt)
{ {
logDebug("resolving dns..");
m_lastError = error;
if(error){ if(error){
m_connecting = false; m_connecting = false;
m_errorCallback(error, __FUNCTION__);
return; return;
} }
//lets connect //lets connect
m_socket.async_connect(*endpoint_iterator, boost::bind(&Connection::onConnect, this, boost::asio::placeholders::error)); m_socket.async_connect(*endpointIt, boost::bind(&Connection::onConnect, this, boost::asio::placeholders::error));
} }
void Connection::onConnect(const boost::system::error_code& error) void Connection::onConnect(const boost::system::error_code& error)
{ {
m_lastError = error;
if(error){ if(error){
m_connecting = false; m_connecting = false;
m_errorCallback(error, __FUNCTION__);
return; return;
} }
m_connected = true; m_connected = true;
logInfo("Connected on %s.", m_ip.c_str()); m_connectCallback();
}
void Connection::handleError(const boost::system::error_code& error)
{
stop();
if(isConnected()){
closeSocket();
}
}
void Connection::closeSocket()
{
boost::system::error_code error;
m_socket.shutdown(boost::asio::ip::tcp::socket::shutdown_both, error);
if(error) {
logError("Connection::closeSocket(): %s", error.message().c_str());
}
m_socket.close(error);
if(error) {
logError("Connection::closeSocket(): %s", error.message().c_str());
}
}
void Connection::send(NetworkMessagePtr networkMessage, ConnectionCallback onSend)
{
boost::asio::async_write(m_socket,
boost::asio::buffer(networkMessage->getBuffer(), NetworkMessage::header_length),
boost::bind(&Connection::onSendHeader, shared_from_this(), networkMessage, onSend, boost::asio::placeholders::error));
}
void Connection::recv(RecvCallback onRecv)
{
NetworkMessagePtr networkMessage(new NetworkMessage);
boost::asio::async_read(m_socket,
boost::asio::buffer(networkMessage->getBuffer(), NetworkMessage::header_length),
boost::bind(&Connection::onRecvHeader, shared_from_this(), networkMessage, onRecv, boost::asio::placeholders::error));
}
void Connection::onRecvHeader(ConnectionPtr connection, NetworkMessagePtr networkMessage, RecvCallback onRecv, const boost::system::error_code& error)
{
if(error){
connection->handleError(error);
connection->onError(error, __FUNCTION__);
return;
}
boost::asio::async_read(connection->getSocket(),
boost::asio::buffer(networkMessage->getBodyBuffer(), networkMessage->getMessageLength()),
boost::bind(&Connection::onRecvBody, connection, networkMessage, onRecv, boost::asio::placeholders::error));
}
void Connection::onRecvBody(ConnectionPtr connection, NetworkMessagePtr networkMessage, RecvCallback onRecv, const boost::system::error_code& error)
{
if(error){
connection->handleError(error);
connection->onError(error, __FUNCTION__);
return;
}
onRecv(networkMessage);
}
void Connection::onSendHeader(ConnectionPtr connection, NetworkMessagePtr networkMessage, ConnectionCallback onSend, const boost::system::error_code& error)
{
if(error){
connection->handleError(error);
connection->onError(error, __FUNCTION__);
return;
}
boost::asio::async_write(connection->getSocket(),
boost::asio::buffer(networkMessage->getBodyBuffer(), networkMessage->getMessageLength()),
boost::bind(&Connection::onSendBody, connection, networkMessage, onSend, boost::asio::placeholders::error));
}
void Connection::onSendBody(ConnectionPtr connection, NetworkMessagePtr networkMessage, ConnectionCallback onSend, const boost::system::error_code& error)
{
if(error){
connection->handleError(error);
connection->onError(error, __FUNCTION__);
return;
}
onSend();
} }

View File

@ -28,35 +28,70 @@
#include <boost/asio.hpp> #include <boost/asio.hpp>
class Connection #include "networkmessage.h"
class TestState;
class Protocol;
class Connections;
class Connection : public std::enable_shared_from_this<Connection>
{ {
public: public:
typedef std::function<void()> ConnectionCallback;
typedef std::function<void(NetworkMessagePtr)> RecvCallback;
typedef std::function<void(const boost::system::error_code&, const std::string&)> ErrorCallback;
typedef std::shared_ptr<Connection> ConnectionPtr;
private:
Connection(boost::asio::io_service& ioService); Connection(boost::asio::io_service& ioService);
void connect(const std::string& ip, uint16 port); bool connect(const std::string& ip, uint16 port, ConnectionCallback onConnect);
void stop(); void stop();
void setErrorCallback(ErrorCallback c) { m_errorCallback = c; }
void recv(RecvCallback onSend);
void send(NetworkMessagePtr networkMessage, ConnectionCallback onRecv);
bool isConnecting() const { return m_connecting; } bool isConnecting() const { return m_connecting; }
bool isConnected() const { return m_connected; } bool isConnected() const { return m_connected; }
const boost::system::error_code& getLastError() const { return m_lastError; } boost::asio::ip::tcp::socket& getSocket() { return m_socket; }
void resetLastError() { m_lastError = boost::system::error_code(); } void onError(const boost::system::error_code& error, const std::string& msg) { m_errorCallback(error, msg); }
private: private:
void onResolveDns(const boost::system::error_code& error, boost::asio::ip::tcp::resolver::iterator endpoint_iterator); static void onSendHeader(ConnectionPtr connection, NetworkMessagePtr networkMessage, ConnectionCallback onSend, const boost::system::error_code& error);
static void onSendBody(ConnectionPtr connection, NetworkMessagePtr networkMessage, ConnectionCallback onSend, const boost::system::error_code& error);
static void onRecvHeader(ConnectionPtr connection, NetworkMessagePtr networkMessage, RecvCallback onRecv, const boost::system::error_code& error);
static void onRecvBody(ConnectionPtr connection, NetworkMessagePtr networkMessage, RecvCallback onRecv, const boost::system::error_code& error);
private:
void onResolveDns(const boost::system::error_code& error, boost::asio::ip::tcp::resolver::iterator endpointIt);
void onConnect(const boost::system::error_code& error); void onConnect(const boost::system::error_code& error);
private: private:
void closeSocket();
private:
void handleError(const boost::system::error_code& error);
boost::asio::ip::tcp::socket m_socket; boost::asio::ip::tcp::socket m_socket;
boost::asio::ip::tcp::resolver m_resolver; boost::asio::ip::tcp::resolver m_resolver;
boost::system::error_code m_lastError;
bool m_connecting; bool m_connecting;
bool m_connected; bool m_connected;
std::string m_ip; std::string m_ip;
uint16_t m_port; uint16_t m_port;
ConnectionCallback m_connectCallback;
ErrorCallback m_errorCallback;
friend class Protocol;
friend class Connections;
}; };
typedef std::shared_ptr<Connection> ConnectionPtr; typedef std::shared_ptr<Connection> ConnectionPtr;

View File

@ -0,0 +1,178 @@
/* The MIT License
*
* Copyright (c) 2010 OTClient, https://github.com/edubart/otclient
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
* THE SOFTWARE.
*/
#include "networkmessage.h"
void NetworkMessage::updateHeaderLength()
{
uint16 size = m_msgSize;
memcpy(m_msgBuf, &size, 2);
}
bool NetworkMessage::canAdd(int size) {
return (size + m_readPos < NETWORKMESSAGE_MAXSIZE - 16);
}
std::string NetworkMessage::getString()
{
uint16 stringlen = getU16();
if(stringlen >= (16384 - m_readPos))
return std::string();
char* v = (char*)(m_msgBuf + m_readPos);
m_readPos += stringlen;
return std::string(v, stringlen);
}
std::string NetworkMessage::getRaw()
{
uint16 stringlen = m_msgSize - m_readPos;
if(stringlen >= (16384 - m_readPos))
return std::string();
char* v = (char*)(m_msgBuf + m_readPos);
m_readPos += stringlen;
return std::string(v, stringlen);
}
void NetworkMessage::addString(const char* value)
{
uint32 stringlen = (uint32)strlen(value);
if(!canAdd(stringlen + 2) || stringlen > 8192)
return;
addU16(stringlen);
strcpy((char*)(m_msgBuf + m_readPos), value);
m_readPos += stringlen;
m_msgSize += stringlen;
}
void NetworkMessage::addBytes(const char* bytes, uint32 size)
{
if(!canAdd(size) || size > 8192)
return;
memcpy(m_msgBuf + m_readPos, bytes, size);
m_readPos += size;
m_msgSize += size;
}
void NetworkMessage::addPaddingBytes(uint32 n)
{
if(!canAdd(n))
return;
memset((void*)&m_msgBuf[m_readPos], 0x33, n);
m_msgSize = m_msgSize + n;
}
void NetworkMessage::skipBytes(int count) {
m_readPos += count;
}
// simply write functions for outgoing message
void NetworkMessage::addByte(uint8 value) {
if(!canAdd(1))
return;
m_msgBuf[m_readPos++] = value;
m_msgSize++;
}
void NetworkMessage::addU16(uint16 value) {
if(!canAdd(2))
return;
*(uint16*)(m_msgBuf + m_readPos) = value;
m_readPos += 2;
m_msgSize += 2;
}
void NetworkMessage::addU32(uint32 value) {
if(!canAdd(4))
return;
*(uint32*)(m_msgBuf + m_readPos) = value;
m_readPos += 4;
m_msgSize += 4;
}
void NetworkMessage::addU64(uint64 value) {
if(!canAdd(8))
return;
*(uint64*)(m_msgBuf + m_readPos) = value;
m_readPos += 8;
m_msgSize += 8;
}
void NetworkMessage::addString(const std::string &value) {
addString(value.c_str());
}
int32 NetworkMessage::getMessageLength() const {
return m_msgSize;
}
void NetworkMessage::setMessageLength(int32 newSize) {
m_msgSize = newSize;
}
int32 NetworkMessage::getReadPos() const {
return m_readPos;
}
uint8 NetworkMessage::getByte()
{
return m_msgBuf[m_readPos++];
}
uint16 NetworkMessage::getU16()
{
uint16 v = *(uint16*)(m_msgBuf + m_readPos);
m_readPos += 2;
return v;
}
uint32 NetworkMessage::getU32()
{
uint32 v = *(uint32*)(m_msgBuf + m_readPos);
m_readPos += 4;
return v;
}
uint64 NetworkMessage::getU64()
{
uint64 v = *(uint64*)(m_msgBuf + m_readPos);
m_readPos += 8;
return v;
}
char* NetworkMessage::getBuffer() {
return (char*)&m_msgBuf[0];
}
char* NetworkMessage::getBodyBuffer() {
m_readPos = 2;
return (char*)&m_msgBuf[header_length];
}

View File

@ -0,0 +1,97 @@
/* The MIT License
*
* Copyright (c) 2010 OTClient, https://github.com/edubart/otclient
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
* THE SOFTWARE.
*/
#ifndef NETWORKMESSAGE_H
#define NETWORKMESSAGE_H
#include "../prerequisites.h"
class Rsa;
class NetworkMessage
{
public:
enum {
header_length = 2,
NETWORKMESSAGE_MAXSIZE = 1500
};
enum {
max_body_length = NETWORKMESSAGE_MAXSIZE - header_length
};
// constructor/destructor
NetworkMessage() {
reset();
}
// resets the internal buffer to an empty message
protected:
void reset() {
m_msgSize = 0;
m_readPos = 2;
}
public:
// simply read functions for incoming message
uint8 getByte();
uint16 getU16();
uint32 getU32();
uint64 getU64();
std::string getString();
std::string getRaw();
// skips count unknown/unused bytes in an incoming message
void skipBytes(int count);
// simply write functions for outgoing message
void addByte(uint8 value);
void addU16(uint16 value);
void addU32(uint32 value);
void addU64(uint64 value);
void addBytes(const char* bytes, uint32_t size);
void addPaddingBytes(uint32 n);
void addString(const std::string &value);
void addString(const char* value);
int32 getMessageLength() const;
void setMessageLength(int32 newSize);
int32 getReadPos() const;
int32 getHeaderSize();
char* getBuffer();
char* getBodyBuffer();
void updateHeaderLength();
protected:
inline bool canAdd(int size);
int32 m_msgSize;
int32 m_readPos;
uint8 m_msgBuf[NETWORKMESSAGE_MAXSIZE];
};
typedef std::shared_ptr<NetworkMessage> NetworkMessagePtr;
#endif //NETWORKMESSAGE_H

View File

@ -0,0 +1,50 @@
/* The MIT License
*
* Copyright (c) 2010 OTClient, https://github.com/edubart/otclient
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
* THE SOFTWARE.
*/
#include "protocol.h"
#include "connections.h"
Protocol::Protocol()
{
m_connection = g_connections.createConnection();
m_connection->setErrorCallback(
[this](const boost::system::error_code& error, const std::string& msg){
this->onError(error, msg);
}
);
}
void Protocol::send(NetworkMessagePtr networkMessage, Connection::ConnectionCallback onSend)
{
m_connection->send(networkMessage, onSend);
}
bool Protocol::connect(const std::string& ip, uint16 port, Connection::ConnectionCallback onConnect)
{
return m_connection->connect(ip, port, onConnect);
}
void Protocol::recv(Connection::RecvCallback onRecv)
{
m_connection->recv(onRecv);
}

View File

@ -0,0 +1,48 @@
/* The MIT License
*
* Copyright (c) 2010 OTClient, https://github.com/edubart/otclient
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
* THE SOFTWARE.
*/
#ifndef PROTOCOL_H
#define PROTOCOL_H
#include "../prerequisites.h"
#include "connection.h"
class Protocol
{
public:
Protocol();
virtual void begin() = 0;
protected:
void send(NetworkMessagePtr networkMessage, Connection::ConnectionCallback onSend);
void recv(Connection::RecvCallback onRecv);
bool connect(const std::string& ip, uint16 port, Connection::ConnectionCallback onConnect);
virtual void onError(const boost::system::error_code& error, const std::string& msg) = 0;
ConnectionPtr m_connection;
};
#endif //PROTOCOL_H

View File

@ -31,6 +31,7 @@ typedef unsigned char uchar;
typedef unsigned short ushort; typedef unsigned short ushort;
typedef unsigned int uint; typedef unsigned int uint;
typedef unsigned long ulong; typedef unsigned long ulong;
typedef uint64_t uint64;
typedef uint32_t uint32; typedef uint32_t uint32;
typedef uint16_t uint16; typedef uint16_t uint16;
typedef uint8_t uint8; typedef uint8_t uint8;

View File

@ -28,9 +28,14 @@
#include "framework/engine.h" #include "framework/engine.h"
#include "framework/input.h" #include "framework/input.h"
#include "framework/net/connections.h"
#include "net/protocoltibia87.h"
void TestState::onEnter() void TestState::onEnter()
{ {
m_protocol = ProtocolTibia87Ptr(new ProtocolTibia87);
m_protocol->begin();
} }
void TestState::onLeave() void TestState::onLeave()
@ -62,4 +67,3 @@ void TestState::update(int ticks, int elapsedTicks)
{ {
} }

View File

@ -26,6 +26,7 @@
#define TESTSTATE_H #define TESTSTATE_H
#include "framework/gamestate.h" #include "framework/gamestate.h"
#include "net/protocoltibia87.h"
class TestState : public GameState class TestState : public GameState
{ {
@ -39,8 +40,11 @@ public:
void onInputEvent(InputEvent *event); void onInputEvent(InputEvent *event);
void onResize(const Size& size); void onResize(const Size& size);
void render(); virtual void render();
void update(int ticks, int elapsedTicks); virtual void update(int ticks, int elapsedTicks);
private:
ProtocolTibia87Ptr m_protocol;
}; };
#endif // TESTSTATE_H #endif // TESTSTATE_H