]> git.lizzy.rs Git - rust.git/blob - src/librustc/back/link.rs
auto merge of #15999 : Kimundi/rust/fix_folder, r=nikomatsakis
[rust.git] / src / librustc / back / link.rs
1 // Copyright 2012-2014 The Rust Project Developers. See the COPYRIGHT
2 // file at the top-level directory of this distribution and at
3 // http://rust-lang.org/COPYRIGHT.
4 //
5 // Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
6 // http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
7 // <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
8 // option. This file may not be copied, modified, or distributed
9 // except according to those terms.
10
11 use super::archive::{Archive, ArchiveBuilder, ArchiveConfig, METADATA_FILENAME};
12 use super::rpath;
13 use super::rpath::RPathConfig;
14 use super::svh::Svh;
15 use driver::driver::{CrateTranslation, OutputFilenames, Input, FileInput};
16 use driver::config::NoDebugInfo;
17 use driver::session::Session;
18 use driver::config;
19 use llvm;
20 use llvm::ModuleRef;
21 use metadata::common::LinkMeta;
22 use metadata::{encoder, cstore, filesearch, csearch, loader, creader};
23 use middle::trans::context::CrateContext;
24 use middle::trans::common::gensym_name;
25 use middle::ty;
26 use util::common::time;
27 use util::ppaux;
28 use util::sha2::{Digest, Sha256};
29
30 use std::c_str::{ToCStr, CString};
31 use std::char;
32 use std::collections::HashSet;
33 use std::io::{fs, TempDir, Command};
34 use std::io;
35 use std::ptr;
36 use std::str;
37 use std::string::String;
38 use flate;
39 use serialize::hex::ToHex;
40 use syntax::abi;
41 use syntax::ast;
42 use syntax::ast_map::{PathElem, PathElems, PathName};
43 use syntax::ast_map;
44 use syntax::attr::AttrMetaMethods;
45 use syntax::codemap::Span;
46 use syntax::parse::token;
47
48 #[deriving(Clone, PartialEq, PartialOrd, Ord, Eq)]
49 pub enum OutputType {
50     OutputTypeBitcode,
51     OutputTypeAssembly,
52     OutputTypeLlvmAssembly,
53     OutputTypeObject,
54     OutputTypeExe,
55 }
56
57 pub fn llvm_err(sess: &Session, msg: String) -> ! {
58     unsafe {
59         let cstr = llvm::LLVMRustGetLastError();
60         if cstr == ptr::null() {
61             sess.fatal(msg.as_slice());
62         } else {
63             let err = CString::new(cstr, true);
64             let err = String::from_utf8_lossy(err.as_bytes());
65             sess.fatal(format!("{}: {}",
66                                msg.as_slice(),
67                                err.as_slice()).as_slice());
68         }
69     }
70 }
71
72 pub fn write_output_file(
73         sess: &Session,
74         target: llvm::TargetMachineRef,
75         pm: llvm::PassManagerRef,
76         m: ModuleRef,
77         output: &Path,
78         file_type: llvm::FileType) {
79     unsafe {
80         output.with_c_str(|output| {
81             let result = llvm::LLVMRustWriteOutputFile(
82                     target, pm, m, output, file_type);
83             if !result {
84                 llvm_err(sess, "could not write output".to_string());
85             }
86         })
87     }
88 }
89
90 pub mod write {
91
92     use super::super::lto;
93     use super::{write_output_file, OutputType};
94     use super::{OutputTypeAssembly, OutputTypeBitcode};
95     use super::{OutputTypeExe, OutputTypeLlvmAssembly};
96     use super::{OutputTypeObject};
97     use driver::driver::{CrateTranslation, OutputFilenames};
98     use driver::config::NoDebugInfo;
99     use driver::session::Session;
100     use driver::config;
101     use llvm;
102     use llvm::{ModuleRef, TargetMachineRef, PassManagerRef};
103     use util::common::time;
104     use syntax::abi;
105
106     use std::c_str::ToCStr;
107     use std::io::{Command};
108     use libc::{c_uint, c_int};
109     use std::str;
110
111     // On android, we by default compile for armv7 processors. This enables
112     // things like double word CAS instructions (rather than emulating them)
113     // which are *far* more efficient. This is obviously undesirable in some
114     // cases, so if any sort of target feature is specified we don't append v7
115     // to the feature list.
116     //
117     // On iOS only armv7 and newer are supported. So it is useful to
118     // get all hardware potential via VFP3 (hardware floating point)
119     // and NEON (SIMD) instructions supported by LLVM.
120     // Note that without those flags various linking errors might
121     // arise as some of intrinsics are converted into function calls
122     // and nobody provides implementations those functions
123     fn target_feature<'a>(sess: &'a Session) -> &'a str {
124         match sess.targ_cfg.os {
125             abi::OsAndroid => {
126                 if "" == sess.opts.cg.target_feature.as_slice() {
127                     "+v7"
128                 } else {
129                     sess.opts.cg.target_feature.as_slice()
130                 }
131             },
132             abi::OsiOS if sess.targ_cfg.arch == abi::Arm => {
133                 "+v7,+thumb2,+vfp3,+neon"
134             },
135             _ => sess.opts.cg.target_feature.as_slice()
136         }
137     }
138
139     pub fn run_passes(sess: &Session,
140                       trans: &CrateTranslation,
141                       output_types: &[OutputType],
142                       output: &OutputFilenames) {
143         let llmod = trans.module;
144         let llcx = trans.context;
145         unsafe {
146             configure_llvm(sess);
147
148             if sess.opts.cg.save_temps {
149                 output.with_extension("no-opt.bc").with_c_str(|buf| {
150                     llvm::LLVMWriteBitcodeToFile(llmod, buf);
151                 })
152             }
153
154             let opt_level = match sess.opts.optimize {
155               config::No => llvm::CodeGenLevelNone,
156               config::Less => llvm::CodeGenLevelLess,
157               config::Default => llvm::CodeGenLevelDefault,
158               config::Aggressive => llvm::CodeGenLevelAggressive,
159             };
160             let use_softfp = sess.opts.cg.soft_float;
161
162             // FIXME: #11906: Omitting frame pointers breaks retrieving the value of a parameter.
163             // FIXME: #11954: mac64 unwinding may not work with fp elim
164             let no_fp_elim = (sess.opts.debuginfo != NoDebugInfo) ||
165                              (sess.targ_cfg.os == abi::OsMacos &&
166                               sess.targ_cfg.arch == abi::X86_64);
167
168             // OSX has -dead_strip, which doesn't rely on ffunction_sections
169             // FIXME(#13846) this should be enabled for windows
170             let ffunction_sections = sess.targ_cfg.os != abi::OsMacos &&
171                                      sess.targ_cfg.os != abi::OsWin32;
172             let fdata_sections = ffunction_sections;
173
174             let reloc_model = match sess.opts.cg.relocation_model.as_slice() {
175                 "pic" => llvm::RelocPIC,
176                 "static" => llvm::RelocStatic,
177                 "default" => llvm::RelocDefault,
178                 "dynamic-no-pic" => llvm::RelocDynamicNoPic,
179                 _ => {
180                     sess.err(format!("{} is not a valid relocation mode",
181                                      sess.opts
182                                          .cg
183                                          .relocation_model).as_slice());
184                     sess.abort_if_errors();
185                     return;
186                 }
187             };
188
189             let code_model = match sess.opts.cg.code_model.as_slice() {
190                 "default" => llvm::CodeModelDefault,
191                 "small" => llvm::CodeModelSmall,
192                 "kernel" => llvm::CodeModelKernel,
193                 "medium" => llvm::CodeModelMedium,
194                 "large" => llvm::CodeModelLarge,
195                 _ => {
196                     sess.err(format!("{} is not a valid code model",
197                                      sess.opts
198                                          .cg
199                                          .code_model).as_slice());
200                     sess.abort_if_errors();
201                     return;
202                 }
203             };
204
205             let tm = sess.targ_cfg
206                          .target_strs
207                          .target_triple
208                          .as_slice()
209                          .with_c_str(|t| {
210                 sess.opts.cg.target_cpu.as_slice().with_c_str(|cpu| {
211                     target_feature(sess).with_c_str(|features| {
212                         llvm::LLVMRustCreateTargetMachine(
213                             t, cpu, features,
214                             code_model,
215                             reloc_model,
216                             opt_level,
217                             true /* EnableSegstk */,
218                             use_softfp,
219                             no_fp_elim,
220                             ffunction_sections,
221                             fdata_sections,
222                         )
223                     })
224                 })
225             });
226
227             // Create the two optimizing pass managers. These mirror what clang
228             // does, and are by populated by LLVM's default PassManagerBuilder.
229             // Each manager has a different set of passes, but they also share
230             // some common passes.
231             let fpm = llvm::LLVMCreateFunctionPassManagerForModule(llmod);
232             let mpm = llvm::LLVMCreatePassManager();
233
234             // If we're verifying or linting, add them to the function pass
235             // manager.
236             let addpass = |pass: &str| {
237                 pass.as_slice().with_c_str(|s| llvm::LLVMRustAddPass(fpm, s))
238             };
239             if !sess.no_verify() { assert!(addpass("verify")); }
240
241             if !sess.opts.cg.no_prepopulate_passes {
242                 llvm::LLVMRustAddAnalysisPasses(tm, fpm, llmod);
243                 llvm::LLVMRustAddAnalysisPasses(tm, mpm, llmod);
244                 populate_llvm_passes(fpm, mpm, llmod, opt_level,
245                                      trans.no_builtins);
246             }
247
248             for pass in sess.opts.cg.passes.iter() {
249                 pass.as_slice().with_c_str(|s| {
250                     if !llvm::LLVMRustAddPass(mpm, s) {
251                         sess.warn(format!("unknown pass {}, ignoring",
252                                           *pass).as_slice());
253                     }
254                 })
255             }
256
257             // Finally, run the actual optimization passes
258             time(sess.time_passes(), "llvm function passes", (), |()|
259                  llvm::LLVMRustRunFunctionPassManager(fpm, llmod));
260             time(sess.time_passes(), "llvm module passes", (), |()|
261                  llvm::LLVMRunPassManager(mpm, llmod));
262
263             // Deallocate managers that we're now done with
264             llvm::LLVMDisposePassManager(fpm);
265             llvm::LLVMDisposePassManager(mpm);
266
267             // Emit the bytecode if we're either saving our temporaries or
268             // emitting an rlib. Whenever an rlib is created, the bytecode is
269             // inserted into the archive in order to allow LTO against it.
270             if sess.opts.cg.save_temps ||
271                (sess.crate_types.borrow().contains(&config::CrateTypeRlib) &&
272                 sess.opts.output_types.contains(&OutputTypeExe)) {
273                 output.temp_path(OutputTypeBitcode).with_c_str(|buf| {
274                     llvm::LLVMWriteBitcodeToFile(llmod, buf);
275                 })
276             }
277
278             if sess.lto() {
279                 time(sess.time_passes(), "all lto passes", (), |()|
280                      lto::run(sess, llmod, tm, trans.reachable.as_slice()));
281
282                 if sess.opts.cg.save_temps {
283                     output.with_extension("lto.bc").with_c_str(|buf| {
284                         llvm::LLVMWriteBitcodeToFile(llmod, buf);
285                     })
286                 }
287             }
288
289             // A codegen-specific pass manager is used to generate object
290             // files for an LLVM module.
291             //
292             // Apparently each of these pass managers is a one-shot kind of
293             // thing, so we create a new one for each type of output. The
294             // pass manager passed to the closure should be ensured to not
295             // escape the closure itself, and the manager should only be
296             // used once.
297             fn with_codegen(tm: TargetMachineRef, llmod: ModuleRef,
298                             no_builtins: bool, f: |PassManagerRef|) {
299                 unsafe {
300                     let cpm = llvm::LLVMCreatePassManager();
301                     llvm::LLVMRustAddAnalysisPasses(tm, cpm, llmod);
302                     llvm::LLVMRustAddLibraryInfo(cpm, llmod, no_builtins);
303                     f(cpm);
304                     llvm::LLVMDisposePassManager(cpm);
305                 }
306             }
307
308             let mut object_file = None;
309             let mut needs_metadata = false;
310             for output_type in output_types.iter() {
311                 let path = output.path(*output_type);
312                 match *output_type {
313                     OutputTypeBitcode => {
314                         path.with_c_str(|buf| {
315                             llvm::LLVMWriteBitcodeToFile(llmod, buf);
316                         })
317                     }
318                     OutputTypeLlvmAssembly => {
319                         path.with_c_str(|output| {
320                             with_codegen(tm, llmod, trans.no_builtins, |cpm| {
321                                 llvm::LLVMRustPrintModule(cpm, llmod, output);
322                             })
323                         })
324                     }
325                     OutputTypeAssembly => {
326                         // If we're not using the LLVM assembler, this function
327                         // could be invoked specially with output_type_assembly,
328                         // so in this case we still want the metadata object
329                         // file.
330                         let ty = OutputTypeAssembly;
331                         let path = if sess.opts.output_types.contains(&ty) {
332                            path
333                         } else {
334                             needs_metadata = true;
335                             output.temp_path(OutputTypeAssembly)
336                         };
337                         with_codegen(tm, llmod, trans.no_builtins, |cpm| {
338                             write_output_file(sess, tm, cpm, llmod, &path,
339                                             llvm::AssemblyFile);
340                         });
341                     }
342                     OutputTypeObject => {
343                         object_file = Some(path);
344                     }
345                     OutputTypeExe => {
346                         object_file = Some(output.temp_path(OutputTypeObject));
347                         needs_metadata = true;
348                     }
349                 }
350             }
351
352             time(sess.time_passes(), "codegen passes", (), |()| {
353                 match object_file {
354                     Some(ref path) => {
355                         with_codegen(tm, llmod, trans.no_builtins, |cpm| {
356                             write_output_file(sess, tm, cpm, llmod, path,
357                                             llvm::ObjectFile);
358                         });
359                     }
360                     None => {}
361                 }
362                 if needs_metadata {
363                     with_codegen(tm, trans.metadata_module,
364                                  trans.no_builtins, |cpm| {
365                         let out = output.temp_path(OutputTypeObject)
366                                         .with_extension("metadata.o");
367                         write_output_file(sess, tm, cpm,
368                                         trans.metadata_module, &out,
369                                         llvm::ObjectFile);
370                     })
371                 }
372             });
373
374             llvm::LLVMRustDisposeTargetMachine(tm);
375             llvm::LLVMDisposeModule(trans.metadata_module);
376             llvm::LLVMDisposeModule(llmod);
377             llvm::LLVMContextDispose(llcx);
378             if sess.time_llvm_passes() { llvm::LLVMRustPrintPassTimings(); }
379         }
380     }
381
382     pub fn run_assembler(sess: &Session, outputs: &OutputFilenames) {
383         let pname = super::get_cc_prog(sess);
384         let mut cmd = Command::new(pname.as_slice());
385
386         cmd.arg("-c").arg("-o").arg(outputs.path(OutputTypeObject))
387                                .arg(outputs.temp_path(OutputTypeAssembly));
388         debug!("{}", &cmd);
389
390         match cmd.output() {
391             Ok(prog) => {
392                 if !prog.status.success() {
393                     sess.err(format!("linking with `{}` failed: {}",
394                                      pname,
395                                      prog.status).as_slice());
396                     sess.note(format!("{}", &cmd).as_slice());
397                     let mut note = prog.error.clone();
398                     note.push_all(prog.output.as_slice());
399                     sess.note(str::from_utf8(note.as_slice()).unwrap());
400                     sess.abort_if_errors();
401                 }
402             },
403             Err(e) => {
404                 sess.err(format!("could not exec the linker `{}`: {}",
405                                  pname,
406                                  e).as_slice());
407                 sess.abort_if_errors();
408             }
409         }
410     }
411
412     unsafe fn configure_llvm(sess: &Session) {
413         use std::sync::{Once, ONCE_INIT};
414         static mut INIT: Once = ONCE_INIT;
415
416         // Copy what clang does by turning on loop vectorization at O2 and
417         // slp vectorization at O3
418         let vectorize_loop = !sess.opts.cg.no_vectorize_loops &&
419                              (sess.opts.optimize == config::Default ||
420                               sess.opts.optimize == config::Aggressive);
421         let vectorize_slp = !sess.opts.cg.no_vectorize_slp &&
422                             sess.opts.optimize == config::Aggressive;
423
424         let mut llvm_c_strs = Vec::new();
425         let mut llvm_args = Vec::new();
426         {
427             let add = |arg: &str| {
428                 let s = arg.to_c_str();
429                 llvm_args.push(s.as_ptr());
430                 llvm_c_strs.push(s);
431             };
432             add("rustc"); // fake program name
433             if vectorize_loop { add("-vectorize-loops"); }
434             if vectorize_slp  { add("-vectorize-slp");   }
435             if sess.time_llvm_passes() { add("-time-passes"); }
436             if sess.print_llvm_passes() { add("-debug-pass=Structure"); }
437
438             for arg in sess.opts.cg.llvm_args.iter() {
439                 add((*arg).as_slice());
440             }
441         }
442
443         INIT.doit(|| {
444             llvm::LLVMInitializePasses();
445
446             // Only initialize the platforms supported by Rust here, because
447             // using --llvm-root will have multiple platforms that rustllvm
448             // doesn't actually link to and it's pointless to put target info
449             // into the registry that Rust cannot generate machine code for.
450             llvm::LLVMInitializeX86TargetInfo();
451             llvm::LLVMInitializeX86Target();
452             llvm::LLVMInitializeX86TargetMC();
453             llvm::LLVMInitializeX86AsmPrinter();
454             llvm::LLVMInitializeX86AsmParser();
455
456             llvm::LLVMInitializeARMTargetInfo();
457             llvm::LLVMInitializeARMTarget();
458             llvm::LLVMInitializeARMTargetMC();
459             llvm::LLVMInitializeARMAsmPrinter();
460             llvm::LLVMInitializeARMAsmParser();
461
462             llvm::LLVMInitializeMipsTargetInfo();
463             llvm::LLVMInitializeMipsTarget();
464             llvm::LLVMInitializeMipsTargetMC();
465             llvm::LLVMInitializeMipsAsmPrinter();
466             llvm::LLVMInitializeMipsAsmParser();
467
468             llvm::LLVMRustSetLLVMOptions(llvm_args.len() as c_int,
469                                          llvm_args.as_ptr());
470         });
471     }
472
473     unsafe fn populate_llvm_passes(fpm: llvm::PassManagerRef,
474                                    mpm: llvm::PassManagerRef,
475                                    llmod: ModuleRef,
476                                    opt: llvm::CodeGenOptLevel,
477                                    no_builtins: bool) {
478         // Create the PassManagerBuilder for LLVM. We configure it with
479         // reasonable defaults and prepare it to actually populate the pass
480         // manager.
481         let builder = llvm::LLVMPassManagerBuilderCreate();
482         match opt {
483             llvm::CodeGenLevelNone => {
484                 // Don't add lifetime intrinsics at O0
485                 llvm::LLVMRustAddAlwaysInlinePass(builder, false);
486             }
487             llvm::CodeGenLevelLess => {
488                 llvm::LLVMRustAddAlwaysInlinePass(builder, true);
489             }
490             // numeric values copied from clang
491             llvm::CodeGenLevelDefault => {
492                 llvm::LLVMPassManagerBuilderUseInlinerWithThreshold(builder,
493                                                                     225);
494             }
495             llvm::CodeGenLevelAggressive => {
496                 llvm::LLVMPassManagerBuilderUseInlinerWithThreshold(builder,
497                                                                     275);
498             }
499         }
500         llvm::LLVMPassManagerBuilderSetOptLevel(builder, opt as c_uint);
501         llvm::LLVMRustAddBuilderLibraryInfo(builder, llmod, no_builtins);
502
503         // Use the builder to populate the function/module pass managers.
504         llvm::LLVMPassManagerBuilderPopulateFunctionPassManager(builder, fpm);
505         llvm::LLVMPassManagerBuilderPopulateModulePassManager(builder, mpm);
506         llvm::LLVMPassManagerBuilderDispose(builder);
507     }
508 }
509
510
511 /*
512  * Name mangling and its relationship to metadata. This is complex. Read
513  * carefully.
514  *
515  * The semantic model of Rust linkage is, broadly, that "there's no global
516  * namespace" between crates. Our aim is to preserve the illusion of this
517  * model despite the fact that it's not *quite* possible to implement on
518  * modern linkers. We initially didn't use system linkers at all, but have
519  * been convinced of their utility.
520  *
521  * There are a few issues to handle:
522  *
523  *  - Linkers operate on a flat namespace, so we have to flatten names.
524  *    We do this using the C++ namespace-mangling technique. Foo::bar
525  *    symbols and such.
526  *
527  *  - Symbols with the same name but different types need to get different
528  *    linkage-names. We do this by hashing a string-encoding of the type into
529  *    a fixed-size (currently 16-byte hex) cryptographic hash function (CHF:
530  *    we use SHA256) to "prevent collisions". This is not airtight but 16 hex
531  *    digits on uniform probability means you're going to need 2**32 same-name
532  *    symbols in the same process before you're even hitting birthday-paradox
533  *    collision probability.
534  *
535  *  - Symbols in different crates but with same names "within" the crate need
536  *    to get different linkage-names.
537  *
538  *  - The hash shown in the filename needs to be predictable and stable for
539  *    build tooling integration. It also needs to be using a hash function
540  *    which is easy to use from Python, make, etc.
541  *
542  * So here is what we do:
543  *
544  *  - Consider the package id; every crate has one (specified with crate_id
545  *    attribute).  If a package id isn't provided explicitly, we infer a
546  *    versionless one from the output name. The version will end up being 0.0
547  *    in this case. CNAME and CVERS are taken from this package id. For
548  *    example, github.com/mozilla/CNAME#CVERS.
549  *
550  *  - Define CMH as SHA256(crateid).
551  *
552  *  - Define CMH8 as the first 8 characters of CMH.
553  *
554  *  - Compile our crate to lib CNAME-CMH8-CVERS.so
555  *
556  *  - Define STH(sym) as SHA256(CMH, type_str(sym))
557  *
558  *  - Suffix a mangled sym with ::STH@CVERS, so that it is unique in the
559  *    name, non-name metadata, and type sense, and versioned in the way
560  *    system linkers understand.
561  */
562
563 pub fn find_crate_name(sess: Option<&Session>,
564                        attrs: &[ast::Attribute],
565                        input: &Input) -> String {
566     use syntax::crateid::CrateId;
567
568     let validate = |s: String, span: Option<Span>| {
569         creader::validate_crate_name(sess, s.as_slice(), span);
570         s
571     };
572
573     // Look in attributes 100% of the time to make sure the attribute is marked
574     // as used. After doing this, however, we still prioritize a crate name from
575     // the command line over one found in the #[crate_name] attribute. If we
576     // find both we ensure that they're the same later on as well.
577     let attr_crate_name = attrs.iter().find(|at| at.check_name("crate_name"))
578                                .and_then(|at| at.value_str().map(|s| (at, s)));
579
580     match sess {
581         Some(sess) => {
582             match sess.opts.crate_name {
583                 Some(ref s) => {
584                     match attr_crate_name {
585                         Some((attr, ref name)) if s.as_slice() != name.get() => {
586                             let msg = format!("--crate-name and #[crate_name] \
587                                                are required to match, but `{}` \
588                                                != `{}`", s, name);
589                             sess.span_err(attr.span, msg.as_slice());
590                         }
591                         _ => {},
592                     }
593                     return validate(s.clone(), None);
594                 }
595                 None => {}
596             }
597         }
598         None => {}
599     }
600
601     match attr_crate_name {
602         Some((attr, s)) => return validate(s.get().to_string(), Some(attr.span)),
603         None => {}
604     }
605     let crate_id = attrs.iter().find(|at| at.check_name("crate_id"))
606                         .and_then(|at| at.value_str().map(|s| (at, s)))
607                         .and_then(|(at, s)| {
608                             from_str::<CrateId>(s.get()).map(|id| (at, id))
609                         });
610     match crate_id {
611         Some((attr, id)) => {
612             match sess {
613                 Some(sess) => {
614                     sess.span_warn(attr.span, "the #[crate_id] attribute is \
615                                                deprecated for the \
616                                                #[crate_name] attribute");
617                 }
618                 None => {}
619             }
620             return validate(id.name, Some(attr.span))
621         }
622         None => {}
623     }
624     match *input {
625         FileInput(ref path) => {
626             match path.filestem_str() {
627                 Some(s) => return validate(s.to_string(), None),
628                 None => {}
629             }
630         }
631         _ => {}
632     }
633
634     "rust-out".to_string()
635 }
636
637 pub fn build_link_meta(sess: &Session, krate: &ast::Crate,
638                        name: String) -> LinkMeta {
639     let r = LinkMeta {
640         crate_name: name,
641         crate_hash: Svh::calculate(&sess.opts.cg.metadata, krate),
642     };
643     info!("{}", r);
644     return r;
645 }
646
647 fn truncated_hash_result(symbol_hasher: &mut Sha256) -> String {
648     let output = symbol_hasher.result_bytes();
649     // 64 bits should be enough to avoid collisions.
650     output.slice_to(8).to_hex().to_string()
651 }
652
653
654 // This calculates STH for a symbol, as defined above
655 fn symbol_hash(tcx: &ty::ctxt,
656                symbol_hasher: &mut Sha256,
657                t: ty::t,
658                link_meta: &LinkMeta)
659                -> String {
660     // NB: do *not* use abbrevs here as we want the symbol names
661     // to be independent of one another in the crate.
662
663     symbol_hasher.reset();
664     symbol_hasher.input_str(link_meta.crate_name.as_slice());
665     symbol_hasher.input_str("-");
666     symbol_hasher.input_str(link_meta.crate_hash.as_str());
667     for meta in tcx.sess.crate_metadata.borrow().iter() {
668         symbol_hasher.input_str(meta.as_slice());
669     }
670     symbol_hasher.input_str("-");
671     symbol_hasher.input_str(encoder::encoded_ty(tcx, t).as_slice());
672     // Prefix with 'h' so that it never blends into adjacent digits
673     let mut hash = String::from_str("h");
674     hash.push_str(truncated_hash_result(symbol_hasher).as_slice());
675     hash
676 }
677
678 fn get_symbol_hash(ccx: &CrateContext, t: ty::t) -> String {
679     match ccx.type_hashcodes.borrow().find(&t) {
680         Some(h) => return h.to_string(),
681         None => {}
682     }
683
684     let mut symbol_hasher = ccx.symbol_hasher.borrow_mut();
685     let hash = symbol_hash(ccx.tcx(), &mut *symbol_hasher, t, &ccx.link_meta);
686     ccx.type_hashcodes.borrow_mut().insert(t, hash.clone());
687     hash
688 }
689
690
691 // Name sanitation. LLVM will happily accept identifiers with weird names, but
692 // gas doesn't!
693 // gas accepts the following characters in symbols: a-z, A-Z, 0-9, ., _, $
694 pub fn sanitize(s: &str) -> String {
695     let mut result = String::new();
696     for c in s.chars() {
697         match c {
698             // Escape these with $ sequences
699             '@' => result.push_str("$SP$"),
700             '~' => result.push_str("$UP$"),
701             '*' => result.push_str("$RP$"),
702             '&' => result.push_str("$BP$"),
703             '<' => result.push_str("$LT$"),
704             '>' => result.push_str("$GT$"),
705             '(' => result.push_str("$LP$"),
706             ')' => result.push_str("$RP$"),
707             ',' => result.push_str("$C$"),
708
709             // '.' doesn't occur in types and functions, so reuse it
710             // for ':' and '-'
711             '-' | ':' => result.push_char('.'),
712
713             // These are legal symbols
714             'a' .. 'z'
715             | 'A' .. 'Z'
716             | '0' .. '9'
717             | '_' | '.' | '$' => result.push_char(c),
718
719             _ => {
720                 let mut tstr = String::new();
721                 char::escape_unicode(c, |c| tstr.push_char(c));
722                 result.push_char('$');
723                 result.push_str(tstr.as_slice().slice_from(1));
724             }
725         }
726     }
727
728     // Underscore-qualify anything that didn't start as an ident.
729     if result.len() > 0u &&
730         result.as_bytes()[0] != '_' as u8 &&
731         ! char::is_XID_start(result.as_bytes()[0] as char) {
732         return format!("_{}", result.as_slice());
733     }
734
735     return result;
736 }
737
738 pub fn mangle<PI: Iterator<PathElem>>(mut path: PI,
739                                       hash: Option<&str>) -> String {
740     // Follow C++ namespace-mangling style, see
741     // http://en.wikipedia.org/wiki/Name_mangling for more info.
742     //
743     // It turns out that on OSX you can actually have arbitrary symbols in
744     // function names (at least when given to LLVM), but this is not possible
745     // when using unix's linker. Perhaps one day when we just use a linker from LLVM
746     // we won't need to do this name mangling. The problem with name mangling is
747     // that it seriously limits the available characters. For example we can't
748     // have things like &T or ~[T] in symbol names when one would theoretically
749     // want them for things like impls of traits on that type.
750     //
751     // To be able to work on all platforms and get *some* reasonable output, we
752     // use C++ name-mangling.
753
754     let mut n = String::from_str("_ZN"); // _Z == Begin name-sequence, N == nested
755
756     fn push(n: &mut String, s: &str) {
757         let sani = sanitize(s);
758         n.push_str(format!("{}{}", sani.len(), sani).as_slice());
759     }
760
761     // First, connect each component with <len, name> pairs.
762     for e in path {
763         push(&mut n, token::get_name(e.name()).get().as_slice())
764     }
765
766     match hash {
767         Some(s) => push(&mut n, s),
768         None => {}
769     }
770
771     n.push_char('E'); // End name-sequence.
772     n
773 }
774
775 pub fn exported_name(path: PathElems, hash: &str) -> String {
776     mangle(path, Some(hash))
777 }
778
779 pub fn mangle_exported_name(ccx: &CrateContext, path: PathElems,
780                             t: ty::t, id: ast::NodeId) -> String {
781     let mut hash = get_symbol_hash(ccx, t);
782
783     // Paths can be completely identical for different nodes,
784     // e.g. `fn foo() { { fn a() {} } { fn a() {} } }`, so we
785     // generate unique characters from the node id. For now
786     // hopefully 3 characters is enough to avoid collisions.
787     static EXTRA_CHARS: &'static str =
788         "abcdefghijklmnopqrstuvwxyz\
789          ABCDEFGHIJKLMNOPQRSTUVWXYZ\
790          0123456789";
791     let id = id as uint;
792     let extra1 = id % EXTRA_CHARS.len();
793     let id = id / EXTRA_CHARS.len();
794     let extra2 = id % EXTRA_CHARS.len();
795     let id = id / EXTRA_CHARS.len();
796     let extra3 = id % EXTRA_CHARS.len();
797     hash.push_char(EXTRA_CHARS.as_bytes()[extra1] as char);
798     hash.push_char(EXTRA_CHARS.as_bytes()[extra2] as char);
799     hash.push_char(EXTRA_CHARS.as_bytes()[extra3] as char);
800
801     exported_name(path, hash.as_slice())
802 }
803
804 pub fn mangle_internal_name_by_type_and_seq(ccx: &CrateContext,
805                                             t: ty::t,
806                                             name: &str) -> String {
807     let s = ppaux::ty_to_string(ccx.tcx(), t);
808     let path = [PathName(token::intern(s.as_slice())),
809                 gensym_name(name)];
810     let hash = get_symbol_hash(ccx, t);
811     mangle(ast_map::Values(path.iter()), Some(hash.as_slice()))
812 }
813
814 pub fn mangle_internal_name_by_path_and_seq(path: PathElems, flav: &str) -> String {
815     mangle(path.chain(Some(gensym_name(flav)).move_iter()), None)
816 }
817
818 pub fn get_cc_prog(sess: &Session) -> String {
819     match sess.opts.cg.linker {
820         Some(ref linker) => return linker.to_string(),
821         None => {}
822     }
823
824     // In the future, FreeBSD will use clang as default compiler.
825     // It would be flexible to use cc (system's default C compiler)
826     // instead of hard-coded gcc.
827     // For win32, there is no cc command, so we add a condition to make it use gcc.
828     match sess.targ_cfg.os {
829         abi::OsWin32 => "gcc",
830         _ => "cc",
831     }.to_string()
832 }
833
834 pub fn get_ar_prog(sess: &Session) -> String {
835     match sess.opts.cg.ar {
836         Some(ref ar) => (*ar).clone(),
837         None => "ar".to_string()
838     }
839 }
840
841 fn remove(sess: &Session, path: &Path) {
842     match fs::unlink(path) {
843         Ok(..) => {}
844         Err(e) => {
845             sess.err(format!("failed to remove {}: {}",
846                              path.display(),
847                              e).as_slice());
848         }
849     }
850 }
851
852 /// Perform the linkage portion of the compilation phase. This will generate all
853 /// of the requested outputs for this compilation session.
854 pub fn link_binary(sess: &Session,
855                    trans: &CrateTranslation,
856                    outputs: &OutputFilenames,
857                    crate_name: &str) -> Vec<Path> {
858     let mut out_filenames = Vec::new();
859     for &crate_type in sess.crate_types.borrow().iter() {
860         if invalid_output_for_target(sess, crate_type) {
861             sess.bug(format!("invalid output type `{}` for target os `{}`",
862                              crate_type, sess.targ_cfg.os).as_slice());
863         }
864         let out_file = link_binary_output(sess, trans, crate_type, outputs,
865                                           crate_name);
866         out_filenames.push(out_file);
867     }
868
869     // Remove the temporary object file and metadata if we aren't saving temps
870     if !sess.opts.cg.save_temps {
871         let obj_filename = outputs.temp_path(OutputTypeObject);
872         if !sess.opts.output_types.contains(&OutputTypeObject) {
873             remove(sess, &obj_filename);
874         }
875         remove(sess, &obj_filename.with_extension("metadata.o"));
876     }
877
878     out_filenames
879 }
880
881
882 /// Returns default crate type for target
883 ///
884 /// Default crate type is used when crate type isn't provided neither
885 /// through cmd line arguments nor through crate attributes
886 ///
887 /// It is CrateTypeExecutable for all platforms but iOS as there is no
888 /// way to run iOS binaries anyway without jailbreaking and
889 /// interaction with Rust code through static library is the only
890 /// option for now
891 pub fn default_output_for_target(sess: &Session) -> config::CrateType {
892     match sess.targ_cfg.os {
893         abi::OsiOS => config::CrateTypeStaticlib,
894         _ => config::CrateTypeExecutable
895     }
896 }
897
898 /// Checks if target supports crate_type as output
899 pub fn invalid_output_for_target(sess: &Session,
900                                  crate_type: config::CrateType) -> bool {
901     match (sess.targ_cfg.os, crate_type) {
902         (abi::OsiOS, config::CrateTypeDylib) => true,
903         _ => false
904     }
905 }
906
907 fn is_writeable(p: &Path) -> bool {
908     match p.stat() {
909         Err(..) => true,
910         Ok(m) => m.perm & io::UserWrite == io::UserWrite
911     }
912 }
913
914 pub fn filename_for_input(sess: &Session,
915                           crate_type: config::CrateType,
916                           name: &str,
917                           out_filename: &Path) -> Path {
918     let libname = format!("{}{}", name, sess.opts.cg.extra_filename);
919     match crate_type {
920         config::CrateTypeRlib => {
921             out_filename.with_filename(format!("lib{}.rlib", libname))
922         }
923         config::CrateTypeDylib => {
924             let (prefix, suffix) = match sess.targ_cfg.os {
925                 abi::OsWin32 => (loader::WIN32_DLL_PREFIX, loader::WIN32_DLL_SUFFIX),
926                 abi::OsMacos => (loader::MACOS_DLL_PREFIX, loader::MACOS_DLL_SUFFIX),
927                 abi::OsLinux => (loader::LINUX_DLL_PREFIX, loader::LINUX_DLL_SUFFIX),
928                 abi::OsAndroid => (loader::ANDROID_DLL_PREFIX, loader::ANDROID_DLL_SUFFIX),
929                 abi::OsFreebsd => (loader::FREEBSD_DLL_PREFIX, loader::FREEBSD_DLL_SUFFIX),
930                 abi::OsDragonfly => (loader::DRAGONFLY_DLL_PREFIX, loader::DRAGONFLY_DLL_SUFFIX),
931                 abi::OsiOS => unreachable!(),
932             };
933             out_filename.with_filename(format!("{}{}{}",
934                                                prefix,
935                                                libname,
936                                                suffix))
937         }
938         config::CrateTypeStaticlib => {
939             out_filename.with_filename(format!("lib{}.a", libname))
940         }
941         config::CrateTypeExecutable => {
942             match sess.targ_cfg.os {
943                 abi::OsWin32 => out_filename.with_extension("exe"),
944                 abi::OsMacos |
945                 abi::OsLinux |
946                 abi::OsAndroid |
947                 abi::OsFreebsd |
948                 abi::OsDragonfly |
949                 abi::OsiOS => out_filename.clone(),
950             }
951         }
952     }
953 }
954
955 fn link_binary_output(sess: &Session,
956                       trans: &CrateTranslation,
957                       crate_type: config::CrateType,
958                       outputs: &OutputFilenames,
959                       crate_name: &str) -> Path {
960     let obj_filename = outputs.temp_path(OutputTypeObject);
961     let out_filename = match outputs.single_output_file {
962         Some(ref file) => file.clone(),
963         None => {
964             let out_filename = outputs.path(OutputTypeExe);
965             filename_for_input(sess, crate_type, crate_name, &out_filename)
966         }
967     };
968
969     // Make sure the output and obj_filename are both writeable.
970     // Mac, FreeBSD, and Windows system linkers check this already --
971     // however, the Linux linker will happily overwrite a read-only file.
972     // We should be consistent.
973     let obj_is_writeable = is_writeable(&obj_filename);
974     let out_is_writeable = is_writeable(&out_filename);
975     if !out_is_writeable {
976         sess.fatal(format!("output file {} is not writeable -- check its \
977                             permissions.",
978                            out_filename.display()).as_slice());
979     }
980     else if !obj_is_writeable {
981         sess.fatal(format!("object file {} is not writeable -- check its \
982                             permissions.",
983                            obj_filename.display()).as_slice());
984     }
985
986     match crate_type {
987         config::CrateTypeRlib => {
988             link_rlib(sess, Some(trans), &obj_filename, &out_filename).build();
989         }
990         config::CrateTypeStaticlib => {
991             link_staticlib(sess, &obj_filename, &out_filename);
992         }
993         config::CrateTypeExecutable => {
994             link_natively(sess, trans, false, &obj_filename, &out_filename);
995         }
996         config::CrateTypeDylib => {
997             link_natively(sess, trans, true, &obj_filename, &out_filename);
998         }
999     }
1000
1001     out_filename
1002 }
1003
1004 fn archive_search_paths(sess: &Session) -> Vec<Path> {
1005     let mut rustpath = filesearch::rust_path();
1006     rustpath.push(sess.target_filesearch().get_lib_path());
1007     // FIXME: Addl lib search paths are an unordered HashSet?
1008     // Shouldn't this search be done in some order?
1009     let addl_lib_paths: HashSet<Path> = sess.opts.addl_lib_search_paths.borrow().clone();
1010     let mut search: Vec<Path> = addl_lib_paths.move_iter().collect();
1011     search.push_all(rustpath.as_slice());
1012     return search;
1013 }
1014
1015 // Create an 'rlib'
1016 //
1017 // An rlib in its current incarnation is essentially a renamed .a file. The
1018 // rlib primarily contains the object file of the crate, but it also contains
1019 // all of the object files from native libraries. This is done by unzipping
1020 // native libraries and inserting all of the contents into this archive.
1021 fn link_rlib<'a>(sess: &'a Session,
1022                  trans: Option<&CrateTranslation>, // None == no metadata/bytecode
1023                  obj_filename: &Path,
1024                  out_filename: &Path) -> ArchiveBuilder<'a> {
1025     let handler = &sess.diagnostic().handler;
1026     let config = ArchiveConfig {
1027         handler: handler,
1028         dst: out_filename.clone(),
1029         lib_search_paths: archive_search_paths(sess),
1030         os: sess.targ_cfg.os,
1031         maybe_ar_prog: sess.opts.cg.ar.clone()
1032     };
1033     let mut ab = ArchiveBuilder::create(config);
1034     ab.add_file(obj_filename).unwrap();
1035
1036     for &(ref l, kind) in sess.cstore.get_used_libraries().borrow().iter() {
1037         match kind {
1038             cstore::NativeStatic => {
1039                 ab.add_native_library(l.as_slice()).unwrap();
1040             }
1041             cstore::NativeFramework | cstore::NativeUnknown => {}
1042         }
1043     }
1044
1045     // After adding all files to the archive, we need to update the
1046     // symbol table of the archive.
1047     ab.update_symbols();
1048
1049     let mut ab = match sess.targ_cfg.os {
1050         // For OSX/iOS, we must be careful to update symbols only when adding
1051         // object files.  We're about to start adding non-object files, so run
1052         // `ar` now to process the object files.
1053         abi::OsMacos | abi::OsiOS => ab.build().extend(),
1054         _ => ab,
1055     };
1056
1057     // Note that it is important that we add all of our non-object "magical
1058     // files" *after* all of the object files in the archive. The reason for
1059     // this is as follows:
1060     //
1061     // * When performing LTO, this archive will be modified to remove
1062     //   obj_filename from above. The reason for this is described below.
1063     //
1064     // * When the system linker looks at an archive, it will attempt to
1065     //   determine the architecture of the archive in order to see whether its
1066     //   linkable.
1067     //
1068     //   The algorithm for this detection is: iterate over the files in the
1069     //   archive. Skip magical SYMDEF names. Interpret the first file as an
1070     //   object file. Read architecture from the object file.
1071     //
1072     // * As one can probably see, if "metadata" and "foo.bc" were placed
1073     //   before all of the objects, then the architecture of this archive would
1074     //   not be correctly inferred once 'foo.o' is removed.
1075     //
1076     // Basically, all this means is that this code should not move above the
1077     // code above.
1078     match trans {
1079         Some(trans) => {
1080             // Instead of putting the metadata in an object file section, rlibs
1081             // contain the metadata in a separate file. We use a temp directory
1082             // here so concurrent builds in the same directory don't try to use
1083             // the same filename for metadata (stomping over one another)
1084             let tmpdir = TempDir::new("rustc").expect("needs a temp dir");
1085             let metadata = tmpdir.path().join(METADATA_FILENAME);
1086             match fs::File::create(&metadata).write(trans.metadata
1087                                                          .as_slice()) {
1088                 Ok(..) => {}
1089                 Err(e) => {
1090                     sess.err(format!("failed to write {}: {}",
1091                                      metadata.display(),
1092                                      e).as_slice());
1093                     sess.abort_if_errors();
1094                 }
1095             }
1096             ab.add_file(&metadata).unwrap();
1097             remove(sess, &metadata);
1098
1099             // For LTO purposes, the bytecode of this library is also inserted
1100             // into the archive.
1101             //
1102             // Note that we make sure that the bytecode filename in the archive
1103             // is never exactly 16 bytes long by adding a 16 byte extension to
1104             // it. This is to work around a bug in LLDB that would cause it to
1105             // crash if the name of a file in an archive was exactly 16 bytes.
1106             let bc = obj_filename.with_extension("bc");
1107             let bc_deflated = obj_filename.with_extension("bytecode.deflate");
1108             match fs::File::open(&bc).read_to_end().and_then(|data| {
1109                 fs::File::create(&bc_deflated)
1110                     .write(match flate::deflate_bytes(data.as_slice()) {
1111                         Some(compressed) => compressed,
1112                         None => sess.fatal("failed to compress bytecode")
1113                      }.as_slice())
1114             }) {
1115                 Ok(()) => {}
1116                 Err(e) => {
1117                     sess.err(format!("failed to write compressed bytecode: \
1118                                       {}",
1119                                      e).as_slice());
1120                     sess.abort_if_errors()
1121                 }
1122             }
1123             ab.add_file(&bc_deflated).unwrap();
1124             remove(sess, &bc_deflated);
1125             if !sess.opts.cg.save_temps &&
1126                !sess.opts.output_types.contains(&OutputTypeBitcode) {
1127                 remove(sess, &bc);
1128             }
1129         }
1130
1131         None => {}
1132     }
1133
1134     ab
1135 }
1136
1137 // Create a static archive
1138 //
1139 // This is essentially the same thing as an rlib, but it also involves adding
1140 // all of the upstream crates' objects into the archive. This will slurp in
1141 // all of the native libraries of upstream dependencies as well.
1142 //
1143 // Additionally, there's no way for us to link dynamic libraries, so we warn
1144 // about all dynamic library dependencies that they're not linked in.
1145 //
1146 // There's no need to include metadata in a static archive, so ensure to not
1147 // link in the metadata object file (and also don't prepare the archive with a
1148 // metadata file).
1149 fn link_staticlib(sess: &Session, obj_filename: &Path, out_filename: &Path) {
1150     let ab = link_rlib(sess, None, obj_filename, out_filename);
1151     let mut ab = match sess.targ_cfg.os {
1152         abi::OsMacos | abi::OsiOS => ab.build().extend(),
1153         _ => ab,
1154     };
1155     ab.add_native_library("morestack").unwrap();
1156     ab.add_native_library("compiler-rt").unwrap();
1157
1158     let crates = sess.cstore.get_used_crates(cstore::RequireStatic);
1159     let mut all_native_libs = vec![];
1160
1161     for &(cnum, ref path) in crates.iter() {
1162         let name = sess.cstore.get_crate_data(cnum).name.clone();
1163         let p = match *path {
1164             Some(ref p) => p.clone(), None => {
1165                 sess.err(format!("could not find rlib for: `{}`",
1166                                  name).as_slice());
1167                 continue
1168             }
1169         };
1170         ab.add_rlib(&p, name.as_slice(), sess.lto()).unwrap();
1171
1172         let native_libs = csearch::get_native_libraries(&sess.cstore, cnum);
1173         all_native_libs.extend(native_libs.move_iter());
1174     }
1175
1176     ab.update_symbols();
1177     let _ = ab.build();
1178
1179     if !all_native_libs.is_empty() {
1180         sess.warn("link against the following native artifacts when linking against \
1181                   this static library");
1182         sess.note("the order and any duplication can be significant on some platforms, \
1183                   and so may need to be preserved");
1184     }
1185
1186     for &(kind, ref lib) in all_native_libs.iter() {
1187         let name = match kind {
1188             cstore::NativeStatic => "static library",
1189             cstore::NativeUnknown => "library",
1190             cstore::NativeFramework => "framework",
1191         };
1192         sess.note(format!("{}: {}", name, *lib).as_slice());
1193     }
1194 }
1195
1196 // Create a dynamic library or executable
1197 //
1198 // This will invoke the system linker/cc to create the resulting file. This
1199 // links to all upstream files as well.
1200 fn link_natively(sess: &Session, trans: &CrateTranslation, dylib: bool,
1201                  obj_filename: &Path, out_filename: &Path) {
1202     let tmpdir = TempDir::new("rustc").expect("needs a temp dir");
1203
1204     // The invocations of cc share some flags across platforms
1205     let pname = get_cc_prog(sess);
1206     let mut cmd = Command::new(pname.as_slice());
1207
1208     cmd.args(sess.targ_cfg.target_strs.cc_args.as_slice());
1209     link_args(&mut cmd, sess, dylib, tmpdir.path(),
1210               trans, obj_filename, out_filename);
1211
1212     if (sess.opts.debugging_opts & config::PRINT_LINK_ARGS) != 0 {
1213         println!("{}", &cmd);
1214     }
1215
1216     // May have not found libraries in the right formats.
1217     sess.abort_if_errors();
1218
1219     // Invoke the system linker
1220     debug!("{}", &cmd);
1221     let prog = time(sess.time_passes(), "running linker", (), |()| cmd.output());
1222     match prog {
1223         Ok(prog) => {
1224             if !prog.status.success() {
1225                 sess.err(format!("linking with `{}` failed: {}",
1226                                  pname,
1227                                  prog.status).as_slice());
1228                 sess.note(format!("{}", &cmd).as_slice());
1229                 let mut output = prog.error.clone();
1230                 output.push_all(prog.output.as_slice());
1231                 sess.note(str::from_utf8(output.as_slice()).unwrap());
1232                 sess.abort_if_errors();
1233             }
1234         },
1235         Err(e) => {
1236             sess.err(format!("could not exec the linker `{}`: {}",
1237                              pname,
1238                              e).as_slice());
1239             sess.abort_if_errors();
1240         }
1241     }
1242
1243
1244     // On OSX, debuggers need this utility to get run to do some munging of
1245     // the symbols
1246     if (sess.targ_cfg.os == abi::OsMacos || sess.targ_cfg.os == abi::OsiOS)
1247         && (sess.opts.debuginfo != NoDebugInfo) {
1248             match Command::new("dsymutil").arg(out_filename).output() {
1249                 Ok(..) => {}
1250                 Err(e) => {
1251                     sess.err(format!("failed to run dsymutil: {}", e).as_slice());
1252                     sess.abort_if_errors();
1253                 }
1254             }
1255         }
1256 }
1257
1258 fn link_args(cmd: &mut Command,
1259              sess: &Session,
1260              dylib: bool,
1261              tmpdir: &Path,
1262              trans: &CrateTranslation,
1263              obj_filename: &Path,
1264              out_filename: &Path) {
1265
1266     // The default library location, we need this to find the runtime.
1267     // The location of crates will be determined as needed.
1268     let lib_path = sess.target_filesearch().get_lib_path();
1269     cmd.arg("-L").arg(&lib_path);
1270
1271     cmd.arg("-o").arg(out_filename).arg(obj_filename);
1272
1273     // Stack growth requires statically linking a __morestack function. Note
1274     // that this is listed *before* all other libraries. Due to the usage of the
1275     // --as-needed flag below, the standard library may only be useful for its
1276     // rust_stack_exhausted function. In this case, we must ensure that the
1277     // libmorestack.a file appears *before* the standard library (so we put it
1278     // at the very front).
1279     //
1280     // Most of the time this is sufficient, except for when LLVM gets super
1281     // clever. If, for example, we have a main function `fn main() {}`, LLVM
1282     // will optimize out calls to `__morestack` entirely because the function
1283     // doesn't need any stack at all!
1284     //
1285     // To get around this snag, we specially tell the linker to always include
1286     // all contents of this library. This way we're guaranteed that the linker
1287     // will include the __morestack symbol 100% of the time, always resolving
1288     // references to it even if the object above didn't use it.
1289     match sess.targ_cfg.os {
1290         abi::OsMacos | abi::OsiOS => {
1291             let morestack = lib_path.join("libmorestack.a");
1292
1293             let mut v = b"-Wl,-force_load,".to_vec();
1294             v.push_all(morestack.as_vec());
1295             cmd.arg(v.as_slice());
1296         }
1297         _ => {
1298             cmd.args(["-Wl,--whole-archive", "-lmorestack",
1299                       "-Wl,--no-whole-archive"]);
1300         }
1301     }
1302
1303     // When linking a dynamic library, we put the metadata into a section of the
1304     // executable. This metadata is in a separate object file from the main
1305     // object file, so we link that in here.
1306     if dylib {
1307         cmd.arg(obj_filename.with_extension("metadata.o"));
1308     }
1309
1310     // We want to prevent the compiler from accidentally leaking in any system
1311     // libraries, so we explicitly ask gcc to not link to any libraries by
1312     // default. Note that this does not happen for windows because windows pulls
1313     // in some large number of libraries and I couldn't quite figure out which
1314     // subset we wanted.
1315     //
1316     // FIXME(#11937) we should invoke the system linker directly
1317     if sess.targ_cfg.os != abi::OsWin32 {
1318         cmd.arg("-nodefaultlibs");
1319     }
1320
1321     // If we're building a dylib, we don't use --gc-sections because LLVM has
1322     // already done the best it can do, and we also don't want to eliminate the
1323     // metadata. If we're building an executable, however, --gc-sections drops
1324     // the size of hello world from 1.8MB to 597K, a 67% reduction.
1325     if !dylib && sess.targ_cfg.os != abi::OsMacos && sess.targ_cfg.os != abi::OsiOS {
1326         cmd.arg("-Wl,--gc-sections");
1327     }
1328
1329     if sess.targ_cfg.os == abi::OsLinux || sess.targ_cfg.os == abi::OsDragonfly {
1330         // GNU-style linkers will use this to omit linking to libraries which
1331         // don't actually fulfill any relocations, but only for libraries which
1332         // follow this flag. Thus, use it before specifying libraries to link to.
1333         cmd.arg("-Wl,--as-needed");
1334
1335         // GNU-style linkers support optimization with -O. GNU ld doesn't need a
1336         // numeric argument, but other linkers do.
1337         if sess.opts.optimize == config::Default ||
1338            sess.opts.optimize == config::Aggressive {
1339             cmd.arg("-Wl,-O1");
1340         }
1341     } else if sess.targ_cfg.os == abi::OsMacos || sess.targ_cfg.os == abi::OsiOS {
1342         // The dead_strip option to the linker specifies that functions and data
1343         // unreachable by the entry point will be removed. This is quite useful
1344         // with Rust's compilation model of compiling libraries at a time into
1345         // one object file. For example, this brings hello world from 1.7MB to
1346         // 458K.
1347         //
1348         // Note that this is done for both executables and dynamic libraries. We
1349         // won't get much benefit from dylibs because LLVM will have already
1350         // stripped away as much as it could. This has not been seen to impact
1351         // link times negatively.
1352         cmd.arg("-Wl,-dead_strip");
1353     }
1354
1355     if sess.targ_cfg.os == abi::OsWin32 {
1356         // Make sure that we link to the dynamic libgcc, otherwise cross-module
1357         // DWARF stack unwinding will not work.
1358         // This behavior may be overridden by --link-args "-static-libgcc"
1359         cmd.arg("-shared-libgcc");
1360
1361         // And here, we see obscure linker flags #45. On windows, it has been
1362         // found to be necessary to have this flag to compile liblibc.
1363         //
1364         // First a bit of background. On Windows, the file format is not ELF,
1365         // but COFF (at least according to LLVM). COFF doesn't officially allow
1366         // for section names over 8 characters, apparently. Our metadata
1367         // section, ".note.rustc", you'll note is over 8 characters.
1368         //
1369         // On more recent versions of gcc on mingw, apparently the section name
1370         // is *not* truncated, but rather stored elsewhere in a separate lookup
1371         // table. On older versions of gcc, they apparently always truncated the
1372         // section names (at least in some cases). Truncating the section name
1373         // actually creates "invalid" objects [1] [2], but only for some
1374         // introspection tools, not in terms of whether it can be loaded.
1375         //
1376         // Long story short, passing this flag forces the linker to *not*
1377         // truncate section names (so we can find the metadata section after
1378         // it's compiled). The real kicker is that rust compiled just fine on
1379         // windows for quite a long time *without* this flag, so I have no idea
1380         // why it suddenly started failing for liblibc. Regardless, we
1381         // definitely don't want section name truncation, so we're keeping this
1382         // flag for windows.
1383         //
1384         // [1] - https://sourceware.org/bugzilla/show_bug.cgi?id=13130
1385         // [2] - https://code.google.com/p/go/issues/detail?id=2139
1386         cmd.arg("-Wl,--enable-long-section-names");
1387     }
1388
1389     if sess.targ_cfg.os == abi::OsAndroid {
1390         // Many of the symbols defined in compiler-rt are also defined in libgcc.
1391         // Android linker doesn't like that by default.
1392         cmd.arg("-Wl,--allow-multiple-definition");
1393     }
1394
1395     // Take careful note of the ordering of the arguments we pass to the linker
1396     // here. Linkers will assume that things on the left depend on things to the
1397     // right. Things on the right cannot depend on things on the left. This is
1398     // all formally implemented in terms of resolving symbols (libs on the right
1399     // resolve unknown symbols of libs on the left, but not vice versa).
1400     //
1401     // For this reason, we have organized the arguments we pass to the linker as
1402     // such:
1403     //
1404     //  1. The local object that LLVM just generated
1405     //  2. Upstream rust libraries
1406     //  3. Local native libraries
1407     //  4. Upstream native libraries
1408     //
1409     // This is generally fairly natural, but some may expect 2 and 3 to be
1410     // swapped. The reason that all native libraries are put last is that it's
1411     // not recommended for a native library to depend on a symbol from a rust
1412     // crate. If this is the case then a staticlib crate is recommended, solving
1413     // the problem.
1414     //
1415     // Additionally, it is occasionally the case that upstream rust libraries
1416     // depend on a local native library. In the case of libraries such as
1417     // lua/glfw/etc the name of the library isn't the same across all platforms,
1418     // so only the consumer crate of a library knows the actual name. This means
1419     // that downstream crates will provide the #[link] attribute which upstream
1420     // crates will depend on. Hence local native libraries are after out
1421     // upstream rust crates.
1422     //
1423     // In theory this means that a symbol in an upstream native library will be
1424     // shadowed by a local native library when it wouldn't have been before, but
1425     // this kind of behavior is pretty platform specific and generally not
1426     // recommended anyway, so I don't think we're shooting ourself in the foot
1427     // much with that.
1428     add_upstream_rust_crates(cmd, sess, dylib, tmpdir, trans);
1429     add_local_native_libraries(cmd, sess);
1430     add_upstream_native_libraries(cmd, sess);
1431
1432     // # Telling the linker what we're doing
1433
1434     if dylib {
1435         // On mac we need to tell the linker to let this library be rpathed
1436         if sess.targ_cfg.os == abi::OsMacos {
1437             cmd.args(["-dynamiclib", "-Wl,-dylib"]);
1438
1439             if sess.opts.cg.rpath {
1440                 let mut v = Vec::from_slice("-Wl,-install_name,@rpath/".as_bytes());
1441                 v.push_all(out_filename.filename().unwrap());
1442                 cmd.arg(v.as_slice());
1443             }
1444         } else {
1445             cmd.arg("-shared");
1446         }
1447     }
1448
1449     if sess.targ_cfg.os == abi::OsFreebsd {
1450         cmd.args(["-L/usr/local/lib",
1451                   "-L/usr/local/lib/gcc46",
1452                   "-L/usr/local/lib/gcc44"]);
1453     }
1454     else if sess.targ_cfg.os == abi::OsDragonfly {
1455         cmd.args(["-L/usr/local/lib",
1456                   "-L/usr/lib/gcc47",
1457                   "-L/usr/lib/gcc44"]);
1458     }
1459
1460
1461     // FIXME (#2397): At some point we want to rpath our guesses as to
1462     // where extern libraries might live, based on the
1463     // addl_lib_search_paths
1464     if sess.opts.cg.rpath {
1465         let sysroot = sess.sysroot();
1466         let target_triple = sess.opts.target_triple.as_slice();
1467         let get_install_prefix_lib_path = || {
1468             let install_prefix = option_env!("CFG_PREFIX").expect("CFG_PREFIX");
1469             let tlib = filesearch::relative_target_lib_path(sysroot, target_triple);
1470             let mut path = Path::new(install_prefix);
1471             path.push(&tlib);
1472
1473             path
1474         };
1475         let rpath_config = RPathConfig {
1476             os: sess.targ_cfg.os,
1477             used_crates: sess.cstore.get_used_crates(cstore::RequireDynamic),
1478             out_filename: out_filename.clone(),
1479             get_install_prefix_lib_path: get_install_prefix_lib_path,
1480             realpath: ::util::fs::realpath
1481         };
1482         cmd.args(rpath::get_rpath_flags(rpath_config).as_slice());
1483     }
1484
1485     // compiler-rt contains implementations of low-level LLVM helpers. This is
1486     // used to resolve symbols from the object file we just created, as well as
1487     // any system static libraries that may be expecting gcc instead. Most
1488     // symbols in libgcc also appear in compiler-rt.
1489     //
1490     // This is the end of the command line, so this library is used to resolve
1491     // *all* undefined symbols in all other libraries, and this is intentional.
1492     cmd.arg("-lcompiler-rt");
1493
1494     // Finally add all the linker arguments provided on the command line along
1495     // with any #[link_args] attributes found inside the crate
1496     cmd.args(sess.opts.cg.link_args.as_slice());
1497     for arg in sess.cstore.get_used_link_args().borrow().iter() {
1498         cmd.arg(arg.as_slice());
1499     }
1500 }
1501
1502 // # Native library linking
1503 //
1504 // User-supplied library search paths (-L on the command line). These are
1505 // the same paths used to find Rust crates, so some of them may have been
1506 // added already by the previous crate linking code. This only allows them
1507 // to be found at compile time so it is still entirely up to outside
1508 // forces to make sure that library can be found at runtime.
1509 //
1510 // Also note that the native libraries linked here are only the ones located
1511 // in the current crate. Upstream crates with native library dependencies
1512 // may have their native library pulled in above.
1513 fn add_local_native_libraries(cmd: &mut Command, sess: &Session) {
1514     for path in sess.opts.addl_lib_search_paths.borrow().iter() {
1515         cmd.arg("-L").arg(path);
1516     }
1517
1518     let rustpath = filesearch::rust_path();
1519     for path in rustpath.iter() {
1520         cmd.arg("-L").arg(path);
1521     }
1522
1523     // Some platforms take hints about whether a library is static or dynamic.
1524     // For those that support this, we ensure we pass the option if the library
1525     // was flagged "static" (most defaults are dynamic) to ensure that if
1526     // libfoo.a and libfoo.so both exist that the right one is chosen.
1527     let takes_hints = sess.targ_cfg.os != abi::OsMacos && sess.targ_cfg.os != abi::OsiOS;
1528
1529     for &(ref l, kind) in sess.cstore.get_used_libraries().borrow().iter() {
1530         match kind {
1531             cstore::NativeUnknown | cstore::NativeStatic => {
1532                 if takes_hints {
1533                     if kind == cstore::NativeStatic {
1534                         cmd.arg("-Wl,-Bstatic");
1535                     } else {
1536                         cmd.arg("-Wl,-Bdynamic");
1537                     }
1538                 }
1539                 cmd.arg(format!("-l{}", *l));
1540             }
1541             cstore::NativeFramework => {
1542                 cmd.arg("-framework");
1543                 cmd.arg(l.as_slice());
1544             }
1545         }
1546     }
1547     if takes_hints {
1548         cmd.arg("-Wl,-Bdynamic");
1549     }
1550 }
1551
1552 // # Rust Crate linking
1553 //
1554 // Rust crates are not considered at all when creating an rlib output. All
1555 // dependencies will be linked when producing the final output (instead of
1556 // the intermediate rlib version)
1557 fn add_upstream_rust_crates(cmd: &mut Command, sess: &Session,
1558                             dylib: bool, tmpdir: &Path,
1559                             trans: &CrateTranslation) {
1560     // All of the heavy lifting has previously been accomplished by the
1561     // dependency_format module of the compiler. This is just crawling the
1562     // output of that module, adding crates as necessary.
1563     //
1564     // Linking to a rlib involves just passing it to the linker (the linker
1565     // will slurp up the object files inside), and linking to a dynamic library
1566     // involves just passing the right -l flag.
1567
1568     let data = if dylib {
1569         trans.crate_formats.get(&config::CrateTypeDylib)
1570     } else {
1571         trans.crate_formats.get(&config::CrateTypeExecutable)
1572     };
1573
1574     // Invoke get_used_crates to ensure that we get a topological sorting of
1575     // crates.
1576     let deps = sess.cstore.get_used_crates(cstore::RequireDynamic);
1577
1578     for &(cnum, _) in deps.iter() {
1579         // We may not pass all crates through to the linker. Some crates may
1580         // appear statically in an existing dylib, meaning we'll pick up all the
1581         // symbols from the dylib.
1582         let kind = match *data.get(cnum as uint - 1) {
1583             Some(t) => t,
1584             None => continue
1585         };
1586         let src = sess.cstore.get_used_crate_source(cnum).unwrap();
1587         match kind {
1588             cstore::RequireDynamic => {
1589                 add_dynamic_crate(cmd, sess, src.dylib.unwrap())
1590             }
1591             cstore::RequireStatic => {
1592                 add_static_crate(cmd, sess, tmpdir, src.rlib.unwrap())
1593             }
1594         }
1595
1596     }
1597
1598     // Converts a library file-stem into a cc -l argument
1599     fn unlib<'a>(config: &config::Config, stem: &'a [u8]) -> &'a [u8] {
1600         if stem.starts_with("lib".as_bytes()) && config.os != abi::OsWin32 {
1601             stem.tailn(3)
1602         } else {
1603             stem
1604         }
1605     }
1606
1607     // Adds the static "rlib" versions of all crates to the command line.
1608     fn add_static_crate(cmd: &mut Command, sess: &Session, tmpdir: &Path,
1609                         cratepath: Path) {
1610         // When performing LTO on an executable output, all of the
1611         // bytecode from the upstream libraries has already been
1612         // included in our object file output. We need to modify all of
1613         // the upstream archives to remove their corresponding object
1614         // file to make sure we don't pull the same code in twice.
1615         //
1616         // We must continue to link to the upstream archives to be sure
1617         // to pull in native static dependencies. As the final caveat,
1618         // on linux it is apparently illegal to link to a blank archive,
1619         // so if an archive no longer has any object files in it after
1620         // we remove `lib.o`, then don't link against it at all.
1621         //
1622         // If we're not doing LTO, then our job is simply to just link
1623         // against the archive.
1624         if sess.lto() {
1625             let name = cratepath.filename_str().unwrap();
1626             let name = name.slice(3, name.len() - 5); // chop off lib/.rlib
1627             time(sess.time_passes(),
1628                  format!("altering {}.rlib", name).as_slice(),
1629                  (), |()| {
1630                 let dst = tmpdir.join(cratepath.filename().unwrap());
1631                 match fs::copy(&cratepath, &dst) {
1632                     Ok(..) => {}
1633                     Err(e) => {
1634                         sess.err(format!("failed to copy {} to {}: {}",
1635                                          cratepath.display(),
1636                                          dst.display(),
1637                                          e).as_slice());
1638                         sess.abort_if_errors();
1639                     }
1640                 }
1641                 let handler = &sess.diagnostic().handler;
1642                 let config = ArchiveConfig {
1643                     handler: handler,
1644                     dst: dst.clone(),
1645                     lib_search_paths: archive_search_paths(sess),
1646                     os: sess.targ_cfg.os,
1647                     maybe_ar_prog: sess.opts.cg.ar.clone()
1648                 };
1649                 let mut archive = Archive::open(config);
1650                 archive.remove_file(format!("{}.o", name).as_slice());
1651                 let files = archive.files();
1652                 if files.iter().any(|s| s.as_slice().ends_with(".o")) {
1653                     cmd.arg(dst);
1654                 }
1655             });
1656         } else {
1657             cmd.arg(cratepath);
1658         }
1659     }
1660
1661     // Same thing as above, but for dynamic crates instead of static crates.
1662     fn add_dynamic_crate(cmd: &mut Command, sess: &Session, cratepath: Path) {
1663         // If we're performing LTO, then it should have been previously required
1664         // that all upstream rust dependencies were available in an rlib format.
1665         assert!(!sess.lto());
1666
1667         // Just need to tell the linker about where the library lives and
1668         // what its name is
1669         let dir = cratepath.dirname();
1670         if !dir.is_empty() { cmd.arg("-L").arg(dir); }
1671
1672         let mut v = Vec::from_slice("-l".as_bytes());
1673         v.push_all(unlib(&sess.targ_cfg, cratepath.filestem().unwrap()));
1674         cmd.arg(v.as_slice());
1675     }
1676 }
1677
1678 // Link in all of our upstream crates' native dependencies. Remember that
1679 // all of these upstream native dependencies are all non-static
1680 // dependencies. We've got two cases then:
1681 //
1682 // 1. The upstream crate is an rlib. In this case we *must* link in the
1683 // native dependency because the rlib is just an archive.
1684 //
1685 // 2. The upstream crate is a dylib. In order to use the dylib, we have to
1686 // have the dependency present on the system somewhere. Thus, we don't
1687 // gain a whole lot from not linking in the dynamic dependency to this
1688 // crate as well.
1689 //
1690 // The use case for this is a little subtle. In theory the native
1691 // dependencies of a crate are purely an implementation detail of the crate
1692 // itself, but the problem arises with generic and inlined functions. If a
1693 // generic function calls a native function, then the generic function must
1694 // be instantiated in the target crate, meaning that the native symbol must
1695 // also be resolved in the target crate.
1696 fn add_upstream_native_libraries(cmd: &mut Command, sess: &Session) {
1697     // Be sure to use a topological sorting of crates because there may be
1698     // interdependencies between native libraries. When passing -nodefaultlibs,
1699     // for example, almost all native libraries depend on libc, so we have to
1700     // make sure that's all the way at the right (liblibc is near the base of
1701     // the dependency chain).
1702     //
1703     // This passes RequireStatic, but the actual requirement doesn't matter,
1704     // we're just getting an ordering of crate numbers, we're not worried about
1705     // the paths.
1706     let crates = sess.cstore.get_used_crates(cstore::RequireStatic);
1707     for (cnum, _) in crates.move_iter() {
1708         let libs = csearch::get_native_libraries(&sess.cstore, cnum);
1709         for &(kind, ref lib) in libs.iter() {
1710             match kind {
1711                 cstore::NativeUnknown => {
1712                     cmd.arg(format!("-l{}", *lib));
1713                 }
1714                 cstore::NativeFramework => {
1715                     cmd.arg("-framework");
1716                     cmd.arg(lib.as_slice());
1717                 }
1718                 cstore::NativeStatic => {
1719                     sess.bug("statics shouldn't be propagated");
1720                 }
1721             }
1722         }
1723     }
1724 }