PE/utils.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_UTILS_H
#define LIEF_PE_UTILS_H
#include <vector>
#include <string>
 
#include "LIEF/PE/enums.hpp"
#include "LIEF/visibility.h"
#include "LIEF/errors.hpp"
 
namespace LIEF {
class BinaryStream;
 
namespace PE {
class Binary;
class Import;
 
enum class IMPHASH_MODE {
  DEFAULT = 0,    
  LIEF = DEFAULT, 
  PEFILE,         
  VT = PEFILE,    
};
 
//Check if the given stream wraps a PE binary
LIEF_API bool is_pe(BinaryStream& stream);
 
LIEF_API bool is_pe(const std::string& file);
 
LIEF_API bool is_pe(const std::vector<uint8_t>& raw);
 
LIEF_API result<PE_TYPE> get_type(const std::string& file);
 
LIEF_API result<PE_TYPE> get_type(const std::vector<uint8_t>& raw);
 
// In this case we assume that stream contains a valid PE stream
LIEF_LOCAL result<PE_TYPE> get_type_from_stream(BinaryStream& stream);
 
LIEF_API std::string get_imphash(const Binary& binary, IMPHASH_MODE mode = IMPHASH_MODE::DEFAULT);
 
LIEF_API result<Import> resolve_ordinals(const Import& import, bool strict=false, bool use_std=false);
 
LIEF_API ALGORITHMS algo_from_oid(const std::string& oid);
}
}
#endif