src/developer/feedback/crashpad_agent/database.h

// Copyright 2019 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef SRC_DEVELOPER_FEEDBACk_CRASHPAD_AGENT_DATABASE_H_
#define SRC_DEVELOPER_FEEDBACk_CRASHPAD_AGENT_DATABASE_H_

#include <fuchsia/mem/cpp/fidl.h>

#include <map>
#include <optional>
#include <unordered_map>

#include "src/developer/feedback/crashpad_agent/config.h"
#include "src/developer/feedback/crashpad_agent/upload_report.h"
#include "src/lib/fxl/macros.h"
#include "third_party/crashpad/client/crash_report_database.h"
#include "third_party/crashpad/util/misc/uuid.h"

namespace feedback {

// Wrapper around the Crashpad database that also stores annotations.
class Database {
 public:
  static std::unique_ptr<Database> TryCreate(CrashpadDatabaseConfig config);

  // Returns the path to the underlying Crashpad database.
  const char* path();

  // Make a new report in |database_|.
  //
  // Return false if there is an error with the database.
  bool MakeNewReport(const std::map<std::string, fuchsia::mem::Buffer>& attachments,
                     const std::optional<fuchsia::mem::Buffer>& minidump,
                     const std::map<std::string, std::string>& annotations,
                     crashpad::UUID* local_report_id);

  // Construct and return the |UploadReport| for |local_report_id|.
  //
  // If there's an error with |database_| return nullptr.
  std::unique_ptr<UploadReport> GetUploadReport(const crashpad::UUID& local_report_id);

  // Record |upload_report| as uploaded and clean up the report's annotations
  //
  // Return false if there is an error with |database_|.
  bool MarkAsUploaded(std::unique_ptr<UploadReport> upload_report,
                      const std::string& server_report_id);

  // Record |upload_report| as having too many upload attempts and clean up the report's annotations
  //
  // Return false if there is an error with |database_|.
  bool MarkAsTooManyUploadAttempts(std::unique_ptr<UploadReport> upload_report);

  // Record |local_report_id| as skipped in |database_| and clean up the report's annotations
  //
  // Return false if there is an error with |database_|.
  bool Archive(const crashpad::UUID& local_report_id);

  // Deletes oldest (determined by creation_time) crash reports to keep |database_| under a maximum
  // size read from |config_| and removes expired lockfiles, metadata without report files, report
  // files without metadata from |database_|, and orphaned attachments. Removes all data from
  // |additional_data_| that is not in |database_|.
  //
  // Return the number of reports that are removed from |database_|.
  size_t GarbageCollect();

  ~Database() = default;

 private:
  // Allows for crashpad::UUID to be used as the key in an |std::unordered_map|.
  struct UUIDHasher {
    size_t operator()(const crashpad::UUID& uuid) const {
      return std::hash<std::string>()(uuid.ToString());
    }
  };

  // Data pertinent to a crash report not stored in |database_|.
  struct AdditionalData {
    std::map<std::string, std::string> annotations;
    bool has_minidump;
  };

  Database(CrashpadDatabaseConfig config, std::unique_ptr<crashpad::CrashReportDatabase> database);

  // Removes |local_report_id| from |additional_data_|.
  void CleanUp(const crashpad::UUID& local_report_id);

  const CrashpadDatabaseConfig config_;
  std::unique_ptr<crashpad::CrashReportDatabase> database_;
  std::unordered_map<crashpad::UUID, AdditionalData, UUIDHasher> additional_data_;

  FXL_DISALLOW_COPY_AND_ASSIGN(Database);
};

}  // namespace feedback
#endif  // SRC_DEVELOPER_FEEDBACk_CRASHPAD_AGENT_DATABASE_H_