diff options
Diffstat (limited to 'node/stack.go')
-rw-r--r-- | node/stack.go | 32 |
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 +} |