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