浏览代码

RemoveComments: Fix out-of-bounds read when file ends with a comment

Turo Lamminen 7 年之前
父节点
当前提交
90cdd0f20c
共有 1 个文件被更改,包括 4 次插入0 次删除
  1. 4 0
      code/RemoveComments.cpp

+ 4 - 0
code/RemoveComments.cpp

@@ -66,6 +66,10 @@ void CommentRemover::RemoveLineComments(const char* szComment,
         if (!strncmp(szBuffer,szComment,len)) {
         if (!strncmp(szBuffer,szComment,len)) {
             while (!IsLineEnd(*szBuffer))
             while (!IsLineEnd(*szBuffer))
                 *szBuffer++ = chReplacement;
                 *szBuffer++ = chReplacement;
+
+            if (!*szBuffer) {
+                break;
+            }
         }
         }
         ++szBuffer;
         ++szBuffer;
     }
     }