lib/Support/Unix/Host.inc

   1 //===- llvm/Support/Unix/Host.inc -------------------------------*- C++ -*-===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8 //
   9 // This file implements the UNIX Host support.
  10 //
  11 //===----------------------------------------------------------------------===//
  12
  13 //===----------------------------------------------------------------------===//
  14 //=== WARNING: Implementation here must contain only generic UNIX code that
  15 //===          is guaranteed to work on *all* UNIX variants.
  16 //===----------------------------------------------------------------------===//
  17
  18 #include "Unix.h"
  19 #include "llvm/ADT/StringRef.h"
  20 #include "llvm/Config/config.h"
  21 #include <cctype>
  22 #include <string>
  23 #include <sys/utsname.h>
  24
  25 using namespace llvm;
  26
  27 static std::string getOSVersion() {
  28   struct utsname info;
  29
  30   if (uname(&info))
  31     return "";
  32
  33   return info.release;
  34 }
  35
  36 static std::string updateTripleOSVersion(std::string TargetTripleString) {
  37   // On darwin, we want to update the version to match that of the target.
  38   std::string::size_type DarwinDashIdx = TargetTripleString.find("-darwin");
  39   if (DarwinDashIdx != std::string::npos) {
  40     TargetTripleString.resize(DarwinDashIdx + strlen("-darwin"));
  41     TargetTripleString += getOSVersion();
  42     return TargetTripleString;
  43   }
  44   std::string::size_type MacOSDashIdx = TargetTripleString.find("-macos");
  45   if (MacOSDashIdx != std::string::npos) {
  46     TargetTripleString.resize(MacOSDashIdx);
  47     // Reset the OS to darwin as the OS version from `uname` doesn't use the
  48     // macOS version scheme.
  49     TargetTripleString += "-darwin";
  50     TargetTripleString += getOSVersion();
  51   }
  52   return TargetTripleString;
  53 }
  54
  55 std::string sys::getDefaultTargetTriple() {
  56   std::string TargetTripleString =
  57       updateTripleOSVersion(LLVM_DEFAULT_TARGET_TRIPLE);
  58
  59   // Override the default target with an environment variable named by
  60   // LLVM_TARGET_TRIPLE_ENV.
  61 #if defined(LLVM_TARGET_TRIPLE_ENV)
  62   if (const char *EnvTriple = std::getenv(LLVM_TARGET_TRIPLE_ENV))
  63     TargetTripleString = EnvTriple;
  64 #endif
  65
  66   return TargetTripleString;
  67 }