Revision:
0:25fa8795676b
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/simple-mbed-cloud-client/mbed-cloud-client/update-client-hub/source/update_client_hub.c	Sun Apr 18 15:20:23 2021 +0000
@@ -0,0 +1,489 @@
+// ----------------------------------------------------------------------------
+// Copyright 2016-2017 ARM Ltd.
+//
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+// ----------------------------------------------------------------------------
+
+#include "update-client-hub/update_client_hub.h"
+
+#include "update-client-common/arm_uc_common.h"
+#include "update-client-control-center/arm_uc_control_center.h"
+#include "update-client-control-center/arm_uc_pre_shared_key.h"
+#include "update-client-control-center/arm_uc_certificate.h"
+#include "update-client-source-manager/arm_uc_source_manager.h"
+#include "update-client-firmware-manager/arm_uc_firmware_manager.h"
+#include "update-client-manifest-manager/update-client-manifest-manager.h"
+
+#include "update_client_hub_state_machine.h"
+#include "update_client_hub_event_handlers.h"
+#include "update_client_hub_error_handler.h"
+
+#include "pal4life-device-identity/pal_device_identity.h"
+
+#define HANDLE_INIT_ERROR(retval, msg, ...)\
+    if (retval.error != ERR_NONE)\
+    {\
+        ARM_UC_HUB_setState(ARM_UC_HUB_STATE_UNINITIALIZED);\
+        UC_HUB_ERR_MSG(msg " error code %s", ##__VA_ARGS__, ARM_UC_err2Str(retval));\
+        return retval;\
+    }
+
+static const ARM_UPDATE_SOURCE **arm_uc_sources = NULL;
+static uint8_t arm_uc_sources_size = 0;
+extern arm_uc_mmContext_t *pManifestManagerContext;
+
+/**
+ * @brief Handle any errors posted by the scheduler.
+ * @details This explicitly runs *not* in interrupt context, the scheduler has a dedicated
+ *            callback structure to ensure it can post at least this event.
+ *          ARM_UC_HUB_ErrorHandler() will invoke the HUB callback that was set up.
+ *          It is up to the external application to go about inducing a reset etc,
+ *            if that is what it decides. Note that the HUB is no longer operable
+ *            and the app should probably Uninitialize it and report an error.
+ *            However, the HUB will attempt some cleanup after it returns.
+ * @param an_event the type of the event causing the error callback.
+ *        The only possible errors from the scheduler are currently:
+ *            ARM_UC_EQ_ERR_POOL_EXHAUSTED
+ *            ARM_UC_EQ_ERR_FAILED_TAKE
+ *        These are passed on to the Hub error handler as an internal error,
+ *          and the hub state is now considered unknown from this perspective.
+ *          (An internal error is considered fatal by the hub.)
+ */
+void UC_HUB_scheduler_error_handler(uint32_t an_event)
+{
+    UC_HUB_ERR_MSG("scheduler error: %" PRIu32, an_event);
+    ARM_UC_HUB_ErrorHandler(HUB_ERR_INTERNAL_ERROR, ARM_UC_HUB_getState());
+}
+
+/**
+ * @brief Call initialiser of all components of the client.
+ *        finish asynchronously, will invoke callback when initialization is done.
+ * @param init_cb the callback to be invoked at the end of initialization.
+ */
+arm_uc_error_t ARM_UC_HUB_Initialize(void (*init_cb)(uintptr_t))
+{
+    arm_uc_error_t retval;
+
+    if (ARM_UC_HUB_getState() != ARM_UC_HUB_STATE_UNINITIALIZED) {
+        UC_HUB_ERR_MSG("Already Initialized/Initializing");
+        return (arm_uc_error_t) { ERR_INVALID_STATE };
+    }
+    ARM_UC_HUB_setState(ARM_UC_HUB_STATE_INITIALIZING);
+
+    ARM_UC_SchedulerInit();
+    ARM_UC_HUB_setInitializationCallback(init_cb);
+    ARM_UC_SetSchedulerErrorHandler(UC_HUB_scheduler_error_handler);
+
+    /* Register event handler with Control Center. */
+    retval = ARM_UC_ControlCenter_Initialize(ARM_UC_HUB_ControlCenterEventHandler);
+    HANDLE_INIT_ERROR(retval, "Control Center init failed")
+
+    /* Register event handler with Firmware Manager */
+    retval = ARM_UC_FirmwareManager.Initialize(ARM_UC_HUB_FirmwareManagerEventHandler);
+    HANDLE_INIT_ERROR(retval, "Firmware Manager init failed")
+
+    /* Register event handler with Source Manager */
+    retval = ARM_UC_SourceManager.Initialize(ARM_UC_HUB_SourceManagerEventHandler);
+    HANDLE_INIT_ERROR(retval, "Source Manager init failed")
+
+    for (uint8_t index = 0; index < arm_uc_sources_size; index++) {
+        ARM_UC_SourceManager.AddSource(arm_uc_sources[index]);
+    }
+
+    /* Register event handler and add config store implementation to manifest
+       manager.
+    */
+    retval = ARM_UC_mmInit(&pManifestManagerContext,
+                           ARM_UC_HUB_ManifestManagerEventHandler,
+                           NULL);
+    HANDLE_INIT_ERROR(retval, "Manifest manager init failed")
+
+    /* add hard coded certificates to the manifest manager */
+    // retval = ARM_UC_mmStoreCertificate(CA_PATH, cert, CERT_SIZE);
+    // if ((retval.error != ERR_NONE) && (retval.code != MFST_ERR_PENDING))
+    // {
+    //     HANDLE_INIT_ERROR(retval, "Manifest manager StoreCertificate failed")
+    // }
+
+    return (arm_uc_error_t) { ERR_NONE };
+}
+
+/**
+ * @brief Process events in the event queue.
+ */
+arm_uc_error_t ARM_UC_HUB_ProcessEvents()
+{
+    ARM_UC_ProcessQueue();
+
+    return (arm_uc_error_t) { ERR_NONE };
+}
+
+/**
+ * @brief Register callback function for when callbacks are added to an empty queue.
+ */
+arm_uc_error_t ARM_UC_HUB_AddNotificationHandler(void (*handler)(void))
+{
+    ARM_UC_AddNotificationHandler(handler);
+
+    return (arm_uc_error_t) { ERR_NONE };
+}
+
+/**
+ * @brief Add source to the Update Client.
+ */
+arm_uc_error_t ARM_UC_HUB_SetSources(const ARM_UPDATE_SOURCE *sources[],
+                                     uint8_t size)
+{
+    arm_uc_sources = sources;
+    arm_uc_sources_size = size;
+
+    return (arm_uc_error_t) { ERR_NONE };
+}
+
+/**
+ * Set PAAL Update implementation
+ */
+arm_uc_error_t ARM_UC_HUB_SetStorage(const ARM_UC_PAAL_UPDATE *implementation)
+{
+    return ARM_UCP_SetPAALUpdate(implementation);
+}
+
+/**
+ * @brief Add monitor to the control center.
+ */
+arm_uc_error_t ARM_UC_HUB_AddMonitor(const ARM_UPDATE_MONITOR *monitor)
+{
+    return ARM_UC_ControlCenter_AddMonitor(monitor);
+}
+
+/**
+ * @brief Temporary error reporting function.
+ */
+void ARM_UC_HUB_AddErrorCallback(void (*callback)(int32_t error))
+{
+    ARM_UC_HUB_AddErrorCallbackInternal(callback);
+}
+
+/**
+ * @brief Authorize request.
+ */
+arm_uc_error_t ARM_UC_Authorize(arm_uc_request_t request)
+{
+    return ARM_UC_ControlCenter_Authorize(request);
+}
+
+/**
+ * @brief Set callback for receiving download progress.
+ * @details User application call for setting callback handler.
+ *          The callback function takes the progreess in percent as argument.
+ *
+ * @param callback Function pointer to the progress function.
+ * @return Error code.
+ */
+arm_uc_error_t ARM_UC_SetProgressHandler(void (*callback)(uint32_t progress, uint32_t total))
+{
+    return ARM_UC_ControlCenter_SetProgressHandler(callback);
+}
+
+/**
+ * @brief Set callback function for authorizing requests.
+ * @details User application call for setting callback handler.
+ *          The callback function takes an enum request and an authorization
+ *          function pointer. To authorize the given request, the caller
+ *          invokes the authorization function.
+ *
+ * @param callback Function pointer to the authorization function.
+ * @return Error code.
+ */
+arm_uc_error_t ARM_UC_SetAuthorizeHandler(void (*callback)(int32_t))
+{
+    return ARM_UC_ControlCenter_SetAuthorityHandler(callback);
+}
+
+/**
+ * @brief Override update authorization handler.
+ * @details Force download and update to progress regardless of authorization
+ *          handler. This function is used for unblocking an update in a buggy
+ *          application.
+ */
+void ARM_UC_OverrideAuthorization(void)
+{
+    ARM_UC_ControlCenter_OverrideAuthorization();
+}
+
+#if defined(ARM_UC_FEATURE_MANIFEST_PUBKEY) && (ARM_UC_FEATURE_MANIFEST_PUBKEY == 1)
+/**
+ * @brief Add certificate.
+ * @details [long description]
+ *
+ * @param certificate Pointer to certiface being added.
+ * @param certificate_length Certificate length.
+ * @param fingerprint Pointer to the fingerprint of the certificate being added.
+ * @param fingerprint_length Fingerprint length.
+ * @return Error code.
+ */
+arm_uc_error_t ARM_UC_AddCertificate(const uint8_t *certificate,
+                                     uint16_t certificate_length,
+                                     const uint8_t *fingerprint,
+                                     uint16_t fingerprint_length,
+                                     void (*callback)(arm_uc_error_t, const arm_uc_buffer_t *))
+{
+    return ARM_UC_Certificate_Add(certificate,
+                                  certificate_length,
+                                  fingerprint,
+                                  fingerprint_length,
+                                  callback);
+}
+#endif /* ARM_UC_FEATURE_MANIFEST_PUBKEY */
+
+#if defined(ARM_UC_FEATURE_MANIFEST_PSK) && (ARM_UC_FEATURE_MANIFEST_PSK == 1)
+/**
+ * @brief Set pointer to pre-shared-key with the given size.
+ *
+ * @param key Pointer to pre-shared-key.
+ * @param bits Key size in bits.
+ *
+ * @return Error code.
+ */
+arm_uc_error_t ARM_UC_AddPreSharedKey(const uint8_t *key, uint16_t bits)
+{
+    return ARM_UC_PreSharedKey_SetSecret(key, bits);
+}
+#endif
+
+/**
+ * @brief Function for setting the vendor ID.
+ * @details The ID is copied to a 16 byte struct. Any data after the first
+ *          16 bytes will be ignored.
+ * @param id Pointer to ID.
+ * @param length Length of ID.
+ * @return Error code.
+ */
+arm_uc_error_t ARM_UC_SetVendorId(const uint8_t *id, uint8_t length)
+{
+    arm_uc_guid_t uuid = { 0 };
+
+    if (id) {
+        for (uint8_t index = 0;
+                (index < sizeof(arm_uc_guid_t) && (index < length));
+                index++) {
+            ((uint8_t *) uuid)[index] = id[index];
+        }
+    }
+
+    return pal_setVendorGuid(&uuid);
+}
+
+/**
+ * @brief Function for setting the class ID.
+ * @details The ID is copied to a 16 byte struct. Any data after the first
+ *          16 bytes will be ignored.
+ * @param id Pointer to ID.
+ * @param length Length of ID.
+ * @return Error code.
+ */
+arm_uc_error_t ARM_UC_SetClassId(const uint8_t *id, uint8_t length)
+{
+    arm_uc_guid_t uuid = { 0 };
+
+    if (id) {
+        for (uint8_t index = 0;
+                (index < sizeof(arm_uc_guid_t) && (index < length));
+                index++) {
+            ((uint8_t *) uuid)[index] = id[index];
+        }
+    }
+
+    return pal_setClassGuid(&uuid);
+}
+
+/**
+ * @brief Function for setting the device ID.
+ * @details The ID is copied to a 16 byte struct. Any data after the first
+ *          16 bytes will be ignored.
+ * @param id Pointer to ID.
+ * @param length Length of ID.
+ * @return Error code.
+ */
+arm_uc_error_t ARM_UC_SetDeviceId(const uint8_t *id, uint8_t length)
+{
+    arm_uc_guid_t uuid = { 0 };
+
+    if (id) {
+        for (uint8_t index = 0;
+                (index < sizeof(arm_uc_guid_t) && (index < length));
+                index++) {
+            ((uint8_t *) uuid)[index] = id[index];
+        }
+    }
+
+    return pal_setDeviceGuid(&uuid);
+}
+
+/**
+ * @brief Function for reporting the vendor ID.
+ * @details 16 bytes are copied into the supplied buffer.
+ * @param id Pointer to storage for ID. MUST be at least 16 bytes long.
+ * @param id_max the size of the ID buffer
+ * @param id_size pointer to a variable to receive the size of the ID
+ *                written into the buffer (always 16).
+ * @return Error code.
+ */
+arm_uc_error_t ARM_UC_GetVendorId(uint8_t *id,
+                                  const size_t id_max,
+                                  size_t *id_size)
+{
+    arm_uc_guid_t guid = {0};
+    arm_uc_error_t err = {ERR_NONE};
+    if (id_max < sizeof(arm_uc_guid_t)) {
+        err.code = ARM_UC_DI_ERR_SIZE;
+    }
+    if (err.error == ERR_NONE) {
+        err = pal_getVendorGuid(&guid);
+    }
+    if (err.error == ERR_NONE) {
+        memcpy(id, guid, sizeof(arm_uc_guid_t));
+        if (id_size != NULL) {
+            *id_size = sizeof(arm_uc_guid_t);
+        }
+    }
+    return err;
+}
+
+/**
+ * @brief Function for reporting the class ID.
+ * @details 16 bytes are copied into the supplied buffer.
+ * @param id Pointer to storage for ID. MUST be at least 16 bytes long.
+ * @param id_max the size of the ID buffer
+ * @param id_size pointer to a variable to receive the size of the ID
+ *                written into the buffer (always 16).
+ * @return Error code.
+ */
+arm_uc_error_t ARM_UC_GetClassId(uint8_t *id,
+                                 const size_t id_max,
+                                 size_t *id_size)
+{
+    arm_uc_guid_t guid = {0};
+    arm_uc_error_t err = {ERR_NONE};
+    if (id_max < sizeof(arm_uc_guid_t)) {
+        err.code = ARM_UC_DI_ERR_SIZE;
+    }
+    if (err.error == ERR_NONE) {
+        err = pal_getClassGuid(&guid);
+    }
+    if (err.error == ERR_NONE) {
+        memcpy(id, guid, sizeof(arm_uc_guid_t));
+        if (id_size != NULL) {
+            *id_size = sizeof(arm_uc_guid_t);
+        }
+    }
+    return err;
+}
+
+/**
+ * @brief Function for reporting the device ID.
+ * @details 16 bytes are copied into the supplied buffer.
+ * @param id Pointer to storage for ID. MUST be at least 16 bytes long.
+ * @param id_max the size of the ID buffer
+ * @param id_size pointer to a variable to receive the size of the ID
+ *                written into the buffer (always 16).
+ * @return Error code.
+ */
+arm_uc_error_t ARM_UC_GetDeviceId(uint8_t *id,
+                                  const size_t id_max,
+                                  size_t *id_size)
+{
+    arm_uc_guid_t guid = {0};
+    arm_uc_error_t err = {ERR_NONE};
+    if (id_max < sizeof(arm_uc_guid_t)) {
+        err.code = ARM_UC_DI_ERR_SIZE;
+    }
+    if (err.error == ERR_NONE) {
+        err = pal_getDeviceGuid(&guid);
+    }
+    if (err.error == ERR_NONE) {
+        memcpy(id, guid, sizeof(arm_uc_guid_t));
+        if (id_size != NULL) {
+            *id_size = sizeof(arm_uc_guid_t);
+        }
+    }
+    return err;
+}
+
+arm_uc_error_t ARM_UC_HUB_Uninitialize(void)
+{
+    if (ARM_UC_HUB_getState() <= ARM_UC_HUB_STATE_INITIALIZED) {
+        UC_HUB_ERR_MSG("Update Client not initialized");
+        return (arm_uc_error_t) { ERR_INVALID_STATE };
+    }
+
+    arm_uc_error_t err = ARM_UC_SourceManager.Uninitialize();
+    ARM_UC_HUB_setState(ARM_UC_HUB_STATE_UNINITIALIZED);
+    return err;
+}
+
+/**
+ * @brief Return the details of the active firmware.
+ * @param details Pointer to the firmware details structure.
+ * @return ARM_UC_HUB_ERR_NOT_AVAILABLE if the active firmware details
+ *         are not yet available, ERR_INVALID_PARAMETER if "details" is
+ *         NULL or ERR_NONE for success.
+ */
+arm_uc_error_t ARM_UC_API_GetActiveFirmwareDetails(arm_uc_firmware_details_t *details)
+{
+    arm_uc_error_t err = {ARM_UC_HUB_ERR_NOT_AVAILABLE};
+
+    if (details == NULL) {
+        err.code = ERR_INVALID_PARAMETER;
+    } else {
+        arm_uc_firmware_details_t *hub_details = ARM_UC_HUB_getActiveFirmwareDetails();
+        if (hub_details) {
+            memcpy(details, hub_details, sizeof(arm_uc_firmware_details_t));
+            err.code = ERR_NONE;
+        }
+    }
+    return err;
+}
+
+/**
+ * @brief Return whether or not the given state is a valid defined one.
+ */
+bool ARM_UC_IsValidState(arm_uc_update_state_t an_update_state)
+{
+    int val = (int)an_update_state;
+    bool is_valid = ((val >= (int)ARM_UC_UPDATE_STATE_FIRST)
+                     && (val <= (int)ARM_UC_UPDATE_STATE_LAST));
+    if (!is_valid) UC_ERROR_ERR_MSG("Invalid UC HUB reported state");
+    return is_valid;
+}
+
+/**
+ * @brief Return whether or not the given result is a valid defined one.
+ */
+bool ARM_UC_IsValidResult(arm_uc_update_result_t an_update_result)
+{
+    bool is_valid = ((an_update_result >= ARM_UC_UPDATE_RESULT_UPDATE_FIRST)
+                     && (an_update_result <= ARM_UC_UPDATE_RESULT_UPDATE_LAST))
+                    || ((an_update_result >= ARM_UC_UPDATE_RESULT_FETCHER_FIRST)
+                        && (an_update_result <= ARM_UC_UPDATE_RESULT_FETCHER_LAST))
+                    || ((an_update_result >= ARM_UC_UPDATE_RESULT_WRITER_FIRST)
+                        && (an_update_result <= ARM_UC_UPDATE_RESULT_WRITER_LAST))
+                    || ((an_update_result >= ARM_UC_UPDATE_RESULT_PROCESSOR_FIRST)
+                        && (an_update_result <= ARM_UC_UPDATE_RESULT_PROCESSOR_LAST))
+                    || ((an_update_result >= ARM_UC_UPDATE_RESULT_MANIFEST_FIRST)
+                        && (an_update_result <= ARM_UC_UPDATE_RESULT_MANIFEST_LAST));
+    if (!is_valid) UC_ERROR_ERR_MSG("Invalid UC HUB reported state");
+    return is_valid;
+}