Magic-Mirror/llvm-capstone
1
0
Fork 0
mirror of https://github.com/capstone-engine/llvm-capstone.git synced 2025-01-06 08:02:18 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Define a helper function to simplify. NFC.

Browse Source 
llvm-svn: 265757
This commit is contained in:
Rui Ueyama 2016-04-08 00:14:55 +00:00
parent 466a15ef7f
commit 4655ea3ff8
1 changed files with 13 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
lld/ELF/InputFiles.cpp
View File

@ -553,11 +553,15 @@ static std::unique_ptr<InputFile> createELFFile(MemoryBufferRef MB) {
fatal("invalid file class: " + MB.getBufferIdentifier());
}
static bool isBitcode(MemoryBufferRef MB) {
using namespace sys::fs;
return identify_magic(MB.getBuffer()) == file_magic::bitcode;
}
std::unique_ptr<InputFile> elf::createObjectFile(MemoryBufferRef MB,
StringRef ArchiveName) {
using namespace sys::fs;
std::unique_ptr<InputFile> F;
if (identify_magic(MB.getBuffer()) == file_magic::bitcode)
if (isBitcode(MB))
F.reset(new BitcodeFile(MB));
else
F = createELFFile<ObjectFile>(MB);
@ -612,19 +616,18 @@ std::vector<StringRef> LazyObjectFile::getBitcodeSymbols() {
// Returns a vector of globally-visible symbol names.
std::vector<StringRef> LazyObjectFile::getSymbols() {
using namespace sys::fs;
StringRef Buf = this->MB.getBuffer();
if (identify_magic(Buf) == file_magic::bitcode)
if (isBitcode(this->MB))
return getBitcodeSymbols();
std::pair<unsigned char, unsigned char> Type = getElfArchType(Buf);
if (Type.first == ELF::ELFCLASS32) {
if (Type.second == ELF::ELFDATA2LSB)
unsigned char Size;
unsigned char Endian;
std::tie(Size, Endian) = getElfArchType(this->MB.getBuffer());
if (Size == ELFCLASS32) {
if (Endian == ELFDATA2LSB)
return getElfSymbols<ELF32LE>();
return getElfSymbols<ELF32BE>();
}
if (Type.second == ELF::ELFDATA2LSB)
if (Endian == ELFDATA2LSB)
return getElfSymbols<ELF64LE>();
return getElfSymbols<ELF64BE>();
}