1 /*
2 * Copyright (C) 2019 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17 #include <gtest/gtest.h>
18 #include <gmock/gmock.h>
19
20 #include <binder/Binder.h>
21 #include <binder/ProcessState.h>
22 #include <binder/IServiceManager.h>
23
24 #include "fakeservicemanager/FakeServiceManager.h"
25 #include "rust/wrappers/FakeServiceManagerWrapper.hpp"
26
27 using android::sp;
28 using android::BBinder;
29 using android::IBinder;
30 using android::OK;
31 using android::status_t;
32 using android::FakeServiceManager;
33 using android::String16;
34 using android::IServiceManager;
35 using android::defaultServiceManager;
36 using testing::ElementsAre;
37
getBinder()38 static sp<IBinder> getBinder() {
39 class LinkableBinder : public BBinder {
40 status_t linkToDeath(const sp<DeathRecipient>&, void*, uint32_t) override {
41 // let SM linkToDeath
42 return OK;
43 }
44 };
45
46 return new LinkableBinder;
47 }
48
TEST(AddService,HappyHappy)49 TEST(AddService, HappyHappy) {
50 auto sm = new FakeServiceManager();
51 EXPECT_EQ(sm->addService(String16("foo"), getBinder(), false /*allowIsolated*/,
52 IServiceManager::DUMP_FLAG_PRIORITY_DEFAULT), OK);
53 }
54
TEST(AddService,SadNullBinder)55 TEST(AddService, SadNullBinder) {
56 auto sm = new FakeServiceManager();
57 EXPECT_EQ(sm->addService(String16("foo"), nullptr, false /*allowIsolated*/,
58 IServiceManager::DUMP_FLAG_PRIORITY_DEFAULT), android::UNEXPECTED_NULL);
59 }
60
TEST(AddService,HappyOverExistingService)61 TEST(AddService, HappyOverExistingService) {
62 auto sm = new FakeServiceManager();
63 EXPECT_EQ(sm->addService(String16("foo"), getBinder(), false /*allowIsolated*/,
64 IServiceManager::DUMP_FLAG_PRIORITY_DEFAULT), OK);
65 EXPECT_EQ(sm->addService(String16("foo"), getBinder(), false /*allowIsolated*/,
66 IServiceManager::DUMP_FLAG_PRIORITY_DEFAULT), OK);
67 }
68
TEST(AddService,HappyClearAddedService)69 TEST(AddService, HappyClearAddedService) {
70 auto sm = new FakeServiceManager();
71 EXPECT_EQ(sm->addService(String16("foo"), getBinder(), false /*allowIsolated*/,
72 IServiceManager::DUMP_FLAG_PRIORITY_DEFAULT), OK);
73 EXPECT_NE(sm->getService(String16("foo")), nullptr);
74 sm->clear();
75 EXPECT_EQ(sm->getService(String16("foo")), nullptr);
76 }
77
TEST(GetService,HappyHappy)78 TEST(GetService, HappyHappy) {
79 auto sm = new FakeServiceManager();
80 sp<IBinder> service = getBinder();
81
82 EXPECT_EQ(sm->addService(String16("foo"), service, false /*allowIsolated*/,
83 IServiceManager::DUMP_FLAG_PRIORITY_DEFAULT), OK);
84
85 EXPECT_EQ(sm->getService(String16("foo")), service);
86 }
87
TEST(GetService,NonExistent)88 TEST(GetService, NonExistent) {
89 auto sm = new FakeServiceManager();
90
91 EXPECT_EQ(sm->getService(String16("foo")), nullptr);
92 }
93
TEST(ListServices,AllServices)94 TEST(ListServices, AllServices) {
95 auto sm = new FakeServiceManager();
96
97 EXPECT_EQ(sm->addService(String16("sd"), getBinder(), false /*allowIsolated*/,
98 IServiceManager::DUMP_FLAG_PRIORITY_DEFAULT), OK);
99 EXPECT_EQ(sm->addService(String16("sc"), getBinder(), false /*allowIsolated*/,
100 IServiceManager::DUMP_FLAG_PRIORITY_NORMAL), OK);
101 EXPECT_EQ(sm->addService(String16("sb"), getBinder(), false /*allowIsolated*/,
102 IServiceManager::DUMP_FLAG_PRIORITY_HIGH), OK);
103 EXPECT_EQ(sm->addService(String16("sa"), getBinder(), false /*allowIsolated*/,
104 IServiceManager::DUMP_FLAG_PRIORITY_CRITICAL), OK);
105
106 android::Vector<String16> out = sm->listServices(IServiceManager::DUMP_FLAG_PRIORITY_ALL);
107
108 // all there and in the right order
109 EXPECT_THAT(out, ElementsAre(String16("sa"), String16("sb"), String16("sc"),
110 String16("sd")));
111 }
112
TEST(WaitForService,NonExistent)113 TEST(WaitForService, NonExistent) {
114 auto sm = new FakeServiceManager();
115
116 EXPECT_EQ(sm->waitForService(String16("foo")), nullptr);
117 }
118
TEST(WaitForService,HappyHappy)119 TEST(WaitForService, HappyHappy) {
120 auto sm = new FakeServiceManager();
121 sp<IBinder> service = getBinder();
122
123 EXPECT_EQ(sm->addService(String16("foo"), service, false /*allowIsolated*/,
124 IServiceManager::DUMP_FLAG_PRIORITY_DEFAULT), OK);
125
126 EXPECT_EQ(sm->waitForService(String16("foo")), service);
127 }
128
TEST(IsDeclared,NonExistent)129 TEST(IsDeclared, NonExistent) {
130 auto sm = new FakeServiceManager();
131
132 EXPECT_FALSE(sm->isDeclared(String16("foo")));
133 }
134
TEST(IsDeclared,HappyHappy)135 TEST(IsDeclared, HappyHappy) {
136 auto sm = new FakeServiceManager();
137 sp<IBinder> service = getBinder();
138
139 EXPECT_EQ(sm->addService(String16("foo"), service, false /*allowIsolated*/,
140 IServiceManager::DUMP_FLAG_PRIORITY_DEFAULT), OK);
141
142 EXPECT_TRUE(sm->isDeclared(String16("foo")));
143 }
144
TEST(SetupFakeServiceManager,NonExistent)145 TEST(SetupFakeServiceManager, NonExistent) {
146 setupFakeServiceManager();
147
148 EXPECT_EQ(defaultServiceManager()->getService(String16("foo")), nullptr);
149 }
150
TEST(SetupFakeServiceManager,GetExistingService)151 TEST(SetupFakeServiceManager, GetExistingService) {
152 setupFakeServiceManager();
153 sp<IBinder> service = getBinder();
154
155 EXPECT_EQ(defaultServiceManager()->addService(String16("foo"), service, false /*allowIsolated*/,
156 IServiceManager::DUMP_FLAG_PRIORITY_DEFAULT), OK);
157
158 EXPECT_EQ(defaultServiceManager()->getService(String16("foo")), service);
159 clearFakeServiceManager();
160 }
161
TEST(ClearFakeServiceManager,GetServiceAfterClear)162 TEST(ClearFakeServiceManager, GetServiceAfterClear) {
163 setupFakeServiceManager();
164
165 sp<IBinder> service = getBinder();
166 EXPECT_EQ(defaultServiceManager()->addService(String16("foo"), service, false /*allowIsolated*/,
167 IServiceManager::DUMP_FLAG_PRIORITY_DEFAULT), OK);
168
169 clearFakeServiceManager();
170 EXPECT_EQ(defaultServiceManager()->getService(String16("foo")), nullptr);
171 }