Reduced memory consumption on network level.

This commit is contained in:
XMRig 2020-03-20 20:53:27 +07:00
parent bb96684daf
commit 1b875fdabb
No known key found for this signature in database
GPG key ID: 446A53638BE94409
17 changed files with 462 additions and 205 deletions

View file

@ -0,0 +1,103 @@
/* XMRig
* Copyright 2018-2020 SChernykh <https://github.com/SChernykh>
* Copyright 2016-2020 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/tools/LineReader.h"
#include "base/net/tools/NetBuffer.h"
#include "base/kernel/interfaces/ILineListener.h"
#include <cassert>
#include <cstring>
xmrig::LineReader::~LineReader()
{
NetBuffer::release(m_buf);
}
void xmrig::LineReader::parse(char *data, size_t size)
{
assert(m_listener != nullptr && size > 0);
if (!m_listener || size == 0) {
return;
}
if (!m_buf) {
return getline(data, size);
}
add(data, size);
getline(m_buf, m_pos);
}
void xmrig::LineReader::reset()
{
if (m_buf) {
NetBuffer::release(m_buf);
m_buf = nullptr;
m_pos = 0;
}
}
void xmrig::LineReader::add(const char *data, size_t size)
{
if (size > NetBuffer::kChunkSize - m_pos) {
return;
}
if (!m_buf) {
m_buf = NetBuffer::allocate();
m_pos = 0;
}
memcpy(m_buf + m_pos, data, size);
m_pos += size;
}
void xmrig::LineReader::getline(char *data, size_t size)
{
char *end = nullptr;
char *start = data;
size_t remaining = size;
while ((end = static_cast<char*>(memchr(start, '\n', remaining))) != nullptr) {
*end = '\0';
end++;
const auto len = static_cast<size_t>(end - start);
if (len > 1) {
m_listener->onLine(start, len - 1);
}
remaining -= len;
start = end;
}
if (remaining == 0) {
return reset();
}
if (!m_buf || m_buf != data) {
add(start, remaining);
}
}