OLD | NEW |
1 // Copyright (c) 2010 The Chromium OS Authors. All rights reserved. | 1 // Copyright (c) 2010 The Chromium OS Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include <sys/stat.h> | 5 #include <sys/stat.h> |
6 #include <sys/types.h> | 6 #include <sys/types.h> |
7 #include <unistd.h> | 7 #include <unistd.h> |
8 #include <set> | 8 #include <set> |
9 #include <string> | 9 #include <string> |
10 #include <vector> | 10 #include <vector> |
11 #include <gtest/gtest.h> | 11 #include <gtest/gtest.h> |
12 #include "base/basictypes.h" | 12 #include "base/basictypes.h" |
13 #include "update_engine/extent_mapper.h" | 13 #include "update_engine/extent_mapper.h" |
| 14 #include "update_engine/graph_types.h" |
14 #include "update_engine/utils.h" | 15 #include "update_engine/utils.h" |
15 | 16 |
16 using std::set; | 17 using std::set; |
17 using std::string; | 18 using std::string; |
18 using std::vector; | 19 using std::vector; |
19 | 20 |
20 namespace chromeos_update_engine { | 21 namespace chromeos_update_engine { |
21 | 22 |
22 class ExtentMapperTest : public ::testing::Test {}; | 23 class ExtentMapperTest : public ::testing::Test {}; |
23 | 24 |
24 TEST(ExtentMapperTest, RunAsRootSimpleTest) { | 25 TEST(ExtentMapperTest, RunAsRootSimpleTest) { |
25 // It's hard to have a concrete test for extent mapping without including | 26 // It's hard to have a concrete test for extent mapping without including |
26 // a specific filesystem image. | 27 // a specific filesystem image. |
27 // In lieu of this, we do a weak test: make sure the extents of the unittest | 28 // In lieu of this, we do a weak test: make sure the extents of the unittest |
28 // executable are consistent and they match with the size of the file. | 29 // executable are consistent and they match with the size of the file. |
29 const string kFilename = "/proc/self/exe"; | 30 const string kFilename = "/proc/self/exe"; |
30 const off_t kBlockSize = 4096; | |
31 | 31 |
| 32 uint32 block_size = 0; |
| 33 EXPECT_TRUE(extent_mapper::GetFilesystemBlockSize(kFilename, &block_size)); |
| 34 EXPECT_GT(block_size, 0); |
| 35 |
32 vector<Extent> extents; | 36 vector<Extent> extents; |
33 | 37 |
34 ASSERT_TRUE(extent_mapper::ExtentsForFileFibmap(kFilename, &extents)); | 38 ASSERT_TRUE(extent_mapper::ExtentsForFileFibmap(kFilename, &extents)); |
35 | 39 |
36 EXPECT_FALSE(extents.empty()); | 40 EXPECT_FALSE(extents.empty()); |
37 set<uint64> blocks; | 41 set<uint64> blocks; |
38 | 42 |
39 for (vector<Extent>::const_iterator it = extents.begin(); | 43 for (vector<Extent>::const_iterator it = extents.begin(); |
40 it != extents.end(); ++it) { | 44 it != extents.end(); ++it) { |
41 for (uint64 block = it->start_block(); | 45 for (uint64 block = it->start_block(); |
42 block < it->start_block() + it->num_blocks(); | 46 block < it->start_block() + it->num_blocks(); |
43 block++) { | 47 block++) { |
44 EXPECT_FALSE(utils::SetContainsKey(blocks, block)); | 48 EXPECT_FALSE(utils::SetContainsKey(blocks, block)); |
45 blocks.insert(block); | 49 blocks.insert(block); |
46 } | 50 } |
47 } | 51 } |
48 | 52 |
49 struct stat stbuf; | 53 struct stat stbuf; |
50 EXPECT_EQ(0, stat(kFilename.c_str(), &stbuf)); | 54 EXPECT_EQ(0, stat(kFilename.c_str(), &stbuf)); |
51 EXPECT_EQ(blocks.size(), (stbuf.st_size + kBlockSize - 1)/kBlockSize); | 55 EXPECT_EQ(blocks.size(), (stbuf.st_size + block_size - 1)/block_size); |
| 56 } |
| 57 |
| 58 TEST(ExtentMapperTest, RunAsRootSparseFileTest) { |
| 59 // Create sparse file with one real block, then two sparse ones, then a real |
| 60 // block at the end. |
| 61 const char tmp_name_template[] = |
| 62 "/tmp/ExtentMapperTest.RunAsRootSparseFileTest.XXXXXX"; |
| 63 char buf[sizeof(tmp_name_template)]; |
| 64 strncpy(buf, tmp_name_template, sizeof(buf)); |
| 65 COMPILE_ASSERT(sizeof(buf) > 8, buf_size_incorrect); |
| 66 ASSERT_EQ('\0', buf[sizeof(buf) - 1]); |
| 67 |
| 68 int fd = mkstemp(buf); |
| 69 ASSERT_GE(fd, 0); |
| 70 |
| 71 uint32 block_size = 0; |
| 72 EXPECT_TRUE(extent_mapper::GetFilesystemBlockSize(buf, &block_size)); |
| 73 EXPECT_GT(block_size, 0); |
| 74 |
| 75 EXPECT_EQ(1, pwrite(fd, "x", 1, 0)); |
| 76 EXPECT_EQ(1, pwrite(fd, "x", 1, 3 * block_size)); |
| 77 close(fd); |
| 78 |
| 79 vector<Extent> extents; |
| 80 EXPECT_TRUE(extent_mapper::ExtentsForFileFibmap(buf, &extents)); |
| 81 unlink(buf); |
| 82 EXPECT_EQ(3, extents.size()); |
| 83 EXPECT_EQ(1, extents[0].num_blocks()); |
| 84 EXPECT_EQ(2, extents[1].num_blocks()); |
| 85 EXPECT_EQ(1, extents[2].num_blocks()); |
| 86 EXPECT_NE(kSparseHole, extents[0].start_block()); |
| 87 EXPECT_EQ(kSparseHole, extents[1].start_block()); |
| 88 EXPECT_NE(kSparseHole, extents[2].start_block()); |
| 89 EXPECT_NE(extents[2].start_block(), extents[0].start_block()); |
52 } | 90 } |
53 | 91 |
54 } // namespace chromeos_update_engine | 92 } // namespace chromeos_update_engine |
OLD | NEW |