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