![support@xmrig.com](/assets/img/avatar_default.png)
17 changed files with 440 additions and 123 deletions
@ -0,0 +1,52 @@ |
|||
/* XMRig
|
|||
* Copyright 2010 Jeff Garzik <jgarzik@pobox.com> |
|||
* Copyright 2012-2014 pooler <pooler@litecoinpool.org> |
|||
* Copyright 2014 Lucas Jones <https://github.com/lucasjones>
|
|||
* Copyright 2014-2016 Wolf9466 <https://github.com/OhGodAPet>
|
|||
* Copyright 2016 Jay D Dee <jayddee246@gmail.com> |
|||
* Copyright 2017-2018 XMR-Stak <https://github.com/fireice-uk>, <https://github.com/psychocrypt>
|
|||
* Copyright 2018 Lee Clagett <https://github.com/vtnerd>
|
|||
* Copyright 2018-2019 SChernykh <https://github.com/SChernykh>
|
|||
* Copyright 2018-2019 tevador <tevador@gmail.com> |
|||
* Copyright 2016-2019 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/>.
|
|||
*/ |
|||
|
|||
#ifndef XMRIG_IMEMORYPOOL_H |
|||
#define XMRIG_IMEMORYPOOL_H |
|||
|
|||
|
|||
#include <cstdint> |
|||
|
|||
|
|||
namespace xmrig { |
|||
|
|||
|
|||
class IMemoryPool |
|||
{ |
|||
public: |
|||
virtual ~IMemoryPool() = default; |
|||
|
|||
virtual bool isHugePages(uint32_t node) const = 0; |
|||
virtual uint8_t *get(size_t size, uint32_t node) = 0; |
|||
virtual void release(uint32_t node) = 0; |
|||
}; |
|||
|
|||
|
|||
} /* namespace xmrig */ |
|||
|
|||
|
|||
|
|||
#endif /* XMRIG_IMEMORYPOOL_H */ |
@ -0,0 +1,95 @@ |
|||
/* XMRig
|
|||
* Copyright 2010 Jeff Garzik <jgarzik@pobox.com> |
|||
* Copyright 2012-2014 pooler <pooler@litecoinpool.org> |
|||
* Copyright 2014 Lucas Jones <https://github.com/lucasjones>
|
|||
* Copyright 2014-2016 Wolf9466 <https://github.com/OhGodAPet>
|
|||
* Copyright 2016 Jay D Dee <jayddee246@gmail.com> |
|||
* Copyright 2017-2018 XMR-Stak <https://github.com/fireice-uk>, <https://github.com/psychocrypt>
|
|||
* Copyright 2018 Lee Clagett <https://github.com/vtnerd>
|
|||
* Copyright 2018-2019 SChernykh <https://github.com/SChernykh>
|
|||
* Copyright 2018-2019 tevador <tevador@gmail.com> |
|||
* Copyright 2016-2019 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 "crypto/common/MemoryPool.h" |
|||
#include "crypto/common/VirtualMemory.h" |
|||
|
|||
|
|||
#include <cassert> |
|||
|
|||
|
|||
namespace xmrig { |
|||
|
|||
|
|||
constexpr size_t pageSize = 2 * 1024 * 1024; |
|||
|
|||
|
|||
} // namespace xmrig
|
|||
|
|||
|
|||
xmrig::MemoryPool::MemoryPool(size_t size, bool hugePages, uint32_t node) : |
|||
m_size(size) |
|||
{ |
|||
if (!size) { |
|||
return; |
|||
} |
|||
|
|||
m_memory = new VirtualMemory(size * pageSize, hugePages, false, node); |
|||
} |
|||
|
|||
|
|||
xmrig::MemoryPool::~MemoryPool() |
|||
{ |
|||
delete m_memory; |
|||
} |
|||
|
|||
|
|||
bool xmrig::MemoryPool::isHugePages(uint32_t) const |
|||
{ |
|||
return m_memory && m_memory->isHugePages(); |
|||
} |
|||
|
|||
|
|||
uint8_t *xmrig::MemoryPool::get(size_t size, uint32_t) |
|||
{ |
|||
assert(!(size % pageSize)); |
|||
|
|||
if (!m_memory || (m_memory->size() - m_offset) < size) { |
|||
return nullptr; |
|||
} |
|||
|
|||
uint8_t *out = m_memory->scratchpad() + m_offset; |
|||
|
|||
m_offset += size; |
|||
++m_refs; |
|||
|
|||
return out; |
|||
} |
|||
|
|||
|
|||
void xmrig::MemoryPool::release(uint32_t) |
|||
{ |
|||
assert(m_refs > 0); |
|||
|
|||
if (m_refs > 0) { |
|||
--m_refs; |
|||
} |
|||
|
|||
if (m_refs == 0) { |
|||
m_offset = 0; |
|||
} |
|||
} |
@ -0,0 +1,66 @@ |
|||
/* XMRig
|
|||
* Copyright 2010 Jeff Garzik <jgarzik@pobox.com> |
|||
* Copyright 2012-2014 pooler <pooler@litecoinpool.org> |
|||
* Copyright 2014 Lucas Jones <https://github.com/lucasjones>
|
|||
* Copyright 2014-2016 Wolf9466 <https://github.com/OhGodAPet>
|
|||
* Copyright 2016 Jay D Dee <jayddee246@gmail.com> |
|||
* Copyright 2017-2018 XMR-Stak <https://github.com/fireice-uk>, <https://github.com/psychocrypt>
|
|||
* Copyright 2018 Lee Clagett <https://github.com/vtnerd>
|
|||
* Copyright 2018-2019 SChernykh <https://github.com/SChernykh>
|
|||
* Copyright 2018-2019 tevador <tevador@gmail.com> |
|||
* Copyright 2016-2019 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/>.
|
|||
*/ |
|||
|
|||
#ifndef XMRIG_MEMORYPOOL_H |
|||
#define XMRIG_MEMORYPOOL_H |
|||
|
|||
|
|||
#include "backend/common/interfaces/IMemoryPool.h" |
|||
#include "base/tools/Object.h" |
|||
|
|||
|
|||
namespace xmrig { |
|||
|
|||
|
|||
class VirtualMemory; |
|||
|
|||
|
|||
class MemoryPool : public IMemoryPool |
|||
{ |
|||
public: |
|||
XMRIG_DISABLE_COPY_MOVE_DEFAULT(MemoryPool) |
|||
|
|||
MemoryPool(size_t size, bool hugePages, uint32_t node = 0); |
|||
~MemoryPool() override; |
|||
|
|||
protected: |
|||
bool isHugePages(uint32_t node) const override; |
|||
uint8_t *get(size_t size, uint32_t node) override; |
|||
void release(uint32_t node) override; |
|||
|
|||
private: |
|||
size_t m_size = 0; |
|||
size_t m_refs = 0; |
|||
size_t m_offset = 0; |
|||
VirtualMemory *m_memory = nullptr; |
|||
}; |
|||
|
|||
|
|||
} /* namespace xmrig */ |
|||
|
|||
|
|||
|
|||
#endif /* XMRIG_MEMORYPOOL_H */ |
@ -0,0 +1,56 @@ |
|||
/* XMRig
|
|||
* Copyright 2010 Jeff Garzik <jgarzik@pobox.com> |
|||
* Copyright 2012-2014 pooler <pooler@litecoinpool.org> |
|||
* Copyright 2014 Lucas Jones <https://github.com/lucasjones>
|
|||
* Copyright 2014-2016 Wolf9466 <https://github.com/OhGodAPet>
|
|||
* Copyright 2016 Jay D Dee <jayddee246@gmail.com> |
|||
* Copyright 2017-2018 XMR-Stak <https://github.com/fireice-uk>, <https://github.com/psychocrypt>
|
|||
* Copyright 2018 Lee Clagett <https://github.com/vtnerd>
|
|||
* Copyright 2018-2019 SChernykh <https://github.com/SChernykh>
|
|||
* Copyright 2018-2019 tevador <tevador@gmail.com> |
|||
* Copyright 2016-2019 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 "crypto/common/VirtualMemory.h" |
|||
#include "backend/cpu/Cpu.h" |
|||
#include "backend/cpu/platform/HwlocCpuInfo.h" |
|||
#include "base/io/log/Log.h" |
|||
|
|||
|
|||
#include <hwloc.h> |
|||
|
|||
|
|||
uint32_t xmrig::VirtualMemory::bindToNUMANode(int64_t affinity) |
|||
{ |
|||
if (affinity < 0 || Cpu::info()->nodes() < 2) { |
|||
return 0; |
|||
} |
|||
|
|||
auto cpu = static_cast<HwlocCpuInfo *>(Cpu::info()); |
|||
hwloc_obj_t pu = hwloc_get_pu_obj_by_os_index(cpu->topology(), static_cast<unsigned>(affinity)); |
|||
|
|||
char *buffer; |
|||
hwloc_bitmap_asprintf(&buffer, pu->cpuset); |
|||
|
|||
if (pu == nullptr || !cpu->membind(pu->nodeset)) { |
|||
LOG_WARN("CPU #%02" PRId64 " warning: \"can't bind memory\"", affinity); |
|||
|
|||
return 0; |
|||
} |
|||
|
|||
return hwloc_bitmap_first(pu->nodeset); |
|||
} |
Loading…
Reference in new issue