version 1.1.2.1, 2014/02/05 12:37:31
|
version 1.2, 2014/02/05 15:44:06
|
Line 1
|
Line 1
|
|
/************************************************************************* |
|
* (C) 2014 AITNET - Sofia/Bulgaria - <office@aitbg.com> |
|
* by Michael Pounov <misho@aitbg.com> |
|
* |
|
* $Author$ |
|
* $Id$ |
|
* |
|
************************************************************************* |
|
The ELWIX and AITNET software is distributed under the following |
|
terms: |
|
|
|
All of the documentation and software included in the ELWIX and AITNET |
|
Releases is copyrighted by ELWIX - Sofia/Bulgaria <info@elwix.org> |
|
|
|
Copyright 2004 - 2014 |
|
by Michael Pounov <misho@elwix.org>. All rights reserved. |
|
|
|
Redistribution and use in source and binary forms, with or without |
|
modification, are permitted provided that the following conditions |
|
are met: |
|
1. Redistributions of source code must retain the above copyright |
|
notice, this list of conditions and the following disclaimer. |
|
2. Redistributions in binary form must reproduce the above copyright |
|
notice, this list of conditions and the following disclaimer in the |
|
documentation and/or other materials provided with the distribution. |
|
3. All advertising materials mentioning features or use of this software |
|
must display the following acknowledgement: |
|
This product includes software developed by Michael Pounov <misho@elwix.org> |
|
ELWIX - Embedded LightWeight unIX and its contributors. |
|
4. Neither the name of AITNET nor the names of its contributors |
|
may be used to endorse or promote products derived from this software |
|
without specific prior written permission. |
|
|
|
THIS SOFTWARE IS PROVIDED BY AITNET AND CONTRIBUTORS ``AS IS'' AND |
|
ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE |
|
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE |
|
ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE |
|
FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL |
|
DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS |
|
OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) |
|
HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT |
|
LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY |
|
OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF |
|
SUCH DAMAGE. |
|
*/ |
#include "global.h" |
#include "global.h" |
|
|
|
|
char imgName[PATH_MAX], imgFile[PATH_MAX]; |
char imgName[PATH_MAX], imgFile[PATH_MAX]; |
off_t imgSize, fSize, iSize; | off_t imgSize, iSize; |
int Verbose; |
int Verbose; |
extern char compiled[], compiledby[], compilehost[]; |
extern char compiled[], compiledby[], compilehost[]; |
|
|
Line 14 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-s <size>\tStorage size\n" | "\t-t\t\tTruncate Storage file name\n" |
| "\t-s <size>\tStorage size (required for stdin)\n" |
"\t-f <devfile>\tStorage file name\n" |
"\t-f <devfile>\tStorage file name\n" |
"\n", compiled, compiledby, compilehost); |
"\n", compiled, compiledby, compilehost); |
} |
} |
Line 23 static int
|
Line 69 static int
|
EmptyStore(int img) |
EmptyStore(int img) |
{ |
{ |
register int i; |
register int i; |
u_char buf[IMGBLOCK_SIZE]; | u_char buf[IMGBUF_SIZE]; |
ssize_t wlen; |
ssize_t wlen; |
|
|
|
VERB(1) printf("Erase store %s\n", imgName); |
|
|
|
iSize = lseek(img, 0, SEEK_END); |
|
if (iSize == -1) { |
|
ESYSERR(0); |
|
return -1; |
|
} else |
|
imgSize += E_ALIGN(iSize, IMGBUF_SIZE); |
|
|
memset(buf, 0, sizeof buf); |
memset(buf, 0, sizeof buf); |
for (i = 0; i < howmany(imgSize, IMGBLOCK_SIZE); i++) | for (i = howmany(iSize, IMGBUF_SIZE); i < howmany(imgSize, IMGBUF_SIZE); i++) |
if ((wlen = write(img, buf, sizeof buf)) == -1 || |
if ((wlen = write(img, buf, sizeof buf)) == -1 || |
wlen != sizeof buf) { |
wlen != sizeof buf) { |
printf("Error:: at chunk %d init %d bytes, should be %u", | EERROR(EIO, "Error at chunk %d init %d bytes, should be %u\n", |
i, wlen, sizeof buf); |
i, wlen, sizeof buf); |
return -1; |
return -1; |
} else |
} else |
VERB(1) printf("+Written chunk #%d\n", i); |
VERB(1) printf("+Written chunk #%d\n", i); |
|
|
iSize = lseek(img, 0, SEEK_END); | iSize = lseek(img, iSize, SEEK_SET); |
return iSize; |
return iSize; |
} |
} |
|
|
|
static int |
|
FillStore(int img, int fd) |
|
{ |
|
register int i, j; |
|
u_char buf[IMGBUF_SIZE]; |
|
ssize_t rlen, wlen; |
|
|
|
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); |
|
i++, j++) { |
|
memset(buf, 0, sizeof buf); |
|
rlen = read(fd, buf, sizeof buf); |
|
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); |
|
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 |
|
VERB(1) printf("+Written %d bytes at chunk #%d\n", wlen, i); |
|
} |
|
|
|
return 0; |
|
} |
|
|
int |
int |
main(int argc, char **argv) |
main(int argc, char **argv) |
{ |
{ |
char ch; |
char ch; |
int fd, img; | int fd, img, tr = 0; |
|
|
while ((ch = getopt(argc, argv, "hvs:f:")) != -1) | while ((ch = getopt(argc, argv, "hvts:f:")) != -1) |
switch (ch) { |
switch (ch) { |
case 'f': |
case 'f': |
strlcpy(imgName, optarg, sizeof imgName); |
strlcpy(imgName, optarg, sizeof imgName); |
Line 58 main(int argc, char **argv)
|
Line 148 main(int argc, char **argv)
|
return 1; |
return 1; |
} |
} |
break; |
break; |
|
case 't': |
|
tr = O_TRUNC; |
|
break; |
case 'v': |
case 'v': |
Verbose++; |
Verbose++; |
break; |
break; |
Line 68 main(int argc, char **argv)
|
Line 161 main(int argc, char **argv)
|
} |
} |
argc -= optind; |
argc -= optind; |
argv += optind; |
argv += optind; |
if (!argc || !*imgName || !imgSize) { | |
| if (argc) { |
| strlcpy(imgFile, *argv, sizeof imgFile); |
| /* open image file */ |
| fd = open(imgFile, O_RDONLY); |
| if (fd == -1) { |
| ESYSERR(0); |
| return 2; |
| } else |
| iSize = lseek(fd, 0, SEEK_END); |
| if (!imgSize) |
| imgSize = E_ALIGN(iSize, IMGBUF_SIZE); |
| if (iSize == -1 || iSize > imgSize) { |
| close(fd); |
| EERROR(ENOSPC, "Error:: file size %llu is greater from storage size %llu\n", |
| iSize, imgSize); |
| return 2; |
| } else |
| lseek(fd, 0, SEEK_SET); |
| } else if (!imgSize) { |
Usage(); |
Usage(); |
return 1; |
return 1; |
} else |
} else |
strlcpy(imgFile, *argv, sizeof imgFile); | fd = STDIN_FILENO; |
|
|
VERB(1) printf("imgSize=%llu imgName=%s imgFile=%s\n", imgSize, imgName, imgFile); | VERB(1) printf("imgSize=%llu imgName=%s imgFile=%s\n", |
| imgSize, imgName, argc ? imgFile : "<stdin>"); |
|
|
/* open image file */ |
|
fd = open(imgFile, O_RDONLY); |
|
if (fd == -1) { |
|
ESYSERR(0); |
|
return 2; |
|
} else |
|
fSize = lseek(fd, 0, SEEK_END); |
|
if (fSize == -1 || fSize > imgSize) { |
|
close(fd); |
|
printf("Error:: file size %llu is greater from storage size %llu\n", |
|
fSize, imgSize); |
|
return 2; |
|
} else |
|
lseek(fd, 0, SEEK_SET); |
|
|
|
/* open storage device */ |
/* open storage device */ |
img = open(imgName, O_RDWR | O_CREAT, 0644); | img = open(imgName, O_RDWR | O_CREAT | tr, 0644); |
if (img == -1) { |
if (img == -1) { |
ESYSERR(0); |
ESYSERR(0); |
close(fd); | if (fd > 2) |
| close(fd); |
return 3; |
return 3; |
} |
} |
if (EmptyStore(img) == -1 || iSize < imgSize) { | if (EmptyStore(img) == -1) { |
close(fd); | if (fd > 2) |
| close(fd); |
close(img); |
close(img); |
unlink(imgName); |
unlink(imgName); |
printf("Error:: current storage size %llu is smaller from storage size %llu\n", |
|
iSize, imgSize); |
|
return 3; |
return 3; |
} else | } |
lseek(img, 0, SEEK_SET); | |
|
|
|
if (FillStore(img, fd) == -1) { |
|
if (fd > 2) |
|
close(fd); |
|
close(img); |
|
unlink(imgName); |
|
return 4; |
|
} |
|
|
close(img); |
close(img); |
close(fd); | if (fd > 2) |
| close(fd); |
return 0; |
return 0; |
} |
} |