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 "RpcState" |
| 18 | |
| 19 | #include "RpcState.h" |
| 20 | |
Steven Moreland | d730207 | 2021-05-15 01:32:04 +0000 | [diff] [blame] | 21 | #include <android-base/scopeguard.h> |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 22 | #include <binder/BpBinder.h> |
Steven Moreland | d730207 | 2021-05-15 01:32:04 +0000 | [diff] [blame] | 23 | #include <binder/IPCThreadState.h> |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 24 | #include <binder/RpcServer.h> |
| 25 | |
| 26 | #include "Debug.h" |
| 27 | #include "RpcWireFormat.h" |
| 28 | |
Steven Moreland | b817679 | 2021-06-22 20:29:21 +0000 | [diff] [blame] | 29 | #include <random> |
| 30 | |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 31 | #include <inttypes.h> |
| 32 | |
| 33 | namespace android { |
| 34 | |
Steven Moreland | d730207 | 2021-05-15 01:32:04 +0000 | [diff] [blame] | 35 | using base::ScopeGuard; |
| 36 | |
Devin Moore | 0825643 | 2021-07-02 13:03:49 -0700 | [diff] [blame] | 37 | #if RPC_FLAKE_PRONE |
Steven Moreland | b817679 | 2021-06-22 20:29:21 +0000 | [diff] [blame] | 38 | void rpcMaybeWaitToFlake() { |
Devin Moore | 0825643 | 2021-07-02 13:03:49 -0700 | [diff] [blame] | 39 | [[clang::no_destroy]] static std::random_device r; |
| 40 | [[clang::no_destroy]] static std::mutex m; |
Steven Moreland | b817679 | 2021-06-22 20:29:21 +0000 | [diff] [blame] | 41 | unsigned num; |
| 42 | { |
| 43 | std::lock_guard<std::mutex> lock(m); |
| 44 | num = r(); |
| 45 | } |
| 46 | if (num % 10 == 0) usleep(num % 1000); |
| 47 | } |
| 48 | #endif |
| 49 | |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 50 | RpcState::RpcState() {} |
| 51 | RpcState::~RpcState() {} |
| 52 | |
Steven Moreland | bdb53ab | 2021-05-05 17:57:41 +0000 | [diff] [blame] | 53 | status_t RpcState::onBinderLeaving(const sp<RpcSession>& session, const sp<IBinder>& binder, |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 54 | RpcAddress* outAddress) { |
| 55 | bool isRemote = binder->remoteBinder(); |
| 56 | bool isRpc = isRemote && binder->remoteBinder()->isRpcBinder(); |
| 57 | |
Steven Moreland | bdb53ab | 2021-05-05 17:57:41 +0000 | [diff] [blame] | 58 | if (isRpc && binder->remoteBinder()->getPrivateAccessorForId().rpcSession() != session) { |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 59 | // We need to be able to send instructions over the socket for how to |
| 60 | // connect to a different server, and we also need to let the host |
| 61 | // process know that this is happening. |
Steven Moreland | bdb53ab | 2021-05-05 17:57:41 +0000 | [diff] [blame] | 62 | ALOGE("Cannot send binder from unrelated binder RPC session."); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 63 | return INVALID_OPERATION; |
| 64 | } |
| 65 | |
| 66 | if (isRemote && !isRpc) { |
| 67 | // Without additional work, this would have the effect of using this |
| 68 | // process to proxy calls from the socket over to the other process, and |
| 69 | // it would make those calls look like they come from us (not over the |
| 70 | // sockets). In order to make this work transparently like binder, we |
| 71 | // would instead need to send instructions over the socket for how to |
| 72 | // connect to the host process, and we also need to let the host process |
| 73 | // know this was happening. |
| 74 | ALOGE("Cannot send binder proxy %p over sockets", binder.get()); |
| 75 | return INVALID_OPERATION; |
| 76 | } |
| 77 | |
| 78 | std::lock_guard<std::mutex> _l(mNodeMutex); |
Steven Moreland | 7227c8a | 2021-06-02 00:24:32 +0000 | [diff] [blame] | 79 | if (mTerminated) return DEAD_OBJECT; |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 80 | |
| 81 | // TODO(b/182939933): maybe move address out of BpBinder, and keep binder->address map |
| 82 | // in RpcState |
| 83 | for (auto& [addr, node] : mNodeForAddress) { |
| 84 | if (binder == node.binder) { |
| 85 | if (isRpc) { |
| 86 | const RpcAddress& actualAddr = |
| 87 | binder->remoteBinder()->getPrivateAccessorForId().rpcAddress(); |
| 88 | // TODO(b/182939933): this is only checking integrity of data structure |
| 89 | // a different data structure doesn't need this |
| 90 | LOG_ALWAYS_FATAL_IF(addr < actualAddr, "Address mismatch"); |
| 91 | LOG_ALWAYS_FATAL_IF(actualAddr < addr, "Address mismatch"); |
| 92 | } |
| 93 | node.timesSent++; |
| 94 | node.sentRef = binder; // might already be set |
| 95 | *outAddress = addr; |
| 96 | return OK; |
| 97 | } |
| 98 | } |
| 99 | LOG_ALWAYS_FATAL_IF(isRpc, "RPC binder must have known address at this point"); |
| 100 | |
Steven Moreland | 9153824 | 2021-06-10 23:35:35 +0000 | [diff] [blame] | 101 | bool forServer = session->server() != nullptr; |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 102 | |
Steven Moreland | 9153824 | 2021-06-10 23:35:35 +0000 | [diff] [blame] | 103 | for (size_t tries = 0; tries < 5; tries++) { |
| 104 | auto&& [it, inserted] = mNodeForAddress.insert({RpcAddress::random(forServer), |
| 105 | BinderNode{ |
| 106 | .binder = binder, |
| 107 | .timesSent = 1, |
| 108 | .sentRef = binder, |
| 109 | }}); |
| 110 | if (inserted) { |
| 111 | *outAddress = it->first; |
| 112 | return OK; |
| 113 | } |
| 114 | |
| 115 | // well, we don't have visibility into the header here, but still |
| 116 | static_assert(sizeof(RpcWireAddress) == 40, "this log needs updating"); |
| 117 | ALOGW("2**256 is 1e77. If you see this log, you probably have some entropy issue, or maybe " |
| 118 | "you witness something incredible!"); |
| 119 | } |
| 120 | |
| 121 | ALOGE("Unable to create an address in order to send out %p", binder.get()); |
| 122 | return WOULD_BLOCK; |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 123 | } |
| 124 | |
Steven Moreland | 7227c8a | 2021-06-02 00:24:32 +0000 | [diff] [blame] | 125 | status_t RpcState::onBinderEntering(const sp<RpcSession>& session, const RpcAddress& address, |
| 126 | sp<IBinder>* out) { |
Steven Moreland | 9153824 | 2021-06-10 23:35:35 +0000 | [diff] [blame] | 127 | // ensure that: if we want to use addresses for something else in the future (for |
| 128 | // instance, allowing transitive binder sends), that we don't accidentally |
| 129 | // send those addresses to old server. Accidentally ignoring this in that |
| 130 | // case and considering the binder to be recognized could cause this |
| 131 | // process to accidentally proxy transactions for that binder. Of course, |
| 132 | // if we communicate with a binder, it could always be proxying |
| 133 | // information. However, we want to make sure that isn't done on accident |
| 134 | // by a client. |
| 135 | if (!address.isRecognizedType()) { |
| 136 | ALOGE("Address is of an unknown type, rejecting: %s", address.toString().c_str()); |
| 137 | return BAD_VALUE; |
| 138 | } |
| 139 | |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 140 | std::unique_lock<std::mutex> _l(mNodeMutex); |
Steven Moreland | 7227c8a | 2021-06-02 00:24:32 +0000 | [diff] [blame] | 141 | if (mTerminated) return DEAD_OBJECT; |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 142 | |
| 143 | if (auto it = mNodeForAddress.find(address); it != mNodeForAddress.end()) { |
Steven Moreland | 7227c8a | 2021-06-02 00:24:32 +0000 | [diff] [blame] | 144 | *out = it->second.binder.promote(); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 145 | |
| 146 | // implicitly have strong RPC refcount, since we received this binder |
| 147 | it->second.timesRecd++; |
| 148 | |
| 149 | _l.unlock(); |
| 150 | |
| 151 | // We have timesRecd RPC refcounts, but we only need to hold on to one |
| 152 | // when we keep the object. All additional dec strongs are sent |
| 153 | // immediately, we wait to send the last one in BpBinder::onLastDecStrong. |
Steven Moreland | bdb53ab | 2021-05-05 17:57:41 +0000 | [diff] [blame] | 154 | (void)session->sendDecStrong(address); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 155 | |
Steven Moreland | 7227c8a | 2021-06-02 00:24:32 +0000 | [diff] [blame] | 156 | return OK; |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 157 | } |
| 158 | |
Steven Moreland | 9153824 | 2021-06-10 23:35:35 +0000 | [diff] [blame] | 159 | // we don't know about this binder, so the other side of the connection |
| 160 | // should have created it. |
| 161 | if (address.isForServer() == !!session->server()) { |
| 162 | ALOGE("Server received unrecognized address which we should own the creation of %s.", |
| 163 | address.toString().c_str()); |
| 164 | return BAD_VALUE; |
| 165 | } |
| 166 | |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 167 | auto&& [it, inserted] = mNodeForAddress.insert({address, BinderNode{}}); |
| 168 | LOG_ALWAYS_FATAL_IF(!inserted, "Failed to insert binder when creating proxy"); |
| 169 | |
Steven Moreland | bdb53ab | 2021-05-05 17:57:41 +0000 | [diff] [blame] | 170 | // Currently, all binders are assumed to be part of the same session (no |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 171 | // device global binders in the RPC world). |
Steven Moreland | 7227c8a | 2021-06-02 00:24:32 +0000 | [diff] [blame] | 172 | it->second.binder = *out = BpBinder::create(session, it->first); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 173 | it->second.timesRecd = 1; |
Steven Moreland | 7227c8a | 2021-06-02 00:24:32 +0000 | [diff] [blame] | 174 | return OK; |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 175 | } |
| 176 | |
| 177 | size_t RpcState::countBinders() { |
| 178 | std::lock_guard<std::mutex> _l(mNodeMutex); |
| 179 | return mNodeForAddress.size(); |
| 180 | } |
| 181 | |
| 182 | void RpcState::dump() { |
| 183 | std::lock_guard<std::mutex> _l(mNodeMutex); |
Steven Moreland | 583a14a | 2021-06-04 02:04:58 +0000 | [diff] [blame] | 184 | dumpLocked(); |
| 185 | } |
| 186 | |
Steven Moreland | c9d7b53 | 2021-06-04 20:57:41 +0000 | [diff] [blame] | 187 | void RpcState::clear() { |
Steven Moreland | 583a14a | 2021-06-04 02:04:58 +0000 | [diff] [blame] | 188 | std::unique_lock<std::mutex> _l(mNodeMutex); |
Steven Moreland | c9d7b53 | 2021-06-04 20:57:41 +0000 | [diff] [blame] | 189 | |
| 190 | if (mTerminated) { |
| 191 | LOG_ALWAYS_FATAL_IF(!mNodeForAddress.empty(), |
| 192 | "New state should be impossible after terminating!"); |
| 193 | return; |
| 194 | } |
| 195 | |
| 196 | if (SHOULD_LOG_RPC_DETAIL) { |
| 197 | ALOGE("RpcState::clear()"); |
| 198 | dumpLocked(); |
| 199 | } |
| 200 | |
| 201 | // if the destructor of a binder object makes another RPC call, then calling |
| 202 | // decStrong could deadlock. So, we must hold onto these binders until |
| 203 | // mNodeMutex is no longer taken. |
| 204 | std::vector<sp<IBinder>> tempHoldBinder; |
| 205 | |
| 206 | mTerminated = true; |
| 207 | for (auto& [address, node] : mNodeForAddress) { |
| 208 | sp<IBinder> binder = node.binder.promote(); |
| 209 | LOG_ALWAYS_FATAL_IF(binder == nullptr, "Binder %p expected to be owned.", binder.get()); |
| 210 | |
| 211 | if (node.sentRef != nullptr) { |
| 212 | tempHoldBinder.push_back(node.sentRef); |
| 213 | } |
| 214 | } |
| 215 | |
| 216 | mNodeForAddress.clear(); |
| 217 | |
| 218 | _l.unlock(); |
| 219 | tempHoldBinder.clear(); // explicit |
Steven Moreland | 583a14a | 2021-06-04 02:04:58 +0000 | [diff] [blame] | 220 | } |
| 221 | |
| 222 | void RpcState::dumpLocked() { |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 223 | ALOGE("DUMP OF RpcState %p", this); |
| 224 | ALOGE("DUMP OF RpcState (%zu nodes)", mNodeForAddress.size()); |
| 225 | for (const auto& [address, node] : mNodeForAddress) { |
| 226 | sp<IBinder> binder = node.binder.promote(); |
| 227 | |
| 228 | const char* desc; |
| 229 | if (binder) { |
| 230 | if (binder->remoteBinder()) { |
| 231 | if (binder->remoteBinder()->isRpcBinder()) { |
| 232 | desc = "(rpc binder proxy)"; |
| 233 | } else { |
| 234 | desc = "(binder proxy)"; |
| 235 | } |
| 236 | } else { |
| 237 | desc = "(local binder)"; |
| 238 | } |
| 239 | } else { |
| 240 | desc = "(null)"; |
| 241 | } |
| 242 | |
| 243 | ALOGE("- BINDER NODE: %p times sent:%zu times recd: %zu a:%s type:%s", |
| 244 | node.binder.unsafe_get(), node.timesSent, node.timesRecd, address.toString().c_str(), |
| 245 | desc); |
| 246 | } |
| 247 | ALOGE("END DUMP OF RpcState"); |
| 248 | } |
| 249 | |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 250 | |
Steven Moreland | dbe7183 | 2021-05-12 23:31:00 +0000 | [diff] [blame] | 251 | RpcState::CommandData::CommandData(size_t size) : mSize(size) { |
| 252 | // The maximum size for regular binder is 1MB for all concurrent |
| 253 | // transactions. A very small proportion of transactions are even |
| 254 | // larger than a page, but we need to avoid allocating too much |
| 255 | // data on behalf of an arbitrary client, or we could risk being in |
| 256 | // a position where a single additional allocation could run out of |
| 257 | // memory. |
| 258 | // |
| 259 | // Note, this limit may not reflect the total amount of data allocated for a |
| 260 | // transaction (in some cases, additional fixed size amounts are added), |
| 261 | // though for rough consistency, we should avoid cases where this data type |
| 262 | // is used for multiple dynamic allocations for a single transaction. |
| 263 | constexpr size_t kMaxTransactionAllocation = 100 * 1000; |
| 264 | if (size == 0) return; |
| 265 | if (size > kMaxTransactionAllocation) { |
| 266 | ALOGW("Transaction requested too much data allocation %zu", size); |
| 267 | return; |
| 268 | } |
| 269 | mData.reset(new (std::nothrow) uint8_t[size]); |
| 270 | } |
| 271 | |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 272 | status_t RpcState::rpcSend(const sp<RpcSession::RpcConnection>& connection, |
| 273 | const sp<RpcSession>& session, const char* what, const void* data, |
| 274 | size_t size) { |
| 275 | LOG_RPC_DETAIL("Sending %s on fd %d: %s", what, connection->fd.get(), |
| 276 | hexString(data, size).c_str()); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 277 | |
| 278 | if (size > std::numeric_limits<ssize_t>::max()) { |
| 279 | ALOGE("Cannot send %s at size %zu (too big)", what, size); |
Steven Moreland | c9d7b53 | 2021-06-04 20:57:41 +0000 | [diff] [blame] | 280 | (void)session->shutdownAndWait(false); |
Steven Moreland | 1e4c2b8 | 2021-05-25 01:51:31 +0000 | [diff] [blame] | 281 | return BAD_VALUE; |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 282 | } |
| 283 | |
Steven Moreland | 798e0d1 | 2021-07-14 23:19:25 +0000 | [diff] [blame] | 284 | if (status_t status = session->mShutdownTrigger->interruptableWriteFully(connection->fd.get(), |
| 285 | data, size); |
| 286 | status != OK) { |
| 287 | LOG_RPC_DETAIL("Failed to write %s (%zu bytes) on fd %d, error: %s", what, size, |
| 288 | connection->fd.get(), statusToString(status).c_str()); |
Steven Moreland | c9d7b53 | 2021-06-04 20:57:41 +0000 | [diff] [blame] | 289 | (void)session->shutdownAndWait(false); |
Steven Moreland | 798e0d1 | 2021-07-14 23:19:25 +0000 | [diff] [blame] | 290 | return status; |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 291 | } |
| 292 | |
Steven Moreland | 1e4c2b8 | 2021-05-25 01:51:31 +0000 | [diff] [blame] | 293 | return OK; |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 294 | } |
| 295 | |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 296 | status_t RpcState::rpcRec(const sp<RpcSession::RpcConnection>& connection, |
| 297 | const sp<RpcSession>& session, const char* what, void* data, |
| 298 | size_t size) { |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 299 | if (size > std::numeric_limits<ssize_t>::max()) { |
| 300 | ALOGE("Cannot rec %s at size %zu (too big)", what, size); |
Steven Moreland | c9d7b53 | 2021-06-04 20:57:41 +0000 | [diff] [blame] | 301 | (void)session->shutdownAndWait(false); |
Steven Moreland | 1e4c2b8 | 2021-05-25 01:51:31 +0000 | [diff] [blame] | 302 | return BAD_VALUE; |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 303 | } |
| 304 | |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 305 | if (status_t status = |
| 306 | session->mShutdownTrigger->interruptableReadFully(connection->fd.get(), data, size); |
Steven Moreland | ee3f466 | 2021-05-22 01:07:33 +0000 | [diff] [blame] | 307 | status != OK) { |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 308 | LOG_RPC_DETAIL("Failed to read %s (%zu bytes) on fd %d, error: %s", what, size, |
| 309 | connection->fd.get(), statusToString(status).c_str()); |
Steven Moreland | 1e4c2b8 | 2021-05-25 01:51:31 +0000 | [diff] [blame] | 310 | return status; |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 311 | } |
| 312 | |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 313 | LOG_RPC_DETAIL("Received %s on fd %d: %s", what, connection->fd.get(), |
| 314 | hexString(data, size).c_str()); |
Steven Moreland | 1e4c2b8 | 2021-05-25 01:51:31 +0000 | [diff] [blame] | 315 | return OK; |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 316 | } |
| 317 | |
Steven Moreland | bf57bce | 2021-07-26 15:26:12 -0700 | [diff] [blame^] | 318 | status_t RpcState::readNewSessionResponse(const sp<RpcSession::RpcConnection>& connection, |
| 319 | const sp<RpcSession>& session, uint32_t* version) { |
| 320 | RpcNewSessionResponse response; |
| 321 | if (status_t status = |
| 322 | rpcRec(connection, session, "new session response", &response, sizeof(response)); |
| 323 | status != OK) { |
| 324 | return status; |
| 325 | } |
| 326 | *version = response.version; |
| 327 | return OK; |
| 328 | } |
| 329 | |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 330 | status_t RpcState::sendConnectionInit(const sp<RpcSession::RpcConnection>& connection, |
| 331 | const sp<RpcSession>& session) { |
Steven Moreland | 19fc9f7 | 2021-06-10 03:57:30 +0000 | [diff] [blame] | 332 | RpcOutgoingConnectionInit init{ |
Steven Moreland | c88b7fc | 2021-06-10 00:40:39 +0000 | [diff] [blame] | 333 | .msg = RPC_CONNECTION_INIT_OKAY, |
| 334 | }; |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 335 | return rpcSend(connection, session, "connection init", &init, sizeof(init)); |
Steven Moreland | c88b7fc | 2021-06-10 00:40:39 +0000 | [diff] [blame] | 336 | } |
| 337 | |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 338 | status_t RpcState::readConnectionInit(const sp<RpcSession::RpcConnection>& connection, |
| 339 | const sp<RpcSession>& session) { |
Steven Moreland | 19fc9f7 | 2021-06-10 03:57:30 +0000 | [diff] [blame] | 340 | RpcOutgoingConnectionInit init; |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 341 | if (status_t status = rpcRec(connection, session, "connection init", &init, sizeof(init)); |
| 342 | status != OK) |
Steven Moreland | c88b7fc | 2021-06-10 00:40:39 +0000 | [diff] [blame] | 343 | return status; |
| 344 | |
| 345 | static_assert(sizeof(init.msg) == sizeof(RPC_CONNECTION_INIT_OKAY)); |
| 346 | if (0 != strncmp(init.msg, RPC_CONNECTION_INIT_OKAY, sizeof(init.msg))) { |
| 347 | ALOGE("Connection init message unrecognized %.*s", static_cast<int>(sizeof(init.msg)), |
| 348 | init.msg); |
| 349 | return BAD_VALUE; |
| 350 | } |
| 351 | return OK; |
| 352 | } |
| 353 | |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 354 | sp<IBinder> RpcState::getRootObject(const sp<RpcSession::RpcConnection>& connection, |
| 355 | const sp<RpcSession>& session) { |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 356 | Parcel data; |
Steven Moreland | bdb53ab | 2021-05-05 17:57:41 +0000 | [diff] [blame] | 357 | data.markForRpc(session); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 358 | Parcel reply; |
| 359 | |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 360 | status_t status = transactAddress(connection, RpcAddress::zero(), RPC_SPECIAL_TRANSACT_GET_ROOT, |
| 361 | data, session, &reply, 0); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 362 | if (status != OK) { |
| 363 | ALOGE("Error getting root object: %s", statusToString(status).c_str()); |
| 364 | return nullptr; |
| 365 | } |
| 366 | |
| 367 | return reply.readStrongBinder(); |
| 368 | } |
| 369 | |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 370 | status_t RpcState::getMaxThreads(const sp<RpcSession::RpcConnection>& connection, |
| 371 | const sp<RpcSession>& session, size_t* maxThreadsOut) { |
Steven Moreland | f137de9 | 2021-04-24 01:54:26 +0000 | [diff] [blame] | 372 | Parcel data; |
Steven Moreland | bdb53ab | 2021-05-05 17:57:41 +0000 | [diff] [blame] | 373 | data.markForRpc(session); |
Steven Moreland | f137de9 | 2021-04-24 01:54:26 +0000 | [diff] [blame] | 374 | Parcel reply; |
| 375 | |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 376 | status_t status = |
| 377 | transactAddress(connection, RpcAddress::zero(), RPC_SPECIAL_TRANSACT_GET_MAX_THREADS, |
| 378 | data, session, &reply, 0); |
Steven Moreland | f137de9 | 2021-04-24 01:54:26 +0000 | [diff] [blame] | 379 | if (status != OK) { |
| 380 | ALOGE("Error getting max threads: %s", statusToString(status).c_str()); |
| 381 | return status; |
| 382 | } |
| 383 | |
Steven Moreland | 7c5e6c2 | 2021-05-01 02:55:20 +0000 | [diff] [blame] | 384 | int32_t maxThreads; |
| 385 | status = reply.readInt32(&maxThreads); |
Steven Moreland | f137de9 | 2021-04-24 01:54:26 +0000 | [diff] [blame] | 386 | if (status != OK) return status; |
Steven Moreland | 7c5e6c2 | 2021-05-01 02:55:20 +0000 | [diff] [blame] | 387 | if (maxThreads <= 0) { |
| 388 | ALOGE("Error invalid max maxThreads: %d", maxThreads); |
Steven Moreland | f137de9 | 2021-04-24 01:54:26 +0000 | [diff] [blame] | 389 | return BAD_VALUE; |
| 390 | } |
| 391 | |
Steven Moreland | 7c5e6c2 | 2021-05-01 02:55:20 +0000 | [diff] [blame] | 392 | *maxThreadsOut = maxThreads; |
| 393 | return OK; |
| 394 | } |
| 395 | |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 396 | status_t RpcState::getSessionId(const sp<RpcSession::RpcConnection>& connection, |
Steven Moreland | 01a6bad | 2021-06-11 00:59:20 +0000 | [diff] [blame] | 397 | const sp<RpcSession>& session, RpcAddress* sessionIdOut) { |
Steven Moreland | 7c5e6c2 | 2021-05-01 02:55:20 +0000 | [diff] [blame] | 398 | Parcel data; |
Steven Moreland | bdb53ab | 2021-05-05 17:57:41 +0000 | [diff] [blame] | 399 | data.markForRpc(session); |
Steven Moreland | 7c5e6c2 | 2021-05-01 02:55:20 +0000 | [diff] [blame] | 400 | Parcel reply; |
| 401 | |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 402 | status_t status = |
| 403 | transactAddress(connection, RpcAddress::zero(), RPC_SPECIAL_TRANSACT_GET_SESSION_ID, |
| 404 | data, session, &reply, 0); |
Steven Moreland | 7c5e6c2 | 2021-05-01 02:55:20 +0000 | [diff] [blame] | 405 | if (status != OK) { |
Steven Moreland | bdb53ab | 2021-05-05 17:57:41 +0000 | [diff] [blame] | 406 | ALOGE("Error getting session ID: %s", statusToString(status).c_str()); |
Steven Moreland | 7c5e6c2 | 2021-05-01 02:55:20 +0000 | [diff] [blame] | 407 | return status; |
| 408 | } |
| 409 | |
Steven Moreland | 01a6bad | 2021-06-11 00:59:20 +0000 | [diff] [blame] | 410 | return sessionIdOut->readFromParcel(reply); |
Steven Moreland | f137de9 | 2021-04-24 01:54:26 +0000 | [diff] [blame] | 411 | } |
| 412 | |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 413 | status_t RpcState::transact(const sp<RpcSession::RpcConnection>& connection, |
| 414 | const sp<IBinder>& binder, uint32_t code, const Parcel& data, |
| 415 | const sp<RpcSession>& session, Parcel* reply, uint32_t flags) { |
Steven Moreland | f517427 | 2021-05-25 00:39:28 +0000 | [diff] [blame] | 416 | if (!data.isForRpc()) { |
| 417 | ALOGE("Refusing to send RPC with parcel not crafted for RPC"); |
| 418 | return BAD_TYPE; |
| 419 | } |
| 420 | |
| 421 | if (data.objectsCount() != 0) { |
| 422 | ALOGE("Parcel at %p has attached objects but is being used in an RPC call", &data); |
| 423 | return BAD_TYPE; |
| 424 | } |
| 425 | |
| 426 | RpcAddress address = RpcAddress::zero(); |
| 427 | if (status_t status = onBinderLeaving(session, binder, &address); status != OK) return status; |
| 428 | |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 429 | return transactAddress(connection, address, code, data, session, reply, flags); |
Steven Moreland | f517427 | 2021-05-25 00:39:28 +0000 | [diff] [blame] | 430 | } |
| 431 | |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 432 | status_t RpcState::transactAddress(const sp<RpcSession::RpcConnection>& connection, |
| 433 | const RpcAddress& address, uint32_t code, const Parcel& data, |
| 434 | const sp<RpcSession>& session, Parcel* reply, uint32_t flags) { |
Steven Moreland | f517427 | 2021-05-25 00:39:28 +0000 | [diff] [blame] | 435 | LOG_ALWAYS_FATAL_IF(!data.isForRpc()); |
| 436 | LOG_ALWAYS_FATAL_IF(data.objectsCount() != 0); |
| 437 | |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 438 | uint64_t asyncNumber = 0; |
| 439 | |
| 440 | if (!address.isZero()) { |
Steven Moreland | 583a14a | 2021-06-04 02:04:58 +0000 | [diff] [blame] | 441 | std::unique_lock<std::mutex> _l(mNodeMutex); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 442 | if (mTerminated) return DEAD_OBJECT; // avoid fatal only, otherwise races |
| 443 | auto it = mNodeForAddress.find(address); |
| 444 | LOG_ALWAYS_FATAL_IF(it == mNodeForAddress.end(), "Sending transact on unknown address %s", |
| 445 | address.toString().c_str()); |
| 446 | |
| 447 | if (flags & IBinder::FLAG_ONEWAY) { |
Steven Moreland | 583a14a | 2021-06-04 02:04:58 +0000 | [diff] [blame] | 448 | asyncNumber = it->second.asyncNumber; |
Steven Moreland | c9d7b53 | 2021-06-04 20:57:41 +0000 | [diff] [blame] | 449 | if (!nodeProgressAsyncNumber(&it->second)) { |
| 450 | _l.unlock(); |
| 451 | (void)session->shutdownAndWait(false); |
| 452 | return DEAD_OBJECT; |
| 453 | } |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 454 | } |
| 455 | } |
| 456 | |
Steven Moreland | 77c3011 | 2021-06-02 20:45:46 +0000 | [diff] [blame] | 457 | LOG_ALWAYS_FATAL_IF(std::numeric_limits<int32_t>::max() - sizeof(RpcWireHeader) - |
| 458 | sizeof(RpcWireTransaction) < |
| 459 | data.dataSize(), |
| 460 | "Too much data %zu", data.dataSize()); |
| 461 | |
| 462 | RpcWireHeader command{ |
| 463 | .command = RPC_COMMAND_TRANSACT, |
| 464 | .bodySize = static_cast<uint32_t>(sizeof(RpcWireTransaction) + data.dataSize()), |
| 465 | }; |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 466 | RpcWireTransaction transaction{ |
| 467 | .address = address.viewRawEmbedded(), |
| 468 | .code = code, |
| 469 | .flags = flags, |
| 470 | .asyncNumber = asyncNumber, |
| 471 | }; |
Steven Moreland | 77c3011 | 2021-06-02 20:45:46 +0000 | [diff] [blame] | 472 | CommandData transactionData(sizeof(RpcWireHeader) + sizeof(RpcWireTransaction) + |
| 473 | data.dataSize()); |
Steven Moreland | e839334 | 2021-05-05 23:27:53 +0000 | [diff] [blame] | 474 | if (!transactionData.valid()) { |
| 475 | return NO_MEMORY; |
| 476 | } |
| 477 | |
Steven Moreland | 77c3011 | 2021-06-02 20:45:46 +0000 | [diff] [blame] | 478 | memcpy(transactionData.data() + 0, &command, sizeof(RpcWireHeader)); |
| 479 | memcpy(transactionData.data() + sizeof(RpcWireHeader), &transaction, |
| 480 | sizeof(RpcWireTransaction)); |
| 481 | memcpy(transactionData.data() + sizeof(RpcWireHeader) + sizeof(RpcWireTransaction), data.data(), |
| 482 | data.dataSize()); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 483 | |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 484 | if (status_t status = rpcSend(connection, session, "transaction", transactionData.data(), |
| 485 | transactionData.size()); |
Steven Moreland | 1e4c2b8 | 2021-05-25 01:51:31 +0000 | [diff] [blame] | 486 | status != OK) |
Steven Moreland | a5036f0 | 2021-06-08 02:26:57 +0000 | [diff] [blame] | 487 | // TODO(b/167966510): need to undo onBinderLeaving - we know the |
| 488 | // refcount isn't successfully transferred. |
Steven Moreland | 1e4c2b8 | 2021-05-25 01:51:31 +0000 | [diff] [blame] | 489 | return status; |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 490 | |
| 491 | if (flags & IBinder::FLAG_ONEWAY) { |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 492 | LOG_RPC_DETAIL("Oneway command, so no longer waiting on %d", connection->fd.get()); |
Steven Moreland | 52eee94 | 2021-06-03 00:59:28 +0000 | [diff] [blame] | 493 | |
| 494 | // Do not wait on result. |
| 495 | // However, too many oneway calls may cause refcounts to build up and fill up the socket, |
| 496 | // so process those. |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 497 | return drainCommands(connection, session, CommandType::CONTROL_ONLY); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 498 | } |
| 499 | |
| 500 | LOG_ALWAYS_FATAL_IF(reply == nullptr, "Reply parcel must be used for synchronous transaction."); |
| 501 | |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 502 | return waitForReply(connection, session, reply); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 503 | } |
| 504 | |
Steven Moreland | 438cce8 | 2021-04-02 18:04:08 +0000 | [diff] [blame] | 505 | static void cleanup_reply_data(Parcel* p, const uint8_t* data, size_t dataSize, |
| 506 | const binder_size_t* objects, size_t objectsCount) { |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 507 | (void)p; |
| 508 | delete[] const_cast<uint8_t*>(data - offsetof(RpcWireReply, data)); |
| 509 | (void)dataSize; |
| 510 | LOG_ALWAYS_FATAL_IF(objects != nullptr); |
Yifan Hong | 239a2ca | 2021-06-24 16:05:16 -0700 | [diff] [blame] | 511 | LOG_ALWAYS_FATAL_IF(objectsCount != 0, "%zu objects remaining", objectsCount); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 512 | } |
| 513 | |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 514 | status_t RpcState::waitForReply(const sp<RpcSession::RpcConnection>& connection, |
| 515 | const sp<RpcSession>& session, Parcel* reply) { |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 516 | RpcWireHeader command; |
| 517 | while (true) { |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 518 | if (status_t status = |
| 519 | rpcRec(connection, session, "command header", &command, sizeof(command)); |
Steven Moreland | 1e4c2b8 | 2021-05-25 01:51:31 +0000 | [diff] [blame] | 520 | status != OK) |
| 521 | return status; |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 522 | |
| 523 | if (command.command == RPC_COMMAND_REPLY) break; |
| 524 | |
Steven Moreland | 19fc9f7 | 2021-06-10 03:57:30 +0000 | [diff] [blame] | 525 | if (status_t status = processCommand(connection, session, command, CommandType::ANY); |
Steven Moreland | 52eee94 | 2021-06-03 00:59:28 +0000 | [diff] [blame] | 526 | status != OK) |
Steven Moreland | 1e4c2b8 | 2021-05-25 01:51:31 +0000 | [diff] [blame] | 527 | return status; |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 528 | } |
| 529 | |
Steven Moreland | dbe7183 | 2021-05-12 23:31:00 +0000 | [diff] [blame] | 530 | CommandData data(command.bodySize); |
Steven Moreland | 1e4c2b8 | 2021-05-25 01:51:31 +0000 | [diff] [blame] | 531 | if (!data.valid()) return NO_MEMORY; |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 532 | |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 533 | if (status_t status = rpcRec(connection, session, "reply body", data.data(), command.bodySize); |
Steven Moreland | 1e4c2b8 | 2021-05-25 01:51:31 +0000 | [diff] [blame] | 534 | status != OK) |
| 535 | return status; |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 536 | |
| 537 | if (command.bodySize < sizeof(RpcWireReply)) { |
| 538 | ALOGE("Expecting %zu but got %" PRId32 " bytes for RpcWireReply. Terminating!", |
| 539 | sizeof(RpcWireReply), command.bodySize); |
Steven Moreland | c9d7b53 | 2021-06-04 20:57:41 +0000 | [diff] [blame] | 540 | (void)session->shutdownAndWait(false); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 541 | return BAD_VALUE; |
| 542 | } |
Steven Moreland | e839334 | 2021-05-05 23:27:53 +0000 | [diff] [blame] | 543 | RpcWireReply* rpcReply = reinterpret_cast<RpcWireReply*>(data.data()); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 544 | if (rpcReply->status != OK) return rpcReply->status; |
| 545 | |
Steven Moreland | e839334 | 2021-05-05 23:27:53 +0000 | [diff] [blame] | 546 | data.release(); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 547 | reply->ipcSetDataReference(rpcReply->data, command.bodySize - offsetof(RpcWireReply, data), |
Steven Moreland | 438cce8 | 2021-04-02 18:04:08 +0000 | [diff] [blame] | 548 | nullptr, 0, cleanup_reply_data); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 549 | |
Steven Moreland | bdb53ab | 2021-05-05 17:57:41 +0000 | [diff] [blame] | 550 | reply->markForRpc(session); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 551 | |
| 552 | return OK; |
| 553 | } |
| 554 | |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 555 | status_t RpcState::sendDecStrong(const sp<RpcSession::RpcConnection>& connection, |
| 556 | const sp<RpcSession>& session, const RpcAddress& addr) { |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 557 | { |
| 558 | std::lock_guard<std::mutex> _l(mNodeMutex); |
| 559 | if (mTerminated) return DEAD_OBJECT; // avoid fatal only, otherwise races |
| 560 | auto it = mNodeForAddress.find(addr); |
| 561 | LOG_ALWAYS_FATAL_IF(it == mNodeForAddress.end(), "Sending dec strong on unknown address %s", |
| 562 | addr.toString().c_str()); |
| 563 | LOG_ALWAYS_FATAL_IF(it->second.timesRecd <= 0, "Bad dec strong %s", |
| 564 | addr.toString().c_str()); |
| 565 | |
| 566 | it->second.timesRecd--; |
Steven Moreland | 31bde7a | 2021-06-04 00:57:36 +0000 | [diff] [blame] | 567 | LOG_ALWAYS_FATAL_IF(nullptr != tryEraseNode(it), |
| 568 | "Bad state. RpcState shouldn't own received binder"); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 569 | } |
| 570 | |
| 571 | RpcWireHeader cmd = { |
| 572 | .command = RPC_COMMAND_DEC_STRONG, |
| 573 | .bodySize = sizeof(RpcWireAddress), |
| 574 | }; |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 575 | if (status_t status = rpcSend(connection, session, "dec ref header", &cmd, sizeof(cmd)); |
| 576 | status != OK) |
Steven Moreland | 1e4c2b8 | 2021-05-25 01:51:31 +0000 | [diff] [blame] | 577 | return status; |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 578 | if (status_t status = rpcSend(connection, session, "dec ref body", &addr.viewRawEmbedded(), |
Steven Moreland | c9d7b53 | 2021-06-04 20:57:41 +0000 | [diff] [blame] | 579 | sizeof(RpcWireAddress)); |
Steven Moreland | 1e4c2b8 | 2021-05-25 01:51:31 +0000 | [diff] [blame] | 580 | status != OK) |
| 581 | return status; |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 582 | return OK; |
| 583 | } |
| 584 | |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 585 | status_t RpcState::getAndExecuteCommand(const sp<RpcSession::RpcConnection>& connection, |
| 586 | const sp<RpcSession>& session, CommandType type) { |
| 587 | LOG_RPC_DETAIL("getAndExecuteCommand on fd %d", connection->fd.get()); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 588 | |
| 589 | RpcWireHeader command; |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 590 | if (status_t status = rpcRec(connection, session, "command header", &command, sizeof(command)); |
Steven Moreland | 1e4c2b8 | 2021-05-25 01:51:31 +0000 | [diff] [blame] | 591 | status != OK) |
| 592 | return status; |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 593 | |
Steven Moreland | 19fc9f7 | 2021-06-10 03:57:30 +0000 | [diff] [blame] | 594 | return processCommand(connection, session, command, type); |
Steven Moreland | 52eee94 | 2021-06-03 00:59:28 +0000 | [diff] [blame] | 595 | } |
| 596 | |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 597 | status_t RpcState::drainCommands(const sp<RpcSession::RpcConnection>& connection, |
| 598 | const sp<RpcSession>& session, CommandType type) { |
Steven Moreland | 52eee94 | 2021-06-03 00:59:28 +0000 | [diff] [blame] | 599 | uint8_t buf; |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 600 | while (0 < TEMP_FAILURE_RETRY( |
| 601 | recv(connection->fd.get(), &buf, sizeof(buf), MSG_PEEK | MSG_DONTWAIT))) { |
| 602 | status_t status = getAndExecuteCommand(connection, session, type); |
Steven Moreland | 52eee94 | 2021-06-03 00:59:28 +0000 | [diff] [blame] | 603 | if (status != OK) return status; |
| 604 | } |
| 605 | return OK; |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 606 | } |
| 607 | |
Steven Moreland | 19fc9f7 | 2021-06-10 03:57:30 +0000 | [diff] [blame] | 608 | status_t RpcState::processCommand(const sp<RpcSession::RpcConnection>& connection, |
| 609 | const sp<RpcSession>& session, const RpcWireHeader& command, |
| 610 | CommandType type) { |
Steven Moreland | d730207 | 2021-05-15 01:32:04 +0000 | [diff] [blame] | 611 | IPCThreadState* kernelBinderState = IPCThreadState::selfOrNull(); |
| 612 | IPCThreadState::SpGuard spGuard{ |
| 613 | .address = __builtin_frame_address(0), |
| 614 | .context = "processing binder RPC command", |
| 615 | }; |
| 616 | const IPCThreadState::SpGuard* origGuard; |
| 617 | if (kernelBinderState != nullptr) { |
| 618 | origGuard = kernelBinderState->pushGetCallingSpGuard(&spGuard); |
| 619 | } |
| 620 | ScopeGuard guardUnguard = [&]() { |
| 621 | if (kernelBinderState != nullptr) { |
| 622 | kernelBinderState->restoreGetCallingSpGuard(origGuard); |
| 623 | } |
| 624 | }; |
| 625 | |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 626 | switch (command.command) { |
| 627 | case RPC_COMMAND_TRANSACT: |
Steven Moreland | 52eee94 | 2021-06-03 00:59:28 +0000 | [diff] [blame] | 628 | if (type != CommandType::ANY) return BAD_TYPE; |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 629 | return processTransact(connection, session, command); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 630 | case RPC_COMMAND_DEC_STRONG: |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 631 | return processDecStrong(connection, session, command); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 632 | } |
| 633 | |
| 634 | // We should always know the version of the opposing side, and since the |
| 635 | // RPC-binder-level wire protocol is not self synchronizing, we have no way |
| 636 | // to understand where the current command ends and the next one begins. We |
| 637 | // also can't consider it a fatal error because this would allow any client |
Steven Moreland | bdb53ab | 2021-05-05 17:57:41 +0000 | [diff] [blame] | 638 | // to kill us, so ending the session for misbehaving client. |
| 639 | ALOGE("Unknown RPC command %d - terminating session", command.command); |
Steven Moreland | c9d7b53 | 2021-06-04 20:57:41 +0000 | [diff] [blame] | 640 | (void)session->shutdownAndWait(false); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 641 | return DEAD_OBJECT; |
| 642 | } |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 643 | status_t RpcState::processTransact(const sp<RpcSession::RpcConnection>& connection, |
| 644 | const sp<RpcSession>& session, const RpcWireHeader& command) { |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 645 | LOG_ALWAYS_FATAL_IF(command.command != RPC_COMMAND_TRANSACT, "command: %d", command.command); |
| 646 | |
Steven Moreland | dbe7183 | 2021-05-12 23:31:00 +0000 | [diff] [blame] | 647 | CommandData transactionData(command.bodySize); |
Steven Moreland | e839334 | 2021-05-05 23:27:53 +0000 | [diff] [blame] | 648 | if (!transactionData.valid()) { |
| 649 | return NO_MEMORY; |
| 650 | } |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 651 | if (status_t status = rpcRec(connection, session, "transaction body", transactionData.data(), |
Steven Moreland | 1e4c2b8 | 2021-05-25 01:51:31 +0000 | [diff] [blame] | 652 | transactionData.size()); |
| 653 | status != OK) |
| 654 | return status; |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 655 | |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 656 | return processTransactInternal(connection, session, std::move(transactionData)); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 657 | } |
| 658 | |
Steven Moreland | 438cce8 | 2021-04-02 18:04:08 +0000 | [diff] [blame] | 659 | static void do_nothing_to_transact_data(Parcel* p, const uint8_t* data, size_t dataSize, |
| 660 | const binder_size_t* objects, size_t objectsCount) { |
| 661 | (void)p; |
| 662 | (void)data; |
| 663 | (void)dataSize; |
| 664 | (void)objects; |
| 665 | (void)objectsCount; |
| 666 | } |
| 667 | |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 668 | status_t RpcState::processTransactInternal(const sp<RpcSession::RpcConnection>& connection, |
| 669 | const sp<RpcSession>& session, |
Steven Moreland | ada72bd | 2021-06-09 23:29:13 +0000 | [diff] [blame] | 670 | CommandData transactionData) { |
| 671 | // for 'recursive' calls to this, we have already read and processed the |
| 672 | // binder from the transaction data and taken reference counts into account, |
| 673 | // so it is cached here. |
| 674 | sp<IBinder> targetRef; |
| 675 | processTransactInternalTailCall: |
| 676 | |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 677 | if (transactionData.size() < sizeof(RpcWireTransaction)) { |
| 678 | ALOGE("Expecting %zu but got %zu bytes for RpcWireTransaction. Terminating!", |
| 679 | sizeof(RpcWireTransaction), transactionData.size()); |
Steven Moreland | c9d7b53 | 2021-06-04 20:57:41 +0000 | [diff] [blame] | 680 | (void)session->shutdownAndWait(false); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 681 | return BAD_VALUE; |
| 682 | } |
| 683 | RpcWireTransaction* transaction = reinterpret_cast<RpcWireTransaction*>(transactionData.data()); |
| 684 | |
| 685 | // TODO(b/182939933): heap allocation just for lookup in mNodeForAddress, |
| 686 | // maybe add an RpcAddress 'view' if the type remains 'heavy' |
| 687 | auto addr = RpcAddress::fromRawEmbedded(&transaction->address); |
Steven Moreland | c7d4013 | 2021-06-10 03:42:11 +0000 | [diff] [blame] | 688 | bool oneway = transaction->flags & IBinder::FLAG_ONEWAY; |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 689 | |
| 690 | status_t replyStatus = OK; |
| 691 | sp<IBinder> target; |
| 692 | if (!addr.isZero()) { |
Steven Moreland | f517427 | 2021-05-25 00:39:28 +0000 | [diff] [blame] | 693 | if (!targetRef) { |
Steven Moreland | 7227c8a | 2021-06-02 00:24:32 +0000 | [diff] [blame] | 694 | replyStatus = onBinderEntering(session, addr, &target); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 695 | } else { |
Steven Moreland | f517427 | 2021-05-25 00:39:28 +0000 | [diff] [blame] | 696 | target = targetRef; |
| 697 | } |
| 698 | |
Steven Moreland | 7227c8a | 2021-06-02 00:24:32 +0000 | [diff] [blame] | 699 | if (replyStatus != OK) { |
| 700 | // do nothing |
| 701 | } else if (target == nullptr) { |
Steven Moreland | f517427 | 2021-05-25 00:39:28 +0000 | [diff] [blame] | 702 | // This can happen if the binder is remote in this process, and |
| 703 | // another thread has called the last decStrong on this binder. |
| 704 | // However, for local binders, it indicates a misbehaving client |
| 705 | // (any binder which is being transacted on should be holding a |
| 706 | // strong ref count), so in either case, terminating the |
| 707 | // session. |
| 708 | ALOGE("While transacting, binder has been deleted at address %s. Terminating!", |
| 709 | addr.toString().c_str()); |
Steven Moreland | c9d7b53 | 2021-06-04 20:57:41 +0000 | [diff] [blame] | 710 | (void)session->shutdownAndWait(false); |
Steven Moreland | f517427 | 2021-05-25 00:39:28 +0000 | [diff] [blame] | 711 | replyStatus = BAD_VALUE; |
| 712 | } else if (target->localBinder() == nullptr) { |
| 713 | ALOGE("Unknown binder address or non-local binder, not address %s. Terminating!", |
| 714 | addr.toString().c_str()); |
Steven Moreland | c9d7b53 | 2021-06-04 20:57:41 +0000 | [diff] [blame] | 715 | (void)session->shutdownAndWait(false); |
Steven Moreland | f517427 | 2021-05-25 00:39:28 +0000 | [diff] [blame] | 716 | replyStatus = BAD_VALUE; |
Steven Moreland | c7d4013 | 2021-06-10 03:42:11 +0000 | [diff] [blame] | 717 | } else if (oneway) { |
Steven Moreland | d45be62 | 2021-06-04 02:19:37 +0000 | [diff] [blame] | 718 | std::unique_lock<std::mutex> _l(mNodeMutex); |
Steven Moreland | f517427 | 2021-05-25 00:39:28 +0000 | [diff] [blame] | 719 | auto it = mNodeForAddress.find(addr); |
| 720 | if (it->second.binder.promote() != target) { |
| 721 | ALOGE("Binder became invalid during transaction. Bad client? %s", |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 722 | addr.toString().c_str()); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 723 | replyStatus = BAD_VALUE; |
Steven Moreland | f517427 | 2021-05-25 00:39:28 +0000 | [diff] [blame] | 724 | } else if (transaction->asyncNumber != it->second.asyncNumber) { |
| 725 | // we need to process some other asynchronous transaction |
| 726 | // first |
Steven Moreland | f517427 | 2021-05-25 00:39:28 +0000 | [diff] [blame] | 727 | it->second.asyncTodo.push(BinderNode::AsyncTodo{ |
| 728 | .ref = target, |
| 729 | .data = std::move(transactionData), |
| 730 | .asyncNumber = transaction->asyncNumber, |
| 731 | }); |
Steven Moreland | d45be62 | 2021-06-04 02:19:37 +0000 | [diff] [blame] | 732 | |
| 733 | size_t numPending = it->second.asyncTodo.size(); |
| 734 | LOG_RPC_DETAIL("Enqueuing %" PRId64 " on %s (%zu pending)", |
| 735 | transaction->asyncNumber, addr.toString().c_str(), numPending); |
| 736 | |
| 737 | constexpr size_t kArbitraryOnewayCallTerminateLevel = 10000; |
| 738 | constexpr size_t kArbitraryOnewayCallWarnLevel = 1000; |
| 739 | constexpr size_t kArbitraryOnewayCallWarnPer = 1000; |
| 740 | |
| 741 | if (numPending >= kArbitraryOnewayCallWarnLevel) { |
| 742 | if (numPending >= kArbitraryOnewayCallTerminateLevel) { |
| 743 | ALOGE("WARNING: %zu pending oneway transactions. Terminating!", numPending); |
| 744 | _l.unlock(); |
| 745 | (void)session->shutdownAndWait(false); |
| 746 | return FAILED_TRANSACTION; |
| 747 | } |
| 748 | |
| 749 | if (numPending % kArbitraryOnewayCallWarnPer == 0) { |
| 750 | ALOGW("Warning: many oneway transactions built up on %p (%zu)", |
| 751 | target.get(), numPending); |
| 752 | } |
| 753 | } |
Steven Moreland | f517427 | 2021-05-25 00:39:28 +0000 | [diff] [blame] | 754 | return OK; |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 755 | } |
| 756 | } |
| 757 | } |
| 758 | |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 759 | Parcel reply; |
Steven Moreland | bdb53ab | 2021-05-05 17:57:41 +0000 | [diff] [blame] | 760 | reply.markForRpc(session); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 761 | |
| 762 | if (replyStatus == OK) { |
Steven Moreland | eff77c1 | 2021-04-15 00:37:19 +0000 | [diff] [blame] | 763 | Parcel data; |
| 764 | // transaction->data is owned by this function. Parcel borrows this data and |
| 765 | // only holds onto it for the duration of this function call. Parcel will be |
| 766 | // deleted before the 'transactionData' object. |
| 767 | data.ipcSetDataReference(transaction->data, |
| 768 | transactionData.size() - offsetof(RpcWireTransaction, data), |
| 769 | nullptr /*object*/, 0 /*objectCount*/, |
| 770 | do_nothing_to_transact_data); |
Steven Moreland | bdb53ab | 2021-05-05 17:57:41 +0000 | [diff] [blame] | 771 | data.markForRpc(session); |
Steven Moreland | eff77c1 | 2021-04-15 00:37:19 +0000 | [diff] [blame] | 772 | |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 773 | if (target) { |
Steven Moreland | c7d4013 | 2021-06-10 03:42:11 +0000 | [diff] [blame] | 774 | bool origAllowNested = connection->allowNested; |
| 775 | connection->allowNested = !oneway; |
| 776 | |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 777 | replyStatus = target->transact(transaction->code, data, &reply, transaction->flags); |
Steven Moreland | c7d4013 | 2021-06-10 03:42:11 +0000 | [diff] [blame] | 778 | |
| 779 | connection->allowNested = origAllowNested; |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 780 | } else { |
| 781 | LOG_RPC_DETAIL("Got special transaction %u", transaction->code); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 782 | |
Steven Moreland | 103424e | 2021-06-02 18:16:19 +0000 | [diff] [blame] | 783 | switch (transaction->code) { |
| 784 | case RPC_SPECIAL_TRANSACT_GET_MAX_THREADS: { |
| 785 | replyStatus = reply.writeInt32(session->getMaxThreads()); |
| 786 | break; |
| 787 | } |
| 788 | case RPC_SPECIAL_TRANSACT_GET_SESSION_ID: { |
| 789 | // for client connections, this should always report the value |
Steven Moreland | 01a6bad | 2021-06-11 00:59:20 +0000 | [diff] [blame] | 790 | // originally returned from the server, so this is asserting |
| 791 | // that it exists |
| 792 | replyStatus = session->mId.value().writeToParcel(&reply); |
Steven Moreland | 103424e | 2021-06-02 18:16:19 +0000 | [diff] [blame] | 793 | break; |
| 794 | } |
| 795 | default: { |
Steven Moreland | 7b8bc4c | 2021-06-10 22:50:27 +0000 | [diff] [blame] | 796 | sp<RpcServer> server = session->server(); |
Steven Moreland | 103424e | 2021-06-02 18:16:19 +0000 | [diff] [blame] | 797 | if (server) { |
| 798 | switch (transaction->code) { |
| 799 | case RPC_SPECIAL_TRANSACT_GET_ROOT: { |
| 800 | replyStatus = reply.writeStrongBinder(server->getRootObject()); |
| 801 | break; |
| 802 | } |
| 803 | default: { |
| 804 | replyStatus = UNKNOWN_TRANSACTION; |
| 805 | } |
| 806 | } |
| 807 | } else { |
| 808 | ALOGE("Special command sent, but no server object attached."); |
Steven Moreland | f137de9 | 2021-04-24 01:54:26 +0000 | [diff] [blame] | 809 | } |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 810 | } |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 811 | } |
| 812 | } |
| 813 | } |
| 814 | |
Steven Moreland | c7d4013 | 2021-06-10 03:42:11 +0000 | [diff] [blame] | 815 | if (oneway) { |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 816 | if (replyStatus != OK) { |
| 817 | ALOGW("Oneway call failed with error: %d", replyStatus); |
| 818 | } |
| 819 | |
| 820 | LOG_RPC_DETAIL("Processed async transaction %" PRId64 " on %s", transaction->asyncNumber, |
| 821 | addr.toString().c_str()); |
| 822 | |
| 823 | // Check to see if there is another asynchronous transaction to process. |
| 824 | // This behavior differs from binder behavior, since in the binder |
| 825 | // driver, asynchronous transactions will be processed after existing |
| 826 | // pending binder transactions on the queue. The downside of this is |
| 827 | // that asynchronous transactions can be drowned out by synchronous |
| 828 | // transactions. However, we have no easy way to queue these |
| 829 | // transactions after the synchronous transactions we may want to read |
| 830 | // from the wire. So, in socket binder here, we have the opposite |
| 831 | // downside: asynchronous transactions may drown out synchronous |
| 832 | // transactions. |
| 833 | { |
| 834 | std::unique_lock<std::mutex> _l(mNodeMutex); |
| 835 | auto it = mNodeForAddress.find(addr); |
| 836 | // last refcount dropped after this transaction happened |
| 837 | if (it == mNodeForAddress.end()) return OK; |
| 838 | |
Steven Moreland | c9d7b53 | 2021-06-04 20:57:41 +0000 | [diff] [blame] | 839 | if (!nodeProgressAsyncNumber(&it->second)) { |
| 840 | _l.unlock(); |
| 841 | (void)session->shutdownAndWait(false); |
| 842 | return DEAD_OBJECT; |
| 843 | } |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 844 | |
| 845 | if (it->second.asyncTodo.size() == 0) return OK; |
| 846 | if (it->second.asyncTodo.top().asyncNumber == it->second.asyncNumber) { |
| 847 | LOG_RPC_DETAIL("Found next async transaction %" PRId64 " on %s", |
| 848 | it->second.asyncNumber, addr.toString().c_str()); |
| 849 | |
| 850 | // justification for const_cast (consider avoiding priority_queue): |
Steven Moreland | f517427 | 2021-05-25 00:39:28 +0000 | [diff] [blame] | 851 | // - AsyncTodo operator< doesn't depend on 'data' or 'ref' objects |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 852 | // - gotta go fast |
Steven Moreland | f517427 | 2021-05-25 00:39:28 +0000 | [diff] [blame] | 853 | auto& todo = const_cast<BinderNode::AsyncTodo&>(it->second.asyncTodo.top()); |
| 854 | |
Steven Moreland | ada72bd | 2021-06-09 23:29:13 +0000 | [diff] [blame] | 855 | // reset up arguments |
| 856 | transactionData = std::move(todo.data); |
| 857 | targetRef = std::move(todo.ref); |
Steven Moreland | f517427 | 2021-05-25 00:39:28 +0000 | [diff] [blame] | 858 | |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 859 | it->second.asyncTodo.pop(); |
Steven Moreland | ada72bd | 2021-06-09 23:29:13 +0000 | [diff] [blame] | 860 | goto processTransactInternalTailCall; |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 861 | } |
| 862 | } |
| 863 | return OK; |
| 864 | } |
| 865 | |
Steven Moreland | 77c3011 | 2021-06-02 20:45:46 +0000 | [diff] [blame] | 866 | LOG_ALWAYS_FATAL_IF(std::numeric_limits<int32_t>::max() - sizeof(RpcWireHeader) - |
| 867 | sizeof(RpcWireReply) < |
| 868 | reply.dataSize(), |
| 869 | "Too much data for reply %zu", reply.dataSize()); |
| 870 | |
| 871 | RpcWireHeader cmdReply{ |
| 872 | .command = RPC_COMMAND_REPLY, |
| 873 | .bodySize = static_cast<uint32_t>(sizeof(RpcWireReply) + reply.dataSize()), |
| 874 | }; |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 875 | RpcWireReply rpcReply{ |
| 876 | .status = replyStatus, |
| 877 | }; |
| 878 | |
Steven Moreland | 77c3011 | 2021-06-02 20:45:46 +0000 | [diff] [blame] | 879 | CommandData replyData(sizeof(RpcWireHeader) + sizeof(RpcWireReply) + reply.dataSize()); |
Steven Moreland | e839334 | 2021-05-05 23:27:53 +0000 | [diff] [blame] | 880 | if (!replyData.valid()) { |
| 881 | return NO_MEMORY; |
| 882 | } |
Steven Moreland | 77c3011 | 2021-06-02 20:45:46 +0000 | [diff] [blame] | 883 | memcpy(replyData.data() + 0, &cmdReply, sizeof(RpcWireHeader)); |
| 884 | memcpy(replyData.data() + sizeof(RpcWireHeader), &rpcReply, sizeof(RpcWireReply)); |
| 885 | memcpy(replyData.data() + sizeof(RpcWireHeader) + sizeof(RpcWireReply), reply.data(), |
| 886 | reply.dataSize()); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 887 | |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 888 | return rpcSend(connection, session, "reply", replyData.data(), replyData.size()); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 889 | } |
| 890 | |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 891 | status_t RpcState::processDecStrong(const sp<RpcSession::RpcConnection>& connection, |
| 892 | const sp<RpcSession>& session, const RpcWireHeader& command) { |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 893 | LOG_ALWAYS_FATAL_IF(command.command != RPC_COMMAND_DEC_STRONG, "command: %d", command.command); |
| 894 | |
Steven Moreland | dbe7183 | 2021-05-12 23:31:00 +0000 | [diff] [blame] | 895 | CommandData commandData(command.bodySize); |
Steven Moreland | e839334 | 2021-05-05 23:27:53 +0000 | [diff] [blame] | 896 | if (!commandData.valid()) { |
| 897 | return NO_MEMORY; |
| 898 | } |
Steven Moreland | 1e4c2b8 | 2021-05-25 01:51:31 +0000 | [diff] [blame] | 899 | if (status_t status = |
Steven Moreland | 5ae6256 | 2021-06-10 03:21:42 +0000 | [diff] [blame] | 900 | rpcRec(connection, session, "dec ref body", commandData.data(), commandData.size()); |
Steven Moreland | 1e4c2b8 | 2021-05-25 01:51:31 +0000 | [diff] [blame] | 901 | status != OK) |
| 902 | return status; |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 903 | |
| 904 | if (command.bodySize < sizeof(RpcWireAddress)) { |
| 905 | ALOGE("Expecting %zu but got %" PRId32 " bytes for RpcWireAddress. Terminating!", |
| 906 | sizeof(RpcWireAddress), command.bodySize); |
Steven Moreland | c9d7b53 | 2021-06-04 20:57:41 +0000 | [diff] [blame] | 907 | (void)session->shutdownAndWait(false); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 908 | return BAD_VALUE; |
| 909 | } |
| 910 | RpcWireAddress* address = reinterpret_cast<RpcWireAddress*>(commandData.data()); |
| 911 | |
| 912 | // TODO(b/182939933): heap allocation just for lookup |
| 913 | auto addr = RpcAddress::fromRawEmbedded(address); |
| 914 | std::unique_lock<std::mutex> _l(mNodeMutex); |
| 915 | auto it = mNodeForAddress.find(addr); |
| 916 | if (it == mNodeForAddress.end()) { |
| 917 | ALOGE("Unknown binder address %s for dec strong.", addr.toString().c_str()); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 918 | return OK; |
| 919 | } |
| 920 | |
| 921 | sp<IBinder> target = it->second.binder.promote(); |
| 922 | if (target == nullptr) { |
| 923 | ALOGE("While requesting dec strong, binder has been deleted at address %s. Terminating!", |
| 924 | addr.toString().c_str()); |
Steven Moreland | c9d7b53 | 2021-06-04 20:57:41 +0000 | [diff] [blame] | 925 | _l.unlock(); |
| 926 | (void)session->shutdownAndWait(false); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 927 | return BAD_VALUE; |
| 928 | } |
| 929 | |
| 930 | if (it->second.timesSent == 0) { |
| 931 | ALOGE("No record of sending binder, but requested decStrong: %s", addr.toString().c_str()); |
| 932 | return OK; |
| 933 | } |
| 934 | |
| 935 | LOG_ALWAYS_FATAL_IF(it->second.sentRef == nullptr, "Inconsistent state, lost ref for %s", |
| 936 | addr.toString().c_str()); |
| 937 | |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 938 | it->second.timesSent--; |
Steven Moreland | 31bde7a | 2021-06-04 00:57:36 +0000 | [diff] [blame] | 939 | sp<IBinder> tempHold = tryEraseNode(it); |
| 940 | _l.unlock(); |
| 941 | tempHold = nullptr; // destructor may make binder calls on this session |
| 942 | |
| 943 | return OK; |
| 944 | } |
| 945 | |
| 946 | sp<IBinder> RpcState::tryEraseNode(std::map<RpcAddress, BinderNode>::iterator& it) { |
| 947 | sp<IBinder> ref; |
| 948 | |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 949 | if (it->second.timesSent == 0) { |
Steven Moreland | 31bde7a | 2021-06-04 00:57:36 +0000 | [diff] [blame] | 950 | ref = std::move(it->second.sentRef); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 951 | |
| 952 | if (it->second.timesRecd == 0) { |
Steven Moreland | a6e11cf | 2021-06-04 00:58:31 +0000 | [diff] [blame] | 953 | LOG_ALWAYS_FATAL_IF(!it->second.asyncTodo.empty(), |
| 954 | "Can't delete binder w/ pending async transactions"); |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 955 | mNodeForAddress.erase(it); |
| 956 | } |
| 957 | } |
| 958 | |
Steven Moreland | 31bde7a | 2021-06-04 00:57:36 +0000 | [diff] [blame] | 959 | return ref; |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 960 | } |
| 961 | |
Steven Moreland | c9d7b53 | 2021-06-04 20:57:41 +0000 | [diff] [blame] | 962 | bool RpcState::nodeProgressAsyncNumber(BinderNode* node) { |
Steven Moreland | 583a14a | 2021-06-04 02:04:58 +0000 | [diff] [blame] | 963 | // 2**64 =~ 10**19 =~ 1000 transactions per second for 585 million years to |
| 964 | // a single binder |
| 965 | if (node->asyncNumber >= std::numeric_limits<decltype(node->asyncNumber)>::max()) { |
| 966 | ALOGE("Out of async transaction IDs. Terminating"); |
Steven Moreland | 583a14a | 2021-06-04 02:04:58 +0000 | [diff] [blame] | 967 | return false; |
| 968 | } |
| 969 | node->asyncNumber++; |
| 970 | return true; |
| 971 | } |
| 972 | |
Steven Moreland | 5553ac4 | 2020-11-11 02:14:45 +0000 | [diff] [blame] | 973 | } // namespace android |