deps/LIEF/src/BinaryStream/VectorStream.cpp - main - githubmirror/node

deps/LIEF/src/BinaryStream/VectorStream.cpp

56 строк2 KB

Joyee Cheung

deps: add LIEF as a dependency

23 янв 2026, 01:32

23 янв 2026, 01:32d82ae9e

/* 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 <vector>
#include <string>
#include <fstream>

#include "logging.hpp"
#include "LIEF/BinaryStream/VectorStream.hpp"
#include "LIEF/BinaryStream/SpanStream.hpp"

namespace LIEF {

result<VectorStream> VectorStream::from_file(const std::string& file) {
  std::ifstream ifs(file, std::ios::in | std::ios::binary);
  if (!ifs) {
    LIEF_ERR("Can't open '{}'", file);
    return make_error_code(lief_errors::read_error);
  }

  ifs.unsetf(std::ios::skipws);
  ifs.seekg(0, std::ios::end);
  const auto size = static_cast<uint64_t>(ifs.tellg());
  ifs.seekg(0, std::ios::beg);
  std::vector<uint8_t> data;
  data.resize(size, 0);
  ifs.read(reinterpret_cast<char*>(data.data()), data.size());
  return VectorStream{std::move(data)};
}

std::unique_ptr<SpanStream> VectorStream::slice(uint32_t offset, size_t size) const {
  if (offset > binary_.size() || (offset + size) > binary_.size()) {
    return nullptr;
  }
  const uint8_t* start = binary_.data() + offset;
  return std::unique_ptr<SpanStream>(new SpanStream(start, size));
}


std::unique_ptr<SpanStream> VectorStream::slice(uint32_t offset) const {
  return slice(offset, binary_.size() - offset);
}

}