-/*\r
-* Copyright (c) 2011 Sveriges Television AB <info@casparcg.com>\r
-*\r
-* This file is part of CasparCG (www.casparcg.com).\r
-*\r
-* CasparCG is free software: you can redistribute it and/or modify\r
-* it under the terms of the GNU General Public License as published by\r
-* the Free Software Foundation, either version 3 of the License, or\r
-* (at your option) any later version.\r
-*\r
-* CasparCG is distributed in the hope that it will be useful,\r
-* but WITHOUT ANY WARRANTY; without even the implied warranty of\r
-* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the\r
-* GNU General Public License for more details.\r
-*\r
-* You should have received a copy of the GNU General Public License\r
-* along with CasparCG. If not, see <http://www.gnu.org/licenses/>.\r
-*\r
-* Author: Robert Nagy, ronag89@gmail.com\r
-*/\r
-\r
-#include "..\stdafx.h"\r
-\r
-#include "AsyncEventServer.h"\r
-\r
-#include "ProtocolStrategy.h"\r
-\r
-#include <algorithm>\r
-#include <array>\r
-#include <string>\r
-#include <set>\r
-#include <vector>\r
-\r
-#include <boost/asio.hpp>\r
-#include <boost/thread.hpp>\r
-#include <boost/lexical_cast.hpp>\r
-#include <boost/locale.hpp>\r
-#include <boost/algorithm/string/split.hpp>\r
-\r
-using boost::asio::ip::tcp;\r
-\r
-namespace caspar { namespace IO {\r
- \r
-class connection;\r
-\r
-typedef std::set<spl::shared_ptr<connection>> connection_set;\r
-\r
-class connection : public spl::enable_shared_from_this<connection>, public ClientInfo\r
-{ \r
- const spl::shared_ptr<tcp::socket> socket_; \r
- const spl::shared_ptr<connection_set> connection_set_;\r
- const std::wstring name_;\r
- const spl::shared_ptr<IProtocolStrategy> protocol_;\r
-\r
- std::array<char, 32768> data_;\r
- std::string input_;\r
-\r
-public:\r
- static spl::shared_ptr<connection> create(spl::shared_ptr<tcp::socket> socket, const ProtocolStrategyPtr& protocol, spl::shared_ptr<connection_set> connection_set)\r
- {\r
- spl::shared_ptr<connection> con(new connection(std::move(socket), std::move(protocol), std::move(connection_set)));\r
- con->read_some();\r
- return con;\r
- }\r
-\r
- std::wstring print() const\r
- {\r
- return L"[" + name_ + L"]";\r
- }\r
- \r
- /* ClientInfo */\r
- \r
- virtual void Send(const std::wstring& data)\r
- {\r
- write_some(data);\r
- }\r
-\r
- virtual void Disconnect()\r
- {\r
- stop();\r
- }\r
- \r
- /**************/\r
- \r
- void stop()\r
- {\r
- connection_set_->erase(shared_from_this());\r
- try\r
- {\r
- socket_->close();\r
- }\r
- catch(...)\r
- {\r
- CASPAR_LOG_CURRENT_EXCEPTION();\r
- }\r
- CASPAR_LOG(info) << print() << L" Disconnected.";\r
- }\r
-\r
-private:\r
- connection(const spl::shared_ptr<tcp::socket>& socket, const ProtocolStrategyPtr& protocol, const spl::shared_ptr<connection_set>& connection_set) \r
- : socket_(socket)\r
- , name_((socket_->is_open() ? u16(socket_->local_endpoint().address().to_string() + ":" + boost::lexical_cast<std::string>(socket_->local_endpoint().port())) : L"no-address"))\r
- , connection_set_(connection_set)\r
- , protocol_(protocol)\r
- {\r
- CASPAR_LOG(info) << print() << L" Connected.";\r
- }\r
- \r
- void handle_read(const boost::system::error_code& error, size_t bytes_transferred) \r
- { \r
- if(!error)\r
- {\r
- try\r
- {\r
- CASPAR_LOG(trace) << print() << L" Received: " << u16(std::string(data_.begin(), data_.begin() + bytes_transferred));\r
-\r
- input_.append(data_.begin(), data_.begin() + bytes_transferred);\r
- \r
- std::vector<std::string> split;\r
- boost::iter_split(split, input_, boost::algorithm::first_finder("\r\n"));\r
- \r
- input_ = std::move(split.back());\r
- split.pop_back();\r
-\r
- BOOST_FOREACH(auto cmd, split)\r
- {\r
- auto u16cmd = boost::locale::conv::to_utf<wchar_t>(cmd, protocol_->GetCodepage()) + L"\r\n";\r
- protocol_->Parse(u16cmd.data(), static_cast<int>(u16cmd.size()), shared_from_this());\r
- }\r
- }\r
- catch(...)\r
- {\r
- CASPAR_LOG_CURRENT_EXCEPTION();\r
- }\r
- \r
- read_some();\r
- } \r
- else if (error != boost::asio::error::operation_aborted)\r
- stop(); \r
- }\r
-\r
- void handle_write(const spl::shared_ptr<std::string>& data, const boost::system::error_code& error, size_t bytes_transferred)\r
- {\r
- if(!error) \r
- CASPAR_LOG(trace) << print() << L" Sent: " << (data->size() < 512 ? u16(*data) : L"more than 512 bytes."); \r
- else if (error != boost::asio::error::operation_aborted) \r
- stop(); \r
- }\r
-\r
- void read_some()\r
- {\r
- socket_->async_read_some(boost::asio::buffer(data_.data(), data_.size()), std::bind(&connection::handle_read, shared_from_this(), std::placeholders::_1, std::placeholders::_2));\r
- }\r
- \r
- void write_some(const std::wstring& data)\r
- {\r
- auto str = spl::make_shared<std::string>(boost::locale::conv::from_utf<wchar_t>(data, protocol_->GetCodepage()));\r
- socket_->async_write_some(boost::asio::buffer(str->data(), str->size()), std::bind(&connection::handle_write, shared_from_this(), str, std::placeholders::_1, std::placeholders::_2));\r
- }\r
-};\r
-\r
-struct AsyncEventServer::implementation\r
-{\r
- boost::asio::io_service service_;\r
- tcp::acceptor acceptor_;\r
- spl::shared_ptr<IProtocolStrategy> protocol_;\r
- spl::shared_ptr<connection_set> connection_set_;\r
- boost::thread thread_;\r
-\r
- implementation(const spl::shared_ptr<IProtocolStrategy>& protocol, unsigned short port)\r
- : acceptor_(service_, tcp::endpoint(tcp::v4(), port))\r
- , protocol_(protocol)\r
- , thread_(std::bind(&boost::asio::io_service::run, &service_))\r
- {\r
- start_accept();\r
- }\r
-\r
- ~implementation()\r
- {\r
- try\r
- {\r
- acceptor_.close(); \r
- }\r
- catch(...)\r
- {\r
- CASPAR_LOG_CURRENT_EXCEPTION();\r
- }\r
-\r
- service_.post([=]\r
- {\r
- auto connections = *connection_set_;\r
- BOOST_FOREACH(auto& connection, connections)\r
- connection->stop(); \r
- });\r
-\r
- thread_.join();\r
- }\r
- \r
- void start_accept() \r
- {\r
- spl::shared_ptr<tcp::socket> socket(new tcp::socket(service_));\r
- acceptor_.async_accept(*socket, std::bind(&implementation::handle_accept, this, socket, std::placeholders::_1));\r
- }\r
-\r
- void handle_accept(const spl::shared_ptr<tcp::socket>& socket, const boost::system::error_code& error) \r
- {\r
- if (!acceptor_.is_open())\r
- return;\r
- \r
- if (!error) \r
- connection_set_->insert(connection::create(socket, protocol_, connection_set_));\r
-\r
- start_accept();\r
- }\r
-};\r
-\r
-AsyncEventServer::AsyncEventServer(const spl::shared_ptr<IProtocolStrategy>& protocol, unsigned short port) : impl_(new implementation(protocol, port)){}\r
-AsyncEventServer::~AsyncEventServer(){}\r
+/*
+* Copyright (c) 2011 Sveriges Television AB <info@casparcg.com>
+*
+* This file is part of CasparCG (www.casparcg.com).
+*
+* CasparCG 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 3 of the License, or
+* (at your option) any later version.
+*
+* CasparCG 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 CasparCG. If not, see <http://www.gnu.org/licenses/>.
+*
+* Author: Robert Nagy, ronag89@gmail.com
+*/
+
+#include "..\stdafx.h"
+
+#include "AsyncEventServer.h"
+
+#include "ProtocolStrategy.h"
+
+#include <algorithm>
+#include <array>
+#include <string>
+#include <set>
+#include <vector>
+
+#include <boost/asio.hpp>
+#include <boost/thread.hpp>
+#include <boost/lexical_cast.hpp>
+#include <boost/locale.hpp>
+#include <boost/algorithm/string/split.hpp>
+
+using boost::asio::ip::tcp;
+
+namespace caspar { namespace IO {
+
+class connection;
+
+typedef std::set<spl::shared_ptr<connection>> connection_set;
+
+class connection : public spl::enable_shared_from_this<connection>, public ClientInfo
+{
+ const spl::shared_ptr<tcp::socket> socket_;
+ const spl::shared_ptr<connection_set> connection_set_;
+ const std::wstring name_;
+ const spl::shared_ptr<IProtocolStrategy> protocol_;
+
+ std::array<char, 32768> data_;
+ std::string input_;
+
+public:
+ static spl::shared_ptr<connection> create(spl::shared_ptr<tcp::socket> socket, const ProtocolStrategyPtr& protocol, spl::shared_ptr<connection_set> connection_set)
+ {
+ spl::shared_ptr<connection> con(new connection(std::move(socket), std::move(protocol), std::move(connection_set)));
+ con->read_some();
+ return con;
+ }
+
+ std::wstring print() const
+ {
+ return L"[" + name_ + L"]";
+ }
+
+ /* ClientInfo */
+
+ virtual void Send(const std::wstring& data)
+ {
+ write_some(data);
+ }
+
+ virtual void Disconnect()
+ {
+ stop();
+ }
+
+ /**************/
+
+ void stop()
+ {
+ connection_set_->erase(shared_from_this());
+ try
+ {
+ socket_->close();
+ }
+ catch(...)
+ {
+ CASPAR_LOG_CURRENT_EXCEPTION();
+ }
+ CASPAR_LOG(info) << print() << L" Disconnected.";
+ }
+
+private:
+ connection(const spl::shared_ptr<tcp::socket>& socket, const ProtocolStrategyPtr& protocol, const spl::shared_ptr<connection_set>& connection_set)
+ : socket_(socket)
+ , name_((socket_->is_open() ? u16(socket_->local_endpoint().address().to_string() + ":" + boost::lexical_cast<std::string>(socket_->local_endpoint().port())) : L"no-address"))
+ , connection_set_(connection_set)
+ , protocol_(protocol)
+ {
+ CASPAR_LOG(info) << print() << L" Connected.";
+ }
+
+ void handle_read(const boost::system::error_code& error, size_t bytes_transferred)
+ {
+ if(!error)
+ {
+ try
+ {
+ CASPAR_LOG(trace) << print() << L" Received: " << u16(std::string(data_.begin(), data_.begin() + bytes_transferred));
+
+ input_.append(data_.begin(), data_.begin() + bytes_transferred);
+
+ std::vector<std::string> split;
+ boost::iter_split(split, input_, boost::algorithm::first_finder("\r\n"));
+
+ input_ = std::move(split.back());
+ split.pop_back();
+
+ BOOST_FOREACH(auto cmd, split)
+ {
+ auto u16cmd = boost::locale::conv::to_utf<wchar_t>(cmd, protocol_->GetCodepage()) + L"\r\n";
+ protocol_->Parse(u16cmd.data(), static_cast<int>(u16cmd.size()), shared_from_this());
+ }
+ }
+ catch(...)
+ {
+ CASPAR_LOG_CURRENT_EXCEPTION();
+ }
+
+ read_some();
+ }
+ else if (error != boost::asio::error::operation_aborted)
+ stop();
+ }
+
+ void handle_write(const spl::shared_ptr<std::string>& data, const boost::system::error_code& error, size_t bytes_transferred)
+ {
+ if(!error)
+ CASPAR_LOG(trace) << print() << L" Sent: " << (data->size() < 512 ? u16(*data) : L"more than 512 bytes.");
+ else if (error != boost::asio::error::operation_aborted)
+ stop();
+ }
+
+ void read_some()
+ {
+ socket_->async_read_some(boost::asio::buffer(data_.data(), data_.size()), std::bind(&connection::handle_read, shared_from_this(), std::placeholders::_1, std::placeholders::_2));
+ }
+
+ void write_some(const std::wstring& data)
+ {
+ auto str = spl::make_shared<std::string>(boost::locale::conv::from_utf<wchar_t>(data, protocol_->GetCodepage()));
+ socket_->async_write_some(boost::asio::buffer(str->data(), str->size()), std::bind(&connection::handle_write, shared_from_this(), str, std::placeholders::_1, std::placeholders::_2));
+ }
+};
+
+struct AsyncEventServer::implementation
+{
+ boost::asio::io_service service_;
+ tcp::acceptor acceptor_;
+ spl::shared_ptr<IProtocolStrategy> protocol_;
+ spl::shared_ptr<connection_set> connection_set_;
+ boost::thread thread_;
+
+ implementation(const spl::shared_ptr<IProtocolStrategy>& protocol, unsigned short port)
+ : acceptor_(service_, tcp::endpoint(tcp::v4(), port))
+ , protocol_(protocol)
+ , thread_(std::bind(&boost::asio::io_service::run, &service_))
+ {
+ start_accept();
+ }
+
+ ~implementation()
+ {
+ try
+ {
+ acceptor_.close();
+ }
+ catch(...)
+ {
+ CASPAR_LOG_CURRENT_EXCEPTION();
+ }
+
+ service_.post([=]
+ {
+ auto connections = *connection_set_;
+ BOOST_FOREACH(auto& connection, connections)
+ connection->stop();
+ });
+
+ thread_.join();
+ }
+
+ void start_accept()
+ {
+ spl::shared_ptr<tcp::socket> socket(new tcp::socket(service_));
+ acceptor_.async_accept(*socket, std::bind(&implementation::handle_accept, this, socket, std::placeholders::_1));
+ }
+
+ void handle_accept(const spl::shared_ptr<tcp::socket>& socket, const boost::system::error_code& error)
+ {
+ if (!acceptor_.is_open())
+ return;
+
+ if (!error)
+ connection_set_->insert(connection::create(socket, protocol_, connection_set_));
+
+ start_accept();
+ }
+};
+
+AsyncEventServer::AsyncEventServer(const spl::shared_ptr<IProtocolStrategy>& protocol, unsigned short port) : impl_(new implementation(protocol, port)){}
+AsyncEventServer::~AsyncEventServer(){}
}}
\ No newline at end of file