java - 如何使用 java.nio.ByteBuffer 从 C++ 返回到 Java

Question

这几乎是 How can I make Swig correctly wrap a char* buffer that is modified in C as a Java Something-or-other? 中问题的重复...
如果我使用字节缓冲区而不是 Stringbuffer，类型映射会有什么变化？

Answer 1

我整理了一个示例，说明如何使用以下头文件/函数作为测试来执行此操作:

#include <stdio.h>

static void foo(char *buf, int len) {
  while(len--)
    putchar(*buf++);
}

我的解决方案是修改this answer这样代理接受一个 ByteBuffer 并将其转换为 byte[] 供我们传递给 JNI 代码，然后 JNI 代码为我们将其转换为指针 + 长度组合.

%module test

%{
#include "test.h"
%}

%typemap(jtype) (char *buf, int len) "byte[]"
%typemap(jstype) (char *buf, int len) "java.nio.ByteBuffer"
%typemap(jni) (char *buf, int len) "jbyteArray"
%typemap(javain,pre="    byte[] temp$javainput = new byte[$javainput.capacity()];"
                    "    $javainput.get(temp$javainput);")
        (char *buf, int len) "temp$javainput"

%typemap(in,numinputs=1) (char *buf, int len) {
  $1 = JCALL2(GetByteArrayElements, jenv, $input, NULL);
  $2 = JCALL1(GetArrayLength, jenv, $input);
}

%typemap(freearg) (const signed char *arr, size_t sz) {
  // Or use  0 instead of ABORT to keep changes if it was a copy
  JCALL3(ReleaseByteArrayElements, jenv, $input, $1, JNI_ABORT); 
}

%include "test.h"

这里的新位在 javain typemap 中，分配一个临时的 byte[] 然后使用 get 来填充它。实际上有一个 array() 函数，如果您使用的 ByteBuffer 支持您应该使用它，即类型映射应该只是:

%typemap(javain) (char *buf, int len) "$javainput.array()"

如果您的实现支持它(该方法是可选的，可能会抛出 UnsuportedOperationException)。

事实上，这可以通过 SWIG 2.0 进一步简化，因为我们期望类型始终是byte，所以我们可以使用内置的 int来自 SWIG 2.0 的类型映射来简化我们的界面，现在变成:

%module test

%{
#include "test.h"
%}

%apply (char *STRING, size_t LENGTH) { (char *buf, int len) }
%typemap(javain) (char *buf, int len) "$javainput.array()"
%typemap(jstype) (char *buf, int len) "java.nio.ByteBuffer"

%include "test.h"

我使用以下 Java 测试了所有三个版本:

public class run {
  public static void main(String[] argv) {
    System.loadLibrary("test");
    byte value[] = "hello world\n".getBytes();
    java.nio.ByteBuffer buf = java.nio.ByteBuffer.wrap(value);
    test.foo(buf);
  }
}

为了安全地使用可能不受支持的 array()，您可能想要做的是在带有编译指示的函数中添加一个 try/catch:

%pragma(java) modulecode = %{
  private static byte[] buf2bytearr(java.nio.ByteBuffer buf) {
    try {
      return buf.array();
    }
    catch (UnsupportedOperationException e) {
      byte arr[] = new byte[buf.capacity()];
      buf.get(arr);
      return arr;
    }
  }
%}

然后修改类型映射以使用它:

%typemap(javain) (char *buf, int len) "buf2bytearr($javainput)"