summaryrefslogtreecommitdiffstats
path: root/third_party/libwebrtc/rtc_base/file_rotating_stream.h
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-19 00:47:55 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-19 00:47:55 +0000
commit26a029d407be480d791972afb5975cf62c9360a6 (patch)
treef435a8308119effd964b339f76abb83a57c29483 /third_party/libwebrtc/rtc_base/file_rotating_stream.h
parentInitial commit. (diff)
downloadfirefox-26a029d407be480d791972afb5975cf62c9360a6.tar.xz
firefox-26a029d407be480d791972afb5975cf62c9360a6.zip
Adding upstream version 124.0.1.upstream/124.0.1
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'third_party/libwebrtc/rtc_base/file_rotating_stream.h')
-rw-r--r--third_party/libwebrtc/rtc_base/file_rotating_stream.h173
1 files changed, 173 insertions, 0 deletions
diff --git a/third_party/libwebrtc/rtc_base/file_rotating_stream.h b/third_party/libwebrtc/rtc_base/file_rotating_stream.h
new file mode 100644
index 0000000000..6ae2753098
--- /dev/null
+++ b/third_party/libwebrtc/rtc_base/file_rotating_stream.h
@@ -0,0 +1,173 @@
+/*
+ * Copyright 2015 The WebRTC Project Authors. All rights reserved.
+ *
+ * Use of this source code is governed by a BSD-style license
+ * that can be found in the LICENSE file in the root of the source
+ * tree. An additional intellectual property rights grant can be found
+ * in the file PATENTS. All contributing project authors may
+ * be found in the AUTHORS file in the root of the source tree.
+ */
+
+#ifndef RTC_BASE_FILE_ROTATING_STREAM_H_
+#define RTC_BASE_FILE_ROTATING_STREAM_H_
+
+#include <stddef.h>
+
+#include <memory>
+#include <string>
+#include <vector>
+
+#include "absl/strings/string_view.h"
+#include "rtc_base/system/file_wrapper.h"
+
+namespace rtc {
+
+// FileRotatingStream writes to a file in the directory specified in the
+// constructor. It rotates the files once the current file is full. The
+// individual file size and the number of files used is configurable in the
+// constructor. Open() must be called before using this stream.
+class FileRotatingStream {
+ public:
+ // Use this constructor for writing to a directory. Files in the directory
+ // matching the prefix will be deleted on open.
+ FileRotatingStream(absl::string_view dir_path,
+ absl::string_view file_prefix,
+ size_t max_file_size,
+ size_t num_files);
+
+ virtual ~FileRotatingStream();
+
+ FileRotatingStream(const FileRotatingStream&) = delete;
+ FileRotatingStream& operator=(const FileRotatingStream&) = delete;
+
+ bool IsOpen() const;
+
+ bool Write(const void* data, size_t data_len);
+ bool Flush();
+ void Close();
+
+ // Opens the appropriate file(s). Call this before using the stream.
+ bool Open();
+
+ // Disabling buffering causes writes to block until disk is updated. This is
+ // enabled by default for performance.
+ bool DisableBuffering();
+
+ // Below two methods are public for testing only.
+
+ // Returns the path used for the i-th newest file, where the 0th file is the
+ // newest file. The file may or may not exist, this is just used for
+ // formatting. Index must be less than GetNumFiles().
+ std::string GetFilePath(size_t index) const;
+
+ // Returns the number of files that will used by this stream.
+ size_t GetNumFiles() const { return file_names_.size(); }
+
+ protected:
+ void SetMaxFileSize(size_t size) { max_file_size_ = size; }
+
+ size_t GetRotationIndex() const { return rotation_index_; }
+
+ void SetRotationIndex(size_t index) { rotation_index_ = index; }
+
+ virtual void OnRotation() {}
+
+ private:
+ bool OpenCurrentFile();
+ void CloseCurrentFile();
+
+ // Rotates the files by creating a new current file, renaming the
+ // existing files, and deleting the oldest one. e.g.
+ // file_0 -> file_1
+ // file_1 -> file_2
+ // file_2 -> delete
+ // create new file_0
+ void RotateFiles();
+
+ // Private version of GetFilePath.
+ std::string GetFilePath(size_t index, size_t num_files) const;
+
+ const std::string dir_path_;
+ const std::string file_prefix_;
+
+ // File we're currently writing to.
+ webrtc::FileWrapper file_;
+ // Convenience storage for file names so we don't generate them over and over.
+ std::vector<std::string> file_names_;
+ size_t max_file_size_;
+ size_t current_file_index_;
+ // The rotation index indicates the index of the file that will be
+ // deleted first on rotation. Indices lower than this index will be rotated.
+ size_t rotation_index_;
+ // Number of bytes written to current file. We need this because with
+ // buffering the file size read from disk might not be accurate.
+ size_t current_bytes_written_;
+ bool disable_buffering_;
+};
+
+// CallSessionFileRotatingStream is meant to be used in situations where we will
+// have limited disk space. Its purpose is to write logs up to a
+// maximum size. Once the maximum size is exceeded, logs from the middle are
+// deleted whereas logs from the beginning and end are preserved. The reason for
+// this is because we anticipate that in WebRTC the beginning and end of the
+// logs are most useful for call diagnostics.
+//
+// This implementation simply writes to a single file until
+// `max_total_log_size` / 2 bytes are written to it, and subsequently writes to
+// a set of rotating files. We do this by inheriting FileRotatingStream and
+// setting the appropriate internal variables so that we don't delete the last
+// (earliest) file on rotate, and that that file's size is bigger.
+//
+// Open() must be called before using this stream.
+
+// To read the logs produced by this class, one can use the companion class
+// CallSessionFileRotatingStreamReader.
+class CallSessionFileRotatingStream : public FileRotatingStream {
+ public:
+ // Use this constructor for writing to a directory. Files in the directory
+ // matching what's used by the stream will be deleted. `max_total_log_size`
+ // must be at least 4.
+ CallSessionFileRotatingStream(absl::string_view dir_path,
+ size_t max_total_log_size);
+ ~CallSessionFileRotatingStream() override {}
+
+ CallSessionFileRotatingStream(const CallSessionFileRotatingStream&) = delete;
+ CallSessionFileRotatingStream& operator=(
+ const CallSessionFileRotatingStream&) = delete;
+
+ protected:
+ void OnRotation() override;
+
+ private:
+ static size_t GetRotatingLogSize(size_t max_total_log_size);
+ static size_t GetNumRotatingLogFiles(size_t max_total_log_size);
+ static const size_t kRotatingLogFileDefaultSize;
+
+ const size_t max_total_log_size_;
+ size_t num_rotations_;
+};
+
+// This is a convenience class, to read all files produced by a
+// FileRotatingStream, all in one go. Typical use calls GetSize and ReadData
+// only once. The list of file names to read is based on the contents of the log
+// directory at construction time.
+class FileRotatingStreamReader {
+ public:
+ FileRotatingStreamReader(absl::string_view dir_path,
+ absl::string_view file_prefix);
+ ~FileRotatingStreamReader();
+ size_t GetSize() const;
+ size_t ReadAll(void* buffer, size_t size) const;
+
+ private:
+ std::vector<std::string> file_names_;
+};
+
+class CallSessionFileRotatingStreamReader : public FileRotatingStreamReader {
+ public:
+ CallSessionFileRotatingStreamReader(absl::string_view dir_path);
+};
+
+} // namespace rtc
+
+#endif // RTC_BASE_FILE_ROTATING_STREAM_H_