Initial commit

Socket.h

@@ -0,0 +1,75 @@
+#ifndef INC_SEXT_SOCKET_H
+#define INC_SEXT_SOCKET_H
+
+#include <stdint.h>
+#include <string>
+#include <queue>
+#include <boost/asio.hpp>
+#include <boost/thread.hpp>
+#include <boost/thread/shared_mutex.hpp>
+
+#include "sdk/smsdk_ext.h"
+#include "Define.h"
+
+class SocketHandler;
+
+template <class SocketType>
+class Socket {
+public:
+	Socket(SM_SocketType st, typename SocketType::socket* asioSocket = NULL);
+	~Socket();
+
+	bool IsOpen();
+
+	bool Bind(const char* hostname, uint16_t port, bool async = true);
+	bool Connect(const char* hostname, uint16_t port, bool async = true);
+	bool Disconnect();
+	bool Listen();
+	bool Send(const std::string& data, bool async = true);
+	bool SendTo(const std::string& data, const char* hostname, uint16_t port, bool async = true);
+	bool SetOption(SM_SocketOption so, int value, bool lock=true);
+
+	IPluginFunction* connectCallback;
+	IPluginFunction* incomingCallback;
+	IPluginFunction* receiveCallback;
+	IPluginFunction* sendqueueEmptyCallback;
+	IPluginFunction* disconnectCallback;
+	IPluginFunction* errorCallback;
+
+	int32_t smHandle;
+	int32_t smCallbackArg;
+	volatile unsigned int sendQueueLength;
+
+private:
+	void ReceiveHandler(char* buf, size_t bufferSize, size_t bytes, const boost::system::error_code&, boost::shared_lock<boost::shared_mutex>*);
+
+	void BindPostResolveHandler(typename SocketType::resolver*, typename SocketType::resolver::iterator, const boost::system::error_code&, boost::shared_lock<boost::shared_mutex>*);
+
+	void ConnectPostResolveHandler(typename SocketType::resolver*, typename SocketType::resolver::iterator, const boost::system::error_code&, boost::shared_lock<boost::shared_mutex>*);
+	void ConnectPostConnectHandler(typename SocketType::resolver*, typename SocketType::resolver::iterator, const boost::system::error_code&, boost::shared_lock<boost::shared_mutex>*);
+
+	void ListenIncomingHandler(boost::asio::ip::tcp::socket* newAsioSocket, const boost::system::error_code&, boost::shared_lock<boost::shared_mutex>*);
+
+	void SendPostSendHandler(char* buf, size_t bytes, const boost::system::error_code& err, boost::shared_lock<boost::shared_mutex>*);
+
+	void SendToPostResolveHandler(typename SocketType::resolver*, typename SocketType::resolver::iterator, char* buf, size_t bufLen, const boost::system::error_code&, boost::shared_lock<boost::shared_mutex>*);
+	void SendToPostSendHandler(typename SocketType::resolver*, typename SocketType::resolver::iterator, char* buf, size_t bufLen, size_t bytesTransferred, const boost::system::error_code&, boost::shared_lock<boost::shared_mutex>*);
+
+	//void InitializeResolver();
+	void InitializeSocket();
+
+	SM_SocketType sm_sockettype;
+	std::queue<SocketOption*> socketOptionQueue;
+
+	typename SocketType::socket* socket;
+	boost::mutex socketMutex;
+	//typename SocketType::resolver* resolver;
+	typename SocketType::endpoint* localEndpoint;
+	boost::mutex* localEndpointMutex;
+	boost::asio::ip::tcp::acceptor* tcpAcceptor;
+	boost::mutex* tcpAcceptorMutex;
+
+	boost::shared_mutex handlerMutex;
+};
+
+#endif