Add boundary check

Signed-off-by: HiFiPhile <admin@hifiphile.com>
This commit is contained in:
HiFiPhile
2025-10-31 10:49:33 +01:00
parent 7c4cc0f7c8
commit cee793796c

View File

@ -187,7 +187,9 @@ uint16_t tud_dfu_upload_cb(uint8_t alt, uint16_t block_num, uint8_t* data, uint1
(void) block_num;
(void) length;
uint16_t const xfer_len = (uint16_t) strlen(upload_image[alt]);
TU_VERIFY(block_num == 0, 0); // for this example we only support single block upload
uint16_t const xfer_len = tu_min16((uint16_t) strlen(upload_image[alt]), length);
memcpy(data, upload_image[alt], xfer_len);
return xfer_len;