The Tor Browser: ipc/chromium/src/base/dir_reader

Cloned upstream origin tor-browser at tor-browser-31.3.0esr-4.5-1-build1
revision ID fc1c9ff7c1b2defdbc039f12214767608f46423f for hacking purpose.

     1 // Copyright (c) 2010 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.

     5 #ifndef BASE_DIR_READER_LINUX_H_

     6 #define BASE_DIR_READER_LINUX_H_

     7 #pragma once

     9 #include <errno.h>

    10 #include <fcntl.h>

    11 #include <stdint.h>

    12 #include <sys/syscall.h>

    13 #include <unistd.h>

    15 #include "base/logging.h"

    16 #include "base/eintr_wrapper.h"

    18 // See the comments in dir_reader_posix.h about this.

    20 namespace base {

    22 struct linux_dirent {

    23   uint64_t        d_ino;

    24   int64_t         d_off;

    25   unsigned short  d_reclen;

    26   unsigned char   d_type;

    27   char            d_name[0];

    28 };

    30 class DirReaderLinux {

    31  public:

    32   explicit DirReaderLinux(const char* directory_path)

    33       : fd_(open(directory_path, O_RDONLY | O_DIRECTORY)),

    34         offset_(0),

    35         size_(0) {

    36     memset(buf_, 0, sizeof(buf_));

    37   }

    39   ~DirReaderLinux() {

    40     if (fd_ >= 0) {

    41       if (HANDLE_EINTR(close(fd_)))

    42         DLOG(ERROR) << "Failed to close directory handle";

    43     }

    44   }

    46   bool IsValid() const {

    47     return fd_ >= 0;

    48   }

    50   // Move to the next entry returning false if the iteration is complete.

    51   bool Next() {

    52     if (size_) {

    53       linux_dirent* dirent = reinterpret_cast<linux_dirent*>(&buf_[offset_]);

    54       offset_ += dirent->d_reclen;

    55     }

    57     if (offset_ != size_)

    58       return true;

    60     const int r = syscall(__NR_getdents64, fd_, buf_, sizeof(buf_));

    61     if (r == 0)

    62       return false;

    63     if (r == -1) {

    64       DLOG(ERROR) << "getdents64 returned an error: " << errno;

    65       return false;

    66     }

    67     size_ = r;

    68     offset_ = 0;

    69     return true;

    70   }

    72   const char* name() const {

    73     if (!size_)

    74       return NULL;

    76     const linux_dirent* dirent =

    77         reinterpret_cast<const linux_dirent*>(&buf_[offset_]);

    78     return dirent->d_name;

    79   }

    81   int fd() const {

    82     return fd_;

    83   }

    85   static bool IsFallback() {

    86     return false;

    87   }

    89  private:

    90   const int fd_;

    91   unsigned char buf_[512];

    92   size_t offset_, size_;

    94   DISALLOW_COPY_AND_ASSIGN(DirReaderLinux);

    95 };

    97 }  // namespace base

    99 #endif // BASE_DIR_READER_LINUX_H_

The Tor Browser / file revision