1*b5893f02SDimitry Andric //===- SymbolRemappingReader.cpp - Read symbol remapping file -------------===//
2*b5893f02SDimitry Andric //
3*b5893f02SDimitry Andric //                      The LLVM Compiler Infrastructure
4*b5893f02SDimitry Andric //
5*b5893f02SDimitry Andric // This file is distributed under the University of Illinois Open Source
6*b5893f02SDimitry Andric // License. See LICENSE.TXT for details.
7*b5893f02SDimitry Andric //
8*b5893f02SDimitry Andric //===----------------------------------------------------------------------===//
9*b5893f02SDimitry Andric //
10*b5893f02SDimitry Andric // This file contains definitions needed for reading and applying symbol
11*b5893f02SDimitry Andric // remapping files.
12*b5893f02SDimitry Andric //
13*b5893f02SDimitry Andric //===----------------------------------------------------------------------===//
14*b5893f02SDimitry Andric 
15*b5893f02SDimitry Andric #include "llvm/Support/SymbolRemappingReader.h"
16*b5893f02SDimitry Andric #include "llvm/ADT/StringSwitch.h"
17*b5893f02SDimitry Andric #include "llvm/ADT/Twine.h"
18*b5893f02SDimitry Andric #include "llvm/Support/LineIterator.h"
19*b5893f02SDimitry Andric 
20*b5893f02SDimitry Andric using namespace llvm;
21*b5893f02SDimitry Andric 
22*b5893f02SDimitry Andric char SymbolRemappingParseError::ID;
23*b5893f02SDimitry Andric 
24*b5893f02SDimitry Andric /// Load a set of name remappings from a text file.
25*b5893f02SDimitry Andric ///
26*b5893f02SDimitry Andric /// See the documentation at the top of the file for an explanation of
27*b5893f02SDimitry Andric /// the expected format.
read(MemoryBuffer & B)28*b5893f02SDimitry Andric Error SymbolRemappingReader::read(MemoryBuffer &B) {
29*b5893f02SDimitry Andric   line_iterator LineIt(B, /*SkipBlanks=*/true, '#');
30*b5893f02SDimitry Andric 
31*b5893f02SDimitry Andric   auto ReportError = [&](Twine Msg) {
32*b5893f02SDimitry Andric     return llvm::make_error<SymbolRemappingParseError>(
33*b5893f02SDimitry Andric         B.getBufferIdentifier(), LineIt.line_number(), Msg);
34*b5893f02SDimitry Andric   };
35*b5893f02SDimitry Andric 
36*b5893f02SDimitry Andric   for (; !LineIt.is_at_eof(); ++LineIt) {
37*b5893f02SDimitry Andric     StringRef Line = *LineIt;
38*b5893f02SDimitry Andric     Line = Line.ltrim(' ');
39*b5893f02SDimitry Andric     // line_iterator only detects comments starting in column 1.
40*b5893f02SDimitry Andric     if (Line.startswith("#") || Line.empty())
41*b5893f02SDimitry Andric       continue;
42*b5893f02SDimitry Andric 
43*b5893f02SDimitry Andric     SmallVector<StringRef, 4> Parts;
44*b5893f02SDimitry Andric     Line.split(Parts, ' ', /*MaxSplits*/-1, /*KeepEmpty*/false);
45*b5893f02SDimitry Andric 
46*b5893f02SDimitry Andric     if (Parts.size() != 3)
47*b5893f02SDimitry Andric       return ReportError("Expected 'kind mangled_name mangled_name', "
48*b5893f02SDimitry Andric                          "found '" + Line + "'");
49*b5893f02SDimitry Andric 
50*b5893f02SDimitry Andric     using FK = ItaniumManglingCanonicalizer::FragmentKind;
51*b5893f02SDimitry Andric     Optional<FK> FragmentKind = StringSwitch<Optional<FK>>(Parts[0])
52*b5893f02SDimitry Andric                                     .Case("name", FK::Name)
53*b5893f02SDimitry Andric                                     .Case("type", FK::Type)
54*b5893f02SDimitry Andric                                     .Case("encoding", FK::Encoding)
55*b5893f02SDimitry Andric                                     .Default(None);
56*b5893f02SDimitry Andric     if (!FragmentKind)
57*b5893f02SDimitry Andric       return ReportError("Invalid kind, expected 'name', 'type', or 'encoding',"
58*b5893f02SDimitry Andric                          " found '" + Parts[0] + "'");
59*b5893f02SDimitry Andric 
60*b5893f02SDimitry Andric     using EE = ItaniumManglingCanonicalizer::EquivalenceError;
61*b5893f02SDimitry Andric     switch (Canonicalizer.addEquivalence(*FragmentKind, Parts[1], Parts[2])) {
62*b5893f02SDimitry Andric     case EE::Success:
63*b5893f02SDimitry Andric       break;
64*b5893f02SDimitry Andric 
65*b5893f02SDimitry Andric     case EE::ManglingAlreadyUsed:
66*b5893f02SDimitry Andric       return ReportError("Manglings '" + Parts[1] + "' and '" + Parts[2] + "' "
67*b5893f02SDimitry Andric                          "have both been used in prior remappings. Move this "
68*b5893f02SDimitry Andric                          "remapping earlier in the file.");
69*b5893f02SDimitry Andric 
70*b5893f02SDimitry Andric     case EE::InvalidFirstMangling:
71*b5893f02SDimitry Andric       return ReportError("Could not demangle '" + Parts[1] + "' "
72*b5893f02SDimitry Andric                          "as a <" + Parts[0] + ">; invalid mangling?");
73*b5893f02SDimitry Andric 
74*b5893f02SDimitry Andric     case EE::InvalidSecondMangling:
75*b5893f02SDimitry Andric       return ReportError("Could not demangle '" + Parts[2] + "' "
76*b5893f02SDimitry Andric                          "as a <" + Parts[0] + ">; invalid mangling?");
77*b5893f02SDimitry Andric     }
78*b5893f02SDimitry Andric   }
79*b5893f02SDimitry Andric 
80*b5893f02SDimitry Andric   return Error::success();
81*b5893f02SDimitry Andric }
82