1 // LimitedStreams.h

     2 

     3 #ifndef __LIMITEDSTREAMS_H

     4 #define __LIMITEDSTREAMS_H

     5 

     6 #include "../../Common/MyCom.h"

     7 #include "../IStream.h"

     8 

     9 class CLimitedSequentialInStream: 

    10   public ISequentialInStream,

    11   public CMyUnknownImp

    12 {

    13   CMyComPtr<ISequentialInStream> _stream;

    14   UInt64 _size;

    15   UInt64 _pos;

    16   bool _wasFinished;

    17 public:

    18   void SetStream(ISequentialInStream *stream) { _stream = stream; }

    19   void Init(UInt64 streamSize)  

    20   { 

    21     _size = streamSize; 

    22     _pos = 0; 

    23     _wasFinished = false; 

    24   }

    25  

    26   MY_UNKNOWN_IMP

    27 

    28   STDMETHOD(Read)(void *data, UInt32 size, UInt32 *processedSize);

    29   UInt64 GetSize() const { return _pos; }

    30   bool WasFinished() const { return _wasFinished; }

    31 };

    32 

    33 #endif