Update getLimitedUseToken

Author: a-maurice

app_check/src/android/common_android.cc

@@ -164,11 +164,25 @@ void AndroidAppCheckProvider::GetToken(
 void AndroidAppCheckProvider::GetLimitedUseToken(
     std::function<void(AppCheckToken, int, const std::string&)>
         completion_callback) {
-  LogWarning(
-      "GetLimitedUseToken() was called, but the AppCheckProvider interface on "
-      "Android does not yet support limited-use tokens. Falling back to "
-      "GetToken().");
-  GetToken(completion_callback);
+  JNIEnv* env = GetJniEnv();
+
+  jobject j_task = env->CallObjectMethod(
+      android_provider_,
+      app_check_provider::GetMethodId(app_check_provider::kGetLimitedUseToken));
+  std::string error = util::GetAndClearExceptionMessage(env);
+  if (error.empty()) {
+    // Create an object to wrap the callback function
+    TokenResultCallbackData* completion_callback_data =
+        new TokenResultCallbackData(completion_callback);
+    util::RegisterCallbackOnTask(
+        env, j_task, TokenResultCallback,
+        reinterpret_cast<void*>(completion_callback_data),
+        jni_task_id_.c_str());
+  } else {
+    AppCheckToken empty_token;
+    completion_callback(empty_token, kAppCheckErrorUnknown, error.c_str());
+  }
+  env->DeleteLocalRef(j_task);
 }
 
 }  // namespace internal

app_check/src/android/common_android.h

@@ -31,8 +31,10 @@ namespace internal {
 // Used to setup the cache of AppCheckProvider interface method IDs to reduce
 // time spent looking up methods by string.
 // clang-format off
-#define APP_CHECK_PROVIDER_METHODS(X)                                                        \
-  X(GetToken, "getToken",                                                          \
+#define APP_CHECK_PROVIDER_METHODS(X)                                 \
+  X(GetToken, "getToken",                                             \
+    "()Lcom/google/android/gms/tasks/Task;"),                         \
+  X(GetToken, "getLimitedUseToken",                                             \
     "()Lcom/google/android/gms/tasks/Task;")
 // clang-format on