WolfSSL
/
wolfBoot
mirror of https://github.com/wolfSSL/wolfBoot.git
1
0
Fork 0
Code Issues Releases Wiki Activity

disk.c: fix silent sz truncation and explicitly bound disk IO

pull/703/head
Marco Oliverio 2026-02-26 19:20:01 +01:00 committed by Daniele Lacamera
parent 8c826d4cd6
commit b0cc295b71
2 changed files with 11 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
include/disk.h
View File

@ -25,6 +25,9 @@
#include <stdint.h>
#include "gpt.h"
/* cap DISK I/O write operation to a reasonable size */
#define DISK_IO_MAX_SIZE 0x7FFFFFFFUL
#ifndef MAX_PARTITIONS
#define MAX_PARTITIONS 16
#endif

10
src/disk.c
View File

@ -276,12 +276,15 @@ static struct disk_partition *open_part(int drv, int part)
int disk_part_read(int drv, int part, uint64_t off, uint64_t sz, uint8_t *buf)
{
struct disk_partition *p = open_part(drv, part);
int len = sz;
uint64_t start;
int len;
int ret;
if (p == NULL) {
return -1;
}
if (sz > DISK_IO_MAX_SIZE)
sz = DISK_IO_MAX_SIZE;
len = (int)sz;
start = p->start + off;
/* overflow */
if (start < p->start) {
@ -325,12 +328,15 @@ int disk_part_read(int drv, int part, uint64_t off, uint64_t sz, uint8_t *buf)
int disk_part_write(int drv, int part, uint64_t off, uint64_t sz, const uint8_t *buf)
{
struct disk_partition *p = open_part(drv, part);
int len = sz;
uint64_t start;
int len;
int ret;
if (p == NULL) {
return -1;
}
if (sz > DISK_IO_MAX_SIZE)
sz = DISK_IO_MAX_SIZE;
len = (int)sz;
start = p->start + off;
/* overflow */
if (start < p->start) {