103 lines
2.4 KiB
C++
103 lines
2.4 KiB
C++
/* XMRig
|
|
* Copyright (c) 2020 cohcho <https://github.com/cohcho>
|
|
* Copyright (c) 2018-2020 SChernykh <https://github.com/SChernykh>
|
|
* Copyright (c) 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;
|
|
}
|
|
|
|
getline(data, size);
|
|
}
|
|
|
|
|
|
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) {
|
|
// it breakes correctness silently for long lines
|
|
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 (m_pos) {
|
|
add(start, len);
|
|
m_listener->onLine(m_buf, m_pos - 1);
|
|
m_pos = 0;
|
|
}
|
|
else if (len > 1) {
|
|
m_listener->onLine(start, len - 1);
|
|
}
|
|
|
|
remaining -= len;
|
|
start = end;
|
|
}
|
|
|
|
if (remaining == 0) {
|
|
return reset();
|
|
}
|
|
|
|
add(start, remaining);
|
|
}
|