98 lines
2.3 KiB
C++
98 lines
2.3 KiB
C++
/* XMRig
|
|
* Copyright (c) 2018-2021 SChernykh <https://github.com/SChernykh>
|
|
* Copyright (c) 2016-2021 XMRig <https://github.com/xmrig>, <support@xmrig.com>
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include "base/net/https/HttpsContext.h"
|
|
#include "3rdparty/llhttp/llhttp.h"
|
|
#include "base/net/tls/TlsContext.h"
|
|
|
|
|
|
#include <openssl/bio.h>
|
|
#include <uv.h>
|
|
|
|
|
|
xmrig::HttpsContext::HttpsContext(TlsContext *tls, const std::weak_ptr<IHttpListener> &listener) :
|
|
HttpContext(HTTP_REQUEST, listener),
|
|
ServerTls(tls ? tls->ctx() : nullptr)
|
|
{
|
|
if (!tls) {
|
|
m_mode = TLS_OFF;
|
|
}
|
|
}
|
|
|
|
|
|
xmrig::HttpsContext::~HttpsContext() = default;
|
|
|
|
|
|
void xmrig::HttpsContext::append(char *data, size_t size)
|
|
{
|
|
if (m_mode == TLS_AUTO) {
|
|
m_mode = isTLS(data, size) ? TLS_ON : TLS_OFF;
|
|
}
|
|
|
|
if (m_mode == TLS_ON) {
|
|
read(data, size);
|
|
}
|
|
else {
|
|
parse(data, size);
|
|
}
|
|
}
|
|
|
|
|
|
bool xmrig::HttpsContext::write(BIO *bio)
|
|
{
|
|
if (uv_is_writable(stream()) != 1) {
|
|
return false;
|
|
}
|
|
|
|
char *data = nullptr;
|
|
const size_t size = BIO_get_mem_data(bio, &data); // NOLINT(cppcoreguidelines-pro-type-cstyle-cast)
|
|
std::string body(data, size);
|
|
|
|
(void) BIO_reset(bio);
|
|
|
|
HttpContext::write(std::move(body), m_close);
|
|
|
|
return true;
|
|
}
|
|
|
|
|
|
void xmrig::HttpsContext::parse(char *data, size_t size)
|
|
{
|
|
if (!HttpContext::parse(data, size)) {
|
|
close();
|
|
}
|
|
}
|
|
|
|
|
|
void xmrig::HttpsContext::shutdown()
|
|
{
|
|
close();
|
|
}
|
|
|
|
|
|
void xmrig::HttpsContext::write(std::string &&data, bool close)
|
|
{
|
|
m_close = close;
|
|
|
|
if (m_mode == TLS_ON) {
|
|
send(data.data(), data.size());
|
|
}
|
|
else {
|
|
HttpContext::write(std::move(data), close);
|
|
}
|
|
}
|