73 lines
2.9 KiB
C++
73 lines
2.9 KiB
C++
#pragma once
|
|
|
|
#include <cstdint>
|
|
#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 = nullptr);
|
|
~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;
|
|
};
|