//===-- ObjectContainerMachOFileset.h ---------------------------*- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

#ifndef LLDB_SOURCE_PLUGINS_OBJECTCONTAINER_MACH_O_FILESET_OBJECTCONTAINERMADCHOFILESET_H
#define LLDB_SOURCE_PLUGINS_OBJECTCONTAINER_MACH_O_FILESET_OBJECTCONTAINERMADCHOFILESET_H

#include "lldb/Host/SafeMachO.h"
#include "lldb/Symbol/ObjectContainer.h"
#include "lldb/Utility/FileSpec.h"

namespace lldb_private {

class ObjectContainerMachOFileset : public lldb_private::ObjectContainer {
public:
  ObjectContainerMachOFileset(const lldb::ModuleSP &module_sp,
                              lldb::DataBufferSP &data_sp,
                              lldb::offset_t data_offset,
                              const lldb_private::FileSpec *file,
                              lldb::offset_t offset, lldb::offset_t length);

  ObjectContainerMachOFileset(const lldb::ModuleSP &module_sp,
                              lldb::WritableDataBufferSP data_sp,
                              const lldb::ProcessSP &process_sp,
                              lldb::addr_t header_addr);

  ~ObjectContainerMachOFileset() override;

  static void Initialize();
  static void Terminate();

  static llvm::StringRef GetPluginNameStatic() { return "mach-o-fileset"; }

  static llvm::StringRef GetPluginDescriptionStatic() {
    return "Mach-O Fileset container reader.";
  }

  static lldb_private::ObjectContainer *
  CreateInstance(const lldb::ModuleSP &module_sp, lldb::DataBufferSP &data_sp,
                 lldb::offset_t data_offset, const lldb_private::FileSpec *file,
                 lldb::offset_t offset, lldb::offset_t length);

  static lldb_private::ObjectContainer *CreateMemoryInstance(
      const lldb::ModuleSP &module_sp, lldb::WritableDataBufferSP data_sp,
      const lldb::ProcessSP &process_sp, lldb::addr_t header_addr);

  static size_t GetModuleSpecifications(const lldb_private::FileSpec &file,
                                        lldb::DataBufferSP &data_sp,
                                        lldb::offset_t data_offset,
                                        lldb::offset_t file_offset,
                                        lldb::offset_t length,
                                        lldb_private::ModuleSpecList &specs);

  static bool MagicBytesMatch(const lldb_private::DataExtractor &data);
  static bool MagicBytesMatch(lldb::DataBufferSP data_sp,
                              lldb::addr_t data_offset,
                              lldb::addr_t data_length);

  bool ParseHeader() override;

  size_t GetNumObjects() const override { return m_entries.size(); }

  lldb::ObjectFileSP GetObjectFile(const lldb_private::FileSpec *file) override;

  llvm::StringRef GetPluginName() override { return GetPluginNameStatic(); }

  struct Entry {
    Entry(uint64_t vmaddr, uint64_t fileoff, std::string id)
        : vmaddr(vmaddr), fileoff(fileoff), id(id) {}
    uint64_t vmaddr;
    uint64_t fileoff;
    std::string id;
  };

  Entry *FindEntry(llvm::StringRef id);

private:
  static bool ParseHeader(lldb_private::DataExtractor &data,
                          const lldb_private::FileSpec &file,
                          lldb::offset_t file_offset,
                          std::vector<Entry> &entries);

  std::vector<Entry> m_entries;
  lldb::ProcessWP m_process_wp;
  const lldb::addr_t m_memory_addr;
};

} // namespace lldb_private

#endif // LLDB_SOURCE_PLUGINS_OBJECTCONTAINER_MACH_O_FILESET_OBJECTCONTAINERMADCHOFILESET_H