xref: /aosp_15_r20/hardware/interfaces/weaver/vts/VtsHalWeaverTargetTest.cpp (revision 4d7e907c777eeecc4c5bd7cf640a754fac206ff7)
1*4d7e907cSAndroid Build Coastguard Worker /*
2*4d7e907cSAndroid Build Coastguard Worker  * Copyright (C) 2020 The Android Open Source Project
3*4d7e907cSAndroid Build Coastguard Worker  *
4*4d7e907cSAndroid Build Coastguard Worker  * Licensed under the Apache License, Version 2.0 (the "License");
5*4d7e907cSAndroid Build Coastguard Worker  * you may not use this file except in compliance with the License.
6*4d7e907cSAndroid Build Coastguard Worker  * You may obtain a copy of the License at
7*4d7e907cSAndroid Build Coastguard Worker  *
8*4d7e907cSAndroid Build Coastguard Worker  *      http://www.apache.org/licenses/LICENSE-2.0
9*4d7e907cSAndroid Build Coastguard Worker  *
10*4d7e907cSAndroid Build Coastguard Worker  * Unless required by applicable law or agreed to in writing, software
11*4d7e907cSAndroid Build Coastguard Worker  * distributed under the License is distributed on an "AS IS" BASIS,
12*4d7e907cSAndroid Build Coastguard Worker  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13*4d7e907cSAndroid Build Coastguard Worker  * See the License for the specific language governing permissions and
14*4d7e907cSAndroid Build Coastguard Worker  * limitations under the License.
15*4d7e907cSAndroid Build Coastguard Worker  */
16*4d7e907cSAndroid Build Coastguard Worker #include <aidl/Gtest.h>
17*4d7e907cSAndroid Build Coastguard Worker #include <aidl/Vintf.h>
18*4d7e907cSAndroid Build Coastguard Worker 
19*4d7e907cSAndroid Build Coastguard Worker #include <aidl/android/hardware/weaver/IWeaver.h>
20*4d7e907cSAndroid Build Coastguard Worker #include <android-base/file.h>
21*4d7e907cSAndroid Build Coastguard Worker #include <android-base/parseint.h>
22*4d7e907cSAndroid Build Coastguard Worker #include <android-base/strings.h>
23*4d7e907cSAndroid Build Coastguard Worker #include <android/binder_manager.h>
24*4d7e907cSAndroid Build Coastguard Worker #include <android/binder_process.h>
25*4d7e907cSAndroid Build Coastguard Worker #include <android/hardware/weaver/1.0/IWeaver.h>
26*4d7e907cSAndroid Build Coastguard Worker #include <hidl/GtestPrinter.h>
27*4d7e907cSAndroid Build Coastguard Worker #include <hidl/ServiceManagement.h>
28*4d7e907cSAndroid Build Coastguard Worker 
29*4d7e907cSAndroid Build Coastguard Worker #include <limits>
30*4d7e907cSAndroid Build Coastguard Worker 
31*4d7e907cSAndroid Build Coastguard Worker using ::aidl::android::hardware::weaver::IWeaver;
32*4d7e907cSAndroid Build Coastguard Worker using ::aidl::android::hardware::weaver::WeaverConfig;
33*4d7e907cSAndroid Build Coastguard Worker using ::aidl::android::hardware::weaver::WeaverReadResponse;
34*4d7e907cSAndroid Build Coastguard Worker using ::aidl::android::hardware::weaver::WeaverReadStatus;
35*4d7e907cSAndroid Build Coastguard Worker 
36*4d7e907cSAndroid Build Coastguard Worker using HidlIWeaver = ::android::hardware::weaver::V1_0::IWeaver;
37*4d7e907cSAndroid Build Coastguard Worker using HidlWeaverConfig = ::android::hardware::weaver::V1_0::WeaverConfig;
38*4d7e907cSAndroid Build Coastguard Worker using HidlWeaverReadStatus = ::android::hardware::weaver::V1_0::WeaverReadStatus;
39*4d7e907cSAndroid Build Coastguard Worker using HidlWeaverReadResponse = ::android::hardware::weaver::V1_0::WeaverReadResponse;
40*4d7e907cSAndroid Build Coastguard Worker using HidlWeaverStatus = ::android::hardware::weaver::V1_0::WeaverStatus;
41*4d7e907cSAndroid Build Coastguard Worker 
42*4d7e907cSAndroid Build Coastguard Worker const std::string kSlotMapFile = "/metadata/password_slots/slot_map";
43*4d7e907cSAndroid Build Coastguard Worker const std::vector<uint8_t> KEY{1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16};
44*4d7e907cSAndroid Build Coastguard Worker const std::vector<uint8_t> WRONG_KEY{0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
45*4d7e907cSAndroid Build Coastguard Worker const std::vector<uint8_t> VALUE{16, 15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1};
46*4d7e907cSAndroid Build Coastguard Worker const std::vector<uint8_t> OTHER_VALUE{0, 1, 1, 2, 3, 5, 8, 13, 21, 34, 55, 89, 144, 233, 255, 255};
47*4d7e907cSAndroid Build Coastguard Worker 
48*4d7e907cSAndroid Build Coastguard Worker class WeaverAdapter {
49*4d7e907cSAndroid Build Coastguard Worker   public:
~WeaverAdapter()50*4d7e907cSAndroid Build Coastguard Worker     virtual ~WeaverAdapter() {}
51*4d7e907cSAndroid Build Coastguard Worker     virtual bool isReady() = 0;
52*4d7e907cSAndroid Build Coastguard Worker     virtual ::ndk::ScopedAStatus getConfig(WeaverConfig* _aidl_return) = 0;
53*4d7e907cSAndroid Build Coastguard Worker     virtual ::ndk::ScopedAStatus read(int32_t in_slotId, const std::vector<uint8_t>& in_key,
54*4d7e907cSAndroid Build Coastguard Worker                                       WeaverReadResponse* _aidl_return) = 0;
55*4d7e907cSAndroid Build Coastguard Worker     virtual ::ndk::ScopedAStatus write(int32_t in_slotId, const std::vector<uint8_t>& in_key,
56*4d7e907cSAndroid Build Coastguard Worker                                        const std::vector<uint8_t>& in_value) = 0;
57*4d7e907cSAndroid Build Coastguard Worker };
58*4d7e907cSAndroid Build Coastguard Worker 
59*4d7e907cSAndroid Build Coastguard Worker class WeaverAidlAdapter : public WeaverAdapter {
60*4d7e907cSAndroid Build Coastguard Worker   public:
WeaverAidlAdapter(const std::string & param)61*4d7e907cSAndroid Build Coastguard Worker     WeaverAidlAdapter(const std::string& param)
62*4d7e907cSAndroid Build Coastguard Worker         : aidl_weaver_(IWeaver::fromBinder(
63*4d7e907cSAndroid Build Coastguard Worker                   ::ndk::SpAIBinder(AServiceManager_waitForService(param.c_str())))) {}
~WeaverAidlAdapter()64*4d7e907cSAndroid Build Coastguard Worker     ~WeaverAidlAdapter() {}
65*4d7e907cSAndroid Build Coastguard Worker 
isReady()66*4d7e907cSAndroid Build Coastguard Worker     bool isReady() { return aidl_weaver_ != nullptr; }
67*4d7e907cSAndroid Build Coastguard Worker 
getConfig(WeaverConfig * _aidl_return)68*4d7e907cSAndroid Build Coastguard Worker     ::ndk::ScopedAStatus getConfig(WeaverConfig* _aidl_return) {
69*4d7e907cSAndroid Build Coastguard Worker         return aidl_weaver_->getConfig(_aidl_return);
70*4d7e907cSAndroid Build Coastguard Worker     }
71*4d7e907cSAndroid Build Coastguard Worker 
read(int32_t in_slotId,const std::vector<uint8_t> & in_key,WeaverReadResponse * _aidl_return)72*4d7e907cSAndroid Build Coastguard Worker     ::ndk::ScopedAStatus read(int32_t in_slotId, const std::vector<uint8_t>& in_key,
73*4d7e907cSAndroid Build Coastguard Worker                               WeaverReadResponse* _aidl_return) {
74*4d7e907cSAndroid Build Coastguard Worker         return aidl_weaver_->read(in_slotId, in_key, _aidl_return);
75*4d7e907cSAndroid Build Coastguard Worker     }
76*4d7e907cSAndroid Build Coastguard Worker 
write(int32_t in_slotId,const std::vector<uint8_t> & in_key,const std::vector<uint8_t> & in_value)77*4d7e907cSAndroid Build Coastguard Worker     ::ndk::ScopedAStatus write(int32_t in_slotId, const std::vector<uint8_t>& in_key,
78*4d7e907cSAndroid Build Coastguard Worker                                const std::vector<uint8_t>& in_value) {
79*4d7e907cSAndroid Build Coastguard Worker         return aidl_weaver_->write(in_slotId, in_key, in_value);
80*4d7e907cSAndroid Build Coastguard Worker     }
81*4d7e907cSAndroid Build Coastguard Worker 
82*4d7e907cSAndroid Build Coastguard Worker   private:
83*4d7e907cSAndroid Build Coastguard Worker     std::shared_ptr<IWeaver> aidl_weaver_;
84*4d7e907cSAndroid Build Coastguard Worker };
85*4d7e907cSAndroid Build Coastguard Worker 
86*4d7e907cSAndroid Build Coastguard Worker class WeaverHidlAdapter : public WeaverAdapter {
87*4d7e907cSAndroid Build Coastguard Worker   public:
WeaverHidlAdapter(const std::string & param)88*4d7e907cSAndroid Build Coastguard Worker     WeaverHidlAdapter(const std::string& param) : hidl_weaver_(HidlIWeaver::getService(param)) {}
~WeaverHidlAdapter()89*4d7e907cSAndroid Build Coastguard Worker     ~WeaverHidlAdapter() {}
90*4d7e907cSAndroid Build Coastguard Worker 
isReady()91*4d7e907cSAndroid Build Coastguard Worker     bool isReady() { return hidl_weaver_ != nullptr; }
92*4d7e907cSAndroid Build Coastguard Worker 
getConfig(WeaverConfig * _aidl_return)93*4d7e907cSAndroid Build Coastguard Worker     ::ndk::ScopedAStatus getConfig(WeaverConfig* _aidl_return) {
94*4d7e907cSAndroid Build Coastguard Worker         bool callbackCalled = false;
95*4d7e907cSAndroid Build Coastguard Worker         HidlWeaverStatus status;
96*4d7e907cSAndroid Build Coastguard Worker         HidlWeaverConfig config;
97*4d7e907cSAndroid Build Coastguard Worker         auto ret = hidl_weaver_->getConfig([&](HidlWeaverStatus s, HidlWeaverConfig c) {
98*4d7e907cSAndroid Build Coastguard Worker             callbackCalled = true;
99*4d7e907cSAndroid Build Coastguard Worker             status = s;
100*4d7e907cSAndroid Build Coastguard Worker             config = c;
101*4d7e907cSAndroid Build Coastguard Worker         });
102*4d7e907cSAndroid Build Coastguard Worker         if (!ret.isOk() || !callbackCalled || status != HidlWeaverStatus::OK) {
103*4d7e907cSAndroid Build Coastguard Worker             return ::ndk::ScopedAStatus::fromStatus(STATUS_FAILED_TRANSACTION);
104*4d7e907cSAndroid Build Coastguard Worker         }
105*4d7e907cSAndroid Build Coastguard Worker         _aidl_return->slots = config.slots;
106*4d7e907cSAndroid Build Coastguard Worker         _aidl_return->keySize = config.keySize;
107*4d7e907cSAndroid Build Coastguard Worker         _aidl_return->valueSize = config.valueSize;
108*4d7e907cSAndroid Build Coastguard Worker         return ::ndk::ScopedAStatus::ok();
109*4d7e907cSAndroid Build Coastguard Worker     }
110*4d7e907cSAndroid Build Coastguard Worker 
read(int32_t in_slotId,const std::vector<uint8_t> & in_key,WeaverReadResponse * _aidl_return)111*4d7e907cSAndroid Build Coastguard Worker     ::ndk::ScopedAStatus read(int32_t in_slotId, const std::vector<uint8_t>& in_key,
112*4d7e907cSAndroid Build Coastguard Worker                               WeaverReadResponse* _aidl_return) {
113*4d7e907cSAndroid Build Coastguard Worker         bool callbackCalled = false;
114*4d7e907cSAndroid Build Coastguard Worker         HidlWeaverReadStatus status;
115*4d7e907cSAndroid Build Coastguard Worker         std::vector<uint8_t> value;
116*4d7e907cSAndroid Build Coastguard Worker         uint32_t timeout;
117*4d7e907cSAndroid Build Coastguard Worker         auto ret = hidl_weaver_->read(in_slotId, in_key,
118*4d7e907cSAndroid Build Coastguard Worker                                       [&](HidlWeaverReadStatus s, HidlWeaverReadResponse r) {
119*4d7e907cSAndroid Build Coastguard Worker                                           callbackCalled = true;
120*4d7e907cSAndroid Build Coastguard Worker                                           status = s;
121*4d7e907cSAndroid Build Coastguard Worker                                           value = r.value;
122*4d7e907cSAndroid Build Coastguard Worker                                           timeout = r.timeout;
123*4d7e907cSAndroid Build Coastguard Worker                                       });
124*4d7e907cSAndroid Build Coastguard Worker         if (!ret.isOk() || !callbackCalled) {
125*4d7e907cSAndroid Build Coastguard Worker             return ::ndk::ScopedAStatus::fromStatus(STATUS_FAILED_TRANSACTION);
126*4d7e907cSAndroid Build Coastguard Worker         }
127*4d7e907cSAndroid Build Coastguard Worker         switch (status) {
128*4d7e907cSAndroid Build Coastguard Worker             case HidlWeaverReadStatus::OK:
129*4d7e907cSAndroid Build Coastguard Worker                 _aidl_return->status = WeaverReadStatus::OK;
130*4d7e907cSAndroid Build Coastguard Worker                 break;
131*4d7e907cSAndroid Build Coastguard Worker             case HidlWeaverReadStatus::FAILED:
132*4d7e907cSAndroid Build Coastguard Worker                 _aidl_return->status = WeaverReadStatus::FAILED;
133*4d7e907cSAndroid Build Coastguard Worker                 break;
134*4d7e907cSAndroid Build Coastguard Worker             case HidlWeaverReadStatus::INCORRECT_KEY:
135*4d7e907cSAndroid Build Coastguard Worker                 _aidl_return->status = WeaverReadStatus::INCORRECT_KEY;
136*4d7e907cSAndroid Build Coastguard Worker                 break;
137*4d7e907cSAndroid Build Coastguard Worker             case HidlWeaverReadStatus::THROTTLE:
138*4d7e907cSAndroid Build Coastguard Worker                 _aidl_return->status = WeaverReadStatus::THROTTLE;
139*4d7e907cSAndroid Build Coastguard Worker                 break;
140*4d7e907cSAndroid Build Coastguard Worker             default:
141*4d7e907cSAndroid Build Coastguard Worker                 ADD_FAILURE() << "Unknown HIDL read status: " << static_cast<uint32_t>(status);
142*4d7e907cSAndroid Build Coastguard Worker                 _aidl_return->status = WeaverReadStatus::FAILED;
143*4d7e907cSAndroid Build Coastguard Worker                 break;
144*4d7e907cSAndroid Build Coastguard Worker         }
145*4d7e907cSAndroid Build Coastguard Worker         _aidl_return->value = value;
146*4d7e907cSAndroid Build Coastguard Worker         _aidl_return->timeout = timeout;
147*4d7e907cSAndroid Build Coastguard Worker         return ::ndk::ScopedAStatus::ok();
148*4d7e907cSAndroid Build Coastguard Worker     }
149*4d7e907cSAndroid Build Coastguard Worker 
write(int32_t in_slotId,const std::vector<uint8_t> & in_key,const std::vector<uint8_t> & in_value)150*4d7e907cSAndroid Build Coastguard Worker     ::ndk::ScopedAStatus write(int32_t in_slotId, const std::vector<uint8_t>& in_key,
151*4d7e907cSAndroid Build Coastguard Worker                                const std::vector<uint8_t>& in_value) {
152*4d7e907cSAndroid Build Coastguard Worker         auto status = hidl_weaver_->write(in_slotId, in_key, in_value);
153*4d7e907cSAndroid Build Coastguard Worker         switch (status) {
154*4d7e907cSAndroid Build Coastguard Worker             case HidlWeaverStatus::OK:
155*4d7e907cSAndroid Build Coastguard Worker                 return ::ndk::ScopedAStatus::ok();
156*4d7e907cSAndroid Build Coastguard Worker             case HidlWeaverStatus::FAILED:
157*4d7e907cSAndroid Build Coastguard Worker                 return ::ndk::ScopedAStatus::fromStatus(STATUS_FAILED_TRANSACTION);
158*4d7e907cSAndroid Build Coastguard Worker             default:
159*4d7e907cSAndroid Build Coastguard Worker                 ADD_FAILURE() << "Unknown HIDL write status: " << status.description();
160*4d7e907cSAndroid Build Coastguard Worker                 return ::ndk::ScopedAStatus::fromStatus(STATUS_FAILED_TRANSACTION);
161*4d7e907cSAndroid Build Coastguard Worker         }
162*4d7e907cSAndroid Build Coastguard Worker     }
163*4d7e907cSAndroid Build Coastguard Worker 
164*4d7e907cSAndroid Build Coastguard Worker   private:
165*4d7e907cSAndroid Build Coastguard Worker     android::sp<HidlIWeaver> hidl_weaver_;
166*4d7e907cSAndroid Build Coastguard Worker };
167*4d7e907cSAndroid Build Coastguard Worker 
168*4d7e907cSAndroid Build Coastguard Worker class WeaverTest : public ::testing::TestWithParam<std::tuple<std::string, std::string>> {
169*4d7e907cSAndroid Build Coastguard Worker   protected:
170*4d7e907cSAndroid Build Coastguard Worker     void SetUp() override;
TearDown()171*4d7e907cSAndroid Build Coastguard Worker     void TearDown() override {}
172*4d7e907cSAndroid Build Coastguard Worker     void FindFreeSlots();
173*4d7e907cSAndroid Build Coastguard Worker 
174*4d7e907cSAndroid Build Coastguard Worker     std::unique_ptr<WeaverAdapter> weaver_;
175*4d7e907cSAndroid Build Coastguard Worker     WeaverConfig config_;
176*4d7e907cSAndroid Build Coastguard Worker     uint32_t first_free_slot_;
177*4d7e907cSAndroid Build Coastguard Worker     uint32_t last_free_slot_;
178*4d7e907cSAndroid Build Coastguard Worker };
179*4d7e907cSAndroid Build Coastguard Worker 
SetUp()180*4d7e907cSAndroid Build Coastguard Worker void WeaverTest::SetUp() {
181*4d7e907cSAndroid Build Coastguard Worker     std::string api, instance_name;
182*4d7e907cSAndroid Build Coastguard Worker     std::tie(api, instance_name) = GetParam();
183*4d7e907cSAndroid Build Coastguard Worker     if (api == "hidl") {
184*4d7e907cSAndroid Build Coastguard Worker         weaver_.reset(new WeaverHidlAdapter(instance_name));
185*4d7e907cSAndroid Build Coastguard Worker     } else if (api == "aidl") {
186*4d7e907cSAndroid Build Coastguard Worker         weaver_.reset(new WeaverAidlAdapter(instance_name));
187*4d7e907cSAndroid Build Coastguard Worker     } else {
188*4d7e907cSAndroid Build Coastguard Worker         FAIL() << "Bad test parameterization";
189*4d7e907cSAndroid Build Coastguard Worker     }
190*4d7e907cSAndroid Build Coastguard Worker     ASSERT_TRUE(weaver_->isReady());
191*4d7e907cSAndroid Build Coastguard Worker 
192*4d7e907cSAndroid Build Coastguard Worker     auto ret = weaver_->getConfig(&config_);
193*4d7e907cSAndroid Build Coastguard Worker     ASSERT_TRUE(ret.isOk());
194*4d7e907cSAndroid Build Coastguard Worker     ASSERT_GT(config_.slots, 0);
195*4d7e907cSAndroid Build Coastguard Worker     GTEST_LOG_(INFO) << "WeaverConfig: slots=" << config_.slots << ", keySize=" << config_.keySize
196*4d7e907cSAndroid Build Coastguard Worker                      << ", valueSize=" << config_.valueSize;
197*4d7e907cSAndroid Build Coastguard Worker 
198*4d7e907cSAndroid Build Coastguard Worker     FindFreeSlots();
199*4d7e907cSAndroid Build Coastguard Worker     GTEST_LOG_(INFO) << "First free slot is " << first_free_slot_ << ", last free slot is "
200*4d7e907cSAndroid Build Coastguard Worker                      << last_free_slot_;
201*4d7e907cSAndroid Build Coastguard Worker }
202*4d7e907cSAndroid Build Coastguard Worker 
FindFreeSlots()203*4d7e907cSAndroid Build Coastguard Worker void WeaverTest::FindFreeSlots() {
204*4d7e907cSAndroid Build Coastguard Worker     // Determine which Weaver slots are in use by the system. These slots can't be used by the test.
205*4d7e907cSAndroid Build Coastguard Worker     std::set<uint32_t> used_slots;
206*4d7e907cSAndroid Build Coastguard Worker     if (access(kSlotMapFile.c_str(), F_OK) == 0) {
207*4d7e907cSAndroid Build Coastguard Worker         std::string contents;
208*4d7e907cSAndroid Build Coastguard Worker         ASSERT_TRUE(android::base::ReadFileToString(kSlotMapFile, &contents))
209*4d7e907cSAndroid Build Coastguard Worker                 << "Failed to read " << kSlotMapFile;
210*4d7e907cSAndroid Build Coastguard Worker         for (const auto& line : android::base::Split(contents, "\n")) {
211*4d7e907cSAndroid Build Coastguard Worker             auto trimmed_line = android::base::Trim(line);
212*4d7e907cSAndroid Build Coastguard Worker             if (trimmed_line[0] == '#' || trimmed_line[0] == '\0') continue;
213*4d7e907cSAndroid Build Coastguard Worker             auto slot_and_user = android::base::Split(trimmed_line, "=");
214*4d7e907cSAndroid Build Coastguard Worker             uint32_t slot;
215*4d7e907cSAndroid Build Coastguard Worker             ASSERT_TRUE(slot_and_user.size() == 2 &&
216*4d7e907cSAndroid Build Coastguard Worker                         android::base::ParseUint(slot_and_user[0], &slot))
217*4d7e907cSAndroid Build Coastguard Worker                     << "Error parsing " << kSlotMapFile << " at \"" << line << "\"";
218*4d7e907cSAndroid Build Coastguard Worker             GTEST_LOG_(INFO) << "Slot " << slot << " is in use by " << slot_and_user[1];
219*4d7e907cSAndroid Build Coastguard Worker             ASSERT_LT(slot, config_.slots);
220*4d7e907cSAndroid Build Coastguard Worker             used_slots.insert(slot);
221*4d7e907cSAndroid Build Coastguard Worker         }
222*4d7e907cSAndroid Build Coastguard Worker     }
223*4d7e907cSAndroid Build Coastguard Worker 
224*4d7e907cSAndroid Build Coastguard Worker     // We should assert !used_slots.empty() here, but that can't be done yet due to
225*4d7e907cSAndroid Build Coastguard Worker     // config_disableWeaverOnUnsecuredUsers being supported.  The value of that option is not
226*4d7e907cSAndroid Build Coastguard Worker     // accessible from here, so we have to assume it might be set to true.
227*4d7e907cSAndroid Build Coastguard Worker 
228*4d7e907cSAndroid Build Coastguard Worker     // Find the first free slot.
229*4d7e907cSAndroid Build Coastguard Worker     int found = 0;
230*4d7e907cSAndroid Build Coastguard Worker     for (uint32_t i = 0; i < config_.slots; i++) {
231*4d7e907cSAndroid Build Coastguard Worker         if (used_slots.find(i) == used_slots.end()) {
232*4d7e907cSAndroid Build Coastguard Worker             first_free_slot_ = i;
233*4d7e907cSAndroid Build Coastguard Worker             found++;
234*4d7e907cSAndroid Build Coastguard Worker             break;
235*4d7e907cSAndroid Build Coastguard Worker         }
236*4d7e907cSAndroid Build Coastguard Worker     }
237*4d7e907cSAndroid Build Coastguard Worker     // Find the last free slot.
238*4d7e907cSAndroid Build Coastguard Worker     for (uint32_t i = config_.slots; i > 0; i--) {
239*4d7e907cSAndroid Build Coastguard Worker         if (used_slots.find(i - 1) == used_slots.end()) {
240*4d7e907cSAndroid Build Coastguard Worker             last_free_slot_ = i - 1;
241*4d7e907cSAndroid Build Coastguard Worker             found++;
242*4d7e907cSAndroid Build Coastguard Worker             break;
243*4d7e907cSAndroid Build Coastguard Worker         }
244*4d7e907cSAndroid Build Coastguard Worker     }
245*4d7e907cSAndroid Build Coastguard Worker     ASSERT_EQ(found, 2) << "All Weaver slots are already in use by the system";
246*4d7e907cSAndroid Build Coastguard Worker }
247*4d7e907cSAndroid Build Coastguard Worker 
248*4d7e907cSAndroid Build Coastguard Worker /*
249*4d7e907cSAndroid Build Coastguard Worker  * Checks config values are suitably large
250*4d7e907cSAndroid Build Coastguard Worker  */
TEST_P(WeaverTest,GetConfig)251*4d7e907cSAndroid Build Coastguard Worker TEST_P(WeaverTest, GetConfig) {
252*4d7e907cSAndroid Build Coastguard Worker     EXPECT_GE(config_.slots, 16u);
253*4d7e907cSAndroid Build Coastguard Worker     EXPECT_GE(config_.keySize, 16u);
254*4d7e907cSAndroid Build Coastguard Worker     EXPECT_GE(config_.valueSize, 16u);
255*4d7e907cSAndroid Build Coastguard Worker }
256*4d7e907cSAndroid Build Coastguard Worker 
257*4d7e907cSAndroid Build Coastguard Worker /*
258*4d7e907cSAndroid Build Coastguard Worker  * Gets the config twice and checks they are the same
259*4d7e907cSAndroid Build Coastguard Worker  */
TEST_P(WeaverTest,GettingConfigMultipleTimesGivesSameResult)260*4d7e907cSAndroid Build Coastguard Worker TEST_P(WeaverTest, GettingConfigMultipleTimesGivesSameResult) {
261*4d7e907cSAndroid Build Coastguard Worker     WeaverConfig config2;
262*4d7e907cSAndroid Build Coastguard Worker 
263*4d7e907cSAndroid Build Coastguard Worker     auto ret = weaver_->getConfig(&config2);
264*4d7e907cSAndroid Build Coastguard Worker     ASSERT_TRUE(ret.isOk());
265*4d7e907cSAndroid Build Coastguard Worker 
266*4d7e907cSAndroid Build Coastguard Worker     EXPECT_EQ(config_, config2);
267*4d7e907cSAndroid Build Coastguard Worker }
268*4d7e907cSAndroid Build Coastguard Worker 
269*4d7e907cSAndroid Build Coastguard Worker /*
270*4d7e907cSAndroid Build Coastguard Worker  * Writes a key and value to the last free slot
271*4d7e907cSAndroid Build Coastguard Worker  */
TEST_P(WeaverTest,WriteToLastSlot)272*4d7e907cSAndroid Build Coastguard Worker TEST_P(WeaverTest, WriteToLastSlot) {
273*4d7e907cSAndroid Build Coastguard Worker     const auto writeRet = weaver_->write(last_free_slot_, KEY, VALUE);
274*4d7e907cSAndroid Build Coastguard Worker     ASSERT_TRUE(writeRet.isOk());
275*4d7e907cSAndroid Build Coastguard Worker }
276*4d7e907cSAndroid Build Coastguard Worker 
277*4d7e907cSAndroid Build Coastguard Worker /*
278*4d7e907cSAndroid Build Coastguard Worker  * Writes a key and value to a slot
279*4d7e907cSAndroid Build Coastguard Worker  * Reads the slot with the same key and receives the value that was previously written
280*4d7e907cSAndroid Build Coastguard Worker  */
TEST_P(WeaverTest,WriteFollowedByReadGivesTheSameValue)281*4d7e907cSAndroid Build Coastguard Worker TEST_P(WeaverTest, WriteFollowedByReadGivesTheSameValue) {
282*4d7e907cSAndroid Build Coastguard Worker     const uint32_t slotId = first_free_slot_;
283*4d7e907cSAndroid Build Coastguard Worker     const auto ret = weaver_->write(slotId, KEY, VALUE);
284*4d7e907cSAndroid Build Coastguard Worker     ASSERT_TRUE(ret.isOk());
285*4d7e907cSAndroid Build Coastguard Worker 
286*4d7e907cSAndroid Build Coastguard Worker     WeaverReadResponse response;
287*4d7e907cSAndroid Build Coastguard Worker     const auto readRet = weaver_->read(slotId, KEY, &response);
288*4d7e907cSAndroid Build Coastguard Worker     ASSERT_TRUE(readRet.isOk());
289*4d7e907cSAndroid Build Coastguard Worker     EXPECT_EQ(response.value, VALUE);
290*4d7e907cSAndroid Build Coastguard Worker     EXPECT_EQ(response.timeout, 0u);
291*4d7e907cSAndroid Build Coastguard Worker     EXPECT_EQ(response.status, WeaverReadStatus::OK);
292*4d7e907cSAndroid Build Coastguard Worker }
293*4d7e907cSAndroid Build Coastguard Worker 
294*4d7e907cSAndroid Build Coastguard Worker /*
295*4d7e907cSAndroid Build Coastguard Worker  * Writes a key and value to a slot
296*4d7e907cSAndroid Build Coastguard Worker  * Overwrites the slot with a new key and value
297*4d7e907cSAndroid Build Coastguard Worker  * Reads the slot with the new key and receives the new value
298*4d7e907cSAndroid Build Coastguard Worker  */
TEST_P(WeaverTest,OverwritingSlotUpdatesTheValue)299*4d7e907cSAndroid Build Coastguard Worker TEST_P(WeaverTest, OverwritingSlotUpdatesTheValue) {
300*4d7e907cSAndroid Build Coastguard Worker     const uint32_t slotId = first_free_slot_;
301*4d7e907cSAndroid Build Coastguard Worker     const auto initialWriteRet = weaver_->write(slotId, WRONG_KEY, VALUE);
302*4d7e907cSAndroid Build Coastguard Worker     ASSERT_TRUE(initialWriteRet.isOk());
303*4d7e907cSAndroid Build Coastguard Worker 
304*4d7e907cSAndroid Build Coastguard Worker     const auto overwriteRet = weaver_->write(slotId, KEY, OTHER_VALUE);
305*4d7e907cSAndroid Build Coastguard Worker     ASSERT_TRUE(overwriteRet.isOk());
306*4d7e907cSAndroid Build Coastguard Worker 
307*4d7e907cSAndroid Build Coastguard Worker     WeaverReadResponse response;
308*4d7e907cSAndroid Build Coastguard Worker     const auto readRet = weaver_->read(slotId, KEY, &response);
309*4d7e907cSAndroid Build Coastguard Worker     ASSERT_TRUE(readRet.isOk());
310*4d7e907cSAndroid Build Coastguard Worker     EXPECT_EQ(response.value, OTHER_VALUE);
311*4d7e907cSAndroid Build Coastguard Worker     EXPECT_EQ(response.timeout, 0u);
312*4d7e907cSAndroid Build Coastguard Worker     EXPECT_EQ(response.status, WeaverReadStatus::OK);
313*4d7e907cSAndroid Build Coastguard Worker }
314*4d7e907cSAndroid Build Coastguard Worker 
315*4d7e907cSAndroid Build Coastguard Worker /*
316*4d7e907cSAndroid Build Coastguard Worker  * Writes a key and value to a slot
317*4d7e907cSAndroid Build Coastguard Worker  * Reads the slot with a different key so does not receive the value
318*4d7e907cSAndroid Build Coastguard Worker  */
TEST_P(WeaverTest,WriteFollowedByReadWithWrongKeyDoesNotGiveTheValue)319*4d7e907cSAndroid Build Coastguard Worker TEST_P(WeaverTest, WriteFollowedByReadWithWrongKeyDoesNotGiveTheValue) {
320*4d7e907cSAndroid Build Coastguard Worker     const uint32_t slotId = first_free_slot_;
321*4d7e907cSAndroid Build Coastguard Worker     const auto writeRet = weaver_->write(slotId, KEY, VALUE);
322*4d7e907cSAndroid Build Coastguard Worker     ASSERT_TRUE(writeRet.isOk());
323*4d7e907cSAndroid Build Coastguard Worker 
324*4d7e907cSAndroid Build Coastguard Worker     WeaverReadResponse response;
325*4d7e907cSAndroid Build Coastguard Worker     const auto readRet = weaver_->read(slotId, WRONG_KEY, &response);
326*4d7e907cSAndroid Build Coastguard Worker     ASSERT_TRUE(readRet.isOk());
327*4d7e907cSAndroid Build Coastguard Worker     EXPECT_TRUE(response.value.empty());
328*4d7e907cSAndroid Build Coastguard Worker     EXPECT_EQ(response.status, WeaverReadStatus::INCORRECT_KEY);
329*4d7e907cSAndroid Build Coastguard Worker }
330*4d7e907cSAndroid Build Coastguard Worker 
331*4d7e907cSAndroid Build Coastguard Worker /*
332*4d7e907cSAndroid Build Coastguard Worker  * Writing to an invalid slot fails
333*4d7e907cSAndroid Build Coastguard Worker  */
TEST_P(WeaverTest,WritingToInvalidSlotFails)334*4d7e907cSAndroid Build Coastguard Worker TEST_P(WeaverTest, WritingToInvalidSlotFails) {
335*4d7e907cSAndroid Build Coastguard Worker     if (config_.slots == std::numeric_limits<uint32_t>::max()) {
336*4d7e907cSAndroid Build Coastguard Worker         // If there are no invalid slots then pass
337*4d7e907cSAndroid Build Coastguard Worker         return;
338*4d7e907cSAndroid Build Coastguard Worker     }
339*4d7e907cSAndroid Build Coastguard Worker 
340*4d7e907cSAndroid Build Coastguard Worker     const auto writeRet = weaver_->write(config_.slots, KEY, VALUE);
341*4d7e907cSAndroid Build Coastguard Worker     ASSERT_FALSE(writeRet.isOk());
342*4d7e907cSAndroid Build Coastguard Worker }
343*4d7e907cSAndroid Build Coastguard Worker 
344*4d7e907cSAndroid Build Coastguard Worker /*
345*4d7e907cSAndroid Build Coastguard Worker  * Reading from an invalid slot fails rather than incorrect key
346*4d7e907cSAndroid Build Coastguard Worker  */
TEST_P(WeaverTest,ReadingFromInvalidSlotFails)347*4d7e907cSAndroid Build Coastguard Worker TEST_P(WeaverTest, ReadingFromInvalidSlotFails) {
348*4d7e907cSAndroid Build Coastguard Worker     if (config_.slots == std::numeric_limits<uint32_t>::max()) {
349*4d7e907cSAndroid Build Coastguard Worker         // If there are no invalid slots then pass
350*4d7e907cSAndroid Build Coastguard Worker         return;
351*4d7e907cSAndroid Build Coastguard Worker     }
352*4d7e907cSAndroid Build Coastguard Worker 
353*4d7e907cSAndroid Build Coastguard Worker     WeaverReadResponse response;
354*4d7e907cSAndroid Build Coastguard Worker     const auto readRet = weaver_->read(config_.slots, KEY, &response);
355*4d7e907cSAndroid Build Coastguard Worker     ASSERT_TRUE(readRet.isOk());
356*4d7e907cSAndroid Build Coastguard Worker     EXPECT_TRUE(response.value.empty());
357*4d7e907cSAndroid Build Coastguard Worker     EXPECT_EQ(response.timeout, 0u);
358*4d7e907cSAndroid Build Coastguard Worker     EXPECT_EQ(response.status, WeaverReadStatus::FAILED);
359*4d7e907cSAndroid Build Coastguard Worker }
360*4d7e907cSAndroid Build Coastguard Worker 
361*4d7e907cSAndroid Build Coastguard Worker /*
362*4d7e907cSAndroid Build Coastguard Worker  * Writing a key that is too large fails
363*4d7e907cSAndroid Build Coastguard Worker  */
TEST_P(WeaverTest,WriteWithTooLargeKeyFails)364*4d7e907cSAndroid Build Coastguard Worker TEST_P(WeaverTest, WriteWithTooLargeKeyFails) {
365*4d7e907cSAndroid Build Coastguard Worker     std::vector<uint8_t> bigKey(config_.keySize + 1);
366*4d7e907cSAndroid Build Coastguard Worker 
367*4d7e907cSAndroid Build Coastguard Worker     const auto writeRet = weaver_->write(first_free_slot_, bigKey, VALUE);
368*4d7e907cSAndroid Build Coastguard Worker     ASSERT_FALSE(writeRet.isOk());
369*4d7e907cSAndroid Build Coastguard Worker }
370*4d7e907cSAndroid Build Coastguard Worker 
371*4d7e907cSAndroid Build Coastguard Worker /*
372*4d7e907cSAndroid Build Coastguard Worker  * Writing a value that is too large fails
373*4d7e907cSAndroid Build Coastguard Worker  */
TEST_P(WeaverTest,WriteWithTooLargeValueFails)374*4d7e907cSAndroid Build Coastguard Worker TEST_P(WeaverTest, WriteWithTooLargeValueFails) {
375*4d7e907cSAndroid Build Coastguard Worker     std::vector<uint8_t> bigValue(config_.valueSize + 1);
376*4d7e907cSAndroid Build Coastguard Worker 
377*4d7e907cSAndroid Build Coastguard Worker     const auto writeRet = weaver_->write(first_free_slot_, KEY, bigValue);
378*4d7e907cSAndroid Build Coastguard Worker     ASSERT_FALSE(writeRet.isOk());
379*4d7e907cSAndroid Build Coastguard Worker }
380*4d7e907cSAndroid Build Coastguard Worker 
381*4d7e907cSAndroid Build Coastguard Worker /*
382*4d7e907cSAndroid Build Coastguard Worker  * Reading with a key that is too large fails
383*4d7e907cSAndroid Build Coastguard Worker  */
TEST_P(WeaverTest,ReadWithTooLargeKeyFails)384*4d7e907cSAndroid Build Coastguard Worker TEST_P(WeaverTest, ReadWithTooLargeKeyFails) {
385*4d7e907cSAndroid Build Coastguard Worker     std::vector<uint8_t> bigKey(config_.keySize + 1);
386*4d7e907cSAndroid Build Coastguard Worker 
387*4d7e907cSAndroid Build Coastguard Worker     WeaverReadResponse response;
388*4d7e907cSAndroid Build Coastguard Worker     const auto readRet = weaver_->read(first_free_slot_, bigKey, &response);
389*4d7e907cSAndroid Build Coastguard Worker     ASSERT_TRUE(readRet.isOk());
390*4d7e907cSAndroid Build Coastguard Worker     EXPECT_TRUE(response.value.empty());
391*4d7e907cSAndroid Build Coastguard Worker     EXPECT_EQ(response.timeout, 0u);
392*4d7e907cSAndroid Build Coastguard Worker     EXPECT_EQ(response.status, WeaverReadStatus::FAILED);
393*4d7e907cSAndroid Build Coastguard Worker }
394*4d7e907cSAndroid Build Coastguard Worker 
395*4d7e907cSAndroid Build Coastguard Worker GTEST_ALLOW_UNINSTANTIATED_PARAMETERIZED_TEST(WeaverTest);
396*4d7e907cSAndroid Build Coastguard Worker 
397*4d7e907cSAndroid Build Coastguard Worker // Instantiate the test for each HIDL Weaver service.
398*4d7e907cSAndroid Build Coastguard Worker INSTANTIATE_TEST_SUITE_P(
399*4d7e907cSAndroid Build Coastguard Worker         PerHidlInstance, WeaverTest,
400*4d7e907cSAndroid Build Coastguard Worker         testing::Combine(testing::Values("hidl"),
401*4d7e907cSAndroid Build Coastguard Worker                          testing::ValuesIn(android::hardware::getAllHalInstanceNames(
402*4d7e907cSAndroid Build Coastguard Worker                                  HidlIWeaver::descriptor))),
__anon62c0e71e0302(const testing::TestParamInfo<std::tuple<std::string, std::string>>& info) 403*4d7e907cSAndroid Build Coastguard Worker         [](const testing::TestParamInfo<std::tuple<std::string, std::string>>& info) {
404*4d7e907cSAndroid Build Coastguard Worker             return android::hardware::PrintInstanceNameToString(
405*4d7e907cSAndroid Build Coastguard Worker                     testing::TestParamInfo<std::string>{std::get<1>(info.param), info.index});
406*4d7e907cSAndroid Build Coastguard Worker         });
407*4d7e907cSAndroid Build Coastguard Worker 
408*4d7e907cSAndroid Build Coastguard Worker // Instantiate the test for each AIDL Weaver service.
409*4d7e907cSAndroid Build Coastguard Worker INSTANTIATE_TEST_SUITE_P(
410*4d7e907cSAndroid Build Coastguard Worker         PerAidlInstance, WeaverTest,
411*4d7e907cSAndroid Build Coastguard Worker         testing::Combine(testing::Values("aidl"),
412*4d7e907cSAndroid Build Coastguard Worker                          testing::ValuesIn(android::getAidlHalInstanceNames(IWeaver::descriptor))),
__anon62c0e71e0402(const testing::TestParamInfo<std::tuple<std::string, std::string>>& info) 413*4d7e907cSAndroid Build Coastguard Worker         [](const testing::TestParamInfo<std::tuple<std::string, std::string>>& info) {
414*4d7e907cSAndroid Build Coastguard Worker             // This name_generator makes the instance name be included in the test case names, e.g.
415*4d7e907cSAndroid Build Coastguard Worker             // "PerAidlInstance/WeaverTest#GetConfig/0_android_hardware_weaver_IWeaver_default"
416*4d7e907cSAndroid Build Coastguard Worker             // instead of "PerAidlInstance/WeaverTest#GetConfig/0".
417*4d7e907cSAndroid Build Coastguard Worker             return android::PrintInstanceNameToString(
418*4d7e907cSAndroid Build Coastguard Worker                     testing::TestParamInfo<std::string>{std::get<1>(info.param), info.index});
419*4d7e907cSAndroid Build Coastguard Worker         });
420*4d7e907cSAndroid Build Coastguard Worker 
main(int argc,char ** argv)421*4d7e907cSAndroid Build Coastguard Worker int main(int argc, char** argv) {
422*4d7e907cSAndroid Build Coastguard Worker     ::testing::InitGoogleTest(&argc, argv);
423*4d7e907cSAndroid Build Coastguard Worker     ABinderProcess_setThreadPoolMaxThreadCount(1);
424*4d7e907cSAndroid Build Coastguard Worker     ABinderProcess_startThreadPool();
425*4d7e907cSAndroid Build Coastguard Worker     return RUN_ALL_TESTS();
426*4d7e907cSAndroid Build Coastguard Worker }
427