/* Copyright 2017 - 2022 R. Thomas
 * Copyright 2017 - 2022 Quarkslab
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
#ifndef LIEF_ELF_HEADER_H_
#define LIEF_ELF_HEADER_H_

#include <iostream>
#include <array>
#include <vector>

#include "LIEF/Object.hpp"
#include "LIEF/visibility.h"
#include "LIEF/Abstract/enums.hpp"

#include "LIEF/ELF/enums.hpp"

namespace LIEF {
namespace ELF {
class Parser;

namespace details {
struct Elf32_Ehdr;
struct Elf64_Ehdr;
}

//! Class which represents the ELF's header. This is the ELF structure
//! that starts an ELF file.
class LIEF_API Header : public Object {
  friend class Parser;
  public:
  using identity_t = std::array<uint8_t, 16>;
  using abstract_architecture_t = std::pair<ARCHITECTURES, std::set<MODES>>;

  template<class T>
  using flags_list_t = std::set<T>;

  using arm_flags_list_t     = flags_list_t<ARM_EFLAGS>;
  using mips_flags_list_t    = flags_list_t<MIPS_EFLAGS>;
  using hexagon_flags_list_t = flags_list_t<HEXAGON_EFLAGS>;
  using ppc64_flags_list_t   = flags_list_t<PPC64_EFLAGS>;

  public:
  Header();
  Header(const details::Elf32_Ehdr& header);
  Header(const details::Elf64_Ehdr& header);

  Header& operator=(const Header&);
  Header(const Header&);

  virtual ~Header();

  //! Define the object file type. (e.g. executable, library...)
  E_TYPE file_type() const;

  //! LIEF abstract object type
  OBJECT_TYPES abstract_object_type() const;

  //! Target architecture
  ARCH machine_type() const;

  //! LIEF abstract architecture
  //!
  //! It returns Empty if it can't be abstracted
  abstract_architecture_t abstract_architecture() const;

  //! LIEF abstract endianness
  ENDIANNESS abstract_endianness() const;

  //! Version of the object file format
  VERSION object_file_version() const;

  //! Executable entrypoint
  uint64_t entrypoint() const;

  //! Offset of program table (also known as segments table)
  uint64_t program_headers_offset() const;

  //! Offset of section table
  uint64_t section_headers_offset() const;

  //! Processor-specific flags
  uint32_t processor_flag() const;

  //! Check if the given flag is present in processor_flag()
  bool has(ARM_EFLAGS f) const;

  //! Return a list of ARM_EFLAGS present in processor_flag()
  arm_flags_list_t arm_flags_list() const;

  //! Check if the given flag is present in processor_flag()
  bool has(MIPS_EFLAGS f) const;

  //! Return a list of MIPS_EFLAGS present in processor_flag()
  mips_flags_list_t mips_flags_list() const;

  //! Check if the given flag is present in processor_flag()
  bool has(PPC64_EFLAGS f) const;

  //! Return a list of PPC64_EFLAGS present in processor_flag()
  ppc64_flags_list_t ppc64_flags_list() const;

  //! Check if the given flag is present in processor_flag()
  bool has(HEXAGON_EFLAGS f) const;

  //! Return a list of HEXAGON_EFLAGS present in processor_flag()
  hexagon_flags_list_t hexagon_flags_list() const;

  //! Size of the current header
  //!
  //! This size should be 64 for an ``ELF64`` binary and 52 for an ``ELF32``.
  uint32_t header_size() const;

  //! Return the size of a ``Segment header``
  //!
  //! This size should be 56 for a ``ELF64`` binary and 32 for an ``ELF32``.
  uint32_t program_header_size() const;

  //! Return the the number of segments
  uint32_t numberof_segments() const;

  //! Return the size of a ``Section header``
  //!
  //! This size should be 64 for a ``ELF64`` binary and 40 for an ``ELF32``.
  uint32_t section_header_size() const;

  //! Return the number of sections
  //!
  //! @warning This value could differ from the real number of sections
  //! present in the binary. It must be taken as an *indication*
  uint32_t numberof_sections() const;

  //! Return the section's index which contains sections' names
  uint32_t section_name_table_idx() const;

  //! Return the ELF identity as an ``std::array``
  identity_t&       identity();
  const identity_t& identity() const;

  //! Return the object's class. ``ELF64`` or ``ELF32``
  ELF_CLASS identity_class() const;

  //! Specify the data encoding
  ELF_DATA identity_data() const;

  //! @see object_file_version
  VERSION identity_version() const;

  //! Identifies the version of the ABI for which the object is prepared
  OS_ABI identity_os_abi() const;

  //! ABI Version
  uint32_t identity_abi_version() const;

  void file_type(E_TYPE type);
  void machine_type(ARCH machineType);
  void object_file_version(VERSION version);
  void entrypoint(uint64_t entryPoint);
  void program_headers_offset(uint64_t programHeaderOffset);
  void section_headers_offset(uint64_t sectionHeaderOffset);
  void processor_flag(uint32_t processorFlag);
  void header_size(uint32_t headerSize);
  void program_header_size(uint32_t programHeaderSize);
  void numberof_segments(uint32_t n);
  void section_header_size(uint32_t sizeOfSectionHeaderEntries);
  void numberof_sections(uint32_t n);
  void section_name_table_idx(uint32_t sectionNameStringTableIdx);
  void identity(const std::string& identity);
  void identity(const identity_t& identity);
  void identity_class(ELF_CLASS i_class);
  void identity_data(ELF_DATA data);
  void identity_version(VERSION version);
  void identity_os_abi(OS_ABI osabi);
  void identity_abi_version(uint32_t version);

  void accept(Visitor& visitor) const override;

  bool operator==(const Header& rhs) const;
  bool operator!=(const Header& rhs) const;

  LIEF_API friend std::ostream& operator<<(std::ostream& os, const Header& hdr);

  private:
  //! Field which represents ElfXX_Ehdr->e_ident
  identity_t identity_;

  //! Field which represents ElfXX_Ehdr->e_type
  E_TYPE file_type_;

  //! Field which represents ElfXX_Ehdr->e_machine
  ARCH machine_type_;

  //! Field which represents ElfXX_Ehdr->e_version
  VERSION object_file_version_;

  //! Field which represents ElfXX_Ehdr->e_entry
  uint64_t entrypoint_;

  //! Field which represents ElfXX_Ehdr->e_phoff
  uint64_t program_headers_offset_;

  //! Field which represents ElfXX_Ehdr->e_shoff
  uint64_t section_headers_offset_;

  //! Field which represents ElfXX_Ehdr->e_flags
  uint32_t processor_flags_;

  //! Field which represents ElfXX_Ehdr->e_ehsize
  uint32_t header_size_;

  //! Field which represents ElfXX_Ehdr->e_phentsize
  uint32_t program_header_size_;

  //! Field which represents ElfXX_Ehdr->e_phnum
  uint32_t numberof_segments_;

  //! Field which represents ElfXX_Ehdr->e_shentsize
  uint32_t section_header_size_;

  //! Field which represents ElfXX_Ehdr->e_shnum
  uint32_t numberof_sections_;

  //! Field which represents ElfXX_Ehdr->e_shstrndx
  uint32_t section_string_table_idx_;

};
}
}
#endif