-//
-// Session.cpp
-//
-
-#include "Command.hpp"
-#include "CommandHeader.hpp"
-#include "Session.hpp"
-#include "Utils.hpp"
-#include "../Logger.hpp"
-#include <boost/make_shared.hpp>
-#include <string>
-
-namespace network {
-
- Session::Session(boost::asio::io_service& io_service_tcp) :
- io_service_tcp_(io_service_tcp),
- socket_tcp_(io_service_tcp),
- encryption_(false),
- online_(true),
- login_(false),
- read_start_time_(time(nullptr)),
- write_start_time_(time(nullptr)),
- read_byte_sum_(0),
- write_byte_sum_(0),
- serialized_byte_sum_(0),
- compressed_byte_sum_(0),
- id_(0)
- {
-
- }
-
- Session::~Session()
- {
- Close();
- }
-
- void Session::Close()
- {
- socket_tcp_.close();
- }
-
- void Session::Send(const Command& command)
- {
- auto msg = Serialize(command);
- write_byte_sum_ += msg.size();
- UpdateWriteByteAverage();
-
- io_service_tcp_.post(boost::bind(&Session::DoWriteTCP, this, msg, shared_from_this()));
- }
-
- void Session::SyncSend(const Command& command)
- {
- auto msg = Serialize(command);
- write_byte_sum_ += msg.size();
- UpdateWriteByteAverage();
-
- try {
- boost::asio::write(
- socket_tcp_, boost::asio::buffer(msg.data(), msg.size()),
- boost::asio::transfer_all());
- } catch (std::exception& e) {
- std::cout << e.what() << std::endl;
- }
- }
-
- double Session::GetReadByteAverage() const
- {
- return 1.0f * read_byte_sum_ / (time(nullptr) - read_start_time_);
- }
-
- double Session::GetWriteByteAverage() const
- {
- return 1.0f * write_byte_sum_ / (time(nullptr) - write_start_time_);
- }
-
- void Session::UpdateReadByteAverage()
- {
- unsigned long elapsed_time = time(nullptr) - read_start_time_;
- if (elapsed_time >= BYTE_AVERAGE_REFRESH_SECONDS) {
- read_byte_sum_ /= 2;
- read_start_time_ = time(nullptr) - elapsed_time / 2;
- }
- }
-
- void Session::UpdateWriteByteAverage()
- {
- unsigned long elapsed_time = time(nullptr) - write_start_time_;
- if (elapsed_time >= BYTE_AVERAGE_REFRESH_SECONDS) {
- write_byte_sum_ /= 2;
- write_start_time_ = time(nullptr) - elapsed_time / 2;
- }
- }
-
- void Session::EnableEncryption()
- {
- encryption_ = true;
- }
-
- Encrypter& Session::encrypter()
- {
- return encrypter_;
- }
-
- tcp::socket& Session::tcp_socket()
- {
- return socket_tcp_;
- }
-
- UserID Session::id() const
- {
- return id_;
- }
-
- void Session::set_id(UserID id)
- {
- id_ = id;
- }
-
- bool Session::online() const
- {
- return online_;
- }
-
- std::string Session::global_ip() const
- {
- return global_ip_;
- }
-
- uint16_t Session::udp_port() const{
- return udp_port_;
- }
-
- void Session::set_global_ip(const std::string& global_ip)
- {
- global_ip_ = global_ip;
- }
-
- void Session::set_udp_port(uint16_t udp_port)
- {
- udp_port_ = udp_port;
- }
-
- int Session::serialized_byte_sum() const
- {
- return serialized_byte_sum_;
- }
-
- int Session::compressed_byte_sum() const
- {
- return compressed_byte_sum_;
- }
-
- bool Session::operator==(const Session& s)
- {
- return id_ == s.id_;
- }
-
- bool Session::operator!=(const Session& s)
- {
- return !operator==(s);
- }
-
- std::string Session::Serialize(const Command& command)
- {
- assert(command.header() < 0xFF);
+//\r
+// Session.cpp\r
+//\r
+\r
+#include "Command.hpp"\r
+#include "CommandHeader.hpp"\r
+#include "Session.hpp"\r
+#include "Utils.hpp"\r
+#include "../Logger.hpp"\r
+#include <boost/make_shared.hpp>\r
+#include <string>\r
+\r
+namespace network {\r
+\r
+ Session::Session(boost::asio::io_service& io_service_tcp) :\r
+ io_service_tcp_(io_service_tcp),\r
+ socket_tcp_(io_service_tcp),\r
+ encryption_(false),\r
+ online_(true),\r
+ login_(false),\r
+ read_start_time_(time(nullptr)),\r
+ write_start_time_(time(nullptr)),\r
+ read_byte_sum_(0),\r
+ write_byte_sum_(0),\r
+ serialized_byte_sum_(0),\r
+ compressed_byte_sum_(0),\r
+ write_average_limit_(999999),\r
+ id_(0)\r
+ {\r
+\r
+ }\r
+\r
+ Session::~Session()\r
+ {\r
+ Close();\r
+ }\r
+\r
+ void Session::Close()\r
+ {\r
+ socket_tcp_.close();\r
+ }\r
+\r
+ void Session::Send(const Command& command)\r
+ {\r
+ auto msg = Serialize(command);\r
+ write_byte_sum_ += msg.size();\r
+ UpdateWriteByteAverage();\r
+\r
+ Logger::Debug(_T("%d byte/s"), GetWriteByteAverage());\r
+\r
+ io_service_tcp_.post(boost::bind(&Session::DoWriteTCP, this, msg, shared_from_this()));\r
+ }\r
+\r
+ void Session::SyncSend(const Command& command)\r
+ {\r
+ auto msg = Serialize(command);\r
+ write_byte_sum_ += msg.size();\r
+ UpdateWriteByteAverage();\r
+\r
+ try {\r
+ boost::asio::write(\r
+ socket_tcp_, boost::asio::buffer(msg.data(), msg.size()),\r
+ boost::asio::transfer_all());\r
+ } catch (std::exception& e) {\r
+ std::cout << e.what() << std::endl;\r
+ }\r
+ }\r
+\r
+ double Session::GetReadByteAverage() const\r
+ {\r
+ return 1.0f * read_byte_sum_ / (time(nullptr) - read_start_time_);\r
+ }\r
+\r
+ double Session::GetWriteByteAverage() const\r
+ {\r
+ return 1.0f * write_byte_sum_ / (time(nullptr) - write_start_time_);\r
+ }\r
+\r
+ void Session::UpdateReadByteAverage()\r
+ {\r
+ unsigned long elapsed_time = time(nullptr) - read_start_time_;\r
+ if (elapsed_time >= BYTE_AVERAGE_REFRESH_SECONDS) {\r
+ read_byte_sum_ /= 2;\r
+ read_start_time_ = time(nullptr) - elapsed_time / 2;\r
+ }\r
+ }\r
+\r
+ void Session::UpdateWriteByteAverage()\r
+ {\r
+ unsigned long elapsed_time = time(nullptr) - write_start_time_;\r
+ if (elapsed_time >= BYTE_AVERAGE_REFRESH_SECONDS) {\r
+ write_byte_sum_ /= 2;\r
+ write_start_time_ = time(nullptr) - elapsed_time / 2;\r
+ }\r
+ }\r
+\r
+ void Session::EnableEncryption()\r
+ {\r
+ encryption_ = true;\r
+ }\r
+\r
+ Encrypter& Session::encrypter()\r
+ {\r
+ return encrypter_;\r
+ }\r
+\r
+ tcp::socket& Session::tcp_socket()\r
+ {\r
+ return socket_tcp_;\r
+ }\r
+\r
+ UserID Session::id() const\r
+ {\r
+ return id_;\r
+ }\r
+\r
+ void Session::set_id(UserID id)\r
+ {\r
+ id_ = id;\r
+ }\r
+\r
+ bool Session::online() const\r
+ {\r
+ return online_;\r
+ }\r
+\r
+ std::string Session::global_ip() const\r
+ {\r
+ return global_ip_;\r
+ }\r
+\r
+ uint16_t Session::udp_port() const{\r
+ return udp_port_;\r
+ }\r
+\r
+ void Session::set_global_ip(const std::string& global_ip)\r
+ {\r
+ global_ip_ = global_ip;\r
+ }\r
+\r
+ void Session::set_udp_port(uint16_t udp_port)\r
+ {\r
+ udp_port_ = udp_port;\r
+ }\r
+\r
+ int Session::serialized_byte_sum() const\r
+ {\r
+ return serialized_byte_sum_;\r
+ }\r
+\r
+ int Session::compressed_byte_sum() const\r
+ {\r
+ return compressed_byte_sum_;\r
+ }\r
+\r
+ bool Session::operator==(const Session& s)\r
+ {\r
+ return id_ == s.id_;\r
+ }\r
+\r
+ bool Session::operator!=(const Session& s)\r
+ {\r
+ return !operator==(s);\r
+ }\r
+\r
+ int Session::write_average_limit() const\r
+ {\r
+ return write_average_limit_;\r
+ }\r
+\r
+ void Session::set_write_average_limit(int limit)\r
+ {\r
+ write_average_limit_ = limit;\r
+ }\r
+\r
+ std::string Session::Serialize(const Command& command)\r
+ {\r
+ assert(command.header() < 0xFF);\r
auto header = static_cast<uint8_t>(command.header());\r
- std::string body = command.body();
-
- std::string msg = Utils::Serialize(header) + body;
-
- // 圧縮
- if (body.size() >= COMPRESS_MIN_LENGTH) {
- auto compressed = Utils::LZ4Compress(msg);
+ std::string body = command.body();\r
+\r
+ std::string msg = Utils::Serialize(header) + body;\r
+\r
+ // 圧縮\r
+ if (body.size() >= COMPRESS_MIN_LENGTH) {\r
+ auto compressed = Utils::LZ4Compress(msg);\r
if (msg.size() > compressed.size() + sizeof(uint8_t)) {\r
- assert(msg.size() < 65535);
+ assert(msg.size() < 65535);\r
msg = Utils::Serialize(static_cast<uint8_t>(header::LZ4_COMPRESS_HEADER),\r
static_cast<uint16_t>(msg.size()))\r
- + compressed;
- }
- }
-
- // 暗号化
- if (encryption_) {
+ + compressed;\r
+ }\r
+ }\r
+\r
+ // 暗号化\r
+ if (encryption_) {\r
msg = Utils::Serialize(static_cast<uint8_t>(header::ENCRYPT_HEADER))\r
- + encrypter_.Encrypt(msg);
- }
-
- return Utils::Encode(msg);
- }
-
- Command Session::Deserialize(const std::string& msg)
- {
- std::string decoded_msg = Utils::Decode(msg);
-
+ + encrypter_.Encrypt(msg);\r
+ }\r
+\r
+ return Utils::Encode(msg);\r
+ }\r
+\r
+ Command Session::Deserialize(const std::string& msg)\r
+ {\r
+ std::string decoded_msg = Utils::Decode(msg);\r
+\r
uint8_t header;\r
- Utils::Deserialize(decoded_msg, &header);
-
- // 復号
- if (header == header::ENCRYPT_HEADER) {
- decoded_msg.erase(0, sizeof(header));
- decoded_msg = encrypter_.Decrypt(decoded_msg);
- Utils::Deserialize(decoded_msg, &header);
- }
-
- // 伸長
- if (header == header::LZ4_COMPRESS_HEADER) {
+ Utils::Deserialize(decoded_msg, &header);\r
+\r
+ // 復号\r
+ if (header == header::ENCRYPT_HEADER) {\r
+ decoded_msg.erase(0, sizeof(header));\r
+ decoded_msg = encrypter_.Decrypt(decoded_msg);\r
+ Utils::Deserialize(decoded_msg, &header);\r
+ }\r
+\r
+ // 伸長\r
+ if (header == header::LZ4_COMPRESS_HEADER) {\r
uint16_t original_size;\r
- Utils::Deserialize(decoded_msg, &header, &original_size);
- decoded_msg.erase(0, sizeof(header) + sizeof(original_size));
- decoded_msg = Utils::LZ4Uncompress(decoded_msg, original_size);
- Utils::Deserialize(decoded_msg, &header);
- }
-
- std::string body = decoded_msg.substr(sizeof(header));
-
- return Command(static_cast<header::CommandHeader>(header), body, shared_from_this());
- }
-
- void Session::ReceiveTCP(const boost::system::error_code& error)
- {
- if (!error) {
- std::string buffer(boost::asio::buffer_cast<const char*>(receive_buf_.data()),receive_buf_.size());
- auto length = buffer.find_last_of(NETWORK_UTILS_DELIMITOR);
-
- if (length != std::string::npos) {
-
- receive_buf_.consume(length+1);
- buffer.erase(length+1);
-
- while (!buffer.empty()) {
- std::string msg;
-
- while (!buffer.empty() && buffer[0]!=NETWORK_UTILS_DELIMITOR)
- {
- msg += buffer[0];
- buffer.erase(0,1);
- }
- buffer.erase(0,1);
-
- read_byte_sum_ += msg.size();
- UpdateReadByteAverage();
-
- FetchTCP(msg);
- }
-
- boost::asio::async_read_until(socket_tcp_,
- receive_buf_, NETWORK_UTILS_DELIMITOR,
- boost::bind(
- &Session::ReceiveTCP, shared_from_this(),
- boost::asio::placeholders::error));
-
- }
-
- } else {
- FatalError();
- }
- }
-
- void Session::DoWriteTCP(const std::string msg, SessionPtr session_holder)
- {
- bool write_in_progress = !send_queue_.empty();
- send_queue_.push(msg);
- if (!write_in_progress && !send_queue_.empty())
- {
-
- boost::shared_ptr<std::string> s =
- boost::make_shared<std::string>(msg.data(), msg.size());
-
- boost::asio::async_write(socket_tcp_,
- boost::asio::buffer(s->data(), s->size()),
- boost::bind(&Session::WriteTCP, this,
- boost::asio::placeholders::error, s, session_holder));
- }
- }
-
- void Session::WriteTCP(const boost::system::error_code& error,
- boost::shared_ptr<std::string> holder, SessionPtr session_holder)
- {
- if (!error) {
- if (!send_queue_.empty()) {
- send_queue_.pop();
- if (!send_queue_.empty())
- {
-
- boost::shared_ptr<std::string> s =
- boost::make_shared<std::string>(send_queue_.front().data(), send_queue_.front().size());
-
- boost::asio::async_write(socket_tcp_,
- boost::asio::buffer(s->data(), s->size()),
- boost::bind(&Session::WriteTCP, this,
- boost::asio::placeholders::error, s, session_holder));
- }
- }
- } else {
- FatalError(session_holder);
- }
- }
-
- void Session::FetchTCP(const std::string& msg)
- {
+ Utils::Deserialize(decoded_msg, &header, &original_size);\r
+ decoded_msg.erase(0, sizeof(header) + sizeof(original_size));\r
+ decoded_msg = Utils::LZ4Uncompress(decoded_msg, original_size);\r
+ Utils::Deserialize(decoded_msg, &header);\r
+ }\r
+\r
+ std::string body = decoded_msg.substr(sizeof(header));\r
+\r
+ return Command(static_cast<header::CommandHeader>(header), body, shared_from_this());\r
+ }\r
+\r
+ void Session::ReceiveTCP(const boost::system::error_code& error)\r
+ {\r
+ if (!error) {\r
+ std::string buffer(boost::asio::buffer_cast<const char*>(receive_buf_.data()),receive_buf_.size());\r
+ auto length = buffer.find_last_of(NETWORK_UTILS_DELIMITOR);\r
+\r
+ if (length != std::string::npos) {\r
+\r
+ receive_buf_.consume(length+1);\r
+ buffer.erase(length+1);\r
+\r
+ while (!buffer.empty()) {\r
+ std::string msg;\r
+\r
+ while (!buffer.empty() && buffer[0]!=NETWORK_UTILS_DELIMITOR)\r
+ {\r
+ msg += buffer[0];\r
+ buffer.erase(0,1);\r
+ }\r
+ buffer.erase(0,1);\r
+\r
+ read_byte_sum_ += msg.size();\r
+ UpdateReadByteAverage();\r
+\r
+ FetchTCP(msg);\r
+ }\r
+\r
+ boost::asio::async_read_until(socket_tcp_,\r
+ receive_buf_, NETWORK_UTILS_DELIMITOR,\r
+ boost::bind(\r
+ &Session::ReceiveTCP, shared_from_this(),\r
+ boost::asio::placeholders::error));\r
+\r
+ }\r
+\r
+ } else {\r
+ FatalError();\r
+ }\r
+ }\r
+\r
+ void Session::DoWriteTCP(const std::string msg, SessionPtr session_holder)\r
+ {\r
+ bool write_in_progress = !send_queue_.empty();\r
+ send_queue_.push(msg);\r
+ if (!write_in_progress && !send_queue_.empty())\r
+ {\r
+ \r
+ boost::shared_ptr<std::string> s = \r
+ boost::make_shared<std::string>(msg.data(), msg.size());\r
+\r
+ boost::asio::async_write(socket_tcp_,\r
+ boost::asio::buffer(s->data(), s->size()),\r
+ boost::bind(&Session::WriteTCP, this,\r
+ boost::asio::placeholders::error, s, session_holder));\r
+ }\r
+ }\r
+\r
+ void Session::WriteTCP(const boost::system::error_code& error,\r
+ boost::shared_ptr<std::string> holder, SessionPtr session_holder)\r
+ {\r
+ if (!error) {\r
+ if (!send_queue_.empty()) {\r
+ send_queue_.pop();\r
+ if (!send_queue_.empty())\r
+ {\r
+\r
+ boost::shared_ptr<std::string> s = \r
+ boost::make_shared<std::string>(send_queue_.front().data(), send_queue_.front().size());\r
+\r
+ boost::asio::async_write(socket_tcp_,\r
+ boost::asio::buffer(s->data(), s->size()),\r
+ boost::bind(&Session::WriteTCP, this,\r
+ boost::asio::placeholders::error, s, session_holder));\r
+ }\r
+ }\r
+ } else {\r
+ FatalError(session_holder);\r
+ }\r
+ }\r
+\r
+ void Session::FetchTCP(const std::string& msg)\r
+ {\r
if (msg.size() >= sizeof(uint8_t)) {\r
- if (on_receive_) {
- (*on_receive_)(Deserialize(msg));
- }
- } else {
- Logger::Error(_T("Too short data"));
- }
- }
-
- void Session::FatalError(SessionPtr session_holder)
- {
- if (online_) {
- online_ = false;
- if (on_receive_) {
- if (id_ > 0) {
- (*on_receive_)(FatalConnectionError(id_));
- } else {
- (*on_receive_)(FatalConnectionError());
- }
- }
- }
- }
-
- void Session::set_on_receive(CallbackFuncPtr func)
- {
- on_receive_ = func;
- }
-
-
-}
+ if (on_receive_) {\r
+ (*on_receive_)(Deserialize(msg));\r
+ }\r
+ } else {\r
+ Logger::Error(_T("Too short data"));\r
+ }\r
+ }\r
+\r
+ void Session::FatalError(SessionPtr session_holder)\r
+ {\r
+ if (online_) {\r
+ online_ = false;\r
+ if (on_receive_) {\r
+ if (id_ > 0) {\r
+ (*on_receive_)(FatalConnectionError(id_));\r
+ } else {\r
+ (*on_receive_)(FatalConnectionError());\r
+ }\r
+ }\r
+ }\r
+ }\r
+\r
+ void Session::set_on_receive(CallbackFuncPtr func)\r
+ {\r
+ on_receive_ = func;\r
+ }\r
+\r
+\r
+}\r
OSDN Git Service
