在Rust中使用ctranslate2动态库遇fstream缺失错误的解决问询

错误原因 #

fstream是C标准库的头文件，而bindgen默认以C语言模式解析头文件，无法识别C专属的头文件和语法，因此出现找不到文件的错误。此外，ctranslate2本身是C库，还需要链接系统的C标准库才能正常调用。

解决方案 #

方案1：修改build.rs，让bindgen以C++模式解析 #

直接调整build.rs配置，告诉bindgen用C规则处理头文件，并链接C标准库：

extern crate bindgen;

use std::env;
use std::path::PathBuf;

fn main() {
    // 指定动态库查找路径
    println!("cargo:rustc-link-search=/usr/local/lib");
    // 链接ctranslate2库
    println!("cargo:rustc-link-lib=ctranslate2");
    // 链接C++标准库（Linux用stdc++，macOS替换为c++）
    println!("cargo:rustc-link-lib=stdc++");

    // 头文件变化时触发重新构建
    println!("cargo:rerun-if-changed=/usr/local/include/ctranslate2/translator.h");

    let bindings = bindgen::Builder::default()
        // 目标C++头文件
        .header("/usr/local/include/ctranslate2/translator.h")
        // 告诉clang以C++模式解析代码
        .clang_arg("-x")
        .clang_arg("c++")
        // 指定C++标准（匹配ctranslate2编译时使用的版本，通常为C++17）
        .clang_arg("-std=c++17")
        // 启用C++命名空间支持
        .enable_cxx_namespaces()
        // 自动处理文件变化触发重编译
        .parse_callbacks(Box::new(bindgen::CargoCallbacks))
        // 生成绑定
        .generate()
        .expect("无法生成绑定");

    // 将绑定写入输出目录
    let out_path = PathBuf::from(env::var("OUT_DIR").unwrap());
    bindings
        .write_to_file(out_path.join("bindings.rs"))
        .expect("无法写入绑定文件");
}

方案2：编写C封装层（更稳定的长期方案） #

如果bindgen处理复杂C代码出现兼容性问题，可将ctranslate2的C接口封装为C风格接口，再用bindgen处理C代码：

1. 编写C封装头文件wrapper.h：

#ifndef WRAPPER_H
#define WRAPPER_H

#ifdef __cplusplus
extern "C" {
#endif

typedef struct Translator Translator;

// 创建Translator实例
Translator* translator_create(const char* model_path);
// 销毁Translator实例
void translator_destroy(Translator* translator);
// 示例：添加翻译接口（根据需求扩展）
const char* translator_translate(Translator* translator, const char* input);

#ifdef __cplusplus
}
#endif

#endif

2. 编写C++实现文件wrapper.cpp：

#include "wrapper.h"
#include <ctranslate2/translator.h>
#include <vector>
#include <string>

using namespace ctranslate2;

struct Translator {
    ctranslate2::Translator inner;
    Translator(const std::string& model_path) : inner(model_path) {}
};

Translator* translator_create(const char* model_path) {
    return new Translator(model_path);
}

void translator_destroy(Translator* translator) {
    delete translator;
}

const char* translator_translate(Translator* translator, const char* input) {
    std::vector<std::vector<std::string>> inputs = {{input}};
    auto results = translator->inner.translate_batch(inputs);
    static std::string output;
    output = results[0][0];
    return output.c_str();
}

3. 编译封装层为动态库：

g++ -shared -fPIC wrapper.cpp -o libctranslate2_c.so -I/usr/local/include -L/usr/local/lib -lctranslate2 -std=c++17

4. 修改build.rs处理C封装头：

extern crate bindgen;

use std::env;
use std::path::PathBuf;

fn main() {
    println!("cargo:rustc-link-search=/usr/local/lib");
    println!("cargo:rustc-link-search=./"); // 指向封装库所在目录
    println!("cargo:rustc-link-lib=ctranslate2_c");
    println!("cargo:rustc-link-lib=stdc++");

    println!("cargo:rerun-if-changed=wrapper.h");

    let bindings = bindgen::Builder::default()
        .header("wrapper.h")
        .parse_callbacks(Box::new(bindgen::CargoCallbacks))
        .generate()
        .expect("无法生成绑定");

    let out_path = PathBuf::from(env::var("OUT_DIR").unwrap());
    bindings
        .write_to_file(out_path.join("bindings.rs"))
        .expect("无法写入绑定文件");
}

内容的提问来源于stack exchange，提问作者filif96770