-
Notifications
You must be signed in to change notification settings - Fork 2
/
Cmd_Flash.cpp
142 lines (110 loc) · 3.91 KB
/
Cmd_Flash.cpp
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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
/*
This file is part of PICoBoot Utility.
Copyright (C) 2021 ReimuNotMoe <[email protected]>
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU Affero General Public License as
published by the Free Software Foundation, either version 3 of the
License, or (at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU Affero General Public License for more details.
You should have received a copy of the GNU Affero General Public License
along with this program. If not, see <https://www.gnu.org/licenses/>.
*/
#include "PICoBoot_Utility.hpp"
#include <chrono>
int Command_Flash(const std::vector<std::string>& args) {
if (args.size() != 2) {
puts("error: no file specified");
return 2;
}
auto &filepath = args[1];
if (global_device_path.empty()) {
auto paths = ScanDevice_GetValidDevices();
if (paths.size() > 1) {
puts("More than one devices connected. Please specify one using the `-d' option.");
return 1;
} else {
global_device_path = paths[0];
}
}
Serial tty;
tty.open(global_device_path);
#ifdef __CYGWIN__
try {
#endif
tty.make_raw();
#ifdef __CYGWIN__
} catch (...) {
}
#endif
std::ifstream infile(filepath);
if (!infile.good()) {
printf("error: unable to open file `%s'\n", filepath.c_str());
return 2;
}
infile.seekg(0, std::ios_base::end);
size_t infile_size = infile.tellg();
infile.seekg(0, std::ios_base::beg);
size_t processed_size = 0;
size_t byte_count = 0;
std::string line;
iHexPP::Decoder dec;
uint32_t last_addr = 0;
uint8_t cnt = 0;
dec.read_callback = [&](uint8_t record_type, uint32_t addr, const uint8_t *data, size_t len){
if (record_type == iHexPP::IHEX_DATA_RECORD) {
last_addr = addr;
auto rc_addr = serial_chat(tty, FlasherCommand_FlashSetAddr, &addr);
if (*((uint32_t *)rc_addr.data()) != addr) {
throw std::logic_error("address verification failed!");
}
if (cnt % 16 == 0) {
printf("\033[2K\rFlashing ... [0x%08x+%zu] %zu/%zu %.2f%%", addr, len, processed_size, infile_size, ((float) processed_size / infile_size) * 100);
fflush(stdout);
}
uint8_t buf[32];
uint8_t rc;
if (len == 8) {
memcpy(buf, data, 8);
buf[8] = crc8(buf, 8);
auto rcv = serial_chat(tty, FlasherCommand_FlashWrite8, buf);
rc = *((uint8_t *)rcv.data());
} else if (len == 16) {
memcpy(buf, data, 16);
buf[16] = crc8(buf, 16);
auto rcv = serial_chat(tty, FlasherCommand_FlashWrite16, buf);
rc = *((uint8_t *)rcv.data());
} else {
for (size_t i=0; i<len; i+=4) {
memcpy(buf, data + i, 4);
buf[4] = crc8(buf, 4);
auto rcv = serial_chat(tty, FlasherCommand_FlashWrite4, buf);
rc = *((uint8_t *)rcv.data());
if (rc != FlasherResult_OK) {
if (rc == FlasherResult_ERANGE) {
printf("\nnotice: writes to flash addr 0x%08x (0x%08x) rejected by bootloader.\n", addr, addr / 2);
printf("Maybe you're writing the configuration bits region?\n");
} else {
printf("\nerror: failed to write flash addr 0x%08x (0x%08x): %s\n", addr, addr / 2, FlasherResult_strerror(rc));
abort();
}
}
}
}
byte_count += len;
cnt++;
}
};
auto time_start = std::chrono::high_resolution_clock::now();
while (std::getline(infile, line)) {
dec.read(line);
processed_size += line.size();
}
auto time_end = std::chrono::high_resolution_clock::now();
auto duration = (double)std::chrono::duration_cast<std::chrono::microseconds>(time_end - time_start).count() / 1000000;
printf("\033[2K\rFlashing ... [0x%08x] %zu/%zu %.2f%%\n", last_addr, infile_size, infile_size, (float)100);
printf("%zu bytes written in %.3f seconds, %zu bytes/sec\n", byte_count, duration, (size_t)(byte_count / duration));
return 0;
}