[Driver] Add unit tests for Distro detection

Add a set of unit tests for the distro detection code. The tests use an
in-memory virtual filesystems resembling release files for various
distributions supported. All release files are provided (not only the
ones directly used) in order to guarantee that one of the rules will not
mistakenly recognize the distribution incorrectly due to the additional
files (e.g. Ubuntu as Debian).

Differential Revision: https://reviews.llvm.org/D25869

llvm-svn: 288062

2 files changed, 306 insertions, 0 deletions

diff --git a/clang/unittests/Driver/CMakeLists.txt b/clang/unittests/Driver/CMakeLists.txt
index 2df1eb24fd4..b9c52f7a53f 100644
--- a/clang/unittests/Driver/CMakeLists.txt
+++ b/clang/unittests/Driver/CMakeLists.txt
@@ -3,6 +3,7 @@ set(LLVM_LINK_COMPONENTS
   )
 
 add_clang_unittest(ClangDriverTests
+  DistroTest.cpp
   ToolChainTest.cpp
   MultilibTest.cpp
   )
diff --git a/clang/unittests/Driver/DistroTest.cpp b/clang/unittests/Driver/DistroTest.cpp
new file mode 100644
index 00000000000..e3686e49806
--- /dev/null
+++ b/clang/unittests/Driver/DistroTest.cpp
@@ -0,0 +1,305 @@
+//===- unittests/Driver/DistroTest.cpp --- ToolChains tests ---------------===//
+//
+//                     The LLVM Compiler Infrastructure
+//
+// This file is distributed under the University of Illinois Open Source
+// License. See LICENSE.TXT for details.
+//
+//===----------------------------------------------------------------------===//
+//
+// Unit tests for Distro detection.
+//
+//===----------------------------------------------------------------------===//
+
+#include "clang/Driver/Distro.h"
+#include "clang/Basic/VirtualFileSystem.h"
+#include "llvm/Support/raw_ostream.h"
+#include "gtest/gtest.h"
+using namespace clang;
+using namespace clang::driver;
+
+namespace {
+
+// The tests include all release-related files for each distribution
+// in the VFS, in order to make sure that earlier tests do not
+// accidentally result in incorrect distribution guess.
+
+TEST(DistroTest, DetectUbuntu) {
+  vfs::InMemoryFileSystem UbuntuTrustyFileSystem;
+  // Ubuntu uses Debian Sid version.
+  UbuntuTrustyFileSystem.addFile("/etc/debian_version", 0,
+      llvm::MemoryBuffer::getMemBuffer("jessie/sid\n"));
+  UbuntuTrustyFileSystem.addFile("/etc/lsb-release", 0,
+      llvm::MemoryBuffer::getMemBuffer("DISTRIB_ID=Ubuntu\n"
+                                       "DISTRIB_RELEASE=14.04\n"
+                                       "DISTRIB_CODENAME=trusty\n"
+                                       "DISTRIB_DESCRIPTION=\"Ubuntu 14.04 LTS\"\n"));
+  UbuntuTrustyFileSystem.addFile("/etc/os-release", 0,
+      llvm::MemoryBuffer::getMemBuffer("NAME=\"Ubuntu\"\n"
+                                       "VERSION=\"14.04, Trusty Tahr\"\n"
+                                       "ID=ubuntu\n"
+                                       "ID_LIKE=debian\n"
+                                       "PRETTY_NAME=\"Ubuntu 14.04 LTS\"\n"
+                                       "VERSION_ID=\"14.04\"\n"
+                                       "HOME_URL=\"http://www.ubuntu.com/\"\n"
+                                       "SUPPORT_URL=\"http://help.ubuntu.com/\"\n"
+                                       "BUG_REPORT_URL=\"http://bugs.launchpad.net/ubuntu/\"\n"));
+
+  Distro UbuntuTrusty{UbuntuTrustyFileSystem};
+  ASSERT_EQ(Distro(Distro::UbuntuTrusty), UbuntuTrusty);
+  ASSERT_TRUE(UbuntuTrusty.IsUbuntu());
+  ASSERT_FALSE(UbuntuTrusty.IsRedhat());
+  ASSERT_FALSE(UbuntuTrusty.IsOpenSUSE());
+  ASSERT_FALSE(UbuntuTrusty.IsDebian());
+
+  vfs::InMemoryFileSystem UbuntuYakketyFileSystem;
+  UbuntuYakketyFileSystem.addFile("/etc/debian_version", 0,
+      llvm::MemoryBuffer::getMemBuffer("stretch/sid\n"));
+  UbuntuYakketyFileSystem.addFile("/etc/lsb-release", 0,
+      llvm::MemoryBuffer::getMemBuffer("DISTRIB_ID=Ubuntu\n"
+                                       "DISTRIB_RELEASE=16.10\n"
+                                       "DISTRIB_CODENAME=yakkety\n"
+                                       "DISTRIB_DESCRIPTION=\"Ubuntu 16.10\"\n"));
+  UbuntuYakketyFileSystem.addFile("/etc/os-release", 0,
+      llvm::MemoryBuffer::getMemBuffer("NAME=\"Ubuntu\"\n"
+                                       "VERSION=\"16.10 (Yakkety Yak)\"\n"
+                                       "ID=ubuntu\n"
+                                       "ID_LIKE=debian\n"
+                                       "PRETTY_NAME=\"Ubuntu 16.10\"\n"
+                                       "VERSION_ID=\"16.10\"\n"
+                                       "HOME_URL=\"http://www.ubuntu.com/\"\n"
+                                       "SUPPORT_URL=\"http://help.ubuntu.com/\"\n"
+                                       "BUG_REPORT_URL=\"http://bugs.launchpad.net/ubuntu/\"\n"
+                                       "PRIVACY_POLICY_URL=\"http://www.ubuntu.com/legal/terms-and-policies/privacy-policy\"\n"
+                                       "VERSION_CODENAME=yakkety\n"
+                                       "UBUNTU_CODENAME=yakkety\n"));
+
+  Distro UbuntuYakkety{UbuntuYakketyFileSystem};
+  ASSERT_EQ(Distro(Distro::UbuntuYakkety), UbuntuYakkety);
+  ASSERT_TRUE(UbuntuYakkety.IsUbuntu());
+  ASSERT_FALSE(UbuntuYakkety.IsRedhat());
+  ASSERT_FALSE(UbuntuYakkety.IsOpenSUSE());
+  ASSERT_FALSE(UbuntuYakkety.IsDebian());
+}
+
+TEST(DistroTest, DetectRedhat) {
+  vfs::InMemoryFileSystem Fedora25FileSystem;
+  Fedora25FileSystem.addFile("/etc/system-release-cpe", 0,
+      llvm::MemoryBuffer::getMemBuffer("cpe:/o:fedoraproject:fedora:25\n"));
+  // Both files are symlinks to fedora-release.
+  Fedora25FileSystem.addFile("/etc/system-release", 0,
+      llvm::MemoryBuffer::getMemBuffer("Fedora release 25 (Twenty Five)\n"));
+  Fedora25FileSystem.addFile("/etc/redhat-release", 0,
+      llvm::MemoryBuffer::getMemBuffer("Fedora release 25 (Twenty Five)\n"));
+  Fedora25FileSystem.addFile("/etc/fedora-release", 0,
+      llvm::MemoryBuffer::getMemBuffer("Fedora release 25 (Twenty Five)\n"));
+  Fedora25FileSystem.addFile("/etc/os-release", 0,
+      llvm::MemoryBuffer::getMemBuffer("NAME=Fedora\n"
+                                       "VERSION=\"25 (Twenty Five)\"\n"
+                                       "ID=fedora\n"
+                                       "VERSION_ID=25\n"
+                                       "PRETTY_NAME=\"Fedora 25 (Twenty Five)\"\n"
+                                       "ANSI_COLOR=\"0;34\"\n"
+                                       "CPE_NAME=\"cpe:/o:fedoraproject:fedora:25\"\n"
+                                       "HOME_URL=\"https://fedoraproject.org/\"\n"
+                                       "BUG_REPORT_URL=\"https://bugzilla.redhat.com/\"\n"
+                                       "REDHAT_BUGZILLA_PRODUCT=\"Fedora\"\n"
+                                       "REDHAT_BUGZILLA_PRODUCT_VERSION=25\n"
+                                       "REDHAT_SUPPORT_PRODUCT=\"Fedora\"\n"
+                                       "REDHAT_SUPPORT_PRODUCT_VERSION=25\n"
+                                       "PRIVACY_POLICY_URL=https://fedoraproject.org/wiki/Legal:PrivacyPolicy\n"));
+  Distro Fedora25{Fedora25FileSystem};
+  ASSERT_EQ(Distro(Distro::Fedora), Fedora25);
+  ASSERT_FALSE(Fedora25.IsUbuntu());
+  ASSERT_TRUE(Fedora25.IsRedhat());
+  ASSERT_FALSE(Fedora25.IsOpenSUSE());
+  ASSERT_FALSE(Fedora25.IsDebian());
+
+  vfs::InMemoryFileSystem CentOS7FileSystem;
+  CentOS7FileSystem.addFile("/etc/system-release-cpe", 0,
+      llvm::MemoryBuffer::getMemBuffer("cpe:/o:centos:centos:7\n"));
+  // Both files are symlinks to centos-release.
+  CentOS7FileSystem.addFile("/etc/system-release", 0,
+      llvm::MemoryBuffer::getMemBuffer("CentOS Linux release 7.2.1511 (Core) \n"));
+  CentOS7FileSystem.addFile("/etc/redhat-release", 0,
+      llvm::MemoryBuffer::getMemBuffer("CentOS Linux release 7.2.1511 (Core) \n"));
+  CentOS7FileSystem.addFile("/etc/centos-release", 0,
+      llvm::MemoryBuffer::getMemBuffer("CentOS Linux release 7.2.1511 (Core) \n"));
+  CentOS7FileSystem.addFile("/etc/centos-release-upstream", 0,
+      llvm::MemoryBuffer::getMemBuffer("Derived from Red Hat Enterprise Linux 7.2 (Source)\n"));
+  CentOS7FileSystem.addFile("/etc/os-release", 0,
+      llvm::MemoryBuffer::getMemBuffer("NAME=\"CentOS Linux\"\n"
+                                       "VERSION=\"7 (Core)\"\n"
+                                       "ID=\"centos\"\n"
+                                       "ID_LIKE=\"rhel fedora\"\n"
+                                       "VERSION_ID=\"7\"\n"
+                                       "PRETTY_NAME=\"CentOS Linux 7 (Core)\"\n"
+                                       "ANSI_COLOR=\"0;31\"\n"
+                                       "CPE_NAME=\"cpe:/o:centos:centos:7\"\n"
+                                       "HOME_URL=\"https://www.centos.org/\"\n"
+                                       "BUG_REPORT_URL=\"https://bugs.centos.org/\"\n"
+                                       "\n"
+                                       "CENTOS_MANTISBT_PROJECT=\"CentOS-7\"\n"
+                                       "CENTOS_MANTISBT_PROJECT_VERSION=\"7\"\n"
+                                       "REDHAT_SUPPORT_PRODUCT=\"centos\"\n"
+                                       "REDHAT_SUPPORT_PRODUCT_VERSION=\"7\"\n"));
+
+  Distro CentOS7{CentOS7FileSystem};
+  ASSERT_EQ(Distro(Distro::RHEL7), CentOS7);
+  ASSERT_FALSE(CentOS7.IsUbuntu());
+  ASSERT_TRUE(CentOS7.IsRedhat());
+  ASSERT_FALSE(CentOS7.IsOpenSUSE());
+  ASSERT_FALSE(CentOS7.IsDebian());
+}
+
+TEST(DistroTest, DetectOpenSUSE) {
+  vfs::InMemoryFileSystem OpenSUSELeap421FileSystem;
+  OpenSUSELeap421FileSystem.addFile("/etc/SuSE-release", 0,
+      llvm::MemoryBuffer::getMemBuffer("openSUSE 42.1 (x86_64)\n"
+                                       "VERSION = 42.1\n"
+                                       "CODENAME = Malachite\n"
+                                       "# /etc/SuSE-release is deprecated and will be removed in the future, use /etc/os-release instead\n"));
+  OpenSUSELeap421FileSystem.addFile("/etc/os-release", 0,
+      llvm::MemoryBuffer::getMemBuffer("NAME=\"openSUSE Leap\"\n"
+                                       "VERSION=\"42.1\"\n"
+                                       "VERSION_ID=\"42.1\"\n"
+                                       "PRETTY_NAME=\"openSUSE Leap 42.1 (x86_64)\"\n"
+                                       "ID=opensuse\n"
+                                       "ANSI_COLOR=\"0;32\"\n"
+                                       "CPE_NAME=\"cpe:/o:opensuse:opensuse:42.1\"\n"
+                                       "BUG_REPORT_URL=\"https://bugs.opensuse.org\"\n"
+                                       "HOME_URL=\"https://opensuse.org/\"\n"
+                                       "ID_LIKE=\"suse\"\n"));
+
+  Distro OpenSUSELeap421{OpenSUSELeap421FileSystem};
+  ASSERT_EQ(Distro(Distro::OpenSUSE), OpenSUSELeap421);
+  ASSERT_FALSE(OpenSUSELeap421.IsUbuntu());
+  ASSERT_FALSE(OpenSUSELeap421.IsRedhat());
+  ASSERT_TRUE(OpenSUSELeap421.IsOpenSUSE());
+  ASSERT_FALSE(OpenSUSELeap421.IsDebian());
+
+  vfs::InMemoryFileSystem OpenSUSE132FileSystem;
+  OpenSUSE132FileSystem.addFile("/etc/SuSE-release", 0,
+      llvm::MemoryBuffer::getMemBuffer("openSUSE 13.2 (x86_64)\n"
+                                       "VERSION = 13.2\n"
+                                       "CODENAME = Harlequin\n"
+                                       "# /etc/SuSE-release is deprecated and will be removed in the future, use /etc/os-release instead\n"));
+  OpenSUSE132FileSystem.addFile("/etc/os-release", 0,
+      llvm::MemoryBuffer::getMemBuffer("NAME=openSUSE\n"
+                                       "VERSION=\"13.2 (Harlequin)\"\n"
+                                       "VERSION_ID=\"13.2\"\n"
+                                       "PRETTY_NAME=\"openSUSE 13.2 (Harlequin) (x86_64)\"\n"
+                                       "ID=opensuse\n"
+                                       "ANSI_COLOR=\"0;32\"\n"
+                                       "CPE_NAME=\"cpe:/o:opensuse:opensuse:13.2\"\n"
+                                       "BUG_REPORT_URL=\"https://bugs.opensuse.org\"\n"
+                                       "HOME_URL=\"https://opensuse.org/\"\n"
+                                       "ID_LIKE=\"suse\"\n"));
+
+  Distro OpenSUSE132{OpenSUSE132FileSystem};
+  ASSERT_EQ(Distro(Distro::OpenSUSE), OpenSUSE132);
+  ASSERT_FALSE(OpenSUSE132.IsUbuntu());
+  ASSERT_FALSE(OpenSUSE132.IsRedhat());
+  ASSERT_TRUE(OpenSUSE132.IsOpenSUSE());
+  ASSERT_FALSE(OpenSUSE132.IsDebian());
+
+  vfs::InMemoryFileSystem SLES10FileSystem;
+  SLES10FileSystem.addFile("/etc/SuSE-release", 0,
+      llvm::MemoryBuffer::getMemBuffer("SUSE Linux Enterprise Server 10 (x86_64)\n"
+                                       "VERSION = 10\n"
+                                       "PATCHLEVEL = 4\n"));
+  SLES10FileSystem.addFile("/etc/lsb_release", 0,
+      llvm::MemoryBuffer::getMemBuffer("LSB_VERSION=\"core-2.0-noarch:core-3.0-noarch:core-2.0-x86_64:core-3.0-x86_64\"\n"));
+
+  // SLES10 is unsupported and therefore evaluates to unknown
+  Distro SLES10{SLES10FileSystem};
+  ASSERT_EQ(Distro(Distro::UnknownDistro), SLES10);
+  ASSERT_FALSE(SLES10.IsUbuntu());
+  ASSERT_FALSE(SLES10.IsRedhat());
+  ASSERT_FALSE(SLES10.IsOpenSUSE());
+  ASSERT_FALSE(SLES10.IsDebian());
+}
+
+TEST(DistroTest, DetectDebian) {
+  vfs::InMemoryFileSystem DebianJessieFileSystem;
+  DebianJessieFileSystem.addFile("/etc/debian_version", 0,
+                                 llvm::MemoryBuffer::getMemBuffer("8.6\n"));
+  DebianJessieFileSystem.addFile("/etc/os-release", 0,
+      llvm::MemoryBuffer::getMemBuffer("PRETTY_NAME=\"Debian GNU/Linux 8 (jessie)\"\n"
+                                       "NAME=\"Debian GNU/Linux\"\n"
+                                       "VERSION_ID=\"8\"\n"
+                                       "VERSION=\"8 (jessie)\"\n"
+                                       "ID=debian\n"
+                                       "HOME_URL=\"http://www.debian.org/\"\n"
+                                       "SUPPORT_URL=\"http://www.debian.org/support\"\n"
+                                       "BUG_REPORT_URL=\"https://bugs.debian.org/\"\n"));
+
+  Distro DebianJessie{DebianJessieFileSystem};
+  ASSERT_EQ(Distro(Distro::DebianJessie), DebianJessie);
+  ASSERT_FALSE(DebianJessie.IsUbuntu());
+  ASSERT_FALSE(DebianJessie.IsRedhat());
+  ASSERT_FALSE(DebianJessie.IsOpenSUSE());
+  ASSERT_TRUE(DebianJessie.IsDebian());
+
+  vfs::InMemoryFileSystem DebianStretchSidFileSystem;
+  DebianStretchSidFileSystem.addFile("/etc/debian_version", 0,
+                                 llvm::MemoryBuffer::getMemBuffer("stretch/sid\n"));
+  DebianStretchSidFileSystem.addFile("/etc/os-release", 0,
+      llvm::MemoryBuffer::getMemBuffer("PRETTY_NAME=\"Debian GNU/Linux stretch/sid\"\n"
+                                       "NAME=\"Debian GNU/Linux\"\n"
+                                       "ID=debian\n"
+                                       "HOME_URL=\"http://www.debian.org/\"\n"
+                                       "SUPPORT_URL=\"http://www.debian.org/support\"\n"
+                                       "BUG_REPORT_URL=\"https://bugs.debian.org/\"\n"));
+
+  Distro DebianStretchSid{DebianStretchSidFileSystem};
+  ASSERT_EQ(Distro(Distro::DebianStretch), DebianStretchSid);
+  ASSERT_FALSE(DebianStretchSid.IsUbuntu());
+  ASSERT_FALSE(DebianStretchSid.IsRedhat());
+  ASSERT_FALSE(DebianStretchSid.IsOpenSUSE());
+  ASSERT_TRUE(DebianStretchSid.IsDebian());
+}
+
+TEST(DistroTest, DetectExherbo) {
+  vfs::InMemoryFileSystem ExherboFileSystem;
+  ExherboFileSystem.addFile("/etc/exherbo-release", 0, // (ASCII art)
+                                 llvm::MemoryBuffer::getMemBuffer(""));
+  ExherboFileSystem.addFile("/etc/os-release", 0,
+      llvm::MemoryBuffer::getMemBuffer("NAME=\"Exherbo\"\n"
+                                       "PRETTY_NAME=\"Exherbo Linux\"\n"
+                                       "ID=\"exherbo\"\n"
+                                       "ANSI_COLOR=\"0;32\"\n"
+                                       "HOME_URL=\"https://www.exherbo.org/\"\n"
+                                       "SUPPORT_URL=\"irc://irc.freenode.net/#exherbo\"\n"
+                                       "BUG_REPORT_URL=\"https://bugs.exherbo.org/\"\n"));
+
+  Distro Exherbo{ExherboFileSystem};
+  ASSERT_EQ(Distro(Distro::Exherbo), Exherbo);
+  ASSERT_FALSE(Exherbo.IsUbuntu());
+  ASSERT_FALSE(Exherbo.IsRedhat());
+  ASSERT_FALSE(Exherbo.IsOpenSUSE());
+  ASSERT_FALSE(Exherbo.IsDebian());
+}
+
+TEST(DistroTest, DetectArchLinux) {
+  vfs::InMemoryFileSystem ArchLinuxFileSystem;
+  ArchLinuxFileSystem.addFile("/etc/arch-release", 0, // (empty)
+                                 llvm::MemoryBuffer::getMemBuffer(""));
+  ArchLinuxFileSystem.addFile("/etc/os-release", 0,
+      llvm::MemoryBuffer::getMemBuffer("NAME=\"Arch Linux\"\n"
+                                       "ID=arch\n"
+                                       "PRETTY_NAME=\"Arch Linux\"\n"
+                                       "ANSI_COLOR=\"0;36\"\n"
+                                       "HOME_URL=\"https://www.archlinux.org/\"\n"
+                                       "SUPPORT_URL=\"https://bbs.archlinux.org/\"\n"
+                                       "BUG_REPORT_URL=\"https://bugs.archlinux.org/\"\n"));
+
+  Distro ArchLinux{ArchLinuxFileSystem};
+  ASSERT_EQ(Distro(Distro::ArchLinux), ArchLinux);
+  ASSERT_FALSE(ArchLinux.IsUbuntu());
+  ASSERT_FALSE(ArchLinux.IsRedhat());
+  ASSERT_FALSE(ArchLinux.IsOpenSUSE());
+  ASSERT_FALSE(ArchLinux.IsDebian());
+}
+
+} // end anonymous namespace