102 lines
1.8 KiB
Plaintext
Executable File
102 lines
1.8 KiB
Plaintext
Executable File
#!/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();
|