stack

1 files changed, 32 insertions, 0 deletions

diff --git a/node/stack.go b/node/stack.go
new file mode 100644
index 0000000..a5460a7
--- /dev/null
+++ b/node/stack.go
@@ -0,0 +1,32 @@
+package main
+
+import (
+	"log"
+
+	"github.com/holiman/uint256"
+)
+
+const STACK_CAP = (1 << 10)
+
+type Stack []uint256.Int
+
+func NewStack() *Stack {
+	return &Stack{}
+}
+
+func (s *Stack) Push(x *uint256.Int) {
+	*s = append(*s, *x)
+	if len(*s) > STACK_CAP {
+		log.Fatal("stack overflow")
+	}
+}
+
+func (s *Stack) Pop() *uint256.Int {
+	if len(*s) <= 0 {
+		log.Fatal("stack underflow")
+	}
+
+	r := (*s)[len(*s)-1]
+	*s = (*s)[:len(*s)-1]
+	return &r
+}