OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium 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 "chrome/utility/safe_browsing/mac/hfs.h" | 5 #include "chrome/utility/safe_browsing/mac/hfs.h" |
6 | 6 |
7 #include <stddef.h> | 7 #include <stddef.h> |
8 #include <stdint.h> | 8 #include <stdint.h> |
9 | 9 |
10 #include "base/files/file.h" | 10 #include "base/files/file.h" |
(...skipping 113 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
124 return true; | 124 return true; |
125 } | 125 } |
126 } | 126 } |
127 return false; | 127 return false; |
128 } | 128 } |
129 | 129 |
130 HFSIterator* hfs_reader() { return hfs_reader_.get(); } | 130 HFSIterator* hfs_reader() { return hfs_reader_.get(); } |
131 | 131 |
132 private: | 132 private: |
133 base::File hfs_file_; | 133 base::File hfs_file_; |
134 scoped_ptr<FileReadStream> hfs_stream_; | 134 std::unique_ptr<FileReadStream> hfs_stream_; |
135 scoped_ptr<HFSIterator> hfs_reader_; | 135 std::unique_ptr<HFSIterator> hfs_reader_; |
136 }; | 136 }; |
137 | 137 |
138 TEST_P(HFSFileReadTest, ReadReadme) { | 138 TEST_P(HFSFileReadTest, ReadReadme) { |
139 ASSERT_TRUE(GoToFile("README.txt")); | 139 ASSERT_TRUE(GoToFile("README.txt")); |
140 | 140 |
141 scoped_ptr<ReadStream> stream = hfs_reader()->GetReadStream(); | 141 std::unique_ptr<ReadStream> stream = hfs_reader()->GetReadStream(); |
142 ASSERT_TRUE(stream.get()); | 142 ASSERT_TRUE(stream.get()); |
143 | 143 |
144 EXPECT_FALSE(hfs_reader()->IsSymbolicLink()); | 144 EXPECT_FALSE(hfs_reader()->IsSymbolicLink()); |
145 EXPECT_FALSE(hfs_reader()->IsHardLink()); | 145 EXPECT_FALSE(hfs_reader()->IsHardLink()); |
146 EXPECT_FALSE(hfs_reader()->IsDecmpfsCompressed()); | 146 EXPECT_FALSE(hfs_reader()->IsDecmpfsCompressed()); |
147 | 147 |
148 std::vector<uint8_t> buffer(4, 0); | 148 std::vector<uint8_t> buffer(4, 0); |
149 | 149 |
150 // Read the first four bytes. | 150 // Read the first four bytes. |
151 EXPECT_TRUE(stream->ReadExact(&buffer[0], buffer.size())); | 151 EXPECT_TRUE(stream->ReadExact(&buffer[0], buffer.size())); |
152 const uint8_t expected[] = { 'T', 'h', 'i', 's' }; | 152 const uint8_t expected[] = { 'T', 'h', 'i', 's' }; |
153 EXPECT_EQ(0, memcmp(expected, &buffer[0], sizeof(expected))); | 153 EXPECT_EQ(0, memcmp(expected, &buffer[0], sizeof(expected))); |
154 buffer.clear(); | 154 buffer.clear(); |
155 | 155 |
156 // Rewind back to the start. | 156 // Rewind back to the start. |
157 EXPECT_EQ(0, stream->Seek(0, SEEK_SET)); | 157 EXPECT_EQ(0, stream->Seek(0, SEEK_SET)); |
158 | 158 |
159 // Read the entire file now. | 159 // Read the entire file now. |
160 EXPECT_TRUE(test::ReadEntireStream(stream.get(), &buffer)); | 160 EXPECT_TRUE(test::ReadEntireStream(stream.get(), &buffer)); |
161 EXPECT_EQ("This is a test HFS+ filesystem generated by " | 161 EXPECT_EQ("This is a test HFS+ filesystem generated by " |
162 "chrome/test/data/safe_browsing/dmg/make_hfs.sh.\n", | 162 "chrome/test/data/safe_browsing/dmg/make_hfs.sh.\n", |
163 base::StringPiece(reinterpret_cast<const char*>(&buffer[0]), | 163 base::StringPiece(reinterpret_cast<const char*>(&buffer[0]), |
164 buffer.size())); | 164 buffer.size())); |
165 EXPECT_EQ(92u, buffer.size()); | 165 EXPECT_EQ(92u, buffer.size()); |
166 } | 166 } |
167 | 167 |
168 TEST_P(HFSFileReadTest, ReadRandom) { | 168 TEST_P(HFSFileReadTest, ReadRandom) { |
169 ASSERT_TRUE(GoToFile("fifth/random")); | 169 ASSERT_TRUE(GoToFile("fifth/random")); |
170 | 170 |
171 scoped_ptr<ReadStream> stream = hfs_reader()->GetReadStream(); | 171 std::unique_ptr<ReadStream> stream = hfs_reader()->GetReadStream(); |
172 ASSERT_TRUE(stream.get()); | 172 ASSERT_TRUE(stream.get()); |
173 | 173 |
174 EXPECT_FALSE(hfs_reader()->IsSymbolicLink()); | 174 EXPECT_FALSE(hfs_reader()->IsSymbolicLink()); |
175 EXPECT_FALSE(hfs_reader()->IsHardLink()); | 175 EXPECT_FALSE(hfs_reader()->IsHardLink()); |
176 EXPECT_FALSE(hfs_reader()->IsDecmpfsCompressed()); | 176 EXPECT_FALSE(hfs_reader()->IsDecmpfsCompressed()); |
177 | 177 |
178 std::vector<uint8_t> buffer; | 178 std::vector<uint8_t> buffer; |
179 EXPECT_TRUE(test::ReadEntireStream(stream.get(), &buffer)); | 179 EXPECT_TRUE(test::ReadEntireStream(stream.get(), &buffer)); |
180 EXPECT_EQ(768u, buffer.size()); | 180 EXPECT_EQ(768u, buffer.size()); |
181 } | 181 } |
182 | 182 |
183 TEST_P(HFSFileReadTest, Symlink) { | 183 TEST_P(HFSFileReadTest, Symlink) { |
184 ASSERT_TRUE(GoToFile("symlink-random")); | 184 ASSERT_TRUE(GoToFile("symlink-random")); |
185 | 185 |
186 scoped_ptr<ReadStream> stream = hfs_reader()->GetReadStream(); | 186 std::unique_ptr<ReadStream> stream = hfs_reader()->GetReadStream(); |
187 ASSERT_TRUE(stream.get()); | 187 ASSERT_TRUE(stream.get()); |
188 | 188 |
189 EXPECT_TRUE(hfs_reader()->IsSymbolicLink()); | 189 EXPECT_TRUE(hfs_reader()->IsSymbolicLink()); |
190 EXPECT_FALSE(hfs_reader()->IsHardLink()); | 190 EXPECT_FALSE(hfs_reader()->IsHardLink()); |
191 EXPECT_FALSE(hfs_reader()->IsDecmpfsCompressed()); | 191 EXPECT_FALSE(hfs_reader()->IsDecmpfsCompressed()); |
192 | 192 |
193 std::vector<uint8_t> buffer; | 193 std::vector<uint8_t> buffer; |
194 EXPECT_TRUE(test::ReadEntireStream(stream.get(), &buffer)); | 194 EXPECT_TRUE(test::ReadEntireStream(stream.get(), &buffer)); |
195 | 195 |
196 EXPECT_EQ("fourth/fifth/random", | 196 EXPECT_EQ("fourth/fifth/random", |
197 base::StringPiece(reinterpret_cast<const char*>(&buffer[0]), | 197 base::StringPiece(reinterpret_cast<const char*>(&buffer[0]), |
198 buffer.size())); | 198 buffer.size())); |
199 } | 199 } |
200 | 200 |
201 TEST_P(HFSFileReadTest, HardLink) { | 201 TEST_P(HFSFileReadTest, HardLink) { |
202 ASSERT_TRUE(GoToFile("unicode_name")); | 202 ASSERT_TRUE(GoToFile("unicode_name")); |
203 | 203 |
204 EXPECT_FALSE(hfs_reader()->IsSymbolicLink()); | 204 EXPECT_FALSE(hfs_reader()->IsSymbolicLink()); |
205 EXPECT_TRUE(hfs_reader()->IsHardLink()); | 205 EXPECT_TRUE(hfs_reader()->IsHardLink()); |
206 EXPECT_FALSE(hfs_reader()->IsDecmpfsCompressed()); | 206 EXPECT_FALSE(hfs_reader()->IsDecmpfsCompressed()); |
207 } | 207 } |
208 | 208 |
209 TEST_P(HFSFileReadTest, DecmpfsFile) { | 209 TEST_P(HFSFileReadTest, DecmpfsFile) { |
210 ASSERT_TRUE(GoToFile("first/second/goat-output.txt")); | 210 ASSERT_TRUE(GoToFile("first/second/goat-output.txt")); |
211 | 211 |
212 scoped_ptr<ReadStream> stream = hfs_reader()->GetReadStream(); | 212 std::unique_ptr<ReadStream> stream = hfs_reader()->GetReadStream(); |
213 ASSERT_TRUE(stream.get()); | 213 ASSERT_TRUE(stream.get()); |
214 | 214 |
215 EXPECT_FALSE(hfs_reader()->IsSymbolicLink()); | 215 EXPECT_FALSE(hfs_reader()->IsSymbolicLink()); |
216 EXPECT_FALSE(hfs_reader()->IsHardLink()); | 216 EXPECT_FALSE(hfs_reader()->IsHardLink()); |
217 EXPECT_TRUE(hfs_reader()->IsDecmpfsCompressed()); | 217 EXPECT_TRUE(hfs_reader()->IsDecmpfsCompressed()); |
218 | 218 |
219 std::vector<uint8_t> buffer; | 219 std::vector<uint8_t> buffer; |
220 EXPECT_TRUE(test::ReadEntireStream(stream.get(), &buffer)); | 220 EXPECT_TRUE(test::ReadEntireStream(stream.get(), &buffer)); |
221 EXPECT_EQ(0u, buffer.size()); | 221 EXPECT_EQ(0u, buffer.size()); |
222 } | 222 } |
223 | 223 |
224 INSTANTIATE_TEST_CASE_P(HFSIteratorTest, | 224 INSTANTIATE_TEST_CASE_P(HFSIteratorTest, |
225 HFSFileReadTest, | 225 HFSFileReadTest, |
226 testing::Values( | 226 testing::Values( |
227 "hfs_plus.img", | 227 "hfs_plus.img", |
228 "hfsx_case_sensitive.img")); | 228 "hfsx_case_sensitive.img")); |
229 | 229 |
230 } // namespace | 230 } // namespace |
231 } // namespace dmg | 231 } // namespace dmg |
232 } // namespace safe_browsing | 232 } // namespace safe_browsing |
OLD | NEW |