FileFinder.cpp

   1 //
   2 // Copyright 2011 The Android Open Source Project
   3 //
   4
   5 // File Finder implementation.
   6 // Implementation for the functions declared and documented in FileFinder.h
   7
   8 #include <utils/Vector.h>
   9 #include <utils/String8.h>
  10 #include <utils/KeyedVector.h>
  11
  12 #include <iostream>
  13 #include <dirent.h>
  14 #include <sys/stat.h>
  15
  16 #include "DirectoryWalker.h"
  17 #include "FileFinder.h"
  18
  19 //#define DEBUG
  20
  21 using android::String8;
  22 using std::cout;
  23 using std::endl;
  24
  25 // Private function to check whether a file is a directory or not
  26 bool isDirectory(const char* filename) {
  27     struct stat fileStat;
  28     if (stat(filename, &fileStat) == -1) {
  29         return false;
  30     }
  31     return(S_ISDIR(fileStat.st_mode));
  32 }
  33
  34
  35 // Private function to check whether a file is a regular file or not
  36 bool isFile(const char* filename) {
  37     struct stat fileStat;
  38     if (stat(filename, &fileStat) == -1) {
  39         return false;
  40     }
  41     return(S_ISREG(fileStat.st_mode));
  42 }
  43
  44 bool SystemFileFinder::findFiles(String8 basePath, Vector<String8>& extensions,
  45                                  KeyedVector<String8,time_t>& fileStore,
  46                                  DirectoryWalker* dw)
  47 {
  48     // Scan the directory pointed to by basePath
  49     // check files and recurse into subdirectories.
  50     if (!dw->openDir(basePath)) {
  51         return false;
  52     }
  53 #ifdef DEBUG
  54     cout << "FileFinder looking in " << basePath << endl;
  55 #endif // DEBUG
  56     /*
  57      *  Go through all directory entries. Check each file using checkAndAddFile
  58      *  and recurse into sub-directories.
  59      */
  60     struct dirent* entry;
  61     while ((entry = dw->nextEntry()) != NULL) {
  62         String8 entryName(entry->d_name);
  63         if (entry->d_name[0] == '.') // Skip hidden files and directories
  64             continue;
  65
  66         String8 fullPath = basePath.appendPathCopy(entryName);
  67         // If this entry is a directory we'll recurse into it
  68         if (isDirectory(fullPath.string()) ) {
  69             DirectoryWalker* copy = dw->clone();
  70             findFiles(fullPath, extensions, fileStore,copy);
  71             delete copy;
  72         }
  73
  74         // If this entry is a file, we'll pass it over to checkAndAddFile
  75         if (isFile(fullPath.string()) ) {
  76             checkAndAddFile(fullPath,dw->entryStats(),extensions,fileStore);
  77         }
  78     }
  79
  80     // Clean up
  81     dw->closeDir();
  82
  83     return true;
  84 }
  85
  86 void SystemFileFinder::checkAndAddFile(String8 path, const struct stat* stats,
  87                                        Vector<String8>& extensions,
  88                                        KeyedVector<String8,time_t>& fileStore)
  89 {
  90 #ifdef DEBUG
  91     cout << "Checking file " << path << "...";
  92 #endif // DEBUG
  93     // Loop over the extensions, checking for a match
  94     bool done = false;
  95     String8 ext(path.getPathExtension());
  96     ext.toLower();
  97     for (size_t i = 0; i < extensions.size() && !done; ++i) {
  98         String8 ext2 = extensions[i].getPathExtension();
  99         ext2.toLower();
 100         // Compare the extensions. If a match is found, add to storage.
 101         if (ext == ext2) {
 102 #ifdef DEBUG
 103             cout << "Match";
 104 #endif // DEBUG
 105             done = true;
 106             fileStore.add(path,stats->st_mtime);
 107         }
 108     }
 109 #ifdef DEBUG
 110     cout << endl;
 111 #endif //DEBUG
 112 }
 113