Просмотр исходного кода

Fix heap-buffer overflow in PLY parser

Alexandre Avenel 2 лет назад
Родитель
Сommit
e4e2c63e0c
2 измененных файлов с 3 добавлено и 2 удалено
  1. 2 1
      code/AssetLib/Ply/PlyParser.cpp
  2. 1 1
      include/assimp/IOStreamBuffer.h

+ 2 - 1
code/AssetLib/Ply/PlyParser.cpp

@@ -425,7 +425,8 @@ bool PLY::DOM::ParseHeader(IOStreamBuffer<char> &streamBuffer, std::vector<char>
             break;
         } else {
             // ignore unknown header elements
-            streamBuffer.getNextLine(buffer);
+            if (!streamBuffer.getNextLine(buffer))
+              return false;
         }
     }
 

+ 1 - 1
include/assimp/IOStreamBuffer.h

@@ -287,7 +287,7 @@ static AI_FORCE_INLINE bool isEndOfCache(size_t pos, size_t cacheSize) {
 template <class T>
 AI_FORCE_INLINE bool IOStreamBuffer<T>::getNextLine(std::vector<T> &buffer) {
     buffer.resize(m_cacheSize);
-    if (isEndOfCache(m_cachePos, m_cacheSize) || 0 == m_filePos) {
+    if (m_cachePos >= m_cacheSize || 0 == m_filePos) {
         if (!readNextBlock()) {
             return false;
         }