/* Copyright 2017 - 2026 R. Thomas
 * Copyright 2017 - 2026 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.
 */
#include "LIEF/Visitor.hpp"

#include "LIEF/PE/DosHeader.hpp"
#include "LIEF/PE/enums.hpp"
#include "PE/Structures.hpp"

#include "spdlog/fmt/fmt.h"
#include "spdlog/fmt/ranges.h"


namespace LIEF::PE {

DosHeader::DosHeader(const details::pe_dos_header& header) :
  magic_{header.Magic},
  used_bytes_in_last_page_{header.UsedBytesInTheLastPage},
  file_sz_in_pages_{header.FileSizeInPages},
  nb_relocations_{header.NumberOfRelocationItems},
  header_sz_in_paragraphs_{header.HeaderSizeInParagraphs},
  min_extra_paragraphs_{header.MinimumExtraParagraphs},
  max_extra_paragraphs_{header.MaximumExtraParagraphs},
  init_relative_ss_{header.InitialRelativeSS},
  init_sp_{header.InitialSP},
  checksum_{header.Checksum},
  init_ip_{header.InitialIP},
  init_rel_cs_{header.InitialRelativeCS},
  addr_reloc_table_{header.AddressOfRelocationTable},
  overlay_number_{header.OverlayNumber},
  oem_id_{header.OEMid},
  oem_info_{header.OEMinfo},
  addr_new_exe_header_{header.AddressOfNewExeHeader} {
  std::copy(std::begin(header.Reserved), std::end(header.Reserved),
            std::begin(reserved_));

  std::copy(std::begin(header.Reserved2), std::end(header.Reserved2),
            std::begin(reserved2_));
}

void DosHeader::accept(LIEF::Visitor& visitor) const {
  visitor.visit(*this);
}


DosHeader DosHeader::create(PE_TYPE type) {
  DosHeader hdr;
  hdr.magic(DosHeader::MAGIC);
  hdr.header_size_in_paragraphs(4);
  hdr.addressof_relocation_table(0x40);
  hdr.file_size_in_pages(3);
  hdr.maximum_extra_paragraphs(0xff);
  hdr.initial_sp(0xb8);
  hdr.used_bytes_in_last_page(0x90);
  hdr.addressof_new_exeheader(type == PE_TYPE::PE32 ? 0xe8 : 0xf0);
  return hdr;
}

std::ostream& operator<<(std::ostream& os, const DosHeader& entry) {
  os << fmt::format("Magic:                       {:#x}\n", entry.magic())
     << fmt::format("Used bytes in last page:     {:#x}\n",
                    entry.used_bytes_in_last_page())
     << fmt::format("File size in pages:          {:#x}\n",
                    entry.file_size_in_pages())
     << fmt::format("Number of relocation:        {:#x}\n",
                    entry.numberof_relocation())
     << fmt::format("Header size in paragraphs:   {:#x}\n",
                    entry.header_size_in_paragraphs())
     << fmt::format("Min extra paragraphs:        {:#x}\n",
                    entry.minimum_extra_paragraphs())
     << fmt::format("Max extra paragraphs:        {:#x}\n",
                    entry.maximum_extra_paragraphs())
     << fmt::format("Initial relative SS:         {:#x}\n",
                    entry.initial_relative_ss())
     << fmt::format("Initial SP:                  {:#x}\n", entry.initial_sp())
     << fmt::format("Checksum:                    {:#x}\n", entry.checksum())
     << fmt::format("Initial IP:                  {:#x}\n", entry.initial_ip())
     << fmt::format("Initial relative CS:         {:#x}\n",
                    entry.initial_relative_cs())
     << fmt::format("Address of relocation table: {:#x}\n",
                    entry.addressof_relocation_table())
     << fmt::format("Overlay number:              {:#x}\n", entry.overlay_number())
     << fmt::format("OEM id:                      {:#x}\n", entry.oem_id())
     << fmt::format("OEM info:                    {:#x}\n", entry.oem_info())
     << fmt::format("Address of new exe-header:   {:#x}\n",
                    entry.addressof_new_exeheader())
     << fmt::format("Reserved1:                   {}\n", entry.reserved())
     << fmt::format("Reserved2:                   {}\n", entry.reserved2());
  return os;
}

}