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

How to use a java.nio.ByteBuffer return from C++ to Java

本文关键字:java nio ByteBuffer 返回 何使用 C++      更新时间:2023-10-16

这几乎是我如何使Swig正确包装在C中修改为Java Something或其他的char*缓冲区中的问题的重复
如果我使用字节缓冲区而不是Stringbuffer,那么typemap会有什么变化?

我举了一个例子,说明如何使用以下头文件/函数作为测试:

#include <stdio.h>
static void foo(char *buf, int len) {
  while(len--)
    putchar(*buf++);
}

我的解决方案是修改这个答案,使代理获取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来填充它。实际上,如果您使用的ByteBuffer支持array()函数,则应该使用它,即typemap应该只是:

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

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

事实上,使用SWIG 2.0可以进一步简化这一点,从前面提到的问题来看,因为我们希望的类型总是byte,所以我们可以使用SWIG 2.0中构建的int类型映射来简化我们的接口,现在它变为:

%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 worldn".getBytes();
    java.nio.ByteBuffer buf = java.nio.ByteBuffer.wrap(value);
    test.foo(buf);
  }
}

为了安全地使用可能不受支持的array(),您可能想做的是在带有pragma:的函数中添加一个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)"