// SPDX-License-Identifier: GPL-2.0

#include <linux/module.h>
#include <linux/init.h>
#include <linux/proc_fs.h>

/* Global variables for procfs folder and file */
static struct proc_dir_entry *proc_folder;
static struct proc_dir_entry *proc_file;

/**
 * @brief Read data out of the buffer
 */
static ssize_t my_read(struct file *File, char *user_buffer, size_t count, loff_t *offs)
{
	char text[] = "Hello from a procfs file!\n";
	int to_copy, not_copied, delta;

	/* Get amount of data to copy */
	to_copy = min(count, sizeof(text));

	/* Copy data to user */
	not_copied = copy_to_user(user_buffer, text, to_copy);

	/* Calculate data */
	delta = to_copy - not_copied;

	return delta;
}

/**
 * @brief Write data to buffer
 */
static ssize_t my_write(struct file *File, const char *user_buffer, size_t count, loff_t *offs){
	char text[255];
	int to_copy, not_copied, delta;

	/* Clear text */
	memset(text, 0, sizeof(text));

	/* Get amount of data to copy */
	to_copy = min(count, sizeof(text));

	/* Copy data to user */
	not_copied = copy_from_user(text, user_buffer, to_copy);
	pr_notice("proc_fs - You have written %s to me\n", text);

	/* Calculate data */
	delta = to_copy - not_copied;

	return delta;
}

static struct proc_ops fops =
{
	.proc_read = my_read,
	.proc_write = my_write,
};

/**
 * @brief This function is called when the module is loaded into the kernel
 */
static int __init my_init(void)
{
	/* /proc/hello/dummy */

	proc_folder = proc_mkdir("hello", NULL);
	if(proc_folder == NULL)
	{
		pr_err("proc_fs - Error creating /proc/hello\n");
		return -ENOMEM;
	}

	proc_file = proc_create("dummy", 0666, proc_folder, &fops);
	if(proc_file == NULL)
	{
		pr_err("proc_fs - Error creating /proc/hello/dummy\n");
		return -ENOMEM;
	}

	pr_notice("proc_fs - Created /proc/hello/dummy\n");

	pr_notice("Hello, Kernel!\n");
	return 0;
}

/**
 * @brief This function is called when the module is removed from the kernel
 */
static void __exit my_exit(void)
{
	pr_notice("proc_fs - Removing /proc/hello/dummy\n");
	proc_remove(proc_folder);
	proc_remove(proc_file);
	pr_info("Goodbye, Kernel\n");
}

module_init(my_init);
module_exit(my_exit);

/* Meta Information */
MODULE_LICENSE("GPL v2");
MODULE_AUTHOR("David D'Ulisse");
MODULE_DESCRIPTION("A Hello, World! Test Module");