Diff for /embedtools/src/imgupd.c between versions 1.2.2.1 and 1.6.2.3

version 1.2.2.1, 2014/02/05 21:59:51 version 1.6.2.3, 2014/02/06 00:56:16
Line 47  SUCH DAMAGE. Line 47  SUCH DAMAGE.
   
   
 char imgName[PATH_MAX], imgFile[PATH_MAX];  char imgName[PATH_MAX], imgFile[PATH_MAX];
off_t imgSize, iSize;off_t imgSize, iSize, fromBegin;
int Verbose;int Verbose, bufSize = IMGBUF_SIZE;
 extern char compiled[], compiledby[], compilehost[];  extern char compiled[], compiledby[], compilehost[];
   
 static void  static void
Line 59  Usage() Line 59  Usage()
                 "=== %s === %s@%s ===\n\n"                  "=== %s === %s@%s ===\n\n"
                 "  Syntax: imgupd [options] [image_file]\n\n"                  "  Syntax: imgupd [options] [image_file]\n\n"
                 "\t-v\t\tVerbose ...\n"                  "\t-v\t\tVerbose ...\n"
                   "\t-R\t\tReboot system after complete\n"
                   "\t-p\t\tPipe suitable transfer on little chunks\n"
                   "\t-i\t\tStart fill storage from begin\n"
                 "\t-g\t\tGet image from Storage\n"                  "\t-g\t\tGet image from Storage\n"
                 "\t-t\t\tTruncate Storage file name\n"                  "\t-t\t\tTruncate Storage file name\n"
                 "\t-s <size>\tStorage size (required for stdin)\n"                  "\t-s <size>\tStorage size (required for stdin)\n"
Line 70  static int Line 73  static int
 EmptyStore(int img)  EmptyStore(int img)
 {  {
         register int i;          register int i;
        u_char buf[IMGBUF_SIZE];        u_char buf[bufSize];
         ssize_t wlen;          ssize_t wlen;
   
         VERB(1) printf("Erase store %s\n", imgName);          VERB(1) printf("Erase store %s\n", imgName);
   
        iSize = lseek(img, 0, SEEK_END);        if (!fromBegin) {
        if (iSize == -1) {                iSize = lseek(img, 0, SEEK_END);
                ESYSERR(0);                if (iSize == -1) {
                return -1;                        ESYSERR(0);
                         return -1;
                 } else
                         imgSize += E_ALIGN(iSize, bufSize);
         } else          } else
                imgSize += E_ALIGN(iSize, IMGBUF_SIZE);                iSize ^= iSize;
   
         memset(buf, 0, sizeof buf);          memset(buf, 0, sizeof buf);
        for (i = howmany(iSize, IMGBUF_SIZE); i < howmany(imgSize, IMGBUF_SIZE); i++)        for (i = howmany(iSize, bufSize); i < howmany(imgSize, bufSize); i++)
                if ((wlen = write(img, buf, sizeof buf)) == -1 ||                 if ((wlen = write(img, buf, bufSize)) == -1 || 
                                wlen != sizeof buf) {                                (wlen && wlen != bufSize)) {
                        EERROR(EIO, "Error at chunk %d init %d bytes, should be %u\n",                         EERROR(EIO, "Error at chunk %d init %d bytes, "
                                        i, wlen, sizeof buf);                                        "should be %u\n", i, wlen, bufSize);
                         return -1;                          return -1;
                 } else                  } else
                         VERB(1) printf("+Written chunk #%d\n", i);                          VERB(1) printf("+Written chunk #%d\n", i);
Line 100  static int Line 106  static int
 FillStore(int img, int fd)  FillStore(int img, int fd)
 {  {
         register int i, j;          register int i, j;
        u_char buf[IMGBUF_SIZE];        u_char buf[bufSize];
         ssize_t rlen, wlen;          ssize_t rlen, wlen;
   
         VERB(1) printf("Fill store %s from image file %s\n", imgName, imgFile);          VERB(1) printf("Fill store %s from image file %s\n", imgName, imgFile);
   
        for (j = 0, i = howmany(iSize, IMGBUF_SIZE); i < howmany(imgSize, IMGBUF_SIZE);         for (j = 0, i = howmany(iSize, bufSize); i < howmany(imgSize, bufSize); 
                         i++, j++) {                          i++, j++) {
                 memset(buf, 0, sizeof buf);                  memset(buf, 0, sizeof buf);
                rlen = read(fd, buf, sizeof buf);                rlen = read(fd, buf, bufSize);
                 if (rlen == -1) {                  if (rlen == -1) {
                         ESYSERR(0);                          ESYSERR(0);
                         return -1;                          return -1;
Line 117  FillStore(int img, int fd) Line 123  FillStore(int img, int fd)
                 else                  else
                         VERB(1) printf("+Readed %d bytes for chunk #%d\n", rlen, j);                          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) {                  if (wlen == -1) {
                         ESYSERR(0);                          ESYSERR(0);
                         return -1;                          return -1;
                } else if (!wlen || wlen != rlen) {                } else if (wlen && wlen != bufSize) {
                        EERROR(EIO, "Readed %d bytes are not equal to written %d bytes\n",                         EERROR(EIO, "Error at chunk %d written %d bytes, "
                                        rlen, wlen);                                        "should be %u\n", i, wlen, bufSize);
                 } else                  } else
                         VERB(1) printf("+Written %d bytes at chunk #%d\n", wlen, i);                          VERB(1) printf("+Written %d bytes at chunk #%d\n", wlen, i);
         }          }
Line 134  FillStore(int img, int fd) Line 140  FillStore(int img, int fd)
 int  int
 main(int argc, char **argv)  main(int argc, char **argv)
 {  {
        char ch, m = 0;        char ch, m = 0, R = 0;
         int fd, img, tr = 0;          int fd, img, tr = 0;
   
        while ((ch = getopt(argc, argv, "hvgts:f:")) != -1)        while ((ch = getopt(argc, argv, "hvRipgts:f:")) != -1)
                 switch (ch) {                  switch (ch) {
                         case 'f':                          case 'f':
                                 strlcpy(imgName, optarg, sizeof imgName);                                  strlcpy(imgName, optarg, sizeof imgName);
Line 152  main(int argc, char **argv) Line 158  main(int argc, char **argv)
                         case 't':                          case 't':
                                 tr = O_TRUNC;                                  tr = O_TRUNC;
                                 break;                                  break;
                           case 'p':
                                   bufSize = IMGBUF_SIZE2;
                                   break;
                         case 'g':                          case 'g':
                                 m = 1;                                  m = 1;
                           case 'i':
                                   fromBegin = 1;
                                 break;                                  break;
                           case 'R':
                                   R = 1;
                                   break;
                         case 'v':                          case 'v':
                                 Verbose++;                                  Verbose++;
                                 break;                                  break;
Line 177  main(int argc, char **argv) Line 191  main(int argc, char **argv)
                         } else                          } else
                                 iSize = lseek(fd, 0, SEEK_END);                                  iSize = lseek(fd, 0, SEEK_END);
                         if (!imgSize)                          if (!imgSize)
                                imgSize = E_ALIGN(iSize, IMGBUF_SIZE);                                imgSize = E_ALIGN(iSize, bufSize);
                         if (iSize == -1 || iSize > imgSize) {                          if (iSize == -1 || iSize > imgSize) {
                                 close(fd);                                  close(fd);
                                 EERROR(ENOSPC, "Error:: file size %llu is "                                  EERROR(ENOSPC, "Error:: file size %llu is "
Line 230  main(int argc, char **argv) Line 244  main(int argc, char **argv)
                 } else                  } else
                         iSize = lseek(img, 0, SEEK_END);                          iSize = lseek(img, 0, SEEK_END);
                 if (!imgSize)                  if (!imgSize)
                        imgSize = E_ALIGN(iSize, IMGBUF_SIZE);                        imgSize = E_ALIGN(iSize, bufSize);
                 if (iSize == -1 || iSize > imgSize) {                  if (iSize == -1 || iSize > imgSize) {
                         if (fd > 2)                          if (fd > 2)
                                 close(fd);                                  close(fd);
Line 274  main(int argc, char **argv) Line 288  main(int argc, char **argv)
         close(img);          close(img);
         if (fd > 2)          if (fd > 2)
                 close(fd);                  close(fd);
   
           if (R)
                   reboot(RB_AUTOBOOT);
         return 0;          return 0;
 }  }

Removed from v.1.2.2.1  
changed lines
  Added in v.1.6.2.3


FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>