// SPDX-License-Identifier: GPL-2.0-only
/*
 * Creating audit records for mapped devices.
 *
 * Copyright (C) 2021 Fraunhofer AISEC. All rights reserved.
 *
 * Authors: Michael Weiß <michael.weiss@aisec.fraunhofer.de>
 */

#include <linux/audit.h>
#include <linux/module.h>
#include <linux/device-mapper.h>
#include <linux/bio.h>
#include <linux/blkdev.h>

#include "dm-audit.h"
#include "dm-core.h"

static struct audit_buffer *dm_audit_log_start(int audit_type,
					       const char *dm_msg_prefix,
					       const char *op)
{
	struct audit_buffer *ab;

	if (audit_enabled == AUDIT_OFF)
		return NULL;

	ab = audit_log_start(audit_context(), GFP_KERNEL, audit_type);
	if (unlikely(!ab))
		return NULL;

	audit_log_format(ab, "module=%s op=%s", dm_msg_prefix, op);
	return ab;
}

void dm_audit_log_ti(int audit_type, const char *dm_msg_prefix, const char *op,
		     struct dm_target *ti, int result)
{
	struct audit_buffer *ab = NULL;
	struct mapped_device *md = dm_table_get_md(ti->table);
	int dev_major = dm_disk(md)->major;
	int dev_minor = dm_disk(md)->first_minor;

	switch (audit_type) {
	case AUDIT_DM_CTRL:
		ab = dm_audit_log_start(audit_type, dm_msg_prefix, op);
		if (unlikely(!ab))
			return;
		audit_log_task_info(ab);
		audit_log_format(ab, " dev=%d:%d error_msg='%s'", dev_major,
				 dev_minor, !result ? ti->error : "success");
		break;
	case AUDIT_DM_EVENT:
		ab = dm_audit_log_start(audit_type, dm_msg_prefix, op);
		if (unlikely(!ab))
			return;
		audit_log_format(ab, " dev=%d:%d sector=?", dev_major,
				 dev_minor);
		break;
	default: /* unintended use */
		return;
	}

	audit_log_format(ab, " res=%d", result);
	audit_log_end(ab);
}
EXPORT_SYMBOL_GPL(dm_audit_log_ti);

void dm_audit_log_bio(const char *dm_msg_prefix, const char *op,
		      struct bio *bio, sector_t sector, int result)
{
	struct audit_buffer *ab;
	int dev_major = MAJOR(bio->bi_bdev->bd_dev);
	int dev_minor = MINOR(bio->bi_bdev->bd_dev);

	ab = dm_audit_log_start(AUDIT_DM_EVENT, dm_msg_prefix, op);
	if (unlikely(!ab))
		return;

	audit_log_format(ab, " dev=%d:%d sector=%llu res=%d",
			 dev_major, dev_minor, sector, result);
	audit_log_end(ab);
}
EXPORT_SYMBOL_GPL(dm_audit_log_bio);