1 /*
2  * Copyright (C) 2017 The Android Open Source Project
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *      http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #ifndef AAPT_IO_BIGBUFFERSTREAM_H
18 #define AAPT_IO_BIGBUFFERSTREAM_H
19 
20 #include "io/Io.h"
21 #include "util/BigBuffer.h"
22 
23 namespace aapt {
24 namespace io {
25 
26 class BigBufferInputStream : public KnownSizeInputStream {
27  public:
BigBufferInputStream(const BigBuffer * buffer)28   inline explicit BigBufferInputStream(const BigBuffer* buffer)
29       : buffer_(buffer), iter_(buffer->begin()) {
30   }
31   virtual ~BigBufferInputStream() = default;
32 
33   bool Next(const void** data, size_t* size) override;
34 
35   void BackUp(size_t count) override;
36 
37   bool CanRewind() const override;
38 
39   bool Rewind() override;
40 
41   size_t ByteCount() const override;
42 
43   bool HadError() const override;
44 
45   size_t TotalSize() const override;
46 
47  private:
48   DISALLOW_COPY_AND_ASSIGN(BigBufferInputStream);
49 
50   const BigBuffer* buffer_;
51   BigBuffer::const_iterator iter_;
52   size_t offset_ = 0;
53   size_t bytes_read_ = 0;
54 };
55 
56 class BigBufferOutputStream : public OutputStream {
57  public:
BigBufferOutputStream(BigBuffer * buffer)58   inline explicit BigBufferOutputStream(BigBuffer* buffer) : buffer_(buffer) {
59   }
60   virtual ~BigBufferOutputStream() = default;
61 
62   bool Next(void** data, size_t* size) override;
63 
64   void BackUp(size_t count) override;
65 
66   size_t ByteCount() const override;
67 
68   bool HadError() const override;
69 
70  private:
71   DISALLOW_COPY_AND_ASSIGN(BigBufferOutputStream);
72 
73   BigBuffer* buffer_;
74 };
75 
76 }  // namespace io
77 }  // namespace aapt
78 
79 #endif  // AAPT_IO_BIGBUFFERSTREAM_H
80