espressif/esp-idf
1
0
Fork 1
mirror of https://github.com/espressif/esp-idf.git synced 2025-08-04 13:14:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

spi_flash: fix memory leak when spi_flash_mmap arguments are invalid

Browse Source 
Check src_addr and size first, then allocate new_entry.
This commit is contained in:
Ivan Grokhotkov
2017-02-28 15:11:54 +08:00
parent 75115e1d83
commit f3687f7177
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
components/spi_flash/flash_mmap.c
View File

@@ -93,16 +93,16 @@ esp_err_t IRAM_ATTR spi_flash_mmap(size_t src_addr, size_t size, spi_flash_mmap_
{ {
esp_err_t ret; esp_err_t ret;
bool did_flush, need_flush = false; bool did_flush, need_flush = false;
mmap_entry_t* new_entry = (mmap_entry_t*) malloc(sizeof(mmap_entry_t));
if (new_entry == 0) {
return ESP_ERR_NO_MEM;
}
if (src_addr & 0xffff) { if (src_addr & 0xffff) {
return ESP_ERR_INVALID_ARG; return ESP_ERR_INVALID_ARG;
} }
if (src_addr + size > g_rom_flashchip.chip_size) { if (src_addr + size > g_rom_flashchip.chip_size) {
return ESP_ERR_INVALID_ARG; return ESP_ERR_INVALID_ARG;
} }
mmap_entry_t* new_entry = (mmap_entry_t*) malloc(sizeof(mmap_entry_t));
if (new_entry == 0) {
return ESP_ERR_NO_MEM;
}
spi_flash_disable_interrupts_caches_and_other_cpu(); spi_flash_disable_interrupts_caches_and_other_cpu();