| 
									
										
										
										
											2021-07-05 09:05:40 +00:00
										 |  |  | From 80368f8ba7a8bab13440463a254888311efe3986 Mon Sep 17 00:00:00 2001 | 
					
						
							|  |  |  | From: Stephan Hartmann <stha09@googlemail.com> | 
					
						
							|  |  |  | Date: Tue, 4 May 2021 15:00:19 +0000 | 
					
						
							|  |  |  | Subject: [PATCH] sql: make VirtualCursor standard layout type | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | sql::recover::VirtualCursor needs to be a standard layout type, but | 
					
						
							|  |  |  | has members of type std::unique_ptr. However, std::unique_ptr is not | 
					
						
							|  |  |  | guaranteed to be standard layout. Compiling with clang combined with | 
					
						
							|  |  |  | gcc-11 libstdc++ fails because of this. Replace std::unique_ptr with | 
					
						
							|  |  |  | raw pointers. | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | Bug: 1189788 | 
					
						
							|  |  |  | Change-Id: Ia6dc388cc5ef1c0f2afc75f8ca45b9f12687ca9c | 
					
						
							|  |  |  | ---
 | 
					
						
							|  |  |  |  sql/recover_module/btree.cc  | 21 +++++++++++++++------ | 
					
						
							|  |  |  |  sql/recover_module/btree.h   | 17 +++++++++++++---- | 
					
						
							|  |  |  |  sql/recover_module/cursor.cc | 24 ++++++++++++------------ | 
					
						
							|  |  |  |  sql/recover_module/cursor.h  |  2 +- | 
					
						
							|  |  |  |  sql/recover_module/pager.cc  |  7 +++---- | 
					
						
							|  |  |  |  sql/recover_module/pager.h   |  5 +++-- | 
					
						
							|  |  |  |  6 files changed, 47 insertions(+), 29 deletions(-) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2022-02-04 13:14:22 +00:00
										 |  |  | Index: chromium-98.0.4758.80/sql/recover_module/btree.cc
 | 
					
						
							|  |  |  | ===================================================================
 | 
					
						
							|  |  |  | --- chromium-98.0.4758.80.orig/sql/recover_module/btree.cc
 | 
					
						
							|  |  |  | +++ chromium-98.0.4758.80/sql/recover_module/btree.cc
 | 
					
						
							|  |  |  | @@ -136,16 +136,25 @@ static_assert(std::is_trivially_destruct
 | 
					
						
							| 
									
										
										
										
											2021-07-05 09:05:40 +00:00
										 |  |  |                "Move the destructor to the .cc file if it's non-trival"); | 
					
						
							|  |  |  |  #endif  // !DCHECK_IS_ON() | 
					
						
							|  |  |  |   | 
					
						
							|  |  |  | -LeafPageDecoder::LeafPageDecoder(DatabasePageReader* db_reader) noexcept
 | 
					
						
							|  |  |  | -    : page_id_(db_reader->page_id()),
 | 
					
						
							|  |  |  | -      db_reader_(db_reader),
 | 
					
						
							|  |  |  | -      cell_count_(ComputeCellCount(db_reader)),
 | 
					
						
							|  |  |  | -      next_read_index_(0),
 | 
					
						
							|  |  |  | -      last_record_size_(0) {
 | 
					
						
							|  |  |  | +void LeafPageDecoder::Initialize(DatabasePageReader* db_reader) {
 | 
					
						
							|  |  |  | +  DCHECK(db_reader);
 | 
					
						
							|  |  |  |    DCHECK(IsOnValidPage(db_reader)); | 
					
						
							|  |  |  | +  page_id_ = db_reader->page_id();
 | 
					
						
							|  |  |  | +  db_reader_ = db_reader;
 | 
					
						
							|  |  |  | +  cell_count_ = ComputeCellCount(db_reader);
 | 
					
						
							|  |  |  | +  next_read_index_ = 0;
 | 
					
						
							|  |  |  | +  last_record_size_ = 0;
 | 
					
						
							|  |  |  |    DCHECK(DatabasePageReader::IsValidPageId(page_id_)); | 
					
						
							|  |  |  |  } | 
					
						
							|  |  |  |   | 
					
						
							|  |  |  | +void LeafPageDecoder::Reset() {
 | 
					
						
							|  |  |  | +  db_reader_ = nullptr;
 | 
					
						
							|  |  |  | +  page_id_ = 0;
 | 
					
						
							|  |  |  | +  cell_count_ = 0;
 | 
					
						
							|  |  |  | +  next_read_index_ = 0;
 | 
					
						
							|  |  |  | +  last_record_size_ = 0;
 | 
					
						
							|  |  |  | +}
 | 
					
						
							|  |  |  | +
 | 
					
						
							|  |  |  |  bool LeafPageDecoder::TryAdvance() { | 
					
						
							|  |  |  |    DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_); | 
					
						
							|  |  |  |    DCHECK(CanAdvance()); | 
					
						
							| 
									
										
										
										
											2022-02-04 13:14:22 +00:00
										 |  |  | Index: chromium-98.0.4758.80/sql/recover_module/btree.h
 | 
					
						
							|  |  |  | ===================================================================
 | 
					
						
							|  |  |  | --- chromium-98.0.4758.80.orig/sql/recover_module/btree.h
 | 
					
						
							|  |  |  | +++ chromium-98.0.4758.80/sql/recover_module/btree.h
 | 
					
						
							|  |  |  | @@ -103,7 +103,7 @@ class LeafPageDecoder {
 | 
					
						
							| 
									
										
										
										
											2021-07-05 09:05:40 +00:00
										 |  |  |    // | 
					
						
							|  |  |  |    // |db_reader| must have been used to read an inner page of a table B-tree. | 
					
						
							|  |  |  |    // |db_reader| must outlive this instance. | 
					
						
							|  |  |  | -  explicit LeafPageDecoder(DatabasePageReader* db_reader) noexcept;
 | 
					
						
							|  |  |  | +  explicit LeafPageDecoder() noexcept = default;
 | 
					
						
							|  |  |  |    ~LeafPageDecoder() noexcept = default; | 
					
						
							|  |  |  |   | 
					
						
							|  |  |  |    LeafPageDecoder(const LeafPageDecoder&) = delete; | 
					
						
							| 
									
										
										
										
											2022-02-04 13:14:22 +00:00
										 |  |  | @@ -151,6 +151,15 @@ class LeafPageDecoder {
 | 
					
						
							| 
									
										
										
										
											2021-07-05 09:05:40 +00:00
										 |  |  |    // read as long as CanAdvance() returns true. | 
					
						
							|  |  |  |    bool TryAdvance(); | 
					
						
							|  |  |  |   | 
					
						
							|  |  |  | +  // Initialize with DatabasePageReader
 | 
					
						
							|  |  |  | +  void Initialize(DatabasePageReader* db_reader);
 | 
					
						
							|  |  |  | +
 | 
					
						
							|  |  |  | +  // Reset internal DatabasePageReader
 | 
					
						
							|  |  |  | +  void Reset();
 | 
					
						
							|  |  |  | +
 | 
					
						
							|  |  |  | +  // True if DatabasePageReader is valid
 | 
					
						
							|  |  |  | +  bool IsValid() { return (db_reader_ != nullptr); }
 | 
					
						
							|  |  |  | +
 | 
					
						
							|  |  |  |    // True if the given reader may point to an inner page in a table B-tree. | 
					
						
							|  |  |  |    // | 
					
						
							|  |  |  |    // The last ReadPage() call on |db_reader| must have succeeded. | 
					
						
							| 
									
										
										
										
											2022-02-04 13:14:22 +00:00
										 |  |  | @@ -164,14 +173,14 @@ class LeafPageDecoder {
 | 
					
						
							| 
									
										
										
										
											2021-07-05 09:05:40 +00:00
										 |  |  |    static int ComputeCellCount(DatabasePageReader* db_reader); | 
					
						
							|  |  |  |   | 
					
						
							|  |  |  |    // The number of the B-tree page this reader is reading. | 
					
						
							|  |  |  | -  const int64_t page_id_;
 | 
					
						
							|  |  |  | +  int64_t page_id_;
 | 
					
						
							|  |  |  |    // Used to read the tree page. | 
					
						
							|  |  |  |    // | 
					
						
							|  |  |  |    // Raw pointer usage is acceptable because this instance's owner is expected | 
					
						
							|  |  |  |    // to ensure that the DatabasePageReader outlives this. | 
					
						
							|  |  |  | -  DatabasePageReader* const db_reader_;
 | 
					
						
							|  |  |  | +  DatabasePageReader* db_reader_;
 | 
					
						
							|  |  |  |    // Caches the ComputeCellCount() value for this reader's page. | 
					
						
							|  |  |  | -  const int cell_count_ = ComputeCellCount(db_reader_);
 | 
					
						
							|  |  |  | +  int cell_count_;
 | 
					
						
							|  |  |  |   | 
					
						
							|  |  |  |    // The reader's cursor state. | 
					
						
							|  |  |  |    // | 
					
						
							| 
									
										
										
										
											2022-02-04 13:14:22 +00:00
										 |  |  | Index: chromium-98.0.4758.80/sql/recover_module/cursor.cc
 | 
					
						
							|  |  |  | ===================================================================
 | 
					
						
							|  |  |  | --- chromium-98.0.4758.80.orig/sql/recover_module/cursor.cc
 | 
					
						
							|  |  |  | +++ chromium-98.0.4758.80/sql/recover_module/cursor.cc
 | 
					
						
							|  |  |  | @@ -28,7 +28,7 @@ VirtualCursor::~VirtualCursor() {
 | 
					
						
							| 
									
										
										
										
											2021-07-05 09:05:40 +00:00
										 |  |  |  int VirtualCursor::First() { | 
					
						
							|  |  |  |    DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_); | 
					
						
							|  |  |  |    inner_decoders_.clear(); | 
					
						
							|  |  |  | -  leaf_decoder_ = nullptr;
 | 
					
						
							|  |  |  | +  leaf_decoder_.Reset();
 | 
					
						
							|  |  |  |   | 
					
						
							|  |  |  |    AppendPageDecoder(table_->root_page_id()); | 
					
						
							|  |  |  |    return Next(); | 
					
						
							| 
									
										
										
										
											2022-02-04 13:14:22 +00:00
										 |  |  | @@ -38,18 +38,18 @@ int VirtualCursor::Next() {
 | 
					
						
							| 
									
										
										
										
											2021-07-05 09:05:40 +00:00
										 |  |  |    DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_); | 
					
						
							|  |  |  |    record_reader_.Reset(); | 
					
						
							|  |  |  |   | 
					
						
							|  |  |  | -  while (!inner_decoders_.empty() || leaf_decoder_.get()) {
 | 
					
						
							|  |  |  | -    if (leaf_decoder_.get()) {
 | 
					
						
							|  |  |  | -      if (!leaf_decoder_->CanAdvance()) {
 | 
					
						
							|  |  |  | +  while (!inner_decoders_.empty() || leaf_decoder_.IsValid()) {
 | 
					
						
							|  |  |  | +    if (leaf_decoder_.IsValid()) {
 | 
					
						
							|  |  |  | +      if (!leaf_decoder_.CanAdvance()) {
 | 
					
						
							|  |  |  |          // The leaf has been exhausted. Remove it from the DFS stack. | 
					
						
							|  |  |  | -        leaf_decoder_ = nullptr;
 | 
					
						
							|  |  |  | +        leaf_decoder_.Reset();
 | 
					
						
							|  |  |  |          continue; | 
					
						
							|  |  |  |        } | 
					
						
							|  |  |  | -      if (!leaf_decoder_->TryAdvance())
 | 
					
						
							|  |  |  | +      if (!leaf_decoder_.TryAdvance())
 | 
					
						
							|  |  |  |          continue; | 
					
						
							|  |  |  |   | 
					
						
							|  |  |  | -      if (!payload_reader_.Initialize(leaf_decoder_->last_record_size(),
 | 
					
						
							|  |  |  | -                                      leaf_decoder_->last_record_offset())) {
 | 
					
						
							|  |  |  | +      if (!payload_reader_.Initialize(leaf_decoder_.last_record_size(),
 | 
					
						
							|  |  |  | +                                      leaf_decoder_.last_record_offset())) {
 | 
					
						
							|  |  |  |          continue; | 
					
						
							|  |  |  |        } | 
					
						
							|  |  |  |        if (!record_reader_.Initialize()) | 
					
						
							| 
									
										
										
										
											2022-02-04 13:14:22 +00:00
										 |  |  | @@ -101,13 +101,13 @@ int VirtualCursor::ReadColumn(int column
 | 
					
						
							| 
									
										
										
										
											2021-07-05 09:05:40 +00:00
										 |  |  |  int64_t VirtualCursor::RowId() { | 
					
						
							|  |  |  |    DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_); | 
					
						
							|  |  |  |    DCHECK(record_reader_.IsInitialized()); | 
					
						
							|  |  |  | -  DCHECK(leaf_decoder_.get());
 | 
					
						
							|  |  |  | -  return leaf_decoder_->last_record_rowid();
 | 
					
						
							|  |  |  | +  DCHECK(leaf_decoder_.IsValid());
 | 
					
						
							|  |  |  | +  return leaf_decoder_.last_record_rowid();
 | 
					
						
							|  |  |  |  } | 
					
						
							|  |  |  |   | 
					
						
							|  |  |  |  void VirtualCursor::AppendPageDecoder(int page_id) { | 
					
						
							|  |  |  |    DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_); | 
					
						
							|  |  |  | -  DCHECK(leaf_decoder_.get() == nullptr)
 | 
					
						
							|  |  |  | +  DCHECK(!leaf_decoder_.IsValid())
 | 
					
						
							|  |  |  |        << __func__ | 
					
						
							|  |  |  |        << " must only be called when the current path has no leaf decoder"; | 
					
						
							|  |  |  |   | 
					
						
							| 
									
										
										
										
											2022-02-04 13:14:22 +00:00
										 |  |  | @@ -115,7 +115,7 @@ void VirtualCursor::AppendPageDecoder(in
 | 
					
						
							| 
									
										
										
										
											2021-07-05 09:05:40 +00:00
										 |  |  |      return; | 
					
						
							|  |  |  |   | 
					
						
							|  |  |  |    if (LeafPageDecoder::IsOnValidPage(&db_reader_)) { | 
					
						
							|  |  |  | -    leaf_decoder_ = std::make_unique<LeafPageDecoder>(&db_reader_);
 | 
					
						
							|  |  |  | +    leaf_decoder_.Initialize(&db_reader_);
 | 
					
						
							|  |  |  |      return; | 
					
						
							|  |  |  |    } | 
					
						
							|  |  |  |   | 
					
						
							| 
									
										
										
										
											2022-02-04 13:14:22 +00:00
										 |  |  | Index: chromium-98.0.4758.80/sql/recover_module/cursor.h
 | 
					
						
							|  |  |  | ===================================================================
 | 
					
						
							|  |  |  | --- chromium-98.0.4758.80.orig/sql/recover_module/cursor.h
 | 
					
						
							|  |  |  | +++ chromium-98.0.4758.80/sql/recover_module/cursor.h
 | 
					
						
							|  |  |  | @@ -130,7 +130,7 @@ class VirtualCursor {
 | 
					
						
							| 
									
										
										
										
											2021-07-05 09:05:40 +00:00
										 |  |  |    std::vector<std::unique_ptr<InnerPageDecoder>> inner_decoders_; | 
					
						
							|  |  |  |   | 
					
						
							|  |  |  |    // Decodes the leaf page containing records. | 
					
						
							|  |  |  | -  std::unique_ptr<LeafPageDecoder> leaf_decoder_;
 | 
					
						
							|  |  |  | +  LeafPageDecoder leaf_decoder_;
 | 
					
						
							|  |  |  |   | 
					
						
							|  |  |  |    SEQUENCE_CHECKER(sequence_checker_); | 
					
						
							|  |  |  |  }; | 
					
						
							| 
									
										
										
										
											2022-02-04 13:14:22 +00:00
										 |  |  | Index: chromium-98.0.4758.80/sql/recover_module/pager.cc
 | 
					
						
							|  |  |  | ===================================================================
 | 
					
						
							|  |  |  | --- chromium-98.0.4758.80.orig/sql/recover_module/pager.cc
 | 
					
						
							|  |  |  | +++ chromium-98.0.4758.80/sql/recover_module/pager.cc
 | 
					
						
							|  |  |  | @@ -23,8 +23,7 @@ static_assert(DatabasePageReader::kMaxPa
 | 
					
						
							| 
									
										
										
										
											2021-07-05 09:05:40 +00:00
										 |  |  |                "ints are not appropriate for representing page IDs"); | 
					
						
							|  |  |  |   | 
					
						
							|  |  |  |  DatabasePageReader::DatabasePageReader(VirtualTable* table) | 
					
						
							|  |  |  | -    : page_data_(std::make_unique<uint8_t[]>(table->page_size())),
 | 
					
						
							|  |  |  | -      table_(table) {
 | 
					
						
							|  |  |  | +    : page_data_(), table_(table) {
 | 
					
						
							|  |  |  |    DCHECK(table != nullptr); | 
					
						
							|  |  |  |    DCHECK(IsValidPageSize(table->page_size())); | 
					
						
							|  |  |  |  } | 
					
						
							| 
									
										
										
										
											2022-02-04 13:14:22 +00:00
										 |  |  | @@ -57,8 +56,8 @@ int DatabasePageReader::ReadPage(int pag
 | 
					
						
							| 
									
										
										
										
											2021-07-05 09:05:40 +00:00
										 |  |  |                      std::numeric_limits<int64_t>::max(), | 
					
						
							|  |  |  |                  "The |read_offset| computation above may overflow"); | 
					
						
							|  |  |  |   | 
					
						
							|  |  |  | -  int sqlite_status =
 | 
					
						
							|  |  |  | -      RawRead(sqlite_file, read_size, read_offset, page_data_.get());
 | 
					
						
							|  |  |  | +  int sqlite_status = RawRead(sqlite_file, read_size, read_offset,
 | 
					
						
							|  |  |  | +                              const_cast<uint8_t*>(page_data_.data()));
 | 
					
						
							|  |  |  |   | 
					
						
							|  |  |  |    // |page_id_| needs to be set to kInvalidPageId if the read failed. | 
					
						
							|  |  |  |    // Otherwise, future ReadPage() calls with the previous |page_id_| value | 
					
						
							| 
									
										
										
										
											2022-02-04 13:14:22 +00:00
										 |  |  | Index: chromium-98.0.4758.80/sql/recover_module/pager.h
 | 
					
						
							|  |  |  | ===================================================================
 | 
					
						
							|  |  |  | --- chromium-98.0.4758.80.orig/sql/recover_module/pager.h
 | 
					
						
							|  |  |  | +++ chromium-98.0.4758.80/sql/recover_module/pager.h
 | 
					
						
							| 
									
										
										
										
											2021-07-05 09:05:40 +00:00
										 |  |  | @@ -5,6 +5,7 @@
 | 
					
						
							|  |  |  |  #ifndef SQL_RECOVER_MODULE_PAGER_H_ | 
					
						
							|  |  |  |  #define SQL_RECOVER_MODULE_PAGER_H_ | 
					
						
							|  |  |  |   | 
					
						
							|  |  |  | +#include <array>
 | 
					
						
							|  |  |  |  #include <cstdint> | 
					
						
							|  |  |  |  #include <memory> | 
					
						
							| 
									
										
										
										
											2021-10-03 09:57:13 +00:00
										 |  |  |  #include <ostream> | 
					
						
							| 
									
										
										
										
											2022-02-04 13:14:22 +00:00
										 |  |  | @@ -72,7 +73,7 @@ class DatabasePageReader {
 | 
					
						
							| 
									
										
										
										
											2021-07-05 09:05:40 +00:00
										 |  |  |      DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_); | 
					
						
							|  |  |  |      DCHECK_NE(page_id_, kInvalidPageId) | 
					
						
							|  |  |  |          << "Successful ReadPage() required before accessing pager state"; | 
					
						
							|  |  |  | -    return page_data_.get();
 | 
					
						
							|  |  |  | +    return page_data_.data();
 | 
					
						
							|  |  |  |    } | 
					
						
							|  |  |  |   | 
					
						
							|  |  |  |    // The number of bytes in the page read by the last ReadPage() call. | 
					
						
							| 
									
										
										
										
											2022-02-04 13:14:22 +00:00
										 |  |  | @@ -139,7 +140,7 @@ class DatabasePageReader {
 | 
					
						
							| 
									
										
										
										
											2021-07-05 09:05:40 +00:00
										 |  |  |    int page_id_ = kInvalidPageId; | 
					
						
							|  |  |  |    // Stores the bytes of the last page successfully read by ReadPage(). | 
					
						
							|  |  |  |    // The content is undefined if the last call to ReadPage() did not succeed. | 
					
						
							|  |  |  | -  const std::unique_ptr<uint8_t[]> page_data_;
 | 
					
						
							|  |  |  | +  const std::array<uint8_t, kMaxPageSize> page_data_;
 | 
					
						
							|  |  |  |    // Raw pointer usage is acceptable because this instance's owner is expected | 
					
						
							|  |  |  |    // to ensure that the VirtualTable outlives this. | 
					
						
							| 
									
										
										
										
											2022-02-04 13:14:22 +00:00
										 |  |  |    const raw_ptr<VirtualTable> table_; |