1 files changed, 58 insertions, 0 deletions

diff --git a/drivers/firmware/xilinx/zynqmp.c b/drivers/firmware/xilinx/zynqmp.c
index 765a2ca1b100..af5cffd3ac43 100644
--- a/drivers/firmware/xilinx/zynqmp.c
+++ b/drivers/firmware/xilinx/zynqmp.c
@@ -557,6 +557,61 @@ static int zynqmp_pm_set_suspend_mode(u32 mode)
         return zynqmp_pm_invoke_fn(PM_SET_SUSPEND_MODE, mode, 0, 0, 0, NULL);
 }
 
+/**
+ * zynqmp_pm_request_node() - Request a node with specific capabilities
+ * @node:               Node ID of the slave
+ * @capabilities:       Requested capabilities of the slave
+ * @qos:                Quality of service (not supported)
+ * @ack:                Flag to specify whether acknowledge is requested
+ *
+ * This function is used by master to request particular node from firmware.
+ * Every master must request node before using it.
+ *
+ * Return: Returns status, either success or error+reason
+ */
+static int zynqmp_pm_request_node(const u32 node, const u32 capabilities,
+                                  const u32 qos,
+                                  const enum zynqmp_pm_request_ack ack)
+{
+        return zynqmp_pm_invoke_fn(PM_REQUEST_NODE, node, capabilities,
+                                   qos, ack, NULL);
+}
+
+/**
+ * zynqmp_pm_release_node() - Release a node
+ * @node:       Node ID of the slave
+ *
+ * This function is used by master to inform firmware that master
+ * has released node. Once released, master must not use that node
+ * without re-request.
+ *
+ * Return: Returns status, either success or error+reason
+ */
+static int zynqmp_pm_release_node(const u32 node)
+{
+        return zynqmp_pm_invoke_fn(PM_RELEASE_NODE, node, 0, 0, 0, NULL);
+}
+
+/**
+ * zynqmp_pm_set_requirement() - PM call to set requirement for PM slaves
+ * @node:               Node ID of the slave
+ * @capabilities:       Requested capabilities of the slave
+ * @qos:                Quality of service (not supported)
+ * @ack:                Flag to specify whether acknowledge is requested
+ *
+ * This API function is to be used for slaves a PU already has requested
+ * to change its capabilities.
+ *
+ * Return: Returns status, either success or error+reason
+ */
+static int zynqmp_pm_set_requirement(const u32 node, const u32 capabilities,
+                                     const u32 qos,
+                                     const enum zynqmp_pm_request_ack ack)
+{
+        return zynqmp_pm_invoke_fn(PM_SET_REQUIREMENT, node, capabilities,
+                                   qos, ack, NULL);
+}
+
 static const struct zynqmp_eemi_ops eemi_ops = {
         .get_api_version = zynqmp_pm_get_api_version,
         .get_chipid = zynqmp_pm_get_chipid,
@@ -575,6 +630,9 @@ static const struct zynqmp_eemi_ops eemi_ops = {
         .reset_get_status = zynqmp_pm_reset_get_status,
         .init_finalize = zynqmp_pm_init_finalize,
         .set_suspend_mode = zynqmp_pm_set_suspend_mode,
+        .request_node = zynqmp_pm_request_node,
+        .release_node = zynqmp_pm_release_node,
+        .set_requirement = zynqmp_pm_set_requirement,
 };
 
 /**