@@ -1744,7 +1744,7 @@ static int make_header_ex(int is_diff, uint8_t *pubkey, uint32_t pubkey_sz,
17441744 {
17451745 const char * env_psize = getenv ("WOLFBOOT_PARTITION_SIZE" );
17461746 const char * env_ssize = getenv ("WOLFBOOT_SECTOR_SIZE" );
1747- if (env_psize ) {
1747+ if (env_psize && * env_psize ) {
17481748 char * endptr ;
17491749 unsigned long tmp ;
17501750 uint32_t partition_sz , sector_sz = 0 ;
@@ -1779,10 +1779,17 @@ static int make_header_ex(int is_diff, uint8_t *pubkey, uint32_t pubkey_sz,
17791779 uint32_t max_img_sz = (sector_sz < partition_sz ) ?
17801780 (partition_sz - sector_sz ) : partition_sz ;
17811781 if (total_img_sz > max_img_sz ) {
1782- printf ("Error: Image size %u (header %u + firmware %u) "
1783- "exceeds max %u (partition %u - sector %u)\n" ,
1784- total_img_sz , CMD .header_sz , image_sz ,
1785- max_img_sz , partition_sz , sector_sz );
1782+ if (sector_sz < partition_sz ) {
1783+ printf ("Error: Image size %u (header %u + firmware %u) "
1784+ "exceeds max %u (partition %u - sector %u)\n" ,
1785+ total_img_sz , CMD .header_sz , image_sz ,
1786+ max_img_sz , partition_sz , sector_sz );
1787+ } else {
1788+ printf ("Error: Image size %u (header %u + firmware %u) "
1789+ "exceeds max %u (partition %u)\n" ,
1790+ total_img_sz , CMD .header_sz , image_sz ,
1791+ max_img_sz , partition_sz );
1792+ }
17861793 goto failure ;
17871794 }
17881795 }
0 commit comments