| /* |
| * Copyright 2016 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| #include <trusty_unittest.h> |
| |
| #include "secure_storage_fake.h" |
| |
| namespace avb { |
| |
| TEST(AvbTest, BasicReadTest) { |
| uint32_t slot = 0; |
| uint64_t value = 0; |
| RollbackIndexRequest request(slot, value); |
| RollbackIndexResponse response; |
| AvbManager avb_manager(new SecureStorageFake); |
| avb_manager.ReadRollbackIndex(request, &response); |
| EXPECT_EQ(AvbError::kNone, response.get_error(), "Read failed"); |
| EXPECT_EQ(1, response.get_value(), "Did not read expected value"); |
| } |
| |
| TEST(AvbTest, ValidWriteTest) { |
| uint32_t slot = 0; |
| uint64_t value = 2; // Read always returns 1 |
| RollbackIndexRequest request(slot, value); |
| RollbackIndexResponse response; |
| AvbManager avb_manager(new SecureStorageFake); |
| avb_manager.WriteRollbackIndex(request, &response); |
| EXPECT_EQ(AvbError::kNone, response.get_error(), "Write failed"); |
| EXPECT_EQ(2, response.get_value(), "Did not write expected value"); |
| } |
| |
| TEST(AvbTest, InvalidWriteTest) { |
| uint32_t slot = 0; |
| uint64_t value = 0; // Read always returns 1 |
| RollbackIndexRequest request(slot, value); |
| RollbackIndexResponse response; |
| AvbManager avb_manager(new SecureStorageFake); |
| avb_manager.WriteRollbackIndex(request, &response); |
| EXPECT_EQ(AvbError::kInvalid, response.get_error(), |
| "Allowed writing index value less than existing value"); |
| EXPECT_EQ(1, response.get_value(), |
| "Did not read expected value after failed write"); |
| } |
| |
| TEST(AvbTest, SlotUpperBitsSetTest) { |
| uint32_t slot = 0x00010000; |
| uint64_t value = 0; |
| RollbackIndexRequest request(slot, value); |
| RollbackIndexResponse response; |
| AvbManager avb_manager(new SecureStorageFake); |
| avb_manager.ReadRollbackIndex(request, &response); |
| EXPECT_EQ(AvbError::kInvalid, response.get_error(), |
| "Slot was not rejected"); |
| } |
| |
| TEST(AvbTest, SlotMaxValueTest) { |
| uint32_t slot = kRollbackSlotMax + 1; |
| uint64_t value = 0; |
| RollbackIndexRequest request(slot, value); |
| RollbackIndexResponse response; |
| AvbManager avb_manager(new SecureStorageFake); |
| avb_manager.ReadRollbackIndex(request, &response); |
| EXPECT_EQ(AvbError::kInvalid, response.get_error(), |
| "Failed to reject write to slot > max slot"); |
| } |
| |
| TEST(AvbTest, SlotFlagTest) { |
| uint32_t slot = 0x0000f000; |
| uint64_t value = 0; |
| RollbackIndexRequest request(slot, value); |
| RollbackIndexResponse response; |
| AvbManager avb_manager(new SecureStorageFake); |
| avb_manager.ReadRollbackIndex(request, &response); |
| EXPECT_EQ(AvbError::kNone, response.get_error(), |
| "Could not validate 0xf flag"); |
| } |
| |
| } // namespace avb |
| |
| int main() { |
| return RUN_ALL_TESTS() ? 0 : 1; |
| } |