network stuff

include/Utils/Network/Network.h

@@ -0,0 +1,137 @@
+#ifndef NETWORK_H_INCLUDED
+#define NETWORK_H_INCLUDED
+
+#include <string>
+#include <map>
+#include <vector>
+#include "boost/shared_array.hpp"
+#include "boost/property_tree/ptree.hpp"
+#include "boost/foreach.hpp"
+
+#include "boost/asio.hpp"
+#include "boost/date_time/posix_time/posix_time.hpp"
+#include "boost/signal.hpp"
+#include "boost/variant.hpp"
+
+#include "include/Utils/Network/SignalSender.h"
+
+
+namespace SE
+{
+
+const int CONST_CONNECTION_TIMEOUT_SECONDS = 300;
+
+
+void SendPropertyTree(boost::asio::io_service& ioService, boost::asio::ip::tcp::socket& socket, boost::property_tree::ptree pTree);
+
+
+//Must be stored in shared_ptr only
+struct TDataReader : public boost::enable_shared_from_this<TDataReader>
+{
+	boost::asio::ip::tcp::socket& Socket;
+
+	int DataSize;
+
+	std::vector<char> Data;
+
+	TDataReader(boost::asio::ip::tcp::socket& socket);
+
+	void StartRead();
+
+	void HandleReadDataSize(const boost::system::error_code& error);
+
+	void HandleReadData(const boost::system::error_code& error);
+
+	boost::signal<void(boost::property_tree::ptree)> DataReadSignal;
+	boost::signal<void()> ErrorSignal;
+};
+
+
+
+class TAuthorizationInterface
+{
+public:
+	virtual void Authorize() = 0;
+};
+
+class TSimpleAuthorization : public TAuthorizationInterface
+{
+public:
+
+	boost::asio::io_service& IoService;
+
+	boost::asio::ip::tcp::socket& Socket;
+
+	//std::shared_ptr<TDataReader> DataReader;
+
+	std::string Login;
+	std::string Password;
+
+
+	TSimpleAuthorization(boost::asio::io_service& ioService, boost::asio::ip::tcp::socket& socket);
+
+	virtual void Authorize();
+
+	void HandleGetData(boost::property_tree::ptree pTree);
+
+	boost::signal<void()> AuthorizedSignal;
+	boost::signal<void(std::string, std::string)> SaveLoginPasswordSignal;
+	boost::signal<void()> ErrorSignal;
+};
+
+typedef boost::variant<std::shared_ptr<TSimpleAuthorization>> TAuthorizationVariant;
+
+class TClientSocket : public boost::enable_shared_from_this<TClientSocket>
+{
+protected:
+
+	boost::asio::io_service IoService; //IoService must be declared before Socket
+    
+    boost::thread IoServiceThread;
+
+	boost::asio::ip::tcp::socket Socket;
+	
+	int ReadDataLenLong;
+
+	std::vector<char> ReadData;
+
+	boost::shared_ptr<boost::asio::deadline_timer> ConnectionTimeoutTimer;
+
+	bool Opened;
+
+public:
+	boost::signal<void()> OnAddressNotResolvedSignal;
+	boost::signal<void()> OnConnectedSignal;
+	boost::signal<void()> OnAutorizedSignal;
+	boost::signal<void()> OnDisconnectedSignal;
+
+	TAuthorizationVariant Authorization;
+
+	TClientSocket();
+	~TClientSocket();
+
+	void IoServiceRun();
+
+	void Open(const std::string address, const std::string& port);
+
+	void Close();
+
+	void HandleConnectTimeout(const boost::system::error_code& error);
+	void HandleConnect(const boost::system::error_code& error);
+
+	void HandleAuthorized();
+	void HandleAuthorizationError();
+};
+
+class TServerSocket
+{
+protected:
+public:
+};
+
+
+
+} //namespace SE
+
+
+#endif

include/Utils/Network/SignalSender.h

@@ -0,0 +1,54 @@
+#ifndef SIGNAL_SENDER_H_INCLUDED
+#define SIGNAL_SENDER_H_INCLUDED
+
+#include "boost/asio.hpp"
+#include "boost/bind.hpp"
+#include "boost/shared_ptr.hpp"
+#include "boost/array.hpp"
+#include "boost/enable_shared_from_this.hpp"
+#include "boost/signal.hpp"
+#include "boost/thread.hpp"
+#include "boost/date_time/posix_time/posix_time.hpp"
+#include "boost/property_tree/ptree.hpp"
+#include "boost/property_tree/xml_parser.hpp"
+
+namespace SE
+{
+	//Xperimental -- need to optimize this
+
+struct TSignalSender
+	: public boost::enable_shared_from_this<TSignalSender>
+{
+	boost::asio::ip::tcp::socket& Socket;
+
+	int Length;
+
+	std::string Data;
+
+	boost::signal<void()> ErrorSignal;
+
+	TSignalSender(boost::asio::ip::tcp::socket& socket, int length, std::string data)
+		: Socket(socket)
+		, Length(length)
+		, Data(data)
+	{
+	}
+
+	void Send()
+	{
+		boost::asio::async_write(Socket, boost::asio::buffer(&Length, 4), boost::bind(&TSignalSender::Handle, shared_from_this(), boost::asio::placeholders::error));
+		boost::asio::async_write(Socket, boost::asio::buffer(Data.c_str(), Data.size()), boost::bind(&TSignalSender::Handle, shared_from_this(), boost::asio::placeholders::error));
+	}
+
+	void Handle(const boost::system::error_code& error)
+	{
+		if (error)
+		{
+			ErrorSignal();
+		}
+	}
+};
+
+} //namespace SE
+
+#endif