You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

348 lines
16 KiB
Markdown

2 years ago
# 32 | JNI的运行机制
我们经常会遇见Java语言较难表达甚至是无法表达的应用场景。比如我们希望使用汇编语言如X86\_64的SIMD指令来提升关键代码的性能再比如我们希望调用Java核心类库无法提供的某个体系架构或者操作系统特有的功能。
在这种情况下我们往往会牺牲可移植性在Java代码中调用C/C++代码下面简述为C代码并在其中实现所需功能。这种跨语言的调用便需要借助Java虚拟机的Java Native InterfaceJNI机制。
关于JNI的例子你应该特别熟悉Java中标记为`native`的、没有方法体的方法下面统称为native方法。当在Java代码中调用这些native方法时Java虚拟机将通过JNI调用至对应的C函数下面将native方法对应的C实现统称为C函数中。
```
public class Object {
public native int hashCode();
}
```
举个例子,`Object.hashCode`方法便是一个native方法。它对应的C函数将计算对象的哈希值并缓存在对象头、栈上锁记录轻型锁或对象监视锁重型锁所使用的monitor以确保该值在对象的生命周期之内不会变更。
## native方法的链接
在调用native方法前Java虚拟机需要将该native方法链接至对应的C函数上。
链接方式主要有两种。第一种是让Java虚拟机自动查找符合默认命名规范的C函数并且链接起来。
事实上,我们并不需要记住所谓的命名规范,而是采用`javac -h`命令便可以根据Java程序中的native方法声明自动生成包含符合命名规范的C函数的头文件。
举个例子,在下面这段代码中,`Foo`类有三个native方法分别为静态方法`foo`以及两个重载的实例方法`bar`。
```
package org.example;
public class Foo {
public static native void foo();
public native void bar(int i, long j);
public native void bar(String s, Object o);
}
```
通过执行`javac -h . org/example/Foo.java`命令,我们将在当前文件夹(对应`-h`后面跟着的`.`)生成名为`org_example_Foo.h`的头文件。其内容如下所示:
```
/* DO NOT EDIT THIS FILE - it is machine generated */
#include <jni.h>
/* Header for class org_example_Foo */
#ifndef _Included_org_example_Foo
#define _Included_org_example_Foo
#ifdef __cplusplus
extern "C" {
#endif
/*
* Class: org_example_Foo
* Method: foo
* Signature: ()V
*/
JNIEXPORT void JNICALL Java_org_example_Foo_foo
(JNIEnv *, jclass);
/*
* Class: org_example_Foo
* Method: bar
* Signature: (IJ)V
*/
JNIEXPORT void JNICALL Java_org_example_Foo_bar__IJ
(JNIEnv *, jobject, jint, jlong);
/*
* Class: org_example_Foo
* Method: bar
* Signature: (Ljava/lang/String;Ljava/lang/Object;)V
*/
JNIEXPORT void JNICALL Java_org_example_Foo_bar__Ljava_lang_String_2Ljava_lang_Object_2
(JNIEnv *, jobject, jstring, jobject);
#ifdef __cplusplus
}
#endif
#endif
```
这里我简单讲解一下该命名规范。
首先native方法对应的C函数都需要以`Java_`为前缀之后跟着完整的包名和方法名。由于C函数名不支持`/`字符,因此我们需要将`/`转换为`_`,而原本方法名中的`_`符号,则需要转换为`_1`。
举个例子,`org.example`包下`Foo`类的`foo`方法Java虚拟机会将其自动链接至名为`Java_org_example_Foo_foo`的C函数中。
当某个类出现重载的native方法时Java虚拟机还会将参数类型纳入自动链接对象的考虑范围之中。具体的做法便是在前面C函数名的基础上追加`__`以及方法描述符作为后缀。
方法描述符的特殊符号同样会被替换掉,如引用类型所使用的`;`会被替换为`_2`,数组类型所使用的`[`会被替换为`_3`。
基于此命名规范,你可以手动拼凑上述代码中,`Foo`类的两个`bar`方法所能自动链接的C函数名并用`javac -h`命令所生成的结果来验证一下。
第二种链接方式则是在C代码中主动链接。
这种链接方式对C函数名没有要求。通常我们会使用一个名为`registerNatives`的native方法并按照第一种链接方式定义所能自动链接的C函数。在该C函数中我们将手动链接该类的其他native方法。
举个例子,`Object`类便拥有一个`registerNatives`方法所对应的C代码如下所示
```
// 注Object类的registerNatives方法的实现位于java.base模块里的C代码中
static JNINativeMethod methods[] = {
{"hashCode", "()I", (void *)&JVM_IHashCode},
{"wait", "(J)V", (void *)&JVM_MonitorWait},
{"notify", "()V", (void *)&JVM_MonitorNotify},
{"notifyAll", "()V", (void *)&JVM_MonitorNotifyAll},
{"clone", "()Ljava/lang/Object;", (void *)&JVM_Clone},
};
JNIEXPORT void JNICALL
Java_java_lang_Object_registerNatives(JNIEnv *env, jclass cls)
{
(*env)->RegisterNatives(env, cls,
methods, sizeof(methods)/sizeof(methods[0]));
}
```
我们可以看到上面这段代码中的C函数将调用`RegisterNatives` API注册`Object`类中其他native方法所要链接的C函数。并且这些C函数的名字并不符合默认命名规则。
当使用第二种方式进行链接时我们需要在其他native方法被调用之前完成链接工作。因此我们往往会在类的初始化方法里调用该`registerNatives`方法。具体示例如下所示:
```
public class Object {
private static native void registerNatives();
static {
registerNatives();
}
}
```
下面我们采用第一种链接方式,并且实现其中的`bar(String, Object)`方法。如下所示:
```
// foo.c
#include <stdio.h>
#include "org_example_Foo.h"
JNIEXPORT void JNICALL Java_org_example_Foo_bar__Ljava_lang_String_2Ljava_lang_Object_2
(JNIEnv *env, jobject thisObject, jstring str, jobject obj) {
printf("Hello, World\n");
return;
}
```
然后我们可以通过gcc命令将其编译成为动态链接库
```
# 该命令仅适用于macOS
$ gcc -I$JAVA_HOME/include -I$JAVA_HOME/include/darwin -o libfoo.dylib -shared foo.c
```
这里需要注意的是,动态链接库的名字须以`lib`为前缀,以`.dylib`(或Linux上的`.so`为扩展名。在Java程序中我们可以通过`System.loadLibrary("foo")`方法来加载`libfoo.dylib`,如下述代码所示:
```
package org.example;
public class Foo {
public static native void foo();
public native void bar(int i, long j);
public native void bar(String s, Object o);
int i = 0xDEADBEEF;
public static void main(String[] args) {
try {
System.loadLibrary("foo");
} catch (UnsatisfiedLinkError e) {
e.printStackTrace();
System.exit(1);
}
new Foo().bar("", "");
}
}
```
如果`libfoo.dylib`不在当前路径下我们可以在启动Java虚拟机时配置`java.library.path`参数,使其指向包含`libfoo.dylib`的文件夹。具体命令如下所示:
```
$ java -Djava.library.path=/PATH/TO/DIR/CONTAINING/libfoo.dylib org.example.Foo
Hello, World
```
## JNI的API
在C代码中我们也可以使用Java的语言特性如instanceof测试等。这些功能都是通过特殊的JNI函数[JNI Functions](https://docs.oracle.com/en/java/javase/11/docs/specs/jni/functions.html))来实现的。
Java虚拟机会将所有JNI函数的函数指针聚合到一个名为`JNIEnv`的数据结构之中。
这是一个线程私有的数据结构。Java虚拟机会为每个线程创建一个`JNIEnv`并规定C代码不能将当前线程的`JNIEnv`共享给其他线程否则JNI函数的正确性将无法保证。
这么设计的原因主要有两个。一是给JNI函数提供一个单独命名空间。二是允许Java虚拟机通过更改函数指针替换JNI函数的具体实现例如从附带参数类型检测的慢速版本切换至不做参数类型检测的快速版本。
在HotSpot虚拟机中`JNIEnv`被内嵌至Java线程的数据结构之中。部分虚拟机代码甚至会从`JNIEnv`的地址倒推出Java线程的地址。因此如果在其他线程中使用当前线程的`JNIEnv`,会使这部分代码错误识别当前线程。
JNI会将Java层面的基本类型以及引用类型映射为另一套可供C代码使用的数据结构。其中基本类型的对应关系如下表所示
![](https://static001.geekbang.org/resource/image/cb/ca/cb2c806532449f2c1edfe821990ac9ca.png)
引用类型对应的数据结构之间也存在着继承关系,具体如下所示:
```
jobject
|- jclass (java.lang.Class objects)
|- jstring (java.lang.String objects)
|- jthrowable (java.lang.Throwable objects)
|- jarray (arrays)
|- jobjectArray (object arrays)
|- jbooleanArray (boolean arrays)
|- jbyteArray (byte arrays)
|- jcharArray (char arrays)
|- jshortArray (short arrays)
|- jintArray (int arrays)
|- jlongArray (long arrays)
|- jfloatArray (float arrays)
|- jdoubleArray (double arrays)
```
我们回头看看`Foo`类3个native方法对应的C函数的参数。
```
JNIEXPORT void JNICALL Java_org_example_Foo_foo
(JNIEnv *, jclass);
JNIEXPORT void JNICALL Java_org_example_Foo_bar__IJ
(JNIEnv *, jobject, jint, jlong);
JNIEXPORT void JNICALL Java_org_example_Foo_bar__Ljava_lang_String_2Ljava_lang_Object_2 (JNIEnv *, jobject, jstring, jobject);
```
静态native方法`foo`将接收两个参数分别为存放JNI函数的`JNIEnv`指针,以及一个`jclass`参数用来指代定义该native方法的类即`Foo`类。
两个实例native方法`bar`的第二个参数则是`jobject`类型的用来指代该native方法的调用者也就是`Foo`类的实例。
如果native方法声明了参数那么对应的C函数将接收这些参数。在我们的例子中第一个`bar`方法声明了int型和long型的参数对应的C函数则接收jint和jlong类型的参数第二个`bar`方法声明了String类型和Object类型的参数对应的C函数则接收jstring和jobject类型的参数。
下面,我们继续修改上一小节中的`foo.c`并在C代码中获取`Foo`类实例的`i`字段。
```
// foo.c
#include <stdio.h>
#include "org_example_Foo.h"
JNIEXPORT void JNICALL Java_org_example_Foo_bar__Ljava_lang_String_2Ljava_lang_Object_2
(JNIEnv *env, jobject thisObject, jstring str, jobject obj) {
jclass cls = (*env)->GetObjectClass(env, thisObject);
jfieldID fieldID = (*env)->GetFieldID(env, cls, "i", "I");
jint value = (*env)->GetIntField(env, thisObject, fieldID);
printf("Hello, World 0x%x\n", value);
return;
}
```
我们可以看到在JNI中访问字段类似于反射API我们首先需要通过类实例获得`FieldID`,然后再通过`FieldID`获得某个实例中该字段的值。不过与Java代码相比上述代码貌似不用处理异常。事实果真如此吗
下面我就尝试获取了不存在的字段`j`,运行结果如下所示:
```
$ java org.example.Foo
Hello, World 0x5
Exception in thread "main" java.lang.NoSuchFieldError: j
at org.example.Foo.bar(Native Method)
at org.example.Foo.main(Foo.java:20)
```
我们可以看到,`printf`语句照常执行并打印出`Hello, World 0x5`但这个数值明显是错误的。当从C函数返回至main方法时Java虚拟机又会抛出`NoSuchFieldError`异常。
实际上当调用JNI函数时Java虚拟机便已生成异常实例并缓存在内存中的某个位置。与Java编程不一样的是它并不会显式地跳转至异常处理器或者调用者中而是继续执行接下来的C代码。
因此当从可能触发异常的JNI函数返回时我们需要通过JNI函数`ExceptionOccurred`检查是否发生了异常并且作出相应的处理。如果无须抛出该异常那么我们需要通过JNI函数`ExceptionClear`显式地清空已缓存的异常。
具体示例如下所示(为了控制代码篇幅,我仅在第一个`GetFieldID`后检查异常以及清空异常):
```
// foo.c
#include <stdio.h>
#include "org_example_Foo.h"
JNIEXPORT void JNICALL Java_org_example_Foo_bar__Ljava_lang_String_2Ljava_lang_Object_2
(JNIEnv *env, jobject thisObject, jstring str, jobject obj) {
jclass cls = (*env)->GetObjectClass(env, thisObject);
jfieldID fieldID = (*env)->GetFieldID(env, cls, "j", "I");
if((*env)->ExceptionOccurred(env)) {
printf("Exception!\n");
(*env)->ExceptionClear(env);
}
fieldID = (*env)->GetFieldID(env, cls, "i", "I");
jint value = (*env)->GetIntField(env, thisObject, fieldID);
// we should put an exception guard here as well.
printf("Hello, World 0x%x\n", value);
return;
}
```
## 局部引用与全局引用
在C代码中我们可以访问所传入的引用类型参数也可以通过JNI函数创建新的Java对象。
这些Java对象显然也会受到垃圾回收器的影响。因此Java虚拟机需要一种机制来告知垃圾回收算法不要回收这些C代码中可能引用到的Java对象。
这种机制便是JNI的局部引用Local Reference和全局引用Global Reference。垃圾回收算法会将被这两种引用指向的对象标记为不可回收。
事实上无论是传入的引用类型参数还是通过JNI函数除`NewGlobalRef`及`NewWeakGlobalRef`之外)返回的引用类型对象,都属于局部引用。
不过一旦从C函数中返回至Java方法之中那么局部引用将失效。也就是说垃圾回收器在标记垃圾时不再考虑这些局部引用。
这就意味着我们不能缓存局部引用以供另一C线程或下一次native方法调用时使用。
对于这种应用场景我们需要借助JNI函数`NewGlobalRef`将该局部引用转换为全局引用以确保其指向的Java对象不会被垃圾回收。
相应的我们还可以通过JNI函数`DeleteGlobalRef`来消除全局引用以便回收被全局引用指向的Java对象。
此外当C函数运行时间极其长时我们也应该考虑通过JNI函数`DeleteLocalRef`消除不再使用的局部引用以便回收被引用的Java对象。
另一方面由于垃圾回收器可能会移动对象在内存中的位置因此Java虚拟机需要另一种机制来保证局部引用或者全局引用将正确地指向移动过后的对象。
HotSpot虚拟机是通过句柄handle来完成上述需求的。这里句柄指的是内存中Java对象的指针的指针。当发生垃圾回收时如果Java对象被移动了那么句柄指向的指针值也将发生变动但句柄本身保持不变。
实际上无论是局部引用还是全局引用都是句柄。其中局部引用所对应的句柄有两种存储方式一是在本地方法栈帧中主要用于存放C函数所接收的来自Java层面的引用类型参数另一种则是线程私有的句柄块主要用于存放C函数运行过程中创建的局部引用。
当从C函数返回至Java方法时本地方法栈帧中的句柄将会被自动清除。而线程私有句柄块则需要由Java虚拟机显式清理。
进入C函数时对引用类型参数的句柄化和调整参数位置C调用和Java调用传参的方式不一样以及从C函数返回时清理线程私有句柄块共同造就了JNI调用的额外性能开销具体可参考该stackoverflow上的[回答](https://stackoverflow.com/questions/24746776/what-does-a-jvm-have-to-do-when-calling-a-native-method/24747484#24747484))。
## 总结与实践
今天我介绍了JNI的运行机制。
Java中的native方法的链接方式主要有两种。一是按照JNI的默认规范命名所要链接的C函数并依赖于Java虚拟机自动链接。另一种则是在C代码中主动链接。
JNI提供了一系列API来允许C代码使用Java语言特性。这些API不仅使用了特殊的数据结构来表示Java类还拥有特殊的异常处理模式。
JNI中的引用可分为局部引用和全局引用。这两者都可以阻止垃圾回收器回收被引用的Java对象。不同的是局部引用在native方法调用返回之后便会失效。传入参数以及大部分JNI API函数的返回值都属于局部引用。
* * *
今天的实践环节,请阅读[该文档](https://www.ibm.com/developerworks/java/library/j-jni/index.html)中的Performance pitfalls以及Correctness pitfalls两节。