llvm/lib/DebugInfo/PDB/DIA/DIAInjectedSource.cpp

   1 //===- DIAInjectedSource.cpp - DIA impl for IPDBInjectedSource --*- 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 #include "llvm/DebugInfo/PDB/DIA/DIAInjectedSource.h"
  10 #include "llvm/ADT/ArrayRef.h"
  11 #include "llvm/DebugInfo/PDB/ConcreteSymbolEnumerator.h"
  12 #include "llvm/DebugInfo/PDB/DIA/DIASession.h"
  13 #include "llvm/DebugInfo/PDB/DIA/DIAUtils.h"
  14
  15 using namespace llvm;
  16 using namespace llvm::pdb;
  17
  18 DIAInjectedSource::DIAInjectedSource(CComPtr<IDiaInjectedSource> DiaSourceFile)
  19     : SourceFile(DiaSourceFile) {}
  20
  21 uint32_t DIAInjectedSource::getCrc32() const {
  22   DWORD Crc;
  23   return (S_OK == SourceFile->get_crc(&Crc)) ? Crc : 0;
  24 }
  25
  26 uint64_t DIAInjectedSource::getCodeByteSize() const {
  27   ULONGLONG Size;
  28   return (S_OK == SourceFile->get_length(&Size)) ? Size : 0;
  29 }
  30
  31 std::string DIAInjectedSource::getFileName() const {
  32   return invokeBstrMethod(*SourceFile, &IDiaInjectedSource::get_filename);
  33 }
  34
  35 std::string DIAInjectedSource::getObjectFileName() const {
  36   return invokeBstrMethod(*SourceFile, &IDiaInjectedSource::get_objectFilename);
  37 }
  38
  39 std::string DIAInjectedSource::getVirtualFileName() const {
  40   return invokeBstrMethod(*SourceFile,
  41                           &IDiaInjectedSource::get_virtualFilename);
  42 }
  43
  44 uint32_t DIAInjectedSource::getCompression() const {
  45   DWORD Compression = 0;
  46   if (S_OK != SourceFile->get_sourceCompression(&Compression))
  47     return PDB_SourceCompression::None;
  48   return static_cast<uint32_t>(Compression);
  49 }
  50
  51 std::string DIAInjectedSource::getCode() const {
  52   DWORD DataSize;
  53   if (S_OK != SourceFile->get_source(0, &DataSize, nullptr))
  54     return "";
  55
  56   std::vector<uint8_t> Buffer(DataSize);
  57   if (S_OK != SourceFile->get_source(DataSize, &DataSize, Buffer.data()))
  58     return "";
  59   assert(Buffer.size() == DataSize);
  60   return std::string(reinterpret_cast<const char *>(Buffer.data()),
  61                      Buffer.size());
  62 }