From 548150779dbe40d7a1a4285fb45c7231eb33cc35 Mon Sep 17 00:00:00 2001 From: aozhiwei Date: Sun, 28 May 2023 09:13:38 +0800 Subject: [PATCH] 1 --- a8/asiotcpclient.cc | 2 +- a8/asiotcpclient.h | 2 +- a8/websocketclient.cc | 25 +++++++++++++++++++++++++ a8/websocketclient.h | 28 ++++++++++++++++++++++++++++ 4 files changed, 55 insertions(+), 2 deletions(-) create mode 100644 a8/websocketclient.cc create mode 100644 a8/websocketclient.h diff --git a/a8/asiotcpclient.cc b/a8/asiotcpclient.cc index ed5e253..db32390 100644 --- a/a8/asiotcpclient.cc +++ b/a8/asiotcpclient.cc @@ -8,7 +8,7 @@ #include #include -#ifdef USE_ASIO +#ifdef USE_BOOST const int MAX_RECV_BUFFERSIZE = 1024 * 64; diff --git a/a8/asiotcpclient.h b/a8/asiotcpclient.h index a5b6b11..ca59cb9 100644 --- a/a8/asiotcpclient.h +++ b/a8/asiotcpclient.h @@ -1,6 +1,6 @@ #pragma once -#ifdef USE_ASIO +#ifdef USE_BOOST #include diff --git a/a8/websocketclient.cc b/a8/websocketclient.cc new file mode 100644 index 0000000..f773e16 --- /dev/null +++ b/a8/websocketclient.cc @@ -0,0 +1,25 @@ +#include + +#include + +#ifdef USE_BOOST + +namespace a8 +{ + + WebSocketClient::WebSocketClient(asio::io_context& io_context, const std::string& remote_ip, int remote_port): AsioTcpClient(io_context, remote_ip, remote_port) + { + decoded_buff_ = (char *)malloc(1024 * 64 + 1); + decoded_bufflen_ = 0; + } + + WebSocketClient::~WebSocketClient() + { + free(decoded_buff_); + decoded_buff_ = nullptr; + decoded_bufflen_ = 0; + } + +} + +#endif diff --git a/a8/websocketclient.h b/a8/websocketclient.h new file mode 100644 index 0000000..2d74705 --- /dev/null +++ b/a8/websocketclient.h @@ -0,0 +1,28 @@ +#pragma once + +#include + +#ifdef USE_BOOST + +#include + +using asio::ip::tcp; + +namespace a8 +{ + + class WebSocketClient : public AsioTcpClient + { + public: + WebSocketClient(asio::io_context& io_context, const std::string& remote_ip, int remote_port); + virtual ~WebSocketClient() override; + + private: + char *decoded_buff_ = nullptr; + int decoded_bufflen_ = 0; + bool handshook_ = false; + }; + +} + +#endif