Add llvm-size.
git-svn-id: https://llvm.org/svn/llvm-project/llvm/trunk@140722 91177308-0d34-0410-b5e6-96231b3b80d8
diff --git a/tools/CMakeLists.txt b/tools/CMakeLists.txt
index c1816c6..e66648b 100644
--- a/tools/CMakeLists.txt
+++ b/tools/CMakeLists.txt
@@ -28,6 +28,7 @@
add_subdirectory(llvm-ranlib)
add_subdirectory(llvm-ar)
add_subdirectory(llvm-nm)
+add_subdirectory(llvm-size)
add_subdirectory(llvm-ld)
add_subdirectory(llvm-prof)
diff --git a/tools/Makefile b/tools/Makefile
index 62779be..68ce314 100644
--- a/tools/Makefile
+++ b/tools/Makefile
@@ -26,7 +26,8 @@
lli llvm-extract llvm-mc \
bugpoint llvm-bcanalyzer llvm-stub \
llvm-diff macho-dump llvm-objdump \
- llvm-rtdyld llvm-dwarfdump llvm-cov
+ llvm-rtdyld llvm-dwarfdump llvm-cov \
+ llvm-size
# Let users override the set of tools to build from the command line.
ifdef ONLY_TOOLS
diff --git a/tools/llvm-size/CMakeLists.txt b/tools/llvm-size/CMakeLists.txt
new file mode 100644
index 0000000..933cc75
--- /dev/null
+++ b/tools/llvm-size/CMakeLists.txt
@@ -0,0 +1,5 @@
+set(LLVM_LINK_COMPONENTS object)
+
+add_llvm_tool(llvm-size
+ llvm-size.cpp
+ )
diff --git a/tools/llvm-size/Makefile b/tools/llvm-size/Makefile
new file mode 100644
index 0000000..5d0e27e
--- /dev/null
+++ b/tools/llvm-size/Makefile
@@ -0,0 +1,17 @@
+##===- tools/llvm-size/Makefile ----------------------------*- Makefile -*-===##
+#
+# The LLVM Compiler Infrastructure
+#
+# This file is distributed under the University of Illinois Open Source
+# License. See LICENSE.TXT for details.
+#
+##===----------------------------------------------------------------------===##
+LEVEL = ../..
+
+TOOLNAME = llvm-size
+LINK_COMPONENTS = object
+
+# This tool has no plugins, optimize startup time.
+TOOL_NO_EXPORTS = 1
+
+include $(LEVEL)/Makefile.common
diff --git a/tools/llvm-size/llvm-size.cpp b/tools/llvm-size/llvm-size.cpp
new file mode 100644
index 0000000..2593758
--- /dev/null
+++ b/tools/llvm-size/llvm-size.cpp
@@ -0,0 +1,298 @@
+//===-- llvm-size.cpp - Print the size of each object section -------------===//
+//
+// The LLVM Compiler Infrastructure
+//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+//
+// This program is a utility that works like traditional Unix "size",
+// that is, it prints out the size of each section, and the total size of all
+// sections.
+//
+//===----------------------------------------------------------------------===//
+
+#include "llvm/ADT/APInt.h"
+#include "llvm/Object/Archive.h"
+#include "llvm/Object/ObjectFile.h"
+#include "llvm/Support/Casting.h"
+#include "llvm/Support/CommandLine.h"
+#include "llvm/Support/FileSystem.h"
+#include "llvm/Support/Format.h"
+#include "llvm/Support/ManagedStatic.h"
+#include "llvm/Support/MemoryBuffer.h"
+#include "llvm/Support/PrettyStackTrace.h"
+#include "llvm/Support/raw_ostream.h"
+#include "llvm/Support/Signals.h"
+#include "llvm/Support/system_error.h"
+#include <algorithm>
+#include <string>
+using namespace llvm;
+using namespace object;
+
+namespace {
+ enum OutputFormatTy {berkeley, sysv};
+ cl::opt<OutputFormatTy>
+ OutputFormat("format",
+ cl::desc("Specify output format"),
+ cl::values(clEnumVal(sysv, "System V format"),
+ clEnumVal(berkeley, "Berkeley format"),
+ clEnumValEnd),
+ cl::init(berkeley));
+
+ cl::opt<OutputFormatTy>
+ OutputFormatShort(cl::desc("Specify output format"),
+ cl::values(clEnumValN(sysv, "A", "System V format"),
+ clEnumValN(berkeley, "B", "Berkeley format"),
+ clEnumValEnd),
+ cl::init(berkeley));
+
+ enum RadixTy {octal = 8, decimal = 10, hexadecimal = 16};
+ cl::opt<int>
+ Radix("-radix",
+ cl::desc("Print size in radix. Only 8, 10, and 16 are valid"),
+ cl::init(decimal));
+
+ cl::opt<RadixTy>
+ RadixShort(cl::desc("Print size in radix:"),
+ cl::values(clEnumValN(octal, "o", "Print size in octal"),
+ clEnumValN(decimal, "d", "Print size in decimal"),
+ clEnumValN(hexadecimal, "x", "Print size in hexadecimal"),
+ clEnumValEnd),
+ cl::init(decimal));
+
+ cl::list<std::string>
+ InputFilenames(cl::Positional, cl::desc("<input files>"),
+ cl::ZeroOrMore);
+
+ std::string ToolName;
+}
+
+static bool error(error_code ec) {
+ if (!ec) return false;
+
+ outs() << ToolName << ": error reading file: " << ec.message() << ".\n";
+ outs().flush();
+ return true;
+}
+
+static int getNumLengthAsString(uint64_t num) {
+ APInt conv(64, num);
+ SmallString<32> result;
+ conv.toString(result, unsigned int(Radix), false, true);
+ return result.size();
+}
+
+static void PrintObjectSectionSizes(ObjectFile *o) {
+ uint64_t total = 0;
+ std::string fmtbuf;
+ raw_string_ostream fmt(fmtbuf);
+
+ const char *radix_fmt = 0;
+ switch (Radix) {
+ case octal:
+ radix_fmt = "llo";
+ break;
+ case decimal:
+ radix_fmt = "llu";
+ break;
+ case hexadecimal:
+ radix_fmt = "llx";
+ break;
+ }
+ if (OutputFormat == sysv) {
+ // Run two passes over all sections. The first gets the lengths needed for
+ // formatting the output. The second actually does the output.
+ std::size_t max_name_len = strlen("section");
+ int max_size_len = strlen("size");
+ int max_addr_len = strlen("addr");
+ error_code ec;
+ for (ObjectFile::section_iterator i = o->begin_sections(),
+ e = o->end_sections(); i != e;
+ i.increment(ec)) {
+ if (error(ec))
+ return;
+ uint64_t size = 0;
+ if (error(i->getSize(size)))
+ return;
+ total += size;
+
+ StringRef name;
+ uint64_t addr = 0;
+ if (error(i->getName(name))) return;
+ if (error(i->getAddress(addr))) return;
+ max_name_len = std::max(max_name_len, name.size());
+ max_size_len = std::max(max_size_len, getNumLengthAsString(size));
+ max_addr_len = std::max(max_addr_len, getNumLengthAsString(addr));
+ }
+
+ max_name_len += 2;
+ max_size_len += 2;
+ max_addr_len += 2;
+
+ fmt << "%-" << max_name_len << "s "
+ << "%" << max_size_len << "s "
+ << "%" << max_addr_len << "s\n";
+
+ // Print header
+ outs() << format(fmt.str().c_str(),
+ static_cast<const char*>("section"),
+ static_cast<const char*>("size"),
+ static_cast<const char*>("addr"));
+ fmtbuf.clear();
+
+ // Setup per section format.
+ fmt << "%-" << max_name_len << "s "
+ << "%#" << max_size_len << radix_fmt << " "
+ << "%#" << max_addr_len << radix_fmt << "\n";
+
+ // Print each section.
+ for (ObjectFile::section_iterator i = o->begin_sections(),
+ e = o->end_sections(); i != e;
+ i.increment(ec)) {
+ if (error(ec))
+ return;
+
+ StringRef name;
+ uint64_t size = 0;
+ uint64_t addr = 0;
+ if (error(i->getName(name))) return;
+ if (error(i->getSize(size))) return;
+ if (error(i->getAddress(addr))) return;
+ std::string namestr = name;
+
+ outs() << format(fmt.str().c_str(),
+ namestr.c_str(),
+ size,
+ addr);
+ }
+
+ // Print total.
+ fmtbuf.clear();
+ fmt << "%-" << max_name_len << "s "
+ << "%#" << max_size_len << radix_fmt << "\n";
+ outs() << format(fmt.str().c_str(),
+ static_cast<const char*>("Total"),
+ total);
+ } else {
+ uint64_t total_text = 0;
+ uint64_t total_data = 0;
+ uint64_t total_bss = 0;
+
+ error_code ec;
+ // Collect section data.
+ for (ObjectFile::section_iterator i = o->begin_sections(),
+ e = o->end_sections(); i != e;
+ i.increment(ec)) {
+ if (error(ec))
+ return;
+
+ uint64_t size = 0;
+ bool isText = false;
+ bool isData = false;
+ bool isBSS = false;
+ if (error(i->getSize(size))) return;
+ if (error(i->isText(isText))) return;
+ if (error(i->isData(isData))) return;
+ if (error(i->isBSS(isBSS))) return;
+ if (isText)
+ total_text += size;
+ else if (isData)
+ total_data += size;
+ else if (isBSS)
+ total_bss += size;
+ }
+
+ total = total_text + total_data + total_bss;
+
+ // Print result.
+ fmt << "%#7" << radix_fmt << " "
+ << "%#7" << radix_fmt << " "
+ << "%#7" << radix_fmt << " ";
+ outs() << format(fmt.str().c_str(),
+ total_text,
+ total_data,
+ total_bss);
+ fmtbuf.clear();
+ fmt << "%7" << (Radix == octal ? "llo" : "llu") << " "
+ << "%7llx ";
+ outs() << format(fmt.str().c_str(),
+ total,
+ total);
+ }
+}
+
+static void PrintFileSectionSizes(StringRef file) {
+ // If file is not stdin, check that it exists.
+ if (file != "-") {
+ bool exists;
+ if (sys::fs::exists(file, exists) || !exists) {
+ errs() << ToolName << ": '" << file << "': " << "No such file\n";
+ return;
+ }
+ }
+
+ OwningPtr<Binary> binary;
+ if (error_code ec = createBinary(file, binary)) {
+ errs() << ToolName << ": " << file << ": " << ec.message() << ".\n";
+ return;
+ }
+
+ if (Archive *a = dyn_cast<Archive>(binary.get())) {
+ for (object::Archive::child_iterator i = a->begin_children(),
+ e = a->end_children(); i != e; ++i) {
+ OwningPtr<Binary> child;
+ if (error_code ec = i->getAsBinary(child)) {
+ errs() << ToolName << ": " << file << ": " << ec.message() << ".\n";
+ continue;
+ }
+ if (ObjectFile *o = dyn_cast<ObjectFile>(child.get())) {
+ if (OutputFormat == sysv)
+ outs() << o->getFileName() << " (ex " << a->getFileName()
+ << "):\n";
+ PrintObjectSectionSizes(o);
+ if (OutputFormat == berkeley)
+ outs() << o->getFileName() << " (ex " << a->getFileName() << ")\n";
+ }
+ }
+ } else if (ObjectFile *o = dyn_cast<ObjectFile>(binary.get())) {
+ if (OutputFormat == sysv)
+ outs() << o->getFileName() << " :\n";
+ PrintObjectSectionSizes(o);
+ if (OutputFormat == berkeley)
+ outs() << o->getFileName() << "\n";
+ } else {
+ errs() << ToolName << ": " << file << ": " << "Unrecognized file type.\n";
+ }
+ if (OutputFormat == sysv)
+ outs() << "\n";
+}
+
+int main(int argc, char **argv) {
+ // Print a stack trace if we signal out.
+ sys::PrintStackTraceOnErrorSignal();
+ PrettyStackTraceProgram X(argc, argv);
+
+ llvm_shutdown_obj Y; // Call llvm_shutdown() on exit.
+ cl::ParseCommandLineOptions(argc, argv, "llvm object size dumper\n");
+
+ ToolName = argv[0];
+ if (OutputFormatShort.getNumOccurrences())
+ OutputFormat = OutputFormatShort;
+ if (RadixShort.getNumOccurrences())
+ Radix = int(RadixShort);
+
+ if (InputFilenames.size() == 0)
+ InputFilenames.push_back("a.out");
+
+ if (OutputFormat == berkeley)
+ outs() << " text data bss "
+ << (Radix == int(octal) ? "oct" : "dec")
+ << " hex filename\n";
+
+ std::for_each(InputFilenames.begin(), InputFilenames.end(),
+ PrintFileSectionSizes);
+
+ return 0;
+}
