FWU: Add checking the state of the running image

The Firmware Update partition reads the image_ok flag to check
whether the running image has been confirmed. The running image
state is set according to this flag in psa_fwu_query().

Change-Id: I9cf58b7d6f3b87a50e0ac6f926e30170bb01b8f1
Signed-off-by: Sherry Zhang <sherry.zhang2@arm.com>
diff --git a/interface/include/psa/update.h b/interface/include/psa/update.h
index 65d7d2d..d7de5ac 100644
--- a/interface/include/psa/update.h
+++ b/interface/include/psa/update.h
@@ -207,6 +207,8 @@
 /**
  * \brief Indicates to the implementation that the upgrade was successful.
  *
+ * \param[in] image_id         The image_id of the image to query
+ *
  * \return A status indicating the success/failure of the operation
  *
  * \retval PSA_SUCCESS                  The image and its dependencies have
@@ -217,7 +219,7 @@
  * \retval PSA_ERROR_NOT_PERMITTED      The caller is not permitted to make
  *                                      this call
  */
-psa_status_t psa_fwu_accept(void);
+psa_status_t psa_fwu_accept(psa_image_id_t image_id);
 
 /**
  * \brief Stores a manifest object and associates it with a particular image ID.