@@ -201,6 +201,9 @@ abstract static class IMulNode extends PythonBinaryClinicBuiltinNode {
201201 Object concat (PArray self , int value ) {
202202 try {
203203 int newLength = Math .max (PythonUtils .multiplyExact (self .getLength (), value ), 0 );
204+ if (newLength != self .getLength ()) {
205+ self .checkCanResize (this );
206+ }
204207 int itemsize = self .getFormat ().bytesize ;
205208 int segmentLength = self .getLength () * itemsize ;
206209 self .resize (newLength );
@@ -533,6 +536,7 @@ Object setitem(PArray self, PSlice slice, PArray other,
533536 }
534537 if (simpleStepProfile .profile (step == 1 )) {
535538 if (differentLengthProfile .profile (sliceLength != needed )) {
539+ self .checkCanResize (this );
536540 if (growProfile .profile (sliceLength < needed )) {
537541 if (stop < start ) {
538542 stop = start ;
@@ -579,19 +583,21 @@ abstract static class DelItemNode extends PythonBinaryBuiltinNode {
579583 public abstract Object executeSlice (PArray self , PSlice slice );
580584
581585 @ Specialization (guards = "!isPSlice(idx)" , limit = "3" )
582- static Object delitem (PArray self , Object idx ,
586+ Object delitem (PArray self , Object idx ,
583587 @ CachedLibrary ("idx" ) PythonObjectLibrary lib ,
584588 @ Cached ("forArrayAssign()" ) NormalizeIndexNode normalizeIndexNode ) {
589+ self .checkCanResize (this );
585590 int index = normalizeIndexNode .execute (lib .asIndex (idx ), self .getLength ());
586591 self .delSlice (index , 1 );
587592 return PNone .NONE ;
588593 }
589594
590595 @ Specialization
591- static Object delitem (PArray self , PSlice slice ,
596+ Object delitem (PArray self , PSlice slice ,
592597 @ Cached ConditionProfile simpleStepProfile ,
593598 @ Cached SliceLiteralNode .SliceUnpack sliceUnpack ,
594599 @ Cached SliceLiteralNode .AdjustIndices adjustIndices ) {
600+ self .checkCanResize (this );
595601 int length = self .getLength ();
596602 PSlice .SliceInfo sliceInfo = adjustIndices .execute (length , sliceUnpack .execute (slice ));
597603 int start = sliceInfo .start ;
@@ -713,6 +719,7 @@ Object append(VirtualFrame frame, PArray self, Object value,
713719 try {
714720 int index = self .getLength ();
715721 int newLength = PythonUtils .addExact (index , 1 );
722+ self .checkCanResize (this );
716723 self .resize (newLength );
717724 putValueNode .execute (frame , self , index , value );
718725 return PNone .NONE ;
@@ -730,6 +737,9 @@ abstract static class ExtendNode extends PythonBinaryBuiltinNode {
730737 Object extend (PArray self , PArray value ) {
731738 try {
732739 int newLength = PythonUtils .addExact (self .getLength (), value .getLength ());
740+ if (newLength != self .getLength ()) {
741+ self .checkCanResize (this );
742+ }
733743 int itemsize = self .getFormat ().bytesize ;
734744 self .resizeStorage (newLength );
735745 PythonUtils .arraycopy (value .getBuffer (), 0 , self .getBuffer (), self .getLength () * itemsize , value .getLength () * itemsize );
@@ -750,7 +760,11 @@ Object extend(VirtualFrame frame, PArray self, PSequence value,
750760 SequenceStorage storage = getSequenceStorageNode .execute (value );
751761 int storageLength = lenNode .execute (storage );
752762 try {
753- self .resizeStorage (PythonUtils .addExact (self .getLength (), storageLength ));
763+ int newLength = PythonUtils .addExact (self .getLength (), storageLength );
764+ if (newLength != self .getLength ()) {
765+ self .checkCanResize (this );
766+ }
767+ self .resizeStorage (newLength );
754768 } catch (OverflowException e ) {
755769 CompilerDirectives .transferToInterpreterAndInvalidate ();
756770 throw raise (MemoryError );
@@ -786,6 +800,7 @@ Object extend(VirtualFrame frame, PArray self, Object value,
786800 // in CPython
787801 try {
788802 length = PythonUtils .addExact (length , 1 );
803+ self .checkCanResize (this );
789804 self .resizeStorage (length );
790805 } catch (OverflowException e ) {
791806 CompilerDirectives .transferToInterpreterAndInvalidate ();
@@ -825,6 +840,7 @@ Object insert(VirtualFrame frame, PArray self, int inputIndex, Object value,
825840 // Need to check the validity of the value before moving the memory around to ensure the
826841 // operation can fail atomically
827842 checkValueNode .execute (frame , self , value );
843+ self .checkCanResize (this );
828844 try {
829845 self .shift (index , 1 );
830846 } catch (OverflowException e ) {
@@ -851,6 +867,7 @@ Object remove(VirtualFrame frame, PArray self, Object value,
851867 for (int i = 0 ; i < self .getLength (); i ++) {
852868 Object item = getValueNode .execute (self , i );
853869 if (lib .equalsWithFrame (item , value , lib , frame )) {
870+ self .checkCanResize (this );
854871 self .delSlice (i , 1 );
855872 return PNone .NONE ;
856873 }
@@ -872,6 +889,7 @@ Object pop(PArray self, int inputIndex,
872889 }
873890 int index = normalizeIndexNode .execute (inputIndex , self .getLength ());
874891 Object value = getValueNode .execute (self , index );
892+ self .checkCanResize (this );
875893 self .delSlice (index , 1 );
876894 return value ;
877895 }
@@ -898,6 +916,7 @@ Object frombytes(PArray self, Object buffer,
898916 }
899917 int newLength = PythonUtils .addExact (oldSize , bufferLength / itemsize );
900918 byte [] bufferBytes = lib .getBufferBytes (buffer );
919+ self .checkCanResize (this );
901920 self .resize (newLength );
902921 PythonUtils .arraycopy (bufferBytes , 0 , self .getBuffer (), oldSize * itemsize , bufferLength );
903922 } catch (UnsupportedMessageException e ) {
@@ -965,6 +984,7 @@ Object fromlist(VirtualFrame frame, PArray self, PList list,
965984 SequenceStorage storage = getSequenceStorageNode .execute (list );
966985 int length = lenNode .execute (storage );
967986 int newLength = PythonUtils .addExact (self .getLength (), length );
987+ self .checkCanResize (this );
968988 self .resizeStorage (newLength );
969989 for (int i = 0 ; i < length ; i ++) {
970990 putValueNode .execute (frame , self , self .getLength () + i , getItemScalarNode .execute (storage , i ));
@@ -1017,6 +1037,7 @@ Object fromunicode(VirtualFrame frame, PArray self, String str,
10171037 try {
10181038 int length = PString .codePointCount (str , 0 , str .length ());
10191039 int newLength = PythonUtils .addExact (self .getLength (), length );
1040+ self .checkCanResize (this );
10201041 self .resizeStorage (newLength );
10211042 for (int i = 0 , index = 0 ; i < length ; index ++) {
10221043 int cpCount = PString .charCount (PString .codePointAt (str , i ));
0 commit comments