PEMap: initial capacity bounds check

cosminbasca · cosminbasca · commit 6701a16a5e7f · 2020-09-24T13:47:36.000+02:00
diff --git a/graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/objects/common/PEMap.java b/graalpython/com.oracle.graal.python/src/com/oracle/graal/python/builtins/objects/common/PEMap.java
@@ -146,7 +146,7 @@ private PEMap(int initialCapacity, boolean isSet, boolean hasSideEffect) {
 
     private void init(int size) {
         if (size > INITIAL_CAPACITY) {
-            entries = new Object[size << 1];
+            entries = new Object[(size << 1) < 0 ? MAX_ELEMENT_COUNT : size << 1];
         }
     }
 

Original file line number	Diff line number	Diff line change
`@@ -146,7 +146,7 @@ private PEMap(int initialCapacity, boolean isSet, boolean hasSideEffect) {`
`146`	`146`
`147`	`147`	`private void init(int size) {`
`148`	`148`	`if (size > INITIAL_CAPACITY) {`
`149`		`- entries = new Object[size << 1];`
	`149`	`+ entries = new Object[(size << 1) < 0 ? MAX_ELEMENT_COUNT : size << 1];`
`150`	`150`	`}`
`151`	`151`	`}`
`152`	`152`