libredex/JarLoader.h

/*
 * Copyright (c) Meta Platforms, Inc. and affiliates.
 *
 * This source code is licensed under the MIT license found in the
 * LICENSE file in the root directory of this source tree.
 */

#pragma once

class DexLocation;
class DexField;
class DexMethod;

#include "boost/variant.hpp"

#include "ConfigFiles.h"

#include <functional>

namespace JarLoaderUtil {
uint32_t read32(uint8_t*& buffer);
uint32_t read16(uint8_t*& buffer);
}; // namespace JarLoaderUtil

using attribute_hook_t =
    std::function<void(boost::variant<DexField*, DexMethod*> field_or_method,
                       const char* attribute_name,
                       uint8_t* attribute_pointer)>;

bool load_jar_file(const DexLocation* location,
                   Scope* classes = nullptr,
                   const attribute_hook_t& = nullptr);

bool load_class_file(const std::string& filename, Scope* classes = nullptr);

void init_basic_types();
bool process_jar(const DexLocation* location,
                 const uint8_t* mapping,
                 ssize_t size,
                 Scope* classes,
                 const attribute_hook_t& attr_hook);

bool parse_class(uint8_t* buffer,
                 Scope* classes,
                 attribute_hook_t attr_hook,
                 const DexLocation* jar_location);