1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. 2 // Use of this source code is governed by a BSD-style license that can be 3 // found in the LICENSE file. 4 5 #include "base/files/dir_reader_posix.h" 6 7 #include <fcntl.h> 8 #include <stdio.h> 9 #include <stdlib.h> 10 #include <string.h> 11 #include <unistd.h> 12 13 #include "base/logging.h" 14 #include "testing/gtest/include/gtest/gtest.h" 15 16 #if defined(OS_ANDROID) 17 #include "base/os_compat_android.h" 18 #endif 19 20 namespace base { 21 22 TEST(DirReaderPosixUnittest, Read) { 23 static const unsigned kNumFiles = 100; 24 25 if (DirReaderPosix::IsFallback()) 26 return; 27 28 char kDirTemplate[] = "/tmp/org.chromium.dir-reader-posix-XXXXXX"; 29 const char* dir = mkdtemp(kDirTemplate); 30 ASSERT_TRUE(dir); 31 32 const int prev_wd = open(".", O_RDONLY | O_DIRECTORY); 33 DCHECK_GE(prev_wd, 0); 34 35 PCHECK(chdir(dir) == 0); 36 37 for (unsigned i = 0; i < kNumFiles; i++) { 38 char buf[16]; 39 snprintf(buf, sizeof(buf), "%d", i); 40 const int fd = open(buf, O_CREAT | O_RDONLY | O_EXCL, 0600); 41 PCHECK(fd >= 0); 42 PCHECK(close(fd) == 0); 43 } 44 45 std::set<unsigned> seen; 46 47 DirReaderPosix reader(dir); 48 EXPECT_TRUE(reader.IsValid()); 49 50 if (!reader.IsValid()) 51 return; 52 53 bool seen_dot = false, seen_dotdot = false; 54 55 for (; reader.Next(); ) { 56 if (strcmp(reader.name(), ".") == 0) { 57 seen_dot = true; 58 continue; 59 } 60 if (strcmp(reader.name(), "..") == 0) { 61 seen_dotdot = true; 62 continue; 63 } 64 65 SCOPED_TRACE(testing::Message() << "reader.name(): " << reader.name()); 66 67 char *endptr; 68 const unsigned long value = strtoul(reader.name(), &endptr, 10); 69 70 EXPECT_FALSE(*endptr); 71 EXPECT_LT(value, kNumFiles); 72 EXPECT_EQ(0u, seen.count(value)); 73 seen.insert(value); 74 } 75 76 for (unsigned i = 0; i < kNumFiles; i++) { 77 char buf[16]; 78 snprintf(buf, sizeof(buf), "%d", i); 79 PCHECK(unlink(buf) == 0); 80 } 81 82 PCHECK(rmdir(dir) == 0); 83 84 PCHECK(fchdir(prev_wd) == 0); 85 PCHECK(close(prev_wd) == 0); 86 87 EXPECT_TRUE(seen_dot); 88 EXPECT_TRUE(seen_dotdot); 89 EXPECT_EQ(kNumFiles, seen.size()); 90 } 91 92 } // namespace base 93