mirror of
https://github.com/Gericom/teak-llvm.git
synced 2025-06-25 14:28:54 -04:00
5b2e968264
On Windows at least, llvm-strings was crashing if it encountered bytes that mapped to negative chars, as it was passing these into std::isgraph and std::isblank functions, resulting in undefined behaviour. On debug builds using MSVC, these functions verfiy that the value passed in is representable as an unsigned char. Since the char is promoted to an int, a value greater than 127 would turn into a negative integer value, and fail the check. Using the llvm::isPrint function is sufficient to solve the issue. Reviewed by: ruiu, mstorsjo Differential Revision: https://reviews.llvm.org/D53509 llvm-svn: 345137
119 lines
3.8 KiB
C++
119 lines
3.8 KiB
C++
//===-- llvm-strings.cpp - Printable String dumping utility ---------------===//
|
|
//
|
|
// 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 binutils "strings", that is, it
|
|
// prints out printable strings in a binary, objdump, or archive file.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "llvm/Object/Binary.h"
|
|
#include "llvm/Support/CommandLine.h"
|
|
#include "llvm/Support/Error.h"
|
|
#include "llvm/Support/Format.h"
|
|
#include "llvm/Support/InitLLVM.h"
|
|
#include "llvm/Support/MemoryBuffer.h"
|
|
#include "llvm/Support/Program.h"
|
|
#include <cctype>
|
|
#include <string>
|
|
|
|
using namespace llvm;
|
|
using namespace llvm::object;
|
|
|
|
static cl::list<std::string> InputFileNames(cl::Positional,
|
|
cl::desc("<input object files>"),
|
|
cl::ZeroOrMore);
|
|
|
|
static cl::opt<bool>
|
|
PrintFileName("print-file-name",
|
|
cl::desc("Print the name of the file before each string"));
|
|
static cl::alias PrintFileNameShort("f", cl::desc(""),
|
|
cl::aliasopt(PrintFileName));
|
|
|
|
static cl::opt<int>
|
|
MinLength("bytes", cl::desc("Print sequences of the specified length"),
|
|
cl::init(4));
|
|
static cl::alias MinLengthShort("n", cl::desc(""), cl::aliasopt(MinLength));
|
|
|
|
static cl::opt<bool>
|
|
AllSections("all",
|
|
cl::desc("Check all sections, not just the data section"));
|
|
static cl::alias AllSectionsShort("a", cl::desc(""),
|
|
cl::aliasopt(AllSections));
|
|
|
|
enum radix { none, octal, hexadecimal, decimal };
|
|
static cl::opt<radix>
|
|
Radix("radix", cl::desc("print the offset within the file"),
|
|
cl::values(clEnumValN(octal, "o", "octal"),
|
|
clEnumValN(hexadecimal, "x", "hexadecimal"),
|
|
clEnumValN(decimal, "d", "decimal")),
|
|
cl::init(none));
|
|
static cl::alias RadixShort("t", cl::desc(""), cl::aliasopt(Radix));
|
|
|
|
static void strings(raw_ostream &OS, StringRef FileName, StringRef Contents) {
|
|
auto print = [&OS, FileName](unsigned Offset, StringRef L) {
|
|
if (L.size() < static_cast<size_t>(MinLength))
|
|
return;
|
|
if (PrintFileName)
|
|
OS << FileName << ":";
|
|
switch (Radix) {
|
|
case none:
|
|
break;
|
|
case octal:
|
|
OS << format("%8o", Offset);
|
|
break;
|
|
case hexadecimal:
|
|
OS << format("%8x", Offset);
|
|
break;
|
|
case decimal:
|
|
OS << format("%8u", Offset);
|
|
break;
|
|
}
|
|
OS << " " << L << '\n';
|
|
};
|
|
|
|
const char *B = Contents.begin();
|
|
const char *P = nullptr, *E = nullptr, *S = nullptr;
|
|
for (P = Contents.begin(), E = Contents.end(); P < E; ++P) {
|
|
if (isPrint(*P) || *P == '\t') {
|
|
if (S == nullptr)
|
|
S = P;
|
|
} else if (S) {
|
|
print(S - B, StringRef(S, P - S));
|
|
S = nullptr;
|
|
}
|
|
}
|
|
if (S)
|
|
print(S - B, StringRef(S, E - S));
|
|
}
|
|
|
|
int main(int argc, char **argv) {
|
|
InitLLVM X(argc, argv);
|
|
|
|
cl::ParseCommandLineOptions(argc, argv, "llvm string dumper\n");
|
|
if (MinLength == 0) {
|
|
errs() << "invalid minimum string length 0\n";
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
if (InputFileNames.empty())
|
|
InputFileNames.push_back("-");
|
|
|
|
for (const auto &File : InputFileNames) {
|
|
ErrorOr<std::unique_ptr<MemoryBuffer>> Buffer =
|
|
MemoryBuffer::getFileOrSTDIN(File);
|
|
if (std::error_code EC = Buffer.getError())
|
|
errs() << File << ": " << EC.message() << '\n';
|
|
else
|
|
strings(llvm::outs(), File == "-" ? "{standard input}" : File,
|
|
Buffer.get()->getMemBufferRef().getBuffer());
|
|
}
|
|
|
|
return EXIT_SUCCESS;
|
|
}
|