1 //===--- PS4CPU.cpp - PS4CPU ToolChain Implementations ----------*- C++ -*-===//
3 // The LLVM Compiler Infrastructure
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
8 //===----------------------------------------------------------------------===//
12 #include "CommonArgs.h"
13 #include "clang/Driver/Compilation.h"
14 #include "clang/Driver/Driver.h"
15 #include "clang/Driver/DriverDiagnostic.h"
16 #include "clang/Driver/Options.h"
17 #include "clang/Driver/SanitizerArgs.h"
18 #include "llvm/Option/ArgList.h"
19 #include "llvm/Support/FileSystem.h"
20 #include "llvm/Support/Path.h"
21 #include <cstdlib> // ::getenv
23 using namespace clang::driver;
24 using namespace clang;
25 using namespace llvm::opt;
27 using clang::driver::tools::AddLinkerInputs;
29 void tools::PS4cpu::addProfileRTArgs(const ToolChain &TC, const ArgList &Args,
30 ArgStringList &CmdArgs) {
31 if ((Args.hasFlag(options::OPT_fprofile_arcs, options::OPT_fno_profile_arcs,
33 Args.hasFlag(options::OPT_fprofile_generate,
34 options::OPT_fno_profile_instr_generate, false) ||
35 Args.hasFlag(options::OPT_fprofile_generate_EQ,
36 options::OPT_fno_profile_instr_generate, false) ||
37 Args.hasFlag(options::OPT_fprofile_instr_generate,
38 options::OPT_fno_profile_instr_generate, false) ||
39 Args.hasFlag(options::OPT_fprofile_instr_generate_EQ,
40 options::OPT_fno_profile_instr_generate, false) ||
41 Args.hasArg(options::OPT_fcreate_profile) ||
42 Args.hasArg(options::OPT_coverage)))
43 CmdArgs.push_back("--dependent-lib=libclang_rt.profile-x86_64.a");
46 void tools::PS4cpu::Assemble::ConstructJob(Compilation &C, const JobAction &JA,
47 const InputInfo &Output,
48 const InputInfoList &Inputs,
50 const char *LinkingOutput) const {
51 claimNoWarnArgs(Args);
52 ArgStringList CmdArgs;
54 Args.AddAllArgValues(CmdArgs, options::OPT_Wa_COMMA, options::OPT_Xassembler);
56 CmdArgs.push_back("-o");
57 CmdArgs.push_back(Output.getFilename());
59 assert(Inputs.size() == 1 && "Unexpected number of inputs.");
60 const InputInfo &Input = Inputs[0];
61 assert(Input.isFilename() && "Invalid input.");
62 CmdArgs.push_back(Input.getFilename());
65 Args.MakeArgString(getToolChain().GetProgramPath("orbis-as"));
66 C.addCommand(llvm::make_unique<Command>(JA, *this, Exec, CmdArgs, Inputs));
69 static void AddPS4SanitizerArgs(const ToolChain &TC, ArgStringList &CmdArgs) {
70 const SanitizerArgs &SanArgs = TC.getSanitizerArgs();
71 if (SanArgs.needsUbsanRt()) {
72 CmdArgs.push_back("-lSceDbgUBSanitizer_stub_weak");
74 if (SanArgs.needsAsanRt()) {
75 CmdArgs.push_back("-lSceDbgAddressSanitizer_stub_weak");
79 void tools::PS4cpu::addSanitizerArgs(const ToolChain &TC,
80 ArgStringList &CmdArgs) {
81 const SanitizerArgs &SanArgs = TC.getSanitizerArgs();
82 if (SanArgs.needsUbsanRt())
83 CmdArgs.push_back("--dependent-lib=libSceDbgUBSanitizer_stub_weak.a");
84 if (SanArgs.needsAsanRt())
85 CmdArgs.push_back("--dependent-lib=libSceDbgAddressSanitizer_stub_weak.a");
88 static void ConstructPS4LinkJob(const Tool &T, Compilation &C,
89 const JobAction &JA, const InputInfo &Output,
90 const InputInfoList &Inputs,
92 const char *LinkingOutput) {
93 const toolchains::FreeBSD &ToolChain =
94 static_cast<const toolchains::FreeBSD &>(T.getToolChain());
95 const Driver &D = ToolChain.getDriver();
96 ArgStringList CmdArgs;
98 // Silence warning for "clang -g foo.o -o foo"
99 Args.ClaimAllArgs(options::OPT_g_Group);
100 // and "clang -emit-llvm foo.o -o foo"
101 Args.ClaimAllArgs(options::OPT_emit_llvm);
102 // and for "clang -w foo.o -o foo". Other warning options are already
103 // handled somewhere else.
104 Args.ClaimAllArgs(options::OPT_w);
106 if (!D.SysRoot.empty())
107 CmdArgs.push_back(Args.MakeArgString("--sysroot=" + D.SysRoot));
109 if (Args.hasArg(options::OPT_pie))
110 CmdArgs.push_back("-pie");
112 if (Args.hasArg(options::OPT_rdynamic))
113 CmdArgs.push_back("-export-dynamic");
114 if (Args.hasArg(options::OPT_shared))
115 CmdArgs.push_back("--oformat=so");
117 if (Output.isFilename()) {
118 CmdArgs.push_back("-o");
119 CmdArgs.push_back(Output.getFilename());
121 assert(Output.isNothing() && "Invalid output.");
124 AddPS4SanitizerArgs(ToolChain, CmdArgs);
126 Args.AddAllArgs(CmdArgs, options::OPT_L);
127 Args.AddAllArgs(CmdArgs, options::OPT_T_Group);
128 Args.AddAllArgs(CmdArgs, options::OPT_e);
129 Args.AddAllArgs(CmdArgs, options::OPT_s);
130 Args.AddAllArgs(CmdArgs, options::OPT_t);
131 Args.AddAllArgs(CmdArgs, options::OPT_r);
133 if (Args.hasArg(options::OPT_Z_Xlinker__no_demangle))
134 CmdArgs.push_back("--no-demangle");
136 AddLinkerInputs(ToolChain, Inputs, Args, CmdArgs, JA);
138 if (Args.hasArg(options::OPT_pthread)) {
139 CmdArgs.push_back("-lpthread");
142 const char *Exec = Args.MakeArgString(ToolChain.GetProgramPath("orbis-ld"));
144 C.addCommand(llvm::make_unique<Command>(JA, T, Exec, CmdArgs, Inputs));
147 static void ConstructGoldLinkJob(const Tool &T, Compilation &C,
148 const JobAction &JA, const InputInfo &Output,
149 const InputInfoList &Inputs,
151 const char *LinkingOutput) {
152 const toolchains::FreeBSD &ToolChain =
153 static_cast<const toolchains::FreeBSD &>(T.getToolChain());
154 const Driver &D = ToolChain.getDriver();
155 ArgStringList CmdArgs;
157 // Silence warning for "clang -g foo.o -o foo"
158 Args.ClaimAllArgs(options::OPT_g_Group);
159 // and "clang -emit-llvm foo.o -o foo"
160 Args.ClaimAllArgs(options::OPT_emit_llvm);
161 // and for "clang -w foo.o -o foo". Other warning options are already
162 // handled somewhere else.
163 Args.ClaimAllArgs(options::OPT_w);
165 if (!D.SysRoot.empty())
166 CmdArgs.push_back(Args.MakeArgString("--sysroot=" + D.SysRoot));
168 if (Args.hasArg(options::OPT_pie))
169 CmdArgs.push_back("-pie");
171 if (Args.hasArg(options::OPT_static)) {
172 CmdArgs.push_back("-Bstatic");
174 if (Args.hasArg(options::OPT_rdynamic))
175 CmdArgs.push_back("-export-dynamic");
176 CmdArgs.push_back("--eh-frame-hdr");
177 if (Args.hasArg(options::OPT_shared)) {
178 CmdArgs.push_back("-Bshareable");
180 CmdArgs.push_back("-dynamic-linker");
181 CmdArgs.push_back("/libexec/ld-elf.so.1");
183 CmdArgs.push_back("--enable-new-dtags");
186 if (Output.isFilename()) {
187 CmdArgs.push_back("-o");
188 CmdArgs.push_back(Output.getFilename());
190 assert(Output.isNothing() && "Invalid output.");
193 AddPS4SanitizerArgs(ToolChain, CmdArgs);
195 if (!Args.hasArg(options::OPT_nostdlib, options::OPT_nostartfiles)) {
196 const char *crt1 = nullptr;
197 if (!Args.hasArg(options::OPT_shared)) {
198 if (Args.hasArg(options::OPT_pg))
200 else if (Args.hasArg(options::OPT_pie))
206 CmdArgs.push_back(Args.MakeArgString(ToolChain.GetFilePath(crt1)));
208 CmdArgs.push_back(Args.MakeArgString(ToolChain.GetFilePath("crti.o")));
210 const char *crtbegin = nullptr;
211 if (Args.hasArg(options::OPT_static))
212 crtbegin = "crtbeginT.o";
213 else if (Args.hasArg(options::OPT_shared) || Args.hasArg(options::OPT_pie))
214 crtbegin = "crtbeginS.o";
216 crtbegin = "crtbegin.o";
218 CmdArgs.push_back(Args.MakeArgString(ToolChain.GetFilePath(crtbegin)));
221 Args.AddAllArgs(CmdArgs, options::OPT_L);
222 ToolChain.AddFilePathLibArgs(Args, CmdArgs);
223 Args.AddAllArgs(CmdArgs, options::OPT_T_Group);
224 Args.AddAllArgs(CmdArgs, options::OPT_e);
225 Args.AddAllArgs(CmdArgs, options::OPT_s);
226 Args.AddAllArgs(CmdArgs, options::OPT_t);
227 Args.AddAllArgs(CmdArgs, options::OPT_r);
229 if (Args.hasArg(options::OPT_Z_Xlinker__no_demangle))
230 CmdArgs.push_back("--no-demangle");
232 AddLinkerInputs(ToolChain, Inputs, Args, CmdArgs, JA);
234 if (!Args.hasArg(options::OPT_nostdlib, options::OPT_nodefaultlibs)) {
235 // For PS4, we always want to pass libm, libstdc++ and libkernel
236 // libraries for both C and C++ compilations.
237 CmdArgs.push_back("-lkernel");
239 if (ToolChain.ShouldLinkCXXStdlib(Args))
240 ToolChain.AddCXXStdlibLibArgs(Args, CmdArgs);
241 if (Args.hasArg(options::OPT_pg))
242 CmdArgs.push_back("-lm_p");
244 CmdArgs.push_back("-lm");
246 // FIXME: For some reason GCC passes -lgcc and -lgcc_s before adding
247 // the default system libraries. Just mimic this for now.
248 if (Args.hasArg(options::OPT_pg))
249 CmdArgs.push_back("-lgcc_p");
251 CmdArgs.push_back("-lcompiler_rt");
252 if (Args.hasArg(options::OPT_static)) {
253 CmdArgs.push_back("-lstdc++");
254 } else if (Args.hasArg(options::OPT_pg)) {
255 CmdArgs.push_back("-lgcc_eh_p");
257 CmdArgs.push_back("--as-needed");
258 CmdArgs.push_back("-lstdc++");
259 CmdArgs.push_back("--no-as-needed");
262 if (Args.hasArg(options::OPT_pthread)) {
263 if (Args.hasArg(options::OPT_pg))
264 CmdArgs.push_back("-lpthread_p");
266 CmdArgs.push_back("-lpthread");
269 if (Args.hasArg(options::OPT_pg)) {
270 if (Args.hasArg(options::OPT_shared))
271 CmdArgs.push_back("-lc");
273 if (Args.hasArg(options::OPT_static)) {
274 CmdArgs.push_back("--start-group");
275 CmdArgs.push_back("-lc_p");
276 CmdArgs.push_back("-lpthread_p");
277 CmdArgs.push_back("--end-group");
279 CmdArgs.push_back("-lc_p");
282 CmdArgs.push_back("-lgcc_p");
284 if (Args.hasArg(options::OPT_static)) {
285 CmdArgs.push_back("--start-group");
286 CmdArgs.push_back("-lc");
287 CmdArgs.push_back("-lpthread");
288 CmdArgs.push_back("--end-group");
290 CmdArgs.push_back("-lc");
292 CmdArgs.push_back("-lcompiler_rt");
295 if (Args.hasArg(options::OPT_static)) {
296 CmdArgs.push_back("-lstdc++");
297 } else if (Args.hasArg(options::OPT_pg)) {
298 CmdArgs.push_back("-lgcc_eh_p");
300 CmdArgs.push_back("--as-needed");
301 CmdArgs.push_back("-lstdc++");
302 CmdArgs.push_back("--no-as-needed");
306 if (!Args.hasArg(options::OPT_nostdlib, options::OPT_nostartfiles)) {
307 if (Args.hasArg(options::OPT_shared) || Args.hasArg(options::OPT_pie))
308 CmdArgs.push_back(Args.MakeArgString(ToolChain.GetFilePath("crtendS.o")));
310 CmdArgs.push_back(Args.MakeArgString(ToolChain.GetFilePath("crtend.o")));
311 CmdArgs.push_back(Args.MakeArgString(ToolChain.GetFilePath("crtn.o")));
316 Args.MakeArgString(ToolChain.GetProgramPath("orbis-ld.gold"));
318 Args.MakeArgString(ToolChain.GetProgramPath("orbis-ld"));
321 C.addCommand(llvm::make_unique<Command>(JA, T, Exec, CmdArgs, Inputs));
324 void tools::PS4cpu::Link::ConstructJob(Compilation &C, const JobAction &JA,
325 const InputInfo &Output,
326 const InputInfoList &Inputs,
328 const char *LinkingOutput) const {
329 const toolchains::FreeBSD &ToolChain =
330 static_cast<const toolchains::FreeBSD &>(getToolChain());
331 const Driver &D = ToolChain.getDriver();
333 StringRef LinkerOptName;
334 if (const Arg *A = Args.getLastArg(options::OPT_fuse_ld_EQ)) {
335 LinkerOptName = A->getValue();
336 if (LinkerOptName != "ps4" && LinkerOptName != "gold")
337 D.Diag(diag::err_drv_unsupported_linker) << LinkerOptName;
340 if (LinkerOptName == "gold")
342 else if (LinkerOptName == "ps4")
345 PS4Linker = !Args.hasArg(options::OPT_shared);
348 ConstructPS4LinkJob(*this, C, JA, Output, Inputs, Args, LinkingOutput);
350 ConstructGoldLinkJob(*this, C, JA, Output, Inputs, Args, LinkingOutput);
353 toolchains::PS4CPU::PS4CPU(const Driver &D, const llvm::Triple &Triple,
355 : Generic_ELF(D, Triple, Args) {
356 if (Args.hasArg(clang::driver::options::OPT_static))
357 D.Diag(clang::diag::err_drv_unsupported_opt_for_target) << "-static"
360 // Determine where to find the PS4 libraries. We use SCE_ORBIS_SDK_DIR
361 // if it exists; otherwise use the driver's installation path, which
362 // should be <SDK_DIR>/host_tools/bin.
364 SmallString<512> PS4SDKDir;
365 if (const char *EnvValue = getenv("SCE_ORBIS_SDK_DIR")) {
366 if (!llvm::sys::fs::exists(EnvValue))
367 getDriver().Diag(clang::diag::warn_drv_ps4_sdk_dir) << EnvValue;
368 PS4SDKDir = EnvValue;
370 PS4SDKDir = getDriver().Dir;
371 llvm::sys::path::append(PS4SDKDir, "/../../");
374 // By default, the driver won't report a warning if it can't find
375 // PS4's include or lib directories. This behavior could be changed if
376 // -Weverything or -Winvalid-or-nonexistent-directory options are passed.
377 // If -isysroot was passed, use that as the SDK base path.
378 std::string PrefixDir;
379 if (const Arg *A = Args.getLastArg(options::OPT_isysroot)) {
380 PrefixDir = A->getValue();
381 if (!llvm::sys::fs::exists(PrefixDir))
382 getDriver().Diag(clang::diag::warn_missing_sysroot) << PrefixDir;
384 PrefixDir = PS4SDKDir.str();
386 SmallString<512> PS4SDKIncludeDir(PrefixDir);
387 llvm::sys::path::append(PS4SDKIncludeDir, "target/include");
388 if (!Args.hasArg(options::OPT_nostdinc) &&
389 !Args.hasArg(options::OPT_nostdlibinc) &&
390 !Args.hasArg(options::OPT_isysroot) &&
391 !Args.hasArg(options::OPT__sysroot_EQ) &&
392 !llvm::sys::fs::exists(PS4SDKIncludeDir)) {
393 getDriver().Diag(clang::diag::warn_drv_unable_to_find_directory_expected)
394 << "PS4 system headers" << PS4SDKIncludeDir;
397 SmallString<512> PS4SDKLibDir(PS4SDKDir);
398 llvm::sys::path::append(PS4SDKLibDir, "target/lib");
399 if (!Args.hasArg(options::OPT_nostdlib) &&
400 !Args.hasArg(options::OPT_nodefaultlibs) &&
401 !Args.hasArg(options::OPT__sysroot_EQ) && !Args.hasArg(options::OPT_E) &&
402 !Args.hasArg(options::OPT_c) && !Args.hasArg(options::OPT_S) &&
403 !Args.hasArg(options::OPT_emit_ast) &&
404 !llvm::sys::fs::exists(PS4SDKLibDir)) {
405 getDriver().Diag(clang::diag::warn_drv_unable_to_find_directory_expected)
406 << "PS4 system libraries" << PS4SDKLibDir;
409 getFilePaths().push_back(PS4SDKLibDir.str());
412 Tool *toolchains::PS4CPU::buildAssembler() const {
413 return new tools::PS4cpu::Assemble(*this);
416 Tool *toolchains::PS4CPU::buildLinker() const {
417 return new tools::PS4cpu::Link(*this);
420 bool toolchains::PS4CPU::isPICDefault() const { return true; }
422 bool toolchains::PS4CPU::HasNativeLLVMSupport() const { return true; }
424 SanitizerMask toolchains::PS4CPU::getSupportedSanitizers() const {
425 SanitizerMask Res = ToolChain::getSupportedSanitizers();
426 Res |= SanitizerKind::Address;
427 Res |= SanitizerKind::Vptr;