// SPDX-License-Identifier: GPL-2.0
/*
 * Copyright 2023 Google LLC.
 */

#include <linux/device.h>
#include <linux/spmi.h>

static void devm_spmi_controller_release(struct device *parent, void *res)
{
 spmi_controller_put(*(struct spmi_controller **)res);
}

struct spmi_controller *devm_spmi_controller_alloc(struct device *parent, size_t size)
{
 struct spmi_controller **ptr, *ctrl;

 ptr = devres_alloc(devm_spmi_controller_release, sizeof(*ptr), GFP_KERNEL);
 if (!ptr)
  return ERR_PTR(-ENOMEM);

 ctrl = spmi_controller_alloc(parent, size);
 if (IS_ERR(ctrl)) {
  devres_free(ptr);
  return ctrl;
 }

 *ptr = ctrl;
 devres_add(parent, ptr);

 return ctrl;
}
EXPORT_SYMBOL_GPL(devm_spmi_controller_alloc);

static void devm_spmi_controller_remove(struct device *parent, void *res)
{
 spmi_controller_remove(*(struct spmi_controller **)res);
}

int devm_spmi_controller_add(struct device *parent, struct spmi_controller *ctrl)
{
 struct spmi_controller **ptr;
 int ret;

 ptr = devres_alloc(devm_spmi_controller_remove, sizeof(*ptr), GFP_KERNEL);
 if (!ptr)
  return -ENOMEM;

 ret = spmi_controller_add(ctrl);
 if (ret) {
  devres_free(ptr);
  return ret;
 }

 *ptr = ctrl;
 devres_add(parent, ptr);

 return 0;

}
EXPORT_SYMBOL_GPL(devm_spmi_controller_add);

MODULE_LICENSE("GPL");
MODULE_DESCRIPTION("SPMI devres helpers");