From 635a078180973fa50408c3e2fc2d6b7de24ee1a7 Mon Sep 17 00:00:00 2001
From: xiaobfly <313060639@qq.com>
Date: Wed, 7 Apr 2021 17:42:52 +0800
Subject: [PATCH] fix LogCleaner::IsLogFromCurrentProject for
 filename_extension

---
 src/logging.cc | 21 ++++++++++++++++++---
 1 file changed, 18 insertions(+), 3 deletions(-)

diff --git a/src/logging.cc b/src/logging.cc
index 25614b6..0f4fa4f 100644
--- a/src/logging.cc
+++ b/src/logging.cc
@@ -1380,6 +1380,7 @@ bool LogCleaner::IsLogFromCurrentProject(const string& filepath,
   // after:  "/tmp/<base_filename>.<create_time>.<pid>"
   string cleaned_base_filename;
 
+  size_t real_filepath_size = filepath.size();
   for (size_t i = 0; i < base_filename.size(); ++i) {
     const char& c = base_filename[i];
 
@@ -1400,15 +1401,29 @@ bool LogCleaner::IsLogFromCurrentProject(const string& filepath,
   // `cleaned_base_filename` in `filepath` if the user
   // has set a custom filename extension.
   if (!filename_extension.empty()) {
-    if (filepath.find(filename_extension) != cleaned_base_filename.size()) {
+    if (cleaned_base_filename.size() >= real_filepath_size) {
       return false;
     }
-    cleaned_base_filename += filename_extension;
+    // for origin version, `filename_extension` is middle of the `filepath`.
+    string ext = filepath.substr(cleaned_base_filename.size(), filename_extension.size());
+    if (ext == filename_extension) {
+      cleaned_base_filename += filename_extension;
+    }
+    else {
+      // for new version, `filename_extension` is right of the `filepath`.
+      if (filename_extension.size() >= real_filepath_size) {
+        return false;
+      }
+      real_filepath_size = filepath.size() - filename_extension.size();
+      if (filepath.substr(real_filepath_size) != filename_extension) {
+        return false;
+      }
+    }
   }
 
   // The characters after `cleaned_base_filename` should match the format:
   // YYYYMMDD-HHMMSS.pid
-  for (size_t i = cleaned_base_filename.size(); i < filepath.size(); i++) {
+  for (size_t i = cleaned_base_filename.size(); i < real_filepath_size; i++) {
     const char& c = filepath[i];
 
     if (i <= cleaned_base_filename.size() + 7) { // 0 ~ 7 : YYYYMMDD