Use JNI types for return types of JNI methods

JNI methods should use JNI types for their return types so as to ensure
type compatibility between Java and C++.

Bug: 152250228
Test: N/A
Change-Id: Iae05806b91ebe5f173607d243d8504ec9e938589