1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27#include "libbb.h"
28#include <mtd/mtd-user.h>
29
30
31#define MTD_DEBUG 0
32
33#define OPT_v (1 << 0)
34
35#define BUFSIZE (4 * 1024)
36
37static void progress(int mode, uoff_t count, uoff_t total)
38{
39 uoff_t percent;
40
41 if (!option_mask32)
42 return;
43 percent = count * 100;
44 if (total)
45 percent = (unsigned) (percent / total);
46 printf("\r%s: %"OFF_FMT"u/%"OFF_FMT"u (%u%%) ",
47 (mode < 0) ? "Erasing block" : ((mode == 0) ? "Writing kb" : "Verifying kb"),
48 count, total, (unsigned)percent);
49 fflush_all();
50}
51
52static void progress_newline(void)
53{
54 if (!option_mask32)
55 return;
56 bb_putchar('\n');
57}
58
59int flashcp_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
60int flashcp_main(int argc UNUSED_PARAM, char **argv)
61{
62 int fd_f, fd_d;
63 int i;
64 uoff_t erase_count;
65 struct mtd_info_user mtd;
66 struct erase_info_user e;
67 struct stat statb;
68
69 RESERVE_CONFIG_UBUFFER(buf, BUFSIZE);
70 RESERVE_CONFIG_UBUFFER(buf2, BUFSIZE);
71
72 getopt32(argv, "^" "v" "\0" "=2");
73 argv += optind;
74
75
76#define filename argv[0]
77#define devicename argv[1]
78
79
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
95 erase_count = (uoff_t)(statb.st_size + mtd.erasesize - 1) / mtd.erasesize;
96
97 e.length = mtd.erasesize;
98#if 0
99
100
101
102
103
104 if (!opts) {
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++) {
111 progress(-1, i, erase_count);
112#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
125
126 for (i = 0; i <= 1; i++) {
127 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) {
135 uoff_t rem;
136
137 progress(i, done / 1024, (uoff_t)statb.st_size / 1024);
138 rem = statb.st_size - done;
139 if (rem == 0)
140 break;
141 if (rem < BUFSIZE)
142 count = rem;
143 xread(fd_f, buf, count);
144 if (i == 0) {
145 int ret;
146 if (count < BUFSIZE)
147 memset((char*)buf + count, 0, BUFSIZE - count);
148 errno = 0;
149 ret = full_write(fd_d, buf, BUFSIZE);
150 if (ret != BUFSIZE) {
151 bb_perror_msg_and_die("write error at 0x%"OFF_FMT"x on %s, "
152 "write returned %d",
153 done, devicename, ret);
154 }
155 } else {
156 xread(fd_d, buf2, count);
157 if (memcmp(buf, buf2, count) != 0) {
158 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
168
169 return EXIT_SUCCESS;
170}
171