From db99ce7ced783b3496003d4ce118a35447e89ba4 Mon Sep 17 00:00:00 2001 From: Sebastian Huber Date: Fri, 23 Aug 2013 09:02:40 +0200 Subject: libblock: PR2040: Avoid NULL pointer access This partly reverts commit 08b9d534604fbf437743a9925ef28eb1b848510d. Avoid a NULL pointer access and perform the source segment erase if necessary. --- cpukit/libblock/src/flashdisk.c | 22 ++++++++++++++++------ 1 file changed, 16 insertions(+), 6 deletions(-) (limited to 'cpukit') diff --git a/cpukit/libblock/src/flashdisk.c b/cpukit/libblock/src/flashdisk.c index ff336925c6..48d83e974a 100644 --- a/cpukit/libblock/src/flashdisk.c +++ b/cpukit/libblock/src/flashdisk.c @@ -1263,12 +1263,6 @@ rtems_fdisk_recycle_segment (rtems_flashdisk* fd, { rtems_fdisk_page_desc* spd = &ssc->page_descriptors[spage]; - if (!dsc && ssc->pages_active > 0) - { - rtems_fdisk_error ("recycle: no available dst segment"); - return EIO; - } - if (rtems_fdisk_page_desc_flags_set (spd, RTEMS_FDISK_PAGE_ACTIVE) && !rtems_fdisk_page_desc_flags_set (spd, RTEMS_FDISK_PAGE_USED)) { @@ -1358,7 +1352,23 @@ rtems_fdisk_recycle_segment (rtems_flashdisk* fd, */ dst_pages = rtems_fdisk_seg_pages_available (dsc); if (dst_pages == 0) + { dsc = rtems_fdisk_seg_most_available (&fd->available); + if (!dsc) + { + if (ssc->pages_active == 0) + { + ret = rtems_fdisk_erase_segment (fd, ssc); + } + else + { + rtems_fdisk_error ("recycle: no available dst segment"); + ret = EIO; + } + + return ret; + } + } (*pages)--; } -- cgit v1.2.3