sql-make-VirtualCursor-standard-layout-type.patch 9.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238
  1. From 80368f8ba7a8bab13440463a254888311efe3986 Mon Sep 17 00:00:00 2001
  2. From: Stephan Hartmann <stha09@googlemail.com>
  3. Date: Tue, 4 May 2021 15:00:19 +0000
  4. Subject: [PATCH] sql: make VirtualCursor standard layout type
  5. sql::recover::VirtualCursor needs to be a standard layout type, but
  6. has members of type std::unique_ptr. However, std::unique_ptr is not
  7. guaranteed to be standard layout. Compiling with clang combined with
  8. gcc-11 libstdc++ fails because of this. Replace std::unique_ptr with
  9. raw pointers.
  10. Bug: 1189788
  11. Change-Id: Ia6dc388cc5ef1c0f2afc75f8ca45b9f12687ca9c
  12. ---
  13. sql/recover_module/btree.cc | 21 +++++++++++++++------
  14. sql/recover_module/btree.h | 17 +++++++++++++----
  15. sql/recover_module/cursor.cc | 24 ++++++++++++------------
  16. sql/recover_module/cursor.h | 2 +-
  17. sql/recover_module/pager.cc | 7 +++----
  18. sql/recover_module/pager.h | 5 +++--
  19. 6 files changed, 47 insertions(+), 29 deletions(-)
  20. diff --git a/sql/recover_module/btree.cc b/sql/recover_module/btree.cc
  21. index 9ecaafe8a3..839318abf9 100644
  22. --- a/sql/recover_module/btree.cc
  23. +++ b/sql/recover_module/btree.cc
  24. @@ -135,16 +135,25 @@ static_assert(std::is_trivially_destructible<LeafPageDecoder>::value,
  25. "Move the destructor to the .cc file if it's non-trival");
  26. #endif // !DCHECK_IS_ON()
  27. -LeafPageDecoder::LeafPageDecoder(DatabasePageReader* db_reader) noexcept
  28. - : page_id_(db_reader->page_id()),
  29. - db_reader_(db_reader),
  30. - cell_count_(ComputeCellCount(db_reader)),
  31. - next_read_index_(0),
  32. - last_record_size_(0) {
  33. +void LeafPageDecoder::Initialize(DatabasePageReader* db_reader) {
  34. + DCHECK(db_reader);
  35. DCHECK(IsOnValidPage(db_reader));
  36. + page_id_ = db_reader->page_id();
  37. + db_reader_ = db_reader;
  38. + cell_count_ = ComputeCellCount(db_reader);
  39. + next_read_index_ = 0;
  40. + last_record_size_ = 0;
  41. DCHECK(DatabasePageReader::IsValidPageId(page_id_));
  42. }
  43. +void LeafPageDecoder::Reset() {
  44. + db_reader_ = nullptr;
  45. + page_id_ = 0;
  46. + cell_count_ = 0;
  47. + next_read_index_ = 0;
  48. + last_record_size_ = 0;
  49. +}
  50. +
  51. bool LeafPageDecoder::TryAdvance() {
  52. DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  53. DCHECK(CanAdvance());
  54. diff --git a/sql/recover_module/btree.h b/sql/recover_module/btree.h
  55. index d76d076bf6..33114b01fa 100644
  56. --- a/sql/recover_module/btree.h
  57. +++ b/sql/recover_module/btree.h
  58. @@ -102,7 +102,7 @@ class LeafPageDecoder {
  59. //
  60. // |db_reader| must have been used to read an inner page of a table B-tree.
  61. // |db_reader| must outlive this instance.
  62. - explicit LeafPageDecoder(DatabasePageReader* db_reader) noexcept;
  63. + explicit LeafPageDecoder() noexcept = default;
  64. ~LeafPageDecoder() noexcept = default;
  65. LeafPageDecoder(const LeafPageDecoder&) = delete;
  66. @@ -150,6 +150,15 @@ class LeafPageDecoder {
  67. // read as long as CanAdvance() returns true.
  68. bool TryAdvance();
  69. + // Initialize with DatabasePageReader
  70. + void Initialize(DatabasePageReader* db_reader);
  71. +
  72. + // Reset internal DatabasePageReader
  73. + void Reset();
  74. +
  75. + // True if DatabasePageReader is valid
  76. + bool IsValid() { return (db_reader_ != nullptr); }
  77. +
  78. // True if the given reader may point to an inner page in a table B-tree.
  79. //
  80. // The last ReadPage() call on |db_reader| must have succeeded.
  81. @@ -163,14 +172,14 @@ class LeafPageDecoder {
  82. static int ComputeCellCount(DatabasePageReader* db_reader);
  83. // The number of the B-tree page this reader is reading.
  84. - const int64_t page_id_;
  85. + int64_t page_id_;
  86. // Used to read the tree page.
  87. //
  88. // Raw pointer usage is acceptable because this instance's owner is expected
  89. // to ensure that the DatabasePageReader outlives this.
  90. - DatabasePageReader* const db_reader_;
  91. + DatabasePageReader* db_reader_;
  92. // Caches the ComputeCellCount() value for this reader's page.
  93. - const int cell_count_ = ComputeCellCount(db_reader_);
  94. + int cell_count_;
  95. // The reader's cursor state.
  96. //
  97. diff --git a/sql/recover_module/cursor.cc b/sql/recover_module/cursor.cc
  98. index 0029ff9295..42548bc4b5 100644
  99. --- a/sql/recover_module/cursor.cc
  100. +++ b/sql/recover_module/cursor.cc
  101. @@ -26,7 +26,7 @@ VirtualCursor::~VirtualCursor() {
  102. int VirtualCursor::First() {
  103. DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  104. inner_decoders_.clear();
  105. - leaf_decoder_ = nullptr;
  106. + leaf_decoder_.Reset();
  107. AppendPageDecoder(table_->root_page_id());
  108. return Next();
  109. @@ -36,18 +36,18 @@ int VirtualCursor::Next() {
  110. DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  111. record_reader_.Reset();
  112. - while (!inner_decoders_.empty() || leaf_decoder_.get()) {
  113. - if (leaf_decoder_.get()) {
  114. - if (!leaf_decoder_->CanAdvance()) {
  115. + while (!inner_decoders_.empty() || leaf_decoder_.IsValid()) {
  116. + if (leaf_decoder_.IsValid()) {
  117. + if (!leaf_decoder_.CanAdvance()) {
  118. // The leaf has been exhausted. Remove it from the DFS stack.
  119. - leaf_decoder_ = nullptr;
  120. + leaf_decoder_.Reset();
  121. continue;
  122. }
  123. - if (!leaf_decoder_->TryAdvance())
  124. + if (!leaf_decoder_.TryAdvance())
  125. continue;
  126. - if (!payload_reader_.Initialize(leaf_decoder_->last_record_size(),
  127. - leaf_decoder_->last_record_offset())) {
  128. + if (!payload_reader_.Initialize(leaf_decoder_.last_record_size(),
  129. + leaf_decoder_.last_record_offset())) {
  130. continue;
  131. }
  132. if (!record_reader_.Initialize())
  133. @@ -99,13 +99,13 @@ int VirtualCursor::ReadColumn(int column_index,
  134. int64_t VirtualCursor::RowId() {
  135. DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  136. DCHECK(record_reader_.IsInitialized());
  137. - DCHECK(leaf_decoder_.get());
  138. - return leaf_decoder_->last_record_rowid();
  139. + DCHECK(leaf_decoder_.IsValid());
  140. + return leaf_decoder_.last_record_rowid();
  141. }
  142. void VirtualCursor::AppendPageDecoder(int page_id) {
  143. DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  144. - DCHECK(leaf_decoder_.get() == nullptr)
  145. + DCHECK(!leaf_decoder_.IsValid())
  146. << __func__
  147. << " must only be called when the current path has no leaf decoder";
  148. @@ -113,7 +113,7 @@ void VirtualCursor::AppendPageDecoder(int page_id) {
  149. return;
  150. if (LeafPageDecoder::IsOnValidPage(&db_reader_)) {
  151. - leaf_decoder_ = std::make_unique<LeafPageDecoder>(&db_reader_);
  152. + leaf_decoder_.Initialize(&db_reader_);
  153. return;
  154. }
  155. diff --git a/sql/recover_module/cursor.h b/sql/recover_module/cursor.h
  156. index afcd6900e1..b15c31d425 100644
  157. --- a/sql/recover_module/cursor.h
  158. +++ b/sql/recover_module/cursor.h
  159. @@ -129,7 +129,7 @@ class VirtualCursor {
  160. std::vector<std::unique_ptr<InnerPageDecoder>> inner_decoders_;
  161. // Decodes the leaf page containing records.
  162. - std::unique_ptr<LeafPageDecoder> leaf_decoder_;
  163. + LeafPageDecoder leaf_decoder_;
  164. SEQUENCE_CHECKER(sequence_checker_);
  165. };
  166. diff --git a/sql/recover_module/pager.cc b/sql/recover_module/pager.cc
  167. index 58e75de270..5fe96204e5 100644
  168. --- a/sql/recover_module/pager.cc
  169. +++ b/sql/recover_module/pager.cc
  170. @@ -23,8 +23,7 @@ static_assert(DatabasePageReader::kMaxPageId <= std::numeric_limits<int>::max(),
  171. "ints are not appropriate for representing page IDs");
  172. DatabasePageReader::DatabasePageReader(VirtualTable* table)
  173. - : page_data_(std::make_unique<uint8_t[]>(table->page_size())),
  174. - table_(table) {
  175. + : page_data_(), table_(table) {
  176. DCHECK(table != nullptr);
  177. DCHECK(IsValidPageSize(table->page_size()));
  178. }
  179. @@ -57,8 +56,8 @@ int DatabasePageReader::ReadPage(int page_id) {
  180. std::numeric_limits<int64_t>::max(),
  181. "The |read_offset| computation above may overflow");
  182. - int sqlite_status =
  183. - RawRead(sqlite_file, read_size, read_offset, page_data_.get());
  184. + int sqlite_status = RawRead(sqlite_file, read_size, read_offset,
  185. + const_cast<uint8_t*>(page_data_.data()));
  186. // |page_id_| needs to be set to kInvalidPageId if the read failed.
  187. // Otherwise, future ReadPage() calls with the previous |page_id_| value
  188. diff --git a/sql/recover_module/pager.h b/sql/recover_module/pager.h
  189. index 0e388ddc3b..99314e30ff 100644
  190. --- a/sql/recover_module/pager.h
  191. +++ b/sql/recover_module/pager.h
  192. @@ -5,6 +5,7 @@
  193. #ifndef SQL_RECOVER_MODULE_PAGER_H_
  194. #define SQL_RECOVER_MODULE_PAGER_H_
  195. +#include <array>
  196. #include <cstdint>
  197. #include <memory>
  198. @@ -70,7 +71,7 @@ class DatabasePageReader {
  199. DCHECK_CALLED_ON_VALID_SEQUENCE(sequence_checker_);
  200. DCHECK_NE(page_id_, kInvalidPageId)
  201. << "Successful ReadPage() required before accessing pager state";
  202. - return page_data_.get();
  203. + return page_data_.data();
  204. }
  205. // The number of bytes in the page read by the last ReadPage() call.
  206. @@ -137,7 +138,7 @@ class DatabasePageReader {
  207. int page_id_ = kInvalidPageId;
  208. // Stores the bytes of the last page successfully read by ReadPage().
  209. // The content is undefined if the last call to ReadPage() did not succeed.
  210. - const std::unique_ptr<uint8_t[]> page_data_;
  211. + const std::array<uint8_t, kMaxPageSize> page_data_;
  212. // Raw pointer usage is acceptable because this instance's owner is expected
  213. // to ensure that the VirtualTable outlives this.
  214. VirtualTable* const table_;