// 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);