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 | */ |
Jiyong Park | 002c866 | 2017-08-09 21:10:16 +0900 | [diff] [blame] | 16 | #include <gatekeeper/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 |
Shawn Willden | f73a010 | 2016-09-13 17:39:27 +0000 | [diff] [blame] | 150 | UniquePtr<uint8_t> auth_token_buffer; |
Andres Morales | 48a4f83 | 2015-05-29 10:13:31 -0700 | [diff] [blame] | 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); |
Alexey Polyudov | 84f8f9f | 2016-08-18 13:48:50 -0700 | [diff] [blame] | 184 | const size_t to_sign_size = password_length + metadata_length; |
George Burgess IV | cfa8a41 | 2017-08-29 14:13:20 -0700 | [diff] [blame] | 185 | UniquePtr<uint8_t[]> to_sign(new uint8_t[to_sign_size]); |
Alexey Polyudov | 84f8f9f | 2016-08-18 13:48:50 -0700 | [diff] [blame] | 186 | |
| 187 | if (to_sign.get() == nullptr) { |
| 188 | return false; |
| 189 | } |
| 190 | |
| 191 | memcpy(to_sign.get(), password_handle, metadata_length); |
| 192 | memcpy(to_sign.get() + metadata_length, password, password_length); |
Andres Morales | edd3e3d | 2015-03-12 13:30:15 -0700 | [diff] [blame] | 193 | |
Andres Morales | f10e289 | 2015-03-23 12:03:56 -0700 | [diff] [blame] | 194 | const uint8_t *password_key = NULL; |
Andres Morales | 11ed52a | 2015-03-30 16:47:47 -0700 | [diff] [blame] | 195 | uint32_t password_key_length = 0; |
Andres Morales | edd3e3d | 2015-03-12 13:30:15 -0700 | [diff] [blame] | 196 | GetPasswordKey(&password_key, &password_key_length); |
| 197 | |
Andres Morales | f10e289 | 2015-03-23 12:03:56 -0700 | [diff] [blame] | 198 | if (!password_key || password_key_length == 0) { |
Andres Morales | edd3e3d | 2015-03-12 13:30:15 -0700 | [diff] [blame] | 199 | return false; |
| 200 | } |
| 201 | |
| 202 | ComputePasswordSignature(password_handle->signature, sizeof(password_handle->signature), |
Alexey Polyudov | 84f8f9f | 2016-08-18 13:48:50 -0700 | [diff] [blame] | 203 | password_key, password_key_length, to_sign.get(), to_sign_size, salt); |
Andres Morales | edd3e3d | 2015-03-12 13:30:15 -0700 | [diff] [blame] | 204 | return true; |
| 205 | } |
| 206 | |
Andres Morales | 7d0f040 | 2015-03-19 18:02:55 -0700 | [diff] [blame] | 207 | bool GateKeeper::DoVerify(const password_handle_t *expected_handle, const SizedBuffer &password) { |
Andres Morales | edd3e3d | 2015-03-12 13:30:15 -0700 | [diff] [blame] | 208 | if (!password.buffer.get()) return false; |
| 209 | |
| 210 | SizedBuffer provided_handle; |
| 211 | if (!CreatePasswordHandle(&provided_handle, expected_handle->salt, expected_handle->user_id, |
Andres Morales | 48a4f83 | 2015-05-29 10:13:31 -0700 | [diff] [blame] | 212 | expected_handle->flags, expected_handle->version, |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 213 | password.buffer.get(), password.length)) { |
Andres Morales | edd3e3d | 2015-03-12 13:30:15 -0700 | [diff] [blame] | 214 | return false; |
| 215 | } |
| 216 | |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 217 | password_handle_t *generated_handle = |
| 218 | reinterpret_cast<password_handle_t *>(provided_handle.buffer.get()); |
| 219 | return memcmp_s(generated_handle->signature, expected_handle->signature, |
| 220 | sizeof(expected_handle->signature)) == 0; |
Andres Morales | edd3e3d | 2015-03-12 13:30:15 -0700 | [diff] [blame] | 221 | } |
| 222 | |
Shawn Willden | f73a010 | 2016-09-13 17:39:27 +0000 | [diff] [blame] | 223 | void GateKeeper::MintAuthToken(UniquePtr<uint8_t> *auth_token, uint32_t *length, |
Andres Morales | 652f076 | 2015-04-13 10:23:20 -0700 | [diff] [blame] | 224 | 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] | 225 | uint64_t challenge) { |
Andres Morales | b2abaa8 | 2015-03-03 09:09:18 -0800 | [diff] [blame] | 226 | if (auth_token == NULL) return; |
| 227 | |
Andres Morales | fa104e1 | 2015-04-08 12:36:44 -0700 | [diff] [blame] | 228 | hw_auth_token_t *token = new hw_auth_token_t; |
Andres Morales | ac80818 | 2015-02-26 14:11:04 -0800 | [diff] [blame] | 229 | SizedBuffer serialized_auth_token; |
| 230 | |
Andres Morales | fa104e1 | 2015-04-08 12:36:44 -0700 | [diff] [blame] | 231 | token->version = HW_AUTH_TOKEN_VERSION; |
Andres Morales | 6034309 | 2015-04-09 19:01:10 -0700 | [diff] [blame] | 232 | token->challenge = challenge; |
Andres Morales | fa104e1 | 2015-04-08 12:36:44 -0700 | [diff] [blame] | 233 | token->user_id = user_id; |
| 234 | token->authenticator_id = authenticator_id; |
| 235 | token->authenticator_type = htonl(HW_AUTH_PASSWORD); |
Andres Morales | ec9fd1d | 2015-04-20 07:45:50 -0700 | [diff] [blame] | 236 | token->timestamp = htobe64(timestamp); |
Andres Morales | ac80818 | 2015-02-26 14:11:04 -0800 | [diff] [blame] | 237 | |
Andres Morales | f10e289 | 2015-03-23 12:03:56 -0700 | [diff] [blame] | 238 | const uint8_t *auth_token_key = NULL; |
Andres Morales | 11ed52a | 2015-03-30 16:47:47 -0700 | [diff] [blame] | 239 | uint32_t key_len = 0; |
Andres Morales | 58bb246 | 2015-04-10 18:27:50 -0700 | [diff] [blame] | 240 | if (GetAuthTokenKey(&auth_token_key, &key_len)) { |
| 241 | uint32_t hash_len = (uint32_t)((uint8_t *)&token->hmac - (uint8_t *)token); |
| 242 | ComputeSignature(token->hmac, sizeof(token->hmac), auth_token_key, key_len, |
| 243 | reinterpret_cast<uint8_t *>(token), hash_len); |
| 244 | } else { |
| 245 | memset(token->hmac, 0, sizeof(token->hmac)); |
| 246 | } |
Andres Morales | ac80818 | 2015-02-26 14:11:04 -0800 | [diff] [blame] | 247 | |
Andres Morales | fa104e1 | 2015-04-08 12:36:44 -0700 | [diff] [blame] | 248 | if (length != NULL) *length = sizeof(*token); |
Andres Morales | b2abaa8 | 2015-03-03 09:09:18 -0800 | [diff] [blame] | 249 | auth_token->reset(reinterpret_cast<uint8_t *>(token)); |
Andres Morales | ac80818 | 2015-02-26 14:11:04 -0800 | [diff] [blame] | 250 | } |
Andres Morales | b2abaa8 | 2015-03-03 09:09:18 -0800 | [diff] [blame] | 251 | |
Andres Morales | f436c80 | 2015-10-08 12:21:25 -0700 | [diff] [blame] | 252 | /* |
| 253 | * Calculates the timeout in milliseconds as a function of the failure |
| 254 | * counter 'x' as follows: |
| 255 | * |
| 256 | * [0. 5) -> 0 |
| 257 | * 5 -> 30 |
| 258 | * [6, 10) -> 0 |
| 259 | * [11, 30) -> 30 |
| 260 | * [30, 140) -> 30 * (2^((x - 30)/10)) |
| 261 | * [140, inf) -> 1 day |
| 262 | * |
| 263 | */ |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 264 | uint32_t GateKeeper::ComputeRetryTimeout(const failure_record_t *record) { |
Andres Morales | b6a5cd7 | 2015-06-03 17:59:17 -0700 | [diff] [blame] | 265 | static const int failure_timeout_ms = 30000; |
| 266 | if (record->failure_counter == 0) return 0; |
| 267 | |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 268 | if (record->failure_counter > 0 && record->failure_counter <= 10) { |
| 269 | if (record->failure_counter % 5 == 0) { |
Andres Morales | b6a5cd7 | 2015-06-03 17:59:17 -0700 | [diff] [blame] | 270 | return failure_timeout_ms; |
Andres Morales | d14f472 | 2015-12-18 11:57:34 -0800 | [diff] [blame] | 271 | } else { |
| 272 | return 0; |
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 | } else if (record->failure_counter < 30) { |
Andres Morales | b6a5cd7 | 2015-06-03 17:59:17 -0700 | [diff] [blame] | 275 | return failure_timeout_ms; |
Andres Morales | f436c80 | 2015-10-08 12:21:25 -0700 | [diff] [blame] | 276 | } else if (record->failure_counter < 140) { |
| 277 | return failure_timeout_ms << ((record->failure_counter - 30) / 10); |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 278 | } |
Andres Morales | f436c80 | 2015-10-08 12:21:25 -0700 | [diff] [blame] | 279 | |
| 280 | return DAY_IN_MS; |
Andres Morales | ac80818 | 2015-02-26 14:11:04 -0800 | [diff] [blame] | 281 | } |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 282 | |
Andres Morales | a623e45 | 2015-05-27 12:26:59 -0700 | [diff] [blame] | 283 | bool GateKeeper::ThrottleRequest(uint32_t uid, uint64_t timestamp, |
Andres Morales | 48a4f83 | 2015-05-29 10:13:31 -0700 | [diff] [blame] | 284 | failure_record_t *record, bool secure, GateKeeperMessage *response) { |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 285 | |
| 286 | uint64_t last_checked = record->last_checked_timestamp; |
| 287 | uint32_t timeout = ComputeRetryTimeout(record); |
| 288 | |
| 289 | if (timeout > 0) { |
| 290 | // we have a pending timeout |
| 291 | if (timestamp < last_checked + timeout && timestamp > last_checked) { |
| 292 | // attempt before timeout expired, return remaining time |
| 293 | response->SetRetryTimeout(timeout - (timestamp - last_checked)); |
| 294 | return true; |
| 295 | } else if (timestamp <= last_checked) { |
| 296 | // device was rebooted or timer reset, don't count as new failure but |
| 297 | // reset timeout |
| 298 | record->last_checked_timestamp = timestamp; |
Andres Morales | 48a4f83 | 2015-05-29 10:13:31 -0700 | [diff] [blame] | 299 | if (!WriteFailureRecord(uid, record, secure)) { |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 300 | response->error = ERROR_UNKNOWN; |
| 301 | return true; |
| 302 | } |
| 303 | response->SetRetryTimeout(timeout); |
| 304 | return true; |
| 305 | } |
| 306 | } |
| 307 | |
| 308 | return false; |
| 309 | } |
| 310 | |
| 311 | 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] | 312 | failure_record_t *record, bool secure) { |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 313 | record->secure_user_id = user_id; |
| 314 | record->failure_counter++; |
| 315 | record->last_checked_timestamp = timestamp; |
| 316 | |
Andres Morales | 48a4f83 | 2015-05-29 10:13:31 -0700 | [diff] [blame] | 317 | return WriteFailureRecord(uid, record, secure); |
Andres Morales | aedf605 | 2015-05-14 13:10:30 -0700 | [diff] [blame] | 318 | } |
| 319 | } // namespace gatekeeper |
| 320 | |