#include "llvm/Support/Path.h"
#include "llvm/Support/FileSystem.h"
#include "llvm/Config/config.h"
#include "llvm/Support/FileSystem.h"
#include "llvm/Support/Endian.h"
#include <cassert>
#include <cstring>
#include <ostream>
using namespace llvm;
using namespace sys;
namespace {
using support::ulittle32_t;
}
bool Path::operator==(const Path &that) const {
return path == that.path;
}
bool Path::operator<(const Path& that) const {
return path < that.path;
}
Path
Path::GetLLVMConfigDir() {
Path result;
#ifdef LLVM_ETCDIR
if (result.set(LLVM_ETCDIR))
return result;
#endif
return GetLLVMDefaultConfigDir();
}
LLVMFileType
sys::IdentifyFileType(const char *magic, unsigned length) {
assert(magic && "Invalid magic number string");
assert(length >=4 && "Invalid magic number length");
switch ((unsigned char)magic[0]) {
case 0xDE: if (magic[1] == (char)0xC0 && magic[2] == (char)0x17 &&
magic[3] == (char)0x0B)
return Bitcode_FileType;
break;
case 'B':
if (magic[1] == 'C' && magic[2] == (char)0xC0 && magic[3] == (char)0xDE)
return Bitcode_FileType;
break;
case '!':
if (length >= 8)
if (memcmp(magic,"!<arch>\n",8) == 0)
return Archive_FileType;
break;
case '\177':
if (magic[1] == 'E' && magic[2] == 'L' && magic[3] == 'F') {
if (length >= 18 && magic[17] == 0)
switch (magic[16]) {
default: break;
case 1: return ELF_Relocatable_FileType;
case 2: return ELF_Executable_FileType;
case 3: return ELF_SharedObject_FileType;
case 4: return ELF_Core_FileType;
}
}
break;
case 0xCA:
if (magic[1] == char(0xFE) && magic[2] == char(0xBA) &&
magic[3] == char(0xBE)) {
if (length >= 8 && magic[7] < 43)
return Mach_O_DynamicallyLinkedSharedLib_FileType;
}
break;
case 0xFE:
case 0xCE:
case 0xCF: {
uint16_t type = 0;
if (magic[0] == char(0xFE) && magic[1] == char(0xED) &&
magic[2] == char(0xFA) &&
(magic[3] == char(0xCE) || magic[3] == char(0xCF))) {
if (length >= 16) type = magic[14] << 8 | magic[15];
} else if ((magic[0] == char(0xCE) || magic[0] == char(0xCF)) &&
magic[1] == char(0xFA) && magic[2] == char(0xED) &&
magic[3] == char(0xFE)) {
if (length >= 14) type = magic[13] << 8 | magic[12];
}
switch (type) {
default: break;
case 1: return Mach_O_Object_FileType;
case 2: return Mach_O_Executable_FileType;
case 3: return Mach_O_FixedVirtualMemorySharedLib_FileType;
case 4: return Mach_O_Core_FileType;
case 5: return Mach_O_PreloadExecutable_FileType;
case 6: return Mach_O_DynamicallyLinkedSharedLib_FileType;
case 7: return Mach_O_DynamicLinker_FileType;
case 8: return Mach_O_Bundle_FileType;
case 9: return Mach_O_DynamicallyLinkedSharedLibStub_FileType;
case 10: break; }
break;
}
case 0xF0: case 0x83: case 0x84: case 0x66: case 0x50: case 0x4c: if (magic[1] == 0x01)
return COFF_FileType;
case 0x90: case 0x68: if (magic[1] == 0x02)
return COFF_FileType;
break;
case 0x4d: if (magic[1] == 0x5a) {
uint32_t off = *reinterpret_cast<const ulittle32_t *>(magic + 0x3c);
if (off < length && memcmp(magic + off, "PE\0\0",4) == 0)
return COFF_FileType;
}
break;
case 0x64: if (magic[1] == char(0x86))
return COFF_FileType;
break;
default:
break;
}
return Unknown_FileType;
}
bool
Path::isArchive() const {
LLVMFileType type;
if (fs::identify_magic(str(), type))
return false;
return type == Archive_FileType;
}
bool
Path::isDynamicLibrary() const {
LLVMFileType type;
if (fs::identify_magic(str(), type))
return false;
switch (type) {
default: return false;
case Mach_O_FixedVirtualMemorySharedLib_FileType:
case Mach_O_DynamicallyLinkedSharedLib_FileType:
case Mach_O_DynamicallyLinkedSharedLibStub_FileType:
case ELF_SharedObject_FileType:
case COFF_FileType: return true;
}
}
bool
Path::isObjectFile() const {
LLVMFileType type;
if (fs::identify_magic(str(), type) || type == Unknown_FileType)
return false;
return true;
}
Path
Path::FindLibrary(std::string& name) {
std::vector<sys::Path> LibPaths;
GetSystemLibraryPaths(LibPaths);
for (unsigned i = 0; i < LibPaths.size(); ++i) {
sys::Path FullPath(LibPaths[i]);
FullPath.appendComponent("lib" + name + LTDL_SHLIB_EXT);
if (FullPath.isDynamicLibrary())
return FullPath;
FullPath.eraseSuffix();
FullPath.appendSuffix("a");
if (FullPath.isArchive())
return FullPath;
}
return sys::Path();
}
StringRef Path::GetDLLSuffix() {
return &(LTDL_SHLIB_EXT[1]);
}
void
Path::appendSuffix(StringRef suffix) {
if (!suffix.empty()) {
path.append(".");
path.append(suffix);
}
}
bool
Path::isBitcodeFile() const {
LLVMFileType type;
if (fs::identify_magic(str(), type))
return false;
return type == Bitcode_FileType;
}
bool Path::hasMagicNumber(StringRef Magic) const {
std::string actualMagic;
if (getMagicNumber(actualMagic, static_cast<unsigned>(Magic.size())))
return Magic == actualMagic;
return false;
}
static void getPathList(const char*path, std::vector<Path>& Paths) {
const char* at = path;
const char* delim = strchr(at, PathSeparator);
Path tmpPath;
while (delim != 0) {
std::string tmp(at, size_t(delim-at));
if (tmpPath.set(tmp))
if (tmpPath.canRead())
Paths.push_back(tmpPath);
at = delim + 1;
delim = strchr(at, PathSeparator);
}
if (*at != 0)
if (tmpPath.set(std::string(at)))
if (tmpPath.canRead())
Paths.push_back(tmpPath);
}
static StringRef getDirnameCharSep(StringRef path, const char *Sep) {
assert(Sep[0] != '\0' && Sep[1] == '\0' &&
"Sep must be a 1-character string literal.");
if (path.empty())
return ".";
signed pos = static_cast<signed>(path.size()) - 1;
while (pos >= 0 && path[pos] == Sep[0])
--pos;
if (pos < 0)
return path[0] == Sep[0] ? Sep : ".";
signed i = 0;
while (i < pos && path[i] != Sep[0])
++i;
if (i == pos) return ".";
while (pos >= 0 && path[pos] != Sep[0])
--pos;
while (pos >= 0 && path[pos] == Sep[0])
--pos;
if (pos < 0)
return path[0] == Sep[0] ? Sep : ".";
return path.substr(0, pos+1);
}
#if defined(LLVM_ON_UNIX)
#include "Unix/Path.inc"
#endif
#if defined(LLVM_ON_WIN32)
#include "Windows/Path.inc"
#endif