chrome/utility/safe_browsing/mac/dmg_iterator.cc

   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
   3 // found in the LICENSE file.
   4
   5 #include "chrome/utility/safe_browsing/mac/dmg_iterator.h"
   6
   7 #include "chrome/utility/safe_browsing/mac/hfs.h"
   8 #include "chrome/utility/safe_browsing/mac/read_stream.h"
   9
  10 namespace safe_browsing {
  11 namespace dmg {
  12
  13 DMGIterator::DMGIterator(ReadStream* stream)
  14     : udif_(stream),
  15       partitions_(),
  16       current_partition_(0),
  17       hfs_() {
  18 }
  19
  20 DMGIterator::~DMGIterator() {}
  21
  22 bool DMGIterator::Open() {
  23   if (!udif_.Parse())
  24     return false;
  25
  26   // Collect all the HFS partitions up-front. The data are accessed lazily, so
  27   // this is relatively inexpensive.
  28   for (size_t i = 0; i < udif_.GetNumberOfPartitions(); ++i) {
  29     if (udif_.GetPartitionType(i) == "Apple_HFS" ||
  30         udif_.GetPartitionType(i) == "Apple_HFSX") {
  31       partitions_.push_back(udif_.GetPartitionReadStream(i).Pass());
  32     }
  33   }
  34
  35   return partitions_.size() > 0;
  36 }
  37
  38 bool DMGIterator::Next() {
  39   // Iterate through all the HFS partitions in the DMG file.
  40   for (; current_partition_ < partitions_.size(); ++current_partition_) {
  41     if (!hfs_) {
  42       hfs_.reset(new HFSIterator(partitions_[current_partition_]));
  43       if (!hfs_->Open())
  44         continue;
  45     }
  46
  47     // Iterate through the HFS filesystem until a concrete file is found.
  48     while (true) {
  49       if (!hfs_->Next())
  50         break;
  51
  52       // Skip directories and symlinks.
  53       if (hfs_->IsDirectory() || hfs_->IsSymbolicLink())
  54         continue;
  55
  56       // Hard links are not supported by the HFSIterator.
  57       if (hfs_->IsHardLink())
  58         continue;
  59
  60       // Decmpfs compression is not supported by the HFSIterator.
  61       if (hfs_->IsDecmpfsCompressed())
  62         continue;
  63
  64       // This must be a normal file!
  65       return true;
  66     }
  67
  68     hfs_.reset();
  69   }
  70
  71   return false;
  72 }
  73
  74 base::string16 DMGIterator::GetPath() {
  75   return hfs_->GetPath();
  76 }
  77
  78 scoped_ptr<ReadStream> DMGIterator::GetReadStream() {
  79   return hfs_->GetReadStream();
  80 }
  81
  82 }  // namespace dmg
  83 }  // namespace safe_browsing