1 //===-- TestModuleFileExtension.h - Module Extension Tester -----*- C++ -*-===// 2 // 3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4 // See https://llvm.org/LICENSE.txt for license information. 5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6 // 7 //===----------------------------------------------------------------------===// 8 #ifndef LLVM_CLANG_FRONTEND_TESTMODULEFILEEXTENSION_H 9 #define LLVM_CLANG_FRONTEND_TESTMODULEFILEEXTENSION_H 10 11 #include "clang/Serialization/ModuleFileExtension.h" 12 #include "clang/Basic/LLVM.h" 13 #include "llvm/ADT/StringRef.h" 14 #include "llvm/Bitstream/BitstreamReader.h" 15 #include <string> 16 17 namespace clang { 18 19 /// A module file extension used for testing purposes. 20 class TestModuleFileExtension 21 : public llvm::RTTIExtends<TestModuleFileExtension, 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(Sema &SemaRef, 34 llvm::BitstreamWriter &Stream) override; 35 }; 36 37 class Reader : public ModuleFileExtensionReader { 38 llvm::BitstreamCursor Stream; 39 40 public: 41 ~Reader() override; 42 43 Reader(ModuleFileExtension *Ext, const llvm::BitstreamCursor &InStream); 44 }; 45 46 public: 47 static char ID; 48 49 TestModuleFileExtension(StringRef BlockName, unsigned MajorVersion, 50 unsigned MinorVersion, bool Hashed, 51 StringRef UserInfo) 52 : BlockName(BlockName), MajorVersion(MajorVersion), 53 MinorVersion(MinorVersion), Hashed(Hashed), UserInfo(UserInfo) {} 54 ~TestModuleFileExtension() override; 55 56 ModuleFileExtensionMetadata getExtensionMetadata() const override; 57 58 void hashExtension(ExtensionHashBuilder &HBuilder) 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 std::string str() const; 69 }; 70 71 } // end namespace clang 72 73 #endif // LLVM_CLANG_FRONTEND_TESTMODULEFILEEXTENSION_H 74