Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (C) 2020 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 "RpcServer" |
| 18 | |
Steven Moreland | c503204 | 2021-09-30 15:40:27 -0700 | [diff] [blame] | 19 | #include <inttypes.h> |
Steven Moreland | 798e0d1 | 2021-07-14 23:19:25 +0000 | [diff] [blame] | 20 | #include <poll.h> |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 21 | #include <sys/socket.h> |
| 22 | #include <sys/un.h> |
| 23 | |
Steven Moreland | f137de9 | 2021-04-24 01:54:26 +0000 | [diff] [blame] | 24 | #include <thread> |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 25 | #include <vector> |
| 26 | |
Steven Moreland | 826367f | 2021-09-10 14:05:31 -0700 | [diff] [blame] | 27 | #include <android-base/hex.h> |
Steven Moreland | 5802c2b | 2021-05-12 20:13:04 +0000 | [diff] [blame] | 28 | #include <android-base/scopeguard.h> |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 29 | #include <binder/Parcel.h> |
| 30 | #include <binder/RpcServer.h> |
Yifan Hong | 702115c | 2021-06-24 15:39:18 -0700 | [diff] [blame] | 31 | #include <binder/RpcTransportRaw.h> |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 32 | #include <log/log.h> |
Andrei Homescu | 1d935e8 | 2022-04-25 04:58:23 +0000 | [diff] [blame] | 33 | #include <utils/Compat.h> |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 34 | |
Andrei Homescu | ffa3aaa | 2022-04-07 05:06:33 +0000 | [diff] [blame] | 35 | #include "BuildFlags.h" |
Yifan Hong | 8c95042 | 2021-08-05 17:13:55 -0700 | [diff] [blame] | 36 | #include "FdTrigger.h" |
Andrei Homescu | 7c0b79f | 2022-06-30 02:00:46 +0000 | [diff] [blame] | 37 | #include "OS.h" |
Steven Moreland | 611d15f | 2021-05-01 01:28:27 +0000 | [diff] [blame] | 38 | #include "RpcSocketAddress.h" |
Yifan Hong | 1a23585 | 2021-05-13 16:07:47 -0700 | [diff] [blame] | 39 | #include "RpcState.h" |
David Brazdil | 21c887c | 2022-09-23 12:25:18 +0100 | [diff] [blame] | 40 | #include "RpcTransportUtils.h" |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 41 | #include "RpcWireFormat.h" |
Andrei Homescu | c24c879 | 2022-04-19 00:24:51 +0000 | [diff] [blame] | 42 | #include "Utils.h" |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 43 | |
| 44 | namespace android { |
| 45 | |
Steven Moreland | c503204 | 2021-09-30 15:40:27 -0700 | [diff] [blame] | 46 | constexpr size_t kSessionIdBytes = 32; |
| 47 | |
Steven Moreland | 5802c2b | 2021-05-12 20:13:04 +0000 | [diff] [blame] | 48 | using base::ScopeGuard; |
Steven Moreland | 611d15f | 2021-05-01 01:28:27 +0000 | [diff] [blame] | 49 | using base::unique_fd; |
| 50 | |
Yifan Hong | ecf937d | 2021-08-11 17:29:28 -0700 | [diff] [blame] | 51 | RpcServer::RpcServer(std::unique_ptr<RpcTransportCtx> ctx) : mCtx(std::move(ctx)) {} |
Yifan Hong | 436f0e6 | 2021-05-19 15:25:34 -0700 | [diff] [blame] | 52 | RpcServer::~RpcServer() { |
David Brazdil | 4766a1f | 2022-12-19 21:58:25 +0000 | [diff] [blame] | 53 | RpcMutexUniqueLock _l(mLock); |
| 54 | LOG_ALWAYS_FATAL_IF(mShutdownTrigger != nullptr, "Must call shutdown() before destructor"); |
Yifan Hong | 436f0e6 | 2021-05-19 15:25:34 -0700 | [diff] [blame] | 55 | } |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 56 | |
Yifan Hong | 702115c | 2021-06-24 15:39:18 -0700 | [diff] [blame] | 57 | sp<RpcServer> RpcServer::make(std::unique_ptr<RpcTransportCtxFactory> rpcTransportCtxFactory) { |
| 58 | // Default is without TLS. |
| 59 | if (rpcTransportCtxFactory == nullptr) |
Andrei Homescu | 024727b | 2022-08-24 23:54:59 +0000 | [diff] [blame] | 60 | rpcTransportCtxFactory = makeDefaultRpcTransportCtxFactory(); |
Yifan Hong | ecf937d | 2021-08-11 17:29:28 -0700 | [diff] [blame] | 61 | auto ctx = rpcTransportCtxFactory->newServerCtx(); |
| 62 | if (ctx == nullptr) return nullptr; |
| 63 | return sp<RpcServer>::make(std::move(ctx)); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 64 | } |
| 65 | |
David Brazdil | 21c887c | 2022-09-23 12:25:18 +0100 | [diff] [blame] | 66 | status_t RpcServer::setupUnixDomainSocketBootstrapServer(unique_fd bootstrapFd) { |
| 67 | return setupExternalServer(std::move(bootstrapFd), &RpcServer::recvmsgSocketConnection); |
| 68 | } |
| 69 | |
Steven Moreland | 2372f9d | 2021-08-05 15:42:01 -0700 | [diff] [blame] | 70 | status_t RpcServer::setupUnixDomainServer(const char* path) { |
Steven Moreland | 611d15f | 2021-05-01 01:28:27 +0000 | [diff] [blame] | 71 | return setupSocketServer(UnixSocketAddress(path)); |
| 72 | } |
| 73 | |
David Brazdil | a47dfda | 2022-11-22 22:52:19 +0000 | [diff] [blame] | 74 | status_t RpcServer::setupVsockServer(unsigned int bindCid, unsigned int port) { |
| 75 | return setupSocketServer(VsockSocketAddress(bindCid, port)); |
Steven Moreland | 611d15f | 2021-05-01 01:28:27 +0000 | [diff] [blame] | 76 | } |
| 77 | |
Steven Moreland | 2372f9d | 2021-08-05 15:42:01 -0700 | [diff] [blame] | 78 | status_t RpcServer::setupInetServer(const char* address, unsigned int port, |
| 79 | unsigned int* assignedPort) { |
Steven Moreland | 611d15f | 2021-05-01 01:28:27 +0000 | [diff] [blame] | 80 | if (assignedPort != nullptr) *assignedPort = 0; |
Devin Moore | f3b9c4f | 2021-08-03 15:50:13 +0000 | [diff] [blame] | 81 | auto aiStart = InetSocketAddress::getAddrInfo(address, port); |
Steven Moreland | 2372f9d | 2021-08-05 15:42:01 -0700 | [diff] [blame] | 82 | if (aiStart == nullptr) return UNKNOWN_ERROR; |
Steven Moreland | 611d15f | 2021-05-01 01:28:27 +0000 | [diff] [blame] | 83 | for (auto ai = aiStart.get(); ai != nullptr; ai = ai->ai_next) { |
Devin Moore | f3b9c4f | 2021-08-03 15:50:13 +0000 | [diff] [blame] | 84 | InetSocketAddress socketAddress(ai->ai_addr, ai->ai_addrlen, address, port); |
Steven Moreland | 2372f9d | 2021-08-05 15:42:01 -0700 | [diff] [blame] | 85 | if (status_t status = setupSocketServer(socketAddress); status != OK) { |
Steven Moreland | 611d15f | 2021-05-01 01:28:27 +0000 | [diff] [blame] | 86 | continue; |
| 87 | } |
| 88 | |
| 89 | LOG_ALWAYS_FATAL_IF(socketAddress.addr()->sa_family != AF_INET, "expecting inet"); |
| 90 | sockaddr_in addr{}; |
| 91 | socklen_t len = sizeof(addr); |
Pawan | 49d74cb | 2022-08-03 21:19:11 +0000 | [diff] [blame] | 92 | if (0 != getsockname(mServer.fd.get(), reinterpret_cast<sockaddr*>(&addr), &len)) { |
Steven Moreland | 611d15f | 2021-05-01 01:28:27 +0000 | [diff] [blame] | 93 | int savedErrno = errno; |
| 94 | ALOGE("Could not getsockname at %s: %s", socketAddress.toString().c_str(), |
| 95 | strerror(savedErrno)); |
Steven Moreland | 2372f9d | 2021-08-05 15:42:01 -0700 | [diff] [blame] | 96 | return -savedErrno; |
Steven Moreland | 611d15f | 2021-05-01 01:28:27 +0000 | [diff] [blame] | 97 | } |
| 98 | LOG_ALWAYS_FATAL_IF(len != sizeof(addr), "Wrong socket type: len %zu vs len %zu", |
| 99 | static_cast<size_t>(len), sizeof(addr)); |
| 100 | unsigned int realPort = ntohs(addr.sin_port); |
| 101 | LOG_ALWAYS_FATAL_IF(port != 0 && realPort != port, |
| 102 | "Requesting inet server on %s but it is set up on %u.", |
| 103 | socketAddress.toString().c_str(), realPort); |
| 104 | |
| 105 | if (assignedPort != nullptr) { |
| 106 | *assignedPort = realPort; |
| 107 | } |
| 108 | |
Steven Moreland | 2372f9d | 2021-08-05 15:42:01 -0700 | [diff] [blame] | 109 | return OK; |
Steven Moreland | 611d15f | 2021-05-01 01:28:27 +0000 | [diff] [blame] | 110 | } |
Devin Moore | f3b9c4f | 2021-08-03 15:50:13 +0000 | [diff] [blame] | 111 | ALOGE("None of the socket address resolved for %s:%u can be set up as inet server.", address, |
Steven Moreland | 611d15f | 2021-05-01 01:28:27 +0000 | [diff] [blame] | 112 | port); |
Steven Moreland | 2372f9d | 2021-08-05 15:42:01 -0700 | [diff] [blame] | 113 | return UNKNOWN_ERROR; |
Steven Moreland | 611d15f | 2021-05-01 01:28:27 +0000 | [diff] [blame] | 114 | } |
| 115 | |
Steven Moreland | f137de9 | 2021-04-24 01:54:26 +0000 | [diff] [blame] | 116 | void RpcServer::setMaxThreads(size_t threads) { |
| 117 | LOG_ALWAYS_FATAL_IF(threads <= 0, "RpcServer is useless without threads"); |
Yifan Hong | 1a23585 | 2021-05-13 16:07:47 -0700 | [diff] [blame] | 118 | LOG_ALWAYS_FATAL_IF(mJoinThreadRunning, "Cannot set max threads while running"); |
Steven Moreland | f137de9 | 2021-04-24 01:54:26 +0000 | [diff] [blame] | 119 | mMaxThreads = threads; |
| 120 | } |
| 121 | |
| 122 | size_t RpcServer::getMaxThreads() { |
| 123 | return mMaxThreads; |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 124 | } |
| 125 | |
Steven Moreland | bf57bce | 2021-07-26 15:26:12 -0700 | [diff] [blame] | 126 | void RpcServer::setProtocolVersion(uint32_t version) { |
| 127 | mProtocolVersion = version; |
| 128 | } |
| 129 | |
Frederick Mayle | 69a0c99 | 2022-05-26 20:38:39 +0000 | [diff] [blame] | 130 | void RpcServer::setSupportedFileDescriptorTransportModes( |
| 131 | const std::vector<RpcSession::FileDescriptorTransportMode>& modes) { |
| 132 | mSupportedFileDescriptorTransportModes.reset(); |
| 133 | for (RpcSession::FileDescriptorTransportMode mode : modes) { |
| 134 | mSupportedFileDescriptorTransportModes.set(static_cast<size_t>(mode)); |
| 135 | } |
| 136 | } |
| 137 | |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 138 | void RpcServer::setRootObject(const sp<IBinder>& binder) { |
Andrei Homescu | ffa3aaa | 2022-04-07 05:06:33 +0000 | [diff] [blame] | 139 | RpcMutexLockGuard _l(mLock); |
Steven Moreland | 51c44a9 | 2021-10-14 16:50:35 -0700 | [diff] [blame] | 140 | mRootObjectFactory = nullptr; |
Yifan Hong | 4ffb0c7 | 2021-05-07 18:35:14 -0700 | [diff] [blame] | 141 | mRootObjectWeak = mRootObject = binder; |
| 142 | } |
| 143 | |
| 144 | void RpcServer::setRootObjectWeak(const wp<IBinder>& binder) { |
Andrei Homescu | ffa3aaa | 2022-04-07 05:06:33 +0000 | [diff] [blame] | 145 | RpcMutexLockGuard _l(mLock); |
Yifan Hong | 4ffb0c7 | 2021-05-07 18:35:14 -0700 | [diff] [blame] | 146 | mRootObject.clear(); |
Steven Moreland | 51c44a9 | 2021-10-14 16:50:35 -0700 | [diff] [blame] | 147 | mRootObjectFactory = nullptr; |
Yifan Hong | 4ffb0c7 | 2021-05-07 18:35:14 -0700 | [diff] [blame] | 148 | mRootObjectWeak = binder; |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 149 | } |
Steven Moreland | 51c44a9 | 2021-10-14 16:50:35 -0700 | [diff] [blame] | 150 | void RpcServer::setPerSessionRootObject( |
Andrei Homescu | 86124ca | 2022-04-21 22:22:48 +0000 | [diff] [blame] | 151 | std::function<sp<IBinder>(const void*, size_t)>&& makeObject) { |
Andrei Homescu | ffa3aaa | 2022-04-07 05:06:33 +0000 | [diff] [blame] | 152 | RpcMutexLockGuard _l(mLock); |
Steven Moreland | 51c44a9 | 2021-10-14 16:50:35 -0700 | [diff] [blame] | 153 | mRootObject.clear(); |
| 154 | mRootObjectWeak.clear(); |
| 155 | mRootObjectFactory = std::move(makeObject); |
| 156 | } |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 157 | |
David Brazdil | a47dfda | 2022-11-22 22:52:19 +0000 | [diff] [blame] | 158 | void RpcServer::setConnectionFilter(std::function<bool(const void*, size_t)>&& filter) { |
| 159 | RpcMutexLockGuard _l(mLock); |
| 160 | LOG_ALWAYS_FATAL_IF(mShutdownTrigger != nullptr, "Already joined"); |
| 161 | mConnectionFilter = std::move(filter); |
| 162 | } |
| 163 | |
Tomasz Wasilczyk | 7c8dd48 | 2023-03-23 15:30:03 -0700 | [diff] [blame^] | 164 | void RpcServer::setServerSocketModifier(std::function<void(base::borrowed_fd)>&& modifier) { |
| 165 | RpcMutexLockGuard _l(mLock); |
| 166 | LOG_ALWAYS_FATAL_IF(mServer.fd != -1, "Already started"); |
| 167 | mServerSocketModifier = std::move(modifier); |
| 168 | } |
| 169 | |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 170 | sp<IBinder> RpcServer::getRootObject() { |
Andrei Homescu | ffa3aaa | 2022-04-07 05:06:33 +0000 | [diff] [blame] | 171 | RpcMutexLockGuard _l(mLock); |
Yifan Hong | 4ffb0c7 | 2021-05-07 18:35:14 -0700 | [diff] [blame] | 172 | bool hasWeak = mRootObjectWeak.unsafe_get(); |
| 173 | sp<IBinder> ret = mRootObjectWeak.promote(); |
| 174 | ALOGW_IF(hasWeak && ret == nullptr, "RpcServer root object is freed, returning nullptr"); |
| 175 | return ret; |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 176 | } |
| 177 | |
Yifan Hong | 9734cfc | 2021-09-13 16:14:09 -0700 | [diff] [blame] | 178 | std::vector<uint8_t> RpcServer::getCertificate(RpcCertificateFormat format) { |
Andrei Homescu | ffa3aaa | 2022-04-07 05:06:33 +0000 | [diff] [blame] | 179 | RpcMutexLockGuard _l(mLock); |
Yifan Hong | ecf937d | 2021-08-11 17:29:28 -0700 | [diff] [blame] | 180 | return mCtx->getCertificate(format); |
| 181 | } |
| 182 | |
Yifan Hong | 326afd1 | 2021-05-19 15:24:54 -0700 | [diff] [blame] | 183 | static void joinRpcServer(sp<RpcServer>&& thiz) { |
| 184 | thiz->join(); |
| 185 | } |
| 186 | |
| 187 | void RpcServer::start() { |
Andrei Homescu | ffa3aaa | 2022-04-07 05:06:33 +0000 | [diff] [blame] | 188 | RpcMutexLockGuard _l(mLock); |
Yifan Hong | 326afd1 | 2021-05-19 15:24:54 -0700 | [diff] [blame] | 189 | LOG_ALWAYS_FATAL_IF(mJoinThread.get(), "Already started!"); |
Andrei Homescu | ffa3aaa | 2022-04-07 05:06:33 +0000 | [diff] [blame] | 190 | mJoinThread = |
| 191 | std::make_unique<RpcMaybeThread>(&joinRpcServer, sp<RpcServer>::fromExisting(this)); |
| 192 | rpcJoinIfSingleThreaded(*mJoinThread); |
Yifan Hong | 326afd1 | 2021-05-19 15:24:54 -0700 | [diff] [blame] | 193 | } |
| 194 | |
David Brazdil | 21c887c | 2022-09-23 12:25:18 +0100 | [diff] [blame] | 195 | status_t RpcServer::acceptSocketConnection(const RpcServer& server, RpcTransportFd* out) { |
| 196 | RpcTransportFd clientSocket(unique_fd(TEMP_FAILURE_RETRY( |
| 197 | accept4(server.mServer.fd.get(), nullptr, nullptr, SOCK_CLOEXEC | SOCK_NONBLOCK)))); |
| 198 | if (clientSocket.fd < 0) { |
| 199 | int savedErrno = errno; |
| 200 | ALOGE("Could not accept4 socket: %s", strerror(savedErrno)); |
| 201 | return -savedErrno; |
| 202 | } |
| 203 | |
| 204 | *out = std::move(clientSocket); |
| 205 | return OK; |
| 206 | } |
| 207 | |
| 208 | status_t RpcServer::recvmsgSocketConnection(const RpcServer& server, RpcTransportFd* out) { |
| 209 | int zero = 0; |
| 210 | iovec iov{&zero, sizeof(zero)}; |
| 211 | std::vector<std::variant<base::unique_fd, base::borrowed_fd>> fds; |
| 212 | |
David Brazdil | 8a74190 | 2022-12-17 21:41:04 +0000 | [diff] [blame] | 213 | ssize_t num_bytes = receiveMessageFromSocket(server.mServer, &iov, 1, &fds); |
| 214 | if (num_bytes < 0) { |
David Brazdil | 21c887c | 2022-09-23 12:25:18 +0100 | [diff] [blame] | 215 | int savedErrno = errno; |
| 216 | ALOGE("Failed recvmsg: %s", strerror(savedErrno)); |
| 217 | return -savedErrno; |
| 218 | } |
David Brazdil | 8a74190 | 2022-12-17 21:41:04 +0000 | [diff] [blame] | 219 | if (num_bytes == 0) { |
| 220 | return DEAD_OBJECT; |
| 221 | } |
David Brazdil | 21c887c | 2022-09-23 12:25:18 +0100 | [diff] [blame] | 222 | if (fds.size() != 1) { |
| 223 | ALOGE("Expected exactly one fd from recvmsg, got %zu", fds.size()); |
| 224 | return -EINVAL; |
| 225 | } |
| 226 | |
| 227 | unique_fd fd(std::move(std::get<unique_fd>(fds.back()))); |
| 228 | if (auto res = setNonBlocking(fd); !res.ok()) { |
| 229 | ALOGE("Failed setNonBlocking: %s", res.error().message().c_str()); |
| 230 | return res.error().code() == 0 ? UNKNOWN_ERROR : -res.error().code(); |
| 231 | } |
| 232 | |
| 233 | *out = RpcTransportFd(std::move(fd)); |
| 234 | return OK; |
| 235 | } |
| 236 | |
Steven Moreland | 611d15f | 2021-05-01 01:28:27 +0000 | [diff] [blame] | 237 | void RpcServer::join() { |
Yifan Hong | 1a23585 | 2021-05-13 16:07:47 -0700 | [diff] [blame] | 238 | |
| 239 | { |
Andrei Homescu | ffa3aaa | 2022-04-07 05:06:33 +0000 | [diff] [blame] | 240 | RpcMutexLockGuard _l(mLock); |
Pawan | 49d74cb | 2022-08-03 21:19:11 +0000 | [diff] [blame] | 241 | LOG_ALWAYS_FATAL_IF(!mServer.fd.ok(), "RpcServer must be setup to join."); |
David Brazdil | 21c887c | 2022-09-23 12:25:18 +0100 | [diff] [blame] | 242 | LOG_ALWAYS_FATAL_IF(mAcceptFn == nullptr, "RpcServer must have an accept() function"); |
Yifan Hong | 1a23585 | 2021-05-13 16:07:47 -0700 | [diff] [blame] | 243 | LOG_ALWAYS_FATAL_IF(mShutdownTrigger != nullptr, "Already joined"); |
| 244 | mJoinThreadRunning = true; |
Yifan Hong | 8c95042 | 2021-08-05 17:13:55 -0700 | [diff] [blame] | 245 | mShutdownTrigger = FdTrigger::make(); |
Yifan Hong | 1a23585 | 2021-05-13 16:07:47 -0700 | [diff] [blame] | 246 | LOG_ALWAYS_FATAL_IF(mShutdownTrigger == nullptr, "Cannot create join signaler"); |
Steven Moreland | d539fbf | 2021-05-05 23:40:25 +0000 | [diff] [blame] | 247 | } |
Yifan Hong | 1a23585 | 2021-05-13 16:07:47 -0700 | [diff] [blame] | 248 | |
Steven Moreland | 2b4f380 | 2021-05-22 01:46:27 +0000 | [diff] [blame] | 249 | status_t status; |
Steven Moreland | 798e0d1 | 2021-07-14 23:19:25 +0000 | [diff] [blame] | 250 | while ((status = mShutdownTrigger->triggerablePoll(mServer, POLLIN)) == OK) { |
Andrei Homescu | 86124ca | 2022-04-21 22:22:48 +0000 | [diff] [blame] | 251 | std::array<uint8_t, kRpcAddressSize> addr; |
| 252 | static_assert(addr.size() >= sizeof(sockaddr_storage), "kRpcAddressSize is too small"); |
Andrei Homescu | 86124ca | 2022-04-21 22:22:48 +0000 | [diff] [blame] | 253 | socklen_t addrLen = addr.size(); |
Steven Moreland | 51c44a9 | 2021-10-14 16:50:35 -0700 | [diff] [blame] | 254 | |
David Brazdil | 21c887c | 2022-09-23 12:25:18 +0100 | [diff] [blame] | 255 | RpcTransportFd clientSocket; |
David Brazdil | 8a74190 | 2022-12-17 21:41:04 +0000 | [diff] [blame] | 256 | if ((status = mAcceptFn(*this, &clientSocket)) != OK) { |
| 257 | if (status == DEAD_OBJECT) |
| 258 | break; |
| 259 | else |
| 260 | continue; |
Steven Moreland | 410325a | 2021-06-02 18:37:42 +0000 | [diff] [blame] | 261 | } |
David Brazdil | a47dfda | 2022-11-22 22:52:19 +0000 | [diff] [blame] | 262 | |
| 263 | LOG_RPC_DETAIL("accept on fd %d yields fd %d", mServer.fd.get(), clientSocket.fd.get()); |
| 264 | |
David Brazdil | 21c887c | 2022-09-23 12:25:18 +0100 | [diff] [blame] | 265 | if (getpeername(clientSocket.fd.get(), reinterpret_cast<sockaddr*>(addr.data()), |
| 266 | &addrLen)) { |
| 267 | ALOGE("Could not getpeername socket: %s", strerror(errno)); |
| 268 | continue; |
| 269 | } |
| 270 | |
David Brazdil | a47dfda | 2022-11-22 22:52:19 +0000 | [diff] [blame] | 271 | if (mConnectionFilter != nullptr && !mConnectionFilter(addr.data(), addrLen)) { |
| 272 | ALOGE("Dropped client connection fd %d", clientSocket.fd.get()); |
| 273 | continue; |
| 274 | } |
Steven Moreland | 410325a | 2021-06-02 18:37:42 +0000 | [diff] [blame] | 275 | |
| 276 | { |
Andrei Homescu | ffa3aaa | 2022-04-07 05:06:33 +0000 | [diff] [blame] | 277 | RpcMutexLockGuard _l(mLock); |
Andrei Homescu | 74a5445 | 2021-12-10 05:30:21 +0000 | [diff] [blame] | 278 | RpcMaybeThread thread = |
| 279 | RpcMaybeThread(&RpcServer::establishConnection, |
Pawan | 49d74cb | 2022-08-03 21:19:11 +0000 | [diff] [blame] | 280 | sp<RpcServer>::fromExisting(this), std::move(clientSocket), addr, |
Andrei Homescu | 74a5445 | 2021-12-10 05:30:21 +0000 | [diff] [blame] | 281 | addrLen, RpcSession::join); |
Andrei Homescu | ffa3aaa | 2022-04-07 05:06:33 +0000 | [diff] [blame] | 282 | |
| 283 | auto& threadRef = mConnectingThreads[thread.get_id()]; |
| 284 | threadRef = std::move(thread); |
| 285 | rpcJoinIfSingleThreaded(threadRef); |
Steven Moreland | 410325a | 2021-06-02 18:37:42 +0000 | [diff] [blame] | 286 | } |
Yifan Hong | 1a23585 | 2021-05-13 16:07:47 -0700 | [diff] [blame] | 287 | } |
Steven Moreland | 2b4f380 | 2021-05-22 01:46:27 +0000 | [diff] [blame] | 288 | LOG_RPC_DETAIL("RpcServer::join exiting with %s", statusToString(status).c_str()); |
Yifan Hong | 1a23585 | 2021-05-13 16:07:47 -0700 | [diff] [blame] | 289 | |
Andrei Homescu | ffa3aaa | 2022-04-07 05:06:33 +0000 | [diff] [blame] | 290 | if constexpr (kEnableRpcThreads) { |
| 291 | RpcMutexLockGuard _l(mLock); |
Yifan Hong | 1a23585 | 2021-05-13 16:07:47 -0700 | [diff] [blame] | 292 | mJoinThreadRunning = false; |
Andrei Homescu | ffa3aaa | 2022-04-07 05:06:33 +0000 | [diff] [blame] | 293 | } else { |
| 294 | // Multi-threaded builds clear this in shutdown(), but we need it valid |
| 295 | // so the loop above exits cleanly |
| 296 | mShutdownTrigger = nullptr; |
Yifan Hong | 1a23585 | 2021-05-13 16:07:47 -0700 | [diff] [blame] | 297 | } |
| 298 | mShutdownCv.notify_all(); |
Steven Moreland | d539fbf | 2021-05-05 23:40:25 +0000 | [diff] [blame] | 299 | } |
| 300 | |
Yifan Hong | 1a23585 | 2021-05-13 16:07:47 -0700 | [diff] [blame] | 301 | bool RpcServer::shutdown() { |
Andrei Homescu | ffa3aaa | 2022-04-07 05:06:33 +0000 | [diff] [blame] | 302 | RpcMutexUniqueLock _l(mLock); |
Steven Moreland | 9d11b92 | 2021-05-20 01:22:58 +0000 | [diff] [blame] | 303 | if (mShutdownTrigger == nullptr) { |
Devin Moore | 06add17 | 2023-03-22 22:49:29 +0000 | [diff] [blame] | 304 | LOG_RPC_DETAIL("Cannot shutdown. No shutdown trigger installed (already shutdown, or not " |
| 305 | "joined yet?)"); |
Steven Moreland | 9d11b92 | 2021-05-20 01:22:58 +0000 | [diff] [blame] | 306 | return false; |
| 307 | } |
Yifan Hong | 1a23585 | 2021-05-13 16:07:47 -0700 | [diff] [blame] | 308 | |
| 309 | mShutdownTrigger->trigger(); |
Steven Moreland | ab3f442 | 2021-09-27 18:38:20 -0700 | [diff] [blame] | 310 | |
Steven Moreland | a8b4429 | 2021-06-08 01:27:53 +0000 | [diff] [blame] | 311 | for (auto& [id, session] : mSessions) { |
| 312 | (void)id; |
Steven Moreland | ab3f442 | 2021-09-27 18:38:20 -0700 | [diff] [blame] | 313 | // server lock is a more general lock |
Andrei Homescu | ffa3aaa | 2022-04-07 05:06:33 +0000 | [diff] [blame] | 314 | RpcMutexLockGuard _lSession(session->mMutex); |
Steven Moreland | a8b4429 | 2021-06-08 01:27:53 +0000 | [diff] [blame] | 315 | session->mShutdownTrigger->trigger(); |
| 316 | } |
| 317 | |
Andrei Homescu | ffa3aaa | 2022-04-07 05:06:33 +0000 | [diff] [blame] | 318 | if constexpr (!kEnableRpcThreads) { |
| 319 | // In single-threaded mode we're done here, everything else that |
| 320 | // needs to happen should be at the end of RpcServer::join() |
| 321 | return true; |
| 322 | } |
| 323 | |
Steven Moreland | ee3f466 | 2021-05-22 01:07:33 +0000 | [diff] [blame] | 324 | while (mJoinThreadRunning || !mConnectingThreads.empty() || !mSessions.empty()) { |
Steven Moreland | af4ca71 | 2021-05-24 23:22:08 +0000 | [diff] [blame] | 325 | if (std::cv_status::timeout == mShutdownCv.wait_for(_l, std::chrono::seconds(1))) { |
| 326 | ALOGE("Waiting for RpcServer to shut down (1s w/o progress). Join thread running: %d, " |
| 327 | "Connecting threads: " |
| 328 | "%zu, Sessions: %zu. Is your server deadlocked?", |
| 329 | mJoinThreadRunning, mConnectingThreads.size(), mSessions.size()); |
| 330 | } |
Steven Moreland | 9d11b92 | 2021-05-20 01:22:58 +0000 | [diff] [blame] | 331 | } |
Yifan Hong | 1a23585 | 2021-05-13 16:07:47 -0700 | [diff] [blame] | 332 | |
Yifan Hong | 326afd1 | 2021-05-19 15:24:54 -0700 | [diff] [blame] | 333 | // At this point, we know join() is about to exit, but the thread that calls |
| 334 | // join() may not have exited yet. |
| 335 | // If RpcServer owns the join thread (aka start() is called), make sure the thread exits; |
| 336 | // otherwise ~thread() may call std::terminate(), which may crash the process. |
| 337 | // If RpcServer does not own the join thread (aka join() is called directly), |
| 338 | // then the owner of RpcServer is responsible for cleaning up that thread. |
| 339 | if (mJoinThread.get()) { |
| 340 | mJoinThread->join(); |
| 341 | mJoinThread.reset(); |
| 342 | } |
| 343 | |
Tomasz Wasilczyk | 7c8dd48 | 2023-03-23 15:30:03 -0700 | [diff] [blame^] | 344 | mServer = RpcTransportFd(); |
| 345 | |
Steven Moreland | 1c943ec | 2021-07-13 23:57:56 +0000 | [diff] [blame] | 346 | LOG_RPC_DETAIL("Finished waiting on shutdown."); |
| 347 | |
Yifan Hong | 1a23585 | 2021-05-13 16:07:47 -0700 | [diff] [blame] | 348 | mShutdownTrigger = nullptr; |
| 349 | return true; |
| 350 | } |
| 351 | |
Steven Moreland | bdb53ab | 2021-05-05 17:57:41 +0000 | [diff] [blame] | 352 | std::vector<sp<RpcSession>> RpcServer::listSessions() { |
Andrei Homescu | ffa3aaa | 2022-04-07 05:06:33 +0000 | [diff] [blame] | 353 | RpcMutexLockGuard _l(mLock); |
Steven Moreland | bdb53ab | 2021-05-05 17:57:41 +0000 | [diff] [blame] | 354 | std::vector<sp<RpcSession>> sessions; |
| 355 | for (auto& [id, session] : mSessions) { |
Steven Moreland | 736664b | 2021-05-01 04:27:25 +0000 | [diff] [blame] | 356 | (void)id; |
Steven Moreland | bdb53ab | 2021-05-05 17:57:41 +0000 | [diff] [blame] | 357 | sessions.push_back(session); |
Steven Moreland | 736664b | 2021-05-01 04:27:25 +0000 | [diff] [blame] | 358 | } |
Steven Moreland | bdb53ab | 2021-05-05 17:57:41 +0000 | [diff] [blame] | 359 | return sessions; |
Steven Moreland | 611d15f | 2021-05-01 01:28:27 +0000 | [diff] [blame] | 360 | } |
| 361 | |
Steven Moreland | d539fbf | 2021-05-05 23:40:25 +0000 | [diff] [blame] | 362 | size_t RpcServer::numUninitializedSessions() { |
Andrei Homescu | ffa3aaa | 2022-04-07 05:06:33 +0000 | [diff] [blame] | 363 | RpcMutexLockGuard _l(mLock); |
Steven Moreland | d539fbf | 2021-05-05 23:40:25 +0000 | [diff] [blame] | 364 | return mConnectingThreads.size(); |
| 365 | } |
| 366 | |
Andrei Homescu | 74a5445 | 2021-12-10 05:30:21 +0000 | [diff] [blame] | 367 | void RpcServer::establishConnection( |
Pawan | 3e0061c | 2022-08-26 21:08:34 +0000 | [diff] [blame] | 368 | sp<RpcServer>&& server, RpcTransportFd clientFd, std::array<uint8_t, kRpcAddressSize> addr, |
Andrei Homescu | 74a5445 | 2021-12-10 05:30:21 +0000 | [diff] [blame] | 369 | size_t addrLen, |
| 370 | std::function<void(sp<RpcSession>&&, RpcSession::PreJoinSetupResult&&)>&& joinFn) { |
Steven Moreland | 9d11b92 | 2021-05-20 01:22:58 +0000 | [diff] [blame] | 371 | // mShutdownTrigger can only be cleared once connection threads have joined. |
| 372 | // It must be set before this thread is started |
| 373 | LOG_ALWAYS_FATAL_IF(server->mShutdownTrigger == nullptr); |
Yifan Hong | 702115c | 2021-06-24 15:39:18 -0700 | [diff] [blame] | 374 | LOG_ALWAYS_FATAL_IF(server->mCtx == nullptr); |
| 375 | |
| 376 | status_t status = OK; |
| 377 | |
Pawan | 49d74cb | 2022-08-03 21:19:11 +0000 | [diff] [blame] | 378 | int clientFdForLog = clientFd.fd.get(); |
Yifan Hong | f6d4229 | 2021-08-05 23:43:05 -0700 | [diff] [blame] | 379 | auto client = server->mCtx->newTransport(std::move(clientFd), server->mShutdownTrigger.get()); |
Yifan Hong | 702115c | 2021-06-24 15:39:18 -0700 | [diff] [blame] | 380 | if (client == nullptr) { |
| 381 | ALOGE("Dropping accept4()-ed socket because sslAccept fails"); |
| 382 | status = DEAD_OBJECT; |
| 383 | // still need to cleanup before we can return |
| 384 | } else { |
| 385 | LOG_RPC_DETAIL("Created RpcTransport %p for client fd %d", client.get(), clientFdForLog); |
| 386 | } |
Steven Moreland | 9d11b92 | 2021-05-20 01:22:58 +0000 | [diff] [blame] | 387 | |
Steven Moreland | 659416d | 2021-05-11 00:47:50 +0000 | [diff] [blame] | 388 | RpcConnectionHeader header; |
Yifan Hong | 702115c | 2021-06-24 15:39:18 -0700 | [diff] [blame] | 389 | if (status == OK) { |
Andrei Homescu | a39e4ed | 2021-12-10 08:41:54 +0000 | [diff] [blame] | 390 | iovec iov{&header, sizeof(header)}; |
Devin Moore | 695368f | 2022-06-03 22:29:14 +0000 | [diff] [blame] | 391 | status = client->interruptableReadFully(server->mShutdownTrigger.get(), &iov, 1, |
Frederick Mayle | ffe9ac2 | 2022-06-30 02:07:36 +0000 | [diff] [blame] | 392 | std::nullopt, /*ancillaryFds=*/nullptr); |
Yifan Hong | 702115c | 2021-06-24 15:39:18 -0700 | [diff] [blame] | 393 | if (status != OK) { |
| 394 | ALOGE("Failed to read ID for client connecting to RPC server: %s", |
| 395 | statusToString(status).c_str()); |
| 396 | // still need to cleanup before we can return |
| 397 | } |
Steven Moreland | a63ff93 | 2021-05-12 00:03:15 +0000 | [diff] [blame] | 398 | } |
Steven Moreland | bf57bce | 2021-07-26 15:26:12 -0700 | [diff] [blame] | 399 | |
Steven Moreland | 826367f | 2021-09-10 14:05:31 -0700 | [diff] [blame] | 400 | std::vector<uint8_t> sessionId; |
| 401 | if (status == OK) { |
| 402 | if (header.sessionIdSize > 0) { |
Steven Moreland | c503204 | 2021-09-30 15:40:27 -0700 | [diff] [blame] | 403 | if (header.sessionIdSize == kSessionIdBytes) { |
| 404 | sessionId.resize(header.sessionIdSize); |
Andrei Homescu | a39e4ed | 2021-12-10 08:41:54 +0000 | [diff] [blame] | 405 | iovec iov{sessionId.data(), sessionId.size()}; |
Devin Moore | 695368f | 2022-06-03 22:29:14 +0000 | [diff] [blame] | 406 | status = client->interruptableReadFully(server->mShutdownTrigger.get(), &iov, 1, |
Frederick Mayle | ffe9ac2 | 2022-06-30 02:07:36 +0000 | [diff] [blame] | 407 | std::nullopt, /*ancillaryFds=*/nullptr); |
Steven Moreland | c503204 | 2021-09-30 15:40:27 -0700 | [diff] [blame] | 408 | if (status != OK) { |
| 409 | ALOGE("Failed to read session ID for client connecting to RPC server: %s", |
| 410 | statusToString(status).c_str()); |
| 411 | // still need to cleanup before we can return |
| 412 | } |
| 413 | } else { |
| 414 | ALOGE("Malformed session ID. Expecting session ID of size %zu but got %" PRIu16, |
| 415 | kSessionIdBytes, header.sessionIdSize); |
| 416 | status = BAD_VALUE; |
Steven Moreland | 826367f | 2021-09-10 14:05:31 -0700 | [diff] [blame] | 417 | } |
| 418 | } |
| 419 | } |
| 420 | |
Steven Moreland | bf57bce | 2021-07-26 15:26:12 -0700 | [diff] [blame] | 421 | bool incoming = false; |
| 422 | uint32_t protocolVersion = 0; |
Steven Moreland | bf57bce | 2021-07-26 15:26:12 -0700 | [diff] [blame] | 423 | bool requestingNewSession = false; |
| 424 | |
| 425 | if (status == OK) { |
| 426 | incoming = header.options & RPC_CONNECTION_OPTION_INCOMING; |
| 427 | protocolVersion = std::min(header.version, |
| 428 | server->mProtocolVersion.value_or(RPC_WIRE_PROTOCOL_VERSION)); |
Steven Moreland | 826367f | 2021-09-10 14:05:31 -0700 | [diff] [blame] | 429 | requestingNewSession = sessionId.empty(); |
Steven Moreland | bf57bce | 2021-07-26 15:26:12 -0700 | [diff] [blame] | 430 | |
| 431 | if (requestingNewSession) { |
| 432 | RpcNewSessionResponse response{ |
| 433 | .version = protocolVersion, |
| 434 | }; |
| 435 | |
Andrei Homescu | a39e4ed | 2021-12-10 08:41:54 +0000 | [diff] [blame] | 436 | iovec iov{&response, sizeof(response)}; |
Devin Moore | 695368f | 2022-06-03 22:29:14 +0000 | [diff] [blame] | 437 | status = client->interruptableWriteFully(server->mShutdownTrigger.get(), &iov, 1, |
Frederick Mayle | 69a0c99 | 2022-05-26 20:38:39 +0000 | [diff] [blame] | 438 | std::nullopt, nullptr); |
Steven Moreland | bf57bce | 2021-07-26 15:26:12 -0700 | [diff] [blame] | 439 | if (status != OK) { |
| 440 | ALOGE("Failed to send new session response: %s", statusToString(status).c_str()); |
| 441 | // still need to cleanup before we can return |
| 442 | } |
| 443 | } |
| 444 | } |
Steven Moreland | a63ff93 | 2021-05-12 00:03:15 +0000 | [diff] [blame] | 445 | |
Andrei Homescu | ffa3aaa | 2022-04-07 05:06:33 +0000 | [diff] [blame] | 446 | RpcMaybeThread thisThread; |
Steven Moreland | a63ff93 | 2021-05-12 00:03:15 +0000 | [diff] [blame] | 447 | sp<RpcSession> session; |
| 448 | { |
Andrei Homescu | ffa3aaa | 2022-04-07 05:06:33 +0000 | [diff] [blame] | 449 | RpcMutexUniqueLock _l(server->mLock); |
Steven Moreland | a63ff93 | 2021-05-12 00:03:15 +0000 | [diff] [blame] | 450 | |
Andrei Homescu | ffa3aaa | 2022-04-07 05:06:33 +0000 | [diff] [blame] | 451 | auto threadId = server->mConnectingThreads.find(rpc_this_thread::get_id()); |
Yifan Hong | b300550 | 2021-05-19 15:37:00 -0700 | [diff] [blame] | 452 | LOG_ALWAYS_FATAL_IF(threadId == server->mConnectingThreads.end(), |
Steven Moreland | a63ff93 | 2021-05-12 00:03:15 +0000 | [diff] [blame] | 453 | "Must establish connection on owned thread"); |
| 454 | thisThread = std::move(threadId->second); |
Steven Moreland | adc5dca | 2021-05-25 02:06:03 +0000 | [diff] [blame] | 455 | ScopeGuard detachGuard = [&]() { |
| 456 | thisThread.detach(); |
Steven Moreland | 9d11b92 | 2021-05-20 01:22:58 +0000 | [diff] [blame] | 457 | _l.unlock(); |
| 458 | server->mShutdownCv.notify_all(); |
| 459 | }; |
Steven Moreland | adc5dca | 2021-05-25 02:06:03 +0000 | [diff] [blame] | 460 | server->mConnectingThreads.erase(threadId); |
Steven Moreland | 9d11b92 | 2021-05-20 01:22:58 +0000 | [diff] [blame] | 461 | |
Steven Moreland | bf57bce | 2021-07-26 15:26:12 -0700 | [diff] [blame] | 462 | if (status != OK || server->mShutdownTrigger->isTriggered()) { |
Steven Moreland | 5802c2b | 2021-05-12 20:13:04 +0000 | [diff] [blame] | 463 | return; |
| 464 | } |
| 465 | |
Steven Moreland | bf57bce | 2021-07-26 15:26:12 -0700 | [diff] [blame] | 466 | if (requestingNewSession) { |
Steven Moreland | 1b30429 | 2021-07-15 22:59:34 +0000 | [diff] [blame] | 467 | if (incoming) { |
| 468 | ALOGE("Cannot create a new session with an incoming connection, would leak"); |
Steven Moreland | 659416d | 2021-05-11 00:47:50 +0000 | [diff] [blame] | 469 | return; |
| 470 | } |
| 471 | |
Steven Moreland | 826367f | 2021-09-10 14:05:31 -0700 | [diff] [blame] | 472 | // Uniquely identify session at the application layer. Even if a |
| 473 | // client/server use the same certificates, if they create multiple |
| 474 | // sessions, we still want to distinguish between them. |
Steven Moreland | c503204 | 2021-09-30 15:40:27 -0700 | [diff] [blame] | 475 | sessionId.resize(kSessionIdBytes); |
Steven Moreland | 01a6bad | 2021-06-11 00:59:20 +0000 | [diff] [blame] | 476 | size_t tries = 0; |
| 477 | do { |
| 478 | // don't block if there is some entropy issue |
| 479 | if (tries++ > 5) { |
Steven Moreland | 826367f | 2021-09-10 14:05:31 -0700 | [diff] [blame] | 480 | ALOGE("Cannot find new address: %s", |
| 481 | base::HexString(sessionId.data(), sessionId.size()).c_str()); |
Steven Moreland | 01a6bad | 2021-06-11 00:59:20 +0000 | [diff] [blame] | 482 | return; |
| 483 | } |
| 484 | |
Andrei Homescu | c24c879 | 2022-04-19 00:24:51 +0000 | [diff] [blame] | 485 | auto status = getRandomBytes(sessionId.data(), sessionId.size()); |
| 486 | if (status != OK) { |
| 487 | ALOGE("Failed to read random session ID: %s", strerror(-status)); |
Steven Moreland | 826367f | 2021-09-10 14:05:31 -0700 | [diff] [blame] | 488 | return; |
| 489 | } |
Steven Moreland | 01a6bad | 2021-06-11 00:59:20 +0000 | [diff] [blame] | 490 | } while (server->mSessions.end() != server->mSessions.find(sessionId)); |
Steven Moreland | a63ff93 | 2021-05-12 00:03:15 +0000 | [diff] [blame] | 491 | |
Andrei Homescu | 5e30146 | 2022-05-12 00:52:34 +0000 | [diff] [blame] | 492 | session = sp<RpcSession>::make(nullptr); |
Yifan Hong | 1042306 | 2021-10-08 16:26:32 -0700 | [diff] [blame] | 493 | session->setMaxIncomingThreads(server->mMaxThreads); |
Steven Moreland | bf57bce | 2021-07-26 15:26:12 -0700 | [diff] [blame] | 494 | if (!session->setProtocolVersion(protocolVersion)) return; |
Steven Moreland | 51c44a9 | 2021-10-14 16:50:35 -0700 | [diff] [blame] | 495 | |
Frederick Mayle | 26b494e | 2022-07-08 22:09:58 +0000 | [diff] [blame] | 496 | if (header.fileDescriptorTransportMode < |
| 497 | server->mSupportedFileDescriptorTransportModes.size() && |
| 498 | server->mSupportedFileDescriptorTransportModes.test( |
Frederick Mayle | 69a0c99 | 2022-05-26 20:38:39 +0000 | [diff] [blame] | 499 | header.fileDescriptorTransportMode)) { |
| 500 | session->setFileDescriptorTransportMode( |
| 501 | static_cast<RpcSession::FileDescriptorTransportMode>( |
| 502 | header.fileDescriptorTransportMode)); |
| 503 | } else { |
| 504 | ALOGE("Rejecting connection: FileDescriptorTransportMode is not supported: %hhu", |
| 505 | header.fileDescriptorTransportMode); |
| 506 | return; |
| 507 | } |
| 508 | |
Steven Moreland | 51c44a9 | 2021-10-14 16:50:35 -0700 | [diff] [blame] | 509 | // if null, falls back to server root |
| 510 | sp<IBinder> sessionSpecificRoot; |
| 511 | if (server->mRootObjectFactory != nullptr) { |
Andrei Homescu | 86124ca | 2022-04-21 22:22:48 +0000 | [diff] [blame] | 512 | sessionSpecificRoot = server->mRootObjectFactory(addr.data(), addrLen); |
Steven Moreland | 51c44a9 | 2021-10-14 16:50:35 -0700 | [diff] [blame] | 513 | if (sessionSpecificRoot == nullptr) { |
| 514 | ALOGE("Warning: server returned null from root object factory"); |
| 515 | } |
| 516 | } |
| 517 | |
Steven Moreland | a8b4429 | 2021-06-08 01:27:53 +0000 | [diff] [blame] | 518 | if (!session->setForServer(server, |
| 519 | sp<RpcServer::EventListener>::fromExisting( |
| 520 | static_cast<RpcServer::EventListener*>( |
| 521 | server.get())), |
Steven Moreland | 51c44a9 | 2021-10-14 16:50:35 -0700 | [diff] [blame] | 522 | sessionId, sessionSpecificRoot)) { |
Steven Moreland | a8b4429 | 2021-06-08 01:27:53 +0000 | [diff] [blame] | 523 | ALOGE("Failed to attach server to session"); |
| 524 | return; |
| 525 | } |
Steven Moreland | a63ff93 | 2021-05-12 00:03:15 +0000 | [diff] [blame] | 526 | |
Steven Moreland | 01a6bad | 2021-06-11 00:59:20 +0000 | [diff] [blame] | 527 | server->mSessions[sessionId] = session; |
Steven Moreland | a63ff93 | 2021-05-12 00:03:15 +0000 | [diff] [blame] | 528 | } else { |
Steven Moreland | 01a6bad | 2021-06-11 00:59:20 +0000 | [diff] [blame] | 529 | auto it = server->mSessions.find(sessionId); |
Yifan Hong | b300550 | 2021-05-19 15:37:00 -0700 | [diff] [blame] | 530 | if (it == server->mSessions.end()) { |
Steven Moreland | 01a6bad | 2021-06-11 00:59:20 +0000 | [diff] [blame] | 531 | ALOGE("Cannot add thread, no record of session with ID %s", |
Steven Moreland | 826367f | 2021-09-10 14:05:31 -0700 | [diff] [blame] | 532 | base::HexString(sessionId.data(), sessionId.size()).c_str()); |
Steven Moreland | a63ff93 | 2021-05-12 00:03:15 +0000 | [diff] [blame] | 533 | return; |
| 534 | } |
| 535 | session = it->second; |
| 536 | } |
Steven Moreland | 5802c2b | 2021-05-12 20:13:04 +0000 | [diff] [blame] | 537 | |
Steven Moreland | 1b30429 | 2021-07-15 22:59:34 +0000 | [diff] [blame] | 538 | if (incoming) { |
Steven Moreland | 2372f9d | 2021-08-05 15:42:01 -0700 | [diff] [blame] | 539 | LOG_ALWAYS_FATAL_IF(OK != session->addOutgoingConnection(std::move(client), true), |
Steven Moreland | 659416d | 2021-05-11 00:47:50 +0000 | [diff] [blame] | 540 | "server state must already be initialized"); |
| 541 | return; |
| 542 | } |
| 543 | |
Steven Moreland | 5802c2b | 2021-05-12 20:13:04 +0000 | [diff] [blame] | 544 | detachGuard.Disable(); |
Steven Moreland | c88b7fc | 2021-06-10 00:40:39 +0000 | [diff] [blame] | 545 | session->preJoinThreadOwnership(std::move(thisThread)); |
Steven Moreland | a63ff93 | 2021-05-12 00:03:15 +0000 | [diff] [blame] | 546 | } |
| 547 | |
Yifan Hong | 702115c | 2021-06-24 15:39:18 -0700 | [diff] [blame] | 548 | auto setupResult = session->preJoinSetup(std::move(client)); |
Steven Moreland | c88b7fc | 2021-06-10 00:40:39 +0000 | [diff] [blame] | 549 | |
Steven Moreland | a63ff93 | 2021-05-12 00:03:15 +0000 | [diff] [blame] | 550 | // avoid strong cycle |
| 551 | server = nullptr; |
Steven Moreland | a63ff93 | 2021-05-12 00:03:15 +0000 | [diff] [blame] | 552 | |
Andrei Homescu | 74a5445 | 2021-12-10 05:30:21 +0000 | [diff] [blame] | 553 | joinFn(std::move(session), std::move(setupResult)); |
Steven Moreland | a63ff93 | 2021-05-12 00:03:15 +0000 | [diff] [blame] | 554 | } |
| 555 | |
Steven Moreland | 2372f9d | 2021-08-05 15:42:01 -0700 | [diff] [blame] | 556 | status_t RpcServer::setupSocketServer(const RpcSocketAddress& addr) { |
Steven Moreland | 704fc1a | 2021-05-04 23:13:14 +0000 | [diff] [blame] | 557 | LOG_RPC_DETAIL("Setting up socket server %s", addr.toString().c_str()); |
Yifan Hong | 0eb5a67 | 2021-05-12 18:00:25 -0700 | [diff] [blame] | 558 | LOG_ALWAYS_FATAL_IF(hasServer(), "Each RpcServer can only have one server."); |
Steven Moreland | 611d15f | 2021-05-01 01:28:27 +0000 | [diff] [blame] | 559 | |
Alice Wang | 0b138ed | 2022-11-15 16:22:44 +0000 | [diff] [blame] | 560 | unique_fd socket_fd(TEMP_FAILURE_RETRY( |
| 561 | socket(addr.addr()->sa_family, SOCK_STREAM | SOCK_CLOEXEC | SOCK_NONBLOCK, 0))); |
| 562 | if (!socket_fd.ok()) { |
Steven Moreland | 2372f9d | 2021-08-05 15:42:01 -0700 | [diff] [blame] | 563 | int savedErrno = errno; |
Steven Moreland | 8925ebb | 2023-04-10 21:52:17 +0000 | [diff] [blame] | 564 | ALOGE("Could not create socket at %s: %s", addr.toString().c_str(), strerror(savedErrno)); |
Steven Moreland | 2372f9d | 2021-08-05 15:42:01 -0700 | [diff] [blame] | 565 | return -savedErrno; |
Steven Moreland | 611d15f | 2021-05-01 01:28:27 +0000 | [diff] [blame] | 566 | } |
Tomasz Wasilczyk | 7c8dd48 | 2023-03-23 15:30:03 -0700 | [diff] [blame^] | 567 | |
| 568 | { |
| 569 | RpcMutexLockGuard _l(mLock); |
| 570 | if (mServerSocketModifier != nullptr) { |
| 571 | mServerSocketModifier(socket_fd); |
| 572 | } |
| 573 | } |
| 574 | |
Alice Wang | 0b138ed | 2022-11-15 16:22:44 +0000 | [diff] [blame] | 575 | if (0 != TEMP_FAILURE_RETRY(bind(socket_fd.get(), addr.addr(), addr.addrSize()))) { |
Steven Moreland | 611d15f | 2021-05-01 01:28:27 +0000 | [diff] [blame] | 576 | int savedErrno = errno; |
| 577 | ALOGE("Could not bind socket at %s: %s", addr.toString().c_str(), strerror(savedErrno)); |
Steven Moreland | 2372f9d | 2021-08-05 15:42:01 -0700 | [diff] [blame] | 578 | return -savedErrno; |
Steven Moreland | 611d15f | 2021-05-01 01:28:27 +0000 | [diff] [blame] | 579 | } |
| 580 | |
Alice Wang | 0b138ed | 2022-11-15 16:22:44 +0000 | [diff] [blame] | 581 | return setupRawSocketServer(std::move(socket_fd)); |
Steven Moreland | 611d15f | 2021-05-01 01:28:27 +0000 | [diff] [blame] | 582 | } |
| 583 | |
Alice Wang | 0b138ed | 2022-11-15 16:22:44 +0000 | [diff] [blame] | 584 | status_t RpcServer::setupRawSocketServer(unique_fd socket_fd) { |
Alice Wang | 30c204d | 2022-11-15 12:50:51 +0000 | [diff] [blame] | 585 | LOG_ALWAYS_FATAL_IF(!socket_fd.ok(), "Socket must be setup to listen."); |
Alice Wang | 30c204d | 2022-11-15 12:50:51 +0000 | [diff] [blame] | 586 | |
Alice Wang | 893a991 | 2022-10-24 10:44:09 +0000 | [diff] [blame] | 587 | // Right now, we create all threads at once, making accept4 slow. To avoid hanging the client, |
| 588 | // the backlog is increased to a large number. |
| 589 | // TODO(b/189955605): Once we create threads dynamically & lazily, the backlog can be reduced |
| 590 | // to 1. |
Alice Wang | df74178 | 2022-11-16 08:37:04 +0000 | [diff] [blame] | 591 | if (0 != TEMP_FAILURE_RETRY(listen(socket_fd.get(), 50 /*backlog*/))) { |
Alice Wang | 893a991 | 2022-10-24 10:44:09 +0000 | [diff] [blame] | 592 | int savedErrno = errno; |
| 593 | ALOGE("Could not listen initialized Unix socket: %s", strerror(savedErrno)); |
| 594 | return -savedErrno; |
| 595 | } |
Alice Wang | df74178 | 2022-11-16 08:37:04 +0000 | [diff] [blame] | 596 | if (status_t status = setupExternalServer(std::move(socket_fd)); status != OK) { |
Alice Wang | 893a991 | 2022-10-24 10:44:09 +0000 | [diff] [blame] | 597 | ALOGE("Another thread has set up server while calling setupSocketServer. Race?"); |
| 598 | return status; |
| 599 | } |
| 600 | return OK; |
| 601 | } |
| 602 | |
Steven Moreland | dd67b94 | 2021-07-23 17:15:41 -0700 | [diff] [blame] | 603 | void RpcServer::onSessionAllIncomingThreadsEnded(const sp<RpcSession>& session) { |
Steven Moreland | 826367f | 2021-09-10 14:05:31 -0700 | [diff] [blame] | 604 | const std::vector<uint8_t>& id = session->mId; |
| 605 | LOG_ALWAYS_FATAL_IF(id.empty(), "Server sessions must be initialized with ID"); |
| 606 | LOG_RPC_DETAIL("Dropping session with address %s", |
| 607 | base::HexString(id.data(), id.size()).c_str()); |
Steven Moreland | ee78e76 | 2021-05-05 21:12:51 +0000 | [diff] [blame] | 608 | |
Andrei Homescu | ffa3aaa | 2022-04-07 05:06:33 +0000 | [diff] [blame] | 609 | RpcMutexLockGuard _l(mLock); |
Steven Moreland | 826367f | 2021-09-10 14:05:31 -0700 | [diff] [blame] | 610 | auto it = mSessions.find(id); |
Steven Moreland | 01a6bad | 2021-06-11 00:59:20 +0000 | [diff] [blame] | 611 | LOG_ALWAYS_FATAL_IF(it == mSessions.end(), "Bad state, unknown session id %s", |
Steven Moreland | 826367f | 2021-09-10 14:05:31 -0700 | [diff] [blame] | 612 | base::HexString(id.data(), id.size()).c_str()); |
Steven Moreland | 01a6bad | 2021-06-11 00:59:20 +0000 | [diff] [blame] | 613 | LOG_ALWAYS_FATAL_IF(it->second != session, "Bad state, session has id mismatch %s", |
Steven Moreland | 826367f | 2021-09-10 14:05:31 -0700 | [diff] [blame] | 614 | base::HexString(id.data(), id.size()).c_str()); |
Steven Moreland | ee78e76 | 2021-05-05 21:12:51 +0000 | [diff] [blame] | 615 | (void)mSessions.erase(it); |
| 616 | } |
| 617 | |
Steven Moreland | 19fc9f7 | 2021-06-10 03:57:30 +0000 | [diff] [blame] | 618 | void RpcServer::onSessionIncomingThreadEnded() { |
Steven Moreland | ee3f466 | 2021-05-22 01:07:33 +0000 | [diff] [blame] | 619 | mShutdownCv.notify_all(); |
| 620 | } |
| 621 | |
Yifan Hong | 0eb5a67 | 2021-05-12 18:00:25 -0700 | [diff] [blame] | 622 | bool RpcServer::hasServer() { |
Andrei Homescu | ffa3aaa | 2022-04-07 05:06:33 +0000 | [diff] [blame] | 623 | RpcMutexLockGuard _l(mLock); |
Pawan | 49d74cb | 2022-08-03 21:19:11 +0000 | [diff] [blame] | 624 | return mServer.fd.ok(); |
Yifan Hong | 0eb5a67 | 2021-05-12 18:00:25 -0700 | [diff] [blame] | 625 | } |
| 626 | |
Yifan Hong | 00aeb76 | 2021-05-12 17:07:36 -0700 | [diff] [blame] | 627 | unique_fd RpcServer::releaseServer() { |
Andrei Homescu | ffa3aaa | 2022-04-07 05:06:33 +0000 | [diff] [blame] | 628 | RpcMutexLockGuard _l(mLock); |
Pawan | 49d74cb | 2022-08-03 21:19:11 +0000 | [diff] [blame] | 629 | return std::move(mServer.fd); |
Yifan Hong | 00aeb76 | 2021-05-12 17:07:36 -0700 | [diff] [blame] | 630 | } |
| 631 | |
David Brazdil | 21c887c | 2022-09-23 12:25:18 +0100 | [diff] [blame] | 632 | status_t RpcServer::setupExternalServer( |
| 633 | base::unique_fd serverFd, |
| 634 | std::function<status_t(const RpcServer&, RpcTransportFd*)>&& acceptFn) { |
Andrei Homescu | ffa3aaa | 2022-04-07 05:06:33 +0000 | [diff] [blame] | 635 | RpcMutexLockGuard _l(mLock); |
Pawan | 49d74cb | 2022-08-03 21:19:11 +0000 | [diff] [blame] | 636 | if (mServer.fd.ok()) { |
Yifan Hong | 00aeb76 | 2021-05-12 17:07:36 -0700 | [diff] [blame] | 637 | ALOGE("Each RpcServer can only have one server."); |
Steven Moreland | 2372f9d | 2021-08-05 15:42:01 -0700 | [diff] [blame] | 638 | return INVALID_OPERATION; |
Yifan Hong | 00aeb76 | 2021-05-12 17:07:36 -0700 | [diff] [blame] | 639 | } |
| 640 | mServer = std::move(serverFd); |
David Brazdil | 21c887c | 2022-09-23 12:25:18 +0100 | [diff] [blame] | 641 | mAcceptFn = std::move(acceptFn); |
Steven Moreland | 2372f9d | 2021-08-05 15:42:01 -0700 | [diff] [blame] | 642 | return OK; |
Yifan Hong | 00aeb76 | 2021-05-12 17:07:36 -0700 | [diff] [blame] | 643 | } |
| 644 | |
David Brazdil | 21c887c | 2022-09-23 12:25:18 +0100 | [diff] [blame] | 645 | status_t RpcServer::setupExternalServer(base::unique_fd serverFd) { |
| 646 | return setupExternalServer(std::move(serverFd), &RpcServer::acceptSocketConnection); |
| 647 | } |
| 648 | |
Pawan | 1c24f9c | 2022-08-26 23:23:15 +0000 | [diff] [blame] | 649 | bool RpcServer::hasActiveRequests() { |
| 650 | RpcMutexLockGuard _l(mLock); |
| 651 | for (const auto& [_, session] : mSessions) { |
| 652 | if (session->hasActiveRequests()) { |
| 653 | return true; |
| 654 | } |
| 655 | } |
| 656 | return !mServer.isInPollingState(); |
| 657 | } |
| 658 | |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 659 | } // namespace android |