// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright (C) 2020 MediaTek Inc.
 */

#include <linux/ktime.h>
#include <linux/hrtimer.h>
#include <linux/sched.h>
#include <linux/mutex.h>
#include <linux/slab.h>
#include <linux/rbtree.h>
#include <linux/preempt.h>
#include <linux/trace_events.h>
#include <linux/spinlock.h>
#include <linux/vmalloc.h>
#include <linux/module.h>
#include <asm/page.h>
#include <linux/vmalloc.h>
#include <linux/slab.h>
#include <linux/mm.h>

#define CREATE_TRACE_POINTS
#include <trace/events/pob.h>

#include "pob_int.h"

#define TRACELOG_SIZE 512
struct kobject *pob_kobj;

void *pob_alloc_atomic(int i32Size)
{
	void *pvBuf;

	if (i32Size <= PAGE_SIZE)
		pvBuf = kmalloc(i32Size, GFP_ATOMIC);
	else
		pvBuf = vmalloc(i32Size);

	return pvBuf;
}

void pob_free(void *pvBuf)
{
	kvfree(pvBuf);
}

void pob_trace(const char *fmt, ...)
{
	char log[TRACELOG_SIZE];
	va_list args;
	int len;

	va_start(args, fmt);
	len = vsnprintf(log, sizeof(log), fmt, args);
	va_end(args);

	if (len == TRACELOG_SIZE)
		log[TRACELOG_SIZE - 1] = '\0';

	preempt_disable();
	trace_pob_log(log);
	preempt_enable();
}

void pob_sysfs_create_file(struct kobject *parent,
			struct kobj_attribute *kobj_attr)
{
	int ret;

	if (kobj_attr == NULL || pob_kobj == NULL)
		return;

	ret = sysfs_create_file(pob_kobj, &(kobj_attr->attr));
	if (ret)
		pr_debug("Failed to create sysfs file\n");
}

void pob_sysfs_remove_file(struct kobject *parent,
			struct kobj_attribute *kobj_attr)
{
	if (kobj_attr == NULL || pob_kobj == NULL)
		return;

	sysfs_remove_file(pob_kobj, &(kobj_attr->attr));
}

static int __init pob_init(void)
{
	if (kernel_kobj == NULL)
		return -1;

	pob_kobj = kobject_create_and_add("pob", kernel_kobj);
	if (pob_kobj == NULL)
		return -1;

	pob_qos_init(pob_kobj);

	return 0;
}

static void __exit pob_exit(void)
{
	pob_qos_exit(pob_kobj);

	if (pob_kobj == NULL)
		return;

	kobject_put(pob_kobj);
	pob_kobj = NULL;
}

module_init(pob_init);
module_exit(pob_exit);

MODULE_LICENSE("GPL");
MODULE_DESCRIPTION("MediaTek Performance Observer");
MODULE_AUTHOR("MediaTek Inc.");