// SPDX-License-Identifier: GPL-2.0-or-later
/*
 * Load Analog Devices SigmaStudio firmware files
 *
 * Copyright 2009-2011 Analog Devices Inc.
 */

#include <linux/export.h>
#include <linux/i2c.h>
#include <linux/module.h>
#include <linux/slab.h>
#include <linux/unaligned.h>

#include "sigmadsp.h"

static int sigmadsp_write_i2c(void *control_data,
        unsigned int addr, const uint8_t data[], size_t len)
{
        uint8_t *buf;
        int ret;

        buf = kzalloc(2 + len, GFP_KERNEL | GFP_DMA);
        if (!buf)
                return -ENOMEM;

        put_unaligned_be16(addr, buf);
        memcpy(buf + 2, data, len);

        ret = i2c_master_send(control_data, buf, len + 2);

        kfree(buf);

        if (ret < 0)
                return ret;

        return 0;
}

static int sigmadsp_read_i2c(void *control_data,
        unsigned int addr, uint8_t data[], size_t len)
{
        struct i2c_client *client = control_data;
        struct i2c_msg msgs[2];
        uint8_t buf[2];
        int ret;

        put_unaligned_be16(addr, buf);

        msgs[0].addr = client->addr;
        msgs[0].len = sizeof(buf);
        msgs[0].buf = buf;
        msgs[0].flags = 0;

        msgs[1].addr = client->addr;
        msgs[1].len = len;
        msgs[1].buf = data;
        msgs[1].flags = I2C_M_RD;

        ret = i2c_transfer(client->adapter, msgs, ARRAY_SIZE(msgs));
        if (ret < 0)
                return ret;
        else if (ret != ARRAY_SIZE(msgs))
                return -EIO;
        return 0;
}

/**
 * devm_sigmadsp_init_i2c() - Initialize SigmaDSP instance
 * @client: The parent I2C device
 * @ops: The sigmadsp_ops to use for this instance
 * @firmware_name: Name of the firmware file to load
 *
 * Allocates a SigmaDSP instance and loads the specified firmware file.
 *
 * Returns a pointer to a struct sigmadsp on success, or a PTR_ERR() on error.
 */
struct sigmadsp *devm_sigmadsp_init_i2c(struct i2c_client *client,
        const struct sigmadsp_ops *ops, const char *firmware_name)
{
        struct sigmadsp *sigmadsp;

        sigmadsp = devm_sigmadsp_init(&client->dev, ops, firmware_name);
        if (IS_ERR(sigmadsp))
                return sigmadsp;

        sigmadsp->control_data = client;
        sigmadsp->write = sigmadsp_write_i2c;
        sigmadsp->read = sigmadsp_read_i2c;

        return sigmadsp;
}
EXPORT_SYMBOL_GPL(devm_sigmadsp_init_i2c);

MODULE_AUTHOR("Lars-Peter Clausen <lars@metafoo.de>");
MODULE_DESCRIPTION("SigmaDSP I2C firmware loader");
MODULE_LICENSE("GPL");