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
use crate::{options::ModuleGeneratorOptions, utils::random_string};
use move_binary_format::file_format::{Bytecode, CompiledModule, Signature};
use move_core_types::{account_address::AccountAddress, identifier::Identifier};
use rand::{rngs::StdRng, Rng, SeedableRng};
pub struct Pad {
gen: StdRng,
table_size: usize,
options: ModuleGeneratorOptions,
}
impl Pad {
pub fn pad(table_size: usize, module: &mut CompiledModule, options: ModuleGeneratorOptions) {
let seed: [u8; 32] = [1; 32];
let mut slf = Self {
gen: StdRng::from_seed(seed),
table_size,
options,
};
slf.pad_cosntant_table(module);
slf.pad_identifier_table(module);
slf.pad_address_identifier_table(module);
slf.pad_signatures(module);
slf.pad_function_bodies(module);
}
fn pad_cosntant_table(&mut self, module: &mut CompiledModule) {
module.constant_pool = vec![]
}
fn pad_identifier_table(&mut self, module: &mut CompiledModule) {
module.identifiers = (0..(self.table_size + module.identifiers.len()))
.map(|_| {
let len = self.gen.gen_range(10..self.options.max_string_size);
Identifier::new(random_string(&mut self.gen, len)).unwrap()
})
.collect()
}
fn pad_address_identifier_table(&mut self, module: &mut CompiledModule) {
module.address_identifiers = (0..(self.table_size + module.address_identifiers.len()))
.map(|_| AccountAddress::random())
.collect()
}
fn pad_function_bodies(&mut self, module: &mut CompiledModule) {
for fdef in module.function_defs.iter_mut() {
if let Some(code) = &mut fdef.code {
code.code = vec![
Bytecode::LdTrue,
Bytecode::LdTrue,
Bytecode::Pop,
Bytecode::Pop,
Bytecode::Ret,
]
}
}
}
fn pad_signatures(&mut self, module: &mut CompiledModule) {
if module.signatures.iter().all(|v| !v.is_empty()) {
module.signatures.push(Signature(Vec::new()));
}
}
}