Annotation of embedtools/src/imgupd.c, revision 1.2
1.2 ! misho 1: /*************************************************************************
! 2: * (C) 2014 AITNET - Sofia/Bulgaria - <office@aitbg.com>
! 3: * by Michael Pounov <misho@aitbg.com>
! 4: *
! 5: * $Author: misho $
! 6: * $Id: imgupd.c,v 1.1.2.4 2014/02/05 15:25:37 misho Exp $
! 7: *
! 8: *************************************************************************
! 9: The ELWIX and AITNET software is distributed under the following
! 10: terms:
! 11:
! 12: All of the documentation and software included in the ELWIX and AITNET
! 13: Releases is copyrighted by ELWIX - Sofia/Bulgaria <info@elwix.org>
! 14:
! 15: Copyright 2004 - 2014
! 16: by Michael Pounov <misho@elwix.org>. All rights reserved.
! 17:
! 18: Redistribution and use in source and binary forms, with or without
! 19: modification, are permitted provided that the following conditions
! 20: are met:
! 21: 1. Redistributions of source code must retain the above copyright
! 22: notice, this list of conditions and the following disclaimer.
! 23: 2. Redistributions in binary form must reproduce the above copyright
! 24: notice, this list of conditions and the following disclaimer in the
! 25: documentation and/or other materials provided with the distribution.
! 26: 3. All advertising materials mentioning features or use of this software
! 27: must display the following acknowledgement:
! 28: This product includes software developed by Michael Pounov <misho@elwix.org>
! 29: ELWIX - Embedded LightWeight unIX and its contributors.
! 30: 4. Neither the name of AITNET nor the names of its contributors
! 31: may be used to endorse or promote products derived from this software
! 32: without specific prior written permission.
! 33:
! 34: THIS SOFTWARE IS PROVIDED BY AITNET AND CONTRIBUTORS ``AS IS'' AND
! 35: ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
! 36: IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
! 37: ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
! 38: FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
! 39: DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
! 40: OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
! 41: HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
! 42: LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
! 43: OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
! 44: SUCH DAMAGE.
! 45: */
! 46: #include "global.h"
! 47:
! 48:
! 49: char imgName[PATH_MAX], imgFile[PATH_MAX];
! 50: off_t imgSize, iSize;
! 51: int Verbose;
! 52: extern char compiled[], compiledby[], compilehost[];
! 53:
! 54: static void
! 55: Usage()
! 56: {
! 57:
! 58: printf( "IMGUPD is tool for management of images\n"
! 59: "=== %s === %s@%s ===\n\n"
! 60: " Syntax: imgupd [options] <image_file>\n\n"
! 61: "\t-v\t\tVerbose ...\n"
! 62: "\t-t\t\tTruncate Storage file name\n"
! 63: "\t-s <size>\tStorage size (required for stdin)\n"
! 64: "\t-f <devfile>\tStorage file name\n"
! 65: "\n", compiled, compiledby, compilehost);
! 66: }
! 67:
! 68: static int
! 69: EmptyStore(int img)
! 70: {
! 71: register int i;
! 72: u_char buf[IMGBUF_SIZE];
! 73: ssize_t wlen;
! 74:
! 75: VERB(1) printf("Erase store %s\n", imgName);
! 76:
! 77: iSize = lseek(img, 0, SEEK_END);
! 78: if (iSize == -1) {
! 79: ESYSERR(0);
! 80: return -1;
! 81: } else
! 82: imgSize += E_ALIGN(iSize, IMGBUF_SIZE);
! 83:
! 84: memset(buf, 0, sizeof buf);
! 85: for (i = howmany(iSize, IMGBUF_SIZE); i < howmany(imgSize, IMGBUF_SIZE); i++)
! 86: if ((wlen = write(img, buf, sizeof buf)) == -1 ||
! 87: wlen != sizeof buf) {
! 88: EERROR(EIO, "Error at chunk %d init %d bytes, should be %u\n",
! 89: i, wlen, sizeof buf);
! 90: return -1;
! 91: } else
! 92: VERB(1) printf("+Written chunk #%d\n", i);
! 93:
! 94: iSize = lseek(img, iSize, SEEK_SET);
! 95: return iSize;
! 96: }
! 97:
! 98: static int
! 99: FillStore(int img, int fd)
! 100: {
! 101: register int i, j;
! 102: u_char buf[IMGBUF_SIZE];
! 103: ssize_t rlen, wlen;
! 104:
! 105: VERB(1) printf("Fill store %s from image file %s\n", imgName, imgFile);
! 106:
! 107: for (j = 0, i = howmany(iSize, IMGBUF_SIZE); i < howmany(imgSize, IMGBUF_SIZE);
! 108: i++, j++) {
! 109: memset(buf, 0, sizeof buf);
! 110: rlen = read(fd, buf, sizeof buf);
! 111: if (rlen == -1) {
! 112: ESYSERR(0);
! 113: return -1;
! 114: } else if (!rlen)
! 115: break;
! 116: else
! 117: VERB(1) printf("+Readed %d bytes for chunk #%d\n", rlen, j);
! 118:
! 119: wlen = write(img, buf, rlen);
! 120: if (wlen == -1) {
! 121: ESYSERR(0);
! 122: return -1;
! 123: } else if (!wlen || wlen != rlen) {
! 124: EERROR(EIO, "Readed %d bytes are not equal to written %d bytes\n",
! 125: rlen, wlen);
! 126: } else
! 127: VERB(1) printf("+Written %d bytes at chunk #%d\n", wlen, i);
! 128: }
! 129:
! 130: return 0;
! 131: }
! 132:
! 133: int
! 134: main(int argc, char **argv)
! 135: {
! 136: char ch;
! 137: int fd, img, tr = 0;
! 138:
! 139: while ((ch = getopt(argc, argv, "hvts:f:")) != -1)
! 140: switch (ch) {
! 141: case 'f':
! 142: strlcpy(imgName, optarg, sizeof imgName);
! 143: break;
! 144: case 's':
! 145: imgSize = strtoll(optarg, NULL, 0);
! 146: if (!imgSize) {
! 147: Usage();
! 148: return 1;
! 149: }
! 150: break;
! 151: case 't':
! 152: tr = O_TRUNC;
! 153: break;
! 154: case 'v':
! 155: Verbose++;
! 156: break;
! 157: case 'h':
! 158: default:
! 159: Usage();
! 160: return 1;
! 161: }
! 162: argc -= optind;
! 163: argv += optind;
! 164:
! 165: if (argc) {
! 166: strlcpy(imgFile, *argv, sizeof imgFile);
! 167: /* open image file */
! 168: fd = open(imgFile, O_RDONLY);
! 169: if (fd == -1) {
! 170: ESYSERR(0);
! 171: return 2;
! 172: } else
! 173: iSize = lseek(fd, 0, SEEK_END);
! 174: if (!imgSize)
! 175: imgSize = E_ALIGN(iSize, IMGBUF_SIZE);
! 176: if (iSize == -1 || iSize > imgSize) {
! 177: close(fd);
! 178: EERROR(ENOSPC, "Error:: file size %llu is greater from storage size %llu\n",
! 179: iSize, imgSize);
! 180: return 2;
! 181: } else
! 182: lseek(fd, 0, SEEK_SET);
! 183: } else if (!imgSize) {
! 184: Usage();
! 185: return 1;
! 186: } else
! 187: fd = STDIN_FILENO;
! 188:
! 189: VERB(1) printf("imgSize=%llu imgName=%s imgFile=%s\n",
! 190: imgSize, imgName, argc ? imgFile : "<stdin>");
! 191:
! 192: /* open storage device */
! 193: img = open(imgName, O_RDWR | O_CREAT | tr, 0644);
! 194: if (img == -1) {
! 195: ESYSERR(0);
! 196: if (fd > 2)
! 197: close(fd);
! 198: return 3;
! 199: }
! 200: if (EmptyStore(img) == -1) {
! 201: if (fd > 2)
! 202: close(fd);
! 203: close(img);
! 204: unlink(imgName);
! 205: return 3;
! 206: }
! 207:
! 208: if (FillStore(img, fd) == -1) {
! 209: if (fd > 2)
! 210: close(fd);
! 211: close(img);
! 212: unlink(imgName);
! 213: return 4;
! 214: }
! 215:
! 216: close(img);
! 217: if (fd > 2)
! 218: close(fd);
! 219: return 0;
! 220: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>