File: MemoryBufferCache.h

package info (click to toggle)
llvm-toolchain-5.0 1:5.0.1-2~bpo9+1
  • links: PTS, VCS
  • area: main
  • in suites: stretch-backports
  • size: 553,688 kB
  • sloc: cpp: 2,878,786; ansic: 584,110; asm: 246,252; python: 124,751; objc: 106,925; sh: 21,542; lisp: 8,628; pascal: 5,885; ml: 5,544; perl: 5,312; makefile: 2,208; cs: 2,022; xml: 686; php: 212; csh: 117
file content (80 lines) | stat: -rw-r--r-- 2,646 bytes parent folder | download | duplicates (17)
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
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
//===- MemoryBufferCache.h - Cache for loaded memory buffers ----*- C++ -*-===//
//
//                     The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//

#ifndef LLVM_CLANG_BASIC_MEMORYBUFFERCACHE_H
#define LLVM_CLANG_BASIC_MEMORYBUFFERCACHE_H

#include "llvm/ADT/IntrusiveRefCntPtr.h"
#include "llvm/ADT/StringMap.h"
#include <memory>

namespace llvm {
class MemoryBuffer;
} // end namespace llvm

namespace clang {

/// Manage memory buffers across multiple users.
///
/// Ensures that multiple users have a consistent view of each buffer.  This is
/// used by \a CompilerInstance when building PCMs to ensure that each \a
/// ModuleManager sees the same files.
///
/// \a finalizeCurrentBuffers() should be called before creating a new user.
/// This locks in the current buffers, ensuring that no buffer that has already
/// been accessed can be purged, preventing use-after-frees.
class MemoryBufferCache : public llvm::RefCountedBase<MemoryBufferCache> {
  struct BufferEntry {
    std::unique_ptr<llvm::MemoryBuffer> Buffer;

    /// Track the timeline of when this was added to the cache.
    unsigned Index;
  };

  /// Cache of buffers.
  llvm::StringMap<BufferEntry> Buffers;

  /// Monotonically increasing index.
  unsigned NextIndex = 0;

  /// Bumped to prevent "older" buffers from being removed.
  unsigned FirstRemovableIndex = 0;

public:
  /// Store the Buffer under the Filename.
  ///
  /// \pre There is not already buffer is not already in the cache.
  /// \return a reference to the buffer as a convenience.
  llvm::MemoryBuffer &addBuffer(llvm::StringRef Filename,
                                std::unique_ptr<llvm::MemoryBuffer> Buffer);

  /// Try to remove a buffer from the cache.
  ///
  /// \return false on success, iff \c !isBufferFinal().
  bool tryToRemoveBuffer(llvm::StringRef Filename);

  /// Get a pointer to the buffer if it exists; else nullptr.
  llvm::MemoryBuffer *lookupBuffer(llvm::StringRef Filename);

  /// Check whether the buffer is final.
  ///
  /// \return true iff \a finalizeCurrentBuffers() has been called since the
  /// buffer was added.  This prevents buffers from being removed.
  bool isBufferFinal(llvm::StringRef Filename);

  /// Finalize the current buffers in the cache.
  ///
  /// Should be called when creating a new user to ensure previous uses aren't
  /// invalidated.
  void finalizeCurrentBuffers();
};

} // end namespace clang

#endif // LLVM_CLANG_BASIC_MEMORYBUFFERCACHE_H