Fix OOB read in trim("")

EleutherAI · Apr 8, 2022 · f5e54cd · f5e54cd
1 parent c9317a1
commit f5e54cd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/util/string.h b/src/util/string.h
@@ -295,11 +295,11 @@ inline std::string lowercase(const std::string &str)
 inline std::string trim(const std::string &str)
 {
  size_t front = 0;
+ size_t back = str.size();
 
- while (std::isspace(str[front]))
+ while (front < back && std::isspace(str[front]))
  ++front;
 
- size_t back = str.size();
  while (back > front && std::isspace(str[back - 1]))
  --back;