1 //===-- TestModuleFileExtension.h - Module Extension Tester -----*- C++ -*-===//
2 //
3 //                     The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9 #ifndef LLVM_CLANG_FRONTEND_TESTMODULEFILEEXTENSION_H
10 #define LLVM_CLANG_FRONTEND_TESTMODULEFILEEXTENSION_H
11 
12 #include "clang/Serialization/ModuleFileExtension.h"
13 #include "clang/Basic/LLVM.h"
14 #include "llvm/ADT/StringRef.h"
15 #include "llvm/Bitcode/BitstreamReader.h"
16 #include <string>
17 
18 namespace clang {
19 
20 /// A module file extension used for testing purposes.
21 class TestModuleFileExtension : public ModuleFileExtension {
22   std::string BlockName;
23   unsigned MajorVersion;
24   unsigned MinorVersion;
25   bool Hashed;
26   std::string UserInfo;
27 
28   class Writer : public ModuleFileExtensionWriter {
29   public:
30     Writer(ModuleFileExtension *Ext) : ModuleFileExtensionWriter(Ext) { }
31     ~Writer() override;
32 
33     void writeExtensionContents(llvm::BitstreamWriter &Stream) override;
34   };
35 
36   class Reader : public ModuleFileExtensionReader {
37     llvm::BitstreamCursor Stream;
38 
39   public:
40     ~Reader() override;
41 
42     Reader(ModuleFileExtension *Ext, const llvm::BitstreamCursor &InStream);
43   };
44 
45 public:
46   TestModuleFileExtension(StringRef BlockName,
47                           unsigned MajorVersion,
48                           unsigned MinorVersion,
49                           bool Hashed,
50                           StringRef UserInfo)
51     : BlockName(BlockName),
52       MajorVersion(MajorVersion), MinorVersion(MinorVersion),
53       Hashed(Hashed), UserInfo(UserInfo) { }
54   ~TestModuleFileExtension() override;
55 
56   ModuleFileExtensionMetadata getExtensionMetadata() const override;
57 
58   llvm::hash_code hashExtension(llvm::hash_code Code) const override;
59 
60   std::unique_ptr<ModuleFileExtensionWriter>
61   createExtensionWriter(ASTWriter &Writer) override;
62 
63   std::unique_ptr<ModuleFileExtensionReader>
64   createExtensionReader(const ModuleFileExtensionMetadata &Metadata,
65                         ASTReader &Reader, serialization::ModuleFile &Mod,
66                         const llvm::BitstreamCursor &Stream) override;
67 };
68 
69 } // end namespace clang
70 
71 #endif // LLVM_CLANG_FRONTEND_TESTMODULEFILEEXTENSION_H
72