version 1.1, 2014/02/05 12:37:31
|
version 1.2, 2014/02/05 15:44:06
|
Line 0
|
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" |
|
|
|
|
|
char imgName[PATH_MAX], imgFile[PATH_MAX]; |
|
off_t imgSize, iSize; |
|
int Verbose; |
|
extern char compiled[], compiledby[], compilehost[]; |
|
|
|
static void |
|
Usage() |
|
{ |
|
|
|
printf( "IMGUPD is tool for management of images\n" |
|
"=== %s === %s@%s ===\n\n" |
|
" Syntax: imgupd [options] <image_file>\n\n" |
|
"\t-v\t\tVerbose ...\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" |
|
"\n", compiled, compiledby, compilehost); |
|
} |
|
|
|
static int |
|
EmptyStore(int img) |
|
{ |
|
register int i; |
|
u_char buf[IMGBUF_SIZE]; |
|
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); |
|
for (i = howmany(iSize, IMGBUF_SIZE); i < howmany(imgSize, IMGBUF_SIZE); i++) |
|
if ((wlen = write(img, buf, sizeof buf)) == -1 || |
|
wlen != sizeof buf) { |
|
EERROR(EIO, "Error at chunk %d init %d bytes, should be %u\n", |
|
i, wlen, sizeof buf); |
|
return -1; |
|
} else |
|
VERB(1) printf("+Written chunk #%d\n", i); |
|
|
|
iSize = lseek(img, iSize, SEEK_SET); |
|
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 |
|
main(int argc, char **argv) |
|
{ |
|
char ch; |
|
int fd, img, tr = 0; |
|
|
|
while ((ch = getopt(argc, argv, "hvts:f:")) != -1) |
|
switch (ch) { |
|
case 'f': |
|
strlcpy(imgName, optarg, sizeof imgName); |
|
break; |
|
case 's': |
|
imgSize = strtoll(optarg, NULL, 0); |
|
if (!imgSize) { |
|
Usage(); |
|
return 1; |
|
} |
|
break; |
|
case 't': |
|
tr = O_TRUNC; |
|
break; |
|
case 'v': |
|
Verbose++; |
|
break; |
|
case 'h': |
|
default: |
|
Usage(); |
|
return 1; |
|
} |
|
argc -= optind; |
|
argv += optind; |
|
|
|
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(); |
|
return 1; |
|
} else |
|
fd = STDIN_FILENO; |
|
|
|
VERB(1) printf("imgSize=%llu imgName=%s imgFile=%s\n", |
|
imgSize, imgName, argc ? imgFile : "<stdin>"); |
|
|
|
/* open storage device */ |
|
img = open(imgName, O_RDWR | O_CREAT | tr, 0644); |
|
if (img == -1) { |
|
ESYSERR(0); |
|
if (fd > 2) |
|
close(fd); |
|
return 3; |
|
} |
|
if (EmptyStore(img) == -1) { |
|
if (fd > 2) |
|
close(fd); |
|
close(img); |
|
unlink(imgName); |
|
return 3; |
|
} |
|
|
|
if (FillStore(img, fd) == -1) { |
|
if (fd > 2) |
|
close(fd); |
|
close(img); |
|
unlink(imgName); |
|
return 4; |
|
} |
|
|
|
close(img); |
|
if (fd > 2) |
|
close(fd); |
|
return 0; |
|
} |