1 //===-- SBReproducerPrivate.h -----------------------------------*- C++ -*-===//
2 //
3 //
4 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
5 // See https://llvm.org/LICENSE.txt for license information.
6 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
7 //
8 //===----------------------------------------------------------------------===//
9 
10 #ifndef LLDB_SOURCE_API_SBREPRODUCERPRIVATE_H
11 #define LLDB_SOURCE_API_SBREPRODUCERPRIVATE_H
12 
13 #include "lldb/API/SBReproducer.h"
14 
15 #include "lldb/Utility/FileSpec.h"
16 #include "lldb/Utility/Log.h"
17 #include "lldb/Utility/Reproducer.h"
18 #include "lldb/Utility/ReproducerInstrumentation.h"
19 
20 #include "llvm/ADT/DenseMap.h"
21 
22 #define LLDB_GET_INSTRUMENTATION_DATA()                                        \
23   lldb_private::repro::InstrumentationData::Instance()
24 
25 namespace lldb_private {
26 namespace repro {
27 
28 class SBRegistry : public Registry {
29 public:
30   SBRegistry();
31 };
32 
33 class SBProvider : public Provider<SBProvider> {
34 public:
35   struct Info {
36     static const char *name;
37     static const char *file;
38   };
39 
40   SBProvider(const FileSpec &directory)
41       : Provider(directory),
42         m_stream(directory.CopyByAppendingPathComponent("sbapi.bin").GetPath(),
43                  m_ec, llvm::sys::fs::OpenFlags::OF_None),
44         m_serializer(m_stream) {}
45 
46   Serializer &GetSerializer() { return m_serializer; }
47   Registry &GetRegistry() { return m_registry; }
48 
49   static char ID;
50 
51 private:
52   std::error_code m_ec;
53   llvm::raw_fd_ostream m_stream;
54   Serializer m_serializer;
55   SBRegistry m_registry;
56 };
57 
58 class ReplayData {
59 public:
60   ReplayData(std::unique_ptr<llvm::MemoryBuffer> memory_buffer)
61       : m_memory_buffer(std::move(memory_buffer)), m_registry(),
62         m_deserializer(m_memory_buffer->getBuffer()) {}
63   Deserializer &GetDeserializer() { return m_deserializer; }
64   Registry &GetRegistry() { return m_registry; }
65 
66 private:
67   std::unique_ptr<llvm::MemoryBuffer> m_memory_buffer;
68   SBRegistry m_registry;
69   Deserializer m_deserializer;
70 };
71 
72 template <typename T> void RegisterMethods(Registry &R);
73 
74 } // namespace repro
75 } // namespace lldb_private
76 
77 #endif
78