feat: impl tron's buggy origin opcode

Fix andelf/opentron#42
opentron · Nov 4, 2020 · bf4080d · bf4080d
1 parent a0b1d33
commit bf4080d
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/runtime/src/eval/system.rs b/runtime/src/eval/system.rs
@@ -52,7 +52,11 @@ pub fn selfbalance<H: Handler>(runtime: &mut Runtime, handler: &H) -> Control<H>
 }
 
 pub fn origin<H: Handler>(runtime: &mut Runtime, handler: &H) -> Control<H> {
-	let ret = H256::from(handler.origin());
+	// TRON: Use 21-byte address before allowMultisig.
+	let mut ret = H256::from(handler.origin());
+	if runtime._config.has_buggy_origin {
+		ret.as_bytes_mut()[11] = 0x41;
+	}
 	push!(runtime, ret);
 
 	Control::Continue

diff --git a/runtime/src/lib.rs b/runtime/src/lib.rs
@@ -217,6 +217,8 @@ pub struct Config {
 	pub has_token_issue: bool,
 	/// Has iswitness. (renamed from issrcandidate)
 	pub has_iswitness: bool,
+	/// Has a buggy origin opcode. (21 bytes origin address)
+	pub has_buggy_origin: bool,
 }
 
 impl Config {
@@ -305,6 +307,7 @@ impl Config {
 			has_stake: false,
 			has_token_issue: false,
 			has_iswitness: false,
+			has_buggy_origin: true,
 		}
 	}
 	/// Frontier hard fork configuration.
@@ -352,6 +355,7 @@ impl Config {
 			has_stake: false,
 			has_token_issue: false,
 			has_iswitness: false,
+			has_buggy_origin: false,
 		}
 	}
 
@@ -400,6 +404,7 @@ impl Config {
 			has_stake: false,
 			has_token_issue: false,
 			has_iswitness: false,
+			has_buggy_origin: false,
 		}
 	}
 }