summaryrefslogtreecommitdiff
path: root/buffer_file.h
blob: 0b9261c18679122238d696e9b2f2ffdcebf13d2f (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
// Copyright 2016 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 _BSDIFF_BUFFER_FILE_H_
#define _BSDIFF_BUFFER_FILE_H_

#include <memory>
#include <vector>

#include "bsdiff/file_interface.h"

namespace bsdiff {

class BufferFile : public FileInterface {
 public:
  // Creates a write only BufferFile based on the underlying |file| passed.
  // The BufferFile will cache all the write in a buffer and write everything
  // to |file| at once upon closing. Read and Seek are not supported.
  // |size| should be the estimated total file size, it is used to reserve
  // buffer space.
  BufferFile(std::unique_ptr<FileInterface> file, size_t size);

  ~BufferFile() override;

  // FileInterface overrides.
  bool Read(void* buf, size_t count, size_t* bytes_read) override;
  bool Write(const void* buf, size_t count, size_t* bytes_written) override;
  bool Seek(off_t pos) override;
  bool Close() override;
  bool GetSize(uint64_t* size) override;

 private:
  // The underlying FileInterace instance.
  std::unique_ptr<FileInterface> file_ = nullptr;
  std::vector<uint8_t> buffer_;
};

}  // namespace bsdiff

#endif  // _BSDIFF_BUFFER_FILE_H_