You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
85 lines
2.3 KiB
85 lines
2.3 KiB
// Copyright (c) 2021 Proton Technologies AG |
|
// |
|
// This file is part of ProtonMail Bridge. |
|
// |
|
// ProtonMail Bridge is free software: you can redistribute it and/or modify |
|
// it under the terms of the GNU General Public License as published by |
|
// the Free Software Foundation, either version 3 of the License, or |
|
// (at your option) any later version. |
|
// |
|
// ProtonMail Bridge is distributed in the hope that it will be useful, |
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
// GNU General Public License for more details. |
|
// |
|
// You should have received a copy of the GNU General Public License |
|
// along with ProtonMail Bridge. If not, see <https://www.gnu.org/licenses/>. |
|
|
|
package logging |
|
|
|
import ( |
|
"io/ioutil" |
|
"os" |
|
"path/filepath" |
|
"sort" |
|
|
|
"github.com/sirupsen/logrus" |
|
) |
|
|
|
func clearLogs(logDir string, maxLogs int) error { |
|
files, err := ioutil.ReadDir(logDir) |
|
if err != nil { |
|
return err |
|
} |
|
|
|
var logsWithPrefix []string |
|
var crashesWithPrefix []string |
|
|
|
for _, file := range files { |
|
if matchLogName(file.Name()) { |
|
if matchStackTraceName(file.Name()) { |
|
crashesWithPrefix = append(crashesWithPrefix, file.Name()) |
|
} else { |
|
logsWithPrefix = append(logsWithPrefix, file.Name()) |
|
} |
|
} else { |
|
// Older versions of Bridge stored logs in subfolders for each version. |
|
// That also has to be cleared and the functionality can be removed after some time. |
|
if file.IsDir() { |
|
if err := clearLogs(filepath.Join(logDir, file.Name()), maxLogs); err != nil { |
|
return err |
|
} |
|
} else { |
|
removeLog(logDir, file.Name()) |
|
} |
|
} |
|
} |
|
|
|
removeOldLogs(logDir, logsWithPrefix, maxLogs) |
|
removeOldLogs(logDir, crashesWithPrefix, maxLogs) |
|
|
|
return nil |
|
} |
|
|
|
func removeOldLogs(logDir string, filenames []string, maxLogs int) { |
|
count := len(filenames) |
|
if count <= maxLogs { |
|
return |
|
} |
|
|
|
sort.Strings(filenames) // Sorted by timestamp: oldest first. |
|
for _, filename := range filenames[:count-maxLogs] { |
|
removeLog(logDir, filename) |
|
} |
|
} |
|
|
|
func removeLog(logDir, filename string) { |
|
// We need to be sure to delete only log files. |
|
// Directory with logs can also contain other files. |
|
if !matchLogName(filename) { |
|
return |
|
} |
|
if err := os.RemoveAll(filepath.Join(logDir, filename)); err != nil { |
|
logrus.WithError(err).Error("Failed to remove old logs") |
|
} |
|
}
|
|
|