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
|
/* Copyright 2020 Mozilla Foundation
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#[no_mangle]
pub unsafe extern "C" fn wasm_text_to_binary(
text: *const u16,
text_len: usize,
out_bytes: *mut *mut u8,
out_bytes_len: *mut usize,
out_error: *mut *mut u8,
out_error_len: *mut usize,
) -> bool {
let text_slice = std::slice::from_raw_parts(text, text_len);
let text = String::from_utf16_lossy(text_slice);
match text_to_binary(&text) {
Ok(bytes) => {
let bytes_box = bytes.into_boxed_slice();
let bytes_slice = Box::leak(bytes_box);
out_bytes.write(bytes_slice.as_mut_ptr());
out_bytes_len.write(bytes_slice.len());
true
}
Err(error) => {
let error = Box::leak(format!("{}\0", error).into_boxed_str());
out_error.write(error.as_mut_ptr());
out_error_len.write(error.len());
false
}
}
}
fn text_to_binary(text: &str) -> Result<Vec<u8>, wast::Error> {
let mut lexer = wast::lexer::Lexer::new(text);
// The 'names.wast' spec test has confusable unicode, so disable detection.
// This protection is not very useful for a shell testing function anyways.
lexer.allow_confusing_unicode(true);
let buf = wast::parser::ParseBuffer::new_with_lexer(lexer)?;
let mut ast = wast::parser::parse::<wast::Wat>(&buf)?;
return ast.encode();
}
#[no_mangle]
pub unsafe extern "C" fn wasm_code_offsets(
bytes: *const u8,
bytes_len: usize,
out_offsets: *mut *mut u32,
out_offsets_len: *mut usize,
) {
fn code_offsets(bytes: &[u8]) -> Vec<u32> {
use wasmparser::*;
if bytes.is_empty() {
return Vec::new();
}
let mut offsets = Vec::new();
// Read operators offsets and skip invalid data.
for payload in Parser::new(0).parse_all(bytes) {
if payload.is_err() {
break;
}
match payload.unwrap() {
Payload::CodeSectionEntry(body) => {
let reader = match body.get_operators_reader() {
Ok(r) => r,
Err(_) => {
break;
}
};
for pair in reader.into_iter_with_offsets() {
let offset = match pair {
Ok((_op, offset)) => offset,
Err(_) => {
break;
}
};
offsets.push(offset as u32);
}
}
_ => (),
}
}
offsets
}
let bytes = std::slice::from_raw_parts(bytes, bytes_len);
let offsets = code_offsets(bytes);
if offsets.len() == 0 {
out_offsets.write(std::ptr::null_mut());
out_offsets_len.write(0);
} else {
let offsets_box = offsets.into_boxed_slice();
let offsets_slice = Box::leak(offsets_box);
out_offsets.write(offsets_slice.as_mut_ptr());
out_offsets_len.write(offsets_slice.len());
}
}
|
