/* Copyright 2017 - 2025 R. Thomas
 * Copyright 2017 - 2025 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/debug/Debug.hpp"
#include "LIEF/PE/Section.hpp"
#include "PE/Structures.hpp"

#include "frozen.hpp"
#include "logging.hpp"
#include "spdlog/fmt/fmt.h"
#include "overflow_check.hpp"

namespace LIEF {
namespace PE {

span<uint8_t> Debug::get_payload(Section& section, uint32_t /*rva*/,
                                 uint32_t offset, uint32_t size)
{
  span<uint8_t> content = section.writable_content();
  if (size == 0 || content.empty() || content.size() < size) {
    return {};
  }

  int32_t rel_offset = (int32_t)offset - (int32_t)section.offset();
  if (rel_offset < 0) {
    return {};
  }

  if ((size_t)rel_offset >= content.size() || (rel_offset + size) > content.size()) {
    return {};
  }

  if (check_overflow<uint64_t>((uint32_t)rel_offset, size, content.size())) {
    return {};
  }

  return content.subspan((uint32_t)rel_offset, size);
}

span<uint8_t> Debug::get_payload(Section& section, const details::pe_debug& hdr) {
  return get_payload(section, hdr.AddressOfRawData, hdr.PointerToRawData,
                     hdr.SizeOfData);
}

Debug::Debug(const details::pe_debug& debug_s, Section* sec) :
  type_{static_cast<TYPES>(debug_s.Type)},
  characteristics_{debug_s.Characteristics},
  timestamp_{debug_s.TimeDateStamp},
  major_version_{debug_s.MajorVersion},
  minor_version_{debug_s.MinorVersion},
  sizeof_data_{debug_s.SizeOfData},
  addressof_rawdata_{debug_s.AddressOfRawData},
  pointerto_rawdata_{debug_s.PointerToRawData},
  section_{sec}
{}

span<uint8_t> Debug::payload() {
  if (section_ == nullptr) {
    return {};
  }
  return get_payload(*section_, *this);
}

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

std::string Debug::to_string() const {
  using namespace fmt;
  std::ostringstream os;
  os << format("Characteristics:     0x{:x}\n", characteristics())
     << format("Timestamp:           0x{:x}\n", timestamp())
     << format("Major/Minor version: {}.{}\n", major_version(), minor_version())
     << format("Type:                {}\n", PE::to_string(type()))
     << format("Size of data:        0x{:x}\n", sizeof_data())
     << format("Address of rawdata:  0x{:x}\n", addressof_rawdata())
     << format("Pointer to rawdata:  0x{:x}", pointerto_rawdata());
  return os.str();
}

const char* to_string(Debug::TYPES e) {
  #define ENTRY(X) std::pair(Debug::TYPES::X, #X)
  STRING_MAP enums2str {
    ENTRY(UNKNOWN),
    ENTRY(COFF),
    ENTRY(CODEVIEW),
    ENTRY(FPO),
    ENTRY(MISC),
    ENTRY(EXCEPTION),
    ENTRY(FIXUP),
    ENTRY(OMAP_TO_SRC),
    ENTRY(OMAP_FROM_SRC),
    ENTRY(BORLAND),
    ENTRY(RESERVED10),
    ENTRY(CLSID),
    ENTRY(VC_FEATURE),
    ENTRY(POGO),
    ENTRY(ILTCG),
    ENTRY(MPX),
    ENTRY(REPRO),
    ENTRY(PDBCHECKSUM),
    ENTRY(EX_DLLCHARACTERISTICS),
  };
  #undef ENTRY

  if (const auto it = enums2str.find(e); it != enums2str.end()) {
    return it->second;
  }
  return "UNKNOWN";
}

}
}