Merge pull request wolfSSL#392 from danielinux/fix-image-type-field

wolfBoot_get_blob_type: fix return type
philljj · Nov 17, 2023 · 83aa1d8 · 83aa1d8
2 parents c910b7f + 23894d3
commit 83aa1d8
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/include/wolfboot/wolfboot.h b/include/wolfboot/wolfboot.h
@@ -271,7 +271,7 @@ void wolfBoot_update_trigger(void);
 void wolfBoot_success(void);
 uint32_t wolfBoot_image_size(uint8_t *image);
 uint32_t wolfBoot_get_blob_version(uint8_t *blob);
-uint32_t wolfBoot_get_blob_type(uint8_t *blob);
+uint16_t wolfBoot_get_blob_type(uint8_t *blob);
 uint32_t wolfBoot_get_blob_diffbase_version(uint8_t *blob);
 
 /* Get partition ID from manifest header */

diff --git a/src/libwolfboot.c b/src/libwolfboot.c
@@ -1075,9 +1075,9 @@ uint32_t wolfBoot_get_blob_version(uint8_t *blob)
  *
  * @return The type of the blob, or 0 if the blob is invalid.
  */
-uint32_t wolfBoot_get_blob_type(uint8_t *blob)
+uint16_t wolfBoot_get_blob_type(uint8_t *blob)
 {
-    uint32_t *volatile type_field = NULL;
+    uint16_t *volatile type_field = NULL;
     uint32_t *magic = NULL;
     uint8_t *img_bin = blob;
 #if defined(EXT_ENCRYPTED) && defined(MMU)

diff --git a/tools/unit-tests/unit-image.c b/tools/unit-tests/unit-image.c
@@ -236,7 +236,7 @@ static uint16_t _find_header(uint8_t *haystack, uint16_t type, uint8_t **ptr)
     return 0;
 }
 
-uint32_t wolfBoot_get_blob_type(uint8_t *addr)
+uint16_t wolfBoot_get_blob_type(uint8_t *addr)
 {
     return HDR_IMG_TYPE_APP;
 }