--- embedtools/src/imgupd.c 2014/02/06 00:00:18 1.6.2.1 +++ embedtools/src/imgupd.c 2014/02/06 09:19:51 1.6.2.5 @@ -3,7 +3,7 @@ * by Michael Pounov * * $Author: misho $ - * $Id: imgupd.c,v 1.6.2.1 2014/02/06 00:00:18 misho Exp $ + * $Id: imgupd.c,v 1.6.2.5 2014/02/06 09:19:51 misho Exp $ * ************************************************************************* The ELWIX and AITNET software is distributed under the following @@ -92,8 +92,11 @@ EmptyStore(int img) for (i = howmany(iSize, bufSize); i < howmany(imgSize, bufSize); i++) if ((wlen = write(img, buf, bufSize)) == -1 || (wlen && wlen != bufSize)) { - EERROR(EIO, "Error at chunk %d init %d bytes, " - "should be %u\n", i, wlen, bufSize); + if (wlen == -1) + ESYSERR(0); + else + EERROR(EIO, "Error at chunk %d init %d bytes, " + "should be %u\n", i, wlen, bufSize); return -1; } else VERB(1) printf("+Written chunk #%d\n", i); @@ -106,30 +109,32 @@ static int FillStore(int img, int fd) { register int i, j; - u_char buf[bufSize]; - ssize_t rlen, wlen; + u_char *pos, buf[bufSize]; + ssize_t rlen, wlen, len; VERB(1) printf("Fill store %s from image file %s\n", imgName, imgFile); for (j = 0, i = howmany(iSize, bufSize); i < howmany(imgSize, bufSize); i++, j++) { memset(buf, 0, sizeof buf); - rlen = read(fd, buf, bufSize); - if (rlen == -1) { - ESYSERR(0); - return -1; - } else if (!rlen) - break; - else - VERB(1) printf("+Readed %d bytes for chunk #%d\n", rlen, j); + for (len = bufSize, pos = buf; len > 0; len -= rlen, pos += rlen) { + rlen = read(fd, pos, len); + if (rlen == -1) { + ESYSERR(0); + return -1; + } else if (!rlen) + break; + else + VERB(1) printf("+Readed %d bytes for chunk #%d\n", rlen, j); + } - wlen = write(img, buf, rlen); + wlen = write(img, buf, bufSize); if (wlen == -1) { ESYSERR(0); return -1; - } else if (wlen && wlen != rlen) { - EERROR(EIO, "Readed %d bytes are not equal to written %d bytes\n", - rlen, wlen); + } else if (wlen && wlen != bufSize) { + EERROR(EIO, "Error at chunk %d written %d bytes, " + "should be %u\n", i, wlen, bufSize); } else VERB(1) printf("+Written %d bytes at chunk #%d\n", wlen, i); } @@ -158,12 +163,11 @@ main(int argc, char **argv) case 't': tr = O_TRUNC; break; - case 'g': - m = 1; - break; case 'p': bufSize = IMGBUF_SIZE2; break; + case 'g': + m = 1; case 'i': fromBegin = 1; break; @@ -272,6 +276,7 @@ main(int argc, char **argv) return 4; } } else { /* GET */ + iSize ^= iSize; if (EmptyStore(fd) == -1) { if (fd > 2) close(fd);