Преглед изворни кода

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

Lioncash пре 12 година
родитељ
комит
2f6ef914db
1 измењених фајлова са 8 додато и 3 уклоњено
  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)
 {
-    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);
+        pos += dest.length();
     }
+
     return result;
 }