From: Zachary Turner Date: Wed, 31 Aug 2016 23:14:31 +0000 (+0000) Subject: [codeview] Have visitTypeBegin return the record type. X-Git-Tag: llvmorg-4.0.0-rc1~10931 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=77807637ffb0fffd6f215187258b0e12c6b6a620;p=platform%2Fupstream%2Fllvm.git [codeview] Have visitTypeBegin return the record type. Previously we were assuming that any visitation of types would necessarily be against a type we had binary data for. Reasonable assumption when were just reading PDBs and dumping them, but once we start writing PDBs from Yaml this breaks down, because we have no binary data yet, only Yaml, and from that we need to read the record kind and perform the switch based on that. So this patch does that. Instead of having the visitor switch on the kind that is already in the CVType record, we change the visitTypeBegin() method to return the Kind, and switch on the returned value. This way, the default implementation can still return the value from the CVType, but the implementation which visits Yaml records and serializes binary PDB type records can use the field in the Yaml as the source of the switch. llvm-svn: 280307 --- diff --git a/llvm/include/llvm/DebugInfo/CodeView/TypeDumper.h b/llvm/include/llvm/DebugInfo/CodeView/TypeDumper.h index 49db43d6b7bb..598f97047808 100644 --- a/llvm/include/llvm/DebugInfo/CodeView/TypeDumper.h +++ b/llvm/include/llvm/DebugInfo/CodeView/TypeDumper.h @@ -68,7 +68,8 @@ public: /// Paired begin/end actions for all types. Receives all record data, /// including the fixed-length record prefix. - Error visitTypeBegin(const CVRecord &Record) override; + Expected + visitTypeBegin(const CVRecord &Record) override; Error visitTypeEnd(const CVRecord &Record) override; #define TYPE_RECORD(EnumName, EnumVal, Name) \ diff --git a/llvm/include/llvm/DebugInfo/CodeView/TypeVisitorCallbackPipeline.h b/llvm/include/llvm/DebugInfo/CodeView/TypeVisitorCallbackPipeline.h index 3bfd52d4da44..d158f42947c9 100644 --- a/llvm/include/llvm/DebugInfo/CodeView/TypeVisitorCallbackPipeline.h +++ b/llvm/include/llvm/DebugInfo/CodeView/TypeVisitorCallbackPipeline.h @@ -40,12 +40,16 @@ public: return Error::success(); } - virtual Error visitTypeBegin(const CVRecord &Record) override { + virtual Expected + visitTypeBegin(const CVRecord &Record) override { + TypeLeafKind Kind = Record.Type; for (auto Visitor : Pipeline) { - if (auto EC = Visitor->visitTypeBegin(Record)) - return EC; + if (auto ExpectedKind = Visitor->visitTypeBegin(Record)) + Kind = *ExpectedKind; + else + return ExpectedKind.takeError(); } - return Error::success(); + return Kind; } virtual Error visitTypeEnd(const CVRecord &Record) override { for (auto Visitor : Pipeline) { diff --git a/llvm/include/llvm/DebugInfo/CodeView/TypeVisitorCallbacks.h b/llvm/include/llvm/DebugInfo/CodeView/TypeVisitorCallbacks.h index a734d480fe39..ecae9eef807b 100644 --- a/llvm/include/llvm/DebugInfo/CodeView/TypeVisitorCallbacks.h +++ b/llvm/include/llvm/DebugInfo/CodeView/TypeVisitorCallbacks.h @@ -33,9 +33,11 @@ public: } /// Paired begin/end actions for all types. Receives all record data, - /// including the fixed-length record prefix. - virtual Error visitTypeBegin(const CVRecord &Record) { - return Error::success(); + /// including the fixed-length record prefix. visitTypeBegin() should return + /// the type of the Record, or an error if it cannot be determined. + virtual Expected + visitTypeBegin(const CVRecord &Record) { + return Record.Type; } virtual Error visitTypeEnd(const CVRecord &Record) { return Error::success(); diff --git a/llvm/lib/DebugInfo/CodeView/CVTypeVisitor.cpp b/llvm/lib/DebugInfo/CodeView/CVTypeVisitor.cpp index 18215d355820..b4c3a9f4c21e 100644 --- a/llvm/lib/DebugInfo/CodeView/CVTypeVisitor.cpp +++ b/llvm/lib/DebugInfo/CodeView/CVTypeVisitor.cpp @@ -75,10 +75,13 @@ static Error visitKnownRecord(const CVRecord &Record, } Error CVTypeVisitor::visitTypeRecord(const CVRecord &Record) { - if (auto EC = Callbacks.visitTypeBegin(Record)) - return EC; + TypeLeafKind Kind; + if (auto ExpectedKind = Callbacks.visitTypeBegin(Record)) + Kind = *ExpectedKind; + else + return ExpectedKind.takeError(); - switch (Record.Type) { + switch (Kind) { default: if (auto EC = Callbacks.visitUnknownType(Record)) return EC; @@ -133,8 +136,9 @@ Error CVTypeVisitor::visitFieldListMemberStream(ArrayRef Data) { if (!ExpectedRecord) \ return ExpectedRecord.takeError(); \ auto &Record = *ExpectedRecord; \ - if (auto EC = Callbacks.visitTypeBegin(Record)) \ - return EC; \ + auto ExpectedKind = Callbacks.visitTypeBegin(Record); \ + if (!ExpectedKind || *ExpectedKind != Leaf) \ + return ExpectedKind.takeError(); \ if (auto EC = visitKnownRecord(Record, Callbacks)) \ return EC; \ if (auto EC = Callbacks.visitTypeEnd(Record)) \ diff --git a/llvm/lib/DebugInfo/CodeView/TypeDumper.cpp b/llvm/lib/DebugInfo/CodeView/TypeDumper.cpp index f7dfdd2f3986..43e29b7f451f 100644 --- a/llvm/lib/DebugInfo/CodeView/TypeDumper.cpp +++ b/llvm/lib/DebugInfo/CodeView/TypeDumper.cpp @@ -203,7 +203,8 @@ static StringRef getLeafTypeName(TypeLeafKind LT) { return "UnknownLeaf"; } -Error CVTypeDumper::visitTypeBegin(const CVRecord &Record) { +Expected +CVTypeDumper::visitTypeBegin(const CVRecord &Record) { // Reset Name to the empty string. If the visitor sets it, we know it. Name = ""; @@ -223,7 +224,7 @@ Error CVTypeDumper::visitTypeBegin(const CVRecord &Record) { assert(!IsInFieldList); IsInFieldList = true; } - return Error::success(); + return Record.Type; } Error CVTypeDumper::visitTypeEnd(const CVRecord &Record) { diff --git a/llvm/lib/DebugInfo/CodeView/TypeStreamMerger.cpp b/llvm/lib/DebugInfo/CodeView/TypeStreamMerger.cpp index ddc3f4af58e4..f2a7298a8d1d 100644 --- a/llvm/lib/DebugInfo/CodeView/TypeStreamMerger.cpp +++ b/llvm/lib/DebugInfo/CodeView/TypeStreamMerger.cpp @@ -70,7 +70,8 @@ public: Error visitUnknownType(const CVRecord &Record) override; - Error visitTypeBegin(const CVRecord &Record) override; + Expected + visitTypeBegin(const CVRecord &Record) override; Error visitTypeEnd(const CVRecord &Record) override; bool mergeStream(const CVTypeArray &Types); @@ -123,13 +124,14 @@ private: } // end anonymous namespace -Error TypeStreamMerger::visitTypeBegin(const CVRecord &Rec) { +Expected +TypeStreamMerger::visitTypeBegin(const CVRecord &Rec) { if (Rec.Type == TypeLeafKind::LF_FIELDLIST) { assert(!IsInFieldList); IsInFieldList = true; } else BeginIndexMapSize = IndexMap.size(); - return Error::success(); + return Rec.Type; } Error TypeStreamMerger::visitTypeEnd(const CVRecord &Rec) { diff --git a/llvm/lib/DebugInfo/PDB/Raw/TpiStream.cpp b/llvm/lib/DebugInfo/PDB/Raw/TpiStream.cpp index 74200e70119c..7154b297b124 100644 --- a/llvm/lib/DebugInfo/PDB/Raw/TpiStream.cpp +++ b/llvm/lib/DebugInfo/PDB/Raw/TpiStream.cpp @@ -121,10 +121,11 @@ public: return verify(Rec); } - Error visitTypeBegin(const CVRecord &Rec) override { + Expected + visitTypeBegin(const CVRecord &Rec) override { ++Index; RawRecord = &Rec; - return Error::success(); + return Rec.Type; } private: diff --git a/llvm/tools/llvm-pdbdump/CodeViewYaml.cpp b/llvm/tools/llvm-pdbdump/CodeViewYaml.cpp index 96bf1c19aa6b..e75f454b34a9 100644 --- a/llvm/tools/llvm-pdbdump/CodeViewYaml.cpp +++ b/llvm/tools/llvm-pdbdump/CodeViewYaml.cpp @@ -508,9 +508,13 @@ void ScalarEnumerationTraits::enumeration(IO &io, } } -Error llvm::codeview::yaml::YamlTypeDumperCallbacks::visitTypeBegin( +Expected +llvm::codeview::yaml::YamlTypeDumperCallbacks::visitTypeBegin( const CVRecord &CVR) { + // When we're outputting, `CVR.Type` already has the right value in it. But + // when we're inputting, we need to read the value. Since `CVR.Type` is const + // we do it into a temp variable. TypeLeafKind K = CVR.Type; YamlIO.mapRequired("Kind", K); - return Error::success(); + return K; } diff --git a/llvm/tools/llvm-pdbdump/CodeViewYaml.h b/llvm/tools/llvm-pdbdump/CodeViewYaml.h index bacb55cef407..8932d008ca38 100644 --- a/llvm/tools/llvm-pdbdump/CodeViewYaml.h +++ b/llvm/tools/llvm-pdbdump/CodeViewYaml.h @@ -21,7 +21,8 @@ class YamlTypeDumperCallbacks : public TypeVisitorCallbacks { public: YamlTypeDumperCallbacks(llvm::yaml::IO &IO) : YamlIO(IO) {} - virtual Error visitTypeBegin(const CVRecord &Record) override; + virtual Expected + visitTypeBegin(const CVRecord &Record) override; #define TYPE_RECORD(EnumName, EnumVal, Name) \ Error visitKnownRecord(const CVRecord &CVR, \