Method getModule

src/Interpreters/WasmModuleManager.cpp:220–279

}

std::pair<std::shared_ptr<WasmModule>, UInt256> WasmModuleManager::getModule(std::string_view module_name, FuelMode fuel_mode)
{
    const bool requires_fuel_specialization = engine->requiresFuelSpecialization();
    const size_t cache_idx = requires_fuel_specialization ? fuelModeIndex(fuel_mode) : 0;

    {
        SharedLockGuard lock(modules_mutex);

        auto it = modules.find(module_name);
        if (it != modules.end())
        {
            if (auto module = it->second.ptrs[cache_idx].lock())
                return {module, it->second.hash};
        }
    }

    UniqueLock lock(modules_mutex);
    auto it = modules.find(module_name);
    if (it != modules.end())
    {
        if (auto module = it->second.ptrs[cache_idx].lock())
            return {module, it->second.hash};
    }

    auto module_path = getFilePath(module_name);
    if (!user_scripts_disk->existsFile(module_path))
        throw Exception(ErrorCodes::RESOURCE_NOT_FOUND, "WebAssembly module '{}' not found", module_name);
    if (auto res = validateModuleFile(user_scripts_disk, module_path); !res)
        throw Exception(ErrorCodes::INCORRECT_DATA, "Cannot load WebAssembly module '{}': {}", module_name, res.error().text);

    auto wasm_code = loadModuleImpl(module_name);
    UInt256 local_hash = calculateHash(wasm_code);

    if (it != modules.end())
    {
        if (it->second.hash && local_hash != it->second.hash)
            throw Exception(
                ErrorCodes::INCORRECT_DATA,
                "Hash mismatch for WebAssembly module '{}', expected {}, got {}",
                module_name,
                hashToHex(it->second.hash),
                hashToHex(local_hash));
    }

    std::shared_ptr<WasmModule> local_module = engine->compileModule(module_name, wasm_code, fuel_mode);
    linkHostFunctions(*local_module);

    if (it == modules.end())
    {
        auto [inserted_it, _] = modules.emplace(String(module_name), ModuleRef{{}, local_hash});
        it = inserted_it;
    }
    else if (!it->second.hash)
    {
        it->second.hash = local_hash;
    }

    it->second.ptrs[cache_idx] = local_module;