1 //===--- RISCVToolchain.cpp - RISCV ToolChain Implementations ---*- C++ -*-===//
2 //
3 // The LLVM Compiler Infrastructure
4 //
5 // This file is distributed under the University of Illinois Open Source
6 // License. See LICENSE.TXT for details.
7 //
8 //===----------------------------------------------------------------------===//
9
10 #include "RISCVToolchain.h"
11 #include "CommonArgs.h"
12 #include "InputInfo.h"
13 #include "clang/Driver/Compilation.h"
14 #include "clang/Driver/Options.h"
15 #include "llvm/Option/ArgList.h"
16 #include "llvm/Support/FileSystem.h"
17 #include "llvm/Support/Path.h"
18 #include "llvm/Support/raw_ostream.h"
19
20 using namespace clang::driver;
21 using namespace clang::driver::toolchains;
22 using namespace clang::driver::tools;
23 using namespace clang;
24 using namespace llvm::opt;
25
26 /// RISCV Toolchain
RISCVToolChain(const Driver & D,const llvm::Triple & Triple,const ArgList & Args)27 RISCVToolChain::RISCVToolChain(const Driver &D, const llvm::Triple &Triple,
28 const ArgList &Args)
29 : Generic_ELF(D, Triple, Args) {
30 GCCInstallation.init(Triple, Args);
31 getFilePaths().push_back(computeSysRoot() + "/lib");
32 if (GCCInstallation.isValid()) {
33 getFilePaths().push_back(GCCInstallation.getInstallPath().str());
34 getProgramPaths().push_back(
35 (GCCInstallation.getParentLibPath() + "/../bin").str());
36 }
37 }
38
buildLinker() const39 Tool *RISCVToolChain::buildLinker() const {
40 return new tools::RISCV::Linker(*this);
41 }
42
addClangTargetOptions(const llvm::opt::ArgList & DriverArgs,llvm::opt::ArgStringList & CC1Args,Action::OffloadKind) const43 void RISCVToolChain::addClangTargetOptions(
44 const llvm::opt::ArgList &DriverArgs,
45 llvm::opt::ArgStringList &CC1Args,
46 Action::OffloadKind) const {
47 CC1Args.push_back("-nostdsysteminc");
48 CC1Args.push_back("-fuse-init-array");
49 }
50
AddClangSystemIncludeArgs(const ArgList & DriverArgs,ArgStringList & CC1Args) const51 void RISCVToolChain::AddClangSystemIncludeArgs(const ArgList &DriverArgs,
52 ArgStringList &CC1Args) const {
53 if (DriverArgs.hasArg(options::OPT_nostdinc))
54 return;
55
56 if (!DriverArgs.hasArg(options::OPT_nostdlibinc)) {
57 SmallString<128> Dir(computeSysRoot());
58 llvm::sys::path::append(Dir, "include");
59 addSystemInclude(DriverArgs, CC1Args, Dir.str());
60 }
61 }
62
addLibStdCxxIncludePaths(const llvm::opt::ArgList & DriverArgs,llvm::opt::ArgStringList & CC1Args) const63 void RISCVToolChain::addLibStdCxxIncludePaths(
64 const llvm::opt::ArgList &DriverArgs,
65 llvm::opt::ArgStringList &CC1Args) const {
66 const GCCVersion &Version = GCCInstallation.getVersion();
67 StringRef TripleStr = GCCInstallation.getTriple().str();
68 const Multilib &Multilib = GCCInstallation.getMultilib();
69 addLibStdCXXIncludePaths(computeSysRoot() + "/include/c++/" + Version.Text,
70 "", TripleStr, "", "", Multilib.includeSuffix(), DriverArgs, CC1Args);
71 }
72
computeSysRoot() const73 std::string RISCVToolChain::computeSysRoot() const {
74 if (!getDriver().SysRoot.empty())
75 return getDriver().SysRoot;
76
77 if (!GCCInstallation.isValid())
78 return std::string();
79
80 StringRef LibDir = GCCInstallation.getParentLibPath();
81 StringRef TripleStr = GCCInstallation.getTriple().str();
82 std::string SysRootDir = LibDir.str() + "/../" + TripleStr.str();
83
84 if (!llvm::sys::fs::exists(SysRootDir))
85 return std::string();
86
87 return SysRootDir;
88 }
89
ConstructJob(Compilation & C,const JobAction & JA,const InputInfo & Output,const InputInfoList & Inputs,const ArgList & Args,const char * LinkingOutput) const90 void RISCV::Linker::ConstructJob(Compilation &C, const JobAction &JA,
91 const InputInfo &Output,
92 const InputInfoList &Inputs,
93 const ArgList &Args,
94 const char *LinkingOutput) const {
95 const ToolChain &ToolChain = getToolChain();
96 const Driver &D = ToolChain.getDriver();
97 ArgStringList CmdArgs;
98
99 if (!D.SysRoot.empty())
100 CmdArgs.push_back(Args.MakeArgString("--sysroot=" + D.SysRoot));
101
102 std::string Linker = getToolChain().GetProgramPath(getShortName());
103
104 bool WantCRTs =
105 !Args.hasArg(options::OPT_nostdlib, options::OPT_nostartfiles);
106
107 if (WantCRTs) {
108 CmdArgs.push_back(Args.MakeArgString(ToolChain.GetFilePath("crt0.o")));
109 CmdArgs.push_back(Args.MakeArgString(ToolChain.GetFilePath("crtbegin.o")));
110 }
111
112 Args.AddAllArgs(CmdArgs, options::OPT_L);
113 ToolChain.AddFilePathLibArgs(Args, CmdArgs);
114 Args.AddAllArgs(CmdArgs,
115 {options::OPT_T_Group, options::OPT_e, options::OPT_s,
116 options::OPT_t, options::OPT_Z_Flag, options::OPT_r});
117
118 AddLinkerInputs(ToolChain, Inputs, Args, CmdArgs, JA);
119
120 // TODO: add C++ includes and libs if compiling C++.
121
122 if (!Args.hasArg(options::OPT_nostdlib) &&
123 !Args.hasArg(options::OPT_nodefaultlibs)) {
124 if (ToolChain.ShouldLinkCXXStdlib(Args))
125 ToolChain.AddCXXStdlibLibArgs(Args, CmdArgs);
126 CmdArgs.push_back("--start-group");
127 CmdArgs.push_back("-lc");
128 CmdArgs.push_back("-lgloss");
129 CmdArgs.push_back("--end-group");
130 CmdArgs.push_back("-lgcc");
131 }
132
133 if (WantCRTs)
134 CmdArgs.push_back(Args.MakeArgString(ToolChain.GetFilePath("crtend.o")));
135
136 CmdArgs.push_back("-o");
137 CmdArgs.push_back(Output.getFilename());
138 C.addCommand(llvm::make_unique<Command>(JA, *this, Args.MakeArgString(Linker),
139 CmdArgs, Inputs));
140 }
141 // RISCV tools end.
142