diff options
Diffstat (limited to 'gnu/llvm/tools/llvm-size/llvm-size.cpp')
| -rw-r--r-- | gnu/llvm/tools/llvm-size/llvm-size.cpp | 894 |
1 files changed, 0 insertions, 894 deletions
diff --git a/gnu/llvm/tools/llvm-size/llvm-size.cpp b/gnu/llvm/tools/llvm-size/llvm-size.cpp deleted file mode 100644 index 5d638443451..00000000000 --- a/gnu/llvm/tools/llvm-size/llvm-size.cpp +++ /dev/null @@ -1,894 +0,0 @@ -//===-- llvm-size.cpp - Print the size of each object section ---*- C++ -*-===// -// -// 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/ELFObjectFile.h" -#include "llvm/Object/MachO.h" -#include "llvm/Object/MachOUniversal.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/InitLLVM.h" -#include "llvm/Support/MemoryBuffer.h" -#include "llvm/Support/raw_ostream.h" -#include <algorithm> -#include <string> -#include <system_error> - -using namespace llvm; -using namespace object; - -enum OutputFormatTy { berkeley, sysv, darwin }; -static cl::opt<OutputFormatTy> -OutputFormat("format", cl::desc("Specify output format"), - cl::values(clEnumVal(sysv, "System V format"), - clEnumVal(berkeley, "Berkeley format"), - clEnumVal(darwin, "Darwin -m format")), - cl::init(berkeley)); - -static cl::opt<OutputFormatTy> OutputFormatShort( - cl::desc("Specify output format"), - cl::values(clEnumValN(sysv, "A", "System V format"), - clEnumValN(berkeley, "B", "Berkeley format"), - clEnumValN(darwin, "m", "Darwin -m format")), - cl::init(berkeley)); - -static bool BerkeleyHeaderPrinted = false; -static bool MoreThanOneFile = false; -static uint64_t TotalObjectText = 0; -static uint64_t TotalObjectData = 0; -static uint64_t TotalObjectBss = 0; -static uint64_t TotalObjectTotal = 0; - -cl::opt<bool> -DarwinLongFormat("l", cl::desc("When format is darwin, use long format " - "to include addresses and offsets.")); - -cl::opt<bool> - ELFCommons("common", - cl::desc("Print common symbols in the ELF file. When using " - "Berkely format, this is added to bss."), - cl::init(false)); - -static cl::list<std::string> -ArchFlags("arch", cl::desc("architecture(s) from a Mach-O file to dump"), - cl::ZeroOrMore); -static bool ArchAll = false; - -enum RadixTy { octal = 8, decimal = 10, hexadecimal = 16 }; -static cl::opt<RadixTy> Radix( - "radix", cl::desc("Print size in radix"), cl::init(decimal), - cl::values(clEnumValN(octal, "8", "Print size in octal"), - clEnumValN(decimal, "10", "Print size in decimal"), - clEnumValN(hexadecimal, "16", "Print size in hexadecimal"))); - -static 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")), - cl::init(decimal)); - -static cl::opt<bool> - TotalSizes("totals", - cl::desc("Print totals of all objects - Berkeley format only"), - cl::init(false)); - -static cl::alias TotalSizesShort("t", cl::desc("Short for --totals"), - cl::aliasopt(TotalSizes)); - -static cl::list<std::string> -InputFilenames(cl::Positional, cl::desc("<input files>"), cl::ZeroOrMore); - -static bool HadError = false; - -static std::string ToolName; - -/// If ec is not success, print the error and return true. -static bool error(std::error_code ec) { - if (!ec) - return false; - - HadError = true; - errs() << ToolName << ": error reading file: " << ec.message() << ".\n"; - errs().flush(); - return true; -} - -static bool error(Twine Message) { - HadError = true; - errs() << ToolName << ": " << Message << ".\n"; - errs().flush(); - return true; -} - -// This version of error() prints the archive name and member name, for example: -// "libx.a(foo.o)" after the ToolName before the error message. It sets -// HadError but returns allowing the code to move on to other archive members. -static void error(llvm::Error E, StringRef FileName, const Archive::Child &C, - StringRef ArchitectureName = StringRef()) { - HadError = true; - errs() << ToolName << ": " << FileName; - - Expected<StringRef> NameOrErr = C.getName(); - // TODO: if we have a error getting the name then it would be nice to print - // the index of which archive member this is and or its offset in the - // archive instead of "???" as the name. - if (!NameOrErr) { - consumeError(NameOrErr.takeError()); - errs() << "(" << "???" << ")"; - } else - errs() << "(" << NameOrErr.get() << ")"; - - if (!ArchitectureName.empty()) - errs() << " (for architecture " << ArchitectureName << ") "; - - std::string Buf; - raw_string_ostream OS(Buf); - logAllUnhandledErrors(std::move(E), OS); - OS.flush(); - errs() << " " << Buf << "\n"; -} - -// This version of error() prints the file name and which architecture slice it // is from, for example: "foo.o (for architecture i386)" after the ToolName -// before the error message. It sets HadError but returns allowing the code to -// move on to other architecture slices. -static void error(llvm::Error E, StringRef FileName, - StringRef ArchitectureName = StringRef()) { - HadError = true; - errs() << ToolName << ": " << FileName; - - if (!ArchitectureName.empty()) - errs() << " (for architecture " << ArchitectureName << ") "; - - std::string Buf; - raw_string_ostream OS(Buf); - logAllUnhandledErrors(std::move(E), OS); - OS.flush(); - errs() << " " << Buf << "\n"; -} - -/// Get the length of the string that represents @p num in Radix including the -/// leading 0x or 0 for hexadecimal and octal respectively. -static size_t getNumLengthAsString(uint64_t num) { - APInt conv(64, num); - SmallString<32> result; - conv.toString(result, Radix, false, true); - return result.size(); -} - -/// Return the printing format for the Radix. -static const char *getRadixFmt() { - switch (Radix) { - case octal: - return PRIo64; - case decimal: - return PRIu64; - case hexadecimal: - return PRIx64; - } - return nullptr; -} - -/// Remove unneeded ELF sections from calculation -static bool considerForSize(ObjectFile *Obj, SectionRef Section) { - if (!Obj->isELF()) - return true; - switch (static_cast<ELFSectionRef>(Section).getType()) { - case ELF::SHT_NULL: - case ELF::SHT_SYMTAB: - case ELF::SHT_STRTAB: - case ELF::SHT_REL: - case ELF::SHT_RELA: - return false; - } - return true; -} - -/// Total size of all ELF common symbols -static uint64_t getCommonSize(ObjectFile *Obj) { - uint64_t TotalCommons = 0; - for (auto &Sym : Obj->symbols()) - if (Obj->getSymbolFlags(Sym.getRawDataRefImpl()) & SymbolRef::SF_Common) - TotalCommons += Obj->getCommonSymbolSize(Sym.getRawDataRefImpl()); - return TotalCommons; -} - -/// Print the size of each Mach-O segment and section in @p MachO. -/// -/// This is when used when @c OutputFormat is darwin and produces the same -/// output as darwin's size(1) -m output. -static void printDarwinSectionSizes(MachOObjectFile *MachO) { - std::string fmtbuf; - raw_string_ostream fmt(fmtbuf); - const char *radix_fmt = getRadixFmt(); - if (Radix == hexadecimal) - fmt << "0x"; - fmt << "%" << radix_fmt; - - uint32_t Filetype = MachO->getHeader().filetype; - - uint64_t total = 0; - for (const auto &Load : MachO->load_commands()) { - if (Load.C.cmd == MachO::LC_SEGMENT_64) { - MachO::segment_command_64 Seg = MachO->getSegment64LoadCommand(Load); - outs() << "Segment " << Seg.segname << ": " - << format(fmt.str().c_str(), Seg.vmsize); - if (DarwinLongFormat) - outs() << " (vmaddr 0x" << format("%" PRIx64, Seg.vmaddr) << " fileoff " - << Seg.fileoff << ")"; - outs() << "\n"; - total += Seg.vmsize; - uint64_t sec_total = 0; - for (unsigned J = 0; J < Seg.nsects; ++J) { - MachO::section_64 Sec = MachO->getSection64(Load, J); - if (Filetype == MachO::MH_OBJECT) - outs() << "\tSection (" << format("%.16s", &Sec.segname) << ", " - << format("%.16s", &Sec.sectname) << "): "; - else - outs() << "\tSection " << format("%.16s", &Sec.sectname) << ": "; - outs() << format(fmt.str().c_str(), Sec.size); - if (DarwinLongFormat) - outs() << " (addr 0x" << format("%" PRIx64, Sec.addr) << " offset " - << Sec.offset << ")"; - outs() << "\n"; - sec_total += Sec.size; - } - if (Seg.nsects != 0) - outs() << "\ttotal " << format(fmt.str().c_str(), sec_total) << "\n"; - } else if (Load.C.cmd == MachO::LC_SEGMENT) { - MachO::segment_command Seg = MachO->getSegmentLoadCommand(Load); - uint64_t Seg_vmsize = Seg.vmsize; - outs() << "Segment " << Seg.segname << ": " - << format(fmt.str().c_str(), Seg_vmsize); - if (DarwinLongFormat) - outs() << " (vmaddr 0x" << format("%" PRIx32, Seg.vmaddr) << " fileoff " - << Seg.fileoff << ")"; - outs() << "\n"; - total += Seg.vmsize; - uint64_t sec_total = 0; - for (unsigned J = 0; J < Seg.nsects; ++J) { - MachO::section Sec = MachO->getSection(Load, J); - if (Filetype == MachO::MH_OBJECT) - outs() << "\tSection (" << format("%.16s", &Sec.segname) << ", " - << format("%.16s", &Sec.sectname) << "): "; - else - outs() << "\tSection " << format("%.16s", &Sec.sectname) << ": "; - uint64_t Sec_size = Sec.size; - outs() << format(fmt.str().c_str(), Sec_size); - if (DarwinLongFormat) - outs() << " (addr 0x" << format("%" PRIx32, Sec.addr) << " offset " - << Sec.offset << ")"; - outs() << "\n"; - sec_total += Sec.size; - } - if (Seg.nsects != 0) - outs() << "\ttotal " << format(fmt.str().c_str(), sec_total) << "\n"; - } - } - outs() << "total " << format(fmt.str().c_str(), total) << "\n"; -} - -/// Print the summary sizes of the standard Mach-O segments in @p MachO. -/// -/// This is when used when @c OutputFormat is berkeley with a Mach-O file and -/// produces the same output as darwin's size(1) default output. -static void printDarwinSegmentSizes(MachOObjectFile *MachO) { - uint64_t total_text = 0; - uint64_t total_data = 0; - uint64_t total_objc = 0; - uint64_t total_others = 0; - for (const auto &Load : MachO->load_commands()) { - if (Load.C.cmd == MachO::LC_SEGMENT_64) { - MachO::segment_command_64 Seg = MachO->getSegment64LoadCommand(Load); - if (MachO->getHeader().filetype == MachO::MH_OBJECT) { - for (unsigned J = 0; J < Seg.nsects; ++J) { - MachO::section_64 Sec = MachO->getSection64(Load, J); - StringRef SegmentName = StringRef(Sec.segname); - if (SegmentName == "__TEXT") - total_text += Sec.size; - else if (SegmentName == "__DATA") - total_data += Sec.size; - else if (SegmentName == "__OBJC") - total_objc += Sec.size; - else - total_others += Sec.size; - } - } else { - StringRef SegmentName = StringRef(Seg.segname); - if (SegmentName == "__TEXT") - total_text += Seg.vmsize; - else if (SegmentName == "__DATA") - total_data += Seg.vmsize; - else if (SegmentName == "__OBJC") - total_objc += Seg.vmsize; - else - total_others += Seg.vmsize; - } - } else if (Load.C.cmd == MachO::LC_SEGMENT) { - MachO::segment_command Seg = MachO->getSegmentLoadCommand(Load); - if (MachO->getHeader().filetype == MachO::MH_OBJECT) { - for (unsigned J = 0; J < Seg.nsects; ++J) { - MachO::section Sec = MachO->getSection(Load, J); - StringRef SegmentName = StringRef(Sec.segname); - if (SegmentName == "__TEXT") - total_text += Sec.size; - else if (SegmentName == "__DATA") - total_data += Sec.size; - else if (SegmentName == "__OBJC") - total_objc += Sec.size; - else - total_others += Sec.size; - } - } else { - StringRef SegmentName = StringRef(Seg.segname); - if (SegmentName == "__TEXT") - total_text += Seg.vmsize; - else if (SegmentName == "__DATA") - total_data += Seg.vmsize; - else if (SegmentName == "__OBJC") - total_objc += Seg.vmsize; - else - total_others += Seg.vmsize; - } - } - } - uint64_t total = total_text + total_data + total_objc + total_others; - - if (!BerkeleyHeaderPrinted) { - outs() << "__TEXT\t__DATA\t__OBJC\tothers\tdec\thex\n"; - BerkeleyHeaderPrinted = true; - } - outs() << total_text << "\t" << total_data << "\t" << total_objc << "\t" - << total_others << "\t" << total << "\t" << format("%" PRIx64, total) - << "\t"; -} - -/// Print the size of each section in @p Obj. -/// -/// The format used is determined by @c OutputFormat and @c Radix. -static void printObjectSectionSizes(ObjectFile *Obj) { - uint64_t total = 0; - std::string fmtbuf; - raw_string_ostream fmt(fmtbuf); - const char *radix_fmt = getRadixFmt(); - - // If OutputFormat is darwin and we have a MachOObjectFile print as darwin's - // size(1) -m output, else if OutputFormat is darwin and not a Mach-O object - // let it fall through to OutputFormat berkeley. - MachOObjectFile *MachO = dyn_cast<MachOObjectFile>(Obj); - if (OutputFormat == darwin && MachO) - printDarwinSectionSizes(MachO); - // If we have a MachOObjectFile and the OutputFormat is berkeley print as - // darwin's default berkeley format for Mach-O files. - else if (MachO && OutputFormat == berkeley) - printDarwinSegmentSizes(MachO); - else 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"); - std::size_t max_size_len = strlen("size"); - std::size_t max_addr_len = strlen("addr"); - for (const SectionRef &Section : Obj->sections()) { - if (!considerForSize(Obj, Section)) - continue; - uint64_t size = Section.getSize(); - total += size; - - StringRef name; - if (error(Section.getName(name))) - return; - uint64_t addr = Section.getAddress(); - 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)); - } - - // Add extra padding. - max_name_len += 2; - max_size_len += 2; - max_addr_len += 2; - - // Setup header format. - 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 (const SectionRef &Section : Obj->sections()) { - if (!considerForSize(Obj, Section)) - continue; - StringRef name; - if (error(Section.getName(name))) - return; - uint64_t size = Section.getSize(); - uint64_t addr = Section.getAddress(); - std::string namestr = name; - - outs() << format(fmt.str().c_str(), namestr.c_str(), size, addr); - } - - if (ELFCommons) { - uint64_t CommonSize = getCommonSize(Obj); - total += CommonSize; - outs() << format(fmt.str().c_str(), std::string("*COM*").c_str(), - CommonSize, static_cast<uint64_t>(0)); - } - - // 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 { - // The Berkeley format does not display individual section sizes. It - // displays the cumulative size for each section type. - uint64_t total_text = 0; - uint64_t total_data = 0; - uint64_t total_bss = 0; - - // Make one pass over the section table to calculate sizes. - for (const SectionRef &Section : Obj->sections()) { - uint64_t size = Section.getSize(); - bool isText = Section.isBerkeleyText(); - bool isData = Section.isBerkeleyData(); - bool isBSS = Section.isBSS(); - if (isText) - total_text += size; - else if (isData) - total_data += size; - else if (isBSS) - total_bss += size; - } - - if (ELFCommons) - total_bss += getCommonSize(Obj); - - total = total_text + total_data + total_bss; - - if (TotalSizes) { - TotalObjectText += total_text; - TotalObjectData += total_data; - TotalObjectBss += total_bss; - TotalObjectTotal += total; - } - - if (!BerkeleyHeaderPrinted) { - outs() << " text\t" - " data\t" - " bss\t" - " " - << (Radix == octal ? "oct" : "dec") - << "\t" - " hex\t" - "filename\n"; - BerkeleyHeaderPrinted = true; - } - - // Print result. - fmt << "%#7" << radix_fmt << "\t" - << "%#7" << radix_fmt << "\t" - << "%#7" << radix_fmt << "\t"; - outs() << format(fmt.str().c_str(), total_text, total_data, total_bss); - fmtbuf.clear(); - fmt << "%7" << (Radix == octal ? PRIo64 : PRIu64) << "\t" - << "%7" PRIx64 "\t"; - outs() << format(fmt.str().c_str(), total, total); - } -} - -/// Checks to see if the @p O ObjectFile is a Mach-O file and if it is and there -/// is a list of architecture flags specified then check to make sure this -/// Mach-O file is one of those architectures or all architectures was -/// specificed. If not then an error is generated and this routine returns -/// false. Else it returns true. -static bool checkMachOAndArchFlags(ObjectFile *O, StringRef Filename) { - auto *MachO = dyn_cast<MachOObjectFile>(O); - - if (!MachO || ArchAll || ArchFlags.empty()) - return true; - - MachO::mach_header H; - MachO::mach_header_64 H_64; - Triple T; - if (MachO->is64Bit()) { - H_64 = MachO->MachOObjectFile::getHeader64(); - T = MachOObjectFile::getArchTriple(H_64.cputype, H_64.cpusubtype); - } else { - H = MachO->MachOObjectFile::getHeader(); - T = MachOObjectFile::getArchTriple(H.cputype, H.cpusubtype); - } - if (none_of(ArchFlags, [&](const std::string &Name) { - return Name == T.getArchName(); - })) { - error(Filename + ": No architecture specified"); - return false; - } - return true; -} - -/// Print the section sizes for @p file. If @p file is an archive, print the -/// section sizes for each archive member. -static void printFileSectionSizes(StringRef file) { - - // Attempt to open the binary. - Expected<OwningBinary<Binary>> BinaryOrErr = createBinary(file); - if (!BinaryOrErr) { - error(BinaryOrErr.takeError(), file); - return; - } - Binary &Bin = *BinaryOrErr.get().getBinary(); - - if (Archive *a = dyn_cast<Archive>(&Bin)) { - // This is an archive. Iterate over each member and display its sizes. - Error Err = Error::success(); - for (auto &C : a->children(Err)) { - Expected<std::unique_ptr<Binary>> ChildOrErr = C.getAsBinary(); - if (!ChildOrErr) { - if (auto E = isNotObjectErrorInvalidFileType(ChildOrErr.takeError())) - error(std::move(E), a->getFileName(), C); - continue; - } - if (ObjectFile *o = dyn_cast<ObjectFile>(&*ChildOrErr.get())) { - MachOObjectFile *MachO = dyn_cast<MachOObjectFile>(o); - if (!checkMachOAndArchFlags(o, file)) - return; - if (OutputFormat == sysv) - outs() << o->getFileName() << " (ex " << a->getFileName() << "):\n"; - else if (MachO && OutputFormat == darwin) - outs() << a->getFileName() << "(" << o->getFileName() << "):\n"; - printObjectSectionSizes(o); - if (OutputFormat == berkeley) { - if (MachO) - outs() << a->getFileName() << "(" << o->getFileName() << ")\n"; - else - outs() << o->getFileName() << " (ex " << a->getFileName() << ")\n"; - } - } - } - if (Err) - error(std::move(Err), a->getFileName()); - } else if (MachOUniversalBinary *UB = - dyn_cast<MachOUniversalBinary>(&Bin)) { - // If we have a list of architecture flags specified dump only those. - if (!ArchAll && !ArchFlags.empty()) { - // Look for a slice in the universal binary that matches each ArchFlag. - bool ArchFound; - for (unsigned i = 0; i < ArchFlags.size(); ++i) { - ArchFound = false; - for (MachOUniversalBinary::object_iterator I = UB->begin_objects(), - E = UB->end_objects(); - I != E; ++I) { - if (ArchFlags[i] == I->getArchFlagName()) { - ArchFound = true; - Expected<std::unique_ptr<ObjectFile>> UO = I->getAsObjectFile(); - if (UO) { - if (ObjectFile *o = dyn_cast<ObjectFile>(&*UO.get())) { - MachOObjectFile *MachO = dyn_cast<MachOObjectFile>(o); - if (OutputFormat == sysv) - outs() << o->getFileName() << " :\n"; - else if (MachO && OutputFormat == darwin) { - if (MoreThanOneFile || ArchFlags.size() > 1) - outs() << o->getFileName() << " (for architecture " - << I->getArchFlagName() << "): \n"; - } - printObjectSectionSizes(o); - if (OutputFormat == berkeley) { - if (!MachO || MoreThanOneFile || ArchFlags.size() > 1) - outs() << o->getFileName() << " (for architecture " - << I->getArchFlagName() << ")"; - outs() << "\n"; - } - } - } else if (auto E = isNotObjectErrorInvalidFileType( - UO.takeError())) { - error(std::move(E), file, ArchFlags.size() > 1 ? - StringRef(I->getArchFlagName()) : StringRef()); - return; - } else if (Expected<std::unique_ptr<Archive>> AOrErr = - I->getAsArchive()) { - std::unique_ptr<Archive> &UA = *AOrErr; - // This is an archive. Iterate over each member and display its - // sizes. - Error Err = Error::success(); - for (auto &C : UA->children(Err)) { - Expected<std::unique_ptr<Binary>> ChildOrErr = C.getAsBinary(); - if (!ChildOrErr) { - if (auto E = isNotObjectErrorInvalidFileType( - ChildOrErr.takeError())) - error(std::move(E), UA->getFileName(), C, - ArchFlags.size() > 1 ? - StringRef(I->getArchFlagName()) : StringRef()); - continue; - } - if (ObjectFile *o = dyn_cast<ObjectFile>(&*ChildOrErr.get())) { - MachOObjectFile *MachO = dyn_cast<MachOObjectFile>(o); - if (OutputFormat == sysv) - outs() << o->getFileName() << " (ex " << UA->getFileName() - << "):\n"; - else if (MachO && OutputFormat == darwin) - outs() << UA->getFileName() << "(" << o->getFileName() - << ")" - << " (for architecture " << I->getArchFlagName() - << "):\n"; - printObjectSectionSizes(o); - if (OutputFormat == berkeley) { - if (MachO) { - outs() << UA->getFileName() << "(" << o->getFileName() - << ")"; - if (ArchFlags.size() > 1) - outs() << " (for architecture " << I->getArchFlagName() - << ")"; - outs() << "\n"; - } else - outs() << o->getFileName() << " (ex " << UA->getFileName() - << ")\n"; - } - } - } - if (Err) - error(std::move(Err), UA->getFileName()); - } else { - consumeError(AOrErr.takeError()); - error("Mach-O universal file: " + file + " for architecture " + - StringRef(I->getArchFlagName()) + - " is not a Mach-O file or an archive file"); - } - } - } - if (!ArchFound) { - errs() << ToolName << ": file: " << file - << " does not contain architecture" << ArchFlags[i] << ".\n"; - return; - } - } - return; - } - // No architecture flags were specified so if this contains a slice that - // matches the host architecture dump only that. - if (!ArchAll) { - StringRef HostArchName = MachOObjectFile::getHostArch().getArchName(); - for (MachOUniversalBinary::object_iterator I = UB->begin_objects(), - E = UB->end_objects(); - I != E; ++I) { - if (HostArchName == I->getArchFlagName()) { - Expected<std::unique_ptr<ObjectFile>> UO = I->getAsObjectFile(); - if (UO) { - if (ObjectFile *o = dyn_cast<ObjectFile>(&*UO.get())) { - MachOObjectFile *MachO = dyn_cast<MachOObjectFile>(o); - if (OutputFormat == sysv) - outs() << o->getFileName() << " :\n"; - else if (MachO && OutputFormat == darwin) { - if (MoreThanOneFile) - outs() << o->getFileName() << " (for architecture " - << I->getArchFlagName() << "):\n"; - } - printObjectSectionSizes(o); - if (OutputFormat == berkeley) { - if (!MachO || MoreThanOneFile) - outs() << o->getFileName() << " (for architecture " - << I->getArchFlagName() << ")"; - outs() << "\n"; - } - } - } else if (auto E = isNotObjectErrorInvalidFileType(UO.takeError())) { - error(std::move(E), file); - return; - } else if (Expected<std::unique_ptr<Archive>> AOrErr = - I->getAsArchive()) { - std::unique_ptr<Archive> &UA = *AOrErr; - // This is an archive. Iterate over each member and display its - // sizes. - Error Err = Error::success(); - for (auto &C : UA->children(Err)) { - Expected<std::unique_ptr<Binary>> ChildOrErr = C.getAsBinary(); - if (!ChildOrErr) { - if (auto E = isNotObjectErrorInvalidFileType( - ChildOrErr.takeError())) - error(std::move(E), UA->getFileName(), C); - continue; - } - if (ObjectFile *o = dyn_cast<ObjectFile>(&*ChildOrErr.get())) { - MachOObjectFile *MachO = dyn_cast<MachOObjectFile>(o); - if (OutputFormat == sysv) - outs() << o->getFileName() << " (ex " << UA->getFileName() - << "):\n"; - else if (MachO && OutputFormat == darwin) - outs() << UA->getFileName() << "(" << o->getFileName() << ")" - << " (for architecture " << I->getArchFlagName() - << "):\n"; - printObjectSectionSizes(o); - if (OutputFormat == berkeley) { - if (MachO) - outs() << UA->getFileName() << "(" << o->getFileName() - << ")\n"; - else - outs() << o->getFileName() << " (ex " << UA->getFileName() - << ")\n"; - } - } - } - if (Err) - error(std::move(Err), UA->getFileName()); - } else { - consumeError(AOrErr.takeError()); - error("Mach-O universal file: " + file + " for architecture " + - StringRef(I->getArchFlagName()) + - " is not a Mach-O file or an archive file"); - } - return; - } - } - } - // Either all architectures have been specified or none have been specified - // and this does not contain the host architecture so dump all the slices. - bool MoreThanOneArch = UB->getNumberOfObjects() > 1; - for (MachOUniversalBinary::object_iterator I = UB->begin_objects(), - E = UB->end_objects(); - I != E; ++I) { - Expected<std::unique_ptr<ObjectFile>> UO = I->getAsObjectFile(); - if (UO) { - if (ObjectFile *o = dyn_cast<ObjectFile>(&*UO.get())) { - MachOObjectFile *MachO = dyn_cast<MachOObjectFile>(o); - if (OutputFormat == sysv) - outs() << o->getFileName() << " :\n"; - else if (MachO && OutputFormat == darwin) { - if (MoreThanOneFile || MoreThanOneArch) - outs() << o->getFileName() << " (for architecture " - << I->getArchFlagName() << "):"; - outs() << "\n"; - } - printObjectSectionSizes(o); - if (OutputFormat == berkeley) { - if (!MachO || MoreThanOneFile || MoreThanOneArch) - outs() << o->getFileName() << " (for architecture " - << I->getArchFlagName() << ")"; - outs() << "\n"; - } - } - } else if (auto E = isNotObjectErrorInvalidFileType(UO.takeError())) { - error(std::move(E), file, MoreThanOneArch ? - StringRef(I->getArchFlagName()) : StringRef()); - return; - } else if (Expected<std::unique_ptr<Archive>> AOrErr = - I->getAsArchive()) { - std::unique_ptr<Archive> &UA = *AOrErr; - // This is an archive. Iterate over each member and display its sizes. - Error Err = Error::success(); - for (auto &C : UA->children(Err)) { - Expected<std::unique_ptr<Binary>> ChildOrErr = C.getAsBinary(); - if (!ChildOrErr) { - if (auto E = isNotObjectErrorInvalidFileType( - ChildOrErr.takeError())) - error(std::move(E), UA->getFileName(), C, MoreThanOneArch ? - StringRef(I->getArchFlagName()) : StringRef()); - continue; - } - if (ObjectFile *o = dyn_cast<ObjectFile>(&*ChildOrErr.get())) { - MachOObjectFile *MachO = dyn_cast<MachOObjectFile>(o); - if (OutputFormat == sysv) - outs() << o->getFileName() << " (ex " << UA->getFileName() - << "):\n"; - else if (MachO && OutputFormat == darwin) - outs() << UA->getFileName() << "(" << o->getFileName() << ")" - << " (for architecture " << I->getArchFlagName() << "):\n"; - printObjectSectionSizes(o); - if (OutputFormat == berkeley) { - if (MachO) - outs() << UA->getFileName() << "(" << o->getFileName() << ")" - << " (for architecture " << I->getArchFlagName() - << ")\n"; - else - outs() << o->getFileName() << " (ex " << UA->getFileName() - << ")\n"; - } - } - } - if (Err) - error(std::move(Err), UA->getFileName()); - } else { - consumeError(AOrErr.takeError()); - error("Mach-O universal file: " + file + " for architecture " + - StringRef(I->getArchFlagName()) + - " is not a Mach-O file or an archive file"); - } - } - } else if (ObjectFile *o = dyn_cast<ObjectFile>(&Bin)) { - if (!checkMachOAndArchFlags(o, file)) - return; - MachOObjectFile *MachO = dyn_cast<MachOObjectFile>(o); - if (OutputFormat == sysv) - outs() << o->getFileName() << " :\n"; - else if (MachO && OutputFormat == darwin && MoreThanOneFile) - outs() << o->getFileName() << ":\n"; - printObjectSectionSizes(o); - if (OutputFormat == berkeley) { - if (!MachO || MoreThanOneFile) - outs() << o->getFileName(); - outs() << "\n"; - } - } else { - errs() << ToolName << ": " << file << ": " - << "Unrecognized file type.\n"; - } - // System V adds an extra newline at the end of each file. - if (OutputFormat == sysv) - outs() << "\n"; -} - -static void printBerkelyTotals() { - std::string fmtbuf; - raw_string_ostream fmt(fmtbuf); - const char *radix_fmt = getRadixFmt(); - fmt << "%#7" << radix_fmt << "\t" - << "%#7" << radix_fmt << "\t" - << "%#7" << radix_fmt << "\t"; - outs() << format(fmt.str().c_str(), TotalObjectText, TotalObjectData, - TotalObjectBss); - fmtbuf.clear(); - fmt << "%7" << (Radix == octal ? PRIo64 : PRIu64) << "\t" - << "%7" PRIx64 "\t"; - outs() << format(fmt.str().c_str(), TotalObjectTotal, TotalObjectTotal) - << "(TOTALS)\n"; -} - -int main(int argc, char **argv) { - InitLLVM X(argc, argv); - cl::ParseCommandLineOptions(argc, argv, "llvm object size dumper\n"); - - ToolName = argv[0]; - if (OutputFormatShort.getNumOccurrences()) - OutputFormat = static_cast<OutputFormatTy>(OutputFormatShort); - if (RadixShort.getNumOccurrences()) - Radix = RadixShort.getValue(); - - for (StringRef Arch : ArchFlags) { - if (Arch == "all") { - ArchAll = true; - } else { - if (!MachOObjectFile::isValidArch(Arch)) { - outs() << ToolName << ": for the -arch option: Unknown architecture " - << "named '" << Arch << "'"; - return 1; - } - } - } - - if (InputFilenames.empty()) - InputFilenames.push_back("a.out"); - - MoreThanOneFile = InputFilenames.size() > 1; - llvm::for_each(InputFilenames, printFileSectionSizes); - if (OutputFormat == berkeley && TotalSizes) - printBerkelyTotals(); - - if (HadError) - return 1; -} |