base.rs 49.6 KB
Newer Older
1
// Copyright 2012-2015 The Rust Project Developers. See the COPYRIGHT
2 3 4 5 6 7 8 9
// file at the top-level directory of this distribution and at
// http://rust-lang.org/COPYRIGHT.
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.
10

I
Irina Popa 已提交
11
//! Codegen the completed AST to the LLVM IR.
12
//!
I
Irina Popa 已提交
13 14 15
//! Some functions here, such as codegen_block and codegen_expr, return a value --
//! the result of the codegen to LLVM -- while others, such as codegen_fn
//! and mono_item, are called only for the side effect of adding a
16 17
//! particular definition to the LLVM IR output we're producing.
//!
I
Irina Popa 已提交
18
//! Hopefully useful general knowledge about codegen:
19 20 21 22 23 24
//!
//!   * There's no way to find out the Ty type of a ValueRef.  Doing so
//!     would be "trying to get the eggs out of an omelette" (credit:
//!     pcwalton).  You can, instead, find out its TypeRef by calling val_ty,
//!     but one TypeRef corresponds to many `Ty`s; for instance, tup(int, int,
//!     int) and rec(x=int, y=int, z=int) will have the same TypeRef.
25

26 27
use super::ModuleLlvm;
use super::ModuleSource;
I
Irina Popa 已提交
28
use super::ModuleCodegen;
29
use super::ModuleKind;
30

31
use abi;
32
use back::link;
I
Irina Popa 已提交
33
use back::write::{self, OngoingCodegen, create_target_machine};
34
use llvm::{ContextRef, ModuleRef, ValueRef, Vector, get_param};
35
use llvm;
36
use metadata;
37
use rustc::hir::def_id::{CrateNum, DefId, LOCAL_CRATE};
38
use rustc::middle::lang_items::StartFnLangItem;
39
use rustc::middle::weak_lang_items;
40 41
use rustc::mir::mono::{Linkage, Visibility, Stats};
use rustc::middle::cstore::{EncodedMetadata};
42
use rustc::ty::{self, Ty, TyCtxt};
43
use rustc::ty::layout::{self, Align, TyLayout, LayoutOf};
44
use rustc::ty::query::Providers;
45
use rustc::dep_graph::{DepNode, DepConstructor};
46
use rustc::middle::cstore::{self, LinkMeta, LinkagePreference};
47
use rustc::middle::exported_symbols;
48
use rustc::util::common::{time, print_time_passes_entry};
A
Alex Crichton 已提交
49
use rustc::session::config::{self, NoDebugInfo};
50
use rustc::session::Session;
51
use rustc_incremental;
52
use allocator;
53
use mir::place::PlaceRef;
54
use attributes;
55
use builder::{Builder, MemFlags};
56
use callee;
57
use common::{C_bool, C_bytes_in_context, C_i32, C_usize};
58
use rustc_mir::monomorphize::collector::{self, MonoItemCollectionMode};
59
use rustc_mir::monomorphize::item::DefPathBasedNames;
60
use common::{self, C_struct_in_context, C_array, val_ty};
61
use consts;
62
use context::{self, CodegenCx};
M
Mark-Simulacrum 已提交
63
use debuginfo;
64 65 66
use declare;
use meth;
use mir;
67
use monomorphize::Instance;
M
Maik Klein 已提交
68
use monomorphize::partitioning::{self, PartitioningStrategy, CodegenUnit, CodegenUnitExt};
I
Irina Popa 已提交
69
use rustc_codegen_utils::symbol_names_test;
70
use time_graph;
71
use mono_item::{MonoItem, BaseMonoItemExt, MonoItemExt};
72
use type_::Type;
73
use type_of::LayoutLlvmExt;
74
use rustc::util::nodemap::{FxHashMap, FxHashSet, DefIdSet};
75
use CrateInfo;
76
use rustc_data_structures::sync::Lrc;
J
James Miller 已提交
77

78
use std::any::Any;
79
use std::ffi::CString;
A
Alex Crichton 已提交
80
use std::str;
81
use std::sync::Arc;
82
use std::time::{Instant, Duration};
V
varkor 已提交
83 84
use std::i32;
use std::cmp;
85
use std::sync::mpsc;
86
use syntax_pos::Span;
A
Alex Crichton 已提交
87
use syntax_pos::symbol::InternedString;
88
use syntax::attr;
89
use rustc::hir::{self, CodegenFnAttrs};
T
Tim Chevalier 已提交
90

91
use mir::operand::OperandValue;
92

B
bjorn3 已提交
93 94
use rustc_codegen_utils::check_for_rustc_errors_attr;

95
pub struct StatRecorder<'a, 'tcx: 'a> {
96
    cx: &'a CodegenCx<'a, 'tcx>,
97
    name: Option<String>,
98
    istart: usize,
99 100
}

101
impl<'a, 'tcx> StatRecorder<'a, 'tcx> {
102 103
    pub fn new(cx: &'a CodegenCx<'a, 'tcx>, name: String) -> StatRecorder<'a, 'tcx> {
        let istart = cx.stats.borrow().n_llvm_insns;
104
        StatRecorder {
105
            cx,
106
            name: Some(name),
107
            istart,
108 109 110 111
        }
    }
}

112
impl<'a, 'tcx> Drop for StatRecorder<'a, 'tcx> {
D
Daniel Micay 已提交
113
    fn drop(&mut self) {
I
Irina Popa 已提交
114
        if self.cx.sess().codegen_stats() {
115
            let mut stats = self.cx.stats.borrow_mut();
A
Alex Crichton 已提交
116 117 118
            let iend = stats.n_llvm_insns;
            stats.fn_stats.push((self.name.take().unwrap(), iend - self.istart));
            stats.n_fns += 1;
119
            // Reset LLVM insn count to avoid compound costs.
A
Alex Crichton 已提交
120
            stats.n_llvm_insns = self.istart;
121 122 123 124
        }
    }
}

C
csmoe 已提交
125
pub fn bin_op_to_icmp_predicate(op: hir::BinOpKind,
M
Ms2ger 已提交
126
                                signed: bool)
127 128
                                -> llvm::IntPredicate {
    match op {
C
csmoe 已提交
129 130 131 132 133 134
        hir::BinOpKind::Eq => llvm::IntEQ,
        hir::BinOpKind::Ne => llvm::IntNE,
        hir::BinOpKind::Lt => if signed { llvm::IntSLT } else { llvm::IntULT },
        hir::BinOpKind::Le => if signed { llvm::IntSLE } else { llvm::IntULE },
        hir::BinOpKind::Gt => if signed { llvm::IntSGT } else { llvm::IntUGT },
        hir::BinOpKind::Ge => if signed { llvm::IntSGE } else { llvm::IntUGE },
135
        op => {
136 137 138
            bug!("comparison_op_to_icmp_predicate: expected comparison operator, \
                  found {:?}",
                 op)
139 140 141
        }
    }
}
142

C
csmoe 已提交
143
pub fn bin_op_to_fcmp_predicate(op: hir::BinOpKind) -> llvm::RealPredicate {
144
    match op {
C
csmoe 已提交
145 146 147 148 149 150
        hir::BinOpKind::Eq => llvm::RealOEQ,
        hir::BinOpKind::Ne => llvm::RealUNE,
        hir::BinOpKind::Lt => llvm::RealOLT,
        hir::BinOpKind::Le => llvm::RealOLE,
        hir::BinOpKind::Gt => llvm::RealOGT,
        hir::BinOpKind::Ge => llvm::RealOGE,
151
        op => {
152 153 154
            bug!("comparison_op_to_fcmp_predicate: expected comparison operator, \
                  found {:?}",
                 op);
155 156 157 158
        }
    }
}

M
Mark Simulacrum 已提交
159
pub fn compare_simd_types<'a, 'tcx>(
160
    bx: &Builder<'a, 'tcx>,
M
Mark Simulacrum 已提交
161 162 163 164
    lhs: ValueRef,
    rhs: ValueRef,
    t: Ty<'tcx>,
    ret_ty: Type,
C
csmoe 已提交
165
    op: hir::BinOpKind
M
Mark Simulacrum 已提交
166
) -> ValueRef {
167
    let signed = match t.sty {
168
        ty::TyFloat(_) => {
169
            let cmp = bin_op_to_fcmp_predicate(op);
170
            return bx.sext(bx.fcmp(cmp, lhs, rhs), ret_ty);
171
        },
172 173
        ty::TyUint(_) => false,
        ty::TyInt(_) => true,
174
        _ => bug!("compare_simd_types: invalid SIMD type"),
175
    };
176

177
    let cmp = bin_op_to_icmp_predicate(op, signed);
178 179 180 181
    // LLVM outputs an `< size x i1 >`, so we need to perform a sign extension
    // to get the correctly sized type. This will compile to a single instruction
    // once the IR is converted to assembly if the SIMD instruction is supported
    // by the target architecture.
182
    bx.sext(bx.icmp(cmp, lhs, rhs), ret_ty)
183 184
}

A
Ariel Ben-Yehuda 已提交
185 186 187 188
/// Retrieve the information we are losing (making dynamic) in an unsizing
/// adjustment.
///
/// The `old_info` argument is a bit funny. It is intended for use
B
Bastien Orivel 已提交
189
/// in an upcast, where the new vtable for an object will be derived
A
Ariel Ben-Yehuda 已提交
190
/// from the old one.
191
pub fn unsized_info<'cx, 'tcx>(cx: &CodegenCx<'cx, 'tcx>,
A
Ariel Ben-Yehuda 已提交
192 193
                                source: Ty<'tcx>,
                                target: Ty<'tcx>,
194
                                old_info: Option<ValueRef>)
A
Ariel Ben-Yehuda 已提交
195
                                -> ValueRef {
196
    let (source, target) = cx.tcx.struct_lockstep_tails(source, target);
A
Ariel Ben-Yehuda 已提交
197
    match (&source.sty, &target.sty) {
198
        (&ty::TyArray(_, len), &ty::TySlice(_)) => {
199
            C_usize(cx, len.unwrap_usize(cx.tcx))
200
        }
201
        (&ty::TyDynamic(..), &ty::TyDynamic(..)) => {
A
Ariel Ben-Yehuda 已提交
202 203 204 205 206
            // For now, upcasts are limited to changes in marker
            // traits, and hence never actually require an actual
            // change to the vtable.
            old_info.expect("unsized_info: missing old info for trait upcast")
        }
207
        (_, &ty::TyDynamic(ref data, ..)) => {
208 209 210 211
            let vtable_ptr = cx.layout_of(cx.tcx.mk_mut_ptr(target))
                .field(cx, abi::FAT_PTR_EXTRA);
            consts::ptrcast(meth::get_vtable(cx, source, data.principal()),
                            vtable_ptr.llvm_type(cx))
A
Ariel Ben-Yehuda 已提交
212
        }
213
        _ => bug!("unsized_info: invalid unsizing {:?} -> {:?}",
A
Ariel Ben-Yehuda 已提交
214
                                     source,
215
                                     target),
A
Ariel Ben-Yehuda 已提交
216 217 218 219
    }
}

/// Coerce `src` to `dst_ty`. `src_ty` must be a thin pointer.
M
Mark Simulacrum 已提交
220
pub fn unsize_thin_ptr<'a, 'tcx>(
221
    bx: &Builder<'a, 'tcx>,
M
Mark Simulacrum 已提交
222 223 224 225
    src: ValueRef,
    src_ty: Ty<'tcx>,
    dst_ty: Ty<'tcx>
) -> (ValueRef, ValueRef) {
A
Ariel Ben-Yehuda 已提交
226 227
    debug!("unsize_thin_ptr: {:?} => {:?}", src_ty, dst_ty);
    match (&src_ty.sty, &dst_ty.sty) {
228 229 230
        (&ty::TyRef(_, a, _),
         &ty::TyRef(_, b, _)) |
        (&ty::TyRef(_, a, _),
A
Ariel Ben-Yehuda 已提交
231 232 233
         &ty::TyRawPtr(ty::TypeAndMut { ty: b, .. })) |
        (&ty::TyRawPtr(ty::TypeAndMut { ty: a, .. }),
         &ty::TyRawPtr(ty::TypeAndMut { ty: b, .. })) => {
234 235 236
            assert!(bx.cx.type_is_sized(a));
            let ptr_ty = bx.cx.layout_of(b).llvm_type(bx.cx).ptr_to();
            (bx.pointercast(src, ptr_ty), unsized_info(bx.cx, a, b, None))
A
Ariel Ben-Yehuda 已提交
237
        }
238 239
        (&ty::TyAdt(def_a, _), &ty::TyAdt(def_b, _)) if def_a.is_box() && def_b.is_box() => {
            let (a, b) = (src_ty.boxed_ty(), dst_ty.boxed_ty());
240 241 242
            assert!(bx.cx.type_is_sized(a));
            let ptr_ty = bx.cx.layout_of(b).llvm_type(bx.cx).ptr_to();
            (bx.pointercast(src, ptr_ty), unsized_info(bx.cx, a, b, None))
243
        }
244 245 246
        (&ty::TyAdt(def_a, _), &ty::TyAdt(def_b, _)) => {
            assert_eq!(def_a, def_b);

247 248
            let src_layout = bx.cx.layout_of(src_ty);
            let dst_layout = bx.cx.layout_of(dst_ty);
249 250
            let mut result = None;
            for i in 0..src_layout.fields.count() {
251
                let src_f = src_layout.field(bx.cx, i);
252 253 254 255 256 257 258
                assert_eq!(src_layout.fields.offset(i).bytes(), 0);
                assert_eq!(dst_layout.fields.offset(i).bytes(), 0);
                if src_f.is_zst() {
                    continue;
                }
                assert_eq!(src_layout.size, src_f.size);

259
                let dst_f = dst_layout.field(bx.cx, i);
260 261
                assert_ne!(src_f.ty, dst_f.ty);
                assert_eq!(result, None);
262
                result = Some(unsize_thin_ptr(bx, src, src_f.ty, dst_f.ty));
263 264 265
            }
            let (lldata, llextra) = result.unwrap();
            // HACK(eddyb) have to bitcast pointers until LLVM removes pointee types.
266 267
            (bx.bitcast(lldata, dst_layout.scalar_pair_element_llvm_type(bx.cx, 0, true)),
             bx.bitcast(llextra, dst_layout.scalar_pair_element_llvm_type(bx.cx, 1, true)))
268
        }
269
        _ => bug!("unsize_thin_ptr: called on bad types"),
A
Ariel Ben-Yehuda 已提交
270 271 272 273 274
    }
}

/// Coerce `src`, which is a reference to a value of type `src_ty`,
/// to a value of type `dst_ty` and store the result in `dst`
275
pub fn coerce_unsized_into<'a, 'tcx>(bx: &Builder<'a, 'tcx>,
276 277
                                     src: PlaceRef<'tcx>,
                                     dst: PlaceRef<'tcx>) {
278 279
    let src_ty = src.layout.ty;
    let dst_ty = dst.layout.ty;
280
    let coerce_ptr = || {
281
        let (base, info) = match src.load(bx).val {
282 283 284 285 286
            OperandValue::Pair(base, info) => {
                // fat-ptr to fat-ptr unsize preserves the vtable
                // i.e. &'a fmt::Debug+Send => &'a fmt::Debug
                // So we need to pointercast the base to ensure
                // the types match up.
287 288
                let thin_ptr = dst.layout.field(bx.cx, abi::FAT_PTR_ADDR);
                (bx.pointercast(base, thin_ptr.llvm_type(bx.cx)), info)
289 290
            }
            OperandValue::Immediate(base) => {
291
                unsize_thin_ptr(bx, base, src_ty, dst_ty)
292 293
            }
            OperandValue::Ref(..) => bug!()
294
        };
295
        OperandValue::Pair(base, info).store(bx, dst);
296
    };
A
Ariel Ben-Yehuda 已提交
297 298 299 300
    match (&src_ty.sty, &dst_ty.sty) {
        (&ty::TyRef(..), &ty::TyRef(..)) |
        (&ty::TyRef(..), &ty::TyRawPtr(..)) |
        (&ty::TyRawPtr(..), &ty::TyRawPtr(..)) => {
301 302 303 304
            coerce_ptr()
        }
        (&ty::TyAdt(def_a, _), &ty::TyAdt(def_b, _)) if def_a.is_box() && def_b.is_box() => {
            coerce_ptr()
A
Ariel Ben-Yehuda 已提交
305 306
        }

307
        (&ty::TyAdt(def_a, _), &ty::TyAdt(def_b, _)) => {
A
Ariel Ben-Yehuda 已提交
308 309
            assert_eq!(def_a, def_b);

310
            for i in 0..def_a.variants[0].fields.len() {
311 312
                let src_f = src.project_field(bx, i);
                let dst_f = dst.project_field(bx, i);
A
Ariel Ben-Yehuda 已提交
313

314
                if dst_f.layout.is_zst() {
M
Ms2ger 已提交
315 316
                    continue;
                }
A
Ariel Ben-Yehuda 已提交
317

318
                if src_f.layout.ty == dst_f.layout.ty {
319
                    memcpy_ty(bx, dst_f.llval, src_f.llval, src_f.layout,
320
                              src_f.align.min(dst_f.align), MemFlags::empty());
A
Ariel Ben-Yehuda 已提交
321
                } else {
322
                    coerce_unsized_into(bx, src_f, dst_f);
A
Ariel Ben-Yehuda 已提交
323 324 325
                }
            }
        }
326 327 328
        _ => bug!("coerce_unsized_into: invalid coercion {:?} -> {:?}",
                  src_ty,
                  dst_ty),
A
Ariel Ben-Yehuda 已提交
329 330 331
    }
}

332
pub fn cast_shift_expr_rhs(
C
csmoe 已提交
333
    cx: &Builder, op: hir::BinOpKind, lhs: ValueRef, rhs: ValueRef
334 335
) -> ValueRef {
    cast_shift_rhs(op, lhs, rhs, |a, b| cx.trunc(a, b), |a, b| cx.zext(a, b))
336 337
}

C
csmoe 已提交
338
fn cast_shift_rhs<F, G>(op: hir::BinOpKind,
339 340 341 342
                        lhs: ValueRef,
                        rhs: ValueRef,
                        trunc: F,
                        zext: G)
M
Ms2ger 已提交
343 344 345
                        -> ValueRef
    where F: FnOnce(ValueRef, Type) -> ValueRef,
          G: FnOnce(ValueRef, Type) -> ValueRef
346
{
347
    // Shifts may have any size int on the rhs
348
    if op.is_shift() {
349 350
        let mut rhs_llty = val_ty(rhs);
        let mut lhs_llty = val_ty(lhs);
M
Ms2ger 已提交
351 352 353 354 355 356
        if rhs_llty.kind() == Vector {
            rhs_llty = rhs_llty.element_type()
        }
        if lhs_llty.kind() == Vector {
            lhs_llty = lhs_llty.element_type()
        }
357 358 359 360 361 362 363 364
        let rhs_sz = rhs_llty.int_width();
        let lhs_sz = lhs_llty.int_width();
        if lhs_sz < rhs_sz {
            trunc(rhs, lhs_llty)
        } else if lhs_sz > rhs_sz {
            // FIXME (#1877: If shifting by negative
            // values becomes not undefined then this is wrong.
            zext(rhs, lhs_llty)
365 366 367
        } else {
            rhs
        }
368 369
    } else {
        rhs
370 371 372
    }
}

373 374 375 376 377 378
/// Returns whether this session's target will use SEH-based unwinding.
///
/// This is only true for MSVC targets, and even then the 64-bit MSVC target
/// currently uses SEH-ish unwinding with DWARF info tables to the side (same as
/// 64-bit MinGW) instead of "full SEH".
pub fn wants_msvc_seh(sess: &Session) -> bool {
379
    sess.target.target.options.is_like_msvc
380 381
}

382 383 384
pub fn call_assume<'a, 'tcx>(bx: &Builder<'a, 'tcx>, val: ValueRef) {
    let assume_intrinsic = bx.cx.get_intrinsic("llvm.assume");
    bx.call(assume_intrinsic, &[val], None);
385 386
}

387 388 389
pub fn from_immediate(bx: &Builder, val: ValueRef) -> ValueRef {
    if val_ty(val) == Type::i1(bx.cx) {
        bx.zext(val, Type::i8(bx.cx))
390 391 392 393 394
    } else {
        val
    }
}

395
pub fn to_immediate(bx: &Builder, val: ValueRef, layout: layout::TyLayout) -> ValueRef {
396
    if let layout::Abi::Scalar(ref scalar) = layout.abi {
397 398 399 400 401 402 403 404
        return to_immediate_scalar(bx, val, scalar);
    }
    val
}

pub fn to_immediate_scalar(bx: &Builder, val: ValueRef, scalar: &layout::Scalar) -> ValueRef {
    if scalar.is_bool() {
        return bx.trunc(val, Type::i1(bx.cx));
405
    }
406
    val
407 408
}

409
pub fn call_memcpy(bx: &Builder,
410 411 412
                   dst: ValueRef,
                   src: ValueRef,
                   n_bytes: ValueRef,
413
                   align: Align,
414 415 416 417 418 419 420 421
                   flags: MemFlags) {
    if flags.contains(MemFlags::NONTEMPORAL) {
        // HACK(nox): This is inefficient but there is no nontemporal memcpy.
        let val = bx.load(src, align);
        let ptr = bx.pointercast(dst, val_ty(val).ptr_to());
        bx.store_with_flags(val, ptr, align, flags);
        return;
    }
422
    let cx = bx.cx;
423
    let ptr_width = &cx.sess().target.target.target_pointer_width;
424
    let key = format!("llvm.memcpy.p0i8.p0i8.i{}", ptr_width);
425
    let memcpy = cx.get_intrinsic(&key);
426 427 428
    let src_ptr = bx.pointercast(src, Type::i8p(cx));
    let dst_ptr = bx.pointercast(dst, Type::i8p(cx));
    let size = bx.intcast(n_bytes, cx.isize_ty, false);
429
    let align = C_i32(cx, align.abi() as i32);
430
    let volatile = C_bool(cx, flags.contains(MemFlags::VOLATILE));
431
    bx.call(memcpy, &[dst_ptr, src_ptr, size, align, volatile], None);
432 433
}

434
pub fn memcpy_ty<'a, 'tcx>(
435
    bx: &Builder<'a, 'tcx>,
436 437
    dst: ValueRef,
    src: ValueRef,
438
    layout: TyLayout<'tcx>,
439
    align: Align,
440
    flags: MemFlags,
441
) {
442
    let size = layout.size.bytes();
443
    if size == 0 {
444 445 446
        return;
    }

447
    call_memcpy(bx, dst, src, C_usize(bx.cx, size), align, flags);
448 449
}

450
pub fn call_memset<'a, 'tcx>(bx: &Builder<'a, 'tcx>,
451 452 453 454 455
                             ptr: ValueRef,
                             fill_byte: ValueRef,
                             size: ValueRef,
                             align: ValueRef,
                             volatile: bool) -> ValueRef {
456
    let ptr_width = &bx.cx.sess().target.target.target_pointer_width;
457
    let intrinsic_key = format!("llvm.memset.p0i8.i{}", ptr_width);
458 459 460
    let llintrinsicfn = bx.cx.get_intrinsic(&intrinsic_key);
    let volatile = C_bool(bx.cx, volatile);
    bx.call(llintrinsicfn, &[ptr, fill_byte, size, align, volatile], None)
461 462
}

I
Irina Popa 已提交
463 464
pub fn codegen_instance<'a, 'tcx>(cx: &CodegenCx<'a, 'tcx>, instance: Instance<'tcx>) {
    let _s = if cx.sess().codegen_stats() {
465
        let mut instance_name = String::new();
466
        DefPathBasedNames::new(cx.tcx, true, true)
467
            .push_def_path(instance.def_id(), &mut instance_name);
468
        Some(StatRecorder::new(cx, instance_name))
469 470 471 472
    } else {
        None
    };

473 474 475
    // this is an info! to allow collecting monomorphization statistics
    // and to allow finding the last function before LLVM aborts from
    // release builds.
I
Irina Popa 已提交
476
    info!("codegen_instance({})", instance);
477

478 479
    let fn_ty = instance.ty(cx.tcx);
    let sig = common::ty_fn_sig(cx, fn_ty);
480
    let sig = cx.tcx.normalize_erasing_late_bound_regions(ty::ParamEnv::reveal_all(), &sig);
481

482
    let lldecl = match cx.instances.borrow().get(&instance) {
483 484 485 486
        Some(&val) => val,
        None => bug!("Instance `{:?}` not already declared", instance)
    };

487
    cx.stats.borrow_mut().n_closures += 1;
488

489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504
    // The `uwtable` attribute according to LLVM is:
    //
    //     This attribute indicates that the ABI being targeted requires that an
    //     unwind table entry be produced for this function even if we can show
    //     that no exceptions passes by it. This is normally the case for the
    //     ELF x86-64 abi, but it can be disabled for some compilation units.
    //
    // Typically when we're compiling with `-C panic=abort` (which implies this
    // `no_landing_pads` check) we don't need `uwtable` because we can't
    // generate any exceptions! On Windows, however, exceptions include other
    // events such as illegal instructions, segfaults, etc. This means that on
    // Windows we end up still needing the `uwtable` attribute even if the `-C
    // panic=abort` flag is passed.
    //
    // You can also find more info on why Windows is whitelisted here in:
    //      https://bugzilla.mozilla.org/show_bug.cgi?id=1302078
505
    if !cx.sess().no_landing_pads() ||
506
       cx.sess().target.target.options.requires_uwtable {
507
        attributes::emit_uwtable(lldecl, true);
508
    }
509

510
    let mir = cx.tcx.instance_mir(instance.def);
I
Irina Popa 已提交
511
    mir::codegen_mir(cx, lldecl, &mir, instance, sig);
512 513
}

514 515 516 517 518 519 520 521
pub fn set_link_section(llval: ValueRef, attrs: &CodegenFnAttrs) {
    let sect = match attrs.link_section {
        Some(name) => name,
        None => return,
    };
    unsafe {
        let buf = CString::new(sect.as_str().as_bytes()).unwrap();
        llvm::LLVMSetSection(llval, buf.as_ptr());
E
Eli Friedman 已提交
522 523 524
    }
}

F
Fourchaux 已提交
525
/// Create the `main` function which will initialize the rust runtime and call
526
/// users main function.
527 528
fn maybe_create_entry_wrapper(cx: &CodegenCx) {
    let (main_def_id, span) = match *cx.sess().entry_fn.borrow() {
529
        Some((id, span, _)) => {
530
            (cx.tcx.hir.local_def_id(id), span)
531 532 533 534
        }
        None => return,
    };

535
    let instance = Instance::mono(cx.tcx, main_def_id);
536

537
    if !cx.codegen_unit.contains_item(&MonoItem::Fn(instance)) {
538 539 540
        // We want to create the wrapper in the same codegen unit as Rust's main
        // function.
        return;
541 542
    }

543
    let main_llfn = callee::get_fn(cx, instance);
544

545
    let et = cx.sess().entry_fn.get().map(|e| e.2);
546
    match et {
547 548 549
        Some(config::EntryMain) => create_entry_fn(cx, span, main_llfn, main_def_id, true),
        Some(config::EntryStart) => create_entry_fn(cx, span, main_llfn, main_def_id, false),
        None => {}    // Do nothing.
550
    }
551

552
    fn create_entry_fn<'cx>(cx: &'cx CodegenCx,
S
Simonas Kazlauskas 已提交
553
                       sp: Span,
554
                       rust_main: ValueRef,
555
                       rust_main_def_id: DefId,
556
                       use_start_lang_item: bool) {
557
        let llfty = Type::func(&[Type::c_int(cx), Type::i8p(cx).ptr_to()], &Type::c_int(cx));
558

559
        let main_ret_ty = cx.tcx.fn_sig(rust_main_def_id).output();
560 561 562 563 564
        // Given that `main()` has no arguments,
        // then its return type cannot have
        // late-bound regions, since late-bound
        // regions must appear in the argument
        // listing.
565 566 567
        let main_ret_ty = cx.tcx.erase_regions(
            &main_ret_ty.no_late_bound_regions().unwrap(),
        );
K
kyeongwoon 已提交
568

569
        if declare::get_defined_value(cx, "main").is_some() {
S
Simonas Kazlauskas 已提交
570
            // FIXME: We should be smart and show a better diagnostic here.
571
            cx.sess().struct_span_err(sp, "entry symbol `main` defined multiple times")
N
Nick Cameron 已提交
572 573
                      .help("did you use #[no_mangle] on `fn main`? Use #[start] instead")
                      .emit();
574
            cx.sess().abort_if_errors();
575
            bug!();
576
        }
577
        let llfn = declare::declare_cfn(cx, "main", llfty);
578

579
        // `main` should respect same config for frame pointer elimination as rest of code
580
        attributes::set_frame_pointer_elimination(cx, llfn);
581

582
        let bx = Builder::new_block(cx, llfn, "top");
583

584
        debuginfo::gdb::insert_reference_to_gdb_debug_scripts_section_global(&bx);
585

586 587 588
        // Params from native main() used as args for rust start function
        let param_argc = get_param(llfn, 0);
        let param_argv = get_param(llfn, 1);
589
        let arg_argc = bx.intcast(param_argc, cx.isize_ty, true);
590 591
        let arg_argv = param_argv;

M
Mark-Simulacrum 已提交
592
        let (start_fn, args) = if use_start_lang_item {
593
            let start_def_id = cx.tcx.require_lang_item(StartFnLangItem);
594 595 596
            let start_fn = callee::resolve_and_get_fn(
                cx,
                start_def_id,
597
                cx.tcx.intern_substs(&[main_ret_ty.into()]),
598
            );
599
            (start_fn, vec![bx.pointercast(rust_main, Type::i8p(cx).ptr_to()),
600
                            arg_argc, arg_argv])
M
Mark-Simulacrum 已提交
601 602
        } else {
            debug!("using user-defined start fn");
603
            (rust_main, vec![arg_argc, arg_argv])
M
Mark-Simulacrum 已提交
604
        };
605

606 607
        let result = bx.call(start_fn, &args, None);
        bx.ret(bx.intcast(result, Type::c_int(cx), true));
M
Marijn Haverbeke 已提交
608
    }
609 610
}

611
fn write_metadata<'a, 'gcx>(tcx: TyCtxt<'a, 'gcx, 'gcx>,
612
                            llmod_id: &str,
613
                            link_meta: &LinkMeta)
614
                            -> (ContextRef, ModuleRef, EncodedMetadata) {
A
Alex Crichton 已提交
615 616
    use std::io::Write;
    use flate2::Compression;
617
    use flate2::write::DeflateEncoder;
618

619
    let (metadata_llcx, metadata_llmod) = unsafe {
620
        context::create_context_and_module(tcx.sess, llmod_id)
621 622
    };

N
Nicholas Nethercote 已提交
623 624 625 626 627 628 629
    #[derive(PartialEq, Eq, PartialOrd, Ord)]
    enum MetadataKind {
        None,
        Uncompressed,
        Compressed
    }

630
    let kind = tcx.sess.crate_types.borrow().iter().map(|ty| {
N
Nicholas Nethercote 已提交
631 632 633 634 635
        match *ty {
            config::CrateTypeExecutable |
            config::CrateTypeStaticlib |
            config::CrateTypeCdylib => MetadataKind::None,

636
            config::CrateTypeRlib => MetadataKind::Uncompressed,
N
Nicholas Nethercote 已提交
637 638 639 640

            config::CrateTypeDylib |
            config::CrateTypeProcMacro => MetadataKind::Compressed,
        }
641
    }).max().unwrap_or(MetadataKind::None);
N
Nicholas Nethercote 已提交
642 643

    if kind == MetadataKind::None {
644 645
        return (metadata_llcx,
                metadata_llmod,
646
                EncodedMetadata::new());
647
    }
J
James Miller 已提交
648

649
    let metadata = tcx.encode_metadata(link_meta);
N
Nicholas Nethercote 已提交
650
    if kind == MetadataKind::Uncompressed {
651
        return (metadata_llcx, metadata_llmod, metadata);
N
Nicholas Nethercote 已提交
652 653 654
    }

    assert!(kind == MetadataKind::Compressed);
655
    let mut compressed = tcx.metadata_encoding_version();
656
    DeflateEncoder::new(&mut compressed, Compression::fast())
A
Alex Crichton 已提交
657
        .write_all(&metadata.raw_data).unwrap();
658

659 660
    let llmeta = C_bytes_in_context(metadata_llcx, &compressed);
    let llconst = C_struct_in_context(metadata_llcx, &[llmeta], false);
661
    let name = exported_symbols::metadata_symbol_name(tcx);
662
    let buf = CString::new(name).unwrap();
A
Alex Crichton 已提交
663
    let llglobal = unsafe {
664
        llvm::LLVMAddGlobal(metadata_llmod, val_ty(llconst).to_ref(), buf.as_ptr())
A
Alex Crichton 已提交
665
    };
666 667
    unsafe {
        llvm::LLVMSetInitializer(llglobal, llconst);
668
        let section_name = metadata::metadata_section_name(&tcx.sess.target.target);
669 670 671 672 673 674 675 676
        let name = CString::new(section_name).unwrap();
        llvm::LLVMSetSection(llglobal, name.as_ptr());

        // Also generate a .section directive to force no
        // flags, at least for ELF outputs, so that the
        // metadata doesn't get loaded into memory.
        let directive = format!(".section {}", section_name);
        let directive = CString::new(directive).unwrap();
677
        llvm::LLVMSetModuleInlineAsm(metadata_llmod, directive.as_ptr())
678
    }
679
    return (metadata_llcx, metadata_llmod, metadata);
680 681
}

682
pub struct ValueIter {
683 684 685
    cur: ValueRef,
    step: unsafe extern "C" fn(ValueRef) -> ValueRef,
}
686

687 688
impl Iterator for ValueIter {
    type Item = ValueRef;
689

690 691 692
    fn next(&mut self) -> Option<ValueRef> {
        let old = self.cur;
        if !old.is_null() {
693
            self.cur = unsafe { (self.step)(old) };
694 695 696 697
            Some(old)
        } else {
            None
        }
698
    }
699
}
700

701
pub fn iter_globals(llmod: llvm::ModuleRef) -> ValueIter {
702 703 704 705 706 707 708
    unsafe {
        ValueIter {
            cur: llvm::LLVMGetFirstGlobal(llmod),
            step: llvm::LLVMGetNextGlobal,
        }
    }
}
J
Jorge Aparicio 已提交
709

I
Irina Popa 已提交
710
pub fn codegen_crate<'a, 'tcx>(tcx: TyCtxt<'a, 'tcx, 'tcx>,
711
                             rx: mpsc::Receiver<Box<dyn Any + Send>>)
I
Irina Popa 已提交
712
                             -> OngoingCodegen {
B
bjorn3 已提交
713

B
bjorn3 已提交
714
    check_for_rustc_errors_attr(tcx);
715

716
    if let Some(true) = tcx.sess.opts.debugging_opts.thinlto {
A
Alex Crichton 已提交
717 718 719 720
        if unsafe { !llvm::LLVMRustThinLTOAvailable() } {
            tcx.sess.fatal("this compiler's LLVM does not support ThinLTO");
        }
    }
721

722 723 724 725 726 727 728
    if (tcx.sess.opts.debugging_opts.pgo_gen.is_some() ||
        !tcx.sess.opts.debugging_opts.pgo_use.is_empty()) &&
        unsafe { !llvm::LLVMRustPGOAvailable() }
    {
        tcx.sess.fatal("this compiler's LLVM does not support PGO");
    }

729
    let crate_hash = tcx.crate_hash(LOCAL_CRATE);
730
    let link_meta = link::build_link_meta(crate_hash);
731

I
Irina Popa 已提交
732
    // Codegen the metadata.
733
    let llmod_id = "metadata";
734
    let (metadata_llcx, metadata_llmod, metadata) =
735
        time(tcx.sess, "write metadata", || {
736
            write_metadata(tcx, llmod_id, &link_meta)
737
        });
738

I
Irina Popa 已提交
739
    let metadata_module = ModuleCodegen {
740 741
        name: link::METADATA_MODULE_NAME.to_string(),
        llmod_id: llmod_id.to_string(),
I
Irina Popa 已提交
742
        source: ModuleSource::Codegened(ModuleLlvm {
743 744
            llcx: metadata_llcx,
            llmod: metadata_llmod,
745
            tm: create_target_machine(tcx.sess, false),
746
        }),
747
        kind: ModuleKind::Metadata,
748
    };
749

I
Irina Popa 已提交
750
    let time_graph = if tcx.sess.opts.debugging_opts.codegen_time_graph {
751 752 753 754
        Some(time_graph::TimeGraph::new())
    } else {
        None
    };
755

I
Irina Popa 已提交
756 757 758 759
    // Skip crate items and just output metadata in -Z no-codegen mode.
    if tcx.sess.opts.debugging_opts.no_codegen ||
       !tcx.sess.opts.output_types.should_codegen() {
        let ongoing_codegen = write::start_async_codegen(
760
            tcx,
761
            time_graph.clone(),
762
            link_meta,
763
            metadata,
A
Alex Crichton 已提交
764 765
            rx,
            1);
766

I
Irina Popa 已提交
767 768
        ongoing_codegen.submit_pre_codegened_module_to_llvm(tcx, metadata_module);
        ongoing_codegen.codegen_finished(tcx);
769

770
        assert_and_save_dep_graph(tcx);
771

I
Irina Popa 已提交
772
        ongoing_codegen.check_for_errors(tcx.sess);
773

I
Irina Popa 已提交
774
        return ongoing_codegen;
775 776
    }

I
Irina Popa 已提交
777
    // Run the monomorphization collector and partition the collected items into
778
    // codegen units.
779
    let codegen_units =
I
Irina Popa 已提交
780
        tcx.collect_and_partition_mono_items(LOCAL_CRATE).1;
781
    let codegen_units = (*codegen_units).clone();
782

783 784 785 786 787 788 789 790 791 792 793
    // Force all codegen_unit queries so they are already either red or green
    // when compile_codegen_unit accesses them. We are not able to re-execute
    // the codegen_unit query from just the DepNode, so an unknown color would
    // lead to having to re-execute compile_codegen_unit, possibly
    // unnecessarily.
    if tcx.dep_graph.is_fully_enabled() {
        for cgu in &codegen_units {
            tcx.codegen_unit(cgu.name().clone());
        }
    }

I
Irina Popa 已提交
794
    let ongoing_codegen = write::start_async_codegen(
795
        tcx,
796
        time_graph.clone(),
797
        link_meta,
798
        metadata,
A
Alex Crichton 已提交
799 800
        rx,
        codegen_units.len());
801

I
Irina Popa 已提交
802
    // Codegen an allocator shim, if any
803
    let allocator_module = if let Some(kind) = *tcx.sess.allocator_kind.get() {
804 805 806 807 808 809 810 811 812 813
        unsafe {
            let llmod_id = "allocator";
            let (llcx, llmod) =
                context::create_context_and_module(tcx.sess, llmod_id);
            let modules = ModuleLlvm {
                llmod,
                llcx,
                tm: create_target_machine(tcx.sess, false),
            };
            time(tcx.sess, "write allocator module", || {
I
Irina Popa 已提交
814
                allocator::codegen(tcx, &modules, kind)
815
            });
816

817 818 819 820 821 822 823
            Some(ModuleCodegen {
                name: link::ALLOCATOR_MODULE_NAME.to_string(),
                llmod_id: llmod_id.to_string(),
                source: ModuleSource::Codegened(modules),
                kind: ModuleKind::Allocator,
            })
        }
824 825 826 827 828
    } else {
        None
    };

    if let Some(allocator_module) = allocator_module {
I
Irina Popa 已提交
829
        ongoing_codegen.submit_pre_codegened_module_to_llvm(tcx, allocator_module);
830 831
    }

I
Irina Popa 已提交
832
    ongoing_codegen.submit_pre_codegened_module_to_llvm(tcx, metadata_module);
833

834
    // We sort the codegen units by size. This way we can schedule work for LLVM
835
    // a bit more efficiently.
836 837
    let codegen_units = {
        let mut codegen_units = codegen_units;
V
varkor 已提交
838
        codegen_units.sort_by_cached_key(|cgu| cmp::Reverse(cgu.size_estimate()));
839 840 841
        codegen_units
    };

I
Irina Popa 已提交
842
    let mut total_codegen_time = Duration::new(0, 0);
A
Alex Crichton 已提交
843
    let mut all_stats = Stats::default();
844

A
Alex Crichton 已提交
845
    for cgu in codegen_units.into_iter() {
I
Irina Popa 已提交
846 847
        ongoing_codegen.wait_for_signal_to_codegen_item();
        ongoing_codegen.check_for_errors(tcx.sess);
848

849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865
        // First, if incremental compilation is enabled, we try to re-use the
        // codegen unit from the cache.
        if tcx.dep_graph.is_fully_enabled() {
            let cgu_id = cgu.work_product_id();

            // Check whether there is a previous work-product we can
            // re-use.  Not only must the file exist, and the inputs not
            // be dirty, but the hash of the symbols we will generate must
            // be the same.
            if let Some(buf) = tcx.dep_graph.previous_work_product(&cgu_id) {
                let dep_node = &DepNode::new(tcx,
                    DepConstructor::CompileCodegenUnit(cgu.name().clone()));

                // We try to mark the DepNode::CompileCodegenUnit green. If we
                // succeed it means that none of the dependencies has changed
                // and we can safely re-use.
                if let Some(dep_node_index) = tcx.dep_graph.try_mark_green(tcx, dep_node) {
866 867 868 869 870 871 872 873 874 875
                    // Append ".rs" to LLVM module identifier.
                    //
                    // LLVM code generator emits a ".file filename" directive
                    // for ELF backends. Value of the "filename" is set as the
                    // LLVM module identifier.  Due to a LLVM MC bug[1], LLVM
                    // crashes if the module identifier is same as other symbols
                    // such as a function name in the module.
                    // 1. http://llvm.org/bugs/show_bug.cgi?id=11479
                    let llmod_id = format!("{}.rs", cgu.name());

I
Irina Popa 已提交
876
                    let module = ModuleCodegen {
877 878 879
                        name: cgu.name().to_string(),
                        source: ModuleSource::Preexisting(buf),
                        kind: ModuleKind::Regular,
880
                        llmod_id,
881 882
                    };
                    tcx.dep_graph.mark_loaded_from_cache(dep_node_index, true);
I
Irina Popa 已提交
883
                    write::submit_codegened_module_to_llvm(tcx, module, 0);
884 885 886 887 888 889 890 891 892
                    // Continue to next cgu, this one is done.
                    continue
                }
            } else {
                // This can happen if files were  deleted from the cache
                // directory for some reason. We just re-compile then.
            }
        }

893
        let _timing_guard = time_graph.as_ref().map(|time_graph| {
I
Irina Popa 已提交
894 895
            time_graph.start(write::CODEGEN_WORKER_TIMELINE,
                             write::CODEGEN_WORK_PACKAGE_KIND,
896 897
                             &format!("codegen {}", cgu.name()))
        });
898
        let start_time = Instant::now();
A
Alex Crichton 已提交
899
        all_stats.extend(tcx.compile_codegen_unit(*cgu.name()));
I
Irina Popa 已提交
900 901
        total_codegen_time += start_time.elapsed();
        ongoing_codegen.check_for_errors(tcx.sess);
902 903
    }

I
Irina Popa 已提交
904
    ongoing_codegen.codegen_finished(tcx);
A
Alex Crichton 已提交
905

I
Irina Popa 已提交
906
    // Since the main thread is sometimes blocked during codegen, we keep track
907 908
    // -Ztime-passes output manually.
    print_time_passes_entry(tcx.sess.time_passes(),
I
Irina Popa 已提交
909 910
                            "codegen to LLVM IR",
                            total_codegen_time);
911

A
Alex Crichton 已提交
912
    if tcx.sess.opts.incremental.is_some() {
913
        ::rustc_incremental::assert_module_sources::assert_module_sources(tcx);
914 915
    }

N
Niko Matsakis 已提交
916
    symbol_names_test::report_symbol_names(tcx);
917

I
Irina Popa 已提交
918 919
    if tcx.sess.codegen_stats() {
        println!("--- codegen stats ---");
A
Alex Crichton 已提交
920 921 922
        println!("n_glues_created: {}", all_stats.n_glues_created);
        println!("n_null_glues: {}", all_stats.n_null_glues);
        println!("n_real_glues: {}", all_stats.n_real_glues);
923

A
Alex Crichton 已提交
924 925 926
        println!("n_fns: {}", all_stats.n_fns);
        println!("n_inlines: {}", all_stats.n_inlines);
        println!("n_closures: {}", all_stats.n_closures);
927
        println!("fn stats:");
A
Alex Crichton 已提交
928 929 930
        all_stats.fn_stats.sort_by_key(|&(_, insns)| insns);
        for &(ref name, insns) in all_stats.fn_stats.iter() {
            println!("{} insns, {}", insns, *name);
931
        }
J
James Miller 已提交
932
    }
933

934
    if tcx.sess.count_llvm_insns() {
A
Alex Crichton 已提交
935
        for (k, v) in all_stats.llvm_insns.iter() {
A
Alex Crichton 已提交
936
            println!("{:7} {}", *v, *k);
937 938 939
        }
    }

I
Irina Popa 已提交
940
    ongoing_codegen.check_for_errors(tcx.sess);
941

942
    assert_and_save_dep_graph(tcx);
I
Irina Popa 已提交
943
    ongoing_codegen
944
}
945

946
fn assert_and_save_dep_graph<'a, 'tcx>(tcx: TyCtxt<'a, 'tcx, 'tcx>) {
947
    time(tcx.sess,
948 949 950
         "assert dep graph",
         || rustc_incremental::assert_dep_graph(tcx));

951
    time(tcx.sess,
952
         "serialize dep graph",
953
         || rustc_incremental::save_dep_graph(tcx));
954 955
}

I
Irina Popa 已提交
956
fn collect_and_partition_mono_items<'a, 'tcx>(
957 958
    tcx: TyCtxt<'a, 'tcx, 'tcx>,
    cnum: CrateNum,
959
) -> (Arc<DefIdSet>, Arc<Vec<Arc<CodegenUnit<'tcx>>>>)
960 961
{
    assert_eq!(cnum, LOCAL_CRATE);
962

I
Irina Popa 已提交
963
    let collection_mode = match tcx.sess.opts.debugging_opts.print_mono_items {
964 965 966 967
        Some(ref s) => {
            let mode_string = s.to_lowercase();
            let mode_string = mode_string.trim();
            if mode_string == "eager" {
968
                MonoItemCollectionMode::Eager
969 970 971 972 973
            } else {
                if mode_string != "lazy" {
                    let message = format!("Unknown codegen-item collection mode '{}'. \
                                           Falling back to 'lazy' mode.",
                                           mode_string);
974
                    tcx.sess.warn(&message);
975 976
                }

977
                MonoItemCollectionMode::Lazy
978 979
            }
        }
980 981 982 983 984 985 986
        None => {
            if tcx.sess.opts.cg.link_dead_code {
                MonoItemCollectionMode::Eager
            } else {
                MonoItemCollectionMode::Lazy
            }
        }
987 988
    };

989
    let (items, inlining_map) =
I
Irina Popa 已提交
990
        time(tcx.sess, "monomorphization collection", || {
991
            collector::collect_crate_mono_items(tcx, collection_mode)
992 993
    });

994 995
    tcx.sess.abort_if_errors();

996
    ::rustc_mir::monomorphize::assert_symbols_are_distinct(tcx, items.iter());
997

998
    let strategy = if tcx.sess.opts.incremental.is_some() {
999 1000
        PartitioningStrategy::PerModule
    } else {
1001
        PartitioningStrategy::FixedUnitCount(tcx.sess.codegen_units())
1002 1003
    };

1004
    let codegen_units = time(tcx.sess, "codegen unit partitioning", || {
1005
        partitioning::partition(tcx,
1006 1007
                                items.iter().cloned(),
                                strategy,
1008
                                &inlining_map)
1009 1010 1011
            .into_iter()
            .map(Arc::new)
            .collect::<Vec<_>>()
1012 1013
    });

I
Irina Popa 已提交
1014 1015
    let mono_items: DefIdSet = items.iter().filter_map(|mono_item| {
        match *mono_item {
M
Maik Klein 已提交
1016
            MonoItem::Fn(ref instance) => Some(instance.def_id()),
1017
            MonoItem::Static(def_id) => Some(def_id),
1018 1019 1020
            _ => None,
        }
    }).collect();
1021

I
Irina Popa 已提交
1022
    if tcx.sess.opts.debugging_opts.print_mono_items.is_some() {
1023
        let mut item_to_cgus = FxHashMap();
1024

1025
        for cgu in &codegen_units {
I
Irina Popa 已提交
1026 1027
            for (&mono_item, &linkage) in cgu.items() {
                item_to_cgus.entry(mono_item)
1028
                            .or_insert(Vec::new())
1029
                            .push((cgu.name().clone(), linkage));
1030 1031 1032 1033 1034 1035
            }
        }

        let mut item_keys: Vec<_> = items
            .iter()
            .map(|i| {
1036
                let mut output = i.to_string(tcx);
1037 1038
                output.push_str(" @@");
                let mut empty = Vec::new();
1039
                let cgus = item_to_cgus.get_mut(i).unwrap_or(&mut empty);
1040 1041
                cgus.as_mut_slice().sort_by_key(|&(ref name, _)| name.clone());
                cgus.dedup();
1042
                for &(ref cgu_name, (linkage, _)) in cgus.iter() {
1043
                    output.push_str(" ");
1044
                    output.push_str(&cgu_name.as_str());
1045 1046

                    let linkage_abbrev = match linkage {
1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057
                        Linkage::External => "External",
                        Linkage::AvailableExternally => "Available",
                        Linkage::LinkOnceAny => "OnceAny",
                        Linkage::LinkOnceODR => "OnceODR",
                        Linkage::WeakAny => "WeakAny",
                        Linkage::WeakODR => "WeakODR",
                        Linkage::Appending => "Appending",
                        Linkage::Internal => "Internal",
                        Linkage::Private => "Private",
                        Linkage::ExternalWeak => "ExternalWeak",
                        Linkage::Common => "Common",
1058 1059 1060 1061 1062 1063 1064 1065 1066 1067
                    };

                    output.push_str("[");
                    output.push_str(linkage_abbrev);
                    output.push_str("]");
                }
                output
            })
            .collect();

1068 1069 1070
        item_keys.sort();

        for item in item_keys {
I
Irina Popa 已提交
1071
            println!("MONO_ITEM {}", item);
1072 1073
        }
    }
1074

I
Irina Popa 已提交
1075
    (Arc::new(mono_items), Arc::new(codegen_units))
1076
}
1077 1078

impl CrateInfo {
1079
    pub fn new(tcx: TyCtxt) -> CrateInfo {
1080 1081 1082 1083 1084 1085
        let mut info = CrateInfo {
            panic_runtime: None,
            compiler_builtins: None,
            profiler_runtime: None,
            sanitizer_runtime: None,
            is_no_builtins: FxHashSet(),
1086
            native_libraries: FxHashMap(),
1087 1088
            used_libraries: tcx.native_libraries(LOCAL_CRATE),
            link_args: tcx.link_args(LOCAL_CRATE),
1089
            crate_name: FxHashMap(),
1090 1091 1092
            used_crates_dynamic: cstore::used_crates(tcx, LinkagePreference::RequireDynamic),
            used_crates_static: cstore::used_crates(tcx, LinkagePreference::RequireStatic),
            used_crate_source: FxHashMap(),
1093
            wasm_imports: FxHashMap(),
1094 1095
            lang_item_to_crate: FxHashMap(),
            missing_lang_items: FxHashMap(),
1096
        };
1097
        let lang_items = tcx.lang_items();
1098

1099
        let load_wasm_items = tcx.sess.crate_types.borrow()
1100 1101
            .iter()
            .any(|c| *c != config::CrateTypeRlib) &&
1102
            tcx.sess.opts.target_triple.triple() == "wasm32-unknown-unknown";
1103

1104 1105
        if load_wasm_items {
            info.load_wasm_imports(tcx, LOCAL_CRATE);
1106 1107
        }

1108
        for &cnum in tcx.crates().iter() {
1109
            info.native_libraries.insert(cnum, tcx.native_libraries(cnum));
1110
            info.crate_name.insert(cnum, tcx.crate_name(cnum).to_string());
1111
            info.used_crate_source.insert(cnum, tcx.used_crate_source(cnum));
1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126
            if tcx.is_panic_runtime(cnum) {
                info.panic_runtime = Some(cnum);
            }
            if tcx.is_compiler_builtins(cnum) {
                info.compiler_builtins = Some(cnum);
            }
            if tcx.is_profiler_runtime(cnum) {
                info.profiler_runtime = Some(cnum);
            }
            if tcx.is_sanitizer_runtime(cnum) {
                info.sanitizer_runtime = Some(cnum);
            }
            if tcx.is_no_builtins(cnum) {
                info.is_no_builtins.insert(cnum);
            }
1127 1128
            if load_wasm_items {
                info.load_wasm_imports(tcx, cnum);
1129
            }
1130 1131 1132 1133 1134 1135
            let missing = tcx.missing_lang_items(cnum);
            for &item in missing.iter() {
                if let Ok(id) = lang_items.require(item) {
                    info.lang_item_to_crate.insert(item, id.krate);
                }
            }
1136 1137 1138 1139 1140 1141 1142

            // No need to look for lang items that are whitelisted and don't
            // actually need to exist.
            let missing = missing.iter()
                .cloned()
                .filter(|&l| !weak_lang_items::whitelisted(tcx, l))
                .collect();
1143
            info.missing_lang_items.insert(cnum, missing);
1144 1145 1146 1147
        }

        return info
    }
1148 1149 1150 1151 1152 1153 1154 1155

    fn load_wasm_imports(&mut self, tcx: TyCtxt, cnum: CrateNum) {
        for (&id, module) in tcx.wasm_import_module_map(cnum).iter() {
            let instance = Instance::mono(tcx, id);
            let import_name = tcx.symbol_name(instance);
            self.wasm_imports.insert(import_name.to_string(), module.clone());
        }
    }
1156
}
1157

I
Irina Popa 已提交
1158 1159 1160 1161
fn is_codegened_item(tcx: TyCtxt, id: DefId) -> bool {
    let (all_mono_items, _) =
        tcx.collect_and_partition_mono_items(LOCAL_CRATE);
    all_mono_items.contains(&id)
1162 1163
}

A
Alex Crichton 已提交
1164 1165
fn compile_codegen_unit<'a, 'tcx>(tcx: TyCtxt<'a, 'tcx, 'tcx>,
                                  cgu: InternedString) -> Stats {
1166
    let cgu = tcx.codegen_unit(cgu);
A
Alex Crichton 已提交
1167 1168

    let start_time = Instant::now();
I
Irina Popa 已提交
1169 1170
    let (stats, module) = module_codegen(tcx, cgu);
    let time_to_codegen = start_time.elapsed();
A
Alex Crichton 已提交
1171 1172

    // We assume that the cost to run LLVM on a CGU is proportional to
I
Irina Popa 已提交
1173 1174 1175
    // the time we needed for codegenning it.
    let cost = time_to_codegen.as_secs() * 1_000_000_000 +
               time_to_codegen.subsec_nanos() as u64;
A
Alex Crichton 已提交
1176

I
Irina Popa 已提交
1177
    write::submit_codegened_module_to_llvm(tcx,
A
Alex Crichton 已提交
1178 1179 1180 1181
                                            module,
                                            cost);
    return stats;

I
Irina Popa 已提交
1182
    fn module_codegen<'a, 'tcx>(
1183 1184
        tcx: TyCtxt<'a, 'tcx, 'tcx>,
        cgu: Arc<CodegenUnit<'tcx>>)
I
Irina Popa 已提交
1185
        -> (Stats, ModuleCodegen)
A
Alex Crichton 已提交
1186 1187 1188
    {
        let cgu_name = cgu.name().to_string();

1189 1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201
        // Append ".rs" to LLVM module identifier.
        //
        // LLVM code generator emits a ".file filename" directive
        // for ELF backends. Value of the "filename" is set as the
        // LLVM module identifier.  Due to a LLVM MC bug[1], LLVM
        // crashes if the module identifier is same as other symbols
        // such as a function name in the module.
        // 1. http://llvm.org/bugs/show_bug.cgi?id=11479
        let llmod_id = format!("{}-{}.rs",
                               cgu.name(),
                               tcx.crate_disambiguator(LOCAL_CRATE)
                                   .to_fingerprint().to_hex());

I
Irina Popa 已提交
1202
        // Instantiate monomorphizations without filling out definitions yet...
1203
        let cx = CodegenCx::new(tcx, cgu, &llmod_id);
A
Alex Crichton 已提交
1204
        let module = {
I
Irina Popa 已提交
1205
            let mono_items = cx.codegen_unit
1206
                                 .items_in_deterministic_order(cx.tcx);
I
Irina Popa 已提交
1207 1208
            for &(mono_item, (linkage, visibility)) in &mono_items {
                mono_item.predefine(&cx, linkage, visibility);
A
Alex Crichton 已提交
1209 1210 1211
            }

            // ... and now that we have everything pre-defined, fill out those definitions.
I
Irina Popa 已提交
1212 1213
            for &(mono_item, _) in &mono_items {
                mono_item.define(&cx);
A
Alex Crichton 已提交
1214 1215 1216 1217
            }

            // If this codegen unit contains the main function, also create the
            // wrapper here
1218
            maybe_create_entry_wrapper(&cx);
A
Alex Crichton 已提交
1219 1220

            // Run replace-all-uses-with for statics that need it
1221
            for &(old_g, new_g) in cx.statics_to_rauw.borrow().iter() {
A
Alex Crichton 已提交
1222 1223 1224 1225 1226 1227 1228 1229 1230
                unsafe {
                    let bitcast = llvm::LLVMConstPointerCast(new_g, llvm::LLVMTypeOf(old_g));
                    llvm::LLVMReplaceAllUsesWith(old_g, bitcast);
                    llvm::LLVMDeleteGlobal(old_g);
                }
            }

            // Create the llvm.used variable
            // This variable has type [N x i8*] and is stored in the llvm.metadata section
1231
            if !cx.used_statics.borrow().is_empty() {
A
Alex Crichton 已提交
1232 1233
                let name = CString::new("llvm.used").unwrap();
                let section = CString::new("llvm.metadata").unwrap();
1234
                let array = C_array(Type::i8(&cx).ptr_to(), &*cx.used_statics.borrow());
A
Alex Crichton 已提交
1235 1236

                unsafe {
1237
                    let g = llvm::LLVMAddGlobal(cx.llmod,
A
Alex Crichton 已提交
1238 1239 1240 1241 1242 1243 1244 1245 1246
                                                val_ty(array).to_ref(),
                                                name.as_ptr());
                    llvm::LLVMSetInitializer(g, array);
                    llvm::LLVMRustSetLinkage(g, llvm::Linkage::AppendingLinkage);
                    llvm::LLVMSetSection(g, section.as_ptr());
                }
            }

            // Finalize debuginfo
1247 1248
            if cx.sess().opts.debuginfo != NoDebugInfo {
                debuginfo::finalize(&cx);
A
Alex Crichton 已提交
1249 1250 1251
            }

            let llvm_module = ModuleLlvm {
1252 1253
                llcx: cx.llcx,
                llmod: cx.llmod,
1254
                tm: create_target_machine(cx.sess(), false),
A
Alex Crichton 已提交
1255 1256
            };

I
Irina Popa 已提交
1257
            ModuleCodegen {
A
Alex Crichton 已提交
1258
                name: cgu_name,
I
Irina Popa 已提交
1259
                source: ModuleSource::Codegened(llvm_module),
A
Alex Crichton 已提交
1260
                kind: ModuleKind::Regular,
1261
                llmod_id,
A
Alex Crichton 已提交
1262 1263 1264
            }
        };

1265
        (cx.into_stats(), module)
A
Alex Crichton 已提交
1266 1267 1268
    }
}

1269
pub fn provide(providers: &mut Providers) {
I
Irina Popa 已提交
1270 1271
    providers.collect_and_partition_mono_items =
        collect_and_partition_mono_items;
1272

I
Irina Popa 已提交
1273
    providers.is_codegened_item = is_codegened_item;
A
Alex Crichton 已提交
1274 1275

    providers.codegen_unit = |tcx, name| {
I
Irina Popa 已提交
1276
        let (_, all) = tcx.collect_and_partition_mono_items(LOCAL_CRATE);
A
Alex Crichton 已提交
1277 1278 1279 1280 1281 1282
        all.iter()
            .find(|cgu| *cgu.name() == name)
            .cloned()
            .expect(&format!("failed to find cgu with name {:?}", name))
    };
    providers.compile_codegen_unit = compile_codegen_unit;
1283 1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315

    provide_extern(providers);
}

pub fn provide_extern(providers: &mut Providers) {
    providers.dllimport_foreign_items = |tcx, krate| {
        let module_map = tcx.foreign_modules(krate);
        let module_map = module_map.iter()
            .map(|lib| (lib.def_id, lib))
            .collect::<FxHashMap<_, _>>();

        let dllimports = tcx.native_libraries(krate)
            .iter()
            .filter(|lib| {
                if lib.kind != cstore::NativeLibraryKind::NativeUnknown {
                    return false
                }
                let cfg = match lib.cfg {
                    Some(ref cfg) => cfg,
                    None => return true,
                };
                attr::cfg_matches(cfg, &tcx.sess.parse_sess, None)
            })
            .filter_map(|lib| lib.foreign_module)
            .map(|id| &module_map[&id])
            .flat_map(|module| module.foreign_items.iter().cloned())
            .collect();
        Lrc::new(dllimports)
    };

    providers.is_dllimport_foreign_item = |tcx, def_id| {
        tcx.dllimport_foreign_items(def_id.krate).contains(&def_id)
    };
1316 1317
}

1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336 1337 1338
pub fn linkage_to_llvm(linkage: Linkage) -> llvm::Linkage {
    match linkage {
        Linkage::External => llvm::Linkage::ExternalLinkage,
        Linkage::AvailableExternally => llvm::Linkage::AvailableExternallyLinkage,
        Linkage::LinkOnceAny => llvm::Linkage::LinkOnceAnyLinkage,
        Linkage::LinkOnceODR => llvm::Linkage::LinkOnceODRLinkage,
        Linkage::WeakAny => llvm::Linkage::WeakAnyLinkage,
        Linkage::WeakODR => llvm::Linkage::WeakODRLinkage,
        Linkage::Appending => llvm::Linkage::AppendingLinkage,
        Linkage::Internal => llvm::Linkage::InternalLinkage,
        Linkage::Private => llvm::Linkage::PrivateLinkage,
        Linkage::ExternalWeak => llvm::Linkage::ExternalWeakLinkage,
        Linkage::Common => llvm::Linkage::CommonLinkage,
    }
}

pub fn visibility_to_llvm(linkage: Visibility) -> llvm::Visibility {
    match linkage {
        Visibility::Default => llvm::Visibility::Default,
        Visibility::Hidden => llvm::Visibility::Hidden,
        Visibility::Protected => llvm::Visibility::Protected,
1339
    }
1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352
}

// FIXME(mw): Anything that is produced via DepGraph::with_task() must implement
//            the HashStable trait. Normally DepGraph::with_task() calls are
//            hidden behind queries, but CGU creation is a special case in two
//            ways: (1) it's not a query and (2) CGU are output nodes, so their
//            Fingerprints are not actually needed. It remains to be clarified
//            how exactly this case will be handled in the red/green system but
//            for now we content ourselves with providing a no-op HashStable
//            implementation for CGUs.
mod temp_stable_hash_impls {
    use rustc_data_structures::stable_hasher::{StableHasherResult, StableHasher,
                                               HashStable};
I
Irina Popa 已提交
1353
    use ModuleCodegen;
1354

I
Irina Popa 已提交
1355
    impl<HCX> HashStable<HCX> for ModuleCodegen {
1356 1357 1358 1359 1360
        fn hash_stable<W: StableHasherResult>(&self,
                                              _: &mut HCX,
                                              _: &mut StableHasher<W>) {
            // do nothing
        }
1361 1362
    }
}