| // Copyright 2020 The Pigweed Authors |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); you may not |
| // use this file except in compliance with the License. You may obtain a copy of |
| // the License at |
| // |
| // https://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, WITHOUT |
| // WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the |
| // License for the specific language governing permissions and limitations under |
| // the License. |
| |
| // This file provides a Java Native Interface (JNI) version of the Detokenizer |
| // class. This facilitates using the tokenizer library from Java or other JVM |
| // languages. A corresponding Java class is provided in Detokenizer.java. |
| |
| #include <jni.h> |
| |
| #include <cstring> |
| #include <span> |
| |
| #include "pw_preprocessor/concat.h" |
| #include "pw_tokenizer/detokenize.h" |
| #include "pw_tokenizer/token_database.h" |
| |
| #define DETOKENIZER_METHOD(method) \ |
| JNICALL PW_CONCAT(Java_dev_pigweed_tokenizer_, Detokenizer_, method) |
| |
| namespace pw::tokenizer { |
| namespace { |
| |
| Detokenizer* HandleToPointer(jlong handle) { |
| Detokenizer* detokenizer = nullptr; |
| std::memcpy(&detokenizer, &handle, sizeof(detokenizer)); |
| static_assert(sizeof(detokenizer) <= sizeof(handle)); |
| return detokenizer; |
| } |
| |
| jlong PointerToHandle(Detokenizer* detokenizer) { |
| jlong handle = 0; |
| std::memcpy(&handle, &detokenizer, sizeof(detokenizer)); |
| static_assert(sizeof(handle) >= sizeof(detokenizer)); |
| return handle; |
| } |
| |
| } // namespace |
| |
| extern "C" { |
| |
| static_assert(sizeof(jbyte) == 1u); |
| |
| JNIEXPORT jlong DETOKENIZER_METHOD(newNativeDetokenizer)(JNIEnv* env, |
| jclass, |
| jbyteArray array) { |
| jbyte* const data = env->GetByteArrayElements(array, nullptr); |
| const jsize size = env->GetArrayLength(array); |
| |
| TokenDatabase tokens = TokenDatabase::Create(std::span(data, size)); |
| const jlong handle = |
| PointerToHandle(new Detokenizer(tokens.ok() ? tokens : TokenDatabase())); |
| |
| env->ReleaseByteArrayElements(array, data, 0); |
| return handle; |
| } |
| |
| JNIEXPORT void DETOKENIZER_METHOD(deleteNativeDetokenizer)(JNIEnv*, |
| jclass, |
| jlong handle) { |
| delete HandleToPointer(handle); |
| } |
| |
| JNIEXPORT jstring DETOKENIZER_METHOD(detokenizeNative)(JNIEnv* env, |
| jobject, |
| jlong handle, |
| jbyteArray array) { |
| jbyte* const data = env->GetByteArrayElements(array, nullptr); |
| const jsize size = env->GetArrayLength(array); |
| |
| DetokenizedString result = HandleToPointer(handle)->Detokenize(data, size); |
| |
| env->ReleaseByteArrayElements(array, data, 0); |
| |
| return result.matches().empty() |
| ? nullptr |
| : env->NewStringUTF(result.BestString().c_str()); |
| } |
| |
| } // extern "C" |
| |
| } // namespace pw::tokenizer |