blob: f3c730fc6b391088ddca9c8cb962e4cf85a84b98 [file] [log] [blame]
Zachary Turner7120a472016-06-06 20:37:05 +00001//===- YAMLOutputStyle.cpp ------------------------------------ *- C++ --*-===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9
10#include "YAMLOutputStyle.h"
11
12#include "PdbYaml.h"
13#include "llvm-pdbdump.h"
14
Zachary Turner8848a7a2016-07-06 18:05:57 +000015#include "llvm/DebugInfo/PDB/Raw/InfoStream.h"
Zachary Turner7120a472016-06-06 20:37:05 +000016#include "llvm/DebugInfo/PDB/Raw/PDBFile.h"
Zachary Turnerab58ae82016-06-30 17:43:00 +000017#include "llvm/DebugInfo/PDB/Raw/RawConstants.h"
Zachary Turner7120a472016-06-06 20:37:05 +000018
19using namespace llvm;
20using namespace llvm::pdb;
21
22YAMLOutputStyle::YAMLOutputStyle(PDBFile &File) : File(File), Out(outs()) {}
23
Zachary Turnera30bd1a2016-06-30 17:42:48 +000024Error YAMLOutputStyle::dump() {
Zachary Turner8848a7a2016-07-06 18:05:57 +000025 if (opts::pdb2yaml::StreamDirectory || opts::pdb2yaml::PdbStream)
26 opts::pdb2yaml::StreamMetadata = true;
27
Zachary Turnera30bd1a2016-06-30 17:42:48 +000028 if (auto EC = dumpFileHeaders())
29 return EC;
Zachary Turner7120a472016-06-06 20:37:05 +000030
Zachary Turnera30bd1a2016-06-30 17:42:48 +000031 if (auto EC = dumpStreamMetadata())
32 return EC;
33
34 if (auto EC = dumpStreamDirectory())
35 return EC;
36
Zachary Turner8848a7a2016-07-06 18:05:57 +000037 if (auto EC = dumpPDBStream())
38 return EC;
39
Zachary Turnera30bd1a2016-06-30 17:42:48 +000040 flush();
41 return Error::success();
42}
43
44Error YAMLOutputStyle::dumpFileHeaders() {
Zachary Turnerf6b93822016-07-11 21:45:09 +000045 if (opts::pdb2yaml::NoFileHeaders)
46 return Error::success();
47
Zachary Turner7120a472016-06-06 20:37:05 +000048 yaml::MsfHeaders Headers;
Zachary Turnerf6b93822016-07-11 21:45:09 +000049 Obj.Headers.emplace();
50 Obj.Headers->SuperBlock.NumBlocks = File.getBlockCount();
51 Obj.Headers->SuperBlock.BlockMapAddr = File.getBlockMapIndex();
52 Obj.Headers->BlockMapOffset = File.getBlockMapOffset();
53 Obj.Headers->SuperBlock.BlockSize = File.getBlockSize();
Zachary Turner7120a472016-06-06 20:37:05 +000054 auto Blocks = File.getDirectoryBlockArray();
Zachary Turnerf6b93822016-07-11 21:45:09 +000055 Obj.Headers->DirectoryBlocks.assign(Blocks.begin(), Blocks.end());
56 Obj.Headers->NumDirectoryBlocks = File.getNumDirectoryBlocks();
57 Obj.Headers->SuperBlock.NumDirectoryBytes = File.getNumDirectoryBytes();
58 Obj.Headers->NumStreams =
Zachary Turnerab58ae82016-06-30 17:43:00 +000059 opts::pdb2yaml::StreamMetadata ? File.getNumStreams() : 0;
Zachary Turnerf6b93822016-07-11 21:45:09 +000060 Obj.Headers->SuperBlock.Unknown0 = File.getUnknown0();
61 Obj.Headers->SuperBlock.Unknown1 = File.getUnknown1();
62 Obj.Headers->FileSize = File.getFileSize();
Zachary Turner7120a472016-06-06 20:37:05 +000063
64 return Error::success();
65}
66
Zachary Turnera30bd1a2016-06-30 17:42:48 +000067Error YAMLOutputStyle::dumpStreamMetadata() {
68 if (!opts::pdb2yaml::StreamMetadata)
Zachary Turner7120a472016-06-06 20:37:05 +000069 return Error::success();
70
Zachary Turner1dc9fd32016-06-14 20:48:36 +000071 Obj.StreamSizes = File.getStreamSizes();
Zachary Turner7120a472016-06-06 20:37:05 +000072 return Error::success();
73}
74
Zachary Turnera30bd1a2016-06-30 17:42:48 +000075Error YAMLOutputStyle::dumpStreamDirectory() {
76 if (!opts::pdb2yaml::StreamDirectory)
Zachary Turner7120a472016-06-06 20:37:05 +000077 return Error::success();
78
Zachary Turner1dc9fd32016-06-14 20:48:36 +000079 auto StreamMap = File.getStreamMap();
80 Obj.StreamMap.emplace();
81 for (auto &Stream : StreamMap) {
82 pdb::yaml::StreamBlockList BlockList;
83 BlockList.Blocks = Stream;
84 Obj.StreamMap->push_back(BlockList);
Zachary Turner25e8b052016-06-06 20:37:17 +000085 }
Zachary Turner25e8b052016-06-06 20:37:17 +000086
Zachary Turner7120a472016-06-06 20:37:05 +000087 return Error::success();
88}
89
Zachary Turner8848a7a2016-07-06 18:05:57 +000090Error YAMLOutputStyle::dumpPDBStream() {
91 if (!opts::pdb2yaml::PdbStream)
92 return Error::success();
93
94 auto IS = File.getPDBInfoStream();
95 if (!IS)
96 return IS.takeError();
97
98 auto &InfoS = IS.get();
99 Obj.PdbStream.emplace();
100 Obj.PdbStream->Age = InfoS.getAge();
101 Obj.PdbStream->Guid = InfoS.getGuid();
102 Obj.PdbStream->Signature = InfoS.getSignature();
103 Obj.PdbStream->Version = InfoS.getVersion();
104
105 return Error::success();
106}
107
Zachary Turner7120a472016-06-06 20:37:05 +0000108void YAMLOutputStyle::flush() {
109 Out << Obj;
110 outs().flush();
111}