blob: f4ebc0ce01c37eea89f1b6a70d467fb732dc7a0b [file] [log] [blame]
// Copyright 2015 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CHROME_UTILITY_SAFE_BROWSING_MAC_DMG_ITERATOR_H_
#define CHROME_UTILITY_SAFE_BROWSING_MAC_DMG_ITERATOR_H_
#include <stddef.h>
#include <memory>
#include "base/macros.h"
#include "base/memory/scoped_vector.h"
#include "base/strings/string16.h"
#include "chrome/utility/safe_browsing/mac/udif.h"
namespace safe_browsing {
namespace dmg {
class HFSIterator;
class ReadStream;
// DMGIterator provides iterator access over all of the concrete files located
// on HFS+ and HFSX volumes in a UDIF/DMG file. This class maintains the
// limitations of its composed components.
class DMGIterator {
public:
// Creates a DMGIterator from a ReadStream. In most cases, this will be a
// FileReadStream opened from a DMG file. This does not take ownership
// of the stream.
explicit DMGIterator(ReadStream* stream);
~DMGIterator();
// Opens the DMG file for iteration. This must be called before any other
// method. If this returns false, it is illegal to call any other methods
// on this class. If this returns true, the iterator is advanced to an
// invalid element before the first item.
bool Open();
// Advances the iterator to the next file item. Returns true on success
// and false on end-of-iterator.
bool Next();
// Returns the full path in a DMG filesystem to the current file item.
base::string16 GetPath();
// Returns a ReadStream for the current file item.
std::unique_ptr<ReadStream> GetReadStream();
private:
UDIFParser udif_; // The UDIF parser that accesses the partitions.
ScopedVector<ReadStream> partitions_; // Streams for all the HFS partitions.
size_t current_partition_; // The index in |partitions_| of the current one.
std::unique_ptr<HFSIterator>
hfs_; // The HFSIterator for |current_partition_|.
DISALLOW_COPY_AND_ASSIGN(DMGIterator);
};
} // namespace dmg
} // namespace safe_browsing
#endif // CHROME_UTILITY_SAFE_BROWSING_MAC_DMG_ITERATOR_H_