/*
 * RAM Oops/Panic logger
 *
 * Copyright (C) 2010 Marco Stornelli <marco.stornelli@gmail.com>
 * Copyright (C) 2011 Kees Cook <keescook@chromium.org>
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * version 2 as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301 USA
 *
 */

#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt

#include <linux/kernel.h>
#include <linux/err.h>
#include <linux/module.h>
#include <linux/version.h>
#include <linux/pstore.h>
#include <linux/io.h>
#include <linux/ioport.h>
#include <linux/platform_device.h>
#include <linux/slab.h>
#include <linux/compiler.h>
#include <linux/pstore_ram.h>
#include <linux/of.h>
#include <linux/of_address.h>

#define RAMOOPS_KERNMSG_HDR "===="
#define MIN_MEM_SIZE 4096UL

static ulong record_size = MIN_MEM_SIZE;
module_param(record_size, ulong, 0400);
MODULE_PARM_DESC(record_size,
                "size of each dump done on oops/panic");

static ulong ramoops_console_size = MIN_MEM_SIZE;
module_param_named(console_size, ramoops_console_size, ulong, 0400);
MODULE_PARM_DESC(console_size, "size of kernel console log");

static ulong ramoops_ftrace_size = MIN_MEM_SIZE;
module_param_named(ftrace_size, ramoops_ftrace_size, ulong, 0400);
MODULE_PARM_DESC(ftrace_size, "size of ftrace log");

static ulong ramoops_pmsg_size = MIN_MEM_SIZE;
module_param_named(pmsg_size, ramoops_pmsg_size, ulong, 0400);
MODULE_PARM_DESC(pmsg_size, "size of user space message log");

static unsigned long long mem_address;
module_param_hw(mem_address, ullong, other, 0400);
MODULE_PARM_DESC(mem_address,
                "start of reserved RAM used to store oops/panic logs");

static ulong mem_size;
module_param(mem_size, ulong, 0400);
MODULE_PARM_DESC(mem_size,
                "size of reserved RAM used to store oops/panic logs");

static unsigned int mem_type;
module_param(mem_type, uint, 0600);
MODULE_PARM_DESC(mem_type,
                "set to 1 to try to use unbuffered memory (default 0)");

static int dump_oops = 1;
module_param(dump_oops, int, 0600);
MODULE_PARM_DESC(dump_oops,
                "set to 1 to dump oopses, 0 to only dump panics (default 1)");

static int ramoops_ecc;
module_param_named(ecc, ramoops_ecc, int, 0600);
MODULE_PARM_DESC(ramoops_ecc,
                "if non-zero, the option enables ECC support and specifies "
                "ECC buffer size in bytes (1 is a special value, means 16 "
                "bytes ECC)");

struct ramoops_context {
        struct persistent_ram_zone **dprzs;     /* Oops dump zones */
        struct persistent_ram_zone *cprz;       /* Console zone */
        struct persistent_ram_zone **fprzs;     /* Ftrace zones */
        struct persistent_ram_zone *mprz;       /* PMSG zone */
        phys_addr_t phys_addr;
        unsigned long size;
        unsigned int memtype;
        size_t record_size;
        size_t console_size;
        size_t ftrace_size;
        size_t pmsg_size;
        int dump_oops;
        u32 flags;
        struct persistent_ram_ecc_info ecc_info;
        unsigned int max_dump_cnt;
        unsigned int dump_write_cnt;
        /* _read_cnt need clear on ramoops_pstore_open */
        unsigned int dump_read_cnt;
        unsigned int console_read_cnt;
        unsigned int max_ftrace_cnt;
        unsigned int ftrace_read_cnt;
        unsigned int pmsg_read_cnt;
        struct pstore_info pstore;
};

static struct platform_device *dummy;
static struct ramoops_platform_data *dummy_data;

static int ramoops_pstore_open(struct pstore_info *psi)
{
        struct ramoops_context *cxt = psi->data;

        cxt->dump_read_cnt = 0;
        cxt->console_read_cnt = 0;
        cxt->ftrace_read_cnt = 0;
        cxt->pmsg_read_cnt = 0;
        return 0;
}

static struct persistent_ram_zone *
ramoops_get_next_prz(struct persistent_ram_zone *przs[], uint *c, uint max,
                     u64 *id,
                     enum pstore_type_id *typep, enum pstore_type_id type,
                     bool update)
{
        struct persistent_ram_zone *prz;
        int i = (*c)++;

        /* Give up if we never existed or have hit the end. */
        if (!przs || i >= max)
                return NULL;

        prz = przs[i];
        if (!prz)
                return NULL;

        /* Update old/shadowed buffer. */
        if (update)
                persistent_ram_save_old(prz);

        if (!persistent_ram_old_size(prz))
                return NULL;

        *typep = type;
        *id = i;

        return prz;
}

static int ramoops_read_kmsg_hdr(char *buffer, struct timespec *time,
                                  bool *compressed)
{
        char data_type;
        int header_length = 0;

        if (sscanf(buffer, RAMOOPS_KERNMSG_HDR "%lu.%lu-%c\n%n", &time->tv_sec,
                        &time->tv_nsec, &data_type, &header_length) == 3) {
                if (data_type == 'C')
                        *compressed = true;
                else
                        *compressed = false;
        } else if (sscanf(buffer, RAMOOPS_KERNMSG_HDR "%lu.%lu\n%n",
                        &time->tv_sec, &time->tv_nsec, &header_length) == 2) {
                        *compressed = false;
        } else {
                time->tv_sec = 0;
                time->tv_nsec = 0;
                *compressed = false;
        }
        return header_length;
}

static bool prz_ok(struct persistent_ram_zone *prz)
{
        return !!prz && !!(persistent_ram_old_size(prz) +
                           persistent_ram_ecc_string(prz, NULL, 0));
}

static ssize_t ftrace_log_combine(struct persistent_ram_zone *dest,
                                  struct persistent_ram_zone *src)
{
        size_t dest_size, src_size, total, dest_off, src_off;
        size_t dest_idx = 0, src_idx = 0, merged_idx = 0;
        void *merged_buf;
        struct pstore_ftrace_record *drec, *srec, *mrec;
        size_t record_size = sizeof(struct pstore_ftrace_record);

        dest_off = dest->old_log_size % record_size;
        dest_size = dest->old_log_size - dest_off;

        src_off = src->old_log_size % record_size;
        src_size = src->old_log_size - src_off;

        total = dest_size + src_size;
        merged_buf = kmalloc(total, GFP_KERNEL);
        if (!merged_buf)
                return -ENOMEM;

        drec = (struct pstore_ftrace_record *)(dest->old_log + dest_off);
        srec = (struct pstore_ftrace_record *)(src->old_log + src_off);
        mrec = (struct pstore_ftrace_record *)(merged_buf);

        while (dest_size > 0 && src_size > 0) {
                if (pstore_ftrace_read_timestamp(&drec[dest_idx]) <
                    pstore_ftrace_read_timestamp(&srec[src_idx])) {
                        mrec[merged_idx++] = drec[dest_idx++];
                        dest_size -= record_size;
                } else {
                        mrec[merged_idx++] = srec[src_idx++];
                        src_size -= record_size;
                }
        }

        while (dest_size > 0) {
                mrec[merged_idx++] = drec[dest_idx++];
                dest_size -= record_size;
        }

        while (src_size > 0) {
                mrec[merged_idx++] = srec[src_idx++];
                src_size -= record_size;
        }

        kfree(dest->old_log);
        dest->old_log = merged_buf;
        dest->old_log_size = total;

        return 0;
}

static ssize_t ramoops_pstore_read(struct pstore_record *record)
{
        ssize_t size = 0;
        struct ramoops_context *cxt = record->psi->data;
        struct persistent_ram_zone *prz = NULL;
        int header_length = 0;
        bool free_prz = false;

        /*
         * Ramoops headers provide time stamps for PSTORE_TYPE_DMESG, but
         * PSTORE_TYPE_CONSOLE and PSTORE_TYPE_FTRACE don't currently have
         * valid time stamps, so it is initialized to zero.
         */
        record->time.tv_sec = 0;
        record->time.tv_nsec = 0;
        record->compressed = false;

        /* Find the next valid persistent_ram_zone for DMESG */
        while (cxt->dump_read_cnt < cxt->max_dump_cnt && !prz) {
                prz = ramoops_get_next_prz(cxt->dprzs, &cxt->dump_read_cnt,
                                           cxt->max_dump_cnt, &record->id,
                                           &record->type,
                                           PSTORE_TYPE_DMESG, 1);
                if (!prz_ok(prz))
                        continue;
                header_length = ramoops_read_kmsg_hdr(persistent_ram_old(prz),
                                                      &record->time,
                                                      &record->compressed);
                /* Clear and skip this DMESG record if it has no valid header */
                if (!header_length) {
                        persistent_ram_free_old(prz);
                        persistent_ram_zap(prz);
                        prz = NULL;
                }
        }

        if (!prz_ok(prz))
                prz = ramoops_get_next_prz(&cxt->cprz, &cxt->console_read_cnt,
                                           1, &record->id, &record->type,
                                           PSTORE_TYPE_CONSOLE, 0);

        if (!prz_ok(prz))
                prz = ramoops_get_next_prz(&cxt->mprz, &cxt->pmsg_read_cnt,
                                           1, &record->id, &record->type,
                                           PSTORE_TYPE_PMSG, 0);

        /* ftrace is last since it may want to dynamically allocate memory. */
        if (!prz_ok(prz)) {
                if (!(cxt->flags & RAMOOPS_FLAG_FTRACE_PER_CPU)) {
                        prz = ramoops_get_next_prz(cxt->fprzs,
                                        &cxt->ftrace_read_cnt, 1, &record->id,
                                        &record->type, PSTORE_TYPE_FTRACE, 0);
                } else {
                        /*
                         * Build a new dummy record which combines all the
                         * per-cpu records including metadata and ecc info.
                         */
                        struct persistent_ram_zone *tmp_prz, *prz_next;

                        tmp_prz = kzalloc(sizeof(struct persistent_ram_zone),
                                          GFP_KERNEL);
                        if (!tmp_prz)
                                return -ENOMEM;
                        free_prz = true;

                        while (cxt->ftrace_read_cnt < cxt->max_ftrace_cnt) {
                                prz_next = ramoops_get_next_prz(cxt->fprzs,
                                                &cxt->ftrace_read_cnt,
                                                cxt->max_ftrace_cnt,
                                                &record->id,
                                                &record->type,
                                                PSTORE_TYPE_FTRACE, 0);

                                if (!prz_ok(prz_next))
                                        continue;

                                tmp_prz->ecc_info = prz_next->ecc_info;
                                tmp_prz->corrected_bytes +=
                                                prz_next->corrected_bytes;
                                tmp_prz->bad_blocks += prz_next->bad_blocks;
                                size = ftrace_log_combine(tmp_prz, prz_next);
                                if (size)
                                        goto out;
                        }
                        record->id = 0;
                        prz = tmp_prz;
                }
        }

        if (!prz_ok(prz)) {
                size = 0;
                goto out;
        }

        size = persistent_ram_old_size(prz) - header_length;

        /* ECC correction notice */
        record->ecc_notice_size = persistent_ram_ecc_string(prz, NULL, 0);

        record->buf = kmalloc(size + record->ecc_notice_size + 1, GFP_KERNEL);
        if (record->buf == NULL) {
                size = -ENOMEM;
                goto out;
        }

        memcpy(record->buf, (char *)persistent_ram_old(prz) + header_length,
               size);

        persistent_ram_ecc_string(prz, record->buf + size,
                                  record->ecc_notice_size + 1);

out:
        if (free_prz) {
                kfree(prz->old_log);
                kfree(prz);
        }

        return size;
}

static size_t ramoops_write_kmsg_hdr(struct persistent_ram_zone *prz,
                                     struct pstore_record *record)
{
        char *hdr;
        size_t len;

        hdr = kasprintf(GFP_ATOMIC, RAMOOPS_KERNMSG_HDR "%lu.%lu-%c\n",
                record->time.tv_sec,
                record->time.tv_nsec / 1000,
                record->compressed ? 'C' : 'D');
        WARN_ON_ONCE(!hdr);
        len = hdr ? strlen(hdr) : 0;
        persistent_ram_write(prz, hdr, len);
        kfree(hdr);

        return len;
}

static int notrace ramoops_pstore_write(struct pstore_record *record)
{
        struct ramoops_context *cxt = record->psi->data;
        struct persistent_ram_zone *prz;
        size_t size, hlen;

        if (record->type == PSTORE_TYPE_CONSOLE) {
                if (!cxt->cprz)
                        return -ENOMEM;
                persistent_ram_write(cxt->cprz, record->buf, record->size);
                return 0;
        } else if (record->type == PSTORE_TYPE_FTRACE) {
                int zonenum;

                if (!cxt->fprzs)
                        return -ENOMEM;
                /*
                 * Choose zone by if we're using per-cpu buffers.
                 */
                if (cxt->flags & RAMOOPS_FLAG_FTRACE_PER_CPU)
                        zonenum = smp_processor_id();
                else
                        zonenum = 0;

                persistent_ram_write(cxt->fprzs[zonenum], record->buf,
                                     record->size);
                return 0;
        } else if (record->type == PSTORE_TYPE_PMSG) {
                pr_warn_ratelimited("PMSG shouldn't call %s\n", __func__);
                return -EINVAL;
        }

        if (record->type != PSTORE_TYPE_DMESG)
                return -EINVAL;

        /*
         * Out of the various dmesg dump types, ramoops is currently designed
         * to only store crash logs, rather than storing general kernel logs.
         */
        if (record->reason != KMSG_DUMP_OOPS &&
            record->reason != KMSG_DUMP_PANIC)
                return -EINVAL;

        /* Skip Oopes when configured to do so. */
        if (record->reason == KMSG_DUMP_OOPS && !cxt->dump_oops)
                return -EINVAL;

        /*
         * Explicitly only take the first part of any new crash.
         * If our buffer is larger than kmsg_bytes, this can never happen,
         * and if our buffer is smaller than kmsg_bytes, we don't want the
         * report split across multiple records.
         */
        if (record->part != 1)
                return -ENOSPC;

        if (!cxt->dprzs)
                return -ENOSPC;

        prz = cxt->dprzs[cxt->dump_write_cnt];

        /* Build header and append record contents. */
        hlen = ramoops_write_kmsg_hdr(prz, record);
        size = record->size;
        if (size + hlen > prz->buffer_size)
                size = prz->buffer_size - hlen;
        persistent_ram_write(prz, record->buf, size);

        cxt->dump_write_cnt = (cxt->dump_write_cnt + 1) % cxt->max_dump_cnt;

        return 0;
}

static int notrace ramoops_pstore_write_user(struct pstore_record *record,
                                             const char __user *buf)
{
        if (record->type == PSTORE_TYPE_PMSG) {
                struct ramoops_context *cxt = record->psi->data;

                if (!cxt->mprz)
                        return -ENOMEM;
                return persistent_ram_write_user(cxt->mprz, buf, record->size);
        }

        return -EINVAL;
}

static int ramoops_pstore_erase(struct pstore_record *record)
{
        struct ramoops_context *cxt = record->psi->data;
        struct persistent_ram_zone *prz;

        switch (record->type) {
        case PSTORE_TYPE_DMESG:
                if (record->id >= cxt->max_dump_cnt)
                        return -EINVAL;
                prz = cxt->dprzs[record->id];
                break;
        case PSTORE_TYPE_CONSOLE:
                prz = cxt->cprz;
                break;
        case PSTORE_TYPE_FTRACE:
                if (record->id >= cxt->max_ftrace_cnt)
                        return -EINVAL;
                prz = cxt->fprzs[record->id];
                break;
        case PSTORE_TYPE_PMSG:
                prz = cxt->mprz;
                break;
        default:
                return -EINVAL;
        }

        persistent_ram_free_old(prz);
        persistent_ram_zap(prz);

        return 0;
}

static struct ramoops_context oops_cxt = {
        .pstore = {
                .owner  = THIS_MODULE,
                .name   = "ramoops",
                .open   = ramoops_pstore_open,
                .read   = ramoops_pstore_read,
                .write  = ramoops_pstore_write,
                .write_user     = ramoops_pstore_write_user,
                .erase  = ramoops_pstore_erase,
        },
};

static void ramoops_free_przs(struct ramoops_context *cxt)
{
        int i;

        /* Free dump PRZs */
        if (cxt->dprzs) {
                for (i = 0; i < cxt->max_dump_cnt; i++)
                        persistent_ram_free(cxt->dprzs[i]);

                kfree(cxt->dprzs);
                cxt->max_dump_cnt = 0;
        }

        /* Free ftrace PRZs */
        if (cxt->fprzs) {
                for (i = 0; i < cxt->max_ftrace_cnt; i++)
                        persistent_ram_free(cxt->fprzs[i]);
                kfree(cxt->fprzs);
                cxt->max_ftrace_cnt = 0;
        }
}

static int ramoops_init_przs(const char *name,
                             struct device *dev, struct ramoops_context *cxt,
                             struct persistent_ram_zone ***przs,
                             phys_addr_t *paddr, size_t mem_sz,
                             ssize_t record_size,
                             unsigned int *cnt, u32 sig, u32 flags)
{
        int err = -ENOMEM;
        int i;
        size_t zone_sz;
        struct persistent_ram_zone **prz_ar;

        /* Allocate nothing for 0 mem_sz or 0 record_size. */
        if (mem_sz == 0 || record_size == 0) {
                *cnt = 0;
                return 0;
        }

        /*
         * If we have a negative record size, calculate it based on
         * mem_sz / *cnt. If we have a positive record size, calculate
         * cnt from mem_sz / record_size.
         */
        if (record_size < 0) {
                if (*cnt == 0)
                        return 0;
                record_size = mem_sz / *cnt;
                if (record_size == 0) {
                        dev_err(dev, "%s record size == 0 (%zu / %u)\n",
                                name, mem_sz, *cnt);
                        goto fail;
                }
        } else {
                *cnt = mem_sz / record_size;
                if (*cnt == 0) {
                        dev_err(dev, "%s record count == 0 (%zu / %zu)\n",
                                name, mem_sz, record_size);
                        goto fail;
                }
        }

        if (*paddr + mem_sz - cxt->phys_addr > cxt->size) {
                dev_err(dev, "no room for %s mem region (0x%zx@0x%llx) in (0x%lx@0x%llx)\n",
                        name,
                        mem_sz, (unsigned long long)*paddr,
                        cxt->size, (unsigned long long)cxt->phys_addr);
                goto fail;
        }

        zone_sz = mem_sz / *cnt;
        if (!zone_sz) {
                dev_err(dev, "%s zone size == 0\n", name);
                goto fail;
        }

        prz_ar = kcalloc(*cnt, sizeof(**przs), GFP_KERNEL);
        if (!prz_ar)
                goto fail;

        for (i = 0; i < *cnt; i++) {
                prz_ar[i] = persistent_ram_new(*paddr, zone_sz, sig,
                                                  &cxt->ecc_info,
                                                  cxt->memtype, flags);
                if (IS_ERR(prz_ar[i])) {
                        err = PTR_ERR(prz_ar[i]);
                        dev_err(dev, "failed to request %s mem region (0x%zx@0x%llx): %d\n",
                                name, record_size,
                                (unsigned long long)*paddr, err);

                        while (i > 0) {
                                i--;
                                persistent_ram_free(prz_ar[i]);
                        }
                        kfree(prz_ar);
                        goto fail;
                }
                *paddr += zone_sz;
        }

        *przs = prz_ar;
        return 0;

fail:
        *cnt = 0;
        return err;
}

static int ramoops_init_prz(const char *name,
                            struct device *dev, struct ramoops_context *cxt,
                            struct persistent_ram_zone **prz,
                            phys_addr_t *paddr, size_t sz, u32 sig)
{
        if (!sz)
                return 0;

        if (*paddr + sz - cxt->phys_addr > cxt->size) {
                dev_err(dev, "no room for %s mem region (0x%zx@0x%llx) in (0x%lx@0x%llx)\n",
                        name, sz, (unsigned long long)*paddr,
                        cxt->size, (unsigned long long)cxt->phys_addr);
                return -ENOMEM;
        }

        *prz = persistent_ram_new(*paddr, sz, sig, &cxt->ecc_info,
                                  cxt->memtype, 0);
        if (IS_ERR(*prz)) {
                int err = PTR_ERR(*prz);

                dev_err(dev, "failed to request %s mem region (0x%zx@0x%llx): %d\n",
                        name, sz, (unsigned long long)*paddr, err);
                return err;
        }

        persistent_ram_zap(*prz);

        *paddr += sz;

        return 0;
}

static int ramoops_parse_dt_size(struct platform_device *pdev,
                                 const char *propname, u32 *value)
{
        u32 val32 = 0;
        int ret;

        ret = of_property_read_u32(pdev->dev.of_node, propname, &val32);
        if (ret < 0 && ret != -EINVAL) {
                dev_err(&pdev->dev, "failed to parse property %s: %d\n",
                        propname, ret);
                return ret;
        }

        if (val32 > INT_MAX) {
                dev_err(&pdev->dev, "%s %u > INT_MAX\n", propname, val32);
                return -EOVERFLOW;
        }

        *value = val32;
        return 0;
}

static int ramoops_parse_dt(struct platform_device *pdev,
                            struct ramoops_platform_data *pdata)
{
        struct device_node *of_node = pdev->dev.of_node;
        struct resource *res;
        u32 value;
        int ret;

        dev_dbg(&pdev->dev, "using Device Tree\n");

        res = platform_get_resource(pdev, IORESOURCE_MEM, 0);
        if (!res) {
                dev_err(&pdev->dev,
                        "failed to locate DT /reserved-memory resource\n");
                return -EINVAL;
        }

        pdata->mem_size = resource_size(res);
        pdata->mem_address = res->start;
        pdata->mem_type = of_property_read_bool(of_node, "unbuffered");
        pdata->dump_oops = !of_property_read_bool(of_node, "no-dump-oops");

#define parse_size(name, field) {                                       \
                ret = ramoops_parse_dt_size(pdev, name, &value);        \
                if (ret < 0)                                            \
                        return ret;                                     \
                field = value;                                          \
        }

        parse_size("record-size", pdata->record_size);
        parse_size("console-size", pdata->console_size);
        parse_size("ftrace-size", pdata->ftrace_size);
        parse_size("pmsg-size", pdata->pmsg_size);
        parse_size("ecc-size", pdata->ecc_info.ecc_size);
        parse_size("flags", pdata->flags);

#undef parse_size

        return 0;
}

static int ramoops_probe(struct platform_device *pdev)
{
        struct device *dev = &pdev->dev;
        struct ramoops_platform_data *pdata = dev->platform_data;
        struct ramoops_context *cxt = &oops_cxt;
        size_t dump_mem_sz;
        phys_addr_t paddr;
        int err = -EINVAL;

        if (dev_of_node(dev) && !pdata) {
                pdata = devm_kzalloc(&pdev->dev, sizeof(*pdata), GFP_KERNEL);
                if (!pdata) {
                        pr_err("cannot allocate platform data buffer\n");
                        err = -ENOMEM;
                        goto fail_out;
                }

                err = ramoops_parse_dt(pdev, pdata);
                if (err < 0)
                        goto fail_out;
        }

        /*
         * Only a single ramoops area allowed at a time, so fail extra
         * probes.
         */
        if (cxt->max_dump_cnt) {
                pr_err("already initialized\n");
                goto fail_out;
        }

        /* Make sure we didn't get bogus platform data pointer. */
        if (!pdata) {
                pr_err("NULL platform data\n");
                goto fail_out;
        }

        if (!pdata->mem_size || (!pdata->record_size && !pdata->console_size &&
                        !pdata->ftrace_size && !pdata->pmsg_size)) {
                pr_err("The memory size and the record/console size must be "
                        "non-zero\n");
                goto fail_out;
        }

        if (pdata->record_size && !is_power_of_2(pdata->record_size))
                pdata->record_size = rounddown_pow_of_two(pdata->record_size);
        if (pdata->console_size && !is_power_of_2(pdata->console_size))
                pdata->console_size = rounddown_pow_of_two(pdata->console_size);
        if (pdata->ftrace_size && !is_power_of_2(pdata->ftrace_size))
                pdata->ftrace_size = rounddown_pow_of_two(pdata->ftrace_size);
        if (pdata->pmsg_size && !is_power_of_2(pdata->pmsg_size))
                pdata->pmsg_size = rounddown_pow_of_two(pdata->pmsg_size);

        cxt->size = pdata->mem_size;
        cxt->phys_addr = pdata->mem_address;
        cxt->memtype = pdata->mem_type;
        cxt->record_size = pdata->record_size;
        cxt->console_size = pdata->console_size;
        cxt->ftrace_size = pdata->ftrace_size;
        cxt->pmsg_size = pdata->pmsg_size;
        cxt->dump_oops = pdata->dump_oops;
        cxt->flags = pdata->flags;
        cxt->ecc_info = pdata->ecc_info;

        paddr = cxt->phys_addr;

        dump_mem_sz = cxt->size - cxt->console_size - cxt->ftrace_size
                        - cxt->pmsg_size;
        err = ramoops_init_przs("dump", dev, cxt, &cxt->dprzs, &paddr,
                                dump_mem_sz, cxt->record_size,
                                &cxt->max_dump_cnt, 0, 0);
        if (err)
                goto fail_out;

        err = ramoops_init_prz("console", dev, cxt, &cxt->cprz, &paddr,
                               cxt->console_size, 0);
        if (err)
                goto fail_init_cprz;

        cxt->max_ftrace_cnt = (cxt->flags & RAMOOPS_FLAG_FTRACE_PER_CPU)
                                ? nr_cpu_ids
                                : 1;
        err = ramoops_init_przs("ftrace", dev, cxt, &cxt->fprzs, &paddr,
                                cxt->ftrace_size, -1,
                                &cxt->max_ftrace_cnt, LINUX_VERSION_CODE,
                                (cxt->flags & RAMOOPS_FLAG_FTRACE_PER_CPU)
                                        ? PRZ_FLAG_NO_LOCK : 0);
        if (err)
                goto fail_init_fprz;

        err = ramoops_init_prz("pmsg", dev, cxt, &cxt->mprz, &paddr,
                                cxt->pmsg_size, 0);
        if (err)
                goto fail_init_mprz;

        cxt->pstore.data = cxt;
        /*
         * Console can handle any buffer size, so prefer LOG_LINE_MAX. If we
         * have to handle dumps, we must have at least record_size buffer. And
         * for ftrace, bufsize is irrelevant (if bufsize is 0, buf will be
         * ZERO_SIZE_PTR).
         */
        if (cxt->console_size)
                cxt->pstore.bufsize = 1024; /* LOG_LINE_MAX */
        cxt->pstore.bufsize = max(cxt->record_size, cxt->pstore.bufsize);
        cxt->pstore.buf = kmalloc(cxt->pstore.bufsize, GFP_KERNEL);
        if (!cxt->pstore.buf) {
                pr_err("cannot allocate pstore buffer\n");
                err = -ENOMEM;
                goto fail_clear;
        }
        spin_lock_init(&cxt->pstore.buf_lock);

        cxt->pstore.flags = PSTORE_FLAGS_DMESG;
        if (cxt->console_size)
                cxt->pstore.flags |= PSTORE_FLAGS_CONSOLE;
        if (cxt->ftrace_size)
                cxt->pstore.flags |= PSTORE_FLAGS_FTRACE;
        if (cxt->pmsg_size)
                cxt->pstore.flags |= PSTORE_FLAGS_PMSG;

        err = pstore_register(&cxt->pstore);
        if (err) {
                pr_err("registering with pstore failed\n");
                goto fail_buf;
        }

        /*
         * Update the module parameter variables as well so they are visible
         * through /sys/module/ramoops/parameters/
         */
        mem_size = pdata->mem_size;
        mem_address = pdata->mem_address;
        record_size = pdata->record_size;
        dump_oops = pdata->dump_oops;
        ramoops_console_size = pdata->console_size;
        ramoops_pmsg_size = pdata->pmsg_size;
        ramoops_ftrace_size = pdata->ftrace_size;

        pr_info("attached 0x%lx@0x%llx, ecc: %d/%d\n",
                cxt->size, (unsigned long long)cxt->phys_addr,
                cxt->ecc_info.ecc_size, cxt->ecc_info.block_size);

        return 0;

fail_buf:
        kfree(cxt->pstore.buf);
fail_clear:
        cxt->pstore.bufsize = 0;
        persistent_ram_free(cxt->mprz);
fail_init_mprz:
fail_init_fprz:
        persistent_ram_free(cxt->cprz);
fail_init_cprz:
        ramoops_free_przs(cxt);
fail_out:
        return err;
}

static int ramoops_remove(struct platform_device *pdev)
{
        struct ramoops_context *cxt = &oops_cxt;

        pstore_unregister(&cxt->pstore);

        kfree(cxt->pstore.buf);
        cxt->pstore.bufsize = 0;

        persistent_ram_free(cxt->mprz);
        persistent_ram_free(cxt->cprz);
        ramoops_free_przs(cxt);

        return 0;
}

static const struct of_device_id dt_match[] = {
        { .compatible = "ramoops" },
        {}
};

static struct platform_driver ramoops_driver = {
        .probe          = ramoops_probe,
        .remove         = ramoops_remove,
        .driver         = {
                .name           = "ramoops",
                .of_match_table = dt_match,
        },
};

static void ramoops_register_dummy(void)
{
        if (!mem_size)
                return;

        pr_info("using module parameters\n");

        dummy_data = kzalloc(sizeof(*dummy_data), GFP_KERNEL);
        if (!dummy_data) {
                pr_info("could not allocate pdata\n");
                return;
        }

        dummy_data->mem_size = mem_size;
        dummy_data->mem_address = mem_address;
        dummy_data->mem_type = mem_type;
        dummy_data->record_size = record_size;
        dummy_data->console_size = ramoops_console_size;
        dummy_data->ftrace_size = ramoops_ftrace_size;
        dummy_data->pmsg_size = ramoops_pmsg_size;
        dummy_data->dump_oops = dump_oops;
        dummy_data->flags = RAMOOPS_FLAG_FTRACE_PER_CPU;

        /*
         * For backwards compatibility ramoops.ecc=1 means 16 bytes ECC
         * (using 1 byte for ECC isn't much of use anyway).
         */
        dummy_data->ecc_info.ecc_size = ramoops_ecc == 1 ? 16 : ramoops_ecc;

        dummy = platform_device_register_data(NULL, "ramoops", -1,
                        dummy_data, sizeof(struct ramoops_platform_data));
        if (IS_ERR(dummy)) {
                pr_info("could not create platform device: %ld\n",
                        PTR_ERR(dummy));
        }
}

static int __init ramoops_init(void)
{
        ramoops_register_dummy();
        return platform_driver_register(&ramoops_driver);
}
postcore_initcall(ramoops_init);

static void __exit ramoops_exit(void)
{
        platform_driver_unregister(&ramoops_driver);
        platform_device_unregister(dummy);
        kfree(dummy_data);
}
module_exit(ramoops_exit);

MODULE_LICENSE("GPL");
MODULE_AUTHOR("Marco Stornelli <marco.stornelli@gmail.com>");
MODULE_DESCRIPTION("RAM Oops/Panic logger/driver");