如何从 JNI 中正确释放内存？

Question

在Java中我称之为矩阵乘法（用于计算方阵的幂）：

// Java
public class Matrix {
    // ...
    native void power(double[] a, double[] b, int rows, int cols, int exponent);
    // ...
}

在 C++ 中，这是通过以下方式实现的：

JNIEXPORT void JNICALL power(JNIEnv* env, jobject o,
    jdoubleArray vals, jdoubleArray originalMatrix, jint aRows, jint aCols, jint i) {
            
    jboolean isCopy;

    // Get input array and result array to write values to.
    jdouble* pa = env->GetDoubleArrayElements(vals, &isCopy);
    jdouble* po = env->GetDoubleArrayElements(originalMatrix, &isCopy);

    // Keeping the result of the k-th multiplication.
    jdouble* tmp = new double[aRows * aCols];

    for (int k = 1; k < i; k++) {
        multiply(po, pa, tmp, aRows, aCols, aCols);  // implemented elsewhere (but should be ok)
        memcpy(tmp, pa, aCols * aRows);
    }    
    
    env->ReleaseDoubleArrayElements(vals, pa, 0);    // <- THIS OK?
    env->ReleaseDoubleArrayElements(originalMatrix, po, 0);

    delete[] tmp;
}

Answer 1

似乎是对的。对于输入数组，您可以在ReleaseDoubleArrayElements 调用中将0 替换为JNI_ABORT，以确保在GetDoubleArrayElements 给您副本的情况下不会进行回写。

对于输出数组，您还可以考虑使用输入数组的isCopy 结果来确定是否使用SetDoubleArrayRegion 而不是GetDoubleArrayElements/ReleaseDoubleArrayElements。