OLD | NEW |
1 /* | 1 /* |
2 * Copyright 2015 The WebRTC Project Authors. All rights reserved. | 2 * Copyright 2015 The WebRTC Project Authors. All rights reserved. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license | 4 * Use of this source code is governed by a BSD-style license |
5 * that can be found in the LICENSE file in the root of the source | 5 * that can be found in the LICENSE file in the root of the source |
6 * tree. An additional intellectual property rights grant can be found | 6 * tree. An additional intellectual property rights grant can be found |
7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
9 */ | 9 */ |
10 | 10 |
(...skipping 19 matching lines...) Expand all Loading... |
30 | 30 |
31 FileRotatingStream::FileRotatingStream(const std::string& dir_path, | 31 FileRotatingStream::FileRotatingStream(const std::string& dir_path, |
32 const std::string& file_prefix, | 32 const std::string& file_prefix, |
33 size_t max_file_size, | 33 size_t max_file_size, |
34 size_t num_files) | 34 size_t num_files) |
35 : FileRotatingStream(dir_path, | 35 : FileRotatingStream(dir_path, |
36 file_prefix, | 36 file_prefix, |
37 max_file_size, | 37 max_file_size, |
38 num_files, | 38 num_files, |
39 kWrite) { | 39 kWrite) { |
40 DCHECK_GT(max_file_size, 0u); | 40 RTC_DCHECK_GT(max_file_size, 0u); |
41 DCHECK_GT(num_files, 1u); | 41 RTC_DCHECK_GT(num_files, 1u); |
42 } | 42 } |
43 | 43 |
44 FileRotatingStream::FileRotatingStream(const std::string& dir_path, | 44 FileRotatingStream::FileRotatingStream(const std::string& dir_path, |
45 const std::string& file_prefix, | 45 const std::string& file_prefix, |
46 size_t max_file_size, | 46 size_t max_file_size, |
47 size_t num_files, | 47 size_t num_files, |
48 Mode mode) | 48 Mode mode) |
49 : dir_path_(dir_path), | 49 : dir_path_(dir_path), |
50 file_prefix_(file_prefix), | 50 file_prefix_(file_prefix), |
51 mode_(mode), | 51 mode_(mode), |
52 file_stream_(nullptr), | 52 file_stream_(nullptr), |
53 max_file_size_(max_file_size), | 53 max_file_size_(max_file_size), |
54 current_file_index_(0), | 54 current_file_index_(0), |
55 rotation_index_(0), | 55 rotation_index_(0), |
56 current_bytes_written_(0), | 56 current_bytes_written_(0), |
57 disable_buffering_(false) { | 57 disable_buffering_(false) { |
58 DCHECK(Filesystem::IsFolder(dir_path)); | 58 RTC_DCHECK(Filesystem::IsFolder(dir_path)); |
59 switch (mode) { | 59 switch (mode) { |
60 case kWrite: { | 60 case kWrite: { |
61 file_names_.clear(); | 61 file_names_.clear(); |
62 for (size_t i = 0; i < num_files; ++i) { | 62 for (size_t i = 0; i < num_files; ++i) { |
63 file_names_.push_back(GetFilePath(i, num_files)); | 63 file_names_.push_back(GetFilePath(i, num_files)); |
64 } | 64 } |
65 rotation_index_ = num_files - 1; | 65 rotation_index_ = num_files - 1; |
66 break; | 66 break; |
67 } | 67 } |
68 case kRead: { | 68 case kRead: { |
(...skipping 18 matching lines...) Expand all Loading... |
87 if (!file_stream_) { | 87 if (!file_stream_) { |
88 return SS_CLOSED; | 88 return SS_CLOSED; |
89 } | 89 } |
90 return file_stream_->GetState(); | 90 return file_stream_->GetState(); |
91 } | 91 } |
92 | 92 |
93 StreamResult FileRotatingStream::Read(void* buffer, | 93 StreamResult FileRotatingStream::Read(void* buffer, |
94 size_t buffer_len, | 94 size_t buffer_len, |
95 size_t* read, | 95 size_t* read, |
96 int* error) { | 96 int* error) { |
97 DCHECK(buffer); | 97 RTC_DCHECK(buffer); |
98 if (mode_ != kRead) { | 98 if (mode_ != kRead) { |
99 return SR_EOS; | 99 return SR_EOS; |
100 } | 100 } |
101 if (current_file_index_ >= file_names_.size()) { | 101 if (current_file_index_ >= file_names_.size()) { |
102 return SR_EOS; | 102 return SR_EOS; |
103 } | 103 } |
104 // We will have no file stream initially, and when we are finished with the | 104 // We will have no file stream initially, and when we are finished with the |
105 // previous file. | 105 // previous file. |
106 if (!file_stream_) { | 106 if (!file_stream_) { |
107 if (!OpenCurrentFile()) { | 107 if (!OpenCurrentFile()) { |
(...skipping 37 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
145 size_t* written, | 145 size_t* written, |
146 int* error) { | 146 int* error) { |
147 if (mode_ != kWrite) { | 147 if (mode_ != kWrite) { |
148 return SR_EOS; | 148 return SR_EOS; |
149 } | 149 } |
150 if (!file_stream_) { | 150 if (!file_stream_) { |
151 std::cerr << "Open() must be called before Write." << std::endl; | 151 std::cerr << "Open() must be called before Write." << std::endl; |
152 return SR_ERROR; | 152 return SR_ERROR; |
153 } | 153 } |
154 // Write as much as will fit in to the current file. | 154 // Write as much as will fit in to the current file. |
155 DCHECK_LT(current_bytes_written_, max_file_size_); | 155 RTC_DCHECK_LT(current_bytes_written_, max_file_size_); |
156 size_t remaining_bytes = max_file_size_ - current_bytes_written_; | 156 size_t remaining_bytes = max_file_size_ - current_bytes_written_; |
157 size_t write_length = std::min(data_len, remaining_bytes); | 157 size_t write_length = std::min(data_len, remaining_bytes); |
158 size_t local_written = 0; | 158 size_t local_written = 0; |
159 if (!written) { | 159 if (!written) { |
160 written = &local_written; | 160 written = &local_written; |
161 } | 161 } |
162 StreamResult result = file_stream_->Write(data, write_length, written, error); | 162 StreamResult result = file_stream_->Write(data, write_length, written, error); |
163 current_bytes_written_ += *written; | 163 current_bytes_written_ += *written; |
164 | 164 |
165 // If we're done with this file, rotate it out. | 165 // If we're done with this file, rotate it out. |
166 if (current_bytes_written_ >= max_file_size_) { | 166 if (current_bytes_written_ >= max_file_size_) { |
167 DCHECK_EQ(current_bytes_written_, max_file_size_); | 167 RTC_DCHECK_EQ(current_bytes_written_, max_file_size_); |
168 RotateFiles(); | 168 RotateFiles(); |
169 } | 169 } |
170 return result; | 170 return result; |
171 } | 171 } |
172 | 172 |
173 bool FileRotatingStream::Flush() { | 173 bool FileRotatingStream::Flush() { |
174 if (!file_stream_) { | 174 if (!file_stream_) { |
175 return false; | 175 return false; |
176 } | 176 } |
177 return file_stream_->Flush(); | 177 return file_stream_->Flush(); |
178 } | 178 } |
179 | 179 |
180 bool FileRotatingStream::GetSize(size_t* size) const { | 180 bool FileRotatingStream::GetSize(size_t* size) const { |
181 if (mode_ != kRead) { | 181 if (mode_ != kRead) { |
182 // Not possible to get accurate size on disk when writing because of | 182 // Not possible to get accurate size on disk when writing because of |
183 // potential buffering. | 183 // potential buffering. |
184 return false; | 184 return false; |
185 } | 185 } |
186 DCHECK(size); | 186 RTC_DCHECK(size); |
187 *size = 0; | 187 *size = 0; |
188 size_t total_size = 0; | 188 size_t total_size = 0; |
189 for (auto file_name : file_names_) { | 189 for (auto file_name : file_names_) { |
190 Pathname pathname(file_name); | 190 Pathname pathname(file_name); |
191 size_t file_size = 0; | 191 size_t file_size = 0; |
192 if (Filesystem::GetFileSize(file_name, &file_size)) { | 192 if (Filesystem::GetFileSize(file_name, &file_size)) { |
193 total_size += file_size; | 193 total_size += file_size; |
194 } | 194 } |
195 } | 195 } |
196 *size = total_size; | 196 *size = total_size; |
(...skipping 28 matching lines...) Expand all Loading... |
225 disable_buffering_ = true; | 225 disable_buffering_ = true; |
226 if (!file_stream_) { | 226 if (!file_stream_) { |
227 std::cerr << "Open() must be called before DisableBuffering()." | 227 std::cerr << "Open() must be called before DisableBuffering()." |
228 << std::endl; | 228 << std::endl; |
229 return false; | 229 return false; |
230 } | 230 } |
231 return file_stream_->DisableBuffering(); | 231 return file_stream_->DisableBuffering(); |
232 } | 232 } |
233 | 233 |
234 std::string FileRotatingStream::GetFilePath(size_t index) const { | 234 std::string FileRotatingStream::GetFilePath(size_t index) const { |
235 DCHECK_LT(index, file_names_.size()); | 235 RTC_DCHECK_LT(index, file_names_.size()); |
236 return file_names_[index]; | 236 return file_names_[index]; |
237 } | 237 } |
238 | 238 |
239 bool FileRotatingStream::OpenCurrentFile() { | 239 bool FileRotatingStream::OpenCurrentFile() { |
240 CloseCurrentFile(); | 240 CloseCurrentFile(); |
241 | 241 |
242 // Opens the appropriate file in the appropriate mode. | 242 // Opens the appropriate file in the appropriate mode. |
243 DCHECK_LT(current_file_index_, file_names_.size()); | 243 RTC_DCHECK_LT(current_file_index_, file_names_.size()); |
244 std::string file_path = file_names_[current_file_index_]; | 244 std::string file_path = file_names_[current_file_index_]; |
245 file_stream_.reset(new FileStream()); | 245 file_stream_.reset(new FileStream()); |
246 const char* mode = nullptr; | 246 const char* mode = nullptr; |
247 switch (mode_) { | 247 switch (mode_) { |
248 case kWrite: | 248 case kWrite: |
249 mode = "w+"; | 249 mode = "w+"; |
250 // We should always we writing to the zero-th file. | 250 // We should always we writing to the zero-th file. |
251 DCHECK_EQ(current_file_index_, 0u); | 251 RTC_DCHECK_EQ(current_file_index_, 0u); |
252 break; | 252 break; |
253 case kRead: | 253 case kRead: |
254 mode = "r"; | 254 mode = "r"; |
255 break; | 255 break; |
256 } | 256 } |
257 int error = 0; | 257 int error = 0; |
258 if (!file_stream_->Open(file_path, mode, &error)) { | 258 if (!file_stream_->Open(file_path, mode, &error)) { |
259 std::cerr << "Failed to open: " << file_path << "Error: " << error | 259 std::cerr << "Failed to open: " << file_path << "Error: " << error |
260 << std::endl; | 260 << std::endl; |
261 file_stream_.reset(); | 261 file_stream_.reset(); |
262 return false; | 262 return false; |
263 } | 263 } |
264 if (disable_buffering_) { | 264 if (disable_buffering_) { |
265 file_stream_->DisableBuffering(); | 265 file_stream_->DisableBuffering(); |
266 } | 266 } |
267 return true; | 267 return true; |
268 } | 268 } |
269 | 269 |
270 void FileRotatingStream::CloseCurrentFile() { | 270 void FileRotatingStream::CloseCurrentFile() { |
271 if (!file_stream_) { | 271 if (!file_stream_) { |
272 return; | 272 return; |
273 } | 273 } |
274 current_bytes_written_ = 0; | 274 current_bytes_written_ = 0; |
275 file_stream_.reset(); | 275 file_stream_.reset(); |
276 } | 276 } |
277 | 277 |
278 void FileRotatingStream::RotateFiles() { | 278 void FileRotatingStream::RotateFiles() { |
279 DCHECK_EQ(mode_, kWrite); | 279 RTC_DCHECK_EQ(mode_, kWrite); |
280 CloseCurrentFile(); | 280 CloseCurrentFile(); |
281 // Rotates the files by deleting the file at |rotation_index_|, which is the | 281 // Rotates the files by deleting the file at |rotation_index_|, which is the |
282 // oldest file and then renaming the newer files to have an incremented index. | 282 // oldest file and then renaming the newer files to have an incremented index. |
283 // See header file comments for example. | 283 // See header file comments for example. |
284 DCHECK_LE(rotation_index_, file_names_.size()); | 284 RTC_DCHECK_LE(rotation_index_, file_names_.size()); |
285 std::string file_to_delete = file_names_[rotation_index_]; | 285 std::string file_to_delete = file_names_[rotation_index_]; |
286 if (Filesystem::IsFile(file_to_delete)) { | 286 if (Filesystem::IsFile(file_to_delete)) { |
287 if (!Filesystem::DeleteFile(file_to_delete)) { | 287 if (!Filesystem::DeleteFile(file_to_delete)) { |
288 std::cerr << "Failed to delete: " << file_to_delete << std::endl; | 288 std::cerr << "Failed to delete: " << file_to_delete << std::endl; |
289 } | 289 } |
290 } | 290 } |
291 for (auto i = rotation_index_; i > 0; --i) { | 291 for (auto i = rotation_index_; i > 0; --i) { |
292 std::string rotated_name = file_names_[i]; | 292 std::string rotated_name = file_names_[i]; |
293 std::string unrotated_name = file_names_[i - 1]; | 293 std::string unrotated_name = file_names_[i - 1]; |
294 if (Filesystem::IsFile(unrotated_name)) { | 294 if (Filesystem::IsFile(unrotated_name)) { |
(...skipping 23 matching lines...) Expand all Loading... |
318 current_name.compare(0, file_prefix_.size(), file_prefix_) == 0) { | 318 current_name.compare(0, file_prefix_.size(), file_prefix_) == 0) { |
319 Pathname path(dir_path_, current_name); | 319 Pathname path(dir_path_, current_name); |
320 files.push_back(path.pathname()); | 320 files.push_back(path.pathname()); |
321 } | 321 } |
322 } while (it.Next()); | 322 } while (it.Next()); |
323 return files; | 323 return files; |
324 } | 324 } |
325 | 325 |
326 std::string FileRotatingStream::GetFilePath(size_t index, | 326 std::string FileRotatingStream::GetFilePath(size_t index, |
327 size_t num_files) const { | 327 size_t num_files) const { |
328 DCHECK_LT(index, num_files); | 328 RTC_DCHECK_LT(index, num_files); |
329 std::ostringstream file_name; | 329 std::ostringstream file_name; |
330 // The format will be "_%<num_digits>zu". We want to zero pad the index so | 330 // The format will be "_%<num_digits>zu". We want to zero pad the index so |
331 // that it will sort nicely. | 331 // that it will sort nicely. |
332 size_t max_digits = ((num_files - 1) / 10) + 1; | 332 size_t max_digits = ((num_files - 1) / 10) + 1; |
333 size_t num_digits = (index / 10) + 1; | 333 size_t num_digits = (index / 10) + 1; |
334 DCHECK_LE(num_digits, max_digits); | 334 RTC_DCHECK_LE(num_digits, max_digits); |
335 size_t padding = max_digits - num_digits; | 335 size_t padding = max_digits - num_digits; |
336 | 336 |
337 file_name << file_prefix_ << "_"; | 337 file_name << file_prefix_ << "_"; |
338 for (size_t i = 0; i < padding; ++i) { | 338 for (size_t i = 0; i < padding; ++i) { |
339 file_name << "0"; | 339 file_name << "0"; |
340 } | 340 } |
341 file_name << index; | 341 file_name << index; |
342 | 342 |
343 Pathname file_path(dir_path_, file_name.str()); | 343 Pathname file_path(dir_path_, file_name.str()); |
344 return file_path.pathname(); | 344 return file_path.pathname(); |
345 } | 345 } |
346 | 346 |
347 CallSessionFileRotatingStream::CallSessionFileRotatingStream( | 347 CallSessionFileRotatingStream::CallSessionFileRotatingStream( |
348 const std::string& dir_path) | 348 const std::string& dir_path) |
349 : FileRotatingStream(dir_path, kLogPrefix), | 349 : FileRotatingStream(dir_path, kLogPrefix), |
350 max_total_log_size_(0), | 350 max_total_log_size_(0), |
351 num_rotations_(0) { | 351 num_rotations_(0) { |
352 } | 352 } |
353 | 353 |
354 CallSessionFileRotatingStream::CallSessionFileRotatingStream( | 354 CallSessionFileRotatingStream::CallSessionFileRotatingStream( |
355 const std::string& dir_path, | 355 const std::string& dir_path, |
356 size_t max_total_log_size) | 356 size_t max_total_log_size) |
357 : FileRotatingStream(dir_path, | 357 : FileRotatingStream(dir_path, |
358 kLogPrefix, | 358 kLogPrefix, |
359 max_total_log_size / 2, | 359 max_total_log_size / 2, |
360 GetNumRotatingLogFiles(max_total_log_size) + 1), | 360 GetNumRotatingLogFiles(max_total_log_size) + 1), |
361 max_total_log_size_(max_total_log_size), | 361 max_total_log_size_(max_total_log_size), |
362 num_rotations_(0) { | 362 num_rotations_(0) { |
363 DCHECK_GE(max_total_log_size, 4u); | 363 RTC_DCHECK_GE(max_total_log_size, 4u); |
364 } | 364 } |
365 | 365 |
366 const char* CallSessionFileRotatingStream::kLogPrefix = "webrtc_log"; | 366 const char* CallSessionFileRotatingStream::kLogPrefix = "webrtc_log"; |
367 const size_t CallSessionFileRotatingStream::kRotatingLogFileDefaultSize = | 367 const size_t CallSessionFileRotatingStream::kRotatingLogFileDefaultSize = |
368 1024 * 1024; | 368 1024 * 1024; |
369 | 369 |
370 void CallSessionFileRotatingStream::OnRotation() { | 370 void CallSessionFileRotatingStream::OnRotation() { |
371 ++num_rotations_; | 371 ++num_rotations_; |
372 if (num_rotations_ == 1) { | 372 if (num_rotations_ == 1) { |
373 // On the first rotation adjust the max file size so subsequent files after | 373 // On the first rotation adjust the max file size so subsequent files after |
(...skipping 17 matching lines...) Expand all Loading... |
391 | 391 |
392 size_t CallSessionFileRotatingStream::GetNumRotatingLogFiles( | 392 size_t CallSessionFileRotatingStream::GetNumRotatingLogFiles( |
393 size_t max_total_log_size) { | 393 size_t max_total_log_size) { |
394 // At minimum have two rotating files. Otherwise split the available log size | 394 // At minimum have two rotating files. Otherwise split the available log size |
395 // evenly across 1MB files. | 395 // evenly across 1MB files. |
396 return std::max((size_t)2, | 396 return std::max((size_t)2, |
397 (max_total_log_size / 2) / kRotatingLogFileDefaultSize); | 397 (max_total_log_size / 2) / kRotatingLogFileDefaultSize); |
398 } | 398 } |
399 | 399 |
400 } // namespace rtc | 400 } // namespace rtc |
OLD | NEW |