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

View File

@ -38,54 +38,145 @@ void Connection::stop()
if(m_connecting){
m_resolver.cancel();
m_socket.cancel();
m_connecting = false;
}
}
void Connection::connect(const std::string& ip, uint16 port)
bool Connection::connect(const std::string& ip, uint16 port, ConnectionCallback onConnect)
{
if(m_connecting){
logError("Already is connecting.");
return;
return false;
}
if(m_connected){
logError("Already is connected.");
return false;
}
m_connectCallback = onConnect;
m_connecting = true;
m_ip = ip;
m_port = port;
logDebug("connecting...");
//first resolve dns
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));
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){
m_connecting = false;
m_errorCallback(error, __FUNCTION__);
return;
}
//lets connect
m_socket.async_connect(*endpoint_iterator, boost::bind(&Connection::onConnect, this, boost::asio::placeholders::error));
//lets connect
m_socket.async_connect(*endpointIt, boost::bind(&Connection::onConnect, this, boost::asio::placeholders::error));
}
void Connection::onConnect(const boost::system::error_code& error)
{
m_lastError = error;
if(error){
m_connecting = false;
m_errorCallback(error, __FUNCTION__);
return;
}
}
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>
class Connection
#include "networkmessage.h"
class TestState;
class Protocol;
class Connections;
class Connection : public std::enable_shared_from_this<Connection>
{
public:
Connection(boost::asio::io_service& ioService);
void connect(const std::string& ip, uint16 port);
void stop();
bool isConnecting() const { return m_connecting; }
bool isConnected() const { return m_connected; }
const boost::system::error_code& getLastError() const { return m_lastError; }
typedef std::function<void()> ConnectionCallback;
typedef std::function<void(NetworkMessagePtr)> RecvCallback;
typedef std::function<void(const boost::system::error_code&, const std::string&)> ErrorCallback;
void resetLastError() { m_lastError = boost::system::error_code(); }
typedef std::shared_ptr<Connection> ConnectionPtr;
private:
Connection(boost::asio::io_service& ioService);
bool connect(const std::string& ip, uint16 port, ConnectionCallback onConnect);
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 isConnected() const { return m_connected; }
boost::asio::ip::tcp::socket& getSocket() { return m_socket; }
void onError(const boost::system::error_code& error, const std::string& msg) { m_errorCallback(error, msg); }
private:
void onResolveDns(const boost::system::error_code& error, boost::asio::ip::tcp::resolver::iterator endpoint_iterator);
void onConnect(const boost::system::error_code& error);
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);
private:
void closeSocket();
private:
void handleError(const boost::system::error_code& error);
boost::asio::ip::tcp::socket m_socket;
boost::asio::ip::tcp::resolver m_resolver;
boost::system::error_code m_lastError;
bool m_connecting;
bool m_connected;
std::string m_ip;
uint16_t m_port;
ConnectionCallback m_connectCallback;
ErrorCallback m_errorCallback;
friend class Protocol;
friend class Connections;
};
typedef std::shared_ptr<Connection> ConnectionPtr;

View File

@ -34,6 +34,6 @@ ConnectionPtr Connections::createConnection()
{
ConnectionPtr connection(new Connection(m_ioService));
m_connections.push_back(connection);
return connection;
}

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 int uint;
typedef unsigned long ulong;
typedef uint64_t uint64;
typedef uint32_t uint32;
typedef uint16_t uint16;
typedef uint8_t uint8;

View File

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

View File

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