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

// Copyright 2015-2020 Parity Technologies (UK) Ltd.
// This file is part of Tetsy Vapory.

// Tetsy Vapory is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.

// Tetsy Vapory is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with Tetsy Vapory.  If not, see <http://www.gnu.org/licenses/>.

use tetsy_trie_db::TrieFactory;
use vaptrie::Layout;
use account_db::Factory as AccountFactory;
use vvm::{Factory as VvmFactory};
use tetsy_vm::{Exec, ActionParams, VersionedSchedule, Schedule};
use vapcore_wasm::WasmInterpreter;

const WASM_MAGIC_NUMBER: &'static [u8; 4] = b"\0asm";

/// Virtual machine factory
#[derive(Default, Clone)]
pub struct VmFactory {
	vvm: VvmFactory,
}

impl VmFactory {
	pub fn create(&self, params: ActionParams, schedule: &Schedule, depth: usize) -> Option<Box<dyn Exec>> {
		if params.code_version.is_zero() {
			Some(if schedule.wasm.is_some() && schedule.versions.is_empty() && params.code.as_ref().map_or(false, |code| code.len() > 4 && &code[0..4] == WASM_MAGIC_NUMBER) {
				Box::new(WasmInterpreter::new(params))
			} else {
				self.vvm.create(params, schedule, depth)
			})
		} else {
			let version_config = schedule.versions.get(&params.code_version);

			match version_config {
				Some(VersionedSchedule::TWasm) => {
					Some(Box::new(WasmInterpreter::new(params)))
				},
				None => None,
			}
		}
	}

	pub fn new(cache_size: usize) -> Self {
		VmFactory { vvm: VvmFactory::new(cache_size) }
	}
}

impl From<VvmFactory> for VmFactory {
	fn from(vvm: VvmFactory) -> Self {
		VmFactory { vvm }
	}
}

/// Collection of factories.
#[derive(Default, Clone)]
pub struct Factories {
	/// factory for vvm.
	pub vm: VmFactory,
	/// factory for tries.
	pub trie: TrieFactory<Layout>,
	/// factory for account databases.
	pub accountdb: AccountFactory,
}