Allow to get the current tx

neo-project · Jan 30, 2020 · 8985184 · 8985184
1 parent 6521582
commit 8985184
Showing 1 changed file with 19 additions and 2 deletions.
diff --git a/src/neo/SmartContract/InteropService.Blockchain.cs b/src/neo/SmartContract/InteropService.Blockchain.cs
@@ -45,8 +45,25 @@ private static bool Blockchain_GetBlock(ApplicationEngine engine)
 
             private static bool Blockchain_GetTransaction(ApplicationEngine engine)
             {
-                ReadOnlySpan<byte> hash = engine.CurrentContext.EvaluationStack.Pop().GetSpan();
-                Transaction tx = engine.Snapshot.GetTransaction(new UInt256(hash));
+                Transaction tx;
+                StackItem item = engine.CurrentContext.EvaluationStack.Pop();
+
+                if (item.IsNull)
+                {
+                    if (engine.ScriptContainer is Transaction senderTx)
+                    {
+                        tx = senderTx;
+                    }
+                    else
+                    {
+                        return false;
+                    }
+                }
+                else
+                {
+                    tx = engine.Snapshot.GetTransaction(new UInt256(item.GetSpan()));
+                }
+
                 if (tx == null)
                     engine.CurrentContext.EvaluationStack.Push(StackItem.Null);
                 else