![[BACK]](/icons/cvsweb/back.gif){kind=icon}
Return to imgupd.c CVS log ![[TXT]](/icons/cvsweb/text.gif){kind=icon}
![[DIR]](/icons/cvsweb/dir.gif){kind=icon}
Up to [ELWIX - Embedded LightWeight unIX -] / embedtools / src|
Return to imgupd.c CVS log | Up to [ELWIX - Embedded LightWeight unIX -] / embedtools / src |
1.2 misho 1: /*************************************************************************
2: * (C) 2014 AITNET - Sofia/Bulgaria - <office@aitbg.com>
3: * by Michael Pounov <misho@aitbg.com>
4: *
5: * $Author: misho $
1.7 ! misho 6: * $Id: imgupd.c,v 1.6.2.7 2014/02/06 15:17:58 misho Exp $
1.2 misho 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"
1.7 ! misho 47: #include "imgupd.h"
1.2 misho 48:
49:
50: char imgName[PATH_MAX], imgFile[PATH_MAX];
1.7 ! misho 51: off_t imgSize, iSize, fromBegin;
! 52: int Verbose, nameFlg, fileFlg, bufSize = IMGBUF_SIZE;
1.2 misho 53: extern char compiled[], compiledby[], compilehost[];
54:
55: static void
56: Usage()
57: {
58:
59: printf( "IMGUPD is tool for management of images\n"
60: "=== %s === %s@%s ===\n\n"
1.3 misho 61: " Syntax: imgupd [options] [image_file]\n\n"
1.2 misho 62: "\t-v\t\tVerbose ...\n"
1.4 misho 63: "\t-R\t\tReboot system after complete\n"
1.7 ! misho 64: "\t-i\t\tStart fill storage from begin\n"
1.3 misho 65: "\t-g\t\tGet image from Storage\n"
1.2 misho 66: "\t-t\t\tTruncate Storage file name\n"
67: "\t-s <size>\tStorage size (required for stdin)\n"
68: "\t-f <devfile>\tStorage file name\n"
69: "\n", compiled, compiledby, compilehost);
70: }
71:
72: static int
1.7 ! misho 73: getFDtype(int fd)
! 74: {
! 75: int type, flg = 0;
! 76: struct stat sb;
! 77:
! 78: if (fstat(fd, &sb) == -1) {
! 79: ESYSERR(0);
! 80: return -1;
! 81: }
! 82: if (S_ISREG(sb.st_mode))
! 83: flg |= FD_TRUNC;
! 84: if (S_ISCHR(sb.st_mode) || S_ISBLK(sb.st_mode)) {
! 85: if (ioctl(fd, FIODTYPE, &type) == -1) {
! 86: ESYSERR(0);
! 87: return -1;
! 88: }
! 89: if (type & D_TAPE)
! 90: flg |= FD_TAPE;
! 91: else if (type & (D_DISK | D_MEM))
! 92: flg |= FD_SEEK;
! 93: if (S_ISCHR(sb.st_mode) && !(type & D_TAPE))
! 94: flg |= FD_CHR;
! 95: goto end;
! 96: }
! 97: errno ^= errno;
! 98: if (lseek(fd, 0, SEEK_CUR) == -1 && errno == ESPIPE)
! 99: flg |= FD_PIPE;
! 100: else
! 101: flg |= FD_SEEK;
! 102: end:
! 103: return flg;
! 104: }
! 105:
! 106: static int
! 107: EmptyStore(int img, int flg)
1.2 misho 108: {
109: register int i;
1.4 misho 110: u_char buf[bufSize];
1.2 misho 111: ssize_t wlen;
112:
113: VERB(1) printf("Erase store %s\n", imgName);
114:
1.7 ! misho 115: if (!fromBegin && flg & FD_SEEK) {
! 116: iSize = lseek(img, 0, SEEK_END);
! 117: if (iSize == -1) {
! 118: ESYSERR(0);
! 119: return -1;
! 120: } else
! 121: imgSize += E_ALIGN(iSize, bufSize);
1.2 misho 122: } else
1.7 ! misho 123: iSize ^= iSize;
1.2 misho 124:
125: memset(buf, 0, sizeof buf);
1.4 misho 126: for (i = howmany(iSize, bufSize); i < howmany(imgSize, bufSize); i++)
1.5 misho 127: if ((wlen = write(img, buf, bufSize)) == -1 ||
128: (wlen && wlen != bufSize)) {
1.7 ! misho 129: if (wlen == -1)
! 130: ESYSERR(0);
! 131: else
! 132: EERROR(EIO, "Error at chunk %d init %d bytes, "
! 133: "should be %u\n", i, wlen, bufSize);
1.2 misho 134: return -1;
135: } else
136: VERB(1) printf("+Written chunk #%d\n", i);
137:
1.7 ! misho 138: if (flg & FD_SEEK)
! 139: iSize = lseek(img, iSize, SEEK_SET);
1.2 misho 140: return iSize;
141: }
142:
143: static int
1.7 ! misho 144: FillStore(int img, int iflg, int fd, int fflg)
1.2 misho 145: {
146: register int i, j;
1.7 ! misho 147: u_char *pos, buf[bufSize];
! 148: ssize_t rlen, wlen, len;
1.2 misho 149:
150: VERB(1) printf("Fill store %s from image file %s\n", imgName, imgFile);
151:
1.4 misho 152: for (j = 0, i = howmany(iSize, bufSize); i < howmany(imgSize, bufSize);
1.2 misho 153: i++, j++) {
154: memset(buf, 0, sizeof buf);
1.7 ! misho 155: for (len = bufSize, pos = buf; len > 0; len -= rlen, pos += rlen) {
! 156: rlen = read(fd, pos, len);
! 157: if (rlen == -1) {
! 158: ESYSERR(0);
! 159: return -1;
! 160: } else if (!rlen)
! 161: break;
! 162: else
! 163: VERB(1) printf("+Readed %d bytes for chunk #%d\n", rlen, j);
! 164: }
1.2 misho 165:
1.7 ! misho 166: wlen = write(img, buf, bufSize);
1.2 misho 167: if (wlen == -1) {
168: ESYSERR(0);
169: return -1;
1.7 ! misho 170: } else if (wlen && wlen != bufSize) {
! 171: EERROR(EIO, "Error at chunk %d written %d bytes, "
! 172: "should be %u\n", i, wlen, bufSize);
1.2 misho 173: } else
174: VERB(1) printf("+Written %d bytes at chunk #%d\n", wlen, i);
175: }
176:
177: return 0;
178: }
179:
180: int
181: main(int argc, char **argv)
182: {
1.4 misho 183: char ch, m = 0, R = 0;
1.2 misho 184: int fd, img, tr = 0;
185:
1.7 ! misho 186: while ((ch = getopt(argc, argv, "hvRigts:f:")) != -1)
1.2 misho 187: switch (ch) {
188: case 'f':
189: strlcpy(imgName, optarg, sizeof imgName);
190: break;
191: case 's':
192: imgSize = strtoll(optarg, NULL, 0);
193: if (!imgSize) {
194: Usage();
195: return 1;
196: }
197: break;
198: case 't':
199: tr = O_TRUNC;
200: break;
1.3 misho 201: case 'g':
202: m = 1;
1.7 ! misho 203: case 'i':
! 204: fromBegin = 1;
1.4 misho 205: break;
206: case 'R':
207: R = 1;
208: break;
1.2 misho 209: case 'v':
210: Verbose++;
211: break;
212: case 'h':
213: default:
214: Usage();
215: return 1;
216: }
217: argc -= optind;
218: argv += optind;
219:
1.3 misho 220: if (!m) {
221: if (argc) {
222: strlcpy(imgFile, *argv, sizeof imgFile);
223: /* open image file */
224: fd = open(imgFile, O_RDONLY);
225: if (fd == -1) {
226: ESYSERR(0);
227: return 2;
228: } else
1.7 ! misho 229: fileFlg = getFDtype(fd);
! 230:
! 231: if (fileFlg & FD_SEEK) {
1.3 misho 232: iSize = lseek(fd, 0, SEEK_END);
1.7 ! misho 233: if (!imgSize)
! 234: imgSize = E_ALIGN(iSize, bufSize);
! 235: if (iSize == -1 || iSize > imgSize) {
! 236: close(fd);
! 237: EERROR(ENOSPC, "Error:: file size %llu is "
! 238: "greater from storage size %llu\n",
! 239: iSize, imgSize);
! 240: return 2;
! 241: } else
! 242: lseek(fd, 0, SEEK_SET);
! 243: }
! 244: } else {
! 245: fd = STDIN_FILENO;
! 246: fileFlg = getFDtype(fd);
! 247: }
! 248:
! 249: if (!imgSize) {
! 250: if (fd > 2)
1.3 misho 251: close(fd);
252: Usage();
253: return 1;
1.7 ! misho 254: }
1.3 misho 255: } else { /* GET */
256: if (argc) {
257: strlcpy(imgFile, *argv, sizeof imgFile);
258: /* open image file */
259: fd = open(imgFile, O_WRONLY | O_TRUNC | O_CREAT, 0644);
260: if (fd == -1) {
261: ESYSERR(0);
262: return 2;
1.7 ! misho 263: } else
! 264: fileFlg = getFDtype(fd);
! 265: } else {
1.3 misho 266: fd = STDOUT_FILENO;
1.7 ! misho 267: fileFlg = getFDtype(fd);
! 268: }
1.3 misho 269: }
270:
271: VERB(1) printf("imgSize=%llu imgName=%s imgFile=%s\n",
272: imgSize, imgName, argc ? imgFile : "<stdin>");
273:
274: if (!m) {
275: /* open storage device */
276: img = open(imgName, O_RDWR | O_CREAT | tr, 0644);
277: if (img == -1) {
1.2 misho 278: ESYSERR(0);
1.3 misho 279: if (fd > 2)
280: close(fd);
281: return 3;
1.7 ! misho 282: } else
! 283: nameFlg = getFDtype(img);
1.3 misho 284: } else { /* GET */
285: /* open storage device */
286: img = open(imgName, O_RDONLY);
287: if (img == -1) {
288: ESYSERR(0);
289: if (fd > 2)
290: close(fd);
291: return 3;
1.2 misho 292: } else
1.7 ! misho 293: nameFlg = getFDtype(img);
! 294:
! 295: if (nameFlg & FD_SEEK) {
1.3 misho 296: iSize = lseek(img, 0, SEEK_END);
1.7 ! misho 297: if (!imgSize)
! 298: imgSize = E_ALIGN(iSize, bufSize);
! 299: if (iSize == -1 || iSize > imgSize) {
! 300: if (fd > 2)
! 301: close(fd);
! 302: close(img);
! 303: EERROR(ENOSPC, "Error:: storage size %llu is "
! 304: "greater from file size %llu\n",
! 305: iSize, imgSize);
! 306: return 3;
! 307: } else
! 308: lseek(img, 0, SEEK_SET);
! 309: }
! 310:
! 311: if (!imgSize) {
1.3 misho 312: if (fd > 2)
313: close(fd);
1.7 ! misho 314: Usage();
! 315: return 1;
! 316: }
1.2 misho 317: }
318:
1.3 misho 319: if (!m) {
1.7 ! misho 320: if (EmptyStore(img, nameFlg) == -1) {
1.3 misho 321: if (fd > 2)
322: close(fd);
323: close(img);
324: return 3;
325: }
1.7 ! misho 326: if (FillStore(img, nameFlg, fd, fileFlg) == -1) {
1.3 misho 327: if (fd > 2)
328: close(fd);
329: close(img);
330: return 4;
331: }
332: } else { /* GET */
1.7 ! misho 333: iSize ^= iSize;
! 334: if (!(fileFlg & FD_PIPE))
! 335: if (EmptyStore(fd, fileFlg) == -1) {
! 336: if (fd > 2)
! 337: close(fd);
! 338: close(img);
! 339: return 3;
! 340: }
! 341: if (FillStore(fd, fileFlg, img, nameFlg) == -1) {
1.3 misho 342: if (fd > 2)
343: close(fd);
344: close(img);
345: return 4;
346: }
1.2 misho 347: }
348:
349: close(img);
350: if (fd > 2)
351: close(fd);
1.4 misho 352:
353: if (R)
354: reboot(RB_AUTOBOOT);
1.2 misho 355: return 0;
356: }