blob: a338e7f9f31f67d21bf0800bc4d4d0304b36004c (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
// Copyright 2017 The Chromium OS 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 SRC_MEMORY_STREAM_H_
#define SRC_MEMORY_STREAM_H_
#include <string>
#include <utility>
#include "puffin/common.h"
#include "puffin/stream.h"
namespace puffin {
// A very simple class for reading and writing into memory.
class MemoryStream : public StreamInterface {
public:
~MemoryStream() override = default;
// Creates a stream for reading.
static UniqueStreamPtr CreateForRead(const Buffer& memory);
// Creates a stream for writing. This function will clear the |memory| if
// |clear| is true. An instance of this class does not retain the ownership of
// the |memory|.
static UniqueStreamPtr CreateForWrite(Buffer* memory);
bool GetSize(uint64_t* size) const override;
bool GetOffset(uint64_t* offset) const override;
bool Seek(uint64_t offset) override;
bool Read(void* buffer, size_t length) override;
bool Write(const void* buffer, size_t length) override;
bool Close() override;
private:
// Ctor. Exactly one of the |read_memory| or |write_memory| should be nullptr.
MemoryStream(const Buffer* read_memory, Buffer* write_memory);
// The memory buffer for reading.
const Buffer* read_memory_;
// The memory buffer for writing. It can grow as we write into it.
Buffer* write_memory_;
// The current offset.
uint64_t offset_;
// True if the stream is open.
bool open_;
DISALLOW_COPY_AND_ASSIGN(MemoryStream);
};
} // namespace puffin
#endif // SRC_MEMORY_STREAM_H_
|
