SectionSizes.cpp
4.31 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
//===-- SectionSizes.cpp - Debug section sizes ----------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#include "llvm-dwarfdump.h"
#define DEBUG_TYPE "dwarfdump"
using namespace llvm;
using namespace llvm::dwarfdump;
using namespace llvm::object;
static size_t getNameColumnWidth(const SectionSizes &Sizes,
const StringRef SectionNameTitle) {
// The minimum column width should be the size of "SECTION".
size_t Width = SectionNameTitle.size();
for (const auto &DebugSec : Sizes.DebugSectionSizes) {
StringRef SectionName = DebugSec.getKey();
Width = std::max(Width, SectionName.size());
}
return Width;
}
static size_t getSizeColumnWidth(const SectionSizes &Sizes,
const StringRef SectionSizeTitle) {
// The minimum column width should be the size of the column title.
size_t Width = SectionSizeTitle.size();
for (const auto &DebugSec : Sizes.DebugSectionSizes) {
size_t NumWidth = std::to_string(DebugSec.getValue()).size();
Width = std::max(Width, NumWidth);
}
return Width;
}
static void prettyPrintSectionSizes(const ObjectFile &Obj,
const SectionSizes &Sizes,
raw_ostream &OS) {
const StringRef SectionNameTitle = "SECTION";
const StringRef SectionSizeTitle = "SIZE (b)";
size_t NameColWidth = getNameColumnWidth(Sizes, SectionNameTitle);
size_t SizeColWidth = getSizeColumnWidth(Sizes, SectionSizeTitle);
OS << "----------------------------------------------------" << '\n';
OS << SectionNameTitle;
size_t SectionNameTitleWidth = SectionNameTitle.size();
for (unsigned i = 0; i < (NameColWidth - SectionNameTitleWidth) + 2; i++)
OS << " ";
OS << SectionSizeTitle << '\n';
for (unsigned i = 0; i < NameColWidth; i++)
OS << "-";
OS << " ";
for (unsigned i = 0; i < SizeColWidth; i++)
OS << "-";
OS << '\n';
for (const auto &DebugSec : Sizes.DebugSectionSizes) {
OS << left_justify(DebugSec.getKey(), NameColWidth) << " ";
auto NumBytes = std::to_string(DebugSec.getValue());
OS << right_justify(NumBytes, SizeColWidth) << " ("
<< format("%0.2f", DebugSec.getValue() /
static_cast<double>(Sizes.TotalObjectSize) * 100)
<< "%)\n";
}
OS << '\n';
OS << " Total Size: " << Sizes.TotalDebugSectionsSize << " ("
<< format("%0.2f", Sizes.TotalDebugSectionsSize /
static_cast<double>(Sizes.TotalObjectSize) * 100)
<< "%)\n";
OS << " Total File Size: " << Sizes.TotalObjectSize << '\n';
OS << "----------------------------------------------------" << '\n';
}
void dwarfdump::calculateSectionSizes(const ObjectFile &Obj,
SectionSizes &Sizes,
const Twine &Filename) {
// Get total size.
Sizes.TotalObjectSize = Obj.getData().size();
for (const SectionRef &Section : Obj.sections()) {
StringRef SectionName;
if (Expected<StringRef> NameOrErr = Section.getName())
SectionName = *NameOrErr;
else
WithColor::defaultWarningHandler(
createFileError(Filename, NameOrErr.takeError()));
LLVM_DEBUG(dbgs() << SectionName.str() << ": " << Section.getSize()
<< '\n');
if (!Section.isDebugSection(SectionName))
continue;
Sizes.TotalDebugSectionsSize += Section.getSize();
Sizes.DebugSectionSizes[SectionName] += Section.getSize();
}
}
bool dwarfdump::collectObjectSectionSizes(ObjectFile &Obj,
DWARFContext & /*DICtx*/,
const Twine &Filename,
raw_ostream &OS) {
SectionSizes Sizes;
// Get the section sizes.
calculateSectionSizes(Obj, Sizes, Filename);
OS << "----------------------------------------------------\n";
OS << "file: " << Filename.str() << '\n';
prettyPrintSectionSizes(Obj, Sizes, OS);
// TODO: If the input file is an archive, print the cumulative summary of all
// files from the archive.
return true;
}