save

gno.land/cmd/gnoland/testdata/alloc_array.txtar

@@ -11,6 +11,6 @@ package alloc
 var buffer interface{}
 
 func DoAlloc() {
-	var arr [1_000_000_000]byte
+	var arr [1_000_000_000_000_000]byte
 	buffer = arr
 }

gno.land/cmd/gnoland/testdata/alloc_byte_slice.txtar

@@ -11,6 +11,6 @@ package alloc
 var buffer []byte
 
 func DoAlloc() {
- 	buffer := make([]byte, 1_000_000_000)
+ 	buffer := make([]byte, 1_000_000_000_000)
   buffer[1] = 'a'
 }

gno.land/cmd/gnoland/testdata/alloc_slice.txtar

@@ -11,6 +11,6 @@ package alloc
 var buffer []int
 
 func DoAlloc() {
-    buffer := make([]int, 1_000_000_0)
+    buffer := make([]int, 1_000_000_000_000)
     buffer[1] = 1
 }

gnovm/pkg/gnolang/alloc.go

@@ -2,6 +2,7 @@ package gnolang
 
 import (
 	"reflect"
+	"runtime"
 )
 
 // Keeps track of in-memory allocations.
@@ -115,7 +116,9 @@ func (alloc *Allocator) Allocate(size int64) {
 		if alloc.heap != nil {
 			deleted := alloc.heap.MarkAndSweep()
 			alloc.DeallocDeleted(deleted)
-			if alloc.detonate && alloc.bytes > alloc.maxBytes {
+
+			ca := canAllocate(size)
+			if (alloc.detonate || ca) && alloc.bytes > alloc.maxBytes {
 				panic("allocation limit exceeded")
 			}
 			alloc.detonate = alloc.bytes > alloc.maxBytes
@@ -323,6 +326,15 @@ func (alloc *Allocator) NewDataArray(n int) *ArrayValue {
 	}
 }
 
+func canAllocate(size int64) bool {
+	var memStats runtime.MemStats
+	runtime.ReadMemStats(&memStats)
+
+	availableMemory := memStats.HeapIdle - memStats.HeapReleased
+
+	return uint64(size) < availableMemory
+}
+
 func (alloc *Allocator) NewArrayFromData(data []byte) *ArrayValue {
 	av := alloc.NewDataArray(len(data))
 	copy(av.Data, data)