base.rs 49.8 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
//!   * There's no way to find out the Ty type of a Value.  Doing so
21
//!     would be "trying to get the eggs out of an omelette" (credit:
22 23 24
//!     pcwalton).  You can, instead, find out its llvm::Type by calling val_ty,
//!     but one llvm::Type corresponds to many `Ty`s; for instance, tup(int, int,
//!     int) and rec(x=int, y=int, z=int) will have the same llvm::Type.
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;
33
use back::write::{self, OngoingCodegen};
34
use llvm::{self, TypeKind, get_param};
35
use metadata;
36
use rustc::hir::def_id::{CrateNum, DefId, LOCAL_CRATE};
37
use rustc::middle::lang_items::StartFnLangItem;
38
use rustc::middle::weak_lang_items;
39 40
use rustc::mir::mono::{Linkage, Visibility, Stats};
use rustc::middle::cstore::{EncodedMetadata};
41
use rustc::ty::{self, Ty, TyCtxt};
42
use rustc::ty::layout::{self, Align, TyLayout, LayoutOf};
43
use rustc::ty::query::Providers;
44
use rustc::dep_graph::{DepNode, DepConstructor};
45
use rustc::middle::cstore::{self, LinkMeta, LinkagePreference};
46
use rustc::middle::exported_symbols;
47
use rustc::util::common::{time, print_time_passes_entry};
W
Wesley Wiser 已提交
48
use rustc::util::profiling::ProfileCategory;
49
use rustc::session::config::{self, DebugInfo, EntryFnType};
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::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::small_c_str::SmallCStr;
77
use rustc_data_structures::sync::Lrc;
J
James Miller 已提交
78

79
use std::any::Any;
80
use std::ffi::CString;
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 92
use value::Value;

93
use mir::operand::OperandValue;
94

B
bjorn3 已提交
95 96
use rustc_codegen_utils::check_for_rustc_errors_attr;

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

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

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

C
csmoe 已提交
127
pub fn bin_op_to_icmp_predicate(op: hir::BinOpKind,
M
Ms2ger 已提交
128
                                signed: bool)
129 130
                                -> llvm::IntPredicate {
    match op {
C
csmoe 已提交
131 132 133 134 135 136
        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 },
137
        op => {
138 139 140
            bug!("comparison_op_to_icmp_predicate: expected comparison operator, \
                  found {:?}",
                 op)
141 142 143
        }
    }
}
144

C
csmoe 已提交
145
pub fn bin_op_to_fcmp_predicate(op: hir::BinOpKind) -> llvm::RealPredicate {
146
    match op {
C
csmoe 已提交
147 148 149 150 151 152
        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,
153
        op => {
154 155 156
            bug!("comparison_op_to_fcmp_predicate: expected comparison operator, \
                  found {:?}",
                 op);
157 158 159 160
        }
    }
}

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

179
    let cmp = bin_op_to_icmp_predicate(op, signed);
180 181 182 183
    // 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.
184
    bx.sext(bx.icmp(cmp, lhs, rhs), ret_ty)
185 186
}

A
Ariel Ben-Yehuda 已提交
187 188 189 190
/// 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 已提交
191
/// in an upcast, where the new vtable for an object will be derived
A
Ariel Ben-Yehuda 已提交
192
/// from the old one.
193 194 195 196 197 198
pub fn unsized_info(
    cx: &CodegenCx<'ll, 'tcx>,
    source: Ty<'tcx>,
    target: Ty<'tcx>,
    old_info: Option<&'ll Value>,
) -> &'ll Value {
199
    let (source, target) = cx.tcx.struct_lockstep_tails(source, target);
A
Ariel Ben-Yehuda 已提交
200
    match (&source.sty, &target.sty) {
201
        (&ty::TyArray(_, len), &ty::TySlice(_)) => {
202
            C_usize(cx, len.unwrap_usize(cx.tcx))
203
        }
204
        (&ty::TyDynamic(..), &ty::TyDynamic(..)) => {
A
Ariel Ben-Yehuda 已提交
205 206 207 208 209
            // 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")
        }
210
        (_, &ty::TyDynamic(ref data, ..)) => {
211 212 213 214
            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 已提交
215
        }
216
        _ => bug!("unsized_info: invalid unsizing {:?} -> {:?}",
A
Ariel Ben-Yehuda 已提交
217
                                     source,
218
                                     target),
A
Ariel Ben-Yehuda 已提交
219 220 221 222
    }
}

/// Coerce `src` to `dst_ty`. `src_ty` must be a thin pointer.
223 224
pub fn unsize_thin_ptr(
    bx: &Builder<'a, 'll, 'tcx>,
225
    src: &'ll Value,
M
Mark Simulacrum 已提交
226 227
    src_ty: Ty<'tcx>,
    dst_ty: Ty<'tcx>
228
) -> (&'ll Value, &'ll Value) {
A
Ariel Ben-Yehuda 已提交
229 230
    debug!("unsize_thin_ptr: {:?} => {:?}", src_ty, dst_ty);
    match (&src_ty.sty, &dst_ty.sty) {
231 232 233
        (&ty::TyRef(_, a, _),
         &ty::TyRef(_, b, _)) |
        (&ty::TyRef(_, a, _),
A
Ariel Ben-Yehuda 已提交
234 235 236
         &ty::TyRawPtr(ty::TypeAndMut { ty: b, .. })) |
        (&ty::TyRawPtr(ty::TypeAndMut { ty: a, .. }),
         &ty::TyRawPtr(ty::TypeAndMut { ty: b, .. })) => {
237 238 239
            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 已提交
240
        }
241 242
        (&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());
243 244 245
            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))
246
        }
247 248 249
        (&ty::TyAdt(def_a, _), &ty::TyAdt(def_b, _)) => {
            assert_eq!(def_a, def_b);

250 251
            let src_layout = bx.cx.layout_of(src_ty);
            let dst_layout = bx.cx.layout_of(dst_ty);
252 253
            let mut result = None;
            for i in 0..src_layout.fields.count() {
254
                let src_f = src_layout.field(bx.cx, i);
255 256 257 258 259 260 261
                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);

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

/// 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`
278 279
pub fn coerce_unsized_into(
    bx: &Builder<'a, 'll, 'tcx>,
280 281
    src: PlaceRef<'ll, 'tcx>,
    dst: PlaceRef<'ll, 'tcx>
282
) {
283 284
    let src_ty = src.layout.ty;
    let dst_ty = dst.layout.ty;
285
    let coerce_ptr = || {
286
        let (base, info) = match src.load(bx).val {
287 288 289 290 291
            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.
292 293
                let thin_ptr = dst.layout.field(bx.cx, abi::FAT_PTR_ADDR);
                (bx.pointercast(base, thin_ptr.llvm_type(bx.cx)), info)
294 295
            }
            OperandValue::Immediate(base) => {
296
                unsize_thin_ptr(bx, base, src_ty, dst_ty)
297 298
            }
            OperandValue::Ref(..) => bug!()
299
        };
300
        OperandValue::Pair(base, info).store(bx, dst);
301
    };
A
Ariel Ben-Yehuda 已提交
302 303 304 305
    match (&src_ty.sty, &dst_ty.sty) {
        (&ty::TyRef(..), &ty::TyRef(..)) |
        (&ty::TyRef(..), &ty::TyRawPtr(..)) |
        (&ty::TyRawPtr(..), &ty::TyRawPtr(..)) => {
306 307 308 309
            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 已提交
310 311
        }

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

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

319
                if dst_f.layout.is_zst() {
M
Ms2ger 已提交
320 321
                    continue;
                }
A
Ariel Ben-Yehuda 已提交
322

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

337
pub fn cast_shift_expr_rhs(
338 339
    cx: &Builder<'_, 'll, '_>, op: hir::BinOpKind, lhs: &'ll Value, rhs: &'ll Value
) -> &'ll Value {
340
    cast_shift_rhs(op, lhs, rhs, |a, b| cx.trunc(a, b), |a, b| cx.zext(a, b))
341 342
}

343
fn cast_shift_rhs<'ll, F, G>(op: hir::BinOpKind,
344 345
                        lhs: &'ll Value,
                        rhs: &'ll Value,
346 347
                        trunc: F,
                        zext: G)
348 349 350
                        -> &'ll Value
    where F: FnOnce(&'ll Value, &'ll Type) -> &'ll Value,
          G: FnOnce(&'ll Value, &'ll Type) -> &'ll Value
351
{
352
    // Shifts may have any size int on the rhs
353
    if op.is_shift() {
354 355
        let mut rhs_llty = val_ty(rhs);
        let mut lhs_llty = val_ty(lhs);
356
        if rhs_llty.kind() == TypeKind::Vector {
M
Ms2ger 已提交
357 358
            rhs_llty = rhs_llty.element_type()
        }
359
        if lhs_llty.kind() == TypeKind::Vector {
M
Ms2ger 已提交
360 361
            lhs_llty = lhs_llty.element_type()
        }
362 363 364 365 366 367 368 369
        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)
370 371 372
        } else {
            rhs
        }
373 374
    } else {
        rhs
375 376 377
    }
}

378 379 380 381 382 383
/// 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 {
384
    sess.target.target.options.is_like_msvc
385 386
}

387
pub fn call_assume(bx: &Builder<'_, 'll, '_>, val: &'ll Value) {
388 389
    let assume_intrinsic = bx.cx.get_intrinsic("llvm.assume");
    bx.call(assume_intrinsic, &[val], None);
390 391
}

392
pub fn from_immediate(bx: &Builder<'_, 'll, '_>, val: &'ll Value) -> &'ll Value {
393 394
    if val_ty(val) == Type::i1(bx.cx) {
        bx.zext(val, Type::i8(bx.cx))
395 396 397 398 399
    } else {
        val
    }
}

400 401 402 403 404
pub fn to_immediate(
    bx: &Builder<'_, 'll, '_>,
    val: &'ll Value,
    layout: layout::TyLayout,
) -> &'ll Value {
405
    if let layout::Abi::Scalar(ref scalar) = layout.abi {
406 407 408 409 410
        return to_immediate_scalar(bx, val, scalar);
    }
    val
}

411 412 413 414 415
pub fn to_immediate_scalar(
    bx: &Builder<'_, 'll, '_>,
    val: &'ll Value,
    scalar: &layout::Scalar,
) -> &'ll Value {
416 417
    if scalar.is_bool() {
        return bx.trunc(val, Type::i1(bx.cx));
418
    }
419
    val
420 421
}

422 423 424 425 426 427 428 429
pub fn call_memcpy(
    bx: &Builder<'_, 'll, '_>,
    dst: &'ll Value,
    src: &'ll Value,
    n_bytes: &'ll Value,
    align: Align,
    flags: MemFlags,
) {
430 431 432 433 434 435 436
    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;
    }
437
    let cx = bx.cx;
438
    let ptr_width = &cx.sess().target.target.target_pointer_width;
439
    let key = format!("llvm.memcpy.p0i8.p0i8.i{}", ptr_width);
440
    let memcpy = cx.get_intrinsic(&key);
441 442 443
    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);
444
    let align = C_i32(cx, align.abi() as i32);
445
    let volatile = C_bool(cx, flags.contains(MemFlags::VOLATILE));
446
    bx.call(memcpy, &[dst_ptr, src_ptr, size, align, volatile], None);
447 448
}

449
pub fn memcpy_ty(
450 451 452
    bx: &Builder<'_, 'll, 'tcx>,
    dst: &'ll Value,
    src: &'ll Value,
453
    layout: TyLayout<'tcx>,
454
    align: Align,
455
    flags: MemFlags,
456
) {
457
    let size = layout.size.bytes();
458
    if size == 0 {
459 460 461
        return;
    }

462
    call_memcpy(bx, dst, src, C_usize(bx.cx, size), align, flags);
463 464
}

465
pub fn call_memset(
466 467 468 469 470
    bx: &Builder<'_, 'll, '_>,
    ptr: &'ll Value,
    fill_byte: &'ll Value,
    size: &'ll Value,
    align: &'ll Value,
471
    volatile: bool,
472
) -> &'ll Value {
473
    let ptr_width = &bx.cx.sess().target.target.target_pointer_width;
474
    let intrinsic_key = format!("llvm.memset.p0i8.i{}", ptr_width);
475 476 477
    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)
478 479
}

I
Irina Popa 已提交
480 481
pub fn codegen_instance<'a, 'tcx>(cx: &CodegenCx<'a, 'tcx>, instance: Instance<'tcx>) {
    let _s = if cx.sess().codegen_stats() {
482
        let mut instance_name = String::new();
483
        DefPathBasedNames::new(cx.tcx, true, true)
484
            .push_def_path(instance.def_id(), &mut instance_name);
485
        Some(StatRecorder::new(cx, instance_name))
486 487 488 489
    } else {
        None
    };

490 491 492
    // 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 已提交
493
    info!("codegen_instance({})", instance);
494

495 496
    let fn_ty = instance.ty(cx.tcx);
    let sig = common::ty_fn_sig(cx, fn_ty);
497
    let sig = cx.tcx.normalize_erasing_late_bound_regions(ty::ParamEnv::reveal_all(), &sig);
498

499
    let lldecl = match cx.instances.borrow().get(&instance) {
500 501 502 503
        Some(&val) => val,
        None => bug!("Instance `{:?}` not already declared", instance)
    };

504
    cx.stats.borrow_mut().n_closures += 1;
505

506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521
    // 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
522
    if !cx.sess().no_landing_pads() ||
523
       cx.sess().target.target.options.requires_uwtable {
524
        attributes::emit_uwtable(lldecl, true);
525
    }
526

527
    let mir = cx.tcx.instance_mir(instance.def);
I
Irina Popa 已提交
528
    mir::codegen_mir(cx, lldecl, &mir, instance, sig);
529 530
}

531
pub fn set_link_section(llval: &Value, attrs: &CodegenFnAttrs) {
532 533 534 535 536
    let sect = match attrs.link_section {
        Some(name) => name,
        None => return,
    };
    unsafe {
537
        let buf = SmallCStr::new(&sect.as_str());
538
        llvm::LLVMSetSection(llval, buf.as_ptr());
E
Eli Friedman 已提交
539 540 541
    }
}

F
Fourchaux 已提交
542
/// Create the `main` function which will initialize the rust runtime and call
543
/// users main function.
544 545
fn maybe_create_entry_wrapper(cx: &CodegenCx) {
    let (main_def_id, span) = match *cx.sess().entry_fn.borrow() {
546
        Some((id, span, _)) => {
547
            (cx.tcx.hir.local_def_id(id), span)
548 549 550 551
        }
        None => return,
    };

552
    let instance = Instance::mono(cx.tcx, main_def_id);
553

554
    if !cx.codegen_unit.contains_item(&MonoItem::Fn(instance)) {
555 556 557
        // We want to create the wrapper in the same codegen unit as Rust's main
        // function.
        return;
558 559
    }

560
    let main_llfn = callee::get_fn(cx, instance);
561

562
    let et = cx.sess().entry_fn.get().map(|e| e.2);
563
    match et {
564 565
        Some(EntryFnType::Main) => create_entry_fn(cx, span, main_llfn, main_def_id, true),
        Some(EntryFnType::Start) => create_entry_fn(cx, span, main_llfn, main_def_id, false),
566
        None => {}    // Do nothing.
567
    }
568

569 570 571 572 573 574 575
    fn create_entry_fn(
        cx: &CodegenCx<'ll, '_>,
        sp: Span,
        rust_main: &'ll Value,
        rust_main_def_id: DefId,
        use_start_lang_item: bool,
    ) {
576
        let llfty = Type::func(&[Type::c_int(cx), Type::i8p(cx).ptr_to()], Type::c_int(cx));
577

578
        let main_ret_ty = cx.tcx.fn_sig(rust_main_def_id).output();
579 580 581 582 583
        // 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.
584 585 586
        let main_ret_ty = cx.tcx.erase_regions(
            &main_ret_ty.no_late_bound_regions().unwrap(),
        );
K
kyeongwoon 已提交
587

588
        if declare::get_defined_value(cx, "main").is_some() {
S
Simonas Kazlauskas 已提交
589
            // FIXME: We should be smart and show a better diagnostic here.
590
            cx.sess().struct_span_err(sp, "entry symbol `main` defined multiple times")
N
Nick Cameron 已提交
591 592
                      .help("did you use #[no_mangle] on `fn main`? Use #[start] instead")
                      .emit();
593
            cx.sess().abort_if_errors();
594
            bug!();
595
        }
596
        let llfn = declare::declare_cfn(cx, "main", llfty);
597

598
        // `main` should respect same config for frame pointer elimination as rest of code
599
        attributes::set_frame_pointer_elimination(cx, llfn);
600
        attributes::apply_target_cpu_attr(cx, llfn);
601

602
        let bx = Builder::new_block(cx, llfn, "top");
603

604
        debuginfo::gdb::insert_reference_to_gdb_debug_scripts_section_global(&bx);
605

606 607 608
        // 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);
609
        let arg_argc = bx.intcast(param_argc, cx.isize_ty, true);
610 611
        let arg_argv = param_argv;

M
Mark-Simulacrum 已提交
612
        let (start_fn, args) = if use_start_lang_item {
613
            let start_def_id = cx.tcx.require_lang_item(StartFnLangItem);
614 615 616
            let start_fn = callee::resolve_and_get_fn(
                cx,
                start_def_id,
617
                cx.tcx.intern_substs(&[main_ret_ty.into()]),
618
            );
619
            (start_fn, vec![bx.pointercast(rust_main, Type::i8p(cx).ptr_to()),
620
                            arg_argc, arg_argv])
M
Mark-Simulacrum 已提交
621 622
        } else {
            debug!("using user-defined start fn");
623
            (rust_main, vec![arg_argc, arg_argv])
M
Mark-Simulacrum 已提交
624
        };
625

626 627
        let result = bx.call(start_fn, &args, None);
        bx.ret(bx.intcast(result, Type::c_int(cx), true));
M
Marijn Haverbeke 已提交
628
    }
629 630
}

631
fn write_metadata<'a, 'gcx>(tcx: TyCtxt<'a, 'gcx, 'gcx>,
632
                            llvm_module: &ModuleLlvm,
633
                            link_meta: &LinkMeta)
634
                            -> EncodedMetadata {
A
Alex Crichton 已提交
635 636
    use std::io::Write;
    use flate2::Compression;
637
    use flate2::write::DeflateEncoder;
638

639
    let (metadata_llcx, metadata_llmod) = (&*llvm_module.llcx, llvm_module.llmod());
640

N
Nicholas Nethercote 已提交
641 642 643 644 645 646 647
    #[derive(PartialEq, Eq, PartialOrd, Ord)]
    enum MetadataKind {
        None,
        Uncompressed,
        Compressed
    }

648
    let kind = tcx.sess.crate_types.borrow().iter().map(|ty| {
N
Nicholas Nethercote 已提交
649
        match *ty {
650 651 652
            config::CrateType::Executable |
            config::CrateType::Staticlib |
            config::CrateType::Cdylib => MetadataKind::None,
N
Nicholas Nethercote 已提交
653

654
            config::CrateType::Rlib => MetadataKind::Uncompressed,
N
Nicholas Nethercote 已提交
655

656 657
            config::CrateType::Dylib |
            config::CrateType::ProcMacro => MetadataKind::Compressed,
N
Nicholas Nethercote 已提交
658
        }
659
    }).max().unwrap_or(MetadataKind::None);
N
Nicholas Nethercote 已提交
660 661

    if kind == MetadataKind::None {
662
        return EncodedMetadata::new();
663
    }
J
James Miller 已提交
664

665
    let metadata = tcx.encode_metadata(link_meta);
N
Nicholas Nethercote 已提交
666
    if kind == MetadataKind::Uncompressed {
667
        return metadata;
N
Nicholas Nethercote 已提交
668 669 670
    }

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

675 676
    let llmeta = C_bytes_in_context(metadata_llcx, &compressed);
    let llconst = C_struct_in_context(metadata_llcx, &[llmeta], false);
677
    let name = exported_symbols::metadata_symbol_name(tcx);
678
    let buf = CString::new(name).unwrap();
A
Alex Crichton 已提交
679
    let llglobal = unsafe {
680
        llvm::LLVMAddGlobal(metadata_llmod, val_ty(llconst), buf.as_ptr())
A
Alex Crichton 已提交
681
    };
682 683
    unsafe {
        llvm::LLVMSetInitializer(llglobal, llconst);
684
        let section_name = metadata::metadata_section_name(&tcx.sess.target.target);
685
        let name = SmallCStr::new(section_name);
686 687 688 689 690 691 692
        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();
693
        llvm::LLVMSetModuleInlineAsm(metadata_llmod, directive.as_ptr())
694
    }
695
    return metadata;
696 697
}

698 699 700
pub struct ValueIter<'ll> {
    cur: Option<&'ll Value>,
    step: unsafe extern "C" fn(&'ll Value) -> Option<&'ll Value>,
701
}
702

703 704
impl Iterator for ValueIter<'ll> {
    type Item = &'ll Value;
705

706
    fn next(&mut self) -> Option<&'ll Value> {
707
        let old = self.cur;
708
        if let Some(old) = old {
709
            self.cur = unsafe { (self.step)(old) };
710
        }
711
        old
712
    }
713
}
714

715
pub fn iter_globals(llmod: &'ll llvm::Module) -> ValueIter<'ll> {
716 717 718 719 720 721 722
    unsafe {
        ValueIter {
            cur: llvm::LLVMGetFirstGlobal(llmod),
            step: llvm::LLVMGetNextGlobal,
        }
    }
}
J
Jorge Aparicio 已提交
723

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

B
bjorn3 已提交
728
    check_for_rustc_errors_attr(tcx);
729

730
    if let Some(true) = tcx.sess.opts.debugging_opts.thinlto {
A
Alex Crichton 已提交
731 732 733 734
        if unsafe { !llvm::LLVMRustThinLTOAvailable() } {
            tcx.sess.fatal("this compiler's LLVM does not support ThinLTO");
        }
    }
735

736 737 738 739 740 741 742
    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");
    }

743
    let crate_hash = tcx.crate_hash(LOCAL_CRATE);
744
    let link_meta = link::build_link_meta(crate_hash);
745

I
Irina Popa 已提交
746
    // Codegen the metadata.
W
Wesley Wiser 已提交
747
    tcx.sess.profiler(|p| p.start_activity(ProfileCategory::Codegen));
748
    let llmod_id = "metadata";
749 750 751 752
    let metadata_llvm_module = ModuleLlvm::new(tcx.sess, llmod_id);
    let metadata = time(tcx.sess, "write metadata", || {
        write_metadata(tcx, &metadata_llvm_module, &link_meta)
    });
W
Wesley Wiser 已提交
753
    tcx.sess.profiler(|p| p.end_activity(ProfileCategory::Codegen));
754

I
Irina Popa 已提交
755
    let metadata_module = ModuleCodegen {
756 757
        name: link::METADATA_MODULE_NAME.to_string(),
        llmod_id: llmod_id.to_string(),
758
        source: ModuleSource::Codegened(metadata_llvm_module),
759
        kind: ModuleKind::Metadata,
760
    };
761

I
Irina Popa 已提交
762
    let time_graph = if tcx.sess.opts.debugging_opts.codegen_time_graph {
763 764 765 766
        Some(time_graph::TimeGraph::new())
    } else {
        None
    };
767

I
Irina Popa 已提交
768 769 770 771
    // 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(
772
            tcx,
773
            time_graph.clone(),
774
            link_meta,
775
            metadata,
A
Alex Crichton 已提交
776 777
            rx,
            1);
778

I
Irina Popa 已提交
779 780
        ongoing_codegen.submit_pre_codegened_module_to_llvm(tcx, metadata_module);
        ongoing_codegen.codegen_finished(tcx);
781

782
        assert_and_save_dep_graph(tcx);
783

I
Irina Popa 已提交
784
        ongoing_codegen.check_for_errors(tcx.sess);
785

I
Irina Popa 已提交
786
        return ongoing_codegen;
787 788
    }

I
Irina Popa 已提交
789
    // Run the monomorphization collector and partition the collected items into
790
    // codegen units.
791
    let codegen_units =
I
Irina Popa 已提交
792
        tcx.collect_and_partition_mono_items(LOCAL_CRATE).1;
793
    let codegen_units = (*codegen_units).clone();
794

795 796 797 798 799 800 801 802 803 804 805
    // 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 已提交
806
    let ongoing_codegen = write::start_async_codegen(
807
        tcx,
808
        time_graph.clone(),
809
        link_meta,
810
        metadata,
A
Alex Crichton 已提交
811 812
        rx,
        codegen_units.len());
813

814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835
    // Codegen an allocator shim, if necessary.
    //
    // If the crate doesn't have an `allocator_kind` set then there's definitely
    // no shim to generate. Otherwise we also check our dependency graph for all
    // our output crate types. If anything there looks like its a `Dynamic`
    // linkage, then it's already got an allocator shim and we'll be using that
    // one instead. If nothing exists then it's our job to generate the
    // allocator!
    let any_dynamic_crate = tcx.sess.dependency_formats.borrow()
        .iter()
        .any(|(_, list)| {
            use rustc::middle::dependency_format::Linkage;
            list.iter().any(|linkage| {
                match linkage {
                    Linkage::Dynamic => true,
                    _ => false,
                }
            })
        });
    let allocator_module = if any_dynamic_crate {
        None
    } else if let Some(kind) = *tcx.sess.allocator_kind.get() {
836 837
        unsafe {
            let llmod_id = "allocator";
838
            let modules = ModuleLlvm::new(tcx.sess, llmod_id);
839
            time(tcx.sess, "write allocator module", || {
I
Irina Popa 已提交
840
                allocator::codegen(tcx, &modules, kind)
841
            });
842

843 844 845 846 847 848 849
            Some(ModuleCodegen {
                name: link::ALLOCATOR_MODULE_NAME.to_string(),
                llmod_id: llmod_id.to_string(),
                source: ModuleSource::Codegened(modules),
                kind: ModuleKind::Allocator,
            })
        }
850 851 852 853 854
    } else {
        None
    };

    if let Some(allocator_module) = allocator_module {
I
Irina Popa 已提交
855
        ongoing_codegen.submit_pre_codegened_module_to_llvm(tcx, allocator_module);
856 857
    }

I
Irina Popa 已提交
858
    ongoing_codegen.submit_pre_codegened_module_to_llvm(tcx, metadata_module);
859

860
    // We sort the codegen units by size. This way we can schedule work for LLVM
861
    // a bit more efficiently.
862 863
    let codegen_units = {
        let mut codegen_units = codegen_units;
V
varkor 已提交
864
        codegen_units.sort_by_cached_key(|cgu| cmp::Reverse(cgu.size_estimate()));
865 866 867
        codegen_units
    };

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

A
Alex Crichton 已提交
871
    for cgu in codegen_units.into_iter() {
I
Irina Popa 已提交
872 873
        ongoing_codegen.wait_for_signal_to_codegen_item();
        ongoing_codegen.check_for_errors(tcx.sess);
874

875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891
        // 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) {
892 893 894 895 896 897 898 899 900 901
                    // 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 已提交
902
                    let module = ModuleCodegen {
903 904 905
                        name: cgu.name().to_string(),
                        source: ModuleSource::Preexisting(buf),
                        kind: ModuleKind::Regular,
906
                        llmod_id,
907 908
                    };
                    tcx.dep_graph.mark_loaded_from_cache(dep_node_index, true);
I
Irina Popa 已提交
909
                    write::submit_codegened_module_to_llvm(tcx, module, 0);
910 911 912 913 914 915 916 917 918
                    // 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.
            }
        }

919
        let _timing_guard = time_graph.as_ref().map(|time_graph| {
I
Irina Popa 已提交
920 921
            time_graph.start(write::CODEGEN_WORKER_TIMELINE,
                             write::CODEGEN_WORK_PACKAGE_KIND,
922 923
                             &format!("codegen {}", cgu.name()))
        });
924
        let start_time = Instant::now();
A
Alex Crichton 已提交
925
        all_stats.extend(tcx.compile_codegen_unit(*cgu.name()));
I
Irina Popa 已提交
926 927
        total_codegen_time += start_time.elapsed();
        ongoing_codegen.check_for_errors(tcx.sess);
928 929
    }

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

I
Irina Popa 已提交
932
    // Since the main thread is sometimes blocked during codegen, we keep track
933 934
    // -Ztime-passes output manually.
    print_time_passes_entry(tcx.sess.time_passes(),
I
Irina Popa 已提交
935 936
                            "codegen to LLVM IR",
                            total_codegen_time);
937

A
Alex Crichton 已提交
938
    if tcx.sess.opts.incremental.is_some() {
939
        ::rustc_incremental::assert_module_sources::assert_module_sources(tcx);
940 941
    }

N
Niko Matsakis 已提交
942
    symbol_names_test::report_symbol_names(tcx);
943

I
Irina Popa 已提交
944 945
    if tcx.sess.codegen_stats() {
        println!("--- codegen stats ---");
A
Alex Crichton 已提交
946 947 948
        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);
949

A
Alex Crichton 已提交
950 951 952
        println!("n_fns: {}", all_stats.n_fns);
        println!("n_inlines: {}", all_stats.n_inlines);
        println!("n_closures: {}", all_stats.n_closures);
953
        println!("fn stats:");
A
Alex Crichton 已提交
954 955 956
        all_stats.fn_stats.sort_by_key(|&(_, insns)| insns);
        for &(ref name, insns) in all_stats.fn_stats.iter() {
            println!("{} insns, {}", insns, *name);
957
        }
J
James Miller 已提交
958
    }
959

960
    if tcx.sess.count_llvm_insns() {
A
Alex Crichton 已提交
961
        for (k, v) in all_stats.llvm_insns.iter() {
A
Alex Crichton 已提交
962
            println!("{:7} {}", *v, *k);
963 964 965
        }
    }

I
Irina Popa 已提交
966
    ongoing_codegen.check_for_errors(tcx.sess);
967

968
    assert_and_save_dep_graph(tcx);
I
Irina Popa 已提交
969
    ongoing_codegen
970
}
971

972
fn assert_and_save_dep_graph<'a, 'tcx>(tcx: TyCtxt<'a, 'tcx, 'tcx>) {
973
    time(tcx.sess,
974 975 976
         "assert dep graph",
         || rustc_incremental::assert_dep_graph(tcx));

977
    time(tcx.sess,
978
         "serialize dep graph",
979
         || rustc_incremental::save_dep_graph(tcx));
980 981
}

I
Irina Popa 已提交
982
fn collect_and_partition_mono_items<'a, 'tcx>(
983 984
    tcx: TyCtxt<'a, 'tcx, 'tcx>,
    cnum: CrateNum,
985
) -> (Arc<DefIdSet>, Arc<Vec<Arc<CodegenUnit<'tcx>>>>)
986 987
{
    assert_eq!(cnum, LOCAL_CRATE);
988

I
Irina Popa 已提交
989
    let collection_mode = match tcx.sess.opts.debugging_opts.print_mono_items {
990 991 992 993
        Some(ref s) => {
            let mode_string = s.to_lowercase();
            let mode_string = mode_string.trim();
            if mode_string == "eager" {
994
                MonoItemCollectionMode::Eager
995 996 997 998 999
            } else {
                if mode_string != "lazy" {
                    let message = format!("Unknown codegen-item collection mode '{}'. \
                                           Falling back to 'lazy' mode.",
                                           mode_string);
1000
                    tcx.sess.warn(&message);
1001 1002
                }

1003
                MonoItemCollectionMode::Lazy
1004 1005
            }
        }
1006 1007 1008 1009 1010 1011 1012
        None => {
            if tcx.sess.opts.cg.link_dead_code {
                MonoItemCollectionMode::Eager
            } else {
                MonoItemCollectionMode::Lazy
            }
        }
1013 1014
    };

1015
    let (items, inlining_map) =
I
Irina Popa 已提交
1016
        time(tcx.sess, "monomorphization collection", || {
1017
            collector::collect_crate_mono_items(tcx, collection_mode)
1018 1019
    });

1020 1021
    tcx.sess.abort_if_errors();

1022
    ::rustc_mir::monomorphize::assert_symbols_are_distinct(tcx, items.iter());
1023

1024
    let strategy = if tcx.sess.opts.incremental.is_some() {
1025 1026
        PartitioningStrategy::PerModule
    } else {
1027
        PartitioningStrategy::FixedUnitCount(tcx.sess.codegen_units())
1028 1029
    };

1030
    let codegen_units = time(tcx.sess, "codegen unit partitioning", || {
1031
        partitioning::partition(tcx,
1032 1033
                                items.iter().cloned(),
                                strategy,
1034
                                &inlining_map)
1035 1036 1037
            .into_iter()
            .map(Arc::new)
            .collect::<Vec<_>>()
1038 1039
    });

I
Irina Popa 已提交
1040 1041
    let mono_items: DefIdSet = items.iter().filter_map(|mono_item| {
        match *mono_item {
M
Maik Klein 已提交
1042
            MonoItem::Fn(ref instance) => Some(instance.def_id()),
1043
            MonoItem::Static(def_id) => Some(def_id),
1044 1045 1046
            _ => None,
        }
    }).collect();
1047

I
Irina Popa 已提交
1048
    if tcx.sess.opts.debugging_opts.print_mono_items.is_some() {
1049
        let mut item_to_cgus = FxHashMap();
1050

1051
        for cgu in &codegen_units {
I
Irina Popa 已提交
1052 1053
            for (&mono_item, &linkage) in cgu.items() {
                item_to_cgus.entry(mono_item)
1054
                            .or_insert(Vec::new())
1055
                            .push((cgu.name().clone(), linkage));
1056 1057 1058 1059 1060 1061
            }
        }

        let mut item_keys: Vec<_> = items
            .iter()
            .map(|i| {
1062
                let mut output = i.to_string(tcx);
1063 1064
                output.push_str(" @@");
                let mut empty = Vec::new();
1065
                let cgus = item_to_cgus.get_mut(i).unwrap_or(&mut empty);
1066 1067
                cgus.as_mut_slice().sort_by_key(|&(ref name, _)| name.clone());
                cgus.dedup();
1068
                for &(ref cgu_name, (linkage, _)) in cgus.iter() {
1069
                    output.push_str(" ");
1070
                    output.push_str(&cgu_name.as_str());
1071 1072

                    let linkage_abbrev = match linkage {
1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083
                        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",
1084 1085 1086 1087 1088 1089 1090 1091 1092 1093
                    };

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

1094 1095 1096
        item_keys.sort();

        for item in item_keys {
I
Irina Popa 已提交
1097
            println!("MONO_ITEM {}", item);
1098 1099
        }
    }
1100

I
Irina Popa 已提交
1101
    (Arc::new(mono_items), Arc::new(codegen_units))
1102
}
1103 1104

impl CrateInfo {
1105
    pub fn new(tcx: TyCtxt) -> CrateInfo {
1106 1107 1108 1109 1110 1111
        let mut info = CrateInfo {
            panic_runtime: None,
            compiler_builtins: None,
            profiler_runtime: None,
            sanitizer_runtime: None,
            is_no_builtins: FxHashSet(),
1112
            native_libraries: FxHashMap(),
1113 1114
            used_libraries: tcx.native_libraries(LOCAL_CRATE),
            link_args: tcx.link_args(LOCAL_CRATE),
1115
            crate_name: FxHashMap(),
1116 1117 1118
            used_crates_dynamic: cstore::used_crates(tcx, LinkagePreference::RequireDynamic),
            used_crates_static: cstore::used_crates(tcx, LinkagePreference::RequireStatic),
            used_crate_source: FxHashMap(),
1119
            wasm_imports: FxHashMap(),
1120 1121
            lang_item_to_crate: FxHashMap(),
            missing_lang_items: FxHashMap(),
1122
        };
1123
        let lang_items = tcx.lang_items();
1124

1125
        let load_wasm_items = tcx.sess.crate_types.borrow()
1126
            .iter()
1127
            .any(|c| *c != config::CrateType::Rlib) &&
1128
            tcx.sess.opts.target_triple.triple() == "wasm32-unknown-unknown";
1129

1130 1131
        if load_wasm_items {
            info.load_wasm_imports(tcx, LOCAL_CRATE);
1132 1133
        }

1134
        for &cnum in tcx.crates().iter() {
1135
            info.native_libraries.insert(cnum, tcx.native_libraries(cnum));
1136
            info.crate_name.insert(cnum, tcx.crate_name(cnum).to_string());
1137
            info.used_crate_source.insert(cnum, tcx.used_crate_source(cnum));
1138 1139 1140 1141 1142 1143 1144 1145 1146 1147 1148 1149 1150 1151 1152
            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);
            }
1153 1154
            if load_wasm_items {
                info.load_wasm_imports(tcx, cnum);
1155
            }
1156 1157 1158 1159 1160 1161
            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);
                }
            }
1162 1163 1164 1165 1166 1167 1168

            // 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();
1169
            info.missing_lang_items.insert(cnum, missing);
1170 1171 1172 1173
        }

        return info
    }
1174 1175 1176 1177 1178 1179 1180 1181

    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());
        }
    }
1182
}
1183

I
Irina Popa 已提交
1184 1185 1186 1187
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)
1188 1189
}

A
Alex Crichton 已提交
1190 1191
fn compile_codegen_unit<'a, 'tcx>(tcx: TyCtxt<'a, 'tcx, 'tcx>,
                                  cgu: InternedString) -> Stats {
1192
    let cgu = tcx.codegen_unit(cgu);
A
Alex Crichton 已提交
1193 1194

    let start_time = Instant::now();
I
Irina Popa 已提交
1195 1196
    let (stats, module) = module_codegen(tcx, cgu);
    let time_to_codegen = start_time.elapsed();
A
Alex Crichton 已提交
1197 1198

    // We assume that the cost to run LLVM on a CGU is proportional to
I
Irina Popa 已提交
1199 1200 1201
    // 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 已提交
1202

I
Irina Popa 已提交
1203
    write::submit_codegened_module_to_llvm(tcx,
A
Alex Crichton 已提交
1204 1205 1206 1207
                                            module,
                                            cost);
    return stats;

I
Irina Popa 已提交
1208
    fn module_codegen<'a, 'tcx>(
1209 1210
        tcx: TyCtxt<'a, 'tcx, 'tcx>,
        cgu: Arc<CodegenUnit<'tcx>>)
I
Irina Popa 已提交
1211
        -> (Stats, ModuleCodegen)
A
Alex Crichton 已提交
1212 1213 1214
    {
        let cgu_name = cgu.name().to_string();

1215 1216 1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227
        // 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 已提交
1228
        // Instantiate monomorphizations without filling out definitions yet...
1229 1230 1231
        let llvm_module = ModuleLlvm::new(tcx.sess, &llmod_id);
        let stats = {
            let cx = CodegenCx::new(tcx, cgu, &llvm_module);
I
Irina Popa 已提交
1232
            let mono_items = cx.codegen_unit
1233
                                 .items_in_deterministic_order(cx.tcx);
I
Irina Popa 已提交
1234 1235
            for &(mono_item, (linkage, visibility)) in &mono_items {
                mono_item.predefine(&cx, linkage, visibility);
A
Alex Crichton 已提交
1236 1237 1238
            }

            // ... and now that we have everything pre-defined, fill out those definitions.
I
Irina Popa 已提交
1239 1240
            for &(mono_item, _) in &mono_items {
                mono_item.define(&cx);
A
Alex Crichton 已提交
1241 1242 1243 1244
            }

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

            // Run replace-all-uses-with for statics that need it
1248
            for &(old_g, new_g) in cx.statics_to_rauw.borrow().iter() {
A
Alex Crichton 已提交
1249
                unsafe {
1250
                    let bitcast = llvm::LLVMConstPointerCast(new_g, val_ty(old_g));
A
Alex Crichton 已提交
1251 1252 1253 1254 1255 1256 1257
                    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
1258
            if !cx.used_statics.borrow().is_empty() {
1259 1260
                let name = const_cstr!("llvm.used");
                let section = const_cstr!("llvm.metadata");
1261
                let array = C_array(Type::i8(&cx).ptr_to(), &*cx.used_statics.borrow());
A
Alex Crichton 已提交
1262 1263

                unsafe {
1264
                    let g = llvm::LLVMAddGlobal(cx.llmod,
1265
                                                val_ty(array),
A
Alex Crichton 已提交
1266 1267 1268 1269 1270 1271 1272 1273
                                                name.as_ptr());
                    llvm::LLVMSetInitializer(g, array);
                    llvm::LLVMRustSetLinkage(g, llvm::Linkage::AppendingLinkage);
                    llvm::LLVMSetSection(g, section.as_ptr());
                }
            }

            // Finalize debuginfo
1274
            if cx.sess().opts.debuginfo != DebugInfo::None {
1275
                debuginfo::finalize(&cx);
A
Alex Crichton 已提交
1276 1277
            }

1278
            cx.stats.into_inner()
A
Alex Crichton 已提交
1279 1280
        };

1281 1282 1283 1284 1285 1286
        (stats, ModuleCodegen {
            name: cgu_name,
            source: ModuleSource::Codegened(llvm_module),
            kind: ModuleKind::Regular,
            llmod_id,
        })
A
Alex Crichton 已提交
1287 1288 1289
    }
}

1290
pub fn provide(providers: &mut Providers) {
I
Irina Popa 已提交
1291 1292
    providers.collect_and_partition_mono_items =
        collect_and_partition_mono_items;
1293

I
Irina Popa 已提交
1294
    providers.is_codegened_item = is_codegened_item;
A
Alex Crichton 已提交
1295 1296

    providers.codegen_unit = |tcx, name| {
I
Irina Popa 已提交
1297
        let (_, all) = tcx.collect_and_partition_mono_items(LOCAL_CRATE);
A
Alex Crichton 已提交
1298 1299 1300
        all.iter()
            .find(|cgu| *cgu.name() == name)
            .cloned()
1301
            .unwrap_or_else(|| panic!("failed to find cgu with name {:?}", name))
A
Alex Crichton 已提交
1302 1303
    };
    providers.compile_codegen_unit = compile_codegen_unit;
1304 1305 1306 1307 1308 1309 1310 1311 1312 1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336

    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)
    };
1337 1338
}

1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359
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,
1360
    }
1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373
}

// 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 已提交
1374
    use ModuleCodegen;
1375

I
Irina Popa 已提交
1376
    impl<HCX> HashStable<HCX> for ModuleCodegen {
1377 1378 1379 1380 1381
        fn hash_stable<W: StableHasherResult>(&self,
                                              _: &mut HCX,
                                              _: &mut StableHasher<W>) {
            // do nothing
        }
1382 1383
    }
}