tdlight-team/tdlight
tdlight-team/tdlight
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

binlog_dump: protect from accidental binlog creation and encryption.

Browse Source 
GitOrigin-RevId: e1e624413c7f672e90b705fb954ed3000d6170c2
This commit is contained in:
levlam 2020-07-13 20:46:17 +03:00
parent 029ee5dd37
commit 416188fc67
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tddb/td/db/binlog/binlog_dump.cpp
View File

@ -12,6 +12,7 @@
#include "td/utils/format.h" #include "td/utils/format.h"
#include "td/utils/logging.h" #include "td/utils/logging.h"
#include "td/utils/misc.h" #include "td/utils/misc.h"
#include "td/utils/port/Stat.h"
#include "td/utils/Slice.h" #include "td/utils/Slice.h"
#include "td/utils/StringBuilder.h" #include "td/utils/StringBuilder.h"
#include "td/utils/tl_parsers.h" #include "td/utils/tl_parsers.h"
@ -99,6 +100,13 @@ int main(int argc, char *argv[]) {
LOG(PLAIN) << "Usage: binlog_dump <binlog_file_name>"; LOG(PLAIN) << "Usage: binlog_dump <binlog_file_name>";
return 1; return 1;
} }
td::string binlog_file_name = argv[1];
auto r_stat = td::stat(binlog_file_name);
if (r_stat.is_error() || r_stat.ok().size_ == 0 || !r_stat.ok().is_reg_) {
LOG(PLAIN) << "Wrong binlog file name specified";
LOG(PLAIN) << "Usage: binlog_dump <binlog_file_name>";
return 1;
}
struct Info { struct Info {
std::size_t full_size = 0; std::size_t full_size = 0;
@ -112,7 +120,7 @@ int main(int argc, char *argv[]) {
td::Binlog binlog; td::Binlog binlog;
binlog binlog
.init( .init(
argv[1], binlog_file_name,
[&](auto &event) { [&](auto &event) {
info[0].compressed_size += event.raw_event_.size(); info[0].compressed_size += event.raw_event_.size();
info[event.type_].compressed_size += event.raw_event_.size(); info[event.type_].compressed_size += event.raw_event_.size();