8289365: SegmentAllocator:allocateArray(MemoryLayout, count) does not throw IAEx when count is -1

Reviewed-by: psandoz

Author: mcimadamore

src/java.base/share/classes/java/lang/foreign/SegmentAllocator.java

@@ -319,6 +319,9 @@ default MemorySegment allocate(MemoryLayout layout) {
      */
     default MemorySegment allocateArray(MemoryLayout elementLayout, long count) {
         Objects.requireNonNull(elementLayout);
+        if (count < 0) {
+            throw new IllegalArgumentException("Negative array size");
+        }
         return allocate(MemoryLayout.sequenceLayout(count, elementLayout));
     }
 

test/jdk/java/foreign/TestSegmentAllocators.java

@@ -160,6 +160,11 @@ public void testBadAllocationAlignNotPowerTwo(SegmentAllocator allocator) {
         allocator.allocate(1, 3);
     }
 
+    @Test(dataProvider = "allocators", expectedExceptions = IllegalArgumentException.class)
+    public void testBadAllocationArrayNegSize(SegmentAllocator allocator) {
+        allocator.allocateArray(ValueLayout.JAVA_BYTE, -1);
+    }
+
     @Test(expectedExceptions = OutOfMemoryError.class)
     public void testBadArenaNullReturn() {
         SegmentAllocator segmentAllocator = SegmentAllocator.newNativeArena(MemorySession.openImplicit());