如何从NodeJS中的Rust FFI函数返回字符串值?

时间:2017-02-27 21:58:18

标签: json node.js rust node-ffi

我想生成6个随机数,将它们推送到向量上,然后使用rustc_serialize将该向量编码为NodeJS要使用的JSON字符串。

extern crate rand;
extern crate rustc_serialize;

use rand::{OsRng, Rng};
use rustc_serialize::json::{self, Json, ToJson};

#[no_mangle]
pub extern "C" fn generate() -> String {
    let choices: [u8; 6] = [1, 2, 3, 4, 5, 6];

    let mut rand_vec: Vec<u8> = Vec::new();

    let mut rng = match OsRng::new() {
        Ok(t) => t,
        Err(e) => panic!("Failed to create OsRng!, {}", e),
    };

    for _ in 0..5 {
        rand_vec.push(*rng.choose(&choices).unwrap());
    }

    json::encode(&rand_vec).unwrap()
}

此代码编译为库generate_6_rand.dll。我有一个单独的二进制文件,我用它来测试这段代码。

如果我跑

println!("{:?}", &json::encode(&rand_vec).unwrap());

输出:

"[5,4,3,4,1,3]" //as expected

然后我在NodeJS程序中使用我的.dll

var ffi = require('ffi');
var path = require('path');

var lib = ffi.Library(path.join(__dirname,
  './ffi/generate_6_rand.dll'), {
    generate: [ 'string', [ ]]
  });

console.log(lib.generate());

测试

console.log(lib.generate())

输出:

��.�
它是EcmaScript ArrayBuffer吗?

console.log(new ArrayBuffer(lib.generate())

输出:

ArrayBuffer { byteLength: 0 }
它的原始链属性是什么?

console.log(lib.generate().__proto__)

输出:

[String: '']

将代码更改为:

var ref = require('ref');
var ArrayType = require('ref-array');
var Int32Array = ArrayType(ref.types.int32);


var lib = ffi.Library(path.join(__dirname,
  '../dice_lib/target/release/generate_6_rand.dll'), {
    generate: [ Int32Array, [ ]]
  });

console.log(new ArrayBuffer(lib.generate()));

输出:

ArrayBuffer { byteLength: 0 }

为什么FFI函数不会像我期望的那样返回JSON字符串?

1 个答案:

答案 0 :(得分:4)

谢谢Wesley Wiser给我一个关于CString的大线索。我在The Rust FFI Omnibus找到了答案。

在我的NodeJS程序可以访问它之前,我的预期JSON字符串的内存被释放,无论我是返回JSON字符串还是返回CString

这是基于该文章的解决方案。 对于其他新手程序员,请记住我的解决方案可能是理想的,也可能不是理想的

extern crate rand;
extern crate rustc_serialize;
extern crate libc;

use libc::c_char;
use rand::{OsRng, Rng};
use std::ffi::CString;
use rustc_serialize::json;

#[no_mangle]
pub extern "C" fn generate() -> *mut c_char {
    let choices: [u8; 6] = [1, 2, 3, 4, 5, 6];

    let mut rand_vec: Vec<u8> = Vec::new();

    let mut rng = match OsRng::new() {
        Ok(t) => t,
        Err(e) => panic!("Failed to create OsRng!, {}", e),
    };

    for _ in 0..6 {
        rand_vec.push(*rng.choose(&choices).unwrap());
    }

    let json_string = CString::new(json::encode(&rand_vec).unwrap()).unwrap();

    json_string.into_raw()
}

#[no_mangle]
pub extern "C" fn free_memory(pointer: *mut c_char) {
    unsafe {
        if pointer.is_null() {
            return;
        }
        CString::from_raw(pointer)
    };
}

的NodeJS

var ffi = require('ffi');
var path = require('path');

var lib = ffi.Library(path.join(__dirname,
  './ffi/generate_6_rand.dll'), {
    generate: [ 'char *' , [ ]],
    free_memory: ['void', ['char *']]
  });

var json_string = lib.generate();

var save_json = JSON.parse(json_string.readCString());

console.log( json_string.readCString()); // Output: [6,1,6,4,1,4]
lib.free_memory(json_string);
console.log(json_string.readCString()); // Output: ��x�

我设置了两个console.log来显示解除分配之前和之后的输出。