Explorar el Código

Common: Fix a potential infinite loop in StringUtil's ReplaceAll

Lioncash hace 12 años
padre
commit
2f6ef914db
Se han modificado 1 ficheros con 8 adiciones y 3 borrados
  1. 8 3
      src/common/string_util.cpp

+ 8 - 3
src/common/string_util.cpp

@@ -283,12 +283,17 @@ std::string TabsToSpaces(int tab_size, const std::string &in)
 
 
 std::string ReplaceAll(std::string result, const std::string& src, const std::string& dest)
 std::string ReplaceAll(std::string result, const std::string& src, const std::string& dest)
 {
 {
-    while(1)
+    size_t pos = 0;
+
+    if (src == dest)
+        return result;
+
+    while ((pos = result.find(src, pos)) != std::string::npos)
     {
     {
-        size_t pos = result.find(src);
-        if (pos == std::string::npos) break;
         result.replace(pos, src.size(), dest);
         result.replace(pos, src.size(), dest);
+        pos += dest.length();
     }
     }
+
     return result;
     return result;
 }
 }