ResourcesManager.hpp

/* Copyright 2017 - 2024 R. Thomas
 * Copyright 2017 - 2024 Quarkslab
 * Copyright 2017 - 2021 K. Nakagawa
 *
 * 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_RESOURCES_MANAGER_H
#define LIEF_PE_RESOURCES_MANAGER_H
#include <ostream>
 
#include "LIEF/errors.hpp"
#include "LIEF/visibility.h"
#include "LIEF/Object.hpp"
 
#include "LIEF/PE/resources/ResourceVersion.hpp"
#include "LIEF/PE/resources/ResourceIcon.hpp"
#include "LIEF/PE/resources/ResourceDialog.hpp"
#include "LIEF/PE/resources/ResourceStringTable.hpp"
#include "LIEF/PE/resources/ResourceAccelerator.hpp"
 
namespace LIEF {
class VectorStream;
 
namespace PE {
class ResourceNode;
 
class LIEF_API ResourcesManager : public Object {
  public:
 
  enum class TYPE {
    CURSOR       = 1,
    BITMAP       = 2,
    ICON         = 3,
    MENU         = 4,
    DIALOG       = 5,
    STRING       = 6,
    FONTDIR      = 7,
    FONT         = 8,
    ACCELERATOR  = 9,
    RCDATA       = 10,
    MESSAGETABLE = 11,
    GROUP_CURSOR = 12,
    GROUP_ICON   = 14,
    VERSION      = 16,
    DLGINCLUDE   = 17,
    PLUGPLAY     = 19,
    VXD          = 20,
    ANICURSOR    = 21,
    ANIICON      = 22,
    HTML         = 23,
    MANIFEST     = 24
  };
 
  static constexpr uint32_t lang_from_id(size_t id) {
    return id & 0x3ff;
  }
 
  static constexpr uint32_t sublang_from_id(size_t id) {
    return id >> 10;
  }
 
  public:
  using dialogs_t = std::vector<ResourceDialog>;
  using it_const_dialogs = const_ref_iterator<dialogs_t>;
 
  using icons_t = std::vector<ResourceIcon>;
  using it_const_icons = const_ref_iterator<icons_t>;
 
  using strings_table_t = std::vector<ResourceStringTable>;
  using it_const_strings_table = const_ref_iterator<strings_table_t>;
 
  using accelerators_t = std::vector<ResourceAccelerator>;
  using it_const_accelerators = const_ref_iterator<accelerators_t>;
 
  ResourcesManager() = delete;
  ResourcesManager(ResourceNode& rsrc) :
    resources_{&rsrc}
  {}
 
  ResourcesManager(const ResourcesManager&) = default;
  ResourcesManager& operator=(const ResourcesManager&) = default;
 
  ResourcesManager(ResourcesManager&&) = default;
  ResourcesManager& operator=(ResourcesManager&&) = default;
 
  ~ResourcesManager() override = default;
 
  ResourceNode*       get_node_type(TYPE type);
  const ResourceNode* get_node_type(TYPE type) const;
 
  std::vector<TYPE> get_types() const;
 
  bool has_type(TYPE type) const {
    return get_node_type(type) != nullptr;
  }
 
  bool has_manifest() const {
    return get_node_type(TYPE::MANIFEST) != nullptr;
  }
 
  std::string manifest() const;
 
  void manifest(const std::string& manifest);
 
  bool has_version() const {
    return get_node_type(TYPE::VERSION) != nullptr;
  }
 
  result<ResourceVersion> version() const;
 
  bool has_icons() const {
    return get_node_type(TYPE::ICON)       != nullptr &&
           get_node_type(TYPE::GROUP_ICON) != nullptr;
  }
 
  it_const_icons icons() const;
 
  void add_icon(const ResourceIcon& icon);
 
  void change_icon(const ResourceIcon& original, const ResourceIcon& newone);
 
  bool has_dialogs() const {
    return get_node_type(TYPE::DIALOG) != nullptr;
  }
 
  it_const_dialogs dialogs() const;
 
  bool has_string_table() const {
    return get_node_type(TYPE::STRING) != nullptr;
  }
 
  it_const_strings_table string_table() const;
 
  bool has_html() const {
    return get_node_type(TYPE::HTML) != nullptr;
  }
 
  std::vector<std::string> html() const;
 
  bool has_accelerator() const {
    return get_node_type(TYPE::ACCELERATOR) != nullptr;
  }
 
  it_const_accelerators accelerator() const;
 
  std::string print(uint32_t depth = 0) const;
 
  void accept(Visitor& visitor) const override;
 
  LIEF_API friend std::ostream& operator<<(std::ostream& os, const ResourcesManager& m);
 
  private:
  void print_tree(const ResourceNode& node, std::ostringstream& stream,
                  uint32_t current_depth, uint32_t max_depth) const;
  ResourceNode* resources_ = nullptr;
};
 
LIEF_API const char* to_string(ResourcesManager::TYPE type);
 
} // namespace PE
} // namespace LIEF
 
#endif