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