1 /*
2 * Copyright (C) 2017 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 #define LOG_TAG "hidl_test_servers"
18
19 #include "hidl_test.h"
20
21 #include <android-base/logging.h>
22
23 #include <android/hardware/tests/baz/1.0/IBaz.h>
24
25 #include <hidl/LegacySupport.h>
26
27 #include <hwbinder/IPCThreadState.h>
28
29 #include <sys/wait.h>
30 #include <signal.h>
31
32 #include <string>
33 #include <utility>
34 #include <vector>
35
36 using ::android::hardware::tests::baz::V1_0::IBaz;
37
38 using ::android::hardware::defaultPassthroughServiceImplementation;
39 using ::android::hardware::IPCThreadState;
40
41 static std::vector<std::pair<std::string, pid_t>> gPidList;
42
signal_handler_server(int signal)43 void signal_handler_server(int signal) {
44 if (signal == SIGTERM) {
45 IPCThreadState::shutdown();
46 exit(0);
47 }
48 }
49
50 template <class T>
51 struct ForkServer {
runForkServer52 static void run(const std::string& serviceName) {
53 pid_t pid;
54
55 if ((pid = fork()) == 0) {
56 // in child process
57 signal(SIGTERM, signal_handler_server);
58 int status = defaultPassthroughServiceImplementation<T>(serviceName);
59 exit(status);
60 }
61
62 gPidList.push_back({serviceName, pid});
63 }
64 };
65
killServer(pid_t pid,const char * serverName)66 static void killServer(pid_t pid, const char *serverName) {
67 if (kill(pid, SIGTERM)) {
68 ALOGE("Could not kill %s; errno = %d", serverName, errno);
69 } else {
70 int status;
71 ALOGE("Waiting for %s to exit...", serverName);
72 waitpid(pid, &status, 0);
73 if (status != 0) {
74 ALOGE("%s terminates abnormally with status %d", serverName, status);
75 } else {
76 ALOGE("%s killed successfully", serverName);
77 }
78 ALOGE("Continuing...");
79 }
80 }
81
signal_handler(int signal)82 void signal_handler(int signal) {
83 if (signal == SIGTERM) {
84 for (const auto& p : gPidList) {
85 killServer(p.second, p.first.c_str());
86 }
87 exit(0);
88 }
89 }
90
main(int,char * [])91 int main(int /* argc */, char* /* argv */ []) {
92 setenv("TREBLE_TESTING_OVERRIDE", "true", true);
93
94 runOnEachServer<ForkServer>();
95
96 ForkServer<IBaz>::run("dyingBaz");
97
98 signal(SIGTERM, signal_handler);
99 // Parent process should not exit before the forked child processes.
100 pause();
101
102 return 0;
103 }
104