]> git.lizzy.rs Git - rust.git/blob - src/librustc/back/link.rs
auto merge of #15733 : sanxiyn/rust/use-from-type, r=alexcrichton
[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, 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, favor crate names from the command
575     // line.
576     let attr_crate_name = attrs.iter().find(|at| at.check_name("crate_name"))
577                                .and_then(|at| at.value_str().map(|s| (at, s)));
578
579     match sess {
580         Some(sess) => {
581             match sess.opts.crate_name {
582                 Some(ref s) => return validate(s.clone(), None),
583                 None => {}
584             }
585         }
586         None => {}
587     }
588
589     match attr_crate_name {
590         Some((attr, s)) => return validate(s.get().to_string(), Some(attr.span)),
591         None => {}
592     }
593     let crate_id = attrs.iter().find(|at| at.check_name("crate_id"))
594                         .and_then(|at| at.value_str().map(|s| (at, s)))
595                         .and_then(|(at, s)| {
596                             from_str::<CrateId>(s.get()).map(|id| (at, id))
597                         });
598     match crate_id {
599         Some((attr, id)) => {
600             match sess {
601                 Some(sess) => {
602                     sess.span_warn(attr.span, "the #[crate_id] attribute is \
603                                                deprecated for the \
604                                                #[crate_name] attribute");
605                 }
606                 None => {}
607             }
608             return validate(id.name, Some(attr.span))
609         }
610         None => {}
611     }
612     match *input {
613         FileInput(ref path) => {
614             match path.filestem_str() {
615                 Some(s) => return validate(s.to_string(), None),
616                 None => {}
617             }
618         }
619         _ => {}
620     }
621
622     "rust-out".to_string()
623 }
624
625 pub fn build_link_meta(sess: &Session, krate: &ast::Crate,
626                        name: String) -> LinkMeta {
627     let r = LinkMeta {
628         crate_name: name,
629         crate_hash: Svh::calculate(&sess.opts.cg.metadata, krate),
630     };
631     info!("{}", r);
632     return r;
633 }
634
635 fn truncated_hash_result(symbol_hasher: &mut Sha256) -> String {
636     let output = symbol_hasher.result_bytes();
637     // 64 bits should be enough to avoid collisions.
638     output.slice_to(8).to_hex().to_string()
639 }
640
641
642 // This calculates STH for a symbol, as defined above
643 fn symbol_hash(tcx: &ty::ctxt,
644                symbol_hasher: &mut Sha256,
645                t: ty::t,
646                link_meta: &LinkMeta)
647                -> String {
648     // NB: do *not* use abbrevs here as we want the symbol names
649     // to be independent of one another in the crate.
650
651     symbol_hasher.reset();
652     symbol_hasher.input_str(link_meta.crate_name.as_slice());
653     symbol_hasher.input_str("-");
654     symbol_hasher.input_str(link_meta.crate_hash.as_str());
655     for meta in tcx.sess.crate_metadata.borrow().iter() {
656         symbol_hasher.input_str(meta.as_slice());
657     }
658     symbol_hasher.input_str("-");
659     symbol_hasher.input_str(encoder::encoded_ty(tcx, t).as_slice());
660     // Prefix with 'h' so that it never blends into adjacent digits
661     let mut hash = String::from_str("h");
662     hash.push_str(truncated_hash_result(symbol_hasher).as_slice());
663     hash
664 }
665
666 fn get_symbol_hash(ccx: &CrateContext, t: ty::t) -> String {
667     match ccx.type_hashcodes.borrow().find(&t) {
668         Some(h) => return h.to_string(),
669         None => {}
670     }
671
672     let mut symbol_hasher = ccx.symbol_hasher.borrow_mut();
673     let hash = symbol_hash(ccx.tcx(), &mut *symbol_hasher, t, &ccx.link_meta);
674     ccx.type_hashcodes.borrow_mut().insert(t, hash.clone());
675     hash
676 }
677
678
679 // Name sanitation. LLVM will happily accept identifiers with weird names, but
680 // gas doesn't!
681 // gas accepts the following characters in symbols: a-z, A-Z, 0-9, ., _, $
682 pub fn sanitize(s: &str) -> String {
683     let mut result = String::new();
684     for c in s.chars() {
685         match c {
686             // Escape these with $ sequences
687             '@' => result.push_str("$SP$"),
688             '~' => result.push_str("$UP$"),
689             '*' => result.push_str("$RP$"),
690             '&' => result.push_str("$BP$"),
691             '<' => result.push_str("$LT$"),
692             '>' => result.push_str("$GT$"),
693             '(' => result.push_str("$LP$"),
694             ')' => result.push_str("$RP$"),
695             ',' => result.push_str("$C$"),
696
697             // '.' doesn't occur in types and functions, so reuse it
698             // for ':' and '-'
699             '-' | ':' => result.push_char('.'),
700
701             // These are legal symbols
702             'a' .. 'z'
703             | 'A' .. 'Z'
704             | '0' .. '9'
705             | '_' | '.' | '$' => result.push_char(c),
706
707             _ => {
708                 let mut tstr = String::new();
709                 char::escape_unicode(c, |c| tstr.push_char(c));
710                 result.push_char('$');
711                 result.push_str(tstr.as_slice().slice_from(1));
712             }
713         }
714     }
715
716     // Underscore-qualify anything that didn't start as an ident.
717     if result.len() > 0u &&
718         result.as_bytes()[0] != '_' as u8 &&
719         ! char::is_XID_start(result.as_bytes()[0] as char) {
720         return format!("_{}", result.as_slice());
721     }
722
723     return result;
724 }
725
726 pub fn mangle<PI: Iterator<PathElem>>(mut path: PI,
727                                       hash: Option<&str>) -> String {
728     // Follow C++ namespace-mangling style, see
729     // http://en.wikipedia.org/wiki/Name_mangling for more info.
730     //
731     // It turns out that on OSX you can actually have arbitrary symbols in
732     // function names (at least when given to LLVM), but this is not possible
733     // when using unix's linker. Perhaps one day when we just use a linker from LLVM
734     // we won't need to do this name mangling. The problem with name mangling is
735     // that it seriously limits the available characters. For example we can't
736     // have things like &T or ~[T] in symbol names when one would theoretically
737     // want them for things like impls of traits on that type.
738     //
739     // To be able to work on all platforms and get *some* reasonable output, we
740     // use C++ name-mangling.
741
742     let mut n = String::from_str("_ZN"); // _Z == Begin name-sequence, N == nested
743
744     fn push(n: &mut String, s: &str) {
745         let sani = sanitize(s);
746         n.push_str(format!("{}{}", sani.len(), sani).as_slice());
747     }
748
749     // First, connect each component with <len, name> pairs.
750     for e in path {
751         push(&mut n, token::get_name(e.name()).get().as_slice())
752     }
753
754     match hash {
755         Some(s) => push(&mut n, s),
756         None => {}
757     }
758
759     n.push_char('E'); // End name-sequence.
760     n
761 }
762
763 pub fn exported_name(path: PathElems, hash: &str) -> String {
764     mangle(path, Some(hash))
765 }
766
767 pub fn mangle_exported_name(ccx: &CrateContext, path: PathElems,
768                             t: ty::t, id: ast::NodeId) -> String {
769     let mut hash = get_symbol_hash(ccx, t);
770
771     // Paths can be completely identical for different nodes,
772     // e.g. `fn foo() { { fn a() {} } { fn a() {} } }`, so we
773     // generate unique characters from the node id. For now
774     // hopefully 3 characters is enough to avoid collisions.
775     static EXTRA_CHARS: &'static str =
776         "abcdefghijklmnopqrstuvwxyz\
777          ABCDEFGHIJKLMNOPQRSTUVWXYZ\
778          0123456789";
779     let id = id as uint;
780     let extra1 = id % EXTRA_CHARS.len();
781     let id = id / EXTRA_CHARS.len();
782     let extra2 = id % EXTRA_CHARS.len();
783     let id = id / EXTRA_CHARS.len();
784     let extra3 = id % EXTRA_CHARS.len();
785     hash.push_char(EXTRA_CHARS.as_bytes()[extra1] as char);
786     hash.push_char(EXTRA_CHARS.as_bytes()[extra2] as char);
787     hash.push_char(EXTRA_CHARS.as_bytes()[extra3] as char);
788
789     exported_name(path, hash.as_slice())
790 }
791
792 pub fn mangle_internal_name_by_type_and_seq(ccx: &CrateContext,
793                                             t: ty::t,
794                                             name: &str) -> String {
795     let s = ppaux::ty_to_string(ccx.tcx(), t);
796     let path = [PathName(token::intern(s.as_slice())),
797                 gensym_name(name)];
798     let hash = get_symbol_hash(ccx, t);
799     mangle(ast_map::Values(path.iter()), Some(hash.as_slice()))
800 }
801
802 pub fn mangle_internal_name_by_path_and_seq(path: PathElems, flav: &str) -> String {
803     mangle(path.chain(Some(gensym_name(flav)).move_iter()), None)
804 }
805
806 pub fn get_cc_prog(sess: &Session) -> String {
807     match sess.opts.cg.linker {
808         Some(ref linker) => return linker.to_string(),
809         None => {}
810     }
811
812     // In the future, FreeBSD will use clang as default compiler.
813     // It would be flexible to use cc (system's default C compiler)
814     // instead of hard-coded gcc.
815     // For win32, there is no cc command, so we add a condition to make it use gcc.
816     match sess.targ_cfg.os {
817         abi::OsWin32 => "gcc",
818         _ => "cc",
819     }.to_string()
820 }
821
822 pub fn get_ar_prog(sess: &Session) -> String {
823     match sess.opts.cg.ar {
824         Some(ref ar) => (*ar).clone(),
825         None => "ar".to_string()
826     }
827 }
828
829 fn remove(sess: &Session, path: &Path) {
830     match fs::unlink(path) {
831         Ok(..) => {}
832         Err(e) => {
833             sess.err(format!("failed to remove {}: {}",
834                              path.display(),
835                              e).as_slice());
836         }
837     }
838 }
839
840 /// Perform the linkage portion of the compilation phase. This will generate all
841 /// of the requested outputs for this compilation session.
842 pub fn link_binary(sess: &Session,
843                    trans: &CrateTranslation,
844                    outputs: &OutputFilenames,
845                    crate_name: &str) -> Vec<Path> {
846     let mut out_filenames = Vec::new();
847     for &crate_type in sess.crate_types.borrow().iter() {
848         if invalid_output_for_target(sess, crate_type) {
849             sess.bug(format!("invalid output type `{}` for target os `{}`",
850                              crate_type, sess.targ_cfg.os).as_slice());
851         }
852         let out_file = link_binary_output(sess, trans, crate_type, outputs,
853                                           crate_name);
854         out_filenames.push(out_file);
855     }
856
857     // Remove the temporary object file and metadata if we aren't saving temps
858     if !sess.opts.cg.save_temps {
859         let obj_filename = outputs.temp_path(OutputTypeObject);
860         if !sess.opts.output_types.contains(&OutputTypeObject) {
861             remove(sess, &obj_filename);
862         }
863         remove(sess, &obj_filename.with_extension("metadata.o"));
864     }
865
866     out_filenames
867 }
868
869
870 /// Returns default crate type for target
871 ///
872 /// Default crate type is used when crate type isn't provided neither
873 /// through cmd line arguments nor through crate attributes
874 ///
875 /// It is CrateTypeExecutable for all platforms but iOS as there is no
876 /// way to run iOS binaries anyway without jailbreaking and
877 /// interaction with Rust code through static library is the only
878 /// option for now
879 pub fn default_output_for_target(sess: &Session) -> config::CrateType {
880     match sess.targ_cfg.os {
881         abi::OsiOS => config::CrateTypeStaticlib,
882         _ => config::CrateTypeExecutable
883     }
884 }
885
886 /// Checks if target supports crate_type as output
887 pub fn invalid_output_for_target(sess: &Session,
888                                  crate_type: config::CrateType) -> bool {
889     match (sess.targ_cfg.os, crate_type) {
890         (abi::OsiOS, config::CrateTypeDylib) => true,
891         _ => false
892     }
893 }
894
895 fn is_writeable(p: &Path) -> bool {
896     match p.stat() {
897         Err(..) => true,
898         Ok(m) => m.perm & io::UserWrite == io::UserWrite
899     }
900 }
901
902 pub fn filename_for_input(sess: &Session,
903                           crate_type: config::CrateType,
904                           name: &str,
905                           out_filename: &Path) -> Path {
906     let libname = format!("{}{}", name, sess.opts.cg.extra_filename);
907     match crate_type {
908         config::CrateTypeRlib => {
909             out_filename.with_filename(format!("lib{}.rlib", libname))
910         }
911         config::CrateTypeDylib => {
912             let (prefix, suffix) = match sess.targ_cfg.os {
913                 abi::OsWin32 => (loader::WIN32_DLL_PREFIX, loader::WIN32_DLL_SUFFIX),
914                 abi::OsMacos => (loader::MACOS_DLL_PREFIX, loader::MACOS_DLL_SUFFIX),
915                 abi::OsLinux => (loader::LINUX_DLL_PREFIX, loader::LINUX_DLL_SUFFIX),
916                 abi::OsAndroid => (loader::ANDROID_DLL_PREFIX, loader::ANDROID_DLL_SUFFIX),
917                 abi::OsFreebsd => (loader::FREEBSD_DLL_PREFIX, loader::FREEBSD_DLL_SUFFIX),
918                 abi::OsiOS => unreachable!(),
919             };
920             out_filename.with_filename(format!("{}{}{}",
921                                                prefix,
922                                                libname,
923                                                suffix))
924         }
925         config::CrateTypeStaticlib => {
926             out_filename.with_filename(format!("lib{}.a", libname))
927         }
928         config::CrateTypeExecutable => out_filename.clone(),
929     }
930 }
931
932 fn link_binary_output(sess: &Session,
933                       trans: &CrateTranslation,
934                       crate_type: config::CrateType,
935                       outputs: &OutputFilenames,
936                       crate_name: &str) -> Path {
937     let obj_filename = outputs.temp_path(OutputTypeObject);
938     let out_filename = match outputs.single_output_file {
939         Some(ref file) => file.clone(),
940         None => {
941             let out_filename = outputs.path(OutputTypeExe);
942             filename_for_input(sess, crate_type, crate_name, &out_filename)
943         }
944     };
945
946     // Make sure the output and obj_filename are both writeable.
947     // Mac, FreeBSD, and Windows system linkers check this already --
948     // however, the Linux linker will happily overwrite a read-only file.
949     // We should be consistent.
950     let obj_is_writeable = is_writeable(&obj_filename);
951     let out_is_writeable = is_writeable(&out_filename);
952     if !out_is_writeable {
953         sess.fatal(format!("output file {} is not writeable -- check its \
954                             permissions.",
955                            out_filename.display()).as_slice());
956     }
957     else if !obj_is_writeable {
958         sess.fatal(format!("object file {} is not writeable -- check its \
959                             permissions.",
960                            obj_filename.display()).as_slice());
961     }
962
963     match crate_type {
964         config::CrateTypeRlib => {
965             link_rlib(sess, Some(trans), &obj_filename, &out_filename);
966         }
967         config::CrateTypeStaticlib => {
968             link_staticlib(sess, &obj_filename, &out_filename);
969         }
970         config::CrateTypeExecutable => {
971             link_natively(sess, trans, false, &obj_filename, &out_filename);
972         }
973         config::CrateTypeDylib => {
974             link_natively(sess, trans, true, &obj_filename, &out_filename);
975         }
976     }
977
978     out_filename
979 }
980
981 fn archive_search_paths(sess: &Session) -> Vec<Path> {
982     let mut rustpath = filesearch::rust_path();
983     rustpath.push(sess.target_filesearch().get_lib_path());
984     // FIXME: Addl lib search paths are an unordered HashSet?
985     // Shouldn't this search be done in some order?
986     let addl_lib_paths: HashSet<Path> = sess.opts.addl_lib_search_paths.borrow().clone();
987     let mut search: Vec<Path> = addl_lib_paths.move_iter().collect();
988     search.push_all(rustpath.as_slice());
989     return search;
990 }
991
992 // Create an 'rlib'
993 //
994 // An rlib in its current incarnation is essentially a renamed .a file. The
995 // rlib primarily contains the object file of the crate, but it also contains
996 // all of the object files from native libraries. This is done by unzipping
997 // native libraries and inserting all of the contents into this archive.
998 fn link_rlib<'a>(sess: &'a Session,
999                  trans: Option<&CrateTranslation>, // None == no metadata/bytecode
1000                  obj_filename: &Path,
1001                  out_filename: &Path) -> Archive<'a> {
1002     let handler = &sess.diagnostic().handler;
1003     let config = ArchiveConfig {
1004         handler: handler,
1005         dst: out_filename.clone(),
1006         lib_search_paths: archive_search_paths(sess),
1007         os: sess.targ_cfg.os,
1008         maybe_ar_prog: sess.opts.cg.ar.clone()
1009     };
1010     let mut a = Archive::create(config, obj_filename);
1011
1012     for &(ref l, kind) in sess.cstore.get_used_libraries().borrow().iter() {
1013         match kind {
1014             cstore::NativeStatic => {
1015                 a.add_native_library(l.as_slice()).unwrap();
1016             }
1017             cstore::NativeFramework | cstore::NativeUnknown => {}
1018         }
1019     }
1020
1021     // Note that it is important that we add all of our non-object "magical
1022     // files" *after* all of the object files in the archive. The reason for
1023     // this is as follows:
1024     //
1025     // * When performing LTO, this archive will be modified to remove
1026     //   obj_filename from above. The reason for this is described below.
1027     //
1028     // * When the system linker looks at an archive, it will attempt to
1029     //   determine the architecture of the archive in order to see whether its
1030     //   linkable.
1031     //
1032     //   The algorithm for this detection is: iterate over the files in the
1033     //   archive. Skip magical SYMDEF names. Interpret the first file as an
1034     //   object file. Read architecture from the object file.
1035     //
1036     // * As one can probably see, if "metadata" and "foo.bc" were placed
1037     //   before all of the objects, then the architecture of this archive would
1038     //   not be correctly inferred once 'foo.o' is removed.
1039     //
1040     // Basically, all this means is that this code should not move above the
1041     // code above.
1042     match trans {
1043         Some(trans) => {
1044             // Instead of putting the metadata in an object file section, rlibs
1045             // contain the metadata in a separate file. We use a temp directory
1046             // here so concurrent builds in the same directory don't try to use
1047             // the same filename for metadata (stomping over one another)
1048             let tmpdir = TempDir::new("rustc").expect("needs a temp dir");
1049             let metadata = tmpdir.path().join(METADATA_FILENAME);
1050             match fs::File::create(&metadata).write(trans.metadata
1051                                                          .as_slice()) {
1052                 Ok(..) => {}
1053                 Err(e) => {
1054                     sess.err(format!("failed to write {}: {}",
1055                                      metadata.display(),
1056                                      e).as_slice());
1057                     sess.abort_if_errors();
1058                 }
1059             }
1060             a.add_file(&metadata, false);
1061             remove(sess, &metadata);
1062
1063             // For LTO purposes, the bytecode of this library is also inserted
1064             // into the archive.
1065             //
1066             // Note that we make sure that the bytecode filename in the archive
1067             // is never exactly 16 bytes long by adding a 16 byte extension to
1068             // it. This is to work around a bug in LLDB that would cause it to
1069             // crash if the name of a file in an archive was exactly 16 bytes.
1070             let bc = obj_filename.with_extension("bc");
1071             let bc_deflated = obj_filename.with_extension("bytecode.deflate");
1072             match fs::File::open(&bc).read_to_end().and_then(|data| {
1073                 fs::File::create(&bc_deflated)
1074                     .write(match flate::deflate_bytes(data.as_slice()) {
1075                         Some(compressed) => compressed,
1076                         None => sess.fatal("failed to compress bytecode")
1077                      }.as_slice())
1078             }) {
1079                 Ok(()) => {}
1080                 Err(e) => {
1081                     sess.err(format!("failed to write compressed bytecode: \
1082                                       {}",
1083                                      e).as_slice());
1084                     sess.abort_if_errors()
1085                 }
1086             }
1087             a.add_file(&bc_deflated, false);
1088             remove(sess, &bc_deflated);
1089             if !sess.opts.cg.save_temps &&
1090                !sess.opts.output_types.contains(&OutputTypeBitcode) {
1091                 remove(sess, &bc);
1092             }
1093
1094             // After adding all files to the archive, we need to update the
1095             // symbol table of the archive. This currently dies on OSX (see
1096             // #11162), and isn't necessary there anyway
1097             match sess.targ_cfg.os {
1098                 abi::OsMacos | abi::OsiOS => {}
1099                 _ => { a.update_symbols(); }
1100             }
1101         }
1102
1103         None => {}
1104     }
1105     return a;
1106 }
1107
1108 // Create a static archive
1109 //
1110 // This is essentially the same thing as an rlib, but it also involves adding
1111 // all of the upstream crates' objects into the archive. This will slurp in
1112 // all of the native libraries of upstream dependencies as well.
1113 //
1114 // Additionally, there's no way for us to link dynamic libraries, so we warn
1115 // about all dynamic library dependencies that they're not linked in.
1116 //
1117 // There's no need to include metadata in a static archive, so ensure to not
1118 // link in the metadata object file (and also don't prepare the archive with a
1119 // metadata file).
1120 fn link_staticlib(sess: &Session, obj_filename: &Path, out_filename: &Path) {
1121     let mut a = link_rlib(sess, None, obj_filename, out_filename);
1122     a.add_native_library("morestack").unwrap();
1123     a.add_native_library("compiler-rt").unwrap();
1124
1125     let crates = sess.cstore.get_used_crates(cstore::RequireStatic);
1126     let mut all_native_libs = vec![];
1127
1128     for &(cnum, ref path) in crates.iter() {
1129         let name = sess.cstore.get_crate_data(cnum).name.clone();
1130         let p = match *path {
1131             Some(ref p) => p.clone(), None => {
1132                 sess.err(format!("could not find rlib for: `{}`",
1133                                  name).as_slice());
1134                 continue
1135             }
1136         };
1137         a.add_rlib(&p, name.as_slice(), sess.lto()).unwrap();
1138
1139         let native_libs = csearch::get_native_libraries(&sess.cstore, cnum);
1140         all_native_libs.extend(native_libs.move_iter());
1141     }
1142
1143     if !all_native_libs.is_empty() {
1144         sess.warn("link against the following native artifacts when linking against \
1145                   this static library");
1146         sess.note("the order and any duplication can be significant on some platforms, \
1147                   and so may need to be preserved");
1148     }
1149
1150     for &(kind, ref lib) in all_native_libs.iter() {
1151         let name = match kind {
1152             cstore::NativeStatic => "static library",
1153             cstore::NativeUnknown => "library",
1154             cstore::NativeFramework => "framework",
1155         };
1156         sess.note(format!("{}: {}", name, *lib).as_slice());
1157     }
1158 }
1159
1160 // Create a dynamic library or executable
1161 //
1162 // This will invoke the system linker/cc to create the resulting file. This
1163 // links to all upstream files as well.
1164 fn link_natively(sess: &Session, trans: &CrateTranslation, dylib: bool,
1165                  obj_filename: &Path, out_filename: &Path) {
1166     let tmpdir = TempDir::new("rustc").expect("needs a temp dir");
1167
1168     // The invocations of cc share some flags across platforms
1169     let pname = get_cc_prog(sess);
1170     let mut cmd = Command::new(pname.as_slice());
1171
1172     cmd.args(sess.targ_cfg.target_strs.cc_args.as_slice());
1173     link_args(&mut cmd, sess, dylib, tmpdir.path(),
1174               trans, obj_filename, out_filename);
1175
1176     if (sess.opts.debugging_opts & config::PRINT_LINK_ARGS) != 0 {
1177         println!("{}", &cmd);
1178     }
1179
1180     // May have not found libraries in the right formats.
1181     sess.abort_if_errors();
1182
1183     // Invoke the system linker
1184     debug!("{}", &cmd);
1185     let prog = time(sess.time_passes(), "running linker", (), |()| cmd.output());
1186     match prog {
1187         Ok(prog) => {
1188             if !prog.status.success() {
1189                 sess.err(format!("linking with `{}` failed: {}",
1190                                  pname,
1191                                  prog.status).as_slice());
1192                 sess.note(format!("{}", &cmd).as_slice());
1193                 let mut output = prog.error.clone();
1194                 output.push_all(prog.output.as_slice());
1195                 sess.note(str::from_utf8(output.as_slice()).unwrap());
1196                 sess.abort_if_errors();
1197             }
1198         },
1199         Err(e) => {
1200             sess.err(format!("could not exec the linker `{}`: {}",
1201                              pname,
1202                              e).as_slice());
1203             sess.abort_if_errors();
1204         }
1205     }
1206
1207
1208     // On OSX, debuggers need this utility to get run to do some munging of
1209     // the symbols
1210     if (sess.targ_cfg.os == abi::OsMacos || sess.targ_cfg.os == abi::OsiOS)
1211         && (sess.opts.debuginfo != NoDebugInfo) {
1212             match Command::new("dsymutil").arg(out_filename).status() {
1213                 Ok(..) => {}
1214                 Err(e) => {
1215                     sess.err(format!("failed to run dsymutil: {}", e).as_slice());
1216                     sess.abort_if_errors();
1217                 }
1218             }
1219         }
1220 }
1221
1222 fn link_args(cmd: &mut Command,
1223              sess: &Session,
1224              dylib: bool,
1225              tmpdir: &Path,
1226              trans: &CrateTranslation,
1227              obj_filename: &Path,
1228              out_filename: &Path) {
1229
1230     // The default library location, we need this to find the runtime.
1231     // The location of crates will be determined as needed.
1232     let lib_path = sess.target_filesearch().get_lib_path();
1233     cmd.arg("-L").arg(&lib_path);
1234
1235     cmd.arg("-o").arg(out_filename).arg(obj_filename);
1236
1237     // Stack growth requires statically linking a __morestack function. Note
1238     // that this is listed *before* all other libraries. Due to the usage of the
1239     // --as-needed flag below, the standard library may only be useful for its
1240     // rust_stack_exhausted function. In this case, we must ensure that the
1241     // libmorestack.a file appears *before* the standard library (so we put it
1242     // at the very front).
1243     //
1244     // Most of the time this is sufficient, except for when LLVM gets super
1245     // clever. If, for example, we have a main function `fn main() {}`, LLVM
1246     // will optimize out calls to `__morestack` entirely because the function
1247     // doesn't need any stack at all!
1248     //
1249     // To get around this snag, we specially tell the linker to always include
1250     // all contents of this library. This way we're guaranteed that the linker
1251     // will include the __morestack symbol 100% of the time, always resolving
1252     // references to it even if the object above didn't use it.
1253     match sess.targ_cfg.os {
1254         abi::OsMacos | abi::OsiOS => {
1255             let morestack = lib_path.join("libmorestack.a");
1256
1257             let mut v = b"-Wl,-force_load,".to_vec();
1258             v.push_all(morestack.as_vec());
1259             cmd.arg(v.as_slice());
1260         }
1261         _ => {
1262             cmd.args(["-Wl,--whole-archive", "-lmorestack",
1263                       "-Wl,--no-whole-archive"]);
1264         }
1265     }
1266
1267     // When linking a dynamic library, we put the metadata into a section of the
1268     // executable. This metadata is in a separate object file from the main
1269     // object file, so we link that in here.
1270     if dylib {
1271         cmd.arg(obj_filename.with_extension("metadata.o"));
1272     }
1273
1274     // We want to prevent the compiler from accidentally leaking in any system
1275     // libraries, so we explicitly ask gcc to not link to any libraries by
1276     // default. Note that this does not happen for windows because windows pulls
1277     // in some large number of libraries and I couldn't quite figure out which
1278     // subset we wanted.
1279     //
1280     // FIXME(#11937) we should invoke the system linker directly
1281     if sess.targ_cfg.os != abi::OsWin32 {
1282         cmd.arg("-nodefaultlibs");
1283     }
1284
1285     // If we're building a dylib, we don't use --gc-sections because LLVM has
1286     // already done the best it can do, and we also don't want to eliminate the
1287     // metadata. If we're building an executable, however, --gc-sections drops
1288     // the size of hello world from 1.8MB to 597K, a 67% reduction.
1289     if !dylib && sess.targ_cfg.os != abi::OsMacos && sess.targ_cfg.os != abi::OsiOS {
1290         cmd.arg("-Wl,--gc-sections");
1291     }
1292
1293     if sess.targ_cfg.os == abi::OsLinux {
1294         // GNU-style linkers will use this to omit linking to libraries which
1295         // don't actually fulfill any relocations, but only for libraries which
1296         // follow this flag. Thus, use it before specifying libraries to link to.
1297         cmd.arg("-Wl,--as-needed");
1298
1299         // GNU-style linkers support optimization with -O. GNU ld doesn't need a
1300         // numeric argument, but other linkers do.
1301         if sess.opts.optimize == config::Default ||
1302            sess.opts.optimize == config::Aggressive {
1303             cmd.arg("-Wl,-O1");
1304         }
1305     } else if sess.targ_cfg.os == abi::OsMacos || sess.targ_cfg.os == abi::OsiOS {
1306         // The dead_strip option to the linker specifies that functions and data
1307         // unreachable by the entry point will be removed. This is quite useful
1308         // with Rust's compilation model of compiling libraries at a time into
1309         // one object file. For example, this brings hello world from 1.7MB to
1310         // 458K.
1311         //
1312         // Note that this is done for both executables and dynamic libraries. We
1313         // won't get much benefit from dylibs because LLVM will have already
1314         // stripped away as much as it could. This has not been seen to impact
1315         // link times negatively.
1316         cmd.arg("-Wl,-dead_strip");
1317     }
1318
1319     if sess.targ_cfg.os == abi::OsWin32 {
1320         // Make sure that we link to the dynamic libgcc, otherwise cross-module
1321         // DWARF stack unwinding will not work.
1322         // This behavior may be overridden by --link-args "-static-libgcc"
1323         cmd.arg("-shared-libgcc");
1324
1325         // And here, we see obscure linker flags #45. On windows, it has been
1326         // found to be necessary to have this flag to compile liblibc.
1327         //
1328         // First a bit of background. On Windows, the file format is not ELF,
1329         // but COFF (at least according to LLVM). COFF doesn't officially allow
1330         // for section names over 8 characters, apparently. Our metadata
1331         // section, ".note.rustc", you'll note is over 8 characters.
1332         //
1333         // On more recent versions of gcc on mingw, apparently the section name
1334         // is *not* truncated, but rather stored elsewhere in a separate lookup
1335         // table. On older versions of gcc, they apparently always truncated the
1336         // section names (at least in some cases). Truncating the section name
1337         // actually creates "invalid" objects [1] [2], but only for some
1338         // introspection tools, not in terms of whether it can be loaded.
1339         //
1340         // Long story short, passing this flag forces the linker to *not*
1341         // truncate section names (so we can find the metadata section after
1342         // it's compiled). The real kicker is that rust compiled just fine on
1343         // windows for quite a long time *without* this flag, so I have no idea
1344         // why it suddenly started failing for liblibc. Regardless, we
1345         // definitely don't want section name truncation, so we're keeping this
1346         // flag for windows.
1347         //
1348         // [1] - https://sourceware.org/bugzilla/show_bug.cgi?id=13130
1349         // [2] - https://code.google.com/p/go/issues/detail?id=2139
1350         cmd.arg("-Wl,--enable-long-section-names");
1351     }
1352
1353     if sess.targ_cfg.os == abi::OsAndroid {
1354         // Many of the symbols defined in compiler-rt are also defined in libgcc.
1355         // Android linker doesn't like that by default.
1356         cmd.arg("-Wl,--allow-multiple-definition");
1357     }
1358
1359     // Take careful note of the ordering of the arguments we pass to the linker
1360     // here. Linkers will assume that things on the left depend on things to the
1361     // right. Things on the right cannot depend on things on the left. This is
1362     // all formally implemented in terms of resolving symbols (libs on the right
1363     // resolve unknown symbols of libs on the left, but not vice versa).
1364     //
1365     // For this reason, we have organized the arguments we pass to the linker as
1366     // such:
1367     //
1368     //  1. The local object that LLVM just generated
1369     //  2. Upstream rust libraries
1370     //  3. Local native libraries
1371     //  4. Upstream native libraries
1372     //
1373     // This is generally fairly natural, but some may expect 2 and 3 to be
1374     // swapped. The reason that all native libraries are put last is that it's
1375     // not recommended for a native library to depend on a symbol from a rust
1376     // crate. If this is the case then a staticlib crate is recommended, solving
1377     // the problem.
1378     //
1379     // Additionally, it is occasionally the case that upstream rust libraries
1380     // depend on a local native library. In the case of libraries such as
1381     // lua/glfw/etc the name of the library isn't the same across all platforms,
1382     // so only the consumer crate of a library knows the actual name. This means
1383     // that downstream crates will provide the #[link] attribute which upstream
1384     // crates will depend on. Hence local native libraries are after out
1385     // upstream rust crates.
1386     //
1387     // In theory this means that a symbol in an upstream native library will be
1388     // shadowed by a local native library when it wouldn't have been before, but
1389     // this kind of behavior is pretty platform specific and generally not
1390     // recommended anyway, so I don't think we're shooting ourself in the foot
1391     // much with that.
1392     add_upstream_rust_crates(cmd, sess, dylib, tmpdir, trans);
1393     add_local_native_libraries(cmd, sess);
1394     add_upstream_native_libraries(cmd, sess);
1395
1396     // # Telling the linker what we're doing
1397
1398     if dylib {
1399         // On mac we need to tell the linker to let this library be rpathed
1400         if sess.targ_cfg.os == abi::OsMacos {
1401             cmd.args(["-dynamiclib", "-Wl,-dylib"]);
1402
1403             if sess.opts.cg.rpath {
1404                 let mut v = Vec::from_slice("-Wl,-install_name,@rpath/".as_bytes());
1405                 v.push_all(out_filename.filename().unwrap());
1406                 cmd.arg(v.as_slice());
1407             }
1408         } else {
1409             cmd.arg("-shared");
1410         }
1411     }
1412
1413     if sess.targ_cfg.os == abi::OsFreebsd {
1414         cmd.args(["-L/usr/local/lib",
1415                   "-L/usr/local/lib/gcc46",
1416                   "-L/usr/local/lib/gcc44"]);
1417     }
1418
1419     // FIXME (#2397): At some point we want to rpath our guesses as to
1420     // where extern libraries might live, based on the
1421     // addl_lib_search_paths
1422     if sess.opts.cg.rpath {
1423         let sysroot = sess.sysroot();
1424         let target_triple = sess.opts.target_triple.as_slice();
1425         let get_install_prefix_lib_path = || {
1426             let install_prefix = option_env!("CFG_PREFIX").expect("CFG_PREFIX");
1427             let tlib = filesearch::relative_target_lib_path(sysroot, target_triple);
1428             let mut path = Path::new(install_prefix);
1429             path.push(&tlib);
1430
1431             path
1432         };
1433         let rpath_config = RPathConfig {
1434             os: sess.targ_cfg.os,
1435             used_crates: sess.cstore.get_used_crates(cstore::RequireDynamic),
1436             out_filename: out_filename.clone(),
1437             get_install_prefix_lib_path: get_install_prefix_lib_path,
1438             realpath: ::util::fs::realpath
1439         };
1440         cmd.args(rpath::get_rpath_flags(rpath_config).as_slice());
1441     }
1442
1443     // compiler-rt contains implementations of low-level LLVM helpers. This is
1444     // used to resolve symbols from the object file we just created, as well as
1445     // any system static libraries that may be expecting gcc instead. Most
1446     // symbols in libgcc also appear in compiler-rt.
1447     //
1448     // This is the end of the command line, so this library is used to resolve
1449     // *all* undefined symbols in all other libraries, and this is intentional.
1450     cmd.arg("-lcompiler-rt");
1451
1452     // Finally add all the linker arguments provided on the command line along
1453     // with any #[link_args] attributes found inside the crate
1454     cmd.args(sess.opts.cg.link_args.as_slice());
1455     for arg in sess.cstore.get_used_link_args().borrow().iter() {
1456         cmd.arg(arg.as_slice());
1457     }
1458 }
1459
1460 // # Native library linking
1461 //
1462 // User-supplied library search paths (-L on the command line). These are
1463 // the same paths used to find Rust crates, so some of them may have been
1464 // added already by the previous crate linking code. This only allows them
1465 // to be found at compile time so it is still entirely up to outside
1466 // forces to make sure that library can be found at runtime.
1467 //
1468 // Also note that the native libraries linked here are only the ones located
1469 // in the current crate. Upstream crates with native library dependencies
1470 // may have their native library pulled in above.
1471 fn add_local_native_libraries(cmd: &mut Command, sess: &Session) {
1472     for path in sess.opts.addl_lib_search_paths.borrow().iter() {
1473         cmd.arg("-L").arg(path);
1474     }
1475
1476     let rustpath = filesearch::rust_path();
1477     for path in rustpath.iter() {
1478         cmd.arg("-L").arg(path);
1479     }
1480
1481     // Some platforms take hints about whether a library is static or dynamic.
1482     // For those that support this, we ensure we pass the option if the library
1483     // was flagged "static" (most defaults are dynamic) to ensure that if
1484     // libfoo.a and libfoo.so both exist that the right one is chosen.
1485     let takes_hints = sess.targ_cfg.os != abi::OsMacos && sess.targ_cfg.os != abi::OsiOS;
1486
1487     for &(ref l, kind) in sess.cstore.get_used_libraries().borrow().iter() {
1488         match kind {
1489             cstore::NativeUnknown | cstore::NativeStatic => {
1490                 if takes_hints {
1491                     if kind == cstore::NativeStatic {
1492                         cmd.arg("-Wl,-Bstatic");
1493                     } else {
1494                         cmd.arg("-Wl,-Bdynamic");
1495                     }
1496                 }
1497                 cmd.arg(format!("-l{}", *l));
1498             }
1499             cstore::NativeFramework => {
1500                 cmd.arg("-framework");
1501                 cmd.arg(l.as_slice());
1502             }
1503         }
1504     }
1505     if takes_hints {
1506         cmd.arg("-Wl,-Bdynamic");
1507     }
1508 }
1509
1510 // # Rust Crate linking
1511 //
1512 // Rust crates are not considered at all when creating an rlib output. All
1513 // dependencies will be linked when producing the final output (instead of
1514 // the intermediate rlib version)
1515 fn add_upstream_rust_crates(cmd: &mut Command, sess: &Session,
1516                             dylib: bool, tmpdir: &Path,
1517                             trans: &CrateTranslation) {
1518     // All of the heavy lifting has previously been accomplished by the
1519     // dependency_format module of the compiler. This is just crawling the
1520     // output of that module, adding crates as necessary.
1521     //
1522     // Linking to a rlib involves just passing it to the linker (the linker
1523     // will slurp up the object files inside), and linking to a dynamic library
1524     // involves just passing the right -l flag.
1525
1526     let data = if dylib {
1527         trans.crate_formats.get(&config::CrateTypeDylib)
1528     } else {
1529         trans.crate_formats.get(&config::CrateTypeExecutable)
1530     };
1531
1532     // Invoke get_used_crates to ensure that we get a topological sorting of
1533     // crates.
1534     let deps = sess.cstore.get_used_crates(cstore::RequireDynamic);
1535
1536     for &(cnum, _) in deps.iter() {
1537         // We may not pass all crates through to the linker. Some crates may
1538         // appear statically in an existing dylib, meaning we'll pick up all the
1539         // symbols from the dylib.
1540         let kind = match *data.get(cnum as uint - 1) {
1541             Some(t) => t,
1542             None => continue
1543         };
1544         let src = sess.cstore.get_used_crate_source(cnum).unwrap();
1545         match kind {
1546             cstore::RequireDynamic => {
1547                 add_dynamic_crate(cmd, sess, src.dylib.unwrap())
1548             }
1549             cstore::RequireStatic => {
1550                 add_static_crate(cmd, sess, tmpdir, cnum, src.rlib.unwrap())
1551             }
1552         }
1553
1554     }
1555
1556     // Converts a library file-stem into a cc -l argument
1557     fn unlib<'a>(config: &config::Config, stem: &'a [u8]) -> &'a [u8] {
1558         if stem.starts_with("lib".as_bytes()) && config.os != abi::OsWin32 {
1559             stem.tailn(3)
1560         } else {
1561             stem
1562         }
1563     }
1564
1565     // Adds the static "rlib" versions of all crates to the command line.
1566     fn add_static_crate(cmd: &mut Command, sess: &Session, tmpdir: &Path,
1567                         cnum: ast::CrateNum, cratepath: Path) {
1568         // When performing LTO on an executable output, all of the
1569         // bytecode from the upstream libraries has already been
1570         // included in our object file output. We need to modify all of
1571         // the upstream archives to remove their corresponding object
1572         // file to make sure we don't pull the same code in twice.
1573         //
1574         // We must continue to link to the upstream archives to be sure
1575         // to pull in native static dependencies. As the final caveat,
1576         // on linux it is apparently illegal to link to a blank archive,
1577         // so if an archive no longer has any object files in it after
1578         // we remove `lib.o`, then don't link against it at all.
1579         //
1580         // If we're not doing LTO, then our job is simply to just link
1581         // against the archive.
1582         if sess.lto() {
1583             let name = sess.cstore.get_crate_data(cnum).name.clone();
1584             time(sess.time_passes(),
1585                  format!("altering {}.rlib", name).as_slice(),
1586                  (), |()| {
1587                 let dst = tmpdir.join(cratepath.filename().unwrap());
1588                 match fs::copy(&cratepath, &dst) {
1589                     Ok(..) => {}
1590                     Err(e) => {
1591                         sess.err(format!("failed to copy {} to {}: {}",
1592                                          cratepath.display(),
1593                                          dst.display(),
1594                                          e).as_slice());
1595                         sess.abort_if_errors();
1596                     }
1597                 }
1598                 let handler = &sess.diagnostic().handler;
1599                 let config = ArchiveConfig {
1600                     handler: handler,
1601                     dst: dst.clone(),
1602                     lib_search_paths: archive_search_paths(sess),
1603                     os: sess.targ_cfg.os,
1604                     maybe_ar_prog: sess.opts.cg.ar.clone()
1605                 };
1606                 let mut archive = Archive::open(config);
1607                 archive.remove_file(format!("{}.o", name).as_slice());
1608                 let files = archive.files();
1609                 if files.iter().any(|s| s.as_slice().ends_with(".o")) {
1610                     cmd.arg(dst);
1611                 }
1612             });
1613         } else {
1614             cmd.arg(cratepath);
1615         }
1616     }
1617
1618     // Same thing as above, but for dynamic crates instead of static crates.
1619     fn add_dynamic_crate(cmd: &mut Command, sess: &Session, cratepath: Path) {
1620         // If we're performing LTO, then it should have been previously required
1621         // that all upstream rust dependencies were available in an rlib format.
1622         assert!(!sess.lto());
1623
1624         // Just need to tell the linker about where the library lives and
1625         // what its name is
1626         let dir = cratepath.dirname();
1627         if !dir.is_empty() { cmd.arg("-L").arg(dir); }
1628
1629         let mut v = Vec::from_slice("-l".as_bytes());
1630         v.push_all(unlib(&sess.targ_cfg, cratepath.filestem().unwrap()));
1631         cmd.arg(v.as_slice());
1632     }
1633 }
1634
1635 // Link in all of our upstream crates' native dependencies. Remember that
1636 // all of these upstream native dependencies are all non-static
1637 // dependencies. We've got two cases then:
1638 //
1639 // 1. The upstream crate is an rlib. In this case we *must* link in the
1640 // native dependency because the rlib is just an archive.
1641 //
1642 // 2. The upstream crate is a dylib. In order to use the dylib, we have to
1643 // have the dependency present on the system somewhere. Thus, we don't
1644 // gain a whole lot from not linking in the dynamic dependency to this
1645 // crate as well.
1646 //
1647 // The use case for this is a little subtle. In theory the native
1648 // dependencies of a crate are purely an implementation detail of the crate
1649 // itself, but the problem arises with generic and inlined functions. If a
1650 // generic function calls a native function, then the generic function must
1651 // be instantiated in the target crate, meaning that the native symbol must
1652 // also be resolved in the target crate.
1653 fn add_upstream_native_libraries(cmd: &mut Command, sess: &Session) {
1654     // Be sure to use a topological sorting of crates because there may be
1655     // interdependencies between native libraries. When passing -nodefaultlibs,
1656     // for example, almost all native libraries depend on libc, so we have to
1657     // make sure that's all the way at the right (liblibc is near the base of
1658     // the dependency chain).
1659     //
1660     // This passes RequireStatic, but the actual requirement doesn't matter,
1661     // we're just getting an ordering of crate numbers, we're not worried about
1662     // the paths.
1663     let crates = sess.cstore.get_used_crates(cstore::RequireStatic);
1664     for (cnum, _) in crates.move_iter() {
1665         let libs = csearch::get_native_libraries(&sess.cstore, cnum);
1666         for &(kind, ref lib) in libs.iter() {
1667             match kind {
1668                 cstore::NativeUnknown => {
1669                     cmd.arg(format!("-l{}", *lib));
1670                 }
1671                 cstore::NativeFramework => {
1672                     cmd.arg("-framework");
1673                     cmd.arg(lib.as_slice());
1674                 }
1675                 cstore::NativeStatic => {
1676                     sess.bug("statics shouldn't be propagated");
1677                 }
1678             }
1679         }
1680     }
1681 }