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 #ifndef BASE_FILES_DIR_READER_LINUX_H_ 6 #define BASE_FILES_DIR_READER_LINUX_H_ 7 8 #include <errno.h> 9 #include <fcntl.h> 10 #include <stddef.h> 11 #include <stdint.h> 12 #include <sys/syscall.h> 13 #include <unistd.h> 14 15 #include "base/logging.h" 16 #include "base/macros.h" 17 #include "base/posix/eintr_wrapper.h" 18 19 // See the comments in dir_reader_posix.h about this. 20 21 namespace base { 22 23 struct linux_dirent { 24 uint64_t d_ino; 25 int64_t d_off; 26 unsigned short d_reclen; 27 unsigned char d_type; 28 char d_name[0]; 29 }; 30 31 class DirReaderLinux { 32 public: DirReaderLinux(const char * directory_path)33 explicit DirReaderLinux(const char* directory_path) 34 : fd_(open(directory_path, O_RDONLY | O_DIRECTORY)), 35 offset_(0), 36 size_(0) { 37 memset(buf_, 0, sizeof(buf_)); 38 } 39 ~DirReaderLinux()40 ~DirReaderLinux() { 41 if (fd_ >= 0) { 42 if (IGNORE_EINTR(close(fd_))) 43 RAW_LOG(ERROR, "Failed to close directory handle"); 44 } 45 } 46 IsValid()47 bool IsValid() const { 48 return fd_ >= 0; 49 } 50 51 // Move to the next entry returning false if the iteration is complete. Next()52 bool Next() { 53 if (size_) { 54 linux_dirent* dirent = reinterpret_cast<linux_dirent*>(&buf_[offset_]); 55 offset_ += dirent->d_reclen; 56 } 57 58 if (offset_ != size_) 59 return true; 60 61 const int r = syscall(__NR_getdents64, fd_, buf_, sizeof(buf_)); 62 if (r == 0) 63 return false; 64 if (r == -1) { 65 DPLOG(FATAL) << "getdents64 returned an error: " << errno; 66 return false; 67 } 68 size_ = r; 69 offset_ = 0; 70 return true; 71 } 72 name()73 const char* name() const { 74 if (!size_) 75 return nullptr; 76 77 const linux_dirent* dirent = 78 reinterpret_cast<const linux_dirent*>(&buf_[offset_]); 79 return dirent->d_name; 80 } 81 fd()82 int fd() const { 83 return fd_; 84 } 85 IsFallback()86 static bool IsFallback() { 87 return false; 88 } 89 90 private: 91 const int fd_; 92 unsigned char buf_[512]; 93 size_t offset_; 94 size_t size_; 95 96 DISALLOW_COPY_AND_ASSIGN(DirReaderLinux); 97 }; 98 99 } // namespace base 100 101 #endif // BASE_FILES_DIR_READER_LINUX_H_ 102