diff --git a/lib/System/Path.cpp b/lib/System/Path.cpp index 634ce5dabb0..fcdc5ff22cf 100644 --- a/lib/System/Path.cpp +++ b/lib/System/Path.cpp @@ -43,6 +43,10 @@ sys::IdentifyFileType(const char*magic, unsigned length) { assert(magic && "Invalid magic number string"); assert(length >=4 && "Invalid magic number length"); switch (magic[0]) { + case 'B': + if (magic[1] == 'C' && magic[2] == (char)0xC0 && magic[3] == (char)0xDE) + return Bitcode_FileType; + break; case 'l': if (magic[1] == 'l' && magic[2] == 'v') { if (magic[3] == 'c') @@ -160,6 +164,31 @@ std::string Path::GetDLLSuffix() { return LTDL_SHLIB_EXT; } +bool +Path::isBytecodeFile() const { + std::string actualMagic; + if (!getMagicNumber(actualMagic, 4)) + return false; + return actualMagic == "llvc" || actualMagic == "llvm"; +} + +bool +Path::isBitcodeFile() const { + std::string actualMagic; + if (!getMagicNumber(actualMagic, 4)) + return false; + return actualMagic == "BC\xC0\xDE"; +} + +bool Path::hasMagicNumber(const std::string &Magic) const { + std::string actualMagic; + if (getMagicNumber(actualMagic, Magic.size())) + return Magic == actualMagic; + return false; +} + + + // Include the truly platform-specific parts of this class. #if defined(LLVM_ON_UNIX) #include "Unix/Path.inc" diff --git a/lib/System/Unix/Path.inc b/lib/System/Unix/Path.inc index 53906d99cec..db95e3c216c 100644 --- a/lib/System/Unix/Path.inc +++ b/lib/System/Unix/Path.inc @@ -267,21 +267,6 @@ Path::getBasename() const { return path.substr(slash, dot - slash); } -bool Path::hasMagicNumber(const std::string &Magic) const { - size_t len = Magic.size(); - assert(len < 1024 && "Request for magic string too long"); - char* buf = (char*) alloca(1 + len); - int fd = ::open(path.c_str(), O_RDONLY); - if (fd < 0) - return false; - size_t read_len = ::read(fd, buf, len); - close(fd); - if (len != read_len) - return false; - buf[len] = '\0'; - return Magic == buf; -} - bool Path::getMagicNumber(std::string& Magic, unsigned len) const { assert(len < 1024 && "Request for magic string too long"); char* buf = (char*) alloca(1 + len); @@ -298,22 +283,6 @@ bool Path::getMagicNumber(std::string& Magic, unsigned len) const { return true; } -bool -Path::isBytecodeFile() const { - char buffer[4]; - buffer[0] = 0; - int fd = ::open(path.c_str(), O_RDONLY); - if (fd < 0) - return false; - ssize_t bytes_read = ::read(fd, buffer, 4); - ::close(fd); - if (4 != bytes_read) - return false; - - return (buffer[0] == 'l' && buffer[1] == 'l' && buffer[2] == 'v' && - (buffer[3] == 'c' || buffer[3] == 'm')); -} - bool Path::exists() const { return 0 == access(path.c_str(), F_OK ); diff --git a/lib/System/Win32/Path.inc b/lib/System/Win32/Path.inc index f28fe86b59b..67f16a75b60 100644 --- a/lib/System/Win32/Path.inc +++ b/lib/System/Win32/Path.inc @@ -247,21 +247,6 @@ Path::getBasename() const { return path.substr(slash, dot - slash); } -bool Path::hasMagicNumber(const std::string &Magic) const { - std::string actualMagic; - if (getMagicNumber(actualMagic, Magic.size())) - return Magic == actualMagic; - return false; -} - -bool -Path::isBytecodeFile() const { - std::string actualMagic; - if (!getMagicNumber(actualMagic, 4)) - return false; - return actualMagic == "llvc" || actualMagic == "llvm"; -} - bool Path::exists() const { DWORD attr = GetFileAttributes(path.c_str());