// SPDX-License-Identifier: GPL-2.0
#include <linux/zalloc.h>
#include "mem-info.h"

struct mem_info *mem_info__get(struct mem_info *mi)
{
	struct mem_info *result;

	if (RC_CHK_GET(result, mi))
		refcount_inc(mem_info__refcnt(mi));

	return result;
}

void mem_info__put(struct mem_info *mi)
{
	if (mi && refcount_dec_and_test(mem_info__refcnt(mi))) {
		addr_map_symbol__exit(mem_info__iaddr(mi));
		addr_map_symbol__exit(mem_info__daddr(mi));
		RC_CHK_FREE(mi);
	} else {
		RC_CHK_PUT(mi);
	}
}

struct mem_info *mem_info__new(void)
{
	struct mem_info *result = NULL;
	RC_STRUCT(mem_info) *mi = zalloc(sizeof(*mi));

	if (ADD_RC_CHK(result, mi))
		refcount_set(mem_info__refcnt(result), 1);

	return result;
}