/** Infiniband memory allocator.
 *
 * @author Steffen Vogel <post@steffenvogel.de>
 * @copyright 2014-2022, Institute for Automation of Complex Power Systems, EONERC
 * @license Apache 2.0
 *********************************************************************************/

#include <infiniband/verbs.h>

#include <villas/nodes/infiniband.hpp>
#include <villas/node/memory.hpp>
#include <villas/utils.hpp>
#include <villas/memory/ib.h>
#include <villas/exceptions.hpp>
#include <villas/node_compat.hpp>

using namespace villas;
using namespace villas::utils;
using namespace villas::node;
using namespace villas::node::memory;

struct ibv_mr * villas::node::memory::ib_get_mr(void *ptr)
{
	auto *ma = get_allocation(ptr);

	return ma->ib.mr;
}

static
struct Allocation * ib_alloc(size_t len, size_t alignment, struct Type *m)
{
	auto *mi = (struct IB *) m->_vd;

	auto *ma = new struct Allocation;
	if (!ma)
		throw MemoryAllocationError();

	ma->type = m;
	ma->length = len;
	ma->alignment = alignment;

	ma->parent = mi->parent->alloc(len + sizeof(struct ibv_mr *), alignment, mi->parent);
	ma->address = ma->parent->address;

	if (!mi->pd) {
		auto logger = logging.get("memory:ib");
		logger->error("Protection domain is not registered!");
	}

	ma->ib.mr = ibv_reg_mr(mi->pd, ma->address, ma->length, IBV_ACCESS_LOCAL_WRITE | IBV_ACCESS_REMOTE_WRITE);
	if (!ma->ib.mr) {
		mi->parent->free(ma->parent, mi->parent);
		delete ma;
		return nullptr;
	}

	return ma;
}

static
int ib_free(struct Allocation *ma, struct Type *m)
{
	int ret;
	auto *mi = (struct IB *) m->_vd;

	ibv_dereg_mr(ma->ib.mr);

	ret = mi->parent->free(ma->parent, mi->parent);
	if (ret)
		return ret;

	return 0;
}

struct Type * villas::node::memory::ib(NodeCompat *n, struct Type *parent)
{
	auto *i = n->getData<struct infiniband>();
	auto *mt = (struct Type *) malloc(sizeof(struct Type));

	mt->name = "ib";
	mt->flags = 0;
	mt->alloc = ib_alloc;
	mt->free = ib_free;
	mt->alignment = 1;

	mt->_vd = malloc(sizeof(struct IB));

	auto *mi = (struct memory::IB *) mt->_vd;

	mi->pd = i->ctx.pd;
	mi->parent = parent;

	return mt;
}