1 files changed, 57 insertions, 0 deletions
diff --git a/xmrstak/net/socket.hpp b/xmrstak/net/socket.hpp
new file mode 100644
index 0000000..94bbf03
--- /dev/null
+++ b/xmrstak/net/socket.hpp
@@ -0,0 +1,57 @@
+#pragma once
+#include "socks.h"
+class jpsock;
+
+class base_socket
+{
+public:
+	virtual bool set_hostname(const char* sAddr) = 0;
+	virtual bool connect() = 0;
+	virtual int recv(char* buf, unsigned int len) = 0;
+	virtual bool send(const char* buf) = 0;
+	virtual void close(bool free) = 0;
+};
+
+class plain_socket : public base_socket
+{
+public:
+	plain_socket(jpsock* err_callback);
+
+	bool set_hostname(const char* sAddr);
+	bool connect();
+	int recv(char* buf, unsigned int len);
+	bool send(const char* buf);
+	void close(bool free);
+
+private:
+	jpsock* pCallback;
+	addrinfo *pSockAddr;
+	addrinfo *pAddrRoot;
+	SOCKET hSocket;
+};
+
+typedef struct ssl_ctx_st SSL_CTX;
+typedef struct bio_st BIO;
+typedef struct ssl_st SSL;
+
+class tls_socket : public base_socket
+{
+public:
+	tls_socket(jpsock* err_callback);
+
+	bool set_hostname(const char* sAddr);
+	bool connect();
+	int recv(char* buf, unsigned int len);
+	bool send(const char* buf);
+	void close(bool free);
+
+private:
+	void init_ctx();
+	void print_error();
+
+	jpsock* pCallback;
+
+	SSL_CTX* ctx = nullptr;
+	BIO* bio = nullptr;
+	SSL* ssl = nullptr;
+};