14 #include "llvm/ADT/StringExtras.h"
15 #include "llvm/ADT/StringMap.h"
16 #include "llvm/ADT/StringSwitch.h"
17 #include "llvm/Option/ArgList.h"
18 #include "llvm/Support/Host.h"
22 using namespace clang;
26 const llvm::Triple &Triple) {
27 if (
const Arg *A = Args.getLastArg(clang::driver::options::OPT_march_EQ)) {
28 StringRef CPU = A->getValue();
37 CPU = llvm::sys::getHostCPUName();
38 if (!CPU.empty() && CPU !=
"generic")
42 if (
const Arg *A = Args.getLastArg(options::OPT__SLASH_arch)) {
46 llvm::StringMap<StringRef> ArchMap({
47 {
"AVX",
"sandybridge"},
50 {
"AVX512",
"skylake-avx512"},
52 if (Triple.getArch() == llvm::Triple::x86) {
60 StringRef CPU = ArchMap.lookup(A->getValue());
62 std::vector<StringRef> ValidArchs{ArchMap.keys().begin(),
63 ArchMap.keys().end()};
65 D.
Diag(diag::warn_drv_invalid_arch_name_with_suggestion)
66 << A->getValue() << (Triple.getArch() == llvm::Triple::x86)
67 << join(ValidArchs,
", ");
77 bool Is64Bit = Triple.getArch() == llvm::Triple::x86_64;
80 if (Triple.isOSDarwin()) {
81 if (Triple.getArchName() ==
"x86_64h")
85 if (Triple.isMacOSX() && !Triple.isOSVersionLT(10, 12))
88 if (Triple.isDriverKit())
92 return Is64Bit ?
"core2" :
"yonah";
102 if (Triple.isAndroid())
103 return Is64Bit ?
"x86-64" :
"i686";
109 switch (Triple.getOS()) {
110 case llvm::Triple::NetBSD:
112 case llvm::Triple::Haiku:
113 case llvm::Triple::OpenBSD:
115 case llvm::Triple::FreeBSD:
125 std::vector<StringRef> &Features) {
127 if (
const Arg *A = Args.getLastArg(clang::driver::options::OPT_march_EQ)) {
128 if (StringRef(A->getValue()) ==
"native") {
129 llvm::StringMap<bool> HostFeatures;
130 if (llvm::sys::getHostCPUFeatures(HostFeatures))
131 for (
auto &F : HostFeatures)
133 Args.MakeArgString((F.second ?
"+" :
"-") + F.first()));
137 if (Triple.getArchName() ==
"x86_64h") {
140 Features.push_back(
"-rdrnd");
141 Features.push_back(
"-aes");
142 Features.push_back(
"-pclmul");
143 Features.push_back(
"-rtm");
144 Features.push_back(
"-fsgsbase");
147 const llvm::Triple::ArchType ArchType = Triple.getArch();
149 if (Triple.isAndroid()) {
150 if (ArchType == llvm::Triple::x86_64) {
151 Features.push_back(
"+sse4.2");
152 Features.push_back(
"+popcnt");
153 Features.push_back(
"+cx16");
155 Features.push_back(
"+ssse3");
165 if (Args.hasArgNoClaim(options::OPT_mretpoline, options::OPT_mno_retpoline,
166 options::OPT_mspeculative_load_hardening,
167 options::OPT_mno_speculative_load_hardening)) {
168 if (Args.hasFlag(options::OPT_mretpoline, options::OPT_mno_retpoline,
170 Features.push_back(
"+retpoline-indirect-calls");
171 Features.push_back(
"+retpoline-indirect-branches");
172 SpectreOpt = options::OPT_mretpoline;
173 }
else if (Args.hasFlag(options::OPT_mspeculative_load_hardening,
174 options::OPT_mno_speculative_load_hardening,
178 Features.push_back(
"+retpoline-indirect-calls");
179 SpectreOpt = options::OPT_mspeculative_load_hardening;
181 }
else if (Args.hasFlag(options::OPT_mretpoline_external_thunk,
182 options::OPT_mno_retpoline_external_thunk,
false)) {
185 Features.push_back(
"+retpoline-indirect-calls");
186 Features.push_back(
"+retpoline-indirect-branches");
187 SpectreOpt = options::OPT_mretpoline_external_thunk;
191 if (Args.hasFlag(options::OPT_mlvi_hardening, options::OPT_mno_lvi_hardening,
193 Features.push_back(
"+lvi-load-hardening");
194 Features.push_back(
"+lvi-cfi");
195 LVIOpt = options::OPT_mlvi_hardening;
196 }
else if (Args.hasFlag(options::OPT_mlvi_cfi, options::OPT_mno_lvi_cfi,
198 Features.push_back(
"+lvi-cfi");
199 LVIOpt = options::OPT_mlvi_cfi;
202 if (Args.hasFlag(options::OPT_m_seses, options::OPT_mno_seses,
false)) {
203 if (LVIOpt == options::OPT_mlvi_hardening)
204 D.
Diag(diag::err_drv_argument_not_allowed_with)
205 << D.
getOpts().getOptionName(options::OPT_mlvi_hardening)
206 << D.
getOpts().getOptionName(options::OPT_m_seses);
209 D.
Diag(diag::err_drv_argument_not_allowed_with)
210 << D.
getOpts().getOptionName(SpectreOpt)
211 << D.
getOpts().getOptionName(options::OPT_m_seses);
213 Features.push_back(
"+seses");
214 if (!Args.hasArg(options::OPT_mno_lvi_cfi)) {
215 Features.push_back(
"+lvi-cfi");
216 LVIOpt = options::OPT_mlvi_cfi;
222 D.
Diag(diag::err_drv_argument_not_allowed_with)
223 << D.
getOpts().getOptionName(SpectreOpt)
224 << D.
getOpts().getOptionName(LVIOpt);
229 for (
const Arg *A : Args.filtered(options::OPT_m_x86_Features_Group,
230 options::OPT_mgeneral_regs_only)) {
231 StringRef Name = A->getOption().getName();
235 assert(Name.startswith(
"m") &&
"Invalid feature name.");
236 Name = Name.substr(1);
239 if (A->getOption().getID() == options::OPT_mgeneral_regs_only) {
240 Features.insert(Features.end(), {
"-x87",
"-mmx",
"-sse"});
244 bool IsNegative = Name.startswith(
"no-");
246 Name = Name.substr(3);
247 Features.push_back(Args.MakeArgString((IsNegative ?
"-" :
"+") + Name));
251 if (Arg *A = Args.getLastArg(options::OPT_mharden_sls_EQ)) {
252 StringRef
Scope = A->getValue();
253 if (
Scope ==
"all") {
254 Features.push_back(
"+harden-sls-ijmp");
255 Features.push_back(
"+harden-sls-ret");
256 }
else if (
Scope ==
"return") {
257 Features.push_back(
"+harden-sls-ret");
258 }
else if (
Scope ==
"indirect-jmp") {
259 Features.push_back(
"+harden-sls-ijmp");
260 }
else if (
Scope !=
"none") {
261 D.
Diag(diag::err_drv_unsupported_option_argument)
262 << A->getOption().getName() <<
Scope;