mgorny created this revision. mgorny added reviewers: bruno, bkramer, rafael. mgorny added a subscriber: cfe-commits.
Refactor the DetectDistro() function to take a single vfs::FileSystem reference only, instead of Driver and llvm::Triple::ArchType. The ArchType parameter was not used anyway, and Driver was only used to obtain the VFS. Aside to making the API simpler and more transparent, it makes it easier to add unit tests for the function in the future -- since the tests would need only to provide an appropriate VFS. https://reviews.llvm.org/D25819 Files: lib/Driver/ToolChains.cpp Index: lib/Driver/ToolChains.cpp =================================================================== --- lib/Driver/ToolChains.cpp +++ lib/Driver/ToolChains.cpp @@ -3850,9 +3850,9 @@ return Distro >= UbuntuHardy && Distro <= UbuntuYakkety; } -static Distro DetectDistro(const Driver &D, llvm::Triple::ArchType Arch) { +static Distro DetectDistro(vfs::FileSystem &VFS) { llvm::ErrorOr<std::unique_ptr<llvm::MemoryBuffer>> File = - D.getVFS().getBufferForFile("/etc/lsb-release"); + VFS.getBufferForFile("/etc/lsb-release"); if (File) { StringRef Data = File.get()->getBuffer(); SmallVector<StringRef, 16> Lines; @@ -3884,7 +3884,7 @@ return Version; } - File = D.getVFS().getBufferForFile("/etc/redhat-release"); + File = VFS.getBufferForFile("/etc/redhat-release"); if (File) { StringRef Data = File.get()->getBuffer(); if (Data.startswith("Fedora release")) @@ -3902,7 +3902,7 @@ return UnknownDistro; } - File = D.getVFS().getBufferForFile("/etc/debian_version"); + File = VFS.getBufferForFile("/etc/debian_version"); if (File) { StringRef Data = File.get()->getBuffer(); if (Data[0] == '5') @@ -3918,13 +3918,13 @@ return UnknownDistro; } - if (D.getVFS().exists("/etc/SuSE-release")) + if (VFS.exists("/etc/SuSE-release")) return OpenSUSE; - if (D.getVFS().exists("/etc/exherbo-release")) + if (VFS.exists("/etc/exherbo-release")) return Exherbo; - if (D.getVFS().exists("/etc/arch-release")) + if (VFS.exists("/etc/arch-release")) return ArchLinux; return UnknownDistro; @@ -4109,7 +4109,7 @@ GCCInstallation.getTriple().str() + "/bin") .str()); - Distro Distro = DetectDistro(D, Arch); + Distro Distro = DetectDistro(D.getVFS()); if (IsOpenSUSE(Distro) || IsUbuntu(Distro)) { ExtraOpts.push_back("-z"); @@ -4313,7 +4313,7 @@ const llvm::Triple::ArchType Arch = getArch(); const llvm::Triple &Triple = getTriple(); - const enum Distro Distro = DetectDistro(getDriver(), Arch); + const enum Distro Distro = DetectDistro(getDriver().getVFS()); if (Triple.isAndroid()) return Triple.isArch64Bit() ? "/system/bin/linker64" : "/system/bin/linker";
Index: lib/Driver/ToolChains.cpp =================================================================== --- lib/Driver/ToolChains.cpp +++ lib/Driver/ToolChains.cpp @@ -3850,9 +3850,9 @@ return Distro >= UbuntuHardy && Distro <= UbuntuYakkety; } -static Distro DetectDistro(const Driver &D, llvm::Triple::ArchType Arch) { +static Distro DetectDistro(vfs::FileSystem &VFS) { llvm::ErrorOr<std::unique_ptr<llvm::MemoryBuffer>> File = - D.getVFS().getBufferForFile("/etc/lsb-release"); + VFS.getBufferForFile("/etc/lsb-release"); if (File) { StringRef Data = File.get()->getBuffer(); SmallVector<StringRef, 16> Lines; @@ -3884,7 +3884,7 @@ return Version; } - File = D.getVFS().getBufferForFile("/etc/redhat-release"); + File = VFS.getBufferForFile("/etc/redhat-release"); if (File) { StringRef Data = File.get()->getBuffer(); if (Data.startswith("Fedora release")) @@ -3902,7 +3902,7 @@ return UnknownDistro; } - File = D.getVFS().getBufferForFile("/etc/debian_version"); + File = VFS.getBufferForFile("/etc/debian_version"); if (File) { StringRef Data = File.get()->getBuffer(); if (Data[0] == '5') @@ -3918,13 +3918,13 @@ return UnknownDistro; } - if (D.getVFS().exists("/etc/SuSE-release")) + if (VFS.exists("/etc/SuSE-release")) return OpenSUSE; - if (D.getVFS().exists("/etc/exherbo-release")) + if (VFS.exists("/etc/exherbo-release")) return Exherbo; - if (D.getVFS().exists("/etc/arch-release")) + if (VFS.exists("/etc/arch-release")) return ArchLinux; return UnknownDistro; @@ -4109,7 +4109,7 @@ GCCInstallation.getTriple().str() + "/bin") .str()); - Distro Distro = DetectDistro(D, Arch); + Distro Distro = DetectDistro(D.getVFS()); if (IsOpenSUSE(Distro) || IsUbuntu(Distro)) { ExtraOpts.push_back("-z"); @@ -4313,7 +4313,7 @@ const llvm::Triple::ArchType Arch = getArch(); const llvm::Triple &Triple = getTriple(); - const enum Distro Distro = DetectDistro(getDriver(), Arch); + const enum Distro Distro = DetectDistro(getDriver().getVFS()); if (Triple.isAndroid()) return Triple.isArch64Bit() ? "/system/bin/linker64" : "/system/bin/linker";
_______________________________________________ cfe-commits mailing list cfe-commits@lists.llvm.org http://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits