Andres Morales | ac80818 | 2015-02-26 14:11:04 -0800 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2015 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 | */ |
Andres Morales | b2abaa8 | 2015-03-03 09:09:18 -0800 | [diff] [blame] | 16 | #include <UniquePtr.h> |
Andres Morales | 7d0f040 | 2015-03-19 18:02:55 -0700 | [diff] [blame] | 17 | #include <gatekeeper/gatekeeper.h> |
Andres Morales | ac80818 | 2015-02-26 14:11:04 -0800 | [diff] [blame] | 18 | |
Andres Morales | ec9fd1d | 2015-04-20 07:45:50 -0700 | [diff] [blame] | 19 | #include <endian.h> |
| 20 | |
Andres Morales | f436c80 | 2015-10-08 12:21:25 -0700 | [diff] [blame] | 21 | #define DAY_IN_MS (1000 * 60 * 60 * 24) |
| 22 | |
Andres Morales | 7d0f040 | 2015-03-19 18:02:55 -0700 | [diff] [blame] | 23 | namespace gatekeeper { |
Andres Morales | ac80818 | 2015-02-26 14:11:04 -0800 | [diff] [blame] | 24 | |
Andres Morales | 7d0f040 | 2015-03-19 18:02:55 -0700 | [diff] [blame] | 25 | void GateKeeper::Enroll(const EnrollRequest &request, EnrollResponse *response) { |
Andres Morales | ac80818 | 2015-02-26 14:11:04 -0800 | [diff] [blame] | 26 | if (response == NULL) return; |
| 27 | |
Andres Morales | b2abaa8 | 2015-03-03 09:09:18 -0800 | [diff] [blame] | 28 | if (!request.provided_password.buffer.get()) { |
Andres Morales | 7d0f040 | 2015-03-19 18:02:55 -0700 | [diff] [blame] | 29 | response->error = ERROR_INVALID; |
Andres Morales | ac80818 | 2015-02-26 14:11:04 -0800 | [diff] [blame] | 30 | return; |
| 31 | } |
Andres Morales | b2abaa8 | 2015-03-03 09:09:18 -0800 | [diff] [blame] | 32 | |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 33 | secure_id_t user_id = 0;// todo: rename to policy |
| 34 | uint32_t uid = request.user_id; |
Andres Morales | b2abaa8 | 2015-03-03 09:09:18 -0800 | [diff] [blame] | 35 | |
Andres Morales | edd3e3d | 2015-03-12 13:30:15 -0700 | [diff] [blame] | 36 | if (request.password_handle.buffer.get() == NULL) { |
| 37 | // Password handle does not match what is stored, generate new SecureID |
| 38 | GetRandom(&user_id, sizeof(secure_id_t)); |
| 39 | } else { |
Andres Morales | 426fcfb | 2015-04-01 13:33:45 -0700 | [diff] [blame] | 40 | password_handle_t *pw_handle = |
| 41 | reinterpret_cast<password_handle_t *>(request.password_handle.buffer.get()); |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 42 | |
Andres Morales | 48a4f83 | 2015-05-29 10:13:31 -0700 | [diff] [blame] | 43 | if (pw_handle->version > HANDLE_VERSION) { |
Andres Morales | 426fcfb | 2015-04-01 13:33:45 -0700 | [diff] [blame] | 44 | response->error = ERROR_INVALID; |
| 45 | return; |
Andres Morales | edd3e3d | 2015-03-12 13:30:15 -0700 | [diff] [blame] | 46 | } |
Andres Morales | 426fcfb | 2015-04-01 13:33:45 -0700 | [diff] [blame] | 47 | |
Andres Morales | 48a4f83 | 2015-05-29 10:13:31 -0700 | [diff] [blame] | 48 | user_id = pw_handle->user_id; |
| 49 | |
| 50 | uint64_t timestamp = GetMillisecondsSinceBoot(); |
| 51 | |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 52 | uint32_t timeout = 0; |
Andres Morales | 48a4f83 | 2015-05-29 10:13:31 -0700 | [diff] [blame] | 53 | bool throttle = (pw_handle->version >= HANDLE_VERSION_THROTTLE); |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 54 | if (throttle) { |
Andres Morales | 48a4f83 | 2015-05-29 10:13:31 -0700 | [diff] [blame] | 55 | bool throttle_secure = pw_handle->flags & HANDLE_FLAG_THROTTLE_SECURE; |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 56 | failure_record_t record; |
Andres Morales | 48a4f83 | 2015-05-29 10:13:31 -0700 | [diff] [blame] | 57 | if (!GetFailureRecord(uid, user_id, &record, throttle_secure)) { |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 58 | response->error = ERROR_UNKNOWN; |
| 59 | return; |
| 60 | } |
| 61 | |
Andres Morales | 48a4f83 | 2015-05-29 10:13:31 -0700 | [diff] [blame] | 62 | if (ThrottleRequest(uid, timestamp, &record, throttle_secure, response)) return; |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 63 | |
Andres Morales | 48a4f83 | 2015-05-29 10:13:31 -0700 | [diff] [blame] | 64 | if (!IncrementFailureRecord(uid, user_id, timestamp, &record, throttle_secure)) { |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 65 | response->error = ERROR_UNKNOWN; |
| 66 | return; |
| 67 | } |
| 68 | |
| 69 | timeout = ComputeRetryTimeout(&record); |
| 70 | } |
| 71 | |
| 72 | if (!DoVerify(pw_handle, request.enrolled_password)) { |
| 73 | // incorrect old password |
| 74 | if (throttle && timeout > 0) { |
| 75 | response->SetRetryTimeout(timeout); |
| 76 | } else { |
| 77 | response->error = ERROR_INVALID; |
| 78 | } |
| 79 | return; |
| 80 | } |
Andres Morales | edd3e3d | 2015-03-12 13:30:15 -0700 | [diff] [blame] | 81 | } |
| 82 | |
Andres Morales | 48a4f83 | 2015-05-29 10:13:31 -0700 | [diff] [blame] | 83 | uint64_t flags = 0; |
| 84 | if (ClearFailureRecord(uid, user_id, true)) { |
| 85 | flags |= HANDLE_FLAG_THROTTLE_SECURE; |
| 86 | } else { |
| 87 | ClearFailureRecord(uid, user_id, false); |
| 88 | } |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 89 | |
Andres Morales | edd3e3d | 2015-03-12 13:30:15 -0700 | [diff] [blame] | 90 | salt_t salt; |
| 91 | GetRandom(&salt, sizeof(salt)); |
| 92 | |
Andres Morales | edd3e3d | 2015-03-12 13:30:15 -0700 | [diff] [blame] | 93 | SizedBuffer password_handle; |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 94 | if (!CreatePasswordHandle(&password_handle, |
Andres Morales | 48a4f83 | 2015-05-29 10:13:31 -0700 | [diff] [blame] | 95 | salt, user_id, flags, HANDLE_VERSION, request.provided_password.buffer.get(), |
Andres Morales | edd3e3d | 2015-03-12 13:30:15 -0700 | [diff] [blame] | 96 | request.provided_password.length)) { |
Andres Morales | 7d0f040 | 2015-03-19 18:02:55 -0700 | [diff] [blame] | 97 | response->error = ERROR_INVALID; |
Andres Morales | edd3e3d | 2015-03-12 13:30:15 -0700 | [diff] [blame] | 98 | return; |
| 99 | } |
| 100 | |
Andres Morales | edd3e3d | 2015-03-12 13:30:15 -0700 | [diff] [blame] | 101 | response->SetEnrolledPasswordHandle(&password_handle); |
Andres Morales | ac80818 | 2015-02-26 14:11:04 -0800 | [diff] [blame] | 102 | } |
| 103 | |
Andres Morales | 7d0f040 | 2015-03-19 18:02:55 -0700 | [diff] [blame] | 104 | void GateKeeper::Verify(const VerifyRequest &request, VerifyResponse *response) { |
Andres Morales | ac80818 | 2015-02-26 14:11:04 -0800 | [diff] [blame] | 105 | if (response == NULL) return; |
| 106 | |
Andres Morales | b2abaa8 | 2015-03-03 09:09:18 -0800 | [diff] [blame] | 107 | if (!request.provided_password.buffer.get() || !request.password_handle.buffer.get()) { |
Andres Morales | 7d0f040 | 2015-03-19 18:02:55 -0700 | [diff] [blame] | 108 | response->error = ERROR_INVALID; |
Andres Morales | ac80818 | 2015-02-26 14:11:04 -0800 | [diff] [blame] | 109 | return; |
| 110 | } |
| 111 | |
Andres Morales | edd3e3d | 2015-03-12 13:30:15 -0700 | [diff] [blame] | 112 | password_handle_t *password_handle = reinterpret_cast<password_handle_t *>( |
| 113 | request.password_handle.buffer.get()); |
Andres Morales | b2abaa8 | 2015-03-03 09:09:18 -0800 | [diff] [blame] | 114 | |
Andres Morales | 48a4f83 | 2015-05-29 10:13:31 -0700 | [diff] [blame] | 115 | if (password_handle->version > HANDLE_VERSION) { |
Andres Morales | 7d0f040 | 2015-03-19 18:02:55 -0700 | [diff] [blame] | 116 | response->error = ERROR_INVALID; |
Andres Morales | b2abaa8 | 2015-03-03 09:09:18 -0800 | [diff] [blame] | 117 | return; |
| 118 | } |
| 119 | |
Andres Morales | 426fcfb | 2015-04-01 13:33:45 -0700 | [diff] [blame] | 120 | secure_id_t user_id = password_handle->user_id; |
Andres Morales | 48a4f83 | 2015-05-29 10:13:31 -0700 | [diff] [blame] | 121 | secure_id_t authenticator_id = 0; |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 122 | uint32_t uid = request.user_id; |
Andres Morales | b2abaa8 | 2015-03-03 09:09:18 -0800 | [diff] [blame] | 123 | |
Andres Morales | ec9fd1d | 2015-04-20 07:45:50 -0700 | [diff] [blame] | 124 | uint64_t timestamp = GetMillisecondsSinceBoot(); |
Andres Morales | edd3e3d | 2015-03-12 13:30:15 -0700 | [diff] [blame] | 125 | |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 126 | uint32_t timeout = 0; |
Andres Morales | 48a4f83 | 2015-05-29 10:13:31 -0700 | [diff] [blame] | 127 | bool throttle = (password_handle->version >= HANDLE_VERSION_THROTTLE); |
| 128 | bool throttle_secure = password_handle->flags & HANDLE_FLAG_THROTTLE_SECURE; |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 129 | if (throttle) { |
| 130 | failure_record_t record; |
Andres Morales | 48a4f83 | 2015-05-29 10:13:31 -0700 | [diff] [blame] | 131 | if (!GetFailureRecord(uid, user_id, &record, throttle_secure)) { |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 132 | response->error = ERROR_UNKNOWN; |
| 133 | return; |
| 134 | } |
| 135 | |
Andres Morales | 48a4f83 | 2015-05-29 10:13:31 -0700 | [diff] [blame] | 136 | if (ThrottleRequest(uid, timestamp, &record, throttle_secure, response)) return; |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 137 | |
Andres Morales | 48a4f83 | 2015-05-29 10:13:31 -0700 | [diff] [blame] | 138 | if (!IncrementFailureRecord(uid, user_id, timestamp, &record, throttle_secure)) { |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 139 | response->error = ERROR_UNKNOWN; |
| 140 | return; |
| 141 | } |
| 142 | |
| 143 | timeout = ComputeRetryTimeout(&record); |
Andres Morales | 893fa7f | 2015-06-02 19:00:03 -0700 | [diff] [blame] | 144 | } else { |
| 145 | response->request_reenroll = true; |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 146 | } |
| 147 | |
Andres Morales | edd3e3d | 2015-03-12 13:30:15 -0700 | [diff] [blame] | 148 | if (DoVerify(password_handle, request.provided_password)) { |
Andres Morales | ac80818 | 2015-02-26 14:11:04 -0800 | [diff] [blame] | 149 | // Signature matches |
Andres Morales | 48a4f83 | 2015-05-29 10:13:31 -0700 | [diff] [blame] | 150 | UniquePtr<uint8_t> auth_token_buffer; |
| 151 | uint32_t auth_token_len; |
| 152 | MintAuthToken(&auth_token_buffer, &auth_token_len, timestamp, |
Andres Morales | 6034309 | 2015-04-09 19:01:10 -0700 | [diff] [blame] | 153 | user_id, authenticator_id, request.challenge); |
Andres Morales | 48a4f83 | 2015-05-29 10:13:31 -0700 | [diff] [blame] | 154 | |
| 155 | SizedBuffer auth_token(auth_token_len); |
| 156 | memcpy(auth_token.buffer.get(), auth_token_buffer.get(), auth_token_len); |
Andres Morales | ac80818 | 2015-02-26 14:11:04 -0800 | [diff] [blame] | 157 | response->SetVerificationToken(&auth_token); |
Andres Morales | 48a4f83 | 2015-05-29 10:13:31 -0700 | [diff] [blame] | 158 | if (throttle) ClearFailureRecord(uid, user_id, throttle_secure); |
Andres Morales | ac80818 | 2015-02-26 14:11:04 -0800 | [diff] [blame] | 159 | } else { |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 160 | // compute the new timeout given the incremented record |
| 161 | if (throttle && timeout > 0) { |
| 162 | response->SetRetryTimeout(timeout); |
| 163 | } else { |
| 164 | response->error = ERROR_INVALID; |
| 165 | } |
Andres Morales | ac80818 | 2015-02-26 14:11:04 -0800 | [diff] [blame] | 166 | } |
| 167 | } |
| 168 | |
Andres Morales | 7d0f040 | 2015-03-19 18:02:55 -0700 | [diff] [blame] | 169 | bool GateKeeper::CreatePasswordHandle(SizedBuffer *password_handle_buffer, salt_t salt, |
Andres Morales | 48a4f83 | 2015-05-29 10:13:31 -0700 | [diff] [blame] | 170 | secure_id_t user_id, uint64_t flags, uint8_t handle_version, const uint8_t *password, |
Andres Morales | 11ed52a | 2015-03-30 16:47:47 -0700 | [diff] [blame] | 171 | uint32_t password_length) { |
Andres Morales | edd3e3d | 2015-03-12 13:30:15 -0700 | [diff] [blame] | 172 | password_handle_buffer->buffer.reset(new uint8_t[sizeof(password_handle_t)]); |
| 173 | password_handle_buffer->length = sizeof(password_handle_t); |
| 174 | |
| 175 | password_handle_t *password_handle = reinterpret_cast<password_handle_t *>( |
| 176 | password_handle_buffer->buffer.get()); |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 177 | password_handle->version = handle_version; |
Andres Morales | edd3e3d | 2015-03-12 13:30:15 -0700 | [diff] [blame] | 178 | password_handle->salt = salt; |
| 179 | password_handle->user_id = user_id; |
Andres Morales | 48a4f83 | 2015-05-29 10:13:31 -0700 | [diff] [blame] | 180 | password_handle->flags = flags; |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 181 | password_handle->hardware_backed = IsHardwareBacked(); |
Andres Morales | edd3e3d | 2015-03-12 13:30:15 -0700 | [diff] [blame] | 182 | |
Andres Morales | 48a4f83 | 2015-05-29 10:13:31 -0700 | [diff] [blame] | 183 | uint32_t metadata_length = sizeof(user_id) + sizeof(flags) + sizeof(HANDLE_VERSION); |
Andres Morales | edd3e3d | 2015-03-12 13:30:15 -0700 | [diff] [blame] | 184 | uint8_t to_sign[password_length + metadata_length]; |
Andres Morales | 94b201e | 2015-04-16 13:47:47 -0700 | [diff] [blame] | 185 | memcpy(to_sign, password_handle, metadata_length); |
Andres Morales | edd3e3d | 2015-03-12 13:30:15 -0700 | [diff] [blame] | 186 | memcpy(to_sign + metadata_length, password, password_length); |
| 187 | |
Andres Morales | f10e289 | 2015-03-23 12:03:56 -0700 | [diff] [blame] | 188 | const uint8_t *password_key = NULL; |
Andres Morales | 11ed52a | 2015-03-30 16:47:47 -0700 | [diff] [blame] | 189 | uint32_t password_key_length = 0; |
Andres Morales | edd3e3d | 2015-03-12 13:30:15 -0700 | [diff] [blame] | 190 | GetPasswordKey(&password_key, &password_key_length); |
| 191 | |
Andres Morales | f10e289 | 2015-03-23 12:03:56 -0700 | [diff] [blame] | 192 | if (!password_key || password_key_length == 0) { |
Andres Morales | edd3e3d | 2015-03-12 13:30:15 -0700 | [diff] [blame] | 193 | return false; |
| 194 | } |
| 195 | |
| 196 | ComputePasswordSignature(password_handle->signature, sizeof(password_handle->signature), |
Andres Morales | f10e289 | 2015-03-23 12:03:56 -0700 | [diff] [blame] | 197 | password_key, password_key_length, to_sign, sizeof(to_sign), salt); |
Andres Morales | edd3e3d | 2015-03-12 13:30:15 -0700 | [diff] [blame] | 198 | return true; |
| 199 | } |
| 200 | |
Andres Morales | 7d0f040 | 2015-03-19 18:02:55 -0700 | [diff] [blame] | 201 | bool GateKeeper::DoVerify(const password_handle_t *expected_handle, const SizedBuffer &password) { |
Andres Morales | edd3e3d | 2015-03-12 13:30:15 -0700 | [diff] [blame] | 202 | if (!password.buffer.get()) return false; |
| 203 | |
| 204 | SizedBuffer provided_handle; |
| 205 | if (!CreatePasswordHandle(&provided_handle, expected_handle->salt, expected_handle->user_id, |
Andres Morales | 48a4f83 | 2015-05-29 10:13:31 -0700 | [diff] [blame] | 206 | expected_handle->flags, expected_handle->version, |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 207 | password.buffer.get(), password.length)) { |
Andres Morales | edd3e3d | 2015-03-12 13:30:15 -0700 | [diff] [blame] | 208 | return false; |
| 209 | } |
| 210 | |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 211 | password_handle_t *generated_handle = |
| 212 | reinterpret_cast<password_handle_t *>(provided_handle.buffer.get()); |
| 213 | return memcmp_s(generated_handle->signature, expected_handle->signature, |
| 214 | sizeof(expected_handle->signature)) == 0; |
Andres Morales | edd3e3d | 2015-03-12 13:30:15 -0700 | [diff] [blame] | 215 | } |
| 216 | |
Andres Morales | 11ed52a | 2015-03-30 16:47:47 -0700 | [diff] [blame] | 217 | void GateKeeper::MintAuthToken(UniquePtr<uint8_t> *auth_token, uint32_t *length, |
Andres Morales | 652f076 | 2015-04-13 10:23:20 -0700 | [diff] [blame] | 218 | uint64_t timestamp, secure_id_t user_id, secure_id_t authenticator_id, |
Andres Morales | 6034309 | 2015-04-09 19:01:10 -0700 | [diff] [blame] | 219 | uint64_t challenge) { |
Andres Morales | b2abaa8 | 2015-03-03 09:09:18 -0800 | [diff] [blame] | 220 | if (auth_token == NULL) return; |
| 221 | |
Andres Morales | fa104e1 | 2015-04-08 12:36:44 -0700 | [diff] [blame] | 222 | hw_auth_token_t *token = new hw_auth_token_t; |
Andres Morales | ac80818 | 2015-02-26 14:11:04 -0800 | [diff] [blame] | 223 | SizedBuffer serialized_auth_token; |
| 224 | |
Andres Morales | fa104e1 | 2015-04-08 12:36:44 -0700 | [diff] [blame] | 225 | token->version = HW_AUTH_TOKEN_VERSION; |
Andres Morales | 6034309 | 2015-04-09 19:01:10 -0700 | [diff] [blame] | 226 | token->challenge = challenge; |
Andres Morales | fa104e1 | 2015-04-08 12:36:44 -0700 | [diff] [blame] | 227 | token->user_id = user_id; |
| 228 | token->authenticator_id = authenticator_id; |
| 229 | token->authenticator_type = htonl(HW_AUTH_PASSWORD); |
Andres Morales | ec9fd1d | 2015-04-20 07:45:50 -0700 | [diff] [blame] | 230 | token->timestamp = htobe64(timestamp); |
Andres Morales | ac80818 | 2015-02-26 14:11:04 -0800 | [diff] [blame] | 231 | |
Andres Morales | f10e289 | 2015-03-23 12:03:56 -0700 | [diff] [blame] | 232 | const uint8_t *auth_token_key = NULL; |
Andres Morales | 11ed52a | 2015-03-30 16:47:47 -0700 | [diff] [blame] | 233 | uint32_t key_len = 0; |
Andres Morales | 58bb246 | 2015-04-10 18:27:50 -0700 | [diff] [blame] | 234 | if (GetAuthTokenKey(&auth_token_key, &key_len)) { |
| 235 | uint32_t hash_len = (uint32_t)((uint8_t *)&token->hmac - (uint8_t *)token); |
| 236 | ComputeSignature(token->hmac, sizeof(token->hmac), auth_token_key, key_len, |
| 237 | reinterpret_cast<uint8_t *>(token), hash_len); |
Andres Morales | 48a4f83 | 2015-05-29 10:13:31 -0700 | [diff] [blame] | 238 | delete[] auth_token_key; |
Andres Morales | 58bb246 | 2015-04-10 18:27:50 -0700 | [diff] [blame] | 239 | } else { |
| 240 | memset(token->hmac, 0, sizeof(token->hmac)); |
| 241 | } |
Andres Morales | ac80818 | 2015-02-26 14:11:04 -0800 | [diff] [blame] | 242 | |
Andres Morales | fa104e1 | 2015-04-08 12:36:44 -0700 | [diff] [blame] | 243 | if (length != NULL) *length = sizeof(*token); |
Andres Morales | b2abaa8 | 2015-03-03 09:09:18 -0800 | [diff] [blame] | 244 | auth_token->reset(reinterpret_cast<uint8_t *>(token)); |
Andres Morales | ac80818 | 2015-02-26 14:11:04 -0800 | [diff] [blame] | 245 | } |
Andres Morales | b2abaa8 | 2015-03-03 09:09:18 -0800 | [diff] [blame] | 246 | |
Andres Morales | f436c80 | 2015-10-08 12:21:25 -0700 | [diff] [blame] | 247 | /* |
| 248 | * Calculates the timeout in milliseconds as a function of the failure |
| 249 | * counter 'x' as follows: |
| 250 | * |
| 251 | * [0. 5) -> 0 |
| 252 | * 5 -> 30 |
| 253 | * [6, 10) -> 0 |
| 254 | * [11, 30) -> 30 |
| 255 | * [30, 140) -> 30 * (2^((x - 30)/10)) |
| 256 | * [140, inf) -> 1 day |
| 257 | * |
| 258 | */ |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 259 | uint32_t GateKeeper::ComputeRetryTimeout(const failure_record_t *record) { |
Andres Morales | b6a5cd7 | 2015-06-03 17:59:17 -0700 | [diff] [blame] | 260 | static const int failure_timeout_ms = 30000; |
| 261 | if (record->failure_counter == 0) return 0; |
| 262 | |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 263 | if (record->failure_counter > 0 && record->failure_counter <= 10) { |
| 264 | if (record->failure_counter % 5 == 0) { |
Andres Morales | b6a5cd7 | 2015-06-03 17:59:17 -0700 | [diff] [blame] | 265 | return failure_timeout_ms; |
Andres Morales | d14f472 | 2015-12-18 11:57:34 -0800 | [diff] [blame] | 266 | } else { |
| 267 | return 0; |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 268 | } |
Andres Morales | f436c80 | 2015-10-08 12:21:25 -0700 | [diff] [blame] | 269 | } else if (record->failure_counter < 30) { |
Andres Morales | b6a5cd7 | 2015-06-03 17:59:17 -0700 | [diff] [blame] | 270 | return failure_timeout_ms; |
Andres Morales | f436c80 | 2015-10-08 12:21:25 -0700 | [diff] [blame] | 271 | } else if (record->failure_counter < 140) { |
| 272 | return failure_timeout_ms << ((record->failure_counter - 30) / 10); |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 273 | } |
Andres Morales | f436c80 | 2015-10-08 12:21:25 -0700 | [diff] [blame] | 274 | |
| 275 | return DAY_IN_MS; |
Andres Morales | ac80818 | 2015-02-26 14:11:04 -0800 | [diff] [blame] | 276 | } |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 277 | |
Andres Morales | a623e45 | 2015-05-27 12:26:59 -0700 | [diff] [blame] | 278 | bool GateKeeper::ThrottleRequest(uint32_t uid, uint64_t timestamp, |
Andres Morales | 48a4f83 | 2015-05-29 10:13:31 -0700 | [diff] [blame] | 279 | failure_record_t *record, bool secure, GateKeeperMessage *response) { |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 280 | |
| 281 | uint64_t last_checked = record->last_checked_timestamp; |
| 282 | uint32_t timeout = ComputeRetryTimeout(record); |
| 283 | |
| 284 | if (timeout > 0) { |
| 285 | // we have a pending timeout |
| 286 | if (timestamp < last_checked + timeout && timestamp > last_checked) { |
| 287 | // attempt before timeout expired, return remaining time |
| 288 | response->SetRetryTimeout(timeout - (timestamp - last_checked)); |
| 289 | return true; |
| 290 | } else if (timestamp <= last_checked) { |
| 291 | // device was rebooted or timer reset, don't count as new failure but |
| 292 | // reset timeout |
| 293 | record->last_checked_timestamp = timestamp; |
Andres Morales | 48a4f83 | 2015-05-29 10:13:31 -0700 | [diff] [blame] | 294 | if (!WriteFailureRecord(uid, record, secure)) { |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 295 | response->error = ERROR_UNKNOWN; |
| 296 | return true; |
| 297 | } |
| 298 | response->SetRetryTimeout(timeout); |
| 299 | return true; |
| 300 | } |
| 301 | } |
| 302 | |
| 303 | return false; |
| 304 | } |
| 305 | |
| 306 | bool GateKeeper::IncrementFailureRecord(uint32_t uid, secure_id_t user_id, uint64_t timestamp, |
Andres Morales | 48a4f83 | 2015-05-29 10:13:31 -0700 | [diff] [blame] | 307 | failure_record_t *record, bool secure) { |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 308 | record->secure_user_id = user_id; |
| 309 | record->failure_counter++; |
| 310 | record->last_checked_timestamp = timestamp; |
| 311 | |
Andres Morales | 48a4f83 | 2015-05-29 10:13:31 -0700 | [diff] [blame] | 312 | return WriteFailureRecord(uid, record, secure); |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 313 | } |
| 314 | } // namespace gatekeeper |
| 315 | |