blob: 4bbc67727cb3b70c6dfc2063f63e76b17631a70e [file] [log] [blame]
Stefan Seyfriedd095fd42009-11-21 18:32:19 +01001/* vi: set sw=4 ts=4: */
2/*
3 * busybox reimplementation of flashcp
4 *
5 * (C) 2009 Stefan Seyfried <seife@sphairon.com>
6 *
Denys Vlasenko0ef64bd2010-08-16 20:14:46 +02007 * Licensed under GPLv2, see file LICENSE in this source tree.
Stefan Seyfriedd095fd42009-11-21 18:32:19 +01008 */
Denys Vlasenkofb4da162016-11-22 23:14:24 +01009//config:config FLASHCP
10//config: bool "flashcp"
11//config: default n # doesn't build on Ubuntu 8.04
12//config: help
13//config: The flashcp binary, inspired by mtd-utils as of git head 5eceb74f7.
14//config: This utility is used to copy images into a MTD device.
Stefan Seyfriedd095fd42009-11-21 18:32:19 +010015
Denys Vlasenkof88e3bf2016-11-22 23:54:17 +010016//applet:IF_FLASHCP(APPLET(flashcp, BB_DIR_USR_SBIN, BB_SUID_DROP))
17
18//kbuild:lib-$(CONFIG_FLASHCP) += flashcp.o
19
Pere Orga5bc8c002011-04-11 03:29:49 +020020//usage:#define flashcp_trivial_usage
21//usage: "-v FILE MTD_DEVICE"
22//usage:#define flashcp_full_usage "\n\n"
23//usage: "Copy an image to MTD device\n"
Pere Orga5bc8c002011-04-11 03:29:49 +020024//usage: "\n -v Verbose"
25
Stefan Seyfriedd095fd42009-11-21 18:32:19 +010026#include "libbb.h"
27#include <mtd/mtd-user.h>
28
Denys Vlasenko243e7332013-01-23 11:41:22 +010029/* If 1, simulates "flashing" by writing to existing regular file */
Stefan Seyfriedd095fd42009-11-21 18:32:19 +010030#define MTD_DEBUG 0
31
32#define OPT_v (1 << 0)
33
Jacob Kjaergaardbd7c1f22014-12-10 13:44:27 +010034#define BUFSIZE (4 * 1024)
Stefan Seyfriedd095fd42009-11-21 18:32:19 +010035
36static void progress(int mode, uoff_t count, uoff_t total)
37{
38 uoff_t percent;
39
40 if (!option_mask32) //if (!(option_mask32 & OPT_v))
41 return;
42 percent = count * 100;
43 if (total)
44 percent = (unsigned) (percent / total);
45 printf("\r%s: %"OFF_FMT"u/%"OFF_FMT"u (%u%%) ",
Denys Vlasenko6be6f3b2013-01-28 12:26:29 +010046 (mode < 0) ? "Erasing block" : ((mode == 0) ? "Writing kb" : "Verifying kb"),
Stefan Seyfriedd095fd42009-11-21 18:32:19 +010047 count, total, (unsigned)percent);
48 fflush_all();
49}
50
51static void progress_newline(void)
52{
53 if (!option_mask32) //if (!(option_mask32 & OPT_v))
54 return;
55 bb_putchar('\n');
56}
57
58int flashcp_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
59int flashcp_main(int argc UNUSED_PARAM, char **argv)
60{
61 int fd_f, fd_d; /* input file and mtd device file descriptors */
62 int i;
63 uoff_t erase_count;
Stefan Seyfriedd095fd42009-11-21 18:32:19 +010064 struct mtd_info_user mtd;
65 struct erase_info_user e;
66 struct stat statb;
67// const char *filename, *devicename;
68 RESERVE_CONFIG_UBUFFER(buf, BUFSIZE);
69 RESERVE_CONFIG_UBUFFER(buf2, BUFSIZE);
70
71 opt_complementary = "=2"; /* exactly 2 non-option args: file, dev */
Manuel Zerpiesd1097982012-09-27 16:06:34 +020072 /*opts =*/ getopt32(argv, "v");
Stefan Seyfriedd095fd42009-11-21 18:32:19 +010073 argv += optind;
74// filename = *argv++;
75// devicename = *argv;
76#define filename argv[0]
77#define devicename argv[1]
78
79 /* open input file and mtd device and do sanity checks */
80 fd_f = xopen(filename, O_RDONLY);
81 fstat(fd_f, &statb);
82 fd_d = xopen(devicename, O_SYNC | O_RDWR);
83#if !MTD_DEBUG
84 if (ioctl(fd_d, MEMGETINFO, &mtd) < 0) {
85 bb_error_msg_and_die("%s is not a MTD flash device", devicename);
86 }
87 if (statb.st_size > mtd.size) {
88 bb_error_msg_and_die("%s bigger than %s", filename, devicename);
89 }
90#else
91 mtd.erasesize = 64 * 1024;
92#endif
93
94 /* always erase a complete block */
95 erase_count = (uoff_t)(statb.st_size + mtd.erasesize - 1) / mtd.erasesize;
96 /* erase 1 block at a time to be able to give verbose output */
97 e.length = mtd.erasesize;
98#if 0
99/* (1) bloat
100 * (2) will it work for multi-gigabyte devices?
101 * (3) worse wrt error detection granularity
102 */
103 /* optimization: if not verbose, erase in one go */
104 if (!opts) { // if (!(opts & OPT_v))
105 e.length = mtd.erasesize * erase_count;
106 erase_count = 1;
107 }
108#endif
109 e.start = 0;
110 for (i = 1; i <= erase_count; i++) {
Denys Vlasenko243e7332013-01-23 11:41:22 +0100111 progress(-1, i, erase_count);
Stefan Seyfriedd095fd42009-11-21 18:32:19 +0100112#if !MTD_DEBUG
113 if (ioctl(fd_d, MEMERASE, &e) < 0) {
114 bb_perror_msg_and_die("erase error at 0x%llx on %s",
115 (long long)e.start, devicename);
116 }
117#else
118 usleep(100*1000);
119#endif
120 e.start += mtd.erasesize;
121 }
122 progress_newline();
123
124 /* doing this outer loop gives significantly smaller code
125 * than doing two separate loops for writing and verifying */
Denys Vlasenko243e7332013-01-23 11:41:22 +0100126 for (i = 0; i <= 1; i++) {
Stefan Seyfriedd095fd42009-11-21 18:32:19 +0100127 uoff_t done;
128 unsigned count;
129
130 xlseek(fd_f, 0, SEEK_SET);
131 xlseek(fd_d, 0, SEEK_SET);
132 done = 0;
133 count = BUFSIZE;
134 while (1) {
Denys Vlasenko243e7332013-01-23 11:41:22 +0100135 uoff_t rem;
136
137 progress(i, done / 1024, (uoff_t)statb.st_size / 1024);
138 rem = statb.st_size - done;
Stefan Seyfriedd095fd42009-11-21 18:32:19 +0100139 if (rem == 0)
140 break;
141 if (rem < BUFSIZE)
142 count = rem;
Stefan Seyfriedd095fd42009-11-21 18:32:19 +0100143 xread(fd_f, buf, count);
Denys Vlasenko243e7332013-01-23 11:41:22 +0100144 if (i == 0) {
Stefan Seyfriedd095fd42009-11-21 18:32:19 +0100145 int ret;
Denys Vlasenko243e7332013-01-23 11:41:22 +0100146 if (count < BUFSIZE)
147 memset((char*)buf + count, 0, BUFSIZE - count);
Stefan Seyfriedd095fd42009-11-21 18:32:19 +0100148 errno = 0;
Denys Vlasenko243e7332013-01-23 11:41:22 +0100149 ret = full_write(fd_d, buf, BUFSIZE);
150 if (ret != BUFSIZE) {
Stefan Seyfriedd095fd42009-11-21 18:32:19 +0100151 bb_perror_msg_and_die("write error at 0x%"OFF_FMT"x on %s, "
152 "write returned %d",
153 done, devicename, ret);
154 }
Denys Vlasenko243e7332013-01-23 11:41:22 +0100155 } else { /* i == 1 */
Stefan Seyfriedd095fd42009-11-21 18:32:19 +0100156 xread(fd_d, buf2, count);
Denys Vlasenko243e7332013-01-23 11:41:22 +0100157 if (memcmp(buf, buf2, count) != 0) {
Stefan Seyfriedd095fd42009-11-21 18:32:19 +0100158 bb_error_msg_and_die("verification mismatch at 0x%"OFF_FMT"x", done);
159 }
160 }
161
162 done += count;
163 }
164
165 progress_newline();
166 }
167 /* we won't come here if there was an error */
168
169 return EXIT_SUCCESS;
170}