From mboxrd@z Thu Jan 1 00:00:00 1970 Return-path: Received: from metis.ext.pengutronix.de ([2001:6f8:1178:4:290:27ff:fe1d:cc33]) by merlin.infradead.org with esmtps (Exim 4.80.1 #2 (Red Hat Linux)) id 1UKlLj-0008Cw-NF for barebox@lists.infradead.org; Wed, 27 Mar 2013 08:04:48 +0000 Date: Wed, 27 Mar 2013 09:04:45 +0100 From: Sascha Hauer Message-ID: <20130327080445.GO1906@pengutronix.de> References: <1364224557-1840-1-git-send-email-j.weitzel@phytec.de> MIME-Version: 1.0 Content-Disposition: inline In-Reply-To: <1364224557-1840-1-git-send-email-j.weitzel@phytec.de> List-Id: List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , Content-Type: text/plain; charset="us-ascii" Content-Transfer-Encoding: 7bit Sender: "barebox" Errors-To: barebox-bounces+u.kleine-koenig=pengutronix.de@lists.infradead.org Subject: Re: [PATCH] ubiformat: get buffer from malloc To: Jan Weitzel Cc: barebox@lists.infradead.org On Mon, Mar 25, 2013 at 04:15:57PM +0100, Jan Weitzel wrote: > There was a erase block sized (here 131072) char buf array on the stack. > Changed this to get the space from malloc preventing stack overflows. > Also fix a wrong return without clean up. > > Signed-off-by: Jan Weitzel Applied, thanks Sascha > --- > commands/ubiformat.c | 22 +++++++++++++++------- > 1 files changed, 15 insertions(+), 7 deletions(-) > > diff --git a/commands/ubiformat.c b/commands/ubiformat.c > index 47941be..121816f 100644 > --- a/commands/ubiformat.c > +++ b/commands/ubiformat.c > @@ -296,13 +296,20 @@ static int mark_bad(const struct mtd_dev_info *mtd, struct ubi_scan_info *si, in > static int flash_image(const struct mtd_dev_info *mtd, > const struct ubigen_info *ui, struct ubi_scan_info *si) > { > - int fd, img_ebs, eb, written_ebs = 0, divisor; > + int fd, img_ebs, eb, written_ebs = 0, divisor, ret = -1; > off_t st_size; > + char *buf = NULL; > > fd = open_file(&st_size); > if (fd < 0) > return fd; > > + buf = malloc(mtd->eb_size); > + if (!buf) { > + sys_errmsg("cannot allocate %d bytes of memory", mtd->eb_size); > + goto out_close; > + } > + > img_ebs = st_size / mtd->eb_size; > > if (img_ebs > si->good_cnt) { > @@ -312,8 +319,9 @@ static int flash_image(const struct mtd_dev_info *mtd, > } > > if (st_size % mtd->eb_size) { > - return sys_errmsg("file \"%s\" (size %lld bytes) is not multiple of ""eraseblock size (%d bytes)", > - args.image, (long long)st_size, mtd->eb_size); > + sys_errmsg("file \"%s\" (size %lld bytes) is not multiple of " > + "eraseblock size (%d bytes)", > + args.image, (long long)st_size, mtd->eb_size); > goto out_close; > } > > @@ -321,7 +329,6 @@ static int flash_image(const struct mtd_dev_info *mtd, > divisor = img_ebs; > for (eb = 0; eb < mtd->eb_cnt; eb++) { > int err, new_len; > - char buf[mtd->eb_size]; > long long ec; > > if (!args.quiet && !args.verbose) { > @@ -404,12 +411,13 @@ static int flash_image(const struct mtd_dev_info *mtd, > > if (!args.quiet && !args.verbose) > printf("\n"); > - close(fd); > - return eb + 1; > + > + ret = eb + 1; > > out_close: > + free(buf); > close(fd); > - return -1; > + return ret; > } > > static int format(const struct mtd_dev_info *mtd, > -- > 1.7.0.4 > > > _______________________________________________ > barebox mailing list > barebox@lists.infradead.org > http://lists.infradead.org/mailman/listinfo/barebox > -- Pengutronix e.K. | | Industrial Linux Solutions | http://www.pengutronix.de/ | Peiner Str. 6-8, 31137 Hildesheim, Germany | Phone: +49-5121-206917-0 | Amtsgericht Hildesheim, HRA 2686 | Fax: +49-5121-206917-5555 | _______________________________________________ barebox mailing list barebox@lists.infradead.org http://lists.infradead.org/mailman/listinfo/barebox