CodeViewPDB.hpp

/* Copyright 2017 - 2024 R. Thomas
 * Copyright 2017 - 2024 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_PE_DEBUG_CODE_VIEW_PDB_H
#define LIEF_PE_DEBUG_CODE_VIEW_PDB_H
#include "LIEF/PE/debug/CodeView.hpp"
 
#include <cstdint>
#include <array>
#include <ostream>
 
namespace LIEF {
namespace PE {
class Parser;
class Builder;
 
namespace details {
struct pe_pdb_70;
}
 
class LIEF_API CodeViewPDB : public CodeView {
  friend class Parser;
  friend class Builder;
 
  public:
  using signature_t = std::array<uint8_t, 16>;
  CodeViewPDB() = default;
  CodeViewPDB(const details::pe_debug& debug_info,
              const details::pe_pdb_70& pdb_70);
  CodeViewPDB(const CodeViewPDB& other) = default;
  CodeViewPDB& operator=(const CodeViewPDB& other) = default;
 
  std::string guid() const;
 
  uint32_t age() const {
    return age_;
  }
 
  const signature_t& signature() const {
    return signature_;
  }
 
  const std::string& filename() const {
    return filename_;
  }
 
  void age(uint32_t age) {
    age_ = age;
  }
 
  void signature(const signature_t& sig) {
    signature_ = sig;
  }
 
  void filename(std::string filename) {
    filename_ = std::move(filename);
  }
 
  std::unique_ptr<Debug> clone() const override {
    return std::unique_ptr<Debug>(new CodeViewPDB(*this));
  }
 
  static bool classof(const Debug* debug) {
    if (!CodeView::classof(debug)) {
      return false;
    }
    const auto& cv = static_cast<const CodeView&>(*debug);
 
    return cv.signature() == SIGNATURES::PDB_20 ||
           cv.signature() == SIGNATURES::PDB_70;
  }
 
  void accept(Visitor& visitor) const override;
  LIEF_API friend std::ostream& operator<<(std::ostream& os, const CodeViewPDB& entry);
 
  ~CodeViewPDB() override = default;
 
  protected:
  uint32_t    age_ = 0;
  signature_t signature_;
  std::string filename_;
};
 
} // namespace PE
} // namespace LIEF
#endif