--- /dev/null
+// SPDX-License-Identifier: GPL-2.0+
+/*
+ * Copyright (c), Vaisala Oyj
+ */
+
+#include <common.h>
+#include <dm.h>
+#include <dm/device_compat.h>
+#include <dm/devres.h>
+#include <exports.h>
+#include <reboot-mode/reboot-mode.h>
+
+DECLARE_GLOBAL_DATA_PTR;
+
+int dm_reboot_mode_update(struct udevice *dev)
+{
+ struct reboot_mode_ops *ops = reboot_mode_get_ops(dev);
+ u32 rebootmode;
+ int ret, i;
+
+ assert(ops);
+
+ if (!ops->get)
+ return -ENOSYS;
+
+ ret = ops->get(dev, &rebootmode);
+ if (ret < 0) {
+ dev_err(dev, "Failed to retrieve the reboot mode value\n");
+ return ret;
+ }
+
+ const struct reboot_mode_uclass_platdata *plat_data =
+ dev_get_uclass_plat(dev);
+
+ for (i = 0; i < plat_data->count; i++) {
+ if (plat_data->modes[i].mode_id == rebootmode) {
+ ret = env_set(plat_data->env_variable,
+ plat_data->modes[i].mode_name);
+ if (ret) {
+ dev_err(dev, "Failed to set env: %s\n",
+ plat_data->env_variable);
+ return ret;
+ }
+ }
+ }
+
+ if (ops->set) {
+ /* Clear the value */
+ rebootmode = 0;
+ ret = ops->set(dev, rebootmode);
+ if (ret) {
+ dev_err(dev, "Failed to clear the reboot mode\n");
+ return ret;
+ }
+ }
+
+ return 0;
+}
+
+int dm_reboot_mode_pre_probe(struct udevice *dev)
+{
+ struct reboot_mode_uclass_platdata *plat_data;
+
+ plat_data = dev_get_uclass_plat(dev);
+ if (!plat_data)
+ return -EINVAL;
+
+#if CONFIG_IS_ENABLED(OF_CONTROL)
+ const int node = dev_of_offset(dev);
+ const char *mode_prefix = "mode-";
+ const int mode_prefix_len = strlen(mode_prefix);
+ int property;
+ const u32 *propvalue;
+ const char *propname;
+
+ plat_data->env_variable = fdt_getprop(gd->fdt_blob,
+ node,
+ "u-boot,env-variable",
+ NULL);
+ if (!plat_data->env_variable)
+ plat_data->env_variable = "reboot-mode";
+
+ plat_data->count = 0;
+
+ fdt_for_each_property_offset(property, gd->fdt_blob, node) {
+ propvalue = fdt_getprop_by_offset(gd->fdt_blob,
+ property, &propname, NULL);
+ if (!propvalue) {
+ dev_err(dev, "Could not get the value for property %s\n",
+ propname);
+ return -EINVAL;
+ }
+
+ if (!strncmp(propname, mode_prefix, mode_prefix_len))
+ plat_data->count++;
+ }
+
+ plat_data->modes = devm_kcalloc(dev, plat_data->count,
+ sizeof(struct reboot_mode_mode), 0);
+
+ struct reboot_mode_mode *next = plat_data->modes;
+
+ fdt_for_each_property_offset(property, gd->fdt_blob, node) {
+ propvalue = fdt_getprop_by_offset(gd->fdt_blob,
+ property, &propname, NULL);
+ if (!propvalue) {
+ dev_err(dev, "Could not get the value for property %s\n",
+ propname);
+ return -EINVAL;
+ }
+
+ if (!strncmp(propname, mode_prefix, mode_prefix_len)) {
+ next->mode_name = &propname[mode_prefix_len];
+ next->mode_id = fdt32_to_cpu(*propvalue);
+
+ next++;
+ }
+ }
+#else
+ if (!plat_data->env_variable)
+ plat_data->env_variable = "reboot-mode";
+
+#endif
+
+ return 0;
+}
+
+UCLASS_DRIVER(reboot_mode) = {
+ .name = "reboot-mode",
+ .id = UCLASS_REBOOT_MODE,
+ .pre_probe = dm_reboot_mode_pre_probe,
+ .per_device_plat_auto =
+ sizeof(struct reboot_mode_uclass_platdata),
+};
--- /dev/null
+/* SPDX-License-Identifier: GPL-2.0+ */
+/*
+ * Copyright (c), Vaisala Oyj
+ */
+
+#ifndef REBOOT_MODE_REBOOT_MODE_H__
+#define REBOOT_MODE_REBOOT_MODE_H__
+
+#include <asm/types.h>
+#include <dm/device.h>
+
+struct reboot_mode_mode {
+ const char *mode_name;
+ u32 mode_id;
+};
+
+struct reboot_mode_uclass_platdata {
+ struct reboot_mode_mode *modes;
+ u8 count;
+ const char *env_variable;
+};
+
+struct reboot_mode_ops {
+ /**
+ * get() - get the current reboot mode value
+ *
+ * Returns the current value from the reboot mode backing store.
+ *
+ * @dev: Device to read from
+ * @rebootmode: Address to save the current reboot mode value
+ */
+ int (*get)(struct udevice *dev, u32 *rebootmode);
+
+ /**
+ * set() - set a reboot mode value
+ *
+ * Sets the value in the reboot mode backing store.
+ *
+ * @dev: Device to read from
+ * @rebootmode: New reboot mode value to store
+ */
+ int (*set)(struct udevice *dev, u32 rebootmode);
+};
+
+/* Access the operations for a reboot mode device */
+#define reboot_mode_get_ops(dev) ((struct reboot_mode_ops *)(dev)->driver->ops)
+
+/**
+ * dm_reboot_mode_update() - Update the reboot mode env variable.
+ *
+ * @dev: Device to read from
+ * @return 0 if OK, -ve on error
+ */
+int dm_reboot_mode_update(struct udevice *dev);
+
+#endif /* REBOOT_MODE_REBOOT_MODE_H__ */