From 8c7e9e0936fcb18f1c95e618c2a2bb138a3c9e35 Mon Sep 17 00:00:00 2001 From: Jehan Bing Date: Wed, 5 Aug 2009 17:40:42 -0700 Subject: nandwrite: unified reading from standard input and from file - part2 Use the same code structure when reading the OOB than when reading the regular data. Signed-off-by: Jehan Bing Signed-off-by: Artem Bityutskiy --- nandwrite.c | 31 +++++++++++++++++++++++-------- 1 file changed, 23 insertions(+), 8 deletions(-) diff --git a/nandwrite.c b/nandwrite.c index 3695c20..f6bff60 100644 --- a/nandwrite.c +++ b/nandwrite.c @@ -520,17 +520,32 @@ int main(int argc, char * const argv[]) } if (writeoob) { - int tinycnt = 0; + { + int readlen = meminfo.oobsize; - while(tinycnt < meminfo.oobsize) { - cnt = read(ifd, oobreadbuf + tinycnt, meminfo.oobsize - tinycnt); - if (cnt == 0) { // EOF - break; - } else if (cnt < 0) { - perror ("File I/O error on input file"); + int tinycnt = 0; + + while (tinycnt < readlen) { + cnt = read(ifd, oobreadbuf + tinycnt, readlen - tinycnt); + if (cnt == 0) { // EOF + break; + } else if (cnt < 0) { + perror ("File I/O error on input"); + goto closeall; + } + tinycnt += cnt; + } + + if (tinycnt < readlen) { + fprintf(stderr, "Unexpected EOF. Expecting at least " + "%d more bytes for OOB\n", readlen - tinycnt); goto closeall; } - tinycnt += cnt; + + if ((ifd == STDIN_FILENO) && (cnt == 0)) { + /* No more bytes - we are done after writing the remaining bytes */ + imglen = 0; + } } if (!noecc) { -- cgit v1.2.3