/build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/llvm/include/llvm/Bitstream/BitstreamReader.h

Bug Summary

File:	llvm/include/llvm/Bitstream/BitstreamReader.h
Warning:	line 220, column 39 The result of the right shift is undefined due to shifting by '64', which is greater or equal to the width of type 'llvm::SimpleBitstreamCursor::word_t'

Annotated Source Code

Press '?' to see keyboard shortcuts

Show analyzer invocation

clang -cc1 -triple x86_64-pc-linux-gnu -analyze -disable-free -disable-llvm-verifier -discard-value-names -main-file-name GlobalModuleIndex.cpp -analyzer-store=region -analyzer-opt-analyze-nested-blocks -analyzer-checker=core -analyzer-checker=apiModeling -analyzer-checker=unix -analyzer-checker=deadcode -analyzer-checker=cplusplus -analyzer-checker=security.insecureAPI.UncheckedReturn -analyzer-checker=security.insecureAPI.getpw -analyzer-checker=security.insecureAPI.gets -analyzer-checker=security.insecureAPI.mktemp -analyzer-checker=security.insecureAPI.mkstemp -analyzer-checker=security.insecureAPI.vfork -analyzer-checker=nullability.NullPassedToNonnull -analyzer-checker=nullability.NullReturnedFromNonnull -analyzer-output plist -w -setup-static-analyzer -analyzer-config-compatibility-mode=true -mrelocation-model pic -pic-level 2 -mthread-model posix -mframe-pointer=none -relaxed-aliasing -fmath-errno -fno-rounding-math -masm-verbose -mconstructor-aliases -munwind-tables -target-cpu x86-64 -dwarf-column-info -fno-split-dwarf-inlining -debugger-tuning=gdb -ffunction-sections -fdata-sections -resource-dir /usr/lib/llvm-10/lib/clang/10.0.0 -D CLANG_VENDOR="Debian " -D _DEBUG -D _GNU_SOURCE -D __STDC_CONSTANT_MACROS -D __STDC_FORMAT_MACROS -D __STDC_LIMIT_MACROS -I /build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/build-llvm/tools/clang/lib/Serialization -I /build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/clang/lib/Serialization -I /build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/clang/include -I /build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/build-llvm/tools/clang/include -I /build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/build-llvm/include -I /build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/llvm/include -U NDEBUG -internal-isystem /usr/lib/gcc/x86_64-linux-gnu/6.3.0/../../../../include/c++/6.3.0 -internal-isystem /usr/lib/gcc/x86_64-linux-gnu/6.3.0/../../../../include/x86_64-linux-gnu/c++/6.3.0 -internal-isystem /usr/lib/gcc/x86_64-linux-gnu/6.3.0/../../../../include/x86_64-linux-gnu/c++/6.3.0 -internal-isystem /usr/lib/gcc/x86_64-linux-gnu/6.3.0/../../../../include/c++/6.3.0/backward -internal-isystem /usr/local/include -internal-isystem /usr/lib/llvm-10/lib/clang/10.0.0/include -internal-externc-isystem /usr/include/x86_64-linux-gnu -internal-externc-isystem /include -internal-externc-isystem /usr/include -O2 -Wno-unused-parameter -Wwrite-strings -Wno-missing-field-initializers -Wno-long-long -Wno-maybe-uninitialized -Wno-comment -std=c++14 -fdeprecated-macro -fdebug-compilation-dir /build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/build-llvm/tools/clang/lib/Serialization -fdebug-prefix-map=/build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd=. -ferror-limit 19 -fmessage-length 0 -fvisibility-inlines-hidden -stack-protector 2 -fgnuc-version=4.2.1 -fobjc-runtime=gcc -fno-common -fdiagnostics-show-option -vectorize-loops -vectorize-slp -analyzer-output=html -analyzer-config stable-report-filename=true -faddrsig -o /tmp/scan-build-2020-01-13-084841-49055-1 -x c++ /build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/clang/lib/Serialization/GlobalModuleIndex.cpp

/build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/clang/lib/Serialization/GlobalModuleIndex.cpp

→

1//===--- GlobalModuleIndex.cpp - Global Module Index ------------*- 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//
9// This file implements the GlobalModuleIndex class.
10//
11//===----------------------------------------------------------------------===//

13#include "clang/Serialization/GlobalModuleIndex.h"
14#include "ASTReaderInternals.h"
15#include "clang/Basic/FileManager.h"
16#include "clang/Lex/HeaderSearch.h"
17#include "clang/Serialization/ASTBitCodes.h"
18#include "clang/Serialization/ModuleFile.h"
19#include "clang/Serialization/PCHContainerOperations.h"
20#include "llvm/ADT/DenseMap.h"
21#include "llvm/ADT/MapVector.h"
22#include "llvm/ADT/SmallString.h"
23#include "llvm/ADT/StringRef.h"
24#include "llvm/Bitstream/BitstreamReader.h"
25#include "llvm/Bitstream/BitstreamWriter.h"
26#include "llvm/Support/DJB.h"
27#include "llvm/Support/FileSystem.h"
28#include "llvm/Support/FileUtilities.h"
29#include "llvm/Support/LockFileManager.h"
30#include "llvm/Support/MemoryBuffer.h"
31#include "llvm/Support/OnDiskHashTable.h"
32#include "llvm/Support/Path.h"
33#include "llvm/Support/TimeProfiler.h"
34#include <cstdio>
35using namespace clang;
36using namespace serialization;

38//----------------------------------------------------------------------------//
39// Shared constants
40//----------------------------------------------------------------------------//
41namespace {
enum {
  /// The block containing the index.
  GLOBAL_INDEX_BLOCK_ID = llvm::bitc::FIRST_APPLICATION_BLOCKID
};

/// Describes the record types in the index.
enum IndexRecordTypes {
  /// Contains version information and potentially other metadata,
  /// used to determine if we can read this global index file.
  INDEX_METADATA,
  /// Describes a module, including its file name and dependencies.
  MODULE,
  /// The index for identifiers.
  IDENTIFIER_INDEX
};
57}

59/// The name of the global index file.
60static const char * const IndexFileName = "modules.idx";

62/// The global index file version.
63static const unsigned CurrentVersion = 1;

65//----------------------------------------------------------------------------//
66// Global module index reader.
67//----------------------------------------------------------------------------//

69namespace {

71/// Trait used to read the identifier index from the on-disk hash
72/// table.
73class IdentifierIndexReaderTrait {
74public:
typedef StringRef external_key_type;
typedef StringRef internal_key_type;
typedef SmallVector<unsigned, 2> data_type;
typedef unsigned hash_value_type;
typedef unsigned offset_type;

static bool EqualKey(const internal_key_type& a, const internal_key_type& b) {
  return a == b;
}

static hash_value_type ComputeHash(const internal_key_type& a) {
  return llvm::djbHash(a);
}

static std::pair<unsigned, unsigned>
ReadKeyDataLength(const unsigned char*& d) {
  using namespace llvm::support;
  unsigned KeyLen = endian::readNext<uint16_t, little, unaligned>(d);
  unsigned DataLen = endian::readNext<uint16_t, little, unaligned>(d);
  return std::make_pair(KeyLen, DataLen);
}

static const internal_key_type&
GetInternalKey(const external_key_type& x) { return x; }

static const external_key_type&
GetExternalKey(const internal_key_type& x) { return x; }

static internal_key_type ReadKey(const unsigned char* d, unsigned n) {
  return StringRef((const char *)d, n);
}

static data_type ReadData(const internal_key_type& k,
                          const unsigned char* d,
                          unsigned DataLen) {
  using namespace llvm::support;

  data_type Result;
  while (DataLen > 0) {
    unsigned ID = endian::readNext<uint32_t, little, unaligned>(d);
    Result.push_back(ID);
    DataLen -= 4;
  }

  return Result;
}
121};

123typedef llvm::OnDiskIterableChainedHashTable<IdentifierIndexReaderTrait>
  IdentifierIndexTable;

126}

128GlobalModuleIndex::GlobalModuleIndex(
  std::unique_ptr<llvm::MemoryBuffer> IndexBuffer,
  llvm::BitstreamCursor Cursor)
  : Buffer(std::move(IndexBuffer)), IdentifierIndex(), NumIdentifierLookups(),
    NumIdentifierLookupHits() {
auto Fail = [&](llvm::Error &&Err) {
  report_fatal_error("Module index '" + Buffer->getBufferIdentifier() +
                     "' failed: " + toString(std::move(Err)));
};

llvm::TimeTraceScope TimeScope("Module LoadIndex");
// Read the global index.
bool InGlobalIndexBlock = false;
bool Done = false;
while (!Done) {
  llvm::BitstreamEntry Entry;
  if (Expected<llvm::BitstreamEntry> Res = Cursor.advance())
    Entry = Res.get();
  else
    Fail(Res.takeError());

  switch (Entry.Kind) {
  case llvm::BitstreamEntry::Error:
    return;

  case llvm::BitstreamEntry::EndBlock:
    if (InGlobalIndexBlock) {
      InGlobalIndexBlock = false;
      Done = true;
      continue;
    }
    return;


  case llvm::BitstreamEntry::Record:
    // Entries in the global index block are handled below.
    if (InGlobalIndexBlock)
      break;

    return;

  case llvm::BitstreamEntry::SubBlock:
    if (!InGlobalIndexBlock && Entry.ID == GLOBAL_INDEX_BLOCK_ID) {
      if (llvm::Error Err = Cursor.EnterSubBlock(GLOBAL_INDEX_BLOCK_ID))
        Fail(std::move(Err));
      InGlobalIndexBlock = true;
    } else if (llvm::Error Err = Cursor.SkipBlock())
      Fail(std::move(Err));
    continue;
  }

  SmallVector<uint64_t, 64> Record;
  StringRef Blob;
  Expected<unsigned> MaybeIndexRecord =
      Cursor.readRecord(Entry.ID, Record, &Blob);
  if (!MaybeIndexRecord)
    Fail(MaybeIndexRecord.takeError());
  IndexRecordTypes IndexRecord =
      static_cast<IndexRecordTypes>(MaybeIndexRecord.get());
  switch (IndexRecord) {
  case INDEX_METADATA:
    // Make sure that the version matches.
    if (Record.size() < 1 || Record[0] != CurrentVersion)
      return;
    break;

  case MODULE: {
    unsigned Idx = 0;
    unsigned ID = Record[Idx++];

    // Make room for this module's information.
    if (ID == Modules.size())
      Modules.push_back(ModuleInfo());
    else
      Modules.resize(ID + 1);

    // Size/modification time for this module file at the time the
    // global index was built.
    Modules[ID].Size = Record[Idx++];
    Modules[ID].ModTime = Record[Idx++];

    // File name.
    unsigned NameLen = Record[Idx++];
    Modules[ID].FileName.assign(Record.begin() + Idx,
                                Record.begin() + Idx + NameLen);
    Idx += NameLen;

    // Dependencies
    unsigned NumDeps = Record[Idx++];
    Modules[ID].Dependencies.insert(Modules[ID].Dependencies.end(),
                                    Record.begin() + Idx,
                                    Record.begin() + Idx + NumDeps);
    Idx += NumDeps;

    // Make sure we're at the end of the record.
    assert(Idx == Record.size() && "More module info?")((Idx == Record.size() && "More module info?") ? static_cast
<void> (0) : __assert_fail ("Idx == Record.size() && \"More module info?\""
, "/build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/clang/lib/Serialization/GlobalModuleIndex.cpp"
, 223, __PRETTY_FUNCTION__));

    // Record this module as an unresolved module.
    // FIXME: this doesn't work correctly for module names containing path
    // separators.
    StringRef ModuleName = llvm::sys::path::stem(Modules[ID].FileName);
    // Remove the -<hash of ModuleMapPath>
    ModuleName = ModuleName.rsplit('-').first;
    UnresolvedModules[ModuleName] = ID;
    break;
  }

  case IDENTIFIER_INDEX:
    // Wire up the identifier index.
    if (Record[0]) {
      IdentifierIndex = IdentifierIndexTable::Create(
          (const unsigned char *)Blob.data() + Record[0],
          (const unsigned char *)Blob.data() + sizeof(uint32_t),
          (const unsigned char *)Blob.data(), IdentifierIndexReaderTrait());
    }
    break;
  }
}
246}

248GlobalModuleIndex::~GlobalModuleIndex() {
delete static_cast<IdentifierIndexTable *>(IdentifierIndex);
250}

252std::pair<GlobalModuleIndex *, llvm::Error>
253GlobalModuleIndex::readIndex(StringRef Path) {
// Load the index file, if it's there.
llvm::SmallString<128> IndexPath;
IndexPath += Path;
llvm::sys::path::append(IndexPath, IndexFileName);

llvm::ErrorOr<std::unique_ptr<llvm::MemoryBuffer>> BufferOrErr =
    llvm::MemoryBuffer::getFile(IndexPath.c_str());
if (!BufferOrErr)
  return std::make_pair(nullptr,
                        llvm::errorCodeToError(BufferOrErr.getError()));
std::unique_ptr<llvm::MemoryBuffer> Buffer = std::move(BufferOrErr.get());

/// The main bitstream cursor for the main block.
llvm::BitstreamCursor Cursor(*Buffer);

// Sniff for the signature.
for (unsigned char C : {'B', 'C', 'G', 'I'}) {
  if (Expected<llvm::SimpleBitstreamCursor::word_t> Res = Cursor.Read(8)) {
    if (Res.get() != C)
      return std::make_pair(
          nullptr, llvm::createStringError(std::errc::illegal_byte_sequence,
                                           "expected signature BCGI"));
  } else
    return std::make_pair(nullptr, Res.takeError());
}

return std::make_pair(new GlobalModuleIndex(std::move(Buffer), Cursor),
                      llvm::Error::success());
282}

284void
285GlobalModuleIndex::getKnownModules(SmallVectorImpl<ModuleFile *> &ModuleFiles) {
ModuleFiles.clear();
for (unsigned I = 0, N = Modules.size(); I != N; ++I) {
  if (ModuleFile *MF = Modules[I].File)
    ModuleFiles.push_back(MF);
}
291}

293void GlobalModuleIndex::getModuleDependencies(
     ModuleFile *File,
     SmallVectorImpl<ModuleFile *> &Dependencies) {
// Look for information about this module file.
llvm::DenseMap<ModuleFile *, unsigned>::iterator Known
  = ModulesByFile.find(File);
if (Known == ModulesByFile.end())
  return;

// Record dependencies.
Dependencies.clear();
ArrayRef<unsigned> StoredDependencies = Modules[Known->second].Dependencies;
for (unsigned I = 0, N = StoredDependencies.size(); I != N; ++I) {
  if (ModuleFile *MF = Modules[I].File)
    Dependencies.push_back(MF);
}
309}

311bool GlobalModuleIndex::lookupIdentifier(StringRef Name, HitSet &Hits) {
Hits.clear();

// If there's no identifier index, there is nothing we can do.
if (!IdentifierIndex)
  return false;

// Look into the identifier index.
++NumIdentifierLookups;
IdentifierIndexTable &Table
  = *static_cast<IdentifierIndexTable *>(IdentifierIndex);
IdentifierIndexTable::iterator Known = Table.find(Name);
if (Known == Table.end()) {
  return true;
}

SmallVector<unsigned, 2> ModuleIDs = *Known;
for (unsigned I = 0, N = ModuleIDs.size(); I != N; ++I) {
  if (ModuleFile *MF = Modules[ModuleIDs[I]].File)
    Hits.insert(MF);
}

++NumIdentifierLookupHits;
return true;
335}

337bool GlobalModuleIndex::loadedModuleFile(ModuleFile *File) {
// Look for the module in the global module index based on the module name.
StringRef Name = File->ModuleName;
llvm::StringMap<unsigned>::iterator Known = UnresolvedModules.find(Name);
if (Known == UnresolvedModules.end()) {
  return true;
}

// Rectify this module with the global module index.
ModuleInfo &Info = Modules[Known->second];

//  If the size and modification time match what we expected, record this
// module file.
bool Failed = true;
if (File->File->getSize() == Info.Size &&
    File->File->getModificationTime() == Info.ModTime) {
  Info.File = File;
  ModulesByFile[File] = Known->second;

  Failed = false;
}

// One way or another, we have resolved this module file.
UnresolvedModules.erase(Known);
return Failed;
362}

364void GlobalModuleIndex::printStats() {
std::fprintf(stderrstderr, "*** Global Module Index Statistics:\n");
if (NumIdentifierLookups) {
  fprintf(stderrstderr, "  %u / %u identifier lookups succeeded (%f%%)\n",
          NumIdentifierLookupHits, NumIdentifierLookups,
          (double)NumIdentifierLookupHits*100.0/NumIdentifierLookups);
}
std::fprintf(stderrstderr, "\n");
372}

374LLVM_DUMP_METHOD__attribute__((noinline)) __attribute__((__used__)) void GlobalModuleIndex::dump() {
llvm::errs() << "*** Global Module Index Dump:\n";
llvm::errs() << "Module files:\n";
for (auto &MI : Modules) {
  llvm::errs() << "** " << MI.FileName << "\n";
  if (MI.File)
    MI.File->dump();
  else
    llvm::errs() << "\n";
}
llvm::errs() << "\n";
385}

387//----------------------------------------------------------------------------//
388// Global module index writer.
389//----------------------------------------------------------------------------//

391namespace {
/// Provides information about a specific module file.
struct ModuleFileInfo {
  /// The numberic ID for this module file.
  unsigned ID;

  /// The set of modules on which this module depends. Each entry is
  /// a module ID.
  SmallVector<unsigned, 4> Dependencies;
  ASTFileSignature Signature;
};

struct ImportedModuleFileInfo {
  off_t StoredSize;
  time_t StoredModTime;
  ASTFileSignature StoredSignature;
  ImportedModuleFileInfo(off_t Size, time_t ModTime, ASTFileSignature Sig)
      : StoredSize(Size), StoredModTime(ModTime), StoredSignature(Sig) {}
};

/// Builder that generates the global module index file.
class GlobalModuleIndexBuilder {
  FileManager &FileMgr;
  const PCHContainerReader &PCHContainerRdr;

  /// Mapping from files to module file information.
  typedef llvm::MapVector<const FileEntry *, ModuleFileInfo> ModuleFilesMap;

  /// Information about each of the known module files.
  ModuleFilesMap ModuleFiles;

  /// Mapping from the imported module file to the imported
  /// information.
  typedef std::multimap<const FileEntry *, ImportedModuleFileInfo>
      ImportedModuleFilesMap;

  /// Information about each importing of a module file.
  ImportedModuleFilesMap ImportedModuleFiles;

  /// Mapping from identifiers to the list of module file IDs that
  /// consider this identifier to be interesting.
  typedef llvm::StringMap<SmallVector<unsigned, 2> > InterestingIdentifierMap;

  /// A mapping from all interesting identifiers to the set of module
  /// files in which those identifiers are considered interesting.
  InterestingIdentifierMap InterestingIdentifiers;

  /// Write the block-info block for the global module index file.
  void emitBlockInfoBlock(llvm::BitstreamWriter &Stream);

  /// Retrieve the module file information for the given file.
  ModuleFileInfo &getModuleFileInfo(const FileEntry *File) {
    llvm::MapVector<const FileEntry *, ModuleFileInfo>::iterator Known
      = ModuleFiles.find(File);
    if (Known != ModuleFiles.end())
      return Known->second;

    unsigned NewID = ModuleFiles.size();
    ModuleFileInfo &Info = ModuleFiles[File];
    Info.ID = NewID;
    return Info;
  }

public:
  explicit GlobalModuleIndexBuilder(
      FileManager &FileMgr, const PCHContainerReader &PCHContainerRdr)
      : FileMgr(FileMgr), PCHContainerRdr(PCHContainerRdr) {}

  /// Load the contents of the given module file into the builder.
  llvm::Error loadModuleFile(const FileEntry *File);

  /// Write the index to the given bitstream.
  /// \returns true if an error occurred, false otherwise.
  bool writeIndex(llvm::BitstreamWriter &Stream);
};
466}

468static void emitBlockID(unsigned ID, const char *Name,
                      llvm::BitstreamWriter &Stream,
                      SmallVectorImpl<uint64_t> &Record) {
Record.clear();
Record.push_back(ID);
Stream.EmitRecord(llvm::bitc::BLOCKINFO_CODE_SETBID, Record);

// Emit the block name if present.
if (!Name || Name[0] == 0) return;
Record.clear();
while (*Name)
  Record.push_back(*Name++);
Stream.EmitRecord(llvm::bitc::BLOCKINFO_CODE_BLOCKNAME, Record);
481}

483static void emitRecordID(unsigned ID, const char *Name,
                       llvm::BitstreamWriter &Stream,
                       SmallVectorImpl<uint64_t> &Record) {
Record.clear();
Record.push_back(ID);
while (*Name)
  Record.push_back(*Name++);
Stream.EmitRecord(llvm::bitc::BLOCKINFO_CODE_SETRECORDNAME, Record);
491}

493void
494GlobalModuleIndexBuilder::emitBlockInfoBlock(llvm::BitstreamWriter &Stream) {
SmallVector<uint64_t, 64> Record;
Stream.EnterBlockInfoBlock();

498#define BLOCK(X) emitBlockID(X ## _ID, #X, Stream, Record)
499#define RECORD(X) emitRecordID(X, #X, Stream, Record)
BLOCK(GLOBAL_INDEX_BLOCK);
RECORD(INDEX_METADATA);
RECORD(MODULE);
RECORD(IDENTIFIER_INDEX);
504#undef RECORD
505#undef BLOCK

Stream.ExitBlock();
508}

510namespace {
class InterestingASTIdentifierLookupTrait
  : public serialization::reader::ASTIdentifierLookupTraitBase {

public:
  /// The identifier and whether it is "interesting".
  typedef std::pair<StringRef, bool> data_type;

  data_type ReadData(const internal_key_type& k,
                     const unsigned char* d,
                     unsigned DataLen) {
    // The first bit indicates whether this identifier is interesting.
    // That's all we care about.
    using namespace llvm::support;
    unsigned RawID = endian::readNext<uint32_t, little, unaligned>(d);
    bool IsInteresting = RawID & 0x01;
    return std::make_pair(k, IsInteresting);
  }
};
529}

531llvm::Error GlobalModuleIndexBuilder::loadModuleFile(const FileEntry *File) {
// Open the module file.

auto Buffer = FileMgr.getBufferForFile(File, /*isVolatile=*/true);
if (!Buffer)
1
Taking false branch→
  return llvm::createStringError(Buffer.getError(),
                                 "failed getting buffer for module file");

// Initialize the input stream
llvm::BitstreamCursor InStream(PCHContainerRdr.ExtractPCH(**Buffer));

// Sniff for the signature.
for (unsigned char C : {'C', 'P', 'C', 'H'})
2
←
Assuming '__begin1' is equal to '__end1'→
  if (Expected<llvm::SimpleBitstreamCursor::word_t> Res = InStream.Read(8)) {
    if (Res.get() != C)
      return llvm::createStringError(std::errc::illegal_byte_sequence,
                                     "expected signature CPCH");
  } else
    return Res.takeError();

// Record this module file and assign it a unique ID (if it doesn't have
// one already).
unsigned ID = getModuleFileInfo(File).ID;

// Search for the blocks and records we care about.
enum { Other, ControlBlock, ASTBlock, DiagnosticOptionsBlock } State = Other;
bool Done = false;
while (!Done) {
3
←
Loop condition is true.  Entering loop body→
  Expected<llvm::BitstreamEntry> MaybeEntry = InStream.advance();
4
←
Calling 'BitstreamCursor::advance'→
  if (!MaybeEntry)
    return MaybeEntry.takeError();
  llvm::BitstreamEntry Entry = MaybeEntry.get();

  switch (Entry.Kind) {
  case llvm::BitstreamEntry::Error:
    Done = true;
    continue;

  case llvm::BitstreamEntry::Record:
    // In the 'other' state, just skip the record. We don't care.
    if (State == Other) {
      if (llvm::Expected<unsigned> Skipped = InStream.skipRecord(Entry.ID))
        continue;
      else
        return Skipped.takeError();
    }

    // Handle potentially-interesting records below.
    break;

  case llvm::BitstreamEntry::SubBlock:
    if (Entry.ID == CONTROL_BLOCK_ID) {
      if (llvm::Error Err = InStream.EnterSubBlock(CONTROL_BLOCK_ID))
        return Err;

      // Found the control block.
      State = ControlBlock;
      continue;
    }

    if (Entry.ID == AST_BLOCK_ID) {
      if (llvm::Error Err = InStream.EnterSubBlock(AST_BLOCK_ID))
        return Err;

      // Found the AST block.
      State = ASTBlock;
      continue;
    }

    if (Entry.ID == UNHASHED_CONTROL_BLOCK_ID) {
      if (llvm::Error Err = InStream.EnterSubBlock(UNHASHED_CONTROL_BLOCK_ID))
        return Err;

      // Found the Diagnostic Options block.
      State = DiagnosticOptionsBlock;
      continue;
    }

    if (llvm::Error Err = InStream.SkipBlock())
      return Err;

    continue;

  case llvm::BitstreamEntry::EndBlock:
    State = Other;
    continue;
  }

  // Read the given record.
  SmallVector<uint64_t, 64> Record;
  StringRef Blob;
  Expected<unsigned> MaybeCode = InStream.readRecord(Entry.ID, Record, &Blob);
  if (!MaybeCode)
    return MaybeCode.takeError();
  unsigned Code = MaybeCode.get();

  // Handle module dependencies.
  if (State == ControlBlock && Code == IMPORTS) {
    // Load each of the imported PCH files.
    unsigned Idx = 0, N = Record.size();
    while (Idx < N) {
      // Read information about the AST file.

      // Skip the imported kind
      ++Idx;

      // Skip the import location
      ++Idx;

      // Load stored size/modification time.
      off_t StoredSize = (off_t)Record[Idx++];
      time_t StoredModTime = (time_t)Record[Idx++];

      // Skip the stored signature.
      // FIXME: we could read the signature out of the import and validate it.
      ASTFileSignature StoredSignature = {
          {{(uint32_t)Record[Idx++], (uint32_t)Record[Idx++],
            (uint32_t)Record[Idx++], (uint32_t)Record[Idx++],
            (uint32_t)Record[Idx++]}}};

      // Skip the module name (currently this is only used for prebuilt
      // modules while here we are only dealing with cached).
      Idx += Record[Idx] + 1;

      // Retrieve the imported file name.
      unsigned Length = Record[Idx++];
      SmallString<128> ImportedFile(Record.begin() + Idx,
                                    Record.begin() + Idx + Length);
      Idx += Length;

      // Find the imported module file.
      auto DependsOnFile
        = FileMgr.getFile(ImportedFile, /*OpenFile=*/false,
                          /*CacheFailure=*/false);

      if (!DependsOnFile)
        return llvm::createStringError(std::errc::bad_file_descriptor,
                                       "imported file \"%s\" not found",
                                       ImportedFile.c_str());

      // Save the information in ImportedModuleFileInfo so we can verify after
      // loading all pcms.
      ImportedModuleFiles.insert(std::make_pair(
          *DependsOnFile, ImportedModuleFileInfo(StoredSize, StoredModTime,
                                                 StoredSignature)));

      // Record the dependency.
      unsigned DependsOnID = getModuleFileInfo(*DependsOnFile).ID;
      getModuleFileInfo(File).Dependencies.push_back(DependsOnID);
    }

    continue;
  }

  // Handle the identifier table
  if (State == ASTBlock && Code == IDENTIFIER_TABLE && Record[0] > 0) {
    typedef llvm::OnDiskIterableChainedHashTable<
        InterestingASTIdentifierLookupTrait> InterestingIdentifierTable;
    std::unique_ptr<InterestingIdentifierTable> Table(
        InterestingIdentifierTable::Create(
            (const unsigned char *)Blob.data() + Record[0],
            (const unsigned char *)Blob.data() + sizeof(uint32_t),
            (const unsigned char *)Blob.data()));
    for (InterestingIdentifierTable::data_iterator D = Table->data_begin(),
                                                   DEnd = Table->data_end();
         D != DEnd; ++D) {
      std::pair<StringRef, bool> Ident = *D;
      if (Ident.second)
        InterestingIdentifiers[Ident.first].push_back(ID);
      else
        (void)InterestingIdentifiers[Ident.first];
    }
  }

  // Get Signature.
  if (State == DiagnosticOptionsBlock && Code == SIGNATURE)
    getModuleFileInfo(File).Signature = {
        {{(uint32_t)Record[0], (uint32_t)Record[1], (uint32_t)Record[2],
          (uint32_t)Record[3], (uint32_t)Record[4]}}};

  // We don't care about this record.
}

return llvm::Error::success();
715}

717namespace {

719/// Trait used to generate the identifier index as an on-disk hash
720/// table.
721class IdentifierIndexWriterTrait {
722public:
typedef StringRef key_type;
typedef StringRef key_type_ref;
typedef SmallVector<unsigned, 2> data_type;
typedef const SmallVector<unsigned, 2> &data_type_ref;
typedef unsigned hash_value_type;
typedef unsigned offset_type;

static hash_value_type ComputeHash(key_type_ref Key) {
  return llvm::djbHash(Key);
}

std::pair<unsigned,unsigned>
EmitKeyDataLength(raw_ostream& Out, key_type_ref Key, data_type_ref Data) {
  using namespace llvm::support;
  endian::Writer LE(Out, little);
  unsigned KeyLen = Key.size();
  unsigned DataLen = Data.size() * 4;
  LE.write<uint16_t>(KeyLen);
  LE.write<uint16_t>(DataLen);
  return std::make_pair(KeyLen, DataLen);
}

void EmitKey(raw_ostream& Out, key_type_ref Key, unsigned KeyLen) {
  Out.write(Key.data(), KeyLen);
}

void EmitData(raw_ostream& Out, key_type_ref Key, data_type_ref Data,
              unsigned DataLen) {
  using namespace llvm::support;
  for (unsigned I = 0, N = Data.size(); I != N; ++I)
    endian::write<uint32_t>(Out, Data[I], little);
}
755};

757}

759bool GlobalModuleIndexBuilder::writeIndex(llvm::BitstreamWriter &Stream) {
for (auto MapEntry : ImportedModuleFiles) {
  auto *File = MapEntry.first;
  ImportedModuleFileInfo &Info = MapEntry.second;
  if (getModuleFileInfo(File).Signature) {
    if (getModuleFileInfo(File).Signature != Info.StoredSignature)
      // Verify Signature.
      return true;
  } else if (Info.StoredSize != File->getSize() ||
             Info.StoredModTime != File->getModificationTime())
    // Verify Size and ModTime.
    return true;
}

using namespace llvm;
llvm::TimeTraceScope TimeScope("Module WriteIndex");

// Emit the file header.
Stream.Emit((unsigned)'B', 8);
Stream.Emit((unsigned)'C', 8);
Stream.Emit((unsigned)'G', 8);
Stream.Emit((unsigned)'I', 8);

// Write the block-info block, which describes the records in this bitcode
// file.
emitBlockInfoBlock(Stream);

Stream.EnterSubblock(GLOBAL_INDEX_BLOCK_ID, 3);

// Write the metadata.
SmallVector<uint64_t, 2> Record;
Record.push_back(CurrentVersion);
Stream.EmitRecord(INDEX_METADATA, Record);

// Write the set of known module files.
for (ModuleFilesMap::iterator M = ModuleFiles.begin(),
                              MEnd = ModuleFiles.end();
     M != MEnd; ++M) {
  Record.clear();
  Record.push_back(M->second.ID);
  Record.push_back(M->first->getSize());
  Record.push_back(M->first->getModificationTime());

  // File name
  StringRef Name(M->first->getName());
  Record.push_back(Name.size());
  Record.append(Name.begin(), Name.end());

  // Dependencies
  Record.push_back(M->second.Dependencies.size());
  Record.append(M->second.Dependencies.begin(), M->second.Dependencies.end());
  Stream.EmitRecord(MODULE, Record);
}

// Write the identifier -> module file mapping.
{
  llvm::OnDiskChainedHashTableGenerator<IdentifierIndexWriterTrait> Generator;
  IdentifierIndexWriterTrait Trait;

  // Populate the hash table.
  for (InterestingIdentifierMap::iterator I = InterestingIdentifiers.begin(),
                                          IEnd = InterestingIdentifiers.end();
       I != IEnd; ++I) {
    Generator.insert(I->first(), I->second, Trait);
  }

  // Create the on-disk hash table in a buffer.
  SmallString<4096> IdentifierTable;
  uint32_t BucketOffset;
  {
    using namespace llvm::support;
    llvm::raw_svector_ostream Out(IdentifierTable);
    // Make sure that no bucket is at offset 0
    endian::write<uint32_t>(Out, 0, little);
    BucketOffset = Generator.Emit(Out, Trait);
  }

  // Create a blob abbreviation
  auto Abbrev = std::make_shared<BitCodeAbbrev>();
  Abbrev->Add(BitCodeAbbrevOp(IDENTIFIER_INDEX));
  Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Fixed, 32));
  Abbrev->Add(BitCodeAbbrevOp(BitCodeAbbrevOp::Blob));
  unsigned IDTableAbbrev = Stream.EmitAbbrev(std::move(Abbrev));

  // Write the identifier table
  uint64_t Record[] = {IDENTIFIER_INDEX, BucketOffset};
  Stream.EmitRecordWithBlob(IDTableAbbrev, Record, IdentifierTable);
}

Stream.ExitBlock();
return false;
850}

852llvm::Error
853GlobalModuleIndex::writeIndex(FileManager &FileMgr,
                            const PCHContainerReader &PCHContainerRdr,
                            StringRef Path) {
llvm::SmallString<128> IndexPath;
IndexPath += Path;
llvm::sys::path::append(IndexPath, IndexFileName);

// Coordinate building the global index file with other processes that might
// try to do the same.
llvm::LockFileManager Locked(IndexPath);
switch (Locked) {
case llvm::LockFileManager::LFS_Error:
  return llvm::createStringError(std::errc::io_error, "LFS error");

case llvm::LockFileManager::LFS_Owned:
  // We're responsible for building the index ourselves. Do so below.
  break;

case llvm::LockFileManager::LFS_Shared:
  // Someone else is responsible for building the index. We don't care
  // when they finish, so we're done.
  return llvm::createStringError(std::errc::device_or_resource_busy,
                                 "someone else is building the index");
}

// The module index builder.
GlobalModuleIndexBuilder Builder(FileMgr, PCHContainerRdr);

// Load each of the module files.
std::error_code EC;
for (llvm::sys::fs::directory_iterator D(Path, EC), DEnd;
     D != DEnd && !EC;
     D.increment(EC)) {
  // If this isn't a module file, we don't care.
  if (llvm::sys::path::extension(D->path()) != ".pcm") {
    // ... unless it's a .pcm.lock file, which indicates that someone is
    // in the process of rebuilding a module. They'll rebuild the index
    // at the end of that translation unit, so we don't have to.
    if (llvm::sys::path::extension(D->path()) == ".pcm.lock")
      return llvm::createStringError(std::errc::device_or_resource_busy,
                                     "someone else is building the index");

    continue;
  }

  // If we can't find the module file, skip it.
  auto ModuleFile = FileMgr.getFile(D->path());
  if (!ModuleFile)
    continue;

  // Load this module file.
  if (llvm::Error Err = Builder.loadModuleFile(*ModuleFile))
    return Err;
}

// The output buffer, into which the global index will be written.
SmallVector<char, 16> OutputBuffer;
{
  llvm::BitstreamWriter OutputStream(OutputBuffer);
  if (Builder.writeIndex(OutputStream))
    return llvm::createStringError(std::errc::io_error,
                                   "failed writing index");
}

return llvm::writeFileAtomically(
    (IndexPath + "-%%%%%%%%").str(), IndexPath,
    llvm::StringRef(OutputBuffer.data(), OutputBuffer.size()));
920}

922namespace {
class GlobalIndexIdentifierIterator : public IdentifierIterator {
  /// The current position within the identifier lookup table.
  IdentifierIndexTable::key_iterator Current;

  /// The end position within the identifier lookup table.
  IdentifierIndexTable::key_iterator End;

public:
  explicit GlobalIndexIdentifierIterator(IdentifierIndexTable &Idx) {
    Current = Idx.key_begin();
    End = Idx.key_end();
  }

  StringRef Next() override {
    if (Current == End)
      return StringRef();

    StringRef Result = *Current;
    ++Current;
    return Result;
  }
};
945}

947IdentifierIterator *GlobalModuleIndex::createIdentifierIterator() const {
IdentifierIndexTable &Table =
  *static_cast<IdentifierIndexTable *>(IdentifierIndex);
return new GlobalIndexIdentifierIterator(Table);
951}

←

/build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/llvm/include/llvm/Bitstream/BitstreamReader.h

→

1//===- BitstreamReader.h - Low-level bitstream reader interface -*- 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//
9// This header defines the BitstreamReader class.  This class can be used to
10// read an arbitrary bitstream, regardless of its contents.
11//
12//===----------------------------------------------------------------------===//

14#ifndef LLVM_BITSTREAM_BITSTREAMREADER_H
15#define LLVM_BITSTREAM_BITSTREAMREADER_H

17#include "llvm/ADT/ArrayRef.h"
18#include "llvm/ADT/SmallVector.h"
19#include "llvm/Bitstream/BitCodes.h"
20#include "llvm/Support/Endian.h"
21#include "llvm/Support/ErrorHandling.h"
22#include "llvm/Support/MathExtras.h"
23#include "llvm/Support/MemoryBuffer.h"
24#include <algorithm>
25#include <cassert>
26#include <climits>
27#include <cstddef>
28#include <cstdint>
29#include <memory>
30#include <string>
31#include <utility>
32#include <vector>

34namespace llvm {

36/// This class maintains the abbreviations read from a block info block.
37class BitstreamBlockInfo {
38public:
/// This contains information emitted to BLOCKINFO_BLOCK blocks. These
/// describe abbreviations that all blocks of the specified ID inherit.
struct BlockInfo {
  unsigned BlockID = 0;
  std::vector<std::shared_ptr<BitCodeAbbrev>> Abbrevs;
  std::string Name;
  std::vector<std::pair<unsigned, std::string>> RecordNames;
};

48private:
std::vector<BlockInfo> BlockInfoRecords;

51public:
/// If there is block info for the specified ID, return it, otherwise return
/// null.
const BlockInfo *getBlockInfo(unsigned BlockID) const {
  // Common case, the most recent entry matches BlockID.
  if (!BlockInfoRecords.empty() && BlockInfoRecords.back().BlockID == BlockID)
    return &BlockInfoRecords.back();

  for (unsigned i = 0, e = static_cast<unsigned>(BlockInfoRecords.size());
       i != e; ++i)
    if (BlockInfoRecords[i].BlockID == BlockID)
      return &BlockInfoRecords[i];
  return nullptr;
}

BlockInfo &getOrCreateBlockInfo(unsigned BlockID) {
  if (const BlockInfo *BI = getBlockInfo(BlockID))
    return *const_cast<BlockInfo*>(BI);

  // Otherwise, add a new record.
  BlockInfoRecords.emplace_back();
  BlockInfoRecords.back().BlockID = BlockID;
  return BlockInfoRecords.back();
}
75};

77/// This represents a position within a bitstream. There may be multiple
78/// independent cursors reading within one bitstream, each maintaining their
79/// own local state.
80class SimpleBitstreamCursor {
ArrayRef<uint8_t> BitcodeBytes;
size_t NextChar = 0;

84public:
/// This is the current data we have pulled from the stream but have not
/// returned to the client. This is specifically and intentionally defined to
/// follow the word size of the host machine for efficiency. We use word_t in
/// places that are aware of this to make it perfectly explicit what is going
/// on.
using word_t = size_t;

92private:
word_t CurWord = 0;

/// This is the number of bits in CurWord that are valid. This is always from
/// [0...bits_of(size_t)-1] inclusive.
unsigned BitsInCurWord = 0;

99public:
static const constexpr size_t MaxChunkSize = sizeof(word_t) * 8;

SimpleBitstreamCursor() = default;
explicit SimpleBitstreamCursor(ArrayRef<uint8_t> BitcodeBytes)
    : BitcodeBytes(BitcodeBytes) {}
explicit SimpleBitstreamCursor(StringRef BitcodeBytes)
    : BitcodeBytes(arrayRefFromStringRef(BitcodeBytes)) {}
explicit SimpleBitstreamCursor(MemoryBufferRef BitcodeBytes)
    : SimpleBitstreamCursor(BitcodeBytes.getBuffer()) {}

bool canSkipToPos(size_t pos) const {
  // pos can be skipped to if it is a valid address or one byte past the end.
  return pos <= BitcodeBytes.size();
}

bool AtEndOfStream() {
  return BitsInCurWord == 0 && BitcodeBytes.size() <= NextChar;
}

/// Return the bit # of the bit we are reading.
uint64_t GetCurrentBitNo() const {
  return NextChar*CHAR_BIT8 - BitsInCurWord;
}

// Return the byte # of the current bit.
uint64_t getCurrentByteNo() const { return GetCurrentBitNo() / 8; }

ArrayRef<uint8_t> getBitcodeBytes() const { return BitcodeBytes; }

/// Reset the stream to the specified bit number.
Error JumpToBit(uint64_t BitNo) {
  size_t ByteNo = size_t(BitNo/8) & ~(sizeof(word_t)-1);
  unsigned WordBitNo = unsigned(BitNo & (sizeof(word_t)*8-1));
  assert(canSkipToPos(ByteNo) && "Invalid location")((canSkipToPos(ByteNo) && "Invalid location") ? static_cast
<void> (0) : __assert_fail ("canSkipToPos(ByteNo) && \"Invalid location\""
, "/build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/llvm/include/llvm/Bitstream/BitstreamReader.h"
, 133, __PRETTY_FUNCTION__));

  // Move the cursor to the right word.
  NextChar = ByteNo;
  BitsInCurWord = 0;

  // Skip over any bits that are already consumed.
  if (WordBitNo) {
    if (Expected<word_t> Res = Read(WordBitNo))
      return Error::success();
    else
      return Res.takeError();
  }

  return Error::success();
}

/// Get a pointer into the bitstream at the specified byte offset.
const uint8_t *getPointerToByte(uint64_t ByteNo, uint64_t NumBytes) {
  return BitcodeBytes.data() + ByteNo;
}

/// Get a pointer into the bitstream at the specified bit offset.
///
/// The bit offset must be on a byte boundary.
const uint8_t *getPointerToBit(uint64_t BitNo, uint64_t NumBytes) {
  assert(!(BitNo % 8) && "Expected bit on byte boundary")((!(BitNo % 8) && "Expected bit on byte boundary") ? static_cast
<void> (0) : __assert_fail ("!(BitNo % 8) && \"Expected bit on byte boundary\""
, "/build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/llvm/include/llvm/Bitstream/BitstreamReader.h"
, 159, __PRETTY_FUNCTION__));
  return getPointerToByte(BitNo / 8, NumBytes);
}

Error fillCurWord() {
  if (NextChar >= BitcodeBytes.size())
    return createStringError(std::errc::io_error,
                             "Unexpected end of file reading %u of %u bytes",
                             NextChar, BitcodeBytes.size());

  // Read the next word from the stream.
  const uint8_t *NextCharPtr = BitcodeBytes.data() + NextChar;
  unsigned BytesRead;
  if (BitcodeBytes.size() >= NextChar + sizeof(word_t)) {
    BytesRead = sizeof(word_t);
    CurWord =
        support::endian::read<word_t, support::little, support::unaligned>(
            NextCharPtr);
  } else {
    // Short read.
    BytesRead = BitcodeBytes.size() - NextChar;
    CurWord = 0;
    for (unsigned B = 0; B != BytesRead; ++B)
      CurWord |= uint64_t(NextCharPtr[B]) << (B * 8);
  }
  NextChar += BytesRead;
  BitsInCurWord = BytesRead * 8;
  return Error::success();
}

Expected<word_t> Read(unsigned NumBits) {
  static const unsigned BitsInWord = MaxChunkSize;

  assert(NumBits && NumBits <= BitsInWord &&((NumBits && NumBits <= BitsInWord && "Cannot return zero or more than BitsInWord bits!"
) ? static_cast<void> (0) : __assert_fail ("NumBits && NumBits <= BitsInWord && \"Cannot return zero or more than BitsInWord bits!\""
, "/build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/llvm/include/llvm/Bitstream/BitstreamReader.h"
, 193, __PRETTY_FUNCTION__))
9
←
Assuming 'NumBits' is not equal to 0→
10
←
Assuming 'NumBits' is <= 'BitsInWord'→
11
←
'?' condition is true→
         "Cannot return zero or more than BitsInWord bits!")((NumBits && NumBits <= BitsInWord && "Cannot return zero or more than BitsInWord bits!"
) ? static_cast<void> (0) : __assert_fail ("NumBits && NumBits <= BitsInWord && \"Cannot return zero or more than BitsInWord bits!\""
, "/build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/llvm/include/llvm/Bitstream/BitstreamReader.h"
, 193, __PRETTY_FUNCTION__));

  static const unsigned Mask = sizeof(word_t) > 4 ? 0x3f : 0x1f;
12
←
'?' condition is true→

  // If the field is fully contained by CurWord, return it quickly.
  if (BitsInCurWord >= NumBits) {
13
←
Assuming 'NumBits' is > field 'BitsInCurWord'→
14
←
Taking false branch→
    word_t R = CurWord & (~word_t(0) >> (BitsInWord - NumBits));

    // Use a mask to avoid undefined behavior.
    CurWord >>= (NumBits & Mask);

    BitsInCurWord -= NumBits;
    return R;
  }

  word_t R = BitsInCurWord14.1
Field 'BitsInCurWord' is not equal to 0
1
Field 'BitsInCurWord' is not equal to 0
1
Field 'BitsInCurWord' is not equal to 0
 ? CurWord : 0;
15
←
'?' condition is true→
  unsigned BitsLeft = NumBits - BitsInCurWord;

  if (Error fillResult = fillCurWord())
16
←
Calling 'Error::operator bool'→
18
←
Returning from 'Error::operator bool'→
19
←
Taking false branch→
    return std::move(fillResult);

  // If we run out of data, abort.
  if (BitsLeft > BitsInCurWord)
20
←
Assuming 'BitsLeft' is <= field 'BitsInCurWord'→
21
←
Taking false branch→
    return createStringError(std::errc::io_error,
                             "Unexpected end of file reading %u of %u bits",
                             BitsInCurWord, BitsLeft);

  word_t R2 = CurWord & (~word_t(0) >> (BitsInWord - BitsLeft));
22
←
The result of the right shift is undefined due to shifting by '64', which is greater or equal to the width of type 'llvm::SimpleBitstreamCursor::word_t'

  // Use a mask to avoid undefined behavior.
  CurWord >>= (BitsLeft & Mask);

  BitsInCurWord -= BitsLeft;

  R |= R2 << (NumBits - BitsLeft);

  return R;
}

Expected<uint32_t> ReadVBR(unsigned NumBits) {
  Expected<unsigned> MaybeRead = Read(NumBits);
  if (!MaybeRead)
    return MaybeRead;
  uint32_t Piece = MaybeRead.get();

  if ((Piece & (1U << (NumBits-1))) == 0)
    return Piece;

  uint32_t Result = 0;
  unsigned NextBit = 0;
  while (true) {
    Result |= (Piece & ((1U << (NumBits-1))-1)) << NextBit;

    if ((Piece & (1U << (NumBits-1))) == 0)
      return Result;

    NextBit += NumBits-1;
    MaybeRead = Read(NumBits);
    if (!MaybeRead)
      return MaybeRead;
    Piece = MaybeRead.get();
  }
}

// Read a VBR that may have a value up to 64-bits in size. The chunk size of
// the VBR must still be <= 32 bits though.
Expected<uint64_t> ReadVBR64(unsigned NumBits) {
  Expected<uint64_t> MaybeRead = Read(NumBits);
  if (!MaybeRead)
    return MaybeRead;
  uint32_t Piece = MaybeRead.get();

  if ((Piece & (1U << (NumBits-1))) == 0)
    return uint64_t(Piece);

  uint64_t Result = 0;
  unsigned NextBit = 0;
  while (true) {
    Result |= uint64_t(Piece & ((1U << (NumBits-1))-1)) << NextBit;

    if ((Piece & (1U << (NumBits-1))) == 0)
      return Result;

    NextBit += NumBits-1;
    MaybeRead = Read(NumBits);
    if (!MaybeRead)
      return MaybeRead;
    Piece = MaybeRead.get();
  }
}

void SkipToFourByteBoundary() {
  // If word_t is 64-bits and if we've read less than 32 bits, just dump
  // the bits we have up to the next 32-bit boundary.
  if (sizeof(word_t) > 4 &&
      BitsInCurWord >= 32) {
    CurWord >>= BitsInCurWord-32;
    BitsInCurWord = 32;
    return;
  }

  BitsInCurWord = 0;
}

/// Return the size of the stream in bytes.
size_t SizeInBytes() const { return BitcodeBytes.size(); }

/// Skip to the end of the file.
void skipToEnd() { NextChar = BitcodeBytes.size(); }
302};

304/// When advancing through a bitstream cursor, each advance can discover a few
305/// different kinds of entries:
306struct BitstreamEntry {
enum {
  Error,    // Malformed bitcode was found.
  EndBlock, // We've reached the end of the current block, (or the end of the
            // file, which is treated like a series of EndBlock records.
  SubBlock, // This is the start of a new subblock of a specific ID.
  Record    // This is a record with a specific AbbrevID.
} Kind;

unsigned ID;

static BitstreamEntry getError() {
  BitstreamEntry E; E.Kind = Error; return E;
}

static BitstreamEntry getEndBlock() {
  BitstreamEntry E; E.Kind = EndBlock; return E;
}

static BitstreamEntry getSubBlock(unsigned ID) {
  BitstreamEntry E; E.Kind = SubBlock; E.ID = ID; return E;
}

static BitstreamEntry getRecord(unsigned AbbrevID) {
  BitstreamEntry E; E.Kind = Record; E.ID = AbbrevID; return E;
}
332};

334/// This represents a position within a bitcode file, implemented on top of a
335/// SimpleBitstreamCursor.
336///
337/// Unlike iterators, BitstreamCursors are heavy-weight objects that should not
338/// be passed by value.
339class BitstreamCursor : SimpleBitstreamCursor {
// This is the declared size of code values used for the current block, in
// bits.
unsigned CurCodeSize = 2;

/// Abbrevs installed at in this block.
std::vector<std::shared_ptr<BitCodeAbbrev>> CurAbbrevs;

struct Block {
  unsigned PrevCodeSize;
  std::vector<std::shared_ptr<BitCodeAbbrev>> PrevAbbrevs;

  explicit Block(unsigned PCS) : PrevCodeSize(PCS) {}
};

/// This tracks the codesize of parent blocks.
SmallVector<Block, 8> BlockScope;

BitstreamBlockInfo *BlockInfo = nullptr;

359public:
static const size_t MaxChunkSize = sizeof(word_t) * 8;

BitstreamCursor() = default;
explicit BitstreamCursor(ArrayRef<uint8_t> BitcodeBytes)
    : SimpleBitstreamCursor(BitcodeBytes) {}
explicit BitstreamCursor(StringRef BitcodeBytes)
    : SimpleBitstreamCursor(BitcodeBytes) {}
explicit BitstreamCursor(MemoryBufferRef BitcodeBytes)
    : SimpleBitstreamCursor(BitcodeBytes) {}

using SimpleBitstreamCursor::AtEndOfStream;
using SimpleBitstreamCursor::canSkipToPos;
using SimpleBitstreamCursor::fillCurWord;
using SimpleBitstreamCursor::getBitcodeBytes;
using SimpleBitstreamCursor::GetCurrentBitNo;
using SimpleBitstreamCursor::getCurrentByteNo;
using SimpleBitstreamCursor::getPointerToByte;
using SimpleBitstreamCursor::JumpToBit;
using SimpleBitstreamCursor::Read;
using SimpleBitstreamCursor::ReadVBR;
using SimpleBitstreamCursor::ReadVBR64;
using SimpleBitstreamCursor::SizeInBytes;
using SimpleBitstreamCursor::skipToEnd;

/// Return the number of bits used to encode an abbrev #.
unsigned getAbbrevIDWidth() const { return CurCodeSize; }

/// Flags that modify the behavior of advance().
enum {
  /// If this flag is used, the advance() method does not automatically pop
  /// the block scope when the end of a block is reached.
  AF_DontPopBlockAtEnd = 1,

  /// If this flag is used, abbrev entries are returned just like normal
  /// records.
  AF_DontAutoprocessAbbrevs = 2
};

/// Advance the current bitstream, returning the next entry in the stream.
Expected<BitstreamEntry> advance(unsigned Flags = 0) {
  while (true) {
5
←
Loop condition is true.  Entering loop body→
    if (AtEndOfStream())
6
←
Taking false branch→
      return BitstreamEntry::getError();

    Expected<unsigned> MaybeCode = ReadCode();
7
←
Calling 'BitstreamCursor::ReadCode'→
    if (!MaybeCode)
      return MaybeCode.takeError();
    unsigned Code = MaybeCode.get();

    if (Code == bitc::END_BLOCK) {
      // Pop the end of the block unless Flags tells us not to.
      if (!(Flags & AF_DontPopBlockAtEnd) && ReadBlockEnd())
        return BitstreamEntry::getError();
      return BitstreamEntry::getEndBlock();
    }

    if (Code == bitc::ENTER_SUBBLOCK) {
      if (Expected<unsigned> MaybeSubBlock = ReadSubBlockID())
        return BitstreamEntry::getSubBlock(MaybeSubBlock.get());
      else
        return MaybeSubBlock.takeError();
    }

    if (Code == bitc::DEFINE_ABBREV &&
        !(Flags & AF_DontAutoprocessAbbrevs)) {
      // We read and accumulate abbrev's, the client can't do anything with
      // them anyway.
      if (Error Err = ReadAbbrevRecord())
        return std::move(Err);
      continue;
    }

    return BitstreamEntry::getRecord(Code);
  }
}

/// This is a convenience function for clients that don't expect any
/// subblocks. This just skips over them automatically.
Expected<BitstreamEntry> advanceSkippingSubblocks(unsigned Flags = 0) {
  while (true) {
    // If we found a normal entry, return it.
    Expected<BitstreamEntry> MaybeEntry = advance(Flags);
    if (!MaybeEntry)
      return MaybeEntry;
    BitstreamEntry Entry = MaybeEntry.get();

    if (Entry.Kind != BitstreamEntry::SubBlock)
      return Entry;

    // If we found a sub-block, just skip over it and check the next entry.
    if (Error Err = SkipBlock())
      return std::move(Err);
  }
}

Expected<unsigned> ReadCode() { return Read(CurCodeSize); }
8
←
Calling 'SimpleBitstreamCursor::Read'→

// Block header:
//    [ENTER_SUBBLOCK, blockid, newcodelen, <align4bytes>, blocklen]

/// Having read the ENTER_SUBBLOCK code, read the BlockID for the block.
Expected<unsigned> ReadSubBlockID() { return ReadVBR(bitc::BlockIDWidth); }

/// Having read the ENTER_SUBBLOCK abbrevid and a BlockID, skip over the body
/// of this block.
Error SkipBlock() {
  // Read and ignore the codelen value.
  if (Expected<uint32_t> Res = ReadVBR(bitc::CodeLenWidth))
    ; // Since we are skipping this block, we don't care what code widths are
      // used inside of it.
  else
    return Res.takeError();

  SkipToFourByteBoundary();
  Expected<unsigned> MaybeNum = Read(bitc::BlockSizeWidth);
  if (!MaybeNum)
    return MaybeNum.takeError();
  size_t NumFourBytes = MaybeNum.get();

  // Check that the block wasn't partially defined, and that the offset isn't
  // bogus.
  size_t SkipTo = GetCurrentBitNo() + NumFourBytes * 4 * 8;
  if (AtEndOfStream())
    return createStringError(std::errc::illegal_byte_sequence,
                             "can't skip block: already at end of stream");
  if (!canSkipToPos(SkipTo / 8))
    return createStringError(std::errc::illegal_byte_sequence,
                             "can't skip to bit %zu from %" PRIu64"l" "u", SkipTo,
                             GetCurrentBitNo());

  if (Error Res = JumpToBit(SkipTo))
    return Res;

  return Error::success();
}

/// Having read the ENTER_SUBBLOCK abbrevid, and enter the block.
Error EnterSubBlock(unsigned BlockID, unsigned *NumWordsP = nullptr);

bool ReadBlockEnd() {
  if (BlockScope.empty()) return true;

  // Block tail:
  //    [END_BLOCK, <align4bytes>]
  SkipToFourByteBoundary();

  popBlockScope();
  return false;
}

510private:
void popBlockScope() {
  CurCodeSize = BlockScope.back().PrevCodeSize;

  CurAbbrevs = std::move(BlockScope.back().PrevAbbrevs);
  BlockScope.pop_back();
}

//===--------------------------------------------------------------------===//
// Record Processing
//===--------------------------------------------------------------------===//

522public:
/// Return the abbreviation for the specified AbbrevId.
const BitCodeAbbrev *getAbbrev(unsigned AbbrevID) {
  unsigned AbbrevNo = AbbrevID - bitc::FIRST_APPLICATION_ABBREV;
  if (AbbrevNo >= CurAbbrevs.size())
    report_fatal_error("Invalid abbrev number");
  return CurAbbrevs[AbbrevNo].get();
}

/// Read the current record and discard it, returning the code for the record.
Expected<unsigned> skipRecord(unsigned AbbrevID);

Expected<unsigned> readRecord(unsigned AbbrevID,
                              SmallVectorImpl<uint64_t> &Vals,
                              StringRef *Blob = nullptr);

//===--------------------------------------------------------------------===//
// Abbrev Processing
//===--------------------------------------------------------------------===//
Error ReadAbbrevRecord();

/// Read and return a block info block from the bitstream. If an error was
/// encountered, return None.
///
/// \param ReadBlockInfoNames Whether to read block/record name information in
/// the BlockInfo block. Only llvm-bcanalyzer uses this.
Expected<Optional<BitstreamBlockInfo>>
ReadBlockInfoBlock(bool ReadBlockInfoNames = false);

/// Set the block info to be used by this BitstreamCursor to interpret
/// abbreviated records.
void setBlockInfo(BitstreamBlockInfo *BI) { BlockInfo = BI; }
554};

556} // end llvm namespace

558#endif // LLVM_BITSTREAM_BITSTREAMREADER_H

←

/build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/llvm/include/llvm/Support/Error.h

1//===- llvm/Support/Error.h - Recoverable error handling --------*- 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//
9// This file defines an API used to report recoverable errors.
10//
11//===----------------------------------------------------------------------===//
12 
13#ifndef LLVM_SUPPORT_ERROR_H
14#define LLVM_SUPPORT_ERROR_H
15 
16#include "llvm-c/Error.h"
17#include "llvm/ADT/STLExtras.h"
18#include "llvm/ADT/SmallVector.h"
19#include "llvm/ADT/StringExtras.h"
20#include "llvm/ADT/Twine.h"
21#include "llvm/Config/abi-breaking.h"
22#include "llvm/Support/AlignOf.h"
23#include "llvm/Support/Compiler.h"
24#include "llvm/Support/Debug.h"
25#include "llvm/Support/ErrorHandling.h"
26#include "llvm/Support/ErrorOr.h"
27#include "llvm/Support/Format.h"
28#include "llvm/Support/raw_ostream.h"
29#include <algorithm>
30#include <cassert>
31#include <cstdint>
32#include <cstdlib>
33#include <functional>
34#include <memory>
35#include <new>
36#include <string>
37#include <system_error>
38#include <type_traits>
39#include <utility>
40#include <vector>
41 
42namespace llvm {
43 
44class ErrorSuccess;
45 
46/// Base class for error info classes. Do not extend this directly: Extend
47/// the ErrorInfo template subclass instead.
48class ErrorInfoBase {
49public:
50  virtual ~ErrorInfoBase() = default;
51 
52  /// Print an error message to an output stream.
53  virtual void log(raw_ostream &OS) const = 0;
54 
55  /// Return the error message as a string.
56  virtual std::string message() const {
57    std::string Msg;
58    raw_string_ostream OS(Msg);
59    log(OS);
60    return OS.str();
61  }
62 
63  /// Convert this error to a std::error_code.
64  ///
65  /// This is a temporary crutch to enable interaction with code still
66  /// using std::error_code. It will be removed in the future.
67  virtual std::error_code convertToErrorCode() const = 0;
68 
69  // Returns the class ID for this type.
70  static const void *classID() { return &ID; }
71 
72  // Returns the class ID for the dynamic type of this ErrorInfoBase instance.
73  virtual const void *dynamicClassID() const = 0;
74 
75  // Check whether this instance is a subclass of the class identified by
76  // ClassID.
77  virtual bool isA(const void *const ClassID) const {
78    return ClassID == classID();
79  }
80 
81  // Check whether this instance is a subclass of ErrorInfoT.
82  template <typename ErrorInfoT> bool isA() const {
83    return isA(ErrorInfoT::classID());
84  }
85 
86private:
87  virtual void anchor();
88 
89  static char ID;
90};
91 
92/// Lightweight error class with error context and mandatory checking.
93///
94/// Instances of this class wrap a ErrorInfoBase pointer. Failure states
95/// are represented by setting the pointer to a ErrorInfoBase subclass
96/// instance containing information describing the failure. Success is
97/// represented by a null pointer value.
98///
99/// Instances of Error also contains a 'Checked' flag, which must be set
100/// before the destructor is called, otherwise the destructor will trigger a
101/// runtime error. This enforces at runtime the requirement that all Error
102/// instances be checked or returned to the caller.
103///
104/// There are two ways to set the checked flag, depending on what state the
105/// Error instance is in. For Error instances indicating success, it
106/// is sufficient to invoke the boolean conversion operator. E.g.:
107///
108///   @code{.cpp}
109///   Error foo(<...>);
110///
111///   if (auto E = foo(<...>))
112///     return E; // <- Return E if it is in the error state.
113///   // We have verified that E was in the success state. It can now be safely
114///   // destroyed.
115///   @endcode
116///
117/// A success value *can not* be dropped. For example, just calling 'foo(<...>)'
118/// without testing the return value will raise a runtime error, even if foo
119/// returns success.
120///
121/// For Error instances representing failure, you must use either the
122/// handleErrors or handleAllErrors function with a typed handler. E.g.:
123///
124///   @code{.cpp}
125///   class MyErrorInfo : public ErrorInfo<MyErrorInfo> {
126///     // Custom error info.
127///   };
128///
129///   Error foo(<...>) { return make_error<MyErrorInfo>(...); }
130///
131///   auto E = foo(<...>); // <- foo returns failure with MyErrorInfo.
132///   auto NewE =
133///     handleErrors(E,
134///       [](const MyErrorInfo &M) {
135///         // Deal with the error.
136///       },
137///       [](std::unique_ptr<OtherError> M) -> Error {
138///         if (canHandle(*M)) {
139///           // handle error.
140///           return Error::success();
141///         }
142///         // Couldn't handle this error instance. Pass it up the stack.
143///         return Error(std::move(M));
144///       );
145///   // Note - we must check or return NewE in case any of the handlers
146///   // returned a new error.
147///   @endcode
148///
149/// The handleAllErrors function is identical to handleErrors, except
150/// that it has a void return type, and requires all errors to be handled and
151/// no new errors be returned. It prevents errors (assuming they can all be
152/// handled) from having to be bubbled all the way to the top-level.
153///
154/// *All* Error instances must be checked before destruction, even if
155/// they're moved-assigned or constructed from Success values that have already
156/// been checked. This enforces checking through all levels of the call stack.
157class LLVM_NODISCARD[[clang::warn_unused_result]] Error {
158  // ErrorList needs to be able to yank ErrorInfoBase pointers out of Errors
159  // to add to the error list. It can't rely on handleErrors for this, since
160  // handleErrors does not support ErrorList handlers.
161  friend class ErrorList;
162 
163  // handleErrors needs to be able to set the Checked flag.
164  template <typename... HandlerTs>
165  friend Error handleErrors(Error E, HandlerTs &&... Handlers);
166 
167  // Expected<T> needs to be able to steal the payload when constructed from an
168  // error.
169  template <typename T> friend class Expected;
170 
171  // wrap needs to be able to steal the payload.
172  friend LLVMErrorRef wrap(Error);
173 
174protected:
175  /// Create a success value. Prefer using 'Error::success()' for readability
176  Error() {
177    setPtr(nullptr);
178    setChecked(false);
179  }
180 
181public:
182  /// Create a success value.
183  static ErrorSuccess success();
184 
185  // Errors are not copy-constructable.
186  Error(const Error &Other) = delete;
187 
188  /// Move-construct an error value. The newly constructed error is considered
189  /// unchecked, even if the source error had been checked. The original error
190  /// becomes a checked Success value, regardless of its original state.
191  Error(Error &&Other) {
192    setChecked(true);
193    *this = std::move(Other);
194  }
195 
196  /// Create an error value. Prefer using the 'make_error' function, but
197  /// this constructor can be useful when "re-throwing" errors from handlers.
198  Error(std::unique_ptr<ErrorInfoBase> Payload) {
199    setPtr(Payload.release());
200    setChecked(false);
201  }
202 
203  // Errors are not copy-assignable.
204  Error &operator=(const Error &Other) = delete;
205 
206  /// Move-assign an error value. The current error must represent success, you
207  /// you cannot overwrite an unhandled error. The current error is then
208  /// considered unchecked. The source error becomes a checked success value,
209  /// regardless of its original state.
210  Error &operator=(Error &&Other) {
211    // Don't allow overwriting of unchecked values.
212    assertIsChecked();
213    setPtr(Other.getPtr());
214 
215    // This Error is unchecked, even if the source error was checked.
216    setChecked(false);
217 
218    // Null out Other's payload and set its checked bit.
219    Other.setPtr(nullptr);
220    Other.setChecked(true);
221 
222    return *this;
223  }
224 
225  /// Destroy a Error. Fails with a call to abort() if the error is
226  /// unchecked.
227  ~Error() {
228    assertIsChecked();
229    delete getPtr();
230  }
231 
232  /// Bool conversion. Returns true if this Error is in a failure state,
233  /// and false if it is in an accept state. If the error is in a Success state
234  /// it will be considered checked.
235  explicit operator bool() {
236    setChecked(getPtr() == nullptr);
237    return getPtr() != nullptr;
17
←
Returning zero, which participates in a condition later→
238  }
239 
240  /// Check whether one error is a subclass of another.
241  template <typename ErrT> bool isA() const {
242    return getPtr() && getPtr()->isA(ErrT::classID());
243  }
244 
245  /// Returns the dynamic class id of this error, or null if this is a success
246  /// value.
247  const void* dynamicClassID() const {
248    if (!getPtr())
249      return nullptr;
250    return getPtr()->dynamicClassID();
251  }
252 
253private:
254#if LLVM_ENABLE_ABI_BREAKING_CHECKS1
255  // assertIsChecked() happens very frequently, but under normal circumstances
256  // is supposed to be a no-op.  So we want it to be inlined, but having a bunch
257  // of debug prints can cause the function to be too large for inlining.  So
258  // it's important that we define this function out of line so that it can't be
259  // inlined.
260  LLVM_ATTRIBUTE_NORETURN__attribute__((noreturn))
261  void fatalUncheckedError() const;
262#endif
263 
264  void assertIsChecked() {
265#if LLVM_ENABLE_ABI_BREAKING_CHECKS1
266    if (LLVM_UNLIKELY(!getChecked() || getPtr())__builtin_expect((bool)(!getChecked() || getPtr()), false))
267      fatalUncheckedError();
268#endif
269  }
270 
271  ErrorInfoBase *getPtr() const {
272    return reinterpret_cast<ErrorInfoBase*>(
273             reinterpret_cast<uintptr_t>(Payload) &
274             ~static_cast<uintptr_t>(0x1));
275  }
276 
277  void setPtr(ErrorInfoBase *EI) {
278#if LLVM_ENABLE_ABI_BREAKING_CHECKS1
279    Payload = reinterpret_cast<ErrorInfoBase*>(
280                (reinterpret_cast<uintptr_t>(EI) &
281                 ~static_cast<uintptr_t>(0x1)) |
282                (reinterpret_cast<uintptr_t>(Payload) & 0x1));
283#else
284    Payload = EI;
285#endif
286  }
287 
288  bool getChecked() const {
289#if LLVM_ENABLE_ABI_BREAKING_CHECKS1
290    return (reinterpret_cast<uintptr_t>(Payload) & 0x1) == 0;
291#else
292    return true;
293#endif
294  }
295 
296  void setChecked(bool V) {
297    Payload = reinterpret_cast<ErrorInfoBase*>(
298                (reinterpret_cast<uintptr_t>(Payload) &
299                  ~static_cast<uintptr_t>(0x1)) |
300                  (V ? 0 : 1));
301  }
302 
303  std::unique_ptr<ErrorInfoBase> takePayload() {
304    std::unique_ptr<ErrorInfoBase> Tmp(getPtr());
305    setPtr(nullptr);
306    setChecked(true);
307    return Tmp;
308  }
309 
310  friend raw_ostream &operator<<(raw_ostream &OS, const Error &E) {
311    if (auto P = E.getPtr())
312      P->log(OS);
313    else
314      OS << "success";
315    return OS;
316  }
317 
318  ErrorInfoBase *Payload = nullptr;
319};
320 
321/// Subclass of Error for the sole purpose of identifying the success path in
322/// the type system. This allows to catch invalid conversion to Expected<T> at
323/// compile time.
324class ErrorSuccess final : public Error {};
325 
326inline ErrorSuccess Error::success() { return ErrorSuccess(); }
327 
328/// Make a Error instance representing failure using the given error info
329/// type.
330template <typename ErrT, typename... ArgTs> Error make_error(ArgTs &&... Args) {
331  return Error(std::make_unique<ErrT>(std::forward<ArgTs>(Args)...));
332}
333 
334/// Base class for user error types. Users should declare their error types
335/// like:
336///
337/// class MyError : public ErrorInfo<MyError> {
338///   ....
339/// };
340///
341/// This class provides an implementation of the ErrorInfoBase::kind
342/// method, which is used by the Error RTTI system.
343template <typename ThisErrT, typename ParentErrT = ErrorInfoBase>
344class ErrorInfo : public ParentErrT {
345public:
346  using ParentErrT::ParentErrT; // inherit constructors
347 
348  static const void *classID() { return &ThisErrT::ID; }
349 
350  const void *dynamicClassID() const override { return &ThisErrT::ID; }
351 
352  bool isA(const void *const ClassID) const override {
353    return ClassID == classID() || ParentErrT::isA(ClassID);
354  }
355};
356 
357/// Special ErrorInfo subclass representing a list of ErrorInfos.
358/// Instances of this class are constructed by joinError.
359class ErrorList final : public ErrorInfo<ErrorList> {
360  // handleErrors needs to be able to iterate the payload list of an
361  // ErrorList.
362  template <typename... HandlerTs>
363  friend Error handleErrors(Error E, HandlerTs &&... Handlers);
364 
365  // joinErrors is implemented in terms of join.
366  friend Error joinErrors(Error, Error);
367 
368public:
369  void log(raw_ostream &OS) const override {
370    OS << "Multiple errors:\n";
371    for (auto &ErrPayload : Payloads) {
372      ErrPayload->log(OS);
373      OS << "\n";
374    }
375  }
376 
377  std::error_code convertToErrorCode() const override;
378 
379  // Used by ErrorInfo::classID.
380  static char ID;
381 
382private:
383  ErrorList(std::unique_ptr<ErrorInfoBase> Payload1,
384            std::unique_ptr<ErrorInfoBase> Payload2) {
385    assert(!Payload1->isA<ErrorList>() && !Payload2->isA<ErrorList>() &&((!Payload1->isA<ErrorList>() && !Payload2->
isA<ErrorList>() && "ErrorList constructor payloads should be singleton errors"
) ? static_cast<void> (0) : __assert_fail ("!Payload1->isA<ErrorList>() && !Payload2->isA<ErrorList>() && \"ErrorList constructor payloads should be singleton errors\""
, "/build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/llvm/include/llvm/Support/Error.h"
, 386, __PRETTY_FUNCTION__))
386           "ErrorList constructor payloads should be singleton errors")((!Payload1->isA<ErrorList>() && !Payload2->
isA<ErrorList>() && "ErrorList constructor payloads should be singleton errors"
) ? static_cast<void> (0) : __assert_fail ("!Payload1->isA<ErrorList>() && !Payload2->isA<ErrorList>() && \"ErrorList constructor payloads should be singleton errors\""
, "/build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/llvm/include/llvm/Support/Error.h"
, 386, __PRETTY_FUNCTION__));
387    Payloads.push_back(std::move(Payload1));
388    Payloads.push_back(std::move(Payload2));
389  }
390 
391  static Error join(Error E1, Error E2) {
392    if (!E1)
393      return E2;
394    if (!E2)
395      return E1;
396    if (E1.isA<ErrorList>()) {
397      auto &E1List = static_cast<ErrorList &>(*E1.getPtr());
398      if (E2.isA<ErrorList>()) {
399        auto E2Payload = E2.takePayload();
400        auto &E2List = static_cast<ErrorList &>(*E2Payload);
401        for (auto &Payload : E2List.Payloads)
402          E1List.Payloads.push_back(std::move(Payload));
403      } else
404        E1List.Payloads.push_back(E2.takePayload());
405 
406      return E1;
407    }
408    if (E2.isA<ErrorList>()) {
409      auto &E2List = static_cast<ErrorList &>(*E2.getPtr());
410      E2List.Payloads.insert(E2List.Payloads.begin(), E1.takePayload());
411      return E2;
412    }
413    return Error(std::unique_ptr<ErrorList>(
414        new ErrorList(E1.takePayload(), E2.takePayload())));
415  }
416 
417  std::vector<std::unique_ptr<ErrorInfoBase>> Payloads;
418};
419 
420/// Concatenate errors. The resulting Error is unchecked, and contains the
421/// ErrorInfo(s), if any, contained in E1, followed by the
422/// ErrorInfo(s), if any, contained in E2.
423inline Error joinErrors(Error E1, Error E2) {
424  return ErrorList::join(std::move(E1), std::move(E2));
425}
426 
427/// Tagged union holding either a T or a Error.
428///
429/// This class parallels ErrorOr, but replaces error_code with Error. Since
430/// Error cannot be copied, this class replaces getError() with
431/// takeError(). It also adds an bool errorIsA<ErrT>() method for testing the
432/// error class type.
433template <class T> class LLVM_NODISCARD[[clang::warn_unused_result]] Expected {
434  template <class T1> friend class ExpectedAsOutParameter;
435  template <class OtherT> friend class Expected;
436 
437  static const bool isRef = std::is_reference<T>::value;
438 
439  using wrap = std::reference_wrapper<typename std::remove_reference<T>::type>;
440 
441  using error_type = std::unique_ptr<ErrorInfoBase>;
442 
443public:
444  using storage_type = typename std::conditional<isRef, wrap, T>::type;
445  using value_type = T;
446 
447private:
448  using reference = typename std::remove_reference<T>::type &;
449  using const_reference = const typename std::remove_reference<T>::type &;
450  using pointer = typename std::remove_reference<T>::type *;
451  using const_pointer = const typename std::remove_reference<T>::type *;
452 
453public:
454  /// Create an Expected<T> error value from the given Error.
455  Expected(Error Err)
456      : HasError(true)
457#if LLVM_ENABLE_ABI_BREAKING_CHECKS1
458        // Expected is unchecked upon construction in Debug builds.
459        , Unchecked(true)
460#endif
461  {
462    assert(Err && "Cannot create Expected<T> from Error success value.")((Err && "Cannot create Expected<T> from Error success value."
) ? static_cast<void> (0) : __assert_fail ("Err && \"Cannot create Expected<T> from Error success value.\""
, "/build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/llvm/include/llvm/Support/Error.h"
, 462, __PRETTY_FUNCTION__));
463    new (getErrorStorage()) error_type(Err.takePayload());
464  }
465 
466  /// Forbid to convert from Error::success() implicitly, this avoids having
467  /// Expected<T> foo() { return Error::success(); } which compiles otherwise
468  /// but triggers the assertion above.
469  Expected(ErrorSuccess) = delete;
470 
471  /// Create an Expected<T> success value from the given OtherT value, which
472  /// must be convertible to T.
473  template <typename OtherT>
474  Expected(OtherT &&Val,
475           typename std::enable_if<std::is_convertible<OtherT, T>::value>::type
476               * = nullptr)
477      : HasError(false)
478#if LLVM_ENABLE_ABI_BREAKING_CHECKS1
479        // Expected is unchecked upon construction in Debug builds.
480        , Unchecked(true)
481#endif
482  {
483    new (getStorage()) storage_type(std::forward<OtherT>(Val));
484  }
485 
486  /// Move construct an Expected<T> value.
487  Expected(Expected &&Other) { moveConstruct(std::move(Other)); }
488 
489  /// Move construct an Expected<T> value from an Expected<OtherT>, where OtherT
490  /// must be convertible to T.
491  template <class OtherT>
492  Expected(Expected<OtherT> &&Other,
493           typename std::enable_if<std::is_convertible<OtherT, T>::value>::type
494               * = nullptr) {
495    moveConstruct(std::move(Other));
496  }
497 
498  /// Move construct an Expected<T> value from an Expected<OtherT>, where OtherT
499  /// isn't convertible to T.
500  template <class OtherT>
501  explicit Expected(
502      Expected<OtherT> &&Other,
503      typename std::enable_if<!std::is_convertible<OtherT, T>::value>::type * =
504          nullptr) {
505    moveConstruct(std::move(Other));
506  }
507 
508  /// Move-assign from another Expected<T>.
509  Expected &operator=(Expected &&Other) {
510    moveAssign(std::move(Other));
511    return *this;
512  }
513 
514  /// Destroy an Expected<T>.
515  ~Expected() {
516    assertIsChecked();
517    if (!HasError)
518      getStorage()->~storage_type();
519    else
520      getErrorStorage()->~error_type();
521  }
522 
523  /// Return false if there is an error.
524  explicit operator bool() {
525#if LLVM_ENABLE_ABI_BREAKING_CHECKS1
526    Unchecked = HasError;
527#endif
528    return !HasError;
529  }
530 
531  /// Returns a reference to the stored T value.
532  reference get() {
533    assertIsChecked();
534    return *getStorage();
535  }
536 
537  /// Returns a const reference to the stored T value.
538  const_reference get() const {
539    assertIsChecked();
540    return const_cast<Expected<T> *>(this)->get();
541  }
542 
543  /// Check that this Expected<T> is an error of type ErrT.
544  template <typename ErrT> bool errorIsA() const {
545    return HasError && (*getErrorStorage())->template isA<ErrT>();
546  }
547 
548  /// Take ownership of the stored error.
549  /// After calling this the Expected<T> is in an indeterminate state that can
550  /// only be safely destructed. No further calls (beside the destructor) should
551  /// be made on the Expected<T> value.
552  Error takeError() {
553#if LLVM_ENABLE_ABI_BREAKING_CHECKS1
554    Unchecked = false;
555#endif
556    return HasError ? Error(std::move(*getErrorStorage())) : Error::success();
557  }
558 
559  /// Returns a pointer to the stored T value.
560  pointer operator->() {
561    assertIsChecked();
562    return toPointer(getStorage());
563  }
564 
565  /// Returns a const pointer to the stored T value.
566  const_pointer operator->() const {
567    assertIsChecked();
568    return toPointer(getStorage());
569  }
570 
571  /// Returns a reference to the stored T value.
572  reference operator*() {
573    assertIsChecked();
574    return *getStorage();
575  }
576 
577  /// Returns a const reference to the stored T value.
578  const_reference operator*() const {
579    assertIsChecked();
580    return *getStorage();
581  }
582 
583private:
584  template <class T1>
585  static bool compareThisIfSameType(const T1 &a, const T1 &b) {
586    return &a == &b;
587  }
588 
589  template <class T1, class T2>
590  static bool compareThisIfSameType(const T1 &a, const T2 &b) {
591    return false;
592  }
593 
594  template <class OtherT> void moveConstruct(Expected<OtherT> &&Other) {
595    HasError = Other.HasError;
596#if LLVM_ENABLE_ABI_BREAKING_CHECKS1
597    Unchecked = true;
598    Other.Unchecked = false;
599#endif
600 
601    if (!HasError)
602      new (getStorage()) storage_type(std::move(*Other.getStorage()));
603    else
604      new (getErrorStorage()) error_type(std::move(*Other.getErrorStorage()));
605  }
606 
607  template <class OtherT> void moveAssign(Expected<OtherT> &&Other) {
608    assertIsChecked();
609 
610    if (compareThisIfSameType(*this, Other))
611      return;
612 
613    this->~Expected();
614    new (this) Expected(std::move(Other));
615  }
616 
617  pointer toPointer(pointer Val) { return Val; }
618 
619  const_pointer toPointer(const_pointer Val) const { return Val; }
620 
621  pointer toPointer(wrap *Val) { return &Val->get(); }
622 
623  const_pointer toPointer(const wrap *Val) const { return &Val->get(); }
624 
625  storage_type *getStorage() {
626    assert(!HasError && "Cannot get value when an error exists!")((!HasError && "Cannot get value when an error exists!"
) ? static_cast<void> (0) : __assert_fail ("!HasError && \"Cannot get value when an error exists!\""
, "/build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/llvm/include/llvm/Support/Error.h"
, 626, __PRETTY_FUNCTION__));
627    return reinterpret_cast<storage_type *>(TStorage.buffer);
628  }
629 
630  const storage_type *getStorage() const {
631    assert(!HasError && "Cannot get value when an error exists!")((!HasError && "Cannot get value when an error exists!"
) ? static_cast<void> (0) : __assert_fail ("!HasError && \"Cannot get value when an error exists!\""
, "/build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/llvm/include/llvm/Support/Error.h"
, 631, __PRETTY_FUNCTION__));
632    return reinterpret_cast<const storage_type *>(TStorage.buffer);
633  }
634 
635  error_type *getErrorStorage() {
636    assert(HasError && "Cannot get error when a value exists!")((HasError && "Cannot get error when a value exists!"
) ? static_cast<void> (0) : __assert_fail ("HasError && \"Cannot get error when a value exists!\""
, "/build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/llvm/include/llvm/Support/Error.h"
, 636, __PRETTY_FUNCTION__));
637    return reinterpret_cast<error_type *>(ErrorStorage.buffer);
638  }
639 
640  const error_type *getErrorStorage() const {
641    assert(HasError && "Cannot get error when a value exists!")((HasError && "Cannot get error when a value exists!"
) ? static_cast<void> (0) : __assert_fail ("HasError && \"Cannot get error when a value exists!\""
, "/build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/llvm/include/llvm/Support/Error.h"
, 641, __PRETTY_FUNCTION__));
642    return reinterpret_cast<const error_type *>(ErrorStorage.buffer);
643  }
644 
645  // Used by ExpectedAsOutParameter to reset the checked flag.
646  void setUnchecked() {
647#if LLVM_ENABLE_ABI_BREAKING_CHECKS1
648    Unchecked = true;
649#endif
650  }
651 
652#if LLVM_ENABLE_ABI_BREAKING_CHECKS1
653  LLVM_ATTRIBUTE_NORETURN__attribute__((noreturn))
654  LLVM_ATTRIBUTE_NOINLINE__attribute__((noinline))
655  void fatalUncheckedExpected() const {
656    dbgs() << "Expected<T> must be checked before access or destruction.\n";
657    if (HasError) {
658      dbgs() << "Unchecked Expected<T> contained error:\n";
659      (*getErrorStorage())->log(dbgs());
660    } else
661      dbgs() << "Expected<T> value was in success state. (Note: Expected<T> "
662                "values in success mode must still be checked prior to being "
663                "destroyed).\n";
664    abort();
665  }
666#endif
667 
668  void assertIsChecked() {
669#if LLVM_ENABLE_ABI_BREAKING_CHECKS1
670    if (LLVM_UNLIKELY(Unchecked)__builtin_expect((bool)(Unchecked), false))
671      fatalUncheckedExpected();
672#endif
673  }
674 
675  union {
676    AlignedCharArrayUnion<storage_type> TStorage;
677    AlignedCharArrayUnion<error_type> ErrorStorage;
678  };
679  bool HasError : 1;
680#if LLVM_ENABLE_ABI_BREAKING_CHECKS1
681  bool Unchecked : 1;
682#endif
683};
684 
685/// Report a serious error, calling any installed error handler. See
686/// ErrorHandling.h.
687LLVM_ATTRIBUTE_NORETURN__attribute__((noreturn)) void report_fatal_error(Error Err,
688                                                bool gen_crash_diag = true);
689 
690/// Report a fatal error if Err is a failure value.
691///
692/// This function can be used to wrap calls to fallible functions ONLY when it
693/// is known that the Error will always be a success value. E.g.
694///
695///   @code{.cpp}
696///   // foo only attempts the fallible operation if DoFallibleOperation is
697///   // true. If DoFallibleOperation is false then foo always returns
698///   // Error::success().
699///   Error foo(bool DoFallibleOperation);
700///
701///   cantFail(foo(false));
702///   @endcode
703inline void cantFail(Error Err, const char *Msg = nullptr) {
704  if (Err) {
705    if (!Msg)
706      Msg = "Failure value returned from cantFail wrapped call";
707#ifndef NDEBUG
708    std::string Str;
709    raw_string_ostream OS(Str);
710    OS << Msg << "\n" << Err;
711    Msg = OS.str().c_str();
712#endif
713    llvm_unreachable(Msg)::llvm::llvm_unreachable_internal(Msg, "/build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/llvm/include/llvm/Support/Error.h"
, 713);
714  }
715}
716 
717/// Report a fatal error if ValOrErr is a failure value, otherwise unwraps and
718/// returns the contained value.
719///
720/// This function can be used to wrap calls to fallible functions ONLY when it
721/// is known that the Error will always be a success value. E.g.
722///
723///   @code{.cpp}
724///   // foo only attempts the fallible operation if DoFallibleOperation is
725///   // true. If DoFallibleOperation is false then foo always returns an int.
726///   Expected<int> foo(bool DoFallibleOperation);
727///
728///   int X = cantFail(foo(false));
729///   @endcode
730template <typename T>
731T cantFail(Expected<T> ValOrErr, const char *Msg = nullptr) {
732  if (ValOrErr)
733    return std::move(*ValOrErr);
734  else {
735    if (!Msg)
736      Msg = "Failure value returned from cantFail wrapped call";
737#ifndef NDEBUG
738    std::string Str;
739    raw_string_ostream OS(Str);
740    auto E = ValOrErr.takeError();
741    OS << Msg << "\n" << E;
742    Msg = OS.str().c_str();
743#endif
744    llvm_unreachable(Msg)::llvm::llvm_unreachable_internal(Msg, "/build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/llvm/include/llvm/Support/Error.h"
, 744);
745  }
746}
747 
748/// Report a fatal error if ValOrErr is a failure value, otherwise unwraps and
749/// returns the contained reference.
750///
751/// This function can be used to wrap calls to fallible functions ONLY when it
752/// is known that the Error will always be a success value. E.g.
753///
754///   @code{.cpp}
755///   // foo only attempts the fallible operation if DoFallibleOperation is
756///   // true. If DoFallibleOperation is false then foo always returns a Bar&.
757///   Expected<Bar&> foo(bool DoFallibleOperation);
758///
759///   Bar &X = cantFail(foo(false));
760///   @endcode
761template <typename T>
762T& cantFail(Expected<T&> ValOrErr, const char *Msg = nullptr) {
763  if (ValOrErr)
764    return *ValOrErr;
765  else {
766    if (!Msg)
767      Msg = "Failure value returned from cantFail wrapped call";
768#ifndef NDEBUG
769    std::string Str;
770    raw_string_ostream OS(Str);
771    auto E = ValOrErr.takeError();
772    OS << Msg << "\n" << E;
773    Msg = OS.str().c_str();
774#endif
775    llvm_unreachable(Msg)::llvm::llvm_unreachable_internal(Msg, "/build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/llvm/include/llvm/Support/Error.h"
, 775);
776  }
777}
778 
779/// Helper for testing applicability of, and applying, handlers for
780/// ErrorInfo types.
781template <typename HandlerT>
782class ErrorHandlerTraits
783    : public ErrorHandlerTraits<decltype(
784          &std::remove_reference<HandlerT>::type::operator())> {};
785 
786// Specialization functions of the form 'Error (const ErrT&)'.
787template <typename ErrT> class ErrorHandlerTraits<Error (&)(ErrT &)> {
788public:
789  static bool appliesTo(const ErrorInfoBase &E) {
790    return E.template isA<ErrT>();
791  }
792 
793  template <typename HandlerT>
794  static Error apply(HandlerT &&H, std::unique_ptr<ErrorInfoBase> E) {
795    assert(appliesTo(*E) && "Applying incorrect handler")((appliesTo(*E) && "Applying incorrect handler") ? static_cast
<void> (0) : __assert_fail ("appliesTo(*E) && \"Applying incorrect handler\""
, "/build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/llvm/include/llvm/Support/Error.h"
, 795, __PRETTY_FUNCTION__));
796    return H(static_cast<ErrT &>(*E));
797  }
798};
799 
800// Specialization functions of the form 'void (const ErrT&)'.
801template <typename ErrT> class ErrorHandlerTraits<void (&)(ErrT &)> {
802public:
803  static bool appliesTo(const ErrorInfoBase &E) {
804    return E.template isA<ErrT>();
805  }
806 
807  template <typename HandlerT>
808  static Error apply(HandlerT &&H, std::unique_ptr<ErrorInfoBase> E) {
809    assert(appliesTo(*E) && "Applying incorrect handler")((appliesTo(*E) && "Applying incorrect handler") ? static_cast
<void> (0) : __assert_fail ("appliesTo(*E) && \"Applying incorrect handler\""
, "/build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/llvm/include/llvm/Support/Error.h"
, 809, __PRETTY_FUNCTION__));
810    H(static_cast<ErrT &>(*E));
811    return Error::success();
812  }
813};
814 
815/// Specialization for functions of the form 'Error (std::unique_ptr<ErrT>)'.
816template <typename ErrT>
817class ErrorHandlerTraits<Error (&)(std::unique_ptr<ErrT>)> {
818public:
819  static bool appliesTo(const ErrorInfoBase &E) {
820    return E.template isA<ErrT>();
821  }
822 
823  template <typename HandlerT>
824  static Error apply(HandlerT &&H, std::unique_ptr<ErrorInfoBase> E) {
825    assert(appliesTo(*E) && "Applying incorrect handler")((appliesTo(*E) && "Applying incorrect handler") ? static_cast
<void> (0) : __assert_fail ("appliesTo(*E) && \"Applying incorrect handler\""
, "/build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/llvm/include/llvm/Support/Error.h"
, 825, __PRETTY_FUNCTION__));
826    std::unique_ptr<ErrT> SubE(static_cast<ErrT *>(E.release()));
827    return H(std::move(SubE));
828  }
829};
830 
831/// Specialization for functions of the form 'void (std::unique_ptr<ErrT>)'.
832template <typename ErrT>
833class ErrorHandlerTraits<void (&)(std::unique_ptr<ErrT>)> {
834public:
835  static bool appliesTo(const ErrorInfoBase &E) {
836    return E.template isA<ErrT>();
837  }
838 
839  template <typename HandlerT>
840  static Error apply(HandlerT &&H, std::unique_ptr<ErrorInfoBase> E) {
841    assert(appliesTo(*E) && "Applying incorrect handler")((appliesTo(*E) && "Applying incorrect handler") ? static_cast
<void> (0) : __assert_fail ("appliesTo(*E) && \"Applying incorrect handler\""
, "/build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/llvm/include/llvm/Support/Error.h"
, 841, __PRETTY_FUNCTION__));
842    std::unique_ptr<ErrT> SubE(static_cast<ErrT *>(E.release()));
843    H(std::move(SubE));
844    return Error::success();
845  }
846};
847 
848// Specialization for member functions of the form 'RetT (const ErrT&)'.
849template <typename C, typename RetT, typename ErrT>
850class ErrorHandlerTraits<RetT (C::*)(ErrT &)>
851    : public ErrorHandlerTraits<RetT (&)(ErrT &)> {};
852 
853// Specialization for member functions of the form 'RetT (const ErrT&) const'.
854template <typename C, typename RetT, typename ErrT>
855class ErrorHandlerTraits<RetT (C::*)(ErrT &) const>
856    : public ErrorHandlerTraits<RetT (&)(ErrT &)> {};
857 
858// Specialization for member functions of the form 'RetT (const ErrT&)'.
859template <typename C, typename RetT, typename ErrT>
860class ErrorHandlerTraits<RetT (C::*)(const ErrT &)>
861    : public ErrorHandlerTraits<RetT (&)(ErrT &)> {};
862 
863// Specialization for member functions of the form 'RetT (const ErrT&) const'.
864template <typename C, typename RetT, typename ErrT>
865class ErrorHandlerTraits<RetT (C::*)(const ErrT &) const>
866    : public ErrorHandlerTraits<RetT (&)(ErrT &)> {};
867 
868/// Specialization for member functions of the form
869/// 'RetT (std::unique_ptr<ErrT>)'.
870template <typename C, typename RetT, typename ErrT>
871class ErrorHandlerTraits<RetT (C::*)(std::unique_ptr<ErrT>)>
872    : public ErrorHandlerTraits<RetT (&)(std::unique_ptr<ErrT>)> {};
873 
874/// Specialization for member functions of the form
875/// 'RetT (std::unique_ptr<ErrT>) const'.
876template <typename C, typename RetT, typename ErrT>
877class ErrorHandlerTraits<RetT (C::*)(std::unique_ptr<ErrT>) const>
878    : public ErrorHandlerTraits<RetT (&)(std::unique_ptr<ErrT>)> {};
879 
880inline Error handleErrorImpl(std::unique_ptr<ErrorInfoBase> Payload) {
881  return Error(std::move(Payload));
882}
883 
884template <typename HandlerT, typename... HandlerTs>
885Error handleErrorImpl(std::unique_ptr<ErrorInfoBase> Payload,
886                      HandlerT &&Handler, HandlerTs &&... Handlers) {
887  if (ErrorHandlerTraits<HandlerT>::appliesTo(*Payload))
888    return ErrorHandlerTraits<HandlerT>::apply(std::forward<HandlerT>(Handler),
889                                               std::move(Payload));
890  return handleErrorImpl(std::move(Payload),
891                         std::forward<HandlerTs>(Handlers)...);
892}
893 
894/// Pass the ErrorInfo(s) contained in E to their respective handlers. Any
895/// unhandled errors (or Errors returned by handlers) are re-concatenated and
896/// returned.
897/// Because this function returns an error, its result must also be checked
898/// or returned. If you intend to handle all errors use handleAllErrors
899/// (which returns void, and will abort() on unhandled errors) instead.
900template <typename... HandlerTs>
901Error handleErrors(Error E, HandlerTs &&... Hs) {
902  if (!E)
903    return Error::success();
904 
905  std::unique_ptr<ErrorInfoBase> Payload = E.takePayload();
906 
907  if (Payload->isA<ErrorList>()) {
908    ErrorList &List = static_cast<ErrorList &>(*Payload);
909    Error R;
910    for (auto &P : List.Payloads)
911      R = ErrorList::join(
912          std::move(R),
913          handleErrorImpl(std::move(P), std::forward<HandlerTs>(Hs)...));
914    return R;
915  }
916 
917  return handleErrorImpl(std::move(Payload), std::forward<HandlerTs>(Hs)...);
918}
919 
920/// Behaves the same as handleErrors, except that by contract all errors
921/// *must* be handled by the given handlers (i.e. there must be no remaining
922/// errors after running the handlers, or llvm_unreachable is called).
923template <typename... HandlerTs>
924void handleAllErrors(Error E, HandlerTs &&... Handlers) {
925  cantFail(handleErrors(std::move(E), std::forward<HandlerTs>(Handlers)...));
926}
927 
928/// Check that E is a non-error, then drop it.
929/// If E is an error, llvm_unreachable will be called.
930inline void handleAllErrors(Error E) {
931  cantFail(std::move(E));
932}
933 
934/// Handle any errors (if present) in an Expected<T>, then try a recovery path.
935///
936/// If the incoming value is a success value it is returned unmodified. If it
937/// is a failure value then it the contained error is passed to handleErrors.
938/// If handleErrors is able to handle the error then the RecoveryPath functor
939/// is called to supply the final result. If handleErrors is not able to
940/// handle all errors then the unhandled errors are returned.
941///
942/// This utility enables the follow pattern:
943///
944///   @code{.cpp}
945///   enum FooStrategy { Aggressive, Conservative };
946///   Expected<Foo> foo(FooStrategy S);
947///
948///   auto ResultOrErr =
949///     handleExpected(
950///       foo(Aggressive),
951///       []() { return foo(Conservative); },
952///       [](AggressiveStrategyError&) {
953///         // Implicitly conusme this - we'll recover by using a conservative
954///         // strategy.
955///       });
956///
957///   @endcode
958template <typename T, typename RecoveryFtor, typename... HandlerTs>
959Expected<T> handleExpected(Expected<T> ValOrErr, RecoveryFtor &&RecoveryPath,
960                           HandlerTs &&... Handlers) {
961  if (ValOrErr)
962    return ValOrErr;
963 
964  if (auto Err = handleErrors(ValOrErr.takeError(),
965                              std::forward<HandlerTs>(Handlers)...))
966    return std::move(Err);
967 
968  return RecoveryPath();
969}
970 
971/// Log all errors (if any) in E to OS. If there are any errors, ErrorBanner
972/// will be printed before the first one is logged. A newline will be printed
973/// after each error.
974///
975/// This function is compatible with the helpers from Support/WithColor.h. You
976/// can pass any of them as the OS. Please consider using them instead of
977/// including 'error: ' in the ErrorBanner.
978///
979/// This is useful in the base level of your program to allow clean termination
980/// (allowing clean deallocation of resources, etc.), while reporting error
981/// information to the user.
982void logAllUnhandledErrors(Error E, raw_ostream &OS, Twine ErrorBanner = {});
983 
984/// Write all error messages (if any) in E to a string. The newline character
985/// is used to separate error messages.
986inline std::string toString(Error E) {
987  SmallVector<std::string, 2> Errors;
988  handleAllErrors(std::move(E), [&Errors](const ErrorInfoBase &EI) {
989    Errors.push_back(EI.message());
990  });
991  return join(Errors.begin(), Errors.end(), "\n");
992}
993 
994/// Consume a Error without doing anything. This method should be used
995/// only where an error can be considered a reasonable and expected return
996/// value.
997///
998/// Uses of this method are potentially indicative of design problems: If it's
999/// legitimate to do nothing while processing an "error", the error-producer
1000/// might be more clearly refactored to return an Optional<T>.
1001inline void consumeError(Error Err) {
1002  handleAllErrors(std::move(Err), [](const ErrorInfoBase &) {});
1003}
1004 
1005/// Convert an Expected to an Optional without doing anything. This method
1006/// should be used only where an error can be considered a reasonable and
1007/// expected return value.
1008///
1009/// Uses of this method are potentially indicative of problems: perhaps the
1010/// error should be propagated further, or the error-producer should just
1011/// return an Optional in the first place.
1012template <typename T> Optional<T> expectedToOptional(Expected<T> &&E) {
1013  if (E)
1014    return std::move(*E);
1015  consumeError(E.takeError());
1016  return None;
1017}
1018 
1019/// Helper for converting an Error to a bool.
1020///
1021/// This method returns true if Err is in an error state, or false if it is
1022/// in a success state.  Puts Err in a checked state in both cases (unlike
1023/// Error::operator bool(), which only does this for success states).
1024inline bool errorToBool(Error Err) {
1025  bool IsError = static_cast<bool>(Err);
1026  if (IsError)
1027    consumeError(std::move(Err));
1028  return IsError;
1029}
1030 
1031/// Helper for Errors used as out-parameters.
1032///
1033/// This helper is for use with the Error-as-out-parameter idiom, where an error
1034/// is passed to a function or method by reference, rather than being returned.
1035/// In such cases it is helpful to set the checked bit on entry to the function
1036/// so that the error can be written to (unchecked Errors abort on assignment)
1037/// and clear the checked bit on exit so that clients cannot accidentally forget
1038/// to check the result. This helper performs these actions automatically using
1039/// RAII:
1040///
1041///   @code{.cpp}
1042///   Result foo(Error &Err) {
1043///     ErrorAsOutParameter ErrAsOutParam(&Err); // 'Checked' flag set
1044///     // <body of foo>
1045///     // <- 'Checked' flag auto-cleared when ErrAsOutParam is destructed.
1046///   }
1047///   @endcode
1048///
1049/// ErrorAsOutParameter takes an Error* rather than Error& so that it can be
1050/// used with optional Errors (Error pointers that are allowed to be null). If
1051/// ErrorAsOutParameter took an Error reference, an instance would have to be
1052/// created inside every condition that verified that Error was non-null. By
1053/// taking an Error pointer we can just create one instance at the top of the
1054/// function.
1055class ErrorAsOutParameter {
1056public:
1057  ErrorAsOutParameter(Error *Err) : Err(Err) {
1058    // Raise the checked bit if Err is success.
1059    if (Err)
1060      (void)!!*Err;
1061  }
1062 
1063  ~ErrorAsOutParameter() {
1064    // Clear the checked bit.
1065    if (Err && !*Err)
1066      *Err = Error::success();
1067  }
1068 
1069private:
1070  Error *Err;
1071};
1072 
1073/// Helper for Expected<T>s used as out-parameters.
1074///
1075/// See ErrorAsOutParameter.
1076template <typename T>
1077class ExpectedAsOutParameter {
1078public:
1079  ExpectedAsOutParameter(Expected<T> *ValOrErr)
1080    : ValOrErr(ValOrErr) {
1081    if (ValOrErr)
1082      (void)!!*ValOrErr;
1083  }
1084 
1085  ~ExpectedAsOutParameter() {
1086    if (ValOrErr)
1087      ValOrErr->setUnchecked();
1088  }
1089 
1090private:
1091  Expected<T> *ValOrErr;
1092};
1093 
1094/// This class wraps a std::error_code in a Error.
1095///
1096/// This is useful if you're writing an interface that returns a Error
1097/// (or Expected) and you want to call code that still returns
1098/// std::error_codes.
1099class ECError : public ErrorInfo<ECError> {
1100  friend Error errorCodeToError(std::error_code);
1101 
1102  virtual void anchor() override;
1103 
1104public:
1105  void setErrorCode(std::error_code EC) { this->EC = EC; }
1106  std::error_code convertToErrorCode() const override { return EC; }
1107  void log(raw_ostream &OS) const override { OS << EC.message(); }
1108 
1109  // Used by ErrorInfo::classID.
1110  static char ID;
1111 
1112protected:
1113  ECError() = default;
1114  ECError(std::error_code EC) : EC(EC) {}
1115 
1116  std::error_code EC;
1117};
1118 
1119/// The value returned by this function can be returned from convertToErrorCode
1120/// for Error values where no sensible translation to std::error_code exists.
1121/// It should only be used in this situation, and should never be used where a
1122/// sensible conversion to std::error_code is available, as attempts to convert
1123/// to/from this error will result in a fatal error. (i.e. it is a programmatic
1124///error to try to convert such a value).
1125std::error_code inconvertibleErrorCode();
1126 
1127/// Helper for converting an std::error_code to a Error.
1128Error errorCodeToError(std::error_code EC);
1129 
1130/// Helper for converting an ECError to a std::error_code.
1131///
1132/// This method requires that Err be Error() or an ECError, otherwise it
1133/// will trigger a call to abort().
1134std::error_code errorToErrorCode(Error Err);
1135 
1136/// Convert an ErrorOr<T> to an Expected<T>.
1137template <typename T> Expected<T> errorOrToExpected(ErrorOr<T> &&EO) {
1138  if (auto EC = EO.getError())
1139    return errorCodeToError(EC);
1140  return std::move(*EO);
1141}
1142 
1143/// Convert an Expected<T> to an ErrorOr<T>.
1144template <typename T> ErrorOr<T> expectedToErrorOr(Expected<T> &&E) {
1145  if (auto Err = E.takeError())
1146    return errorToErrorCode(std::move(Err));
1147  return std::move(*E);
1148}
1149 
1150/// This class wraps a string in an Error.
1151///
1152/// StringError is useful in cases where the client is not expected to be able
1153/// to consume the specific error message programmatically (for example, if the
1154/// error message is to be presented to the user).
1155///
1156/// StringError can also be used when additional information is to be printed
1157/// along with a error_code message. Depending on the constructor called, this
1158/// class can either display:
1159///    1. the error_code message (ECError behavior)
1160///    2. a string
1161///    3. the error_code message and a string
1162///
1163/// These behaviors are useful when subtyping is required; for example, when a
1164/// specific library needs an explicit error type. In the example below,
1165/// PDBError is derived from StringError:
1166///
1167///   @code{.cpp}
1168///   Expected<int> foo() {
1169///      return llvm::make_error<PDBError>(pdb_error_code::dia_failed_loading,
1170///                                        "Additional information");
1171///   }
1172///   @endcode
1173///
1174class StringError : public ErrorInfo<StringError> {
1175public:
1176  static char ID;
1177 
1178  // Prints EC + S and converts to EC
1179  StringError(std::error_code EC, const Twine &S = Twine());
1180 
1181  // Prints S and converts to EC
1182  StringError(const Twine &S, std::error_code EC);
1183 
1184  void log(raw_ostream &OS) const override;
1185  std::error_code convertToErrorCode() const override;
1186 
1187  const std::string &getMessage() const { return Msg; }
1188 
1189private:
1190  std::string Msg;
1191  std::error_code EC;
1192  const bool PrintMsgOnly = false;
1193};
1194 
1195/// Create formatted StringError object.
1196template <typename... Ts>
1197inline Error createStringError(std::error_code EC, char const *Fmt,
1198                               const Ts &... Vals) {
1199  std::string Buffer;
1200  raw_string_ostream Stream(Buffer);
1201  Stream << format(Fmt, Vals...);
1202  return make_error<StringError>(Stream.str(), EC);
1203}
1204 
1205Error createStringError(std::error_code EC, char const *Msg);
1206 
1207inline Error createStringError(std::error_code EC, const Twine &S) {
1208  return createStringError(EC, S.str().c_str());
1209}
1210 
1211template <typename... Ts>
1212inline Error createStringError(std::errc EC, char const *Fmt,
1213                               const Ts &... Vals) {
1214  return createStringError(std::make_error_code(EC), Fmt, Vals...);
1215}
1216 
1217/// This class wraps a filename and another Error.
1218///
1219/// In some cases, an error needs to live along a 'source' name, in order to
1220/// show more detailed information to the user.
1221class FileError final : public ErrorInfo<FileError> {
1222 
1223  friend Error createFileError(const Twine &, Error);
1224  friend Error createFileError(const Twine &, size_t, Error);
1225 
1226public:
1227  void log(raw_ostream &OS) const override {
1228    assert(Err && !FileName.empty() && "Trying to log after takeError().")((Err && !FileName.empty() && "Trying to log after takeError()."
) ? static_cast<void> (0) : __assert_fail ("Err && !FileName.empty() && \"Trying to log after takeError().\""
, "/build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/llvm/include/llvm/Support/Error.h"
, 1228, __PRETTY_FUNCTION__));
1229    OS << "'" << FileName << "': ";
1230    if (Line.hasValue())
1231      OS << "line " << Line.getValue() << ": ";
1232    Err->log(OS);
1233  }
1234 
1235  StringRef getFileName() { return FileName; }
1236 
1237  Error takeError() { return Error(std::move(Err)); }
1238 
1239  std::error_code convertToErrorCode() const override;
1240 
1241  // Used by ErrorInfo::classID.
1242  static char ID;
1243 
1244private:
1245  FileError(const Twine &F, Optional<size_t> LineNum,
1246            std::unique_ptr<ErrorInfoBase> E) {
1247    assert(E && "Cannot create FileError from Error success value.")((E && "Cannot create FileError from Error success value."
) ? static_cast<void> (0) : __assert_fail ("E && \"Cannot create FileError from Error success value.\""
, "/build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/llvm/include/llvm/Support/Error.h"
, 1247, __PRETTY_FUNCTION__));
1248    assert(!F.isTriviallyEmpty() &&((!F.isTriviallyEmpty() && "The file name provided to FileError must not be empty."
) ? static_cast<void> (0) : __assert_fail ("!F.isTriviallyEmpty() && \"The file name provided to FileError must not be empty.\""
, "/build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/llvm/include/llvm/Support/Error.h"
, 1249, __PRETTY_FUNCTION__))
1249           "The file name provided to FileError must not be empty.")((!F.isTriviallyEmpty() && "The file name provided to FileError must not be empty."
) ? static_cast<void> (0) : __assert_fail ("!F.isTriviallyEmpty() && \"The file name provided to FileError must not be empty.\""
, "/build/llvm-toolchain-snapshot-10~++20200112100611+7fa5290d5bd/llvm/include/llvm/Support/Error.h"
, 1249, __PRETTY_FUNCTION__));
1250    FileName = F.str();
1251    Err = std::move(E);
1252    Line = std::move(LineNum);
1253  }
1254 
1255  static Error build(const Twine &F, Optional<size_t> Line, Error E) {
1256    std::unique_ptr<ErrorInfoBase> Payload;
1257    handleAllErrors(std::move(E),
1258                    [&](std::unique_ptr<ErrorInfoBase> EIB) -> Error {
1259                      Payload = std::move(EIB);
1260                      return Error::success();
1261                    });
1262    return Error(
1263        std::unique_ptr<FileError>(new FileError(F, Line, std::move(Payload))));
1264  }
1265 
1266  std::string FileName;
1267  Optional<size_t> Line;
1268  std::unique_ptr<ErrorInfoBase> Err;
1269};
1270 
1271/// Concatenate a source file path and/or name with an Error. The resulting
1272/// Error is unchecked.
1273inline Error createFileError(const Twine &F, Error E) {
1274  return FileError::build(F, Optional<size_t>(), std::move(E));
1275}
1276 
1277/// Concatenate a source file path and/or name with line number and an Error.
1278/// The resulting Error is unchecked.
1279inline Error createFileError(const Twine &F, size_t Line, Error E) {
1280  return FileError::build(F, Optional<size_t>(Line), std::move(E));
1281}
1282 
1283/// Concatenate a source file path and/or name with a std::error_code 
1284/// to form an Error object.
1285inline Error createFileError(const Twine &F, std::error_code EC) {
1286  return createFileError(F, errorCodeToError(EC));
1287}
1288 
1289/// Concatenate a source file path and/or name with line number and
1290/// std::error_code to form an Error object.
1291inline Error createFileError(const Twine &F, size_t Line, std::error_code EC) {
1292  return createFileError(F, Line, errorCodeToError(EC));
1293}
1294 
1295Error createFileError(const Twine &F, ErrorSuccess) = delete;
1296 
1297/// Helper for check-and-exit error handling.
1298///
1299/// For tool use only. NOT FOR USE IN LIBRARY CODE.
1300///
1301class ExitOnError {
1302public:
1303  /// Create an error on exit helper.
1304  ExitOnError(std::string Banner = "", int DefaultErrorExitCode = 1)
1305      : Banner(std::move(Banner)),
1306        GetExitCode([=](const Error &) { return DefaultErrorExitCode; }) {}
1307 
1308  /// Set the banner string for any errors caught by operator().
1309  void setBanner(std::string Banner) { this->Banner = std::move(Banner); }
1310 
1311  /// Set the exit-code mapper function.
1312  void setExitCodeMapper(std::function<int(const Error &)> GetExitCode) {
1313    this->GetExitCode = std::move(GetExitCode);
1314  }
1315 
1316  /// Check Err. If it's in a failure state log the error(s) and exit.
1317  void operator()(Error Err) const { checkError(std::move(Err)); }
1318 
1319  /// Check E. If it's in a success state then return the contained value. If
1320  /// it's in a failure state log the error(s) and exit.
1321  template <typename T> T operator()(Expected<T> &&E) const {
1322    checkError(E.takeError());
1323    return std::move(*E);
1324  }
1325 
1326  /// Check E. If it's in a success state then return the contained reference. If
1327  /// it's in a failure state log the error(s) and exit.
1328  template <typename T> T& operator()(Expected<T&> &&E) const {
1329    checkError(E.takeError());
1330    return *E;
1331  }
1332 
1333private:
1334  void checkError(Error Err) const {
1335    if (Err) {
1336      int ExitCode = GetExitCode(Err);
1337      logAllUnhandledErrors(std::move(Err), errs(), Banner);
1338      exit(ExitCode);
1339    }
1340  }
1341 
1342  std::string Banner;
1343  std::function<int(const Error &)> GetExitCode;
1344};
1345 
1346/// Conversion from Error to LLVMErrorRef for C error bindings.
1347inline LLVMErrorRef wrap(Error Err) {
1348  return reinterpret_cast<LLVMErrorRef>(Err.takePayload().release());
1349}
1350 
1351/// Conversion from LLVMErrorRef to Error for C error bindings.
1352inline Error unwrap(LLVMErrorRef ErrRef) {
1353  return Error(std::unique_ptr<ErrorInfoBase>(
1354      reinterpret_cast<ErrorInfoBase *>(ErrRef)));
1355}
1356 
1357} // end namespace llvm
1358 
1359#endif // LLVM_SUPPORT_ERROR_H