77 lines
2.4 KiB
C++
77 lines
2.4 KiB
C++
/*
|
|
**
|
|
** Copyright 2019, The Android Open Source Project
|
|
**
|
|
** 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
|
|
**
|
|
** http://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.
|
|
*/
|
|
|
|
#include <aidl/android/security/apc/BnConfirmationCallback.h>
|
|
#include <aidl/android/security/apc/IProtectedConfirmation.h>
|
|
#include <android/binder_manager.h>
|
|
#include <android/binder_process.h>
|
|
|
|
#include <gtest/gtest.h>
|
|
|
|
#include <chrono>
|
|
#include <future>
|
|
#include <tuple>
|
|
#include <vector>
|
|
|
|
using namespace std::literals::chrono_literals;
|
|
namespace apc = ::aidl::android::security::apc;
|
|
|
|
class ConfirmationListener
|
|
: public apc::BnConfirmationCallback,
|
|
public std::promise<std::tuple<apc::ResponseCode, std::optional<std::vector<uint8_t>>>> {
|
|
public:
|
|
ConfirmationListener() {}
|
|
|
|
virtual ::ndk::ScopedAStatus
|
|
onCompleted(::aidl::android::security::apc::ResponseCode result,
|
|
const std::optional<std::vector<uint8_t>>& dataConfirmed) override {
|
|
this->set_value({result, dataConfirmed});
|
|
return ::ndk::ScopedAStatus::ok();
|
|
};
|
|
};
|
|
|
|
TEST(ConfirmationInvocationTest, InvokeAndCancel) {
|
|
ABinderProcess_startThreadPool();
|
|
|
|
::ndk::SpAIBinder apcBinder(AServiceManager_getService("android.security.apc"));
|
|
auto apcService = apc::IProtectedConfirmation::fromBinder(apcBinder);
|
|
ASSERT_TRUE(apcService);
|
|
|
|
std::string promptText("Just a little test!");
|
|
std::string locale("en");
|
|
std::vector<uint8_t> extraData{0xaa, 0xff, 0x00, 0x55};
|
|
|
|
auto listener = ndk::SharedRefBase::make<ConfirmationListener>();
|
|
|
|
auto future = listener->get_future();
|
|
|
|
auto rc = apcService->presentPrompt(listener, promptText, extraData, locale, 0);
|
|
|
|
ASSERT_TRUE(rc.isOk());
|
|
|
|
auto fstatus = future.wait_for(2s);
|
|
EXPECT_EQ(fstatus, std::future_status::timeout);
|
|
|
|
rc = apcService->cancelPrompt(listener);
|
|
ASSERT_TRUE(rc.isOk());
|
|
|
|
future.wait();
|
|
auto [responseCode, dataThatWasConfirmed] = future.get();
|
|
|
|
ASSERT_EQ(responseCode, apc::ResponseCode::ABORTED);
|
|
}
|