1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
|
// Copyright (c) 2012 The Chromium OS Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
// Unit tests for SecureBlob.
#include "secure_blob.h"
#include <base/logging.h>
#include <gtest/gtest.h>
#include "utility.h"
namespace chromeos {
using std::string;
class SecureBlobTest : public ::testing::Test {
public:
SecureBlobTest() { }
virtual ~SecureBlobTest() { }
static bool FindBlobInBlob(const chromeos::Blob& haystack,
const chromeos::Blob& needle) {
if (needle.size() > haystack.size()) {
return false;
}
for (unsigned int start = 0; start <= (haystack.size() - needle.size());
start++) {
if (chromeos::SafeMemcmp(&haystack[start], &needle[0],
needle.size()) == 0) {
return true;
}
}
return false;
}
private:
DISALLOW_COPY_AND_ASSIGN(SecureBlobTest);
};
TEST_F(SecureBlobTest, AllocationSizeTest) {
// Check that allocating a SecureBlob of a specified size works
SecureBlob blob(32);
EXPECT_EQ(32, blob.size());
}
TEST_F(SecureBlobTest, AllocationCopyTest) {
// Check that allocating a SecureBlob with an iterator works
unsigned char from_data[32];
for (unsigned int i = 0; i < sizeof(from_data); i++) {
from_data[i] = i;
}
SecureBlob blob(from_data, sizeof(from_data));
EXPECT_EQ(sizeof(from_data), blob.size());
for (unsigned int i = 0; i < sizeof(from_data); i++) {
EXPECT_EQ(from_data[i], blob[i]);
}
}
TEST_F(SecureBlobTest, IteratorConstructorTest) {
// Check that allocating a SecureBlob with an iterator works
chromeos::Blob from_blob(32);
for (unsigned int i = 0; i < from_blob.size(); i++) {
from_blob[i] = i;
}
SecureBlob blob(from_blob.begin(), from_blob.end());
EXPECT_EQ(from_blob.size(), blob.size());
EXPECT_TRUE(SecureBlobTest::FindBlobInBlob(from_blob, blob));
}
TEST_F(SecureBlobTest, ResizeTest) {
// Check that resizing a SecureBlob wipes the excess memory. The test assumes
// that resize() down by one will not re-allocate the memory, so the last byte
// will still be part of the SecureBlob's allocation
unsigned int length = 1024;
SecureBlob blob(length);
void* original_data = blob.data();
for (unsigned int i = 0; i < length; i++) {
blob[i] = i;
}
blob.resize(length - 1);
EXPECT_EQ(original_data, blob.data());
EXPECT_EQ(length - 1, blob.size());
EXPECT_EQ(0, static_cast<unsigned char*>(blob.data())[length - 1]);
}
} // namespace cryptohome
|
