plugins/msvc/msvclinker.cpp

   1 #include <msp/builder/builder.h>
   2 #include <msp/builder/component.h>
   3 #include <msp/builder/executable.h>
   4 #include <msp/builder/importlibrary.h>
   5 #include <msp/builder/installedfile.h>
   6 #include <msp/builder/objectfile.h>
   7 #include <msp/builder/sharedlibrary.h>
   8 #include <msp/builder/sourcepackage.h>
   9 #include <msp/builder/staticlibrary.h>
  10 #include <msp/core/environ.h>
  11 #include <msp/fs/utils.h>
  12 #include <msp/strings/utils.h>
  13 #include "microsofttools.h"
  14 #include "msvclinker.h"
  15
  16 using namespace std;
  17 using namespace Msp;
  18
  19 MsvcLinker::MsvcLinker(Builder &b, const Architecture &a, const MicrosoftTools &m):
  20         Tool(b, &a, "LINK"),
  21         ms_tools(m)
  22 {
  23         input_suffixes.push_back(".obj");
  24         input_suffixes.push_back(".lib");
  25
  26         processing_unit = COMPONENT;
  27
  28         set_command((ms_tools.get_vc_bin_dir()/"link.exe").str(), false);
  29         set_run_external(_run);
  30 }
  31
  32 Target *MsvcLinker::create_target(const vector<Target *> &sources, const string &arg)
  33 {
  34         if(sources.empty())
  35                 throw invalid_argument("MsvcLinker::create_target");
  36
  37         if(arg=="import")
  38         {
  39                 if(sources.size()!=1)
  40                         throw invalid_argument("MsvcLinker::create_target");
  41                 SharedLibrary &shlib = dynamic_cast<SharedLibrary &>(*sources.front());
  42
  43                 ImportLibrary *imp = new ImportLibrary(builder, *shlib.get_component(), shlib);
  44                 for(ObjectFile *o: shlib.get_objects())
  45                         imp->add_dependency(*o);
  46                 shlib.add_side_effect(*imp);
  47
  48                 return imp;
  49         }
  50
  51         vector<ObjectFile *> objs;
  52         objs.reserve(sources.size());
  53         for(Target *s: sources)
  54                 objs.push_back(&dynamic_cast<ObjectFile &>(*s));
  55
  56         const Component &comp = *objs.front()->get_component();
  57         Binary *bin = 0;
  58         if(arg=="shared")
  59                 bin = new SharedLibrary(builder, comp, objs);
  60         else
  61                 bin = new Executable(builder, comp, objs);
  62         bin->set_tool(*this);
  63         return bin;
  64 }
  65
  66 string MsvcLinker::create_build_signature(const BuildInfo &binfo) const
  67 {
  68         string result = Tool::create_build_signature(binfo);
  69         result += ',';
  70         if(binfo.strip)
  71                 result += 's';
  72         if(!binfo.libs.empty())
  73         {
  74                 result += ",l";
  75                 result += join(binfo.libs.begin(), binfo.libs.end(), ",l");
  76         }
  77         return result;
  78 }
  79
  80 void MsvcLinker::do_prepare(ToolData &tool) const
  81 {
  82         const std::string &tool_tag = static_cast<Tool &>(tool).get_tag();
  83         const Architecture &arch = *static_cast<Tool &>(tool).get_architecture();
  84         string arch_dir = (arch.get_bits()==64 ? "x64" : "x86");
  85
  86         const FS::Path &vc_base_dir = ms_tools.get_vc_base_dir();
  87         tool.system_path.push_back(vc_base_dir/"lib"/arch_dir);
  88
  89         const FS::Path &win_sdk_dir = ms_tools.get_windows_sdk_dir();
  90         const string &win_sdk_ver = ms_tools.get_windows_sdk_version();
  91         tool.system_path.push_back(win_sdk_dir/"lib"/win_sdk_ver/"ucrt"/arch_dir);
  92         tool.system_path.push_back(win_sdk_dir/"lib"/win_sdk_ver/"um"/arch_dir);
  93
  94         string path;
  95         for(const FS::Path &p: tool.system_path)
  96         {
  97                 append(path, ";", p.str());
  98                 builder.get_logger().log("tools", "Got %s system path: %s", tool_tag, p);
  99         }
 100
 101         setenv("LIB", path);
 102 }
 103
 104 ExternalTask::Arguments MsvcLinker::_run(const Binary &bin, FS::Path &work_dir)
 105 {
 106         const Tool &tool = *bin.get_tool();
 107
 108         vector<string> argv;
 109         argv.push_back(tool.get_executable()->get_path().str());
 110         argv.push_back("/NOLOGO");
 111
 112         string imp_fn;
 113         if(const SharedLibrary *shlib = dynamic_cast<const SharedLibrary *>(&bin))
 114         {
 115                 argv.push_back("/DLL");
 116                 if(const ImportLibrary *imp = shlib->get_import_library())
 117                         imp_fn = relative(imp->get_path(), work_dir).str();
 118         }
 119
 120         BuildInfo binfo;
 121         bin.collect_build_info(binfo);
 122
 123         /*for(const FS::Path &p: binfo.libpath)
 124                 argv.push_back("/LIBPATH:"+p.str());*/
 125         if(binfo.strip)
 126                 argv.push_back("/INCREMENTAL:NO");
 127         else
 128         {
 129                 argv.push_back("/DEBUG:FULL");
 130                 FS::Path temp_dir = bin.get_package()->get_temp_directory()/bin.get_component()->get_name();
 131                 FS::Path ilk_path = temp_dir/(FS::basepart(FS::basename(bin.get_path()))+".ilk");
 132                 argv.push_back("/ILK:"+relative(ilk_path, work_dir).str());
 133         }
 134
 135         argv.push_back("/OUT:"+relative(bin.get_path(), work_dir).str());
 136         if(!imp_fn.empty())
 137                 argv.push_back("/IMPLIB:"+imp_fn);
 138         else
 139                 argv.push_back("/NOIMPLIB");
 140
 141         argv.push_back("/NOEXP");
 142
 143         for(Target *d: bin.get_dependencies())
 144         {
 145                 FileTarget *file = dynamic_cast<FileTarget *>(d);
 146                 Target *tgt = d->get_real_target();
 147
 148                 if(ObjectFile *obj = dynamic_cast<ObjectFile *>(tgt))
 149                         argv.push_back(relative(obj->get_path(), work_dir).str());
 150                 else if(StaticLibrary *stlib = dynamic_cast<StaticLibrary *>(tgt))
 151                         argv.push_back((file?file:stlib)->get_path().str());
 152                 else if(ImportLibrary *imp = dynamic_cast<ImportLibrary *>(tgt))
 153                         argv.push_back((file?file:imp)->get_path().str());
 154         }
 155
 156         argv.push_back("/SUBSYSTEM:CONSOLE");
 157
 158         return argv;
 159 }