ResourceData.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_DATA_H
#define LIEF_PE_RESOURCE_DATA_H
 
#include <vector>
 
#include "LIEF/visibility.h"
#include "LIEF/PE/ResourceNode.hpp"
#include "LIEF/span.hpp"
 
namespace LIEF {
namespace PE {
 
class Parser;
class Builder;
 
class LIEF_API ResourceData : public ResourceNode {
 
  friend class Parser;
  friend class Builder;
 
  public:
  ResourceData() :
    ResourceNode(ResourceNode::TYPE::DATA)
  {}
  ResourceData(std::vector<uint8_t> content, uint32_t code_page) :
    ResourceNode(ResourceNode::TYPE::DATA),
    content_(std::move(content)),
    code_page_(code_page)
  {}
 
  ResourceData(const ResourceData& other) = default;
  ResourceData& operator=(const ResourceData& other) = default;
  void swap(ResourceData& other) noexcept;
 
  ~ResourceData() override = default;
 
  std::unique_ptr<ResourceNode> clone() const override {
    return std::unique_ptr<ResourceNode>{new ResourceData{*this}};
  }
 
  uint32_t code_page() const {
    return code_page_;
  }
 
  span<const uint8_t> content() const {
    return content_;
  }
  span<uint8_t> content() {
    return content_;
  }
 
  uint32_t reserved() const {
    return reserved_;
  }
 
  uint32_t offset() const {
    return offset_;
  }
 
  void code_page(uint32_t code_page) {
    code_page_ = code_page;
  }
 
  void content(std::vector<uint8_t> content) {
    content_ = std::move(content);
  }
 
  void reserved(uint32_t value) {
    reserved_ = value;
  }
 
  static bool classof(const ResourceNode* node) {
    return node->is_data();
  }
 
  void accept(Visitor& visitor) const override;
 
  LIEF_API friend std::ostream& operator<<(std::ostream& os, const ResourceData& data);
 
  private:
  std::vector<uint8_t> content_;
  uint32_t code_page_ = 0;
  uint32_t reserved_ = 0;
  uint32_t offset_ = 0;
 
};
 
} // namespace PE
} // namepsace LIEF
#endif /* RESOURCEDATA_H */