Changed set|get_sector_flag argument to uint16_t

On some platforms with very small sector size this index could overflow

Author: danielinux

include/image.h

@@ -66,9 +66,9 @@ int wolfBoot_open_image(struct wolfBoot_image *img, uint8_t part);
 int wolfBoot_verify_integrity(struct wolfBoot_image *img);
 int wolfBoot_verify_authenticity(struct wolfBoot_image *img);
 int wolfBoot_set_partition_state(uint8_t part, uint8_t newst);
-int wolfBoot_set_sector_flag(uint8_t part, uint8_t sector, uint8_t newflag);
+int wolfBoot_set_sector_flag(uint8_t part, uint16_t sector, uint8_t newflag);
 int wolfBoot_get_partition_state(uint8_t part, uint8_t *st);
-int wolfBoot_get_sector_flag(uint8_t part, uint8_t sector, uint8_t *flag);
+int wolfBoot_get_sector_flag(uint8_t part, uint16_t sector, uint8_t *flag);
 
 uint8_t* wolfBoot_peek_image(struct wolfBoot_image *img, uint32_t offset, uint32_t* sz);
 

src/libwolfboot.c

@@ -207,7 +207,7 @@ int RAMFUNCTION wolfBoot_set_partition_state(uint8_t part, uint8_t newst)
     return 0;
 }
 
-int RAMFUNCTION wolfBoot_set_sector_flag(uint8_t part, uint8_t sector, uint8_t newflag)
+int RAMFUNCTION wolfBoot_set_sector_flag(uint8_t part, uint16_t sector, uint8_t newflag)
 {
     uint32_t *magic;
     uint8_t *flags;
@@ -238,7 +238,7 @@ int RAMFUNCTION wolfBoot_get_partition_state(uint8_t part, uint8_t *st)
     return 0;
 }
 
-int wolfBoot_get_sector_flag(uint8_t part, uint8_t sector, uint8_t *flag)
+int wolfBoot_get_sector_flag(uint8_t part, uint16_t sector, uint8_t *flag)
 {
     uint32_t *magic;
     uint8_t *flags;