sql/meta_table.h

   1 // Copyright 2012 The Chromium Authors
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef SQL_META_TABLE_H_
   6 #define SQL_META_TABLE_H_
   7
   8 #include <cstdint>
   9 #include <string>
  10
  11 #include "base/component_export.h"
  12 #include "base/memory/raw_ptr.h"
  13 #include "base/strings/string_piece_forward.h"
  14
  15 namespace sql {
  16
  17 class Database;
  18
  19 // Creates and manages a table to store generic metadata. The features provided
  20 // are:
  21 // * An interface for storing multi-typed key-value data.
  22 // * Helper methods to assist in database schema version control.
  23 // * Historical data on past attempts to mmap the database to make it possible
  24 //   to avoid unconditionally retrying to load broken databases.
  25 class COMPONENT_EXPORT(SQL) MetaTable {
  26  public:
  27   MetaTable();
  28   MetaTable(const MetaTable&) = delete;
  29   MetaTable& operator=(const MetaTable&) = delete;
  30   MetaTable(MetaTable&&) = delete;
  31   MetaTable& operator=(MetaTable&&) = delete;
  32   ~MetaTable();
  33
  34   // Values for Get/SetMmapStatus(). `kMmapFailure` indicates that there was at
  35   // some point a read error and the database should not be memory-mapped, while
  36   // `kMmapSuccess` indicates that the entire file was read at some point and
  37   // can be memory-mapped without constraint.
  38   static constexpr int64_t kMmapFailure = -2;
  39   static constexpr int64_t kMmapSuccess = -1;
  40
  41   // Returns true if the 'meta' table exists.
  42   static bool DoesTableExist(Database* db);
  43
  44   // Deletes the 'meta' table if it exists, returning false if an internal error
  45   // occurred during the deletion and true otherwise (no matter whether the
  46   // table existed).
  47   static bool DeleteTableForTesting(Database* db);
  48
  49   // If the current version of the database is less than
  50   // `lowest_supported_version`, or the current version is less than the
  51   // database's least compatible version, razes the database. To only enforce
  52   // the latter, pass `kNoLowestSupportedVersion` for
  53   // `lowest_supported_version`.
  54   //
  55   // TODO(crbug.com/1228463): At this time the database is razed IFF meta exists
  56   // and contains a version row with the value not satisfying the constraints.
  57   // It may make sense to also raze if meta exists but has no version row, or if
  58   // meta doesn't exist. In those cases if the database is not already empty, it
  59   // probably resulted from a broken initialization.
  60   // TODO(crbug.com/1228463): Folding this into Init() would allow enforcing
  61   // the version constraint, but Init() is often called in a transaction.
  62   static constexpr int kNoLowestSupportedVersion = 0;
  63   static void RazeIfIncompatible(Database* db,
  64                                  int lowest_supported_version,
  65                                  int current_version);
  66
  67   // Used to tuck some data into the meta table about mmap status. The value
  68   // represents how much data in bytes has successfully been read from the
  69   // database, or `kMmapFailure` or `kMmapSuccess`.
  70   static bool GetMmapStatus(Database* db, int64_t* status);
  71   static bool SetMmapStatus(Database* db, int64_t status);
  72
  73   // Initializes the MetaTableHelper, providing the `Database` pointer and
  74   // creating the meta table if necessary. Must be called before any other
  75   // non-static methods. For new tables, it will initialize the version number
  76   // to `version` and the compatible version number to `compatible_version`.
  77   // Versions must be greater than 0 to distinguish missing versions (see
  78   // GetVersionNumber()). If there was no meta table (proxy for a fresh
  79   // database), mmap status is set to `kMmapSuccess`.
  80   bool Init(Database* db, int version, int compatible_version);
  81
  82   // Resets this MetaTable object, making another call to Init() possible.
  83   void Reset();
  84
  85   // The version number of the database. This should be the version number of
  86   // the creator of the file. The version number will be 0 if there is no
  87   // previously set version number.
  88   //
  89   // See also Get/SetCompatibleVersionNumber().
  90   void SetVersionNumber(int version);
  91   int GetVersionNumber();
  92
  93   // The compatible version number is the lowest current version embedded in
  94   // Chrome code that can still use this database. This is usually the same as
  95   // the current version. In some limited cases, such as adding a column without
  96   // a NOT NULL constraint, the SQL queries embedded in older code can still
  97   // execute successfully.
  98   //
  99   // For example, if an optional column is added to a table in version 3, the
 100   // new code will set the version to 3, and the compatible version to 2, since
 101   // the code expecting version 2 databases can still read and write the table.
 102   //
 103   // Rule of thumb: check the version number when you're upgrading, but check
 104   // the compatible version number to see if you can use the file at all. If
 105   // it's larger than you code is expecting, fail.
 106   //
 107   // The compatible version number will be 0 if there is no previously set
 108   // compatible version number.
 109   void SetCompatibleVersionNumber(int version);
 110   int GetCompatibleVersionNumber();
 111
 112   // Set the given arbitrary key with the given data. Returns true on success.
 113   bool SetValue(base::StringPiece key, const std::string& value);
 114   bool SetValue(base::StringPiece key, int64_t value);
 115
 116   // Retrieves the value associated with the given key. This will use sqlite's
 117   // type conversion rules. It will return true on success.
 118   bool GetValue(base::StringPiece key, std::string* value);
 119   bool GetValue(base::StringPiece key, int* value);
 120   bool GetValue(base::StringPiece key, int64_t* value);
 121
 122   // Deletes the key from the table.
 123   bool DeleteKey(base::StringPiece key);
 124
 125  private:
 126   raw_ptr<Database> db_ = nullptr;
 127 };
 128
 129 }  // namespace sql
 130
 131 #endif  // SQL_META_TABLE_H_