Parcourir la source

Merge pull request #275 from Dmitry-Me/moreOverflowChecks

More overflow checks
Lee Thomason il y a 11 ans
Parent
commit
38b49ae042
1 fichiers modifiés avec 4 ajouts et 0 suppressions
  1. 4 0
      tinyxml2.h

+ 4 - 0
tinyxml2.h

@@ -232,11 +232,14 @@ public:
     }
 
     void Push( T t ) {
+        TIXMLASSERT( _size < (size_t)(-1) );
         EnsureCapacity( _size+1 );
         _mem[_size++] = t;
     }
 
     T* PushArr( int count ) {
+        TIXMLASSERT( count >= 0 );
+        TIXMLASSERT( _size <= (size_t)(-1) - count );
         EnsureCapacity( _size+count );
         T* ret = &_mem[_size];
         _size += count;
@@ -244,6 +247,7 @@ public:
     }
 
     T Pop() {
+        TIXMLASSERT( _size > 0 );
         return _mem[--_size];
     }