blob: bf411c12148e295cc0f54546cde8533c13d8bf4a (
plain)
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
|
//! Disassembly calling function for `wasm32` targets.
use crate::Function;
use std::collections::HashSet;
pub(crate) fn disassemble_myself() -> HashSet<Function> {
// Use `std::env::args` to find the path to our executable. Assume the
// environment is configured such that we can read that file. Read it and
// use the `wasmprinter` crate to transform the binary to text, then search
// the text for appropriately named functions.
let me = std::env::args()
.next()
.expect("failed to find current wasm file");
let output = wasmprinter::print_file(&me).unwrap();
let mut ret: HashSet<Function> = HashSet::new();
let mut lines = output.lines().map(|s| s.trim());
while let Some(line) = lines.next() {
// If this isn't a function, we don't care about it.
if !line.starts_with("(func ") {
continue;
}
let mut function = Function {
name: String::new(),
instrs: Vec::new(),
};
// Empty functions will end in `))` so there's nothing to do, otherwise
// we'll have a bunch of following lines which are instructions.
//
// Lines that have an imbalanced `)` mark the end of a function.
if !line.ends_with("))") {
while let Some(line) = lines.next() {
function.instrs.push(line.to_string());
if !line.starts_with("(") && line.ends_with(")") {
break;
}
}
}
// The second element here split on whitespace should be the name of
// the function, skipping the type/params/results
function.name = line.split_whitespace().nth(1).unwrap().to_string();
if function.name.starts_with("$") {
function.name = function.name[1..].to_string()
}
if !function.name.contains("stdarch_test_shim") {
continue;
}
assert!(ret.insert(function));
}
return ret;
}
|