media/filters/file_data_source.h

   1 // Copyright 2012 The Chromium Authors
   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 MEDIA_FILTERS_FILE_DATA_SOURCE_H_
   6 #define MEDIA_FILTERS_FILE_DATA_SOURCE_H_
   7
   8 #include <stdint.h>
   9
  10 #include "base/files/file.h"
  11 #include "base/files/file_path.h"
  12 #include "base/files/memory_mapped_file.h"
  13 #include "media/base/data_source.h"
  14
  15 namespace media {
  16
  17 // Basic data source that treats the URL as a file path, and uses the file
  18 // system to read data for a media pipeline.
  19 class MEDIA_EXPORT FileDataSource : public DataSource {
  20  public:
  21   FileDataSource();
  22
  23   FileDataSource(const FileDataSource&) = delete;
  24   FileDataSource& operator=(const FileDataSource&) = delete;
  25
  26   ~FileDataSource() override;
  27
  28   [[nodiscard]] bool Initialize(const base::FilePath& file_path);
  29   [[nodiscard]] bool Initialize(base::File file);
  30
  31   // Implementation of DataSource.
  32   void Stop() override;
  33   void Abort() override;
  34   void Read(int64_t position,
  35             int size,
  36             uint8_t* data,
  37             DataSource::ReadCB read_cb) override;
  38   [[nodiscard]] bool GetSize(int64_t* size_out) override;
  39   bool IsStreaming() override;
  40   void SetBitrate(int bitrate) override;
  41   bool PassedTimingAllowOriginCheck() final;
  42   bool WouldTaintOrigin() final;
  43
  44   // Unit test helpers. Recreate the object if you want the default behaviour.
  45   void force_read_errors_for_testing() { force_read_errors_ = true; }
  46   void force_streaming_for_testing() { force_streaming_ = true; }
  47   uint64_t bytes_read_for_testing() { return bytes_read_; }
  48   void reset_bytes_read_for_testing() { bytes_read_ = 0; }
  49
  50  private:
  51   base::MemoryMappedFile file_;
  52
  53   bool force_read_errors_;
  54   bool force_streaming_;
  55   uint64_t bytes_read_;
  56 };
  57
  58 }  // namespace media
  59
  60 #endif  // MEDIA_FILTERS_FILE_DATA_SOURCE_H_