ResourceNode.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_RESOURCE_NODE_H
#define LIEF_PE_RESOURCE_NODE_H
#include <string>
#include <vector>
#include <memory>
 
#include "LIEF/Object.hpp"
#include "LIEF/visibility.h"
#include "LIEF/iterators.hpp"
 
namespace LIEF {
namespace PE {
 
class ResourceDirectory;
class ResourceData;
 
class Parser;
class Builder;
 
class LIEF_API ResourceNode : public Object {
 
  friend class Parser;
  friend class Builder;
 
  public:
  using childs_t        = std::vector<std::unique_ptr<ResourceNode>>;
  using it_childs       = ref_iterator<childs_t&, ResourceNode*>;
  using it_const_childs = const_ref_iterator<const childs_t&, ResourceNode*>;
 
  enum class TYPE {
    UNKNOWN = 0,
    DATA,
    DIRECTORY,
  };
 
  ResourceNode(const ResourceNode& other);
  ResourceNode& operator=(const ResourceNode& other);
 
  ResourceNode(ResourceNode&& other);
  ResourceNode& operator=(ResourceNode&& other);
 
  void swap(ResourceNode& other);
 
  ~ResourceNode() override;
 
  virtual std::unique_ptr<ResourceNode> clone() const = 0;
 
  uint32_t id() const {
    return id_;
  }
 
  const std::u16string& name() const {
    return name_;
  }
 
  it_childs childs() {
    return childs_;
  }
  it_const_childs childs() const {
    return childs_;
  }
 
  bool has_name() const {
    return static_cast<bool>(id() & 0x80000000);
  }
 
  uint32_t depth() const {
    return depth_;
  }
 
  bool is_directory() const {
    return type_ == TYPE::DIRECTORY;
  }
 
  bool is_data() const {
    return type_ == TYPE::DATA;
  }
 
  void id(uint32_t id) {
    id_ = id;
  }
  void name(const std::string& name);
 
  void name(std::u16string name) {
    name_ = std::move(name);
  }
 
  ResourceNode& add_child(const ResourceDirectory& child);
 
  ResourceNode& add_child(const ResourceData& child);
 
  void delete_child(uint32_t id);
 
  void delete_child(const ResourceNode& node);
 
  void accept(Visitor& visitor) const override;
 
  LIEF_API friend std::ostream& operator<<(std::ostream& os, const ResourceNode& node);
 
  protected:
  ResourceNode();
  ResourceNode(TYPE type);
  childs_t::iterator insert_child(std::unique_ptr<ResourceNode> child);
  TYPE           type_ = TYPE::UNKNOWN;
  uint32_t       id_ = 0;
  std::u16string name_;
  childs_t       childs_;
  uint32_t       depth_ = 0;
};
}
}
#endif /* RESOURCENODE_H */