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
|
#!/usr/bin/nickle
import File;
string timed_read(file f, int timeout) {
thread reader = fork func() {
try {
return fgets(f);
} catch Thread::signal(int i) {
return "";
}
}();
thread killer = fork func() {
try {
sleep (timeout);
Thread::send_signal(reader, 1);
} catch Thread::signal(int i) {
return;
}
}();
poly v = Thread::join(reader);
Thread::send_signal(killer, 1);
Thread::join(killer);
if (is_string(v))
return v;
return "";
}
void flush_input(file f) {
for (;;) {
string s = timed_read(f, 200);
if (s == "")
break;
}
}
string[*] flash(file f) {
string[...] x = {};
flush_input(f);
fprintf (f, "f\nv\n");
flush(f);
for (;;) {
string l = timed_read(f, 1000);
if (l == "") {
File::fprintf(stderr, "Read timedout\n");
exit(1);
}
x[dim(x)] = l;
if (String::index(l, "software-version") == 0)
break;
}
return x;
}
string[*] find_flash(string[*] s, string match) {
for (int i = 0; i < dim(s); i++)
if (String::index(s[i], match) >= 0)
return String::wordsplit(s[i], " ");
return (string[*]) {};
}
bool
do_flash(file f, int expected_size) {
string[*] i = flash(f);
string[*] size = find_flash(i, "Storage size:");
string[*] erase = find_flash(i, "Storage erase unit:");
int actual_size = string_to_integer(size[2]);
if (actual_size != expected_size) {
printf ("weird flash size %d != %d\n", actual_size, expected_size);
return false;
}
int actual_erase = string_to_integer(erase[3]);
if (actual_erase != 65536) {
printf ("weird erase size %d\n", actual_erase);
return false;
}
printf ("flash size %d erase block %d\n", actual_size, actual_erase);
return true;
}
void main () {
string name = argv[1];
string size = argv[2];
file f = open(name, "r+");
bool ret = true;
if (!do_flash(f, string_to_integer(size)))
ret = false;
exit (ret? 0 : 1);
}
main();
|
