trans.rs 171.2 KB
Newer Older
1
import std._int;
2
import std._str;
3
import std._uint;
4 5 6
import std._vec;
import std._str.rustrt.sbuf;
import std._vec.rustrt.vbuf;
7
import std.map;
8
import std.map.hashmap;
9 10 11
import std.option;
import std.option.some;
import std.option.none;
12

13
import front.ast;
14
import driver.session;
15
import middle.ty;
16
import back.x86;
17 18
import back.abi;

19
import middle.ty.pat_ty;
20
import middle.ty.plain_ty;
P
Patrick Walton 已提交
21

22
import util.common;
23
import util.common.append;
24
import util.common.istr;
25
import util.common.new_def_hash;
26
import util.common.new_str_hash;
27 28 29

import lib.llvm.llvm;
import lib.llvm.builder;
30
import lib.llvm.target_data;
31
import lib.llvm.type_handle;
32
import lib.llvm.type_names;
33
import lib.llvm.mk_pass_manager;
34
import lib.llvm.mk_target_data;
35
import lib.llvm.mk_type_handle;
36
import lib.llvm.mk_type_names;
37 38 39
import lib.llvm.llvm.ModuleRef;
import lib.llvm.llvm.ValueRef;
import lib.llvm.llvm.TypeRef;
40
import lib.llvm.llvm.TypeHandleRef;
41 42
import lib.llvm.llvm.BuilderRef;
import lib.llvm.llvm.BasicBlockRef;
43

44 45
import lib.llvm.False;
import lib.llvm.True;
46

47 48 49 50 51 52 53
state obj namegen(mutable int i) {
    fn next(str prefix) -> str {
        i += 1;
        ret prefix + istr(i);
    }
}

54 55
type glue_fns = rec(ValueRef activate_glue,
                    ValueRef yield_glue,
56
                    ValueRef exit_task_glue,
57
                    vec[ValueRef] upcall_glues,
58
                    ValueRef no_op_type_glue,
59 60
                    ValueRef memcpy_glue,
                    ValueRef bzero_glue);
61

62 63 64 65 66
type tag_info = rec(
    type_handle th,
    mutable uint size,
    mutable @hashmap[ast.def_id,ValueRef] lldiscrims
);
67

68
state type crate_ctxt = rec(session.session sess,
69
                            ModuleRef llmod,
70
                            target_data td,
71
                            type_names tn,
72
                            ValueRef crate_ptr,
73
                            hashmap[str, ValueRef] upcalls,
74
                            hashmap[str, ValueRef] intrinsics,
75 76
                            hashmap[str, ValueRef] item_names,
                            hashmap[ast.def_id, ValueRef] item_ids,
77
                            hashmap[ast.def_id, @ast.item] items,
G
Graydon Hoare 已提交
78 79
                            hashmap[ast.def_id,
                                    @ast.native_item] native_items,
80
                            hashmap[@ty.t, @tag_info] tags,
81
                            hashmap[ast.def_id, ValueRef] fn_pairs,
82
                            hashmap[ast.def_id, ValueRef] consts,
83
                            hashmap[ast.def_id,()] obj_methods,
84
                            hashmap[@ty.t, ValueRef] tydescs,
85
                            vec[ast.ty_param] obj_typarams,
86
                            vec[ast.obj_field] obj_fields,
87 88 89
                            @glue_fns glues,
                            namegen names,
                            str path);
90

91 92
state type fn_ctxt = rec(ValueRef llfn,
                         ValueRef lltaskptr,
93 94
                         ValueRef llenv,
                         ValueRef llretptr,
95
                         mutable option.t[ValueRef] llself,
96
                         mutable option.t[ValueRef] lliterbody,
97
                         hashmap[ast.def_id, ValueRef] llargs,
98
                         hashmap[ast.def_id, ValueRef] llobjfields,
99
                         hashmap[ast.def_id, ValueRef] lllocals,
100
                         hashmap[ast.def_id, ValueRef] lltydescs,
101
                         @crate_ctxt ccx);
102

103
tag cleanup {
104
    clean(fn(@block_ctxt cx) -> result);
105 106
}

107 108 109 110 111 112

tag block_kind {
    SCOPE_BLOCK;
    NON_SCOPE_BLOCK;
}

113 114
state type block_ctxt = rec(BasicBlockRef llbb,
                            builder build,
115
                            block_parent parent,
116
                            block_kind kind,
117 118 119
                            mutable vec[cleanup] cleanups,
                            @fn_ctxt fcx);

120 121 122 123 124 125 126 127
// FIXME: we should be able to use option.t[@block_parent] here but
// the infinite-tag check in rustboot gets upset.

tag block_parent {
    parent_none;
    parent_some(@block_ctxt);
}

128

129 130 131
state type result = rec(mutable @block_ctxt bcx,
                        mutable ValueRef val);

132 133 134 135
fn sep() -> str {
    ret "_";
}

136 137 138 139 140
fn res(@block_ctxt bcx, ValueRef val) -> result {
    ret rec(mutable bcx = bcx,
            mutable val = val);
}

141 142
fn ty_str(type_names tn, TypeRef t) -> str {
    ret lib.llvm.type_to_str(tn, t);
143 144 145 146 147 148
}

fn val_ty(ValueRef v) -> TypeRef {
    ret llvm.LLVMTypeOf(v);
}

149 150
fn val_str(type_names tn, ValueRef v) -> str {
    ret ty_str(tn, val_ty(v));
151
}
152 153 154 155


// LLVM type constructors.

156 157 158 159 160 161 162 163 164 165 166
fn T_void() -> TypeRef {
    // Note: For the time being llvm is kinda busted here, it has the notion
    // of a 'void' type that can only occur as part of the signature of a
    // function, but no general unit type of 0-sized value. This is, afaict,
    // vestigial from its C heritage, and we'll be attempting to submit a
    // patch upstream to fix it. In the mean time we only model function
    // outputs (Rust functions and C functions) using T_void, and model the
    // Rust general purpose nil type you can construct as 1-bit (always
    // zero). This makes the result incorrect for now -- things like a tuple
    // of 10 nil values will have 10-bit size -- but it doesn't seem like we
    // have any other options until it's fixed upstream.
167 168 169
    ret llvm.LLVMVoidType();
}

170 171 172 173 174
fn T_nil() -> TypeRef {
    // NB: See above in T_void().
    ret llvm.LLVMInt1Type();
}

175 176 177 178
fn T_i1() -> TypeRef {
    ret llvm.LLVMInt1Type();
}

179 180 181 182 183 184 185 186 187
fn T_i8() -> TypeRef {
    ret llvm.LLVMInt8Type();
}

fn T_i16() -> TypeRef {
    ret llvm.LLVMInt16Type();
}

fn T_i32() -> TypeRef {
188 189 190
    ret llvm.LLVMInt32Type();
}

191 192 193 194
fn T_i64() -> TypeRef {
    ret llvm.LLVMInt64Type();
}

195 196 197 198 199 200 201 202
fn T_f32() -> TypeRef {
    ret llvm.LLVMFloatType();
}

fn T_f64() -> TypeRef {
    ret llvm.LLVMDoubleType();
}

203 204 205 206
fn T_bool() -> TypeRef {
    ret T_i1();
}

207 208 209 210 211
fn T_int() -> TypeRef {
    // FIXME: switch on target type.
    ret T_i32();
}

212 213 214 215
fn T_char() -> TypeRef {
    ret T_i32();
}

216 217 218 219
fn T_fn(vec[TypeRef] inputs, TypeRef output) -> TypeRef {
    ret llvm.LLVMFunctionType(output,
                              _vec.buf[TypeRef](inputs),
                              _vec.len[TypeRef](inputs),
220 221 222
                              False);
}

223
fn T_fn_pair(type_names tn, TypeRef tfn) -> TypeRef {
224
    ret T_struct(vec(T_ptr(tfn),
225
                     T_opaque_closure_ptr(tn)));
226 227
}

228 229 230 231 232 233 234 235 236 237 238 239 240 241
fn T_ptr(TypeRef t) -> TypeRef {
    ret llvm.LLVMPointerType(t, 0u);
}

fn T_struct(vec[TypeRef] elts) -> TypeRef {
    ret llvm.LLVMStructType(_vec.buf[TypeRef](elts),
                            _vec.len[TypeRef](elts),
                            False);
}

fn T_opaque() -> TypeRef {
    ret llvm.LLVMOpaqueType();
}

242 243 244 245 246 247 248 249 250 251 252 253 254 255 256 257 258
fn T_task(type_names tn) -> TypeRef {
    auto s = "task";
    if (tn.name_has_type(s)) {
        ret tn.get_type(s);
    }

    auto t = T_struct(vec(T_int(),      // Refcount
                          T_int(),      // Delegate pointer
                          T_int(),      // Stack segment pointer
                          T_int(),      // Runtime SP
                          T_int(),      // Rust SP
                          T_int(),      // GC chain
                          T_int(),      // Domain pointer
                          T_int()       // Crate cache pointer
                          ));
    tn.associate(s, t);
    ret t;
259 260
}

261 262 263 264 265 266
fn T_glue_fn(type_names tn) -> TypeRef {
    auto s = "glue_fn";
    if (tn.name_has_type(s)) {
        ret tn.get_type(s);
    }

267 268
    // Bit of a kludge: pick the fn typeref out of the tydesc..
    let vec[TypeRef] tydesc_elts = _vec.init_elt[TypeRef](T_nil(), 10u);
269
    llvm.LLVMGetStructElementTypes(T_tydesc(tn),
270
                                   _vec.buf[TypeRef](tydesc_elts));
271 272 273 274 275
    auto t =
        llvm.LLVMGetElementType
        (tydesc_elts.(abi.tydesc_field_drop_glue_off));
    tn.associate(s, t);
    ret t;
276 277
}

278 279 280 281 282 283
fn T_tydesc(type_names tn) -> TypeRef {

    auto s = "tydesc";
    if (tn.name_has_type(s)) {
        ret tn.get_type(s);
    }
284 285 286

    auto th = mk_type_handle();
    auto abs_tydesc = llvm.LLVMResolveTypeHandle(th.llth);
287
    auto tydescpp = T_ptr(T_ptr(abs_tydesc));
288
    auto pvoid = T_ptr(T_i8());
289
    auto glue_fn_ty = T_ptr(T_fn(vec(T_ptr(T_nil()),
290
                                     T_taskptr(tn),
291
                                     T_ptr(T_nil()),
292
                                     tydescpp,
293
                                     pvoid), T_void()));
294
    auto tydesc = T_struct(vec(tydescpp,          // first_param
295 296 297 298 299 300 301 302 303 304 305
                               T_int(),           // size
                               T_int(),           // align
                               glue_fn_ty,        // take_glue_off
                               glue_fn_ty,        // drop_glue_off
                               glue_fn_ty,        // free_glue_off
                               glue_fn_ty,        // sever_glue_off
                               glue_fn_ty,        // mark_glue_off
                               glue_fn_ty,        // obj_drop_glue_off
                               glue_fn_ty));      // is_stateful

    llvm.LLVMRefineType(abs_tydesc, tydesc);
306 307 308
    auto t = llvm.LLVMResolveTypeHandle(th.llth);
    tn.associate(s, t);
    ret t;
309 310
}

311 312 313 314
fn T_array(TypeRef t, uint n) -> TypeRef {
    ret llvm.LLVMArrayType(t, n);
}

315 316 317 318 319
fn T_vec(TypeRef t) -> TypeRef {
    ret T_struct(vec(T_int(),       // Refcount
                     T_int(),       // Alloc
                     T_int(),       // Fill
                     T_array(t, 0u) // Body elements
320 321 322
                     ));
}

323 324
fn T_str() -> TypeRef {
    ret T_vec(T_i8());
325 326
}

327 328 329 330
fn T_box(TypeRef t) -> TypeRef {
    ret T_struct(vec(T_int(), t));
}

331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349
fn T_crate(type_names tn) -> TypeRef {
    auto s = "crate";
    if (tn.name_has_type(s)) {
        ret tn.get_type(s);
    }

    auto t = T_struct(vec(T_int(),      // ptrdiff_t image_base_off
                          T_int(),      // uintptr_t self_addr
                          T_int(),      // ptrdiff_t debug_abbrev_off
                          T_int(),      // size_t debug_abbrev_sz
                          T_int(),      // ptrdiff_t debug_info_off
                          T_int(),      // size_t debug_info_sz
                          T_int(),      // size_t activate_glue_off
                          T_int(),      // size_t yield_glue_off
                          T_int(),      // size_t unwind_glue_off
                          T_int(),      // size_t gc_glue_off
                          T_int(),      // size_t main_exit_task_glue_off
                          T_int(),      // int n_rust_syms
                          T_int(),      // int n_c_syms
350 351
                          T_int(),      // int n_libs
                          T_int()       // uintptr_t abi_tag
352 353 354
                          ));
    tn.associate(s, t);
    ret t;
355 356
}

357 358 359 360
fn T_double() -> TypeRef {
    ret llvm.LLVMDoubleType();
}

361 362
fn T_taskptr(type_names tn) -> TypeRef {
    ret T_ptr(T_task(tn));
363 364
}

365 366 367 368 369 370 371 372 373
fn T_typaram_ptr(type_names tn) -> TypeRef {
    auto s = "typaram";
    if (tn.name_has_type(s)) {
        ret tn.get_type(s);
    }

    auto t = T_ptr(T_i8());
    tn.associate(s, t);
    ret t;
374 375
}

376 377
fn T_closure_ptr(type_names tn,
                 TypeRef lltarget_ty,
378 379
                 TypeRef llbindings_ty,
                 uint n_ty_params) -> TypeRef {
380
    ret T_ptr(T_box(T_struct(vec(T_ptr(T_tydesc(tn)),
381
                                 lltarget_ty,
382 383
                                 llbindings_ty,
                                 T_captured_tydescs(tn, n_ty_params))
384 385 386
                             )));
}

387 388 389 390 391 392 393
fn T_opaque_closure_ptr(type_names tn) -> TypeRef {
    auto s = "*closure";
    if (tn.name_has_type(s)) {
        ret tn.get_type(s);
    }
    auto t = T_closure_ptr(tn, T_struct(vec(T_ptr(T_nil()),
                                            T_ptr(T_nil()))),
394 395
                           T_nil(),
                           0u);
396 397
    tn.associate(s, t);
    ret t;
398 399
}

400 401
fn T_captured_tydescs(type_names tn, uint n) -> TypeRef {
    ret T_struct(_vec.init_elt[TypeRef](T_ptr(T_tydesc(tn)), n));
402 403
}

404 405 406 407
fn T_obj(type_names tn, uint n_captured_tydescs,
         TypeRef llfields_ty) -> TypeRef {
    ret T_struct(vec(T_ptr(T_tydesc(tn)),
                     T_captured_tydescs(tn, n_captured_tydescs),
408 409 410
                     llfields_ty));
}

411 412 413
fn T_obj_ptr(type_names tn, uint n_captured_tydescs,
             TypeRef llfields_ty) -> TypeRef {
    ret T_ptr(T_box(T_obj(tn, n_captured_tydescs, llfields_ty)));
414 415
}

416 417
fn T_opaque_obj_ptr(type_names tn) -> TypeRef {
    ret T_obj_ptr(tn, 0u, T_nil());
418 419
}

420

421
fn type_of(@crate_ctxt cx, @ty.t t) -> TypeRef {
422
    ret type_of_inner(cx, t);
423 424
}

425 426 427 428 429 430
fn type_of_explicit_args(@crate_ctxt cx,
                     vec[ty.arg] inputs) -> vec[TypeRef] {
    let vec[TypeRef] atys = vec();
    for (ty.arg arg in inputs) {
        if (ty.type_has_dynamic_size(arg.ty)) {
            check (arg.mode == ast.alias);
431
            atys += T_typaram_ptr(cx.tn);
432
        } else {
433
            let TypeRef t = type_of_inner(cx, arg.ty);
434 435 436 437 438 439 440 441 442 443 444
            alt (arg.mode) {
                case (ast.alias) {
                    t = T_ptr(t);
                }
                case (_) { /* fall through */  }
            }
            atys += t;
        }
    }
    ret atys;
}
445 446 447 448 449 450 451 452

// NB: must keep 4 fns in sync:
//
//  - type_of_fn_full
//  - create_llargs_for_fn_args.
//  - new_fn_ctxt
//  - trans_args

453
fn type_of_fn_full(@crate_ctxt cx,
454
                   ast.proto proto,
455 456 457
                   option.t[TypeRef] obj_self,
                   vec[ty.arg] inputs,
                   @ty.t output) -> TypeRef {
458
    let vec[TypeRef] atys = vec();
459

460
    // Arg 0: Output pointer.
461
    if (ty.type_has_dynamic_size(output)) {
462
        atys += T_typaram_ptr(cx.tn);
463
    } else {
464
        atys += T_ptr(type_of_inner(cx, output));
465 466
    }

467
    // Arg 1: Task pointer.
468
    atys += T_taskptr(cx.tn);
469 470

    // Arg 2: Env (closure-bindings / self-obj)
471 472 473 474 475
    alt (obj_self) {
        case (some[TypeRef](?t)) {
            check (t as int != 0);
            atys += t;
        }
476
        case (_) {
477
            atys += T_opaque_closure_ptr(cx.tn);
478
        }
479 480
    }

481 482 483
    // Args >3: ty params, if not acquired via capture...
    if (obj_self == none[TypeRef]) {
        auto ty_param_count =
484 485 486
            ty.count_ty_params(plain_ty(ty.ty_fn(proto,
                                                 inputs,
                                                 output)));
487 488
        auto i = 0u;
        while (i < ty_param_count) {
489
            atys += T_ptr(T_tydesc(cx.tn));
490 491
            i += 1u;
        }
492 493
    }

494
    if (proto == ast.proto_iter) {
495 496 497
        // If it's an iter, the 'output' type of the iter is actually the
        // *input* type of the function we're given as our iter-block
        // argument.
498
        atys += T_fn_pair(cx.tn,
499
                          type_of_fn_full(cx, ast.proto_fn, none[TypeRef],
500 501 502 503
                                          vec(rec(mode=ast.val, ty=output)),
                                          plain_ty(ty.ty_nil)));
    }

504
    // ... then explicit args.
505
    atys += type_of_explicit_args(cx, inputs);
506

507
    ret T_fn(atys, llvm.LLVMVoidType());
508 509
}

510
fn type_of_fn(@crate_ctxt cx,
511
              ast.proto proto,
512
              vec[ty.arg] inputs, @ty.t output) -> TypeRef {
513
    ret type_of_fn_full(cx, proto, none[TypeRef], inputs, output);
514 515
}

516 517
fn type_of_native_fn(@crate_ctxt cx, ast.native_abi abi,
                     vec[ty.arg] inputs,
518
                     @ty.t output) -> TypeRef {
519 520 521 522 523 524 525 526 527 528 529 530
    let vec[TypeRef] atys = vec();
    if (abi == ast.native_abi_rust) {
        atys += T_taskptr(cx.tn);
        auto t = ty.ty_native_fn(abi, inputs, output);
        auto ty_param_count = ty.count_ty_params(plain_ty(t));
        auto i = 0u;
        while (i < ty_param_count) {
            atys += T_ptr(T_tydesc(cx.tn));
            i += 1u;
        }
    }
    atys += type_of_explicit_args(cx, inputs);
531
    ret T_fn(atys, type_of_inner(cx, output));
532 533
}

534
fn type_of_inner(@crate_ctxt cx, @ty.t t) -> TypeRef {
535 536
    let TypeRef llty = 0 as TypeRef;

537
    alt (t.struct) {
538 539 540 541 542
        case (ty.ty_native) { llty = T_ptr(T_i8()); }
        case (ty.ty_nil) { llty = T_nil(); }
        case (ty.ty_bool) { llty = T_bool(); }
        case (ty.ty_int) { llty = T_int(); }
        case (ty.ty_uint) { llty = T_int(); }
543
        case (ty.ty_machine(?tm)) {
544
            alt (tm) {
545 546 547 548 549 550 551 552 553 554
                case (common.ty_i8) { llty = T_i8(); }
                case (common.ty_u8) { llty = T_i8(); }
                case (common.ty_i16) { llty = T_i16(); }
                case (common.ty_u16) { llty = T_i16(); }
                case (common.ty_i32) { llty = T_i32(); }
                case (common.ty_u32) { llty = T_i32(); }
                case (common.ty_i64) { llty = T_i64(); }
                case (common.ty_u64) { llty = T_i64(); }
                case (common.ty_f32) { llty = T_f32(); }
                case (common.ty_f64) { llty = T_f64(); }
555 556
            }
        }
557 558
        case (ty.ty_char) { llty = T_char(); }
        case (ty.ty_str) { llty = T_ptr(T_str()); }
559
        case (ty.ty_tag(?tag_id, _)) {
560
            llty = llvm.LLVMResolveTypeHandle(cx.tags.get(t).th.llth);
561
        }
562
        case (ty.ty_box(?t)) {
563
            llty = T_ptr(T_box(type_of_inner(cx, t)));
564
        }
565
        case (ty.ty_vec(?t)) {
566
            llty = T_ptr(T_vec(type_of_inner(cx, t)));
567
        }
568
        case (ty.ty_tup(?elts)) {
569
            let vec[TypeRef] tys = vec();
570
            for (@ty.t elt in elts) {
571
                tys += type_of_inner(cx, elt);
572
            }
573
            llty = T_struct(tys);
574
        }
575
        case (ty.ty_rec(?fields)) {
576
            let vec[TypeRef] tys = vec();
577
            for (ty.field f in fields) {
578
                tys += type_of_inner(cx, f.ty);
579
            }
580
            llty = T_struct(tys);
581
        }
582
        case (ty.ty_fn(?proto, ?args, ?out)) {
583
            llty = T_fn_pair(cx.tn, type_of_fn(cx, proto, args, out));
584
        }
585
        case (ty.ty_native_fn(?abi, ?args, ?out)) {
586
            llty = T_fn_pair(cx.tn, type_of_native_fn(cx, abi, args, out));
587
        }
588
        case (ty.ty_obj(?meths)) {
589 590 591
            auto th = mk_type_handle();
            auto self_ty = llvm.LLVMResolveTypeHandle(th.llth);

592
            let vec[TypeRef] mtys = vec();
593
            for (ty.method m in meths) {
594
                let TypeRef mty =
595
                    type_of_fn_full(cx, m.proto,
596 597
                                    some[TypeRef](self_ty),
                                    m.inputs, m.output);
598
                mtys += T_ptr(mty);
599
            }
600
            let TypeRef vtbl = T_struct(mtys);
601
            let TypeRef pair = T_struct(vec(T_ptr(vtbl),
602
                                            T_opaque_obj_ptr(cx.tn)));
603

604 605 606
            auto abs_pair = llvm.LLVMResolveTypeHandle(th.llth);
            llvm.LLVMRefineType(abs_pair, pair);
            abs_pair = llvm.LLVMResolveTypeHandle(th.llth);
607
            llty = abs_pair;
608
        }
609
        case (ty.ty_var(_)) {
610
            log "ty_var in trans.type_of";
611 612
            fail;
        }
613
        case (ty.ty_param(_)) {
614
            llty = T_typaram_ptr(cx.tn);
615
        }
616
        case (ty.ty_type) { llty = T_ptr(T_tydesc(cx.tn)); }
617
    }
618 619 620 621

    check (llty as int != 0);
    llvm.LLVMAddTypeName(cx.llmod, _str.buf(ty.ty_to_str(t)), llty);
    ret llty;
622 623
}

624
fn type_of_arg(@crate_ctxt cx, &ty.arg arg) -> TypeRef {
625
    auto ty = type_of_inner(cx, arg.ty);
626 627 628 629 630 631
    if (arg.mode == ast.alias) {
        ty = T_ptr(ty);
    }
    ret ty;
}

632 633 634 635 636 637 638 639 640
// Name sanitation. LLVM will happily accept identifiers with weird names, but
// gas doesn't!

fn sanitize(str s) -> str {
    auto result = "";
    for (u8 c in s) {
        if (c == ('@' as u8)) {
            result += "boxed_";
        } else {
641 642 643 644 645 646 647
            if (c == (',' as u8)) {
                result += "_";
            } else {
                if (c == ('{' as u8) || c == ('(' as u8)) {
                    result += "_of_";
                } else {
                    if (c != 10u8 && c != ('}' as u8) && c != (')' as u8) &&
G
Graydon Hoare 已提交
648 649
                        c != (' ' as u8) && c != ('\t' as u8) &&
                        c != (';' as u8)) {
650 651 652 653 654
                        auto v = vec(c);
                        result += _str.from_bytes(v);
                    }
                }
            }
655 656 657 658 659
        }
    }
    ret result;
}

660 661 662 663 664 665
// LLVM constant constructors.

fn C_null(TypeRef t) -> ValueRef {
    ret llvm.LLVMConstNull(t);
}

666
fn C_integral(int i, TypeRef t) -> ValueRef {
667 668 669 670 671 672
    // FIXME. We can't use LLVM.ULongLong with our existing minimal native
    // API, which only knows word-sized args.  Lucky for us LLVM has a "take a
    // string encoding" version.  Hilarious. Please fix to handle:
    //
    // ret llvm.LLVMConstInt(T_int(), t as LLVM.ULongLong, False);
    //
673 674 675
    ret llvm.LLVMConstIntOfString(t, _str.buf(istr(i)), 10);
}

676 677 678 679 680
fn C_nil() -> ValueRef {
    // NB: See comment above in T_void().
    ret C_integral(0, T_i1());
}

681 682
fn C_bool(bool b) -> ValueRef {
    if (b) {
683
        ret C_integral(1, T_bool());
684
    } else {
685
        ret C_integral(0, T_bool());
686 687 688
    }
}

689 690
fn C_int(int i) -> ValueRef {
    ret C_integral(i, T_int());
691 692
}

693 694 695
// This is a 'c-like' raw string, which differs from
// our boxed-and-length-annotated strings.
fn C_cstr(@crate_ctxt cx, str s) -> ValueRef {
696
    auto sc = llvm.LLVMConstString(_str.buf(s), _str.byte_len(s), False);
697
    auto g = llvm.LLVMAddGlobal(cx.llmod, val_ty(sc),
698 699
                                _str.buf(cx.names.next("str")));
    llvm.LLVMSetInitializer(g, sc);
700
    llvm.LLVMSetGlobalConstant(g, True);
701 702
    llvm.LLVMSetLinkage(g, lib.llvm.LLVMPrivateLinkage
                        as llvm.Linkage);
703
    ret g;
704 705
}

706 707 708 709 710 711 712 713 714 715 716 717 718 719 720 721 722
// A rust boxed-and-length-annotated string.
fn C_str(@crate_ctxt cx, str s) -> ValueRef {
    auto len = _str.byte_len(s);
    auto box = C_struct(vec(C_int(abi.const_refcount as int),
                            C_int(len + 1u as int), // 'alloc'
                            C_int(len + 1u as int), // 'fill'
                            llvm.LLVMConstString(_str.buf(s),
                                                 len, False)));
    auto g = llvm.LLVMAddGlobal(cx.llmod, val_ty(box),
                                _str.buf(cx.names.next("str")));
    llvm.LLVMSetInitializer(g, box);
    llvm.LLVMSetGlobalConstant(g, True);
    llvm.LLVMSetLinkage(g, lib.llvm.LLVMPrivateLinkage
                        as llvm.Linkage);
    ret llvm.LLVMConstPointerCast(g, T_ptr(T_str()));
}

723 724 725 726 727 728 729 730 731 732 733
fn C_zero_byte_arr(uint size) -> ValueRef {
    auto i = 0u;
    let vec[ValueRef] elts = vec();
    while (i < size) {
        elts += vec(C_integral(0, T_i8()));
        i += 1u;
    }
    ret llvm.LLVMConstArray(T_i8(), _vec.buf[ValueRef](elts),
                            _vec.len[ValueRef](elts));
}

734 735 736 737 738 739
fn C_struct(vec[ValueRef] elts) -> ValueRef {
    ret llvm.LLVMConstStruct(_vec.buf[ValueRef](elts),
                             _vec.len[ValueRef](elts),
                             False);
}

740
fn decl_fn(ModuleRef llmod, str name, uint cc, TypeRef llty) -> ValueRef {
741 742
    let ValueRef llfn =
        llvm.LLVMAddFunction(llmod, _str.buf(name), llty);
743
    llvm.LLVMSetFunctionCallConv(llfn, cc);
744 745 746
    ret llfn;
}

747 748
fn decl_cdecl_fn(ModuleRef llmod, str name, TypeRef llty) -> ValueRef {
    ret decl_fn(llmod, name, lib.llvm.LLVMCCallConv, llty);
749 750
}

751 752
fn decl_fastcall_fn(ModuleRef llmod, str name, TypeRef llty) -> ValueRef {
    ret decl_fn(llmod, name, lib.llvm.LLVMFastCallConv, llty);
753 754
}

755 756
fn decl_glue(ModuleRef llmod, type_names tn, str s) -> ValueRef {
    ret decl_cdecl_fn(llmod, s, T_fn(vec(T_taskptr(tn)), T_void()));
757 758
}

759
fn decl_upcall_glue(ModuleRef llmod, type_names tn, uint _n) -> ValueRef {
760 761 762 763
    // It doesn't actually matter what type we come up with here, at the
    // moment, as we cast the upcall function pointers to int before passing
    // them to the indirect upcall-invocation glue.  But eventually we'd like
    // to call them directly, once we have a calling convention worked out.
764
    let int n = _n as int;
765
    let str s = abi.upcall_glue_name(n);
766
    let vec[TypeRef] args =
767
        vec(T_taskptr(tn), // taskptr
768
            T_int())     // callee
769 770
        + _vec.init_elt[TypeRef](T_int(), n as uint);

771
    ret decl_fastcall_fn(llmod, s, T_fn(args, T_int()));
772 773
}

774
fn get_upcall(@crate_ctxt cx, str name, int n_args) -> ValueRef {
775 776 777
    if (cx.upcalls.contains_key(name)) {
        ret cx.upcalls.get(name);
    }
778
    auto inputs = vec(T_taskptr(cx.tn));
779
    inputs += _vec.init_elt[TypeRef](T_int(), n_args as uint);
780
    auto output = T_int();
781
    auto f = decl_cdecl_fn(cx.llmod, name, T_fn(inputs, output));
782 783 784 785
    cx.upcalls.insert(name, f);
    ret f;
}

786
fn trans_upcall(@block_ctxt cx, str name, vec[ValueRef] args) -> result {
787
    let int n = _vec.len[ValueRef](args) as int;
788
    let ValueRef llupcall = get_upcall(cx.fcx.ccx, name, n);
789 790
    llupcall = llvm.LLVMConstPointerCast(llupcall, T_int());

791
    let ValueRef llglue = cx.fcx.ccx.glues.upcall_glues.(n);
792
    let vec[ValueRef] call_args = vec(cx.fcx.lltaskptr, llupcall);
793

794 795 796
    for (ValueRef a in args) {
        call_args += cx.build.ZExtOrBitCast(a, T_int());
    }
797

798
    ret res(cx, cx.build.FastCall(llglue, call_args));
799 800
}

801 802 803
fn trans_non_gc_free(@block_ctxt cx, ValueRef v) -> result {
    ret trans_upcall(cx, "upcall_free", vec(cx.build.PtrToInt(v, T_int()),
                                            C_int(0)));
804 805
}

806
fn find_scope_cx(@block_ctxt cx) -> @block_ctxt {
807
    if (cx.kind == SCOPE_BLOCK) {
808 809 810 811 812 813 814 815 816 817 818 819
        ret cx;
    }
    alt (cx.parent) {
        case (parent_some(?b)) {
            be find_scope_cx(b);
        }
        case (parent_none) {
            fail;
        }
    }
}

820 821 822 823 824 825 826 827 828 829 830 831
fn umax(@block_ctxt cx, ValueRef a, ValueRef b) -> ValueRef {
    auto cond = cx.build.ICmp(lib.llvm.LLVMIntULT, a, b);
    ret cx.build.Select(cond, b, a);
}

fn align_to(@block_ctxt cx, ValueRef off, ValueRef align) -> ValueRef {
    auto mask = cx.build.Sub(align, C_int(1));
    auto bumped = cx.build.Add(off, mask);
    ret cx.build.And(bumped, cx.build.Not(mask));
}

fn llsize_of(TypeRef t) -> ValueRef {
832 833 834
    ret llvm.LLVMConstIntCast(lib.llvm.llvm.LLVMSizeOf(t), T_int(), False);
}

835
fn llalign_of(TypeRef t) -> ValueRef {
836 837 838
    ret llvm.LLVMConstIntCast(lib.llvm.llvm.LLVMAlignOf(t), T_int(), False);
}

839
fn size_of(@block_ctxt cx, @ty.t t) -> result {
840
    if (!ty.type_has_dynamic_size(t)) {
841
        ret res(cx, llsize_of(type_of(cx.fcx.ccx, t)));
842 843 844 845
    }
    ret dynamic_size_of(cx, t);
}

846
fn align_of(@block_ctxt cx, @ty.t t) -> result {
847
    if (!ty.type_has_dynamic_size(t)) {
848
        ret res(cx, llalign_of(type_of(cx.fcx.ccx, t)));
849 850 851 852
    }
    ret dynamic_align_of(cx, t);
}

853
fn dynamic_size_of(@block_ctxt cx, @ty.t t) -> result {
854 855 856
    alt (t.struct) {
        case (ty.ty_param(?p)) {
            auto szptr = field_of_tydesc(cx, t, abi.tydesc_field_size);
857
            ret res(szptr.bcx, szptr.bcx.build.Load(szptr.val));
858 859 860 861 862 863 864 865 866 867 868 869
        }
        case (ty.ty_tup(?elts)) {
            //
            // C padding rules:
            //
            //
            //   - Pad after each element so that next element is aligned.
            //   - Pad after final structure member so that whole structure
            //     is aligned to max alignment of interior.
            //
            auto off = C_int(0);
            auto max_align = C_int(1);
870
            auto bcx = cx;
871
            for (@ty.t e in elts) {
872 873 874 875 876 877 878
                auto elt_align = align_of(bcx, e);
                bcx = elt_align.bcx;
                auto elt_size = size_of(bcx, e);
                bcx = elt_size.bcx;
                auto aligned_off = align_to(bcx, off, elt_align.val);
                off = cx.build.Add(aligned_off, elt_size.val);
                max_align = umax(bcx, max_align, elt_align.val);
879
            }
880 881
            off = align_to(bcx, off, max_align);
            ret res(bcx, off);
882 883 884 885
        }
        case (ty.ty_rec(?flds)) {
            auto off = C_int(0);
            auto max_align = C_int(1);
886
            auto bcx = cx;
887
            for (ty.field f in flds) {
888 889 890 891 892 893 894
                auto elt_align = align_of(bcx, f.ty);
                bcx = elt_align.bcx;
                auto elt_size = size_of(bcx, f.ty);
                bcx = elt_size.bcx;
                auto aligned_off = align_to(bcx, off, elt_align.val);
                off = cx.build.Add(aligned_off, elt_size.val);
                max_align = umax(bcx, max_align, elt_align.val);
895
            }
896 897
            off = align_to(bcx, off, max_align);
            ret res(bcx, off);
898 899 900 901
        }
    }
}

902
fn dynamic_align_of(@block_ctxt cx, @ty.t t) -> result {
903 904 905
    alt (t.struct) {
        case (ty.ty_param(?p)) {
            auto aptr = field_of_tydesc(cx, t, abi.tydesc_field_align);
906
            ret res(aptr.bcx, aptr.bcx.build.Load(aptr.val));
907 908 909
        }
        case (ty.ty_tup(?elts)) {
            auto a = C_int(1);
910
            auto bcx = cx;
911
            for (@ty.t e in elts) {
912 913 914
                auto align = align_of(bcx, e);
                bcx = align.bcx;
                a = umax(bcx, a, align.val);
915
            }
916
            ret res(bcx, a);
917 918 919
        }
        case (ty.ty_rec(?flds)) {
            auto a = C_int(1);
920
            auto bcx = cx;
921
            for (ty.field f in flds) {
922 923 924
                auto align = align_of(bcx, f.ty);
                bcx = align.bcx;
                a = umax(bcx, a, align.val);
925
            }
926
            ret res(bcx, a);
927 928 929 930
        }
    }
}

931 932 933 934 935 936 937
// Replacement for the LLVM 'GEP' instruction when field-indexing into a
// tuple-like structure (tup, rec, tag) with a static index. This one is
// driven off ty.struct and knows what to do when it runs into a ty_param
// stuck in the middle of the thing it's GEP'ing into. Much like size_of and
// align_of, above.

fn GEP_tup_like(@block_ctxt cx, @ty.t t,
938
                ValueRef base, vec[int] ixs) -> result {
939 940 941 942 943 944 945 946 947 948

    check (ty.type_is_tup_like(t));

    // It might be a static-known type. Handle this.

    if (! ty.type_has_dynamic_size(t)) {
        let vec[ValueRef] v = vec();
        for (int i in ixs) {
            v += C_int(i);
        }
949
        ret res(cx, cx.build.GEP(base, v));
950 951 952 953 954 955 956 957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 974 975 976 977 978 979 980 981 982 983 984 985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018
    }

    // It is a dynamic-containing type that, if we convert directly to an LLVM
    // TypeRef, will be all wrong; there's no proper LLVM type to represent
    // it, and the lowering function will stick in i8* values for each
    // ty_param, which is not right; the ty_params are all of some dynamic
    // size.
    //
    // What we must do instead is sadder. We must look through the indices
    // manually and split the input type into a prefix and a target. We then
    // measure the prefix size, bump the input pointer by that amount, and
    // cast to a pointer-to-target type.


    // Given a type, an index vector and an element number N in that vector,
    // calculate index X and the type that results by taking the first X-1
    // elements of the type and splitting the Xth off. Return the prefix as
    // well as the innermost Xth type.

    fn split_type(@ty.t t, vec[int] ixs, uint n)
        -> rec(vec[@ty.t] prefix, @ty.t target) {

        let uint len = _vec.len[int](ixs);

        // We don't support 0-index or 1-index GEPs. The former is nonsense
        // and the latter would only be meaningful if we supported non-0
        // values for the 0th index (we don't).

        check (len > 1u);

        if (n == 0u) {
            // Since we're starting from a value that's a pointer to a
            // *single* structure, the first index (in GEP-ese) should just be
            // 0, to yield the pointee.
            check (ixs.(n) == 0);
            ret split_type(t, ixs, n+1u);
        }

        check (n < len);

        let int ix = ixs.(n);
        let vec[@ty.t] prefix = vec();
        let int i = 0;
        while (i < ix) {
            append[@ty.t](prefix, ty.get_element_type(t, i as uint));
            i +=1 ;
        }

        auto selected = ty.get_element_type(t, i as uint);

        if (n == len-1u) {
            // We are at the innermost index.
            ret rec(prefix=prefix, target=selected);

        } else {
            // Not the innermost index; call self recursively to dig deeper.
            // Once we get an inner result, append it current prefix and
            // return to caller.
            auto inner = split_type(selected, ixs, n+1u);
            prefix += inner.prefix;
            ret rec(prefix=prefix with inner);
        }
    }

    // We make a fake prefix tuple-type here; luckily for measuring sizes
    // the tuple parens are associative so it doesn't matter that we've
    // flattened the incoming structure.

    auto s = split_type(t, ixs, 0u);
1019
    auto prefix_ty = plain_ty(ty.ty_tup(s.prefix));
1020 1021 1022 1023 1024
    auto bcx = cx;
    auto sz = size_of(bcx, prefix_ty);
    bcx = sz.bcx;
    auto raw = bcx.build.PointerCast(base, T_ptr(T_i8()));
    auto bumped = bcx.build.GEP(raw, vec(sz.val));
1025
    alt (s.target.struct) {
1026
        case (ty.ty_param(_)) { ret res(bcx, bumped); }
1027
        case (_) {
1028 1029
            auto ty = T_ptr(type_of(bcx.fcx.ccx, s.target));
            ret res(bcx, bcx.build.PointerCast(bumped, ty));
1030 1031 1032 1033 1034
        }
    }
}


1035 1036
fn trans_malloc_inner(@block_ctxt cx, TypeRef llptr_ty) -> result {
    auto llbody_ty = lib.llvm.llvm.LLVMGetElementType(llptr_ty);
1037 1038
    // FIXME: need a table to collect tydesc globals.
    auto tydesc = C_int(0);
1039
    auto sz = llsize_of(llbody_ty);
1040
    auto sub = trans_upcall(cx, "upcall_malloc", vec(sz, tydesc));
1041 1042 1043 1044 1045 1046 1047 1048
    sub.val = sub.bcx.build.IntToPtr(sub.val, llptr_ty);
    ret sub;
}

fn trans_malloc(@block_ctxt cx, @ty.t t) -> result {
    auto scope_cx = find_scope_cx(cx);
    auto llptr_ty = type_of(cx.fcx.ccx, t);
    auto sub = trans_malloc_inner(cx, llptr_ty);
1049
    scope_cx.cleanups += clean(bind drop_ty(_, sub.val, t));
1050 1051 1052 1053
    ret sub;
}


1054 1055 1056 1057 1058
// Type descriptor and type glue stuff

// Given a type and a field index into its corresponding type descriptor,
// returns an LLVM ValueRef of that field from the tydesc, generating the
// tydesc if necessary.
1059
fn field_of_tydesc(@block_ctxt cx, @ty.t t, int field) -> result {
1060
    auto tydesc = get_tydesc(cx, t);
1061 1062
    ret res(tydesc.bcx,
            tydesc.bcx.build.GEP(tydesc.val, vec(C_int(0), C_int(field))));
1063
}
1064

1065 1066 1067 1068 1069 1070 1071 1072 1073
// Given a type containing ty params, build a vector containing a ValueRef for
// each of the ty params it uses (from the current frame), as well as a vec
// containing a def_id for each such param. This is used solely for
// constructing derived tydescs.
fn linearize_ty_params(@block_ctxt cx, @ty.t t)
    -> tup(vec[ast.def_id], vec[ValueRef]) {
    let vec[ValueRef] param_vals = vec();
    let vec[ast.def_id] param_defs = vec();
    type rr = rec(@block_ctxt cx,
1074 1075
                  mutable vec[ValueRef] vals,
                  mutable vec[ast.def_id] defs);
1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087

    state obj folder(@rr r) {
        fn fold_simple_ty(@ty.t t) -> @ty.t {
            alt(t.struct) {
                case (ty.ty_param(?pid)) {
                    let bool seen = false;
                    for (ast.def_id d in r.defs) {
                        if (d == pid) {
                            seen = true;
                        }
                    }
                    if (!seen) {
1088
                        r.vals += r.cx.fcx.lltydescs.get(pid);
1089 1090 1091
                        r.defs += pid;
                    }
                }
1092
                case (_) { }
1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107
            }
            ret t;
        }
    }


    auto x = @rec(cx = cx,
                  mutable vals = param_vals,
                  mutable defs = param_defs);

    ty.fold_ty(folder(x), t);

    ret tup(x.defs, x.vals);
}

1108
fn get_tydesc(&@block_ctxt cx, @ty.t t) -> result {
1109
    // Is the supplied type a type param? If so, return the passed-in tydesc.
1110
    alt (ty.type_param(t)) {
1111
        case (some[ast.def_id](?id)) {
1112
            check (cx.fcx.lltydescs.contains_key(id));
1113 1114
            ret res(cx, cx.fcx.lltydescs.get(id));
        }
1115
        case (none[ast.def_id])      { /* fall through */ }
1116
    }
1117 1118

    // Does it contain a type param? If so, generate a derived tydesc.
1119
    let uint n_params = ty.count_ty_params(t);
1120

1121
    if (ty.count_ty_params(t) > 0u) {
1122
        auto tys = linearize_ty_params(cx, t);
1123

1124 1125 1126
        check (n_params == _vec.len[ast.def_id](tys._0));
        check (n_params == _vec.len[ValueRef](tys._1));

1127 1128 1129 1130
        if (!cx.fcx.ccx.tydescs.contains_key(t)) {
            make_tydesc(cx.fcx.ccx, t, tys._0);
        }

1131 1132
        auto root = cx.fcx.ccx.tydescs.get(t);

1133 1134
        auto tydescs = cx.build.Alloca(T_array(T_ptr(T_tydesc(cx.fcx.ccx.tn)),
                                               n_params));
1135

1136
        auto i = 0;
1137 1138 1139
        auto tdp = cx.build.GEP(tydescs, vec(C_int(0), C_int(i)));
        cx.build.Store(root, tdp);
        i += 1;
1140 1141
        for (ValueRef td in tys._1) {
            auto tdp = cx.build.GEP(tydescs, vec(C_int(0), C_int(i)));
1142
            cx.build.Store(td, tdp);
1143
            i += 1;
1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155
        }

        auto bcx = cx;
        auto sz = size_of(bcx, t);
        bcx = sz.bcx;
        auto align = align_of(bcx, t);
        bcx = align.bcx;

        auto v = trans_upcall(bcx, "upcall_get_type_desc",
                              vec(p2i(bcx.fcx.ccx.crate_ptr),
                                  sz.val,
                                  align.val,
1156
                                  C_int((1u + n_params) as int),
1157 1158
                                  bcx.build.PtrToInt(tydescs, T_int())));

1159 1160
        ret res(v.bcx, v.bcx.build.IntToPtr(v.val,
                                            T_ptr(T_tydesc(cx.fcx.ccx.tn))));
1161 1162 1163
    }

    // Otherwise, generate a tydesc if necessary, and return it.
1164
    if (!cx.fcx.ccx.tydescs.contains_key(t)) {
1165 1166
        let vec[ast.def_id] defs = vec();
        make_tydesc(cx.fcx.ccx, t, defs);
1167
    }
1168
    ret res(cx, cx.fcx.ccx.tydescs.get(t));
1169 1170
}

1171
fn make_tydesc(@crate_ctxt cx, @ty.t t, vec[ast.def_id] typaram_defs) {
1172
    auto tg = make_take_glue;
1173
    auto take_glue = make_generic_glue(cx, t, "take", tg, typaram_defs);
1174
    auto dg = make_drop_glue;
1175
    auto drop_glue = make_generic_glue(cx, t, "drop", dg, typaram_defs);
1176

1177
    auto llty = type_of(cx, t);
1178
    auto glue_fn_ty = T_ptr(T_glue_fn(cx.tn));
1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190

    // FIXME: this adjustment has to do with the ridiculous encoding of
    // glue-pointer-constants in the tydesc records: They are tydesc-relative
    // displacements.  This is purely for compatibility with rustboot and
    // should go when it is discarded.
    fn off(ValueRef tydescp,
           ValueRef gluefn) -> ValueRef {
        ret i2p(llvm.LLVMConstSub(p2i(gluefn), p2i(tydescp)),
                val_ty(gluefn));
    }

    auto name = sanitize(cx.names.next("tydesc_" + ty.ty_to_str(t)));
1191 1192 1193
    auto gvar = llvm.LLVMAddGlobal(cx.llmod, T_tydesc(cx.tn),
                                   _str.buf(name));
    auto tydesc = C_struct(vec(C_null(T_ptr(T_ptr(T_tydesc(cx.tn)))),
1194 1195
                               llsize_of(llty),
                               llalign_of(llty),
1196 1197
                               off(gvar, take_glue),  // take_glue_off
                               off(gvar, drop_glue),  // drop_glue_off
1198 1199 1200 1201 1202 1203 1204 1205
                               C_null(glue_fn_ty),    // free_glue_off
                               C_null(glue_fn_ty),    // sever_glue_off
                               C_null(glue_fn_ty),    // mark_glue_off
                               C_null(glue_fn_ty),    // obj_drop_glue_off
                               C_null(glue_fn_ty)));  // is_stateful

    llvm.LLVMSetInitializer(gvar, tydesc);
    llvm.LLVMSetGlobalConstant(gvar, True);
1206 1207
    llvm.LLVMSetLinkage(gvar, lib.llvm.LLVMPrivateLinkage
                        as llvm.Linkage);
1208
    cx.tydescs.insert(t, gvar);
1209 1210
}

1211
fn make_generic_glue(@crate_ctxt cx, @ty.t t, str name,
1212 1213
                     val_and_ty_fn helper,
                     vec[ast.def_id] typaram_defs) -> ValueRef {
1214
    auto llfnty = T_glue_fn(cx.tn);
1215

1216
    auto fn_name = cx.names.next("_rust_" + name) + sep() + ty.ty_to_str(t);
1217 1218 1219 1220 1221 1222
    fn_name = sanitize(fn_name);
    auto llfn = decl_fastcall_fn(cx.llmod, fn_name, llfnty);

    auto fcx = new_fn_ctxt(cx, fn_name, llfn);
    auto bcx = new_top_block_ctxt(fcx);

1223
    auto re;
1224
    if (!ty.type_is_scalar(t)) {
1225
        auto llty;
1226
        if (ty.type_is_structural(t)) {
1227 1228 1229 1230
            llty = T_ptr(type_of(cx, t));
        } else {
            llty = type_of(cx, t);
        }
1231

1232
        auto lltyparams = llvm.LLVMGetParam(llfn, 3u);
1233 1234 1235 1236 1237 1238 1239 1240
        auto p = 0;
        for (ast.def_id d in typaram_defs) {
            auto llparam = bcx.build.GEP(lltyparams, vec(C_int(p)));
            llparam = bcx.build.Load(llparam);
            bcx.fcx.lltydescs.insert(d, llparam);
            p += 1;
        }

1241
        auto llrawptr = llvm.LLVMGetParam(llfn, 4u);
1242
        auto llval = bcx.build.BitCast(llrawptr, llty);
G
Graydon Hoare 已提交
1243

1244 1245 1246 1247
        re = helper(bcx, llval, t);
    } else {
        re = res(bcx, C_nil());
    }
1248

1249
    re.bcx.build.RetVoid();
1250 1251 1252
    ret llfn;
}

1253 1254
fn make_take_glue(@block_ctxt cx, ValueRef v, @ty.t t) -> result {
    if (ty.type_is_boxed(t)) {
1255 1256
        ret incr_refcnt_of_boxed(cx, v);

1257
    } else if (ty.type_is_structural(t)) {
1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282
        ret iter_structural_ty(cx, v, t,
                               bind incr_all_refcnts(_, _, _));
    }
    ret res(cx, C_nil());
}

fn incr_refcnt_of_boxed(@block_ctxt cx, ValueRef box_ptr) -> result {
    auto rc_ptr = cx.build.GEP(box_ptr, vec(C_int(0),
                                            C_int(abi.box_rc_field_refcnt)));
    auto rc = cx.build.Load(rc_ptr);

    auto rc_adj_cx = new_sub_block_ctxt(cx, "rc++");
    auto next_cx = new_sub_block_ctxt(cx, "next");

    auto const_test = cx.build.ICmp(lib.llvm.LLVMIntEQ,
                                    C_int(abi.const_refcount as int), rc);
    cx.build.CondBr(const_test, next_cx.llbb, rc_adj_cx.llbb);

    rc = rc_adj_cx.build.Add(rc, C_int(1));
    rc_adj_cx.build.Store(rc, rc_ptr);
    rc_adj_cx.build.Br(next_cx.llbb);

    ret res(next_cx, C_nil());
}

1283
fn make_drop_glue(@block_ctxt cx, ValueRef v, @ty.t t) -> result {
1284
    alt (t.struct) {
1285
        case (ty.ty_str) {
G
Graydon Hoare 已提交
1286 1287 1288 1289
            ret decr_refcnt_and_if_zero
                (cx, v, bind trans_non_gc_free(_, v),
                 "free string",
                 T_int(), C_int(0));
1290 1291
        }

1292
        case (ty.ty_vec(_)) {
G
Graydon Hoare 已提交
1293 1294 1295 1296
            fn hit_zero(@block_ctxt cx, ValueRef v,
                        @ty.t t) -> result {
                auto res = iter_sequence(cx, v, t,
                                         bind drop_ty(_,_,_));
1297 1298 1299 1300 1301 1302 1303 1304 1305
                // FIXME: switch gc/non-gc on layer of the type.
                ret trans_non_gc_free(res.bcx, v);
            }
            ret decr_refcnt_and_if_zero(cx, v,
                                        bind hit_zero(_, v, t),
                                        "free vector",
                                        T_int(), C_int(0));
        }

1306
        case (ty.ty_box(?body_ty)) {
G
Graydon Hoare 已提交
1307 1308
            fn hit_zero(@block_ctxt cx, ValueRef v,
                        @ty.t body_ty) -> result {
1309 1310 1311 1312
                auto body = cx.build.GEP(v,
                                         vec(C_int(0),
                                             C_int(abi.box_rc_field_body)));

1313
                auto body_val = load_scalar_or_boxed(cx, body, body_ty);
1314 1315 1316 1317 1318 1319 1320 1321 1322 1323
                auto res = drop_ty(cx, body_val, body_ty);
                // FIXME: switch gc/non-gc on layer of the type.
                ret trans_non_gc_free(res.bcx, v);
            }
            ret decr_refcnt_and_if_zero(cx, v,
                                        bind hit_zero(_, v, body_ty),
                                        "free box",
                                        T_int(), C_int(0));
        }

1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336
        case (ty.ty_obj(_)) {
            fn hit_zero(@block_ctxt cx, ValueRef v) -> result {

                // Call through the obj's own fields-drop glue first.
                auto body =
                    cx.build.GEP(v,
                                 vec(C_int(0),
                                     C_int(abi.box_rc_field_body)));

                auto tydescptr =
                    cx.build.GEP(body,
                                 vec(C_int(0),
                                     C_int(abi.obj_body_elt_tydesc)));
1337

1338
                call_tydesc_glue_full(cx, body, cx.build.Load(tydescptr),
1339
                                      abi.tydesc_field_drop_glue_off);
1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357

                // Then free the body.
                // FIXME: switch gc/non-gc on layer of the type.
                ret trans_non_gc_free(cx, v);
            }
            auto box_cell =
                cx.build.GEP(v,
                             vec(C_int(0),
                                 C_int(abi.obj_field_box)));

            auto boxptr = cx.build.Load(box_cell);

            ret decr_refcnt_and_if_zero(cx, boxptr,
                                        bind hit_zero(_, boxptr),
                                        "free obj",
                                        T_int(), C_int(0));
        }

1358
        case (ty.ty_fn(_,_,_)) {
1359 1360 1361 1362 1363 1364 1365
            fn hit_zero(@block_ctxt cx, ValueRef v) -> result {

                // Call through the closure's own fields-drop glue first.
                auto body =
                    cx.build.GEP(v,
                                 vec(C_int(0),
                                     C_int(abi.box_rc_field_body)));
1366 1367 1368 1369
                auto bindings =
                    cx.build.GEP(body,
                                 vec(C_int(0),
                                     C_int(abi.closure_elt_bindings)));
1370 1371 1372 1373 1374

                auto tydescptr =
                    cx.build.GEP(body,
                                 vec(C_int(0),
                                     C_int(abi.closure_elt_tydesc)));
1375

1376
                call_tydesc_glue_full(cx, bindings, cx.build.Load(tydescptr),
1377 1378
                                      abi.tydesc_field_drop_glue_off);

1379 1380 1381 1382 1383 1384 1385 1386 1387 1388 1389 1390 1391 1392 1393 1394 1395 1396

                // Then free the body.
                // FIXME: switch gc/non-gc on layer of the type.
                ret trans_non_gc_free(cx, v);
            }
            auto box_cell =
                cx.build.GEP(v,
                             vec(C_int(0),
                                 C_int(abi.fn_field_box)));

            auto boxptr = cx.build.Load(box_cell);

            ret decr_refcnt_and_if_zero(cx, boxptr,
                                        bind hit_zero(_, boxptr),
                                        "free fn",
                                        T_int(), C_int(0));
        }

1397
        case (_) {
1398
            if (ty.type_is_structural(t)) {
1399 1400 1401
                ret iter_structural_ty(cx, v, t,
                                       bind drop_ty(_, _, _));

1402 1403
            } else if (ty.type_is_scalar(t) ||
                       ty.type_is_nil(t)) {
1404 1405 1406 1407
                ret res(cx, C_nil());
            }
        }
    }
1408
    cx.fcx.ccx.sess.bug("bad type in trans.make_drop_glue_inner: " +
1409
                        ty.ty_to_str(t));
1410 1411 1412
    fail;
}

1413 1414
fn decr_refcnt_and_if_zero(@block_ctxt cx,
                           ValueRef box_ptr,
1415
                           fn(@block_ctxt cx) -> result inner,
1416
                           str inner_name,
1417
                           TypeRef t_else, ValueRef v_else) -> result {
1418

1419
    auto load_rc_cx = new_sub_block_ctxt(cx, "load rc");
1420 1421 1422 1423
    auto rc_adj_cx = new_sub_block_ctxt(cx, "rc--");
    auto inner_cx = new_sub_block_ctxt(cx, inner_name);
    auto next_cx = new_sub_block_ctxt(cx, "next");

1424 1425
    auto null_test = cx.build.IsNull(box_ptr);
    cx.build.CondBr(null_test, next_cx.llbb, load_rc_cx.llbb);
1426

1427 1428 1429 1430 1431 1432 1433 1434 1435 1436

    auto rc_ptr = load_rc_cx.build.GEP(box_ptr,
                                       vec(C_int(0),
                                           C_int(abi.box_rc_field_refcnt)));

    auto rc = load_rc_cx.build.Load(rc_ptr);
    auto const_test =
        load_rc_cx.build.ICmp(lib.llvm.LLVMIntEQ,
                              C_int(abi.const_refcount as int), rc);
    load_rc_cx.build.CondBr(const_test, next_cx.llbb, rc_adj_cx.llbb);
1437 1438 1439 1440

    rc = rc_adj_cx.build.Sub(rc, C_int(1));
    rc_adj_cx.build.Store(rc, rc_ptr);
    auto zero_test = rc_adj_cx.build.ICmp(lib.llvm.LLVMIntEQ, C_int(0), rc);
1441 1442 1443 1444
    rc_adj_cx.build.CondBr(zero_test, inner_cx.llbb, next_cx.llbb);

    auto inner_res = inner(inner_cx);
    inner_res.bcx.build.Br(next_cx.llbb);
1445

1446
    auto phi = next_cx.build.Phi(t_else,
1447
                                 vec(v_else, v_else, v_else, inner_res.val),
1448
                                 vec(cx.llbb,
1449
                                     load_rc_cx.llbb,
1450
                                     rc_adj_cx.llbb,
1451 1452
                                     inner_res.bcx.llbb));

1453
    ret res(next_cx, phi);
1454 1455
}

1456 1457
// Tag information

1458 1459
fn type_of_variant(@crate_ctxt cx, &ast.variant v) -> TypeRef {
    let vec[TypeRef] lltys = vec();
1460
    alt (ty.ann_to_type(v.ann).struct) {
1461
        case (ty.ty_fn(_, ?args, _)) {
1462
            for (ty.arg arg in args) {
1463 1464 1465 1466 1467 1468 1469 1470
                lltys += vec(type_of(cx, arg.ty));
            }
        }
        case (_) { fail; }
    }
    ret T_struct(lltys);
}

1471 1472 1473 1474 1475 1476 1477 1478 1479
// Returns the type parameters of a tag.
fn tag_ty_params(@crate_ctxt cx, ast.def_id id) -> vec[ast.ty_param] {
    check (cx.items.contains_key(id));
    alt (cx.items.get(id).node) {
        case (ast.item_tag(_, _, ?tps, _)) { ret tps; }
    }
    fail;   // not reached
}

1480 1481
// Returns the variants in a tag.
fn tag_variants(@crate_ctxt cx, ast.def_id id) -> vec[ast.variant] {
1482 1483
    check (cx.items.contains_key(id));
    alt (cx.items.get(id).node) {
1484
        case (ast.item_tag(_, ?variants, _, _)) { ret variants; }
1485 1486 1487 1488
    }
    fail;   // not reached
}

1489 1490 1491 1492 1493 1494 1495 1496
// Returns a new plain tag type of the given ID with no type parameters. Don't
// use this function in new code; it's a hack to keep things working for now.
fn mk_plain_tag(ast.def_id tid) -> @ty.t {
    let vec[@ty.t] tps = vec();
    ret ty.plain_ty(ty.ty_tag(tid, tps));
}


1497
type val_and_ty_fn = fn(@block_ctxt cx, ValueRef v, @ty.t t) -> result;
1498

1499 1500 1501
type val_pair_and_ty_fn =
    fn(@block_ctxt cx, ValueRef av, ValueRef bv, @ty.t t) -> result;

1502
// Iterates through the elements of a structural type.
1503 1504
fn iter_structural_ty(@block_ctxt cx,
                      ValueRef v,
1505
                      @ty.t t,
1506 1507
                      val_and_ty_fn f)
    -> result {
1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525
    fn adaptor_fn(val_and_ty_fn f,
                  @block_ctxt cx,
                  ValueRef av,
                  ValueRef bv,
                  @ty.t t) -> result {
        ret f(cx, av, t);
    }
    be iter_structural_ty_full(cx, v, v, t,
                               bind adaptor_fn(f, _, _, _, _));
}


fn iter_structural_ty_full(@block_ctxt cx,
                           ValueRef av,
                           ValueRef bv,
                           @ty.t t,
                           val_pair_and_ty_fn f)
    -> result {
1526
    let result r = res(cx, C_nil());
1527

1528
    fn iter_boxpp(@block_ctxt cx,
1529 1530 1531 1532 1533
                  ValueRef box_a_cell,
                  ValueRef box_b_cell,
                  val_pair_and_ty_fn f) -> result {
        auto box_a_ptr = cx.build.Load(box_a_cell);
        auto box_b_ptr = cx.build.Load(box_b_cell);
1534 1535
        auto tnil = plain_ty(ty.ty_nil);
        auto tbox = plain_ty(ty.ty_box(tnil));
1536 1537 1538

        auto inner_cx = new_sub_block_ctxt(cx, "iter box");
        auto next_cx = new_sub_block_ctxt(cx, "next");
1539
        auto null_test = cx.build.IsNull(box_a_ptr);
1540 1541
        cx.build.CondBr(null_test, next_cx.llbb, inner_cx.llbb);

1542
        auto r = f(inner_cx, box_a_ptr, box_b_ptr, tbox);
1543 1544 1545 1546
        r.bcx.build.Br(next_cx.llbb);
        ret res(next_cx, r.val);
    }

1547
    alt (t.struct) {
1548
        case (ty.ty_tup(?args)) {
1549
            let int i = 0;
1550
            for (@ty.t arg in args) {
1551 1552
                auto elt_a = r.bcx.build.GEP(av, vec(C_int(0), C_int(i)));
                auto elt_b = r.bcx.build.GEP(bv, vec(C_int(0), C_int(i)));
1553
                r = f(r.bcx,
1554 1555
                      load_scalar_or_boxed(r.bcx, elt_a, arg),
                      load_scalar_or_boxed(r.bcx, elt_b, arg),
1556
                      arg);
1557 1558 1559
                i += 1;
            }
        }
1560
        case (ty.ty_rec(?fields)) {
1561
            let int i = 0;
1562
            for (ty.field fld in fields) {
1563 1564
                auto llfld_a = r.bcx.build.GEP(av, vec(C_int(0), C_int(i)));
                auto llfld_b = r.bcx.build.GEP(bv, vec(C_int(0), C_int(i)));
1565
                r = f(r.bcx,
1566 1567
                      load_scalar_or_boxed(r.bcx, llfld_a, fld.ty),
                      load_scalar_or_boxed(r.bcx, llfld_b, fld.ty),
1568
                      fld.ty);
1569 1570 1571
                i += 1;
            }
        }
1572
        case (ty.ty_tag(?tid, ?tps)) {
1573
            auto info = cx.fcx.ccx.tags.get(mk_plain_tag(tid));
1574 1575 1576

            auto variants = tag_variants(cx.fcx.ccx, tid);
            auto n_variants = _vec.len[ast.variant](variants);
1577

1578 1579 1580 1581 1582 1583 1584
            auto lldiscrim_a_ptr = cx.build.GEP(av, vec(C_int(0), C_int(0)));
            auto llunion_a_ptr = cx.build.GEP(av, vec(C_int(0), C_int(1)));
            auto lldiscrim_a = cx.build.Load(lldiscrim_a_ptr);

            auto lldiscrim_b_ptr = cx.build.GEP(bv, vec(C_int(0), C_int(0)));
            auto llunion_b_ptr = cx.build.GEP(bv, vec(C_int(0), C_int(1)));
            auto lldiscrim_b = cx.build.Load(lldiscrim_b_ptr);
G
Graydon Hoare 已提交
1585

1586 1587 1588
            auto unr_cx = new_sub_block_ctxt(cx, "tag-iter-unr");
            unr_cx.build.Unreachable();

1589 1590
            auto llswitch = cx.build.Switch(lldiscrim_a, unr_cx.llbb,
                                             n_variants);
G
Graydon Hoare 已提交
1591

1592 1593 1594
            auto next_cx = new_sub_block_ctxt(cx, "tag-iter-next");

            auto i = 0u;
1595
            for (ast.variant variant in variants) {
1596 1597
                auto variant_cx = new_sub_block_ctxt(cx,
                                                     "tag-iter-variant-" +
1598 1599 1600
                                                     _uint.to_str(i, 10u));
                llvm.LLVMAddCase(llswitch, C_int(i as int), variant_cx.llbb);

1601 1602 1603 1604 1605 1606 1607
                if (_vec.len[ast.variant_arg](variant.args) > 0u) {
                    // N-ary variant.
                    auto llvarty = type_of_variant(cx.fcx.ccx, variants.(i));

                    auto fn_ty = ty.ann_to_type(variants.(i).ann);
                    alt (fn_ty.struct) {
                        case (ty.ty_fn(_, ?args, _)) {
1608 1609 1610 1611 1612
                            auto llvarp_a = variant_cx.build.
                                TruncOrBitCast(llunion_a_ptr, T_ptr(llvarty));

                            auto llvarp_b = variant_cx.build.
                                TruncOrBitCast(llunion_b_ptr, T_ptr(llvarty));
1613

1614 1615
                            auto ty_params = tag_ty_params(cx.fcx.ccx, tid);

1616 1617 1618
                            auto j = 0u;
                            for (ty.arg a in args) {
                                auto v = vec(C_int(0), C_int(j as int));
1619 1620 1621 1622 1623 1624

                                auto llfldp_a =
                                    variant_cx.build.GEP(llvarp_a, v);

                                auto llfldp_b =
                                    variant_cx.build.GEP(llvarp_b, v);
1625 1626

                                auto ty_subst = ty.substitute_ty_params(
1627
                                    ty_params, tps, a.ty);
1628

1629
                                auto llfld_a =
1630
                                    load_scalar_or_boxed(variant_cx,
1631
                                                         llfldp_a,
1632 1633
                                                         ty_subst);

1634 1635 1636 1637 1638 1639 1640
                                auto llfld_b =
                                    load_scalar_or_boxed(variant_cx,
                                                         llfldp_b,
                                                         ty_subst);

                                auto res = f(variant_cx,
                                             llfld_a, llfld_b, ty_subst);
1641 1642
                                variant_cx = res.bcx;
                                j += 1u;
1643 1644
                            }
                        }
1645
                        case (_) { fail; }
1646
                    }
1647 1648 1649 1650 1651

                    variant_cx.build.Br(next_cx.llbb);
                } else {
                    // Nullary variant; nothing to do.
                    variant_cx.build.Br(next_cx.llbb);
1652 1653 1654 1655 1656 1657 1658
                }

                i += 1u;
            }

            ret res(next_cx, C_nil());
        }
1659
        case (ty.ty_fn(_,_,_)) {
1660 1661 1662 1663 1664 1665
            auto box_cell_a =
                cx.build.GEP(av,
                             vec(C_int(0),
                                 C_int(abi.fn_field_box)));
            auto box_cell_b =
                cx.build.GEP(bv,
1666 1667
                             vec(C_int(0),
                                 C_int(abi.fn_field_box)));
1668
            ret iter_boxpp(cx, box_cell_a, box_cell_b, f);
1669
        }
1670
        case (ty.ty_obj(_)) {
1671 1672 1673 1674 1675 1676
            auto box_cell_a =
                cx.build.GEP(av,
                             vec(C_int(0),
                                 C_int(abi.obj_field_box)));
            auto box_cell_b =
                cx.build.GEP(bv,
1677 1678
                             vec(C_int(0),
                                 C_int(abi.obj_field_box)));
1679
            ret iter_boxpp(cx, box_cell_a, box_cell_b, f);
1680
        }
1681
        case (_) {
1682
            cx.fcx.ccx.sess.unimpl("type in iter_structural_ty_full");
1683
        }
1684
    }
1685
    ret r;
1686 1687
}

1688 1689 1690
// Iterates through the elements of a vec or str.
fn iter_sequence(@block_ctxt cx,
                 ValueRef v,
1691
                 @ty.t t,
1692 1693 1694 1695
                 val_and_ty_fn f) -> result {

    fn iter_sequence_body(@block_ctxt cx,
                          ValueRef v,
1696
                          @ty.t elt_ty,
1697 1698 1699 1700 1701 1702 1703
                          val_and_ty_fn f,
                          bool trailing_null) -> result {

        auto p0 = cx.build.GEP(v, vec(C_int(0),
                                      C_int(abi.vec_elt_data)));
        auto lenptr = cx.build.GEP(v, vec(C_int(0),
                                          C_int(abi.vec_elt_fill)));
1704 1705

        auto llunit_ty = type_of(cx.fcx.ccx, elt_ty);
1706 1707 1708
        auto bcx = cx;
        auto unit_sz = size_of(bcx, elt_ty);
        bcx = unit_sz.bcx;
1709

1710
        auto len = bcx.build.Load(lenptr);
1711
        if (trailing_null) {
1712
            len = bcx.build.Sub(len, unit_sz.val);
1713 1714
        }

1715 1716
        auto cond_cx = new_scope_block_ctxt(cx, "sequence-iter cond");
        auto body_cx = new_scope_block_ctxt(cx, "sequence-iter body");
1717 1718
        auto next_cx = new_sub_block_ctxt(cx, "next");

1719
        bcx.build.Br(cond_cx.llbb);
1720

1721
        auto ix = cond_cx.build.Phi(T_int(), vec(C_int(0)), vec(cx.llbb));
1722 1723 1724 1725 1726
        auto scaled_ix = cond_cx.build.Phi(T_int(),
                                           vec(C_int(0)), vec(cx.llbb));

        auto end_test = cond_cx.build.ICmp(lib.llvm.LLVMIntNE,
                                           scaled_ix, len);
1727 1728
        cond_cx.build.CondBr(end_test, body_cx.llbb, next_cx.llbb);

1729
        auto elt = body_cx.build.GEP(p0, vec(C_int(0), ix));
1730
        auto body_res = f(body_cx,
1731
                          load_scalar_or_boxed(body_cx, elt, elt_ty),
1732
                          elt_ty);
1733
        auto next_ix = body_res.bcx.build.Add(ix, C_int(1));
1734
        auto next_scaled_ix = body_res.bcx.build.Add(scaled_ix, unit_sz.val);
1735

1736 1737 1738
        cond_cx.build.AddIncomingToPhi(ix, vec(next_ix),
                                       vec(body_res.bcx.llbb));

1739 1740 1741
        cond_cx.build.AddIncomingToPhi(scaled_ix, vec(next_scaled_ix),
                                       vec(body_res.bcx.llbb));

1742 1743 1744 1745
        body_res.bcx.build.Br(cond_cx.llbb);
        ret res(next_cx, C_nil());
    }

1746 1747
    alt (t.struct) {
        case (ty.ty_vec(?et)) {
1748 1749
            ret iter_sequence_body(cx, v, et, f, false);
        }
1750
        case (ty.ty_str) {
1751
            auto et = plain_ty(ty.ty_machine(common.ty_u8));
1752
            ret iter_sequence_body(cx, v, et, f, true);
1753
        }
1754
        case (_) { fail; }
1755
    }
1756 1757 1758 1759
    cx.fcx.ccx.sess.bug("bad type in trans.iter_sequence");
    fail;
}

1760 1761 1762 1763 1764 1765 1766 1767 1768
fn call_tydesc_glue_full(@block_ctxt cx, ValueRef v,
                         ValueRef tydesc, int field) {
    auto llrawptr = cx.build.BitCast(v, T_ptr(T_i8()));
    auto lltydescs = cx.build.GEP(tydesc,
                                  vec(C_int(0),
                                      C_int(abi.tydesc_field_first_param)));
    lltydescs = cx.build.Load(lltydescs);
    auto llfnptr = cx.build.GEP(tydesc, vec(C_int(0), C_int(field)));
    auto llfn = cx.build.Load(llfnptr);
1769 1770 1771 1772 1773 1774 1775 1776 1777

    // FIXME: this adjustment has to do with the ridiculous encoding of
    // glue-pointer-constants in the tydesc records: They are tydesc-relative
    // displacements.  This is purely for compatibility with rustboot and
    // should go when it is discarded.
    llfn = cx.build.IntToPtr(cx.build.Add(cx.build.PtrToInt(llfn, T_int()),
                                          cx.build.PtrToInt(tydesc, T_int())),
                             val_ty(llfn));

1778 1779 1780 1781 1782
    cx.build.FastCall(llfn, vec(C_null(T_ptr(T_nil())),
                                cx.fcx.lltaskptr,
                                C_null(T_ptr(T_nil())),
                                lltydescs,
                                llrawptr));
1783 1784 1785
}

fn call_tydesc_glue(@block_ctxt cx, ValueRef v, @ty.t t, int field) {
1786 1787
    auto td = get_tydesc(cx, t);
    call_tydesc_glue_full(td.bcx, v, td.val, field);
1788 1789
}

1790 1791
fn incr_all_refcnts(@block_ctxt cx,
                    ValueRef v,
1792
                    @ty.t t) -> result {
1793

1794
    if (!ty.type_is_scalar(t)) {
1795
        call_tydesc_glue(cx, v, t, abi.tydesc_field_take_glue_off);
1796
    }
1797
    ret res(cx, C_nil());
1798 1799
}

1800 1801
fn drop_slot(@block_ctxt cx,
             ValueRef slot,
1802
             @ty.t t) -> result {
1803
    auto llptr = load_scalar_or_boxed(cx, slot, t);
1804 1805 1806 1807 1808 1809
    auto re = drop_ty(cx, llptr, t);

    auto llty = val_ty(slot);
    auto llelemty = lib.llvm.llvm.LLVMGetElementType(llty);
    re.bcx.build.Store(C_null(llelemty), slot);
    ret re;
1810 1811
}

1812 1813
fn drop_ty(@block_ctxt cx,
           ValueRef v,
1814
           @ty.t t) -> result {
1815

1816
    if (!ty.type_is_scalar(t)) {
1817
        call_tydesc_glue(cx, v, t, abi.tydesc_field_drop_glue_off);
1818
    }
1819
    ret res(cx, C_nil());
1820 1821
}

1822 1823 1824 1825
fn call_memcpy(@block_ctxt cx,
               ValueRef dst,
               ValueRef src,
               ValueRef n_bytes) -> result {
1826 1827
    auto src_ptr = cx.build.PointerCast(src, T_ptr(T_i8()));
    auto dst_ptr = cx.build.PointerCast(dst, T_ptr(T_i8()));
1828 1829 1830
    auto size = cx.build.IntCast(n_bytes, T_int());
    ret res(cx, cx.build.FastCall(cx.fcx.ccx.glues.memcpy_glue,
                                  vec(dst_ptr, src_ptr, size)));
1831 1832
}

1833 1834 1835 1836 1837 1838 1839 1840 1841
fn call_bzero(@block_ctxt cx,
              ValueRef dst,
              ValueRef n_bytes) -> result {
    auto dst_ptr = cx.build.PointerCast(dst, T_ptr(T_i8()));
    auto size = cx.build.IntCast(n_bytes, T_int());
    ret res(cx, cx.build.FastCall(cx.fcx.ccx.glues.bzero_glue,
                                  vec(dst_ptr, size)));
}

1842 1843 1844 1845 1846 1847
fn memcpy_ty(@block_ctxt cx,
             ValueRef dst,
             ValueRef src,
             @ty.t t) -> result {
    if (ty.type_has_dynamic_size(t)) {
        auto llszptr = field_of_tydesc(cx, t, abi.tydesc_field_size);
1848 1849
        auto llsz = llszptr.bcx.build.Load(llszptr.val);
        ret call_memcpy(llszptr.bcx, dst, src, llsz);
1850 1851 1852 1853 1854 1855

    } else {
        ret res(cx, cx.build.Store(cx.build.Load(src), dst));
    }
}

1856 1857 1858 1859 1860
tag copy_action {
    INIT;
    DROP_EXISTING;
}

1861
fn copy_ty(@block_ctxt cx,
1862
           copy_action action,
1863 1864
           ValueRef dst,
           ValueRef src,
1865 1866
           @ty.t t) -> result {
    if (ty.type_is_scalar(t)) {
1867 1868
        ret res(cx, cx.build.Store(src, dst));

1869
    } else if (ty.type_is_nil(t)) {
1870 1871
        ret res(cx, C_nil());

1872
    } else if (ty.type_is_boxed(t)) {
1873
        auto r = incr_all_refcnts(cx, src, t);
1874
        if (action == DROP_EXISTING) {
1875
            r = drop_ty(r.bcx, r.bcx.build.Load(dst), t);
1876 1877 1878
        }
        ret res(r.bcx, r.bcx.build.Store(src, dst));

1879 1880
    } else if (ty.type_is_structural(t) ||
               ty.type_has_dynamic_size(t)) {
1881
        auto r = incr_all_refcnts(cx, src, t);
1882
        if (action == DROP_EXISTING) {
1883 1884
            r = drop_ty(r.bcx, dst, t);
        }
1885
        ret memcpy_ty(r.bcx, dst, src, t);
1886 1887 1888
    }

    cx.fcx.ccx.sess.bug("unexpected type in trans.copy_ty: " +
1889
                        ty.ty_to_str(t));
1890 1891 1892
    fail;
}

1893
fn trans_lit(@crate_ctxt cx, &ast.lit lit, &ast.ann ann) -> ValueRef {
1894
    alt (lit.node) {
1895
        case (ast.lit_int(?i)) {
1896
            ret C_int(i);
1897 1898
        }
        case (ast.lit_uint(?u)) {
1899
            ret C_int(u as int);
1900
        }
1901 1902 1903 1904 1905 1906
        case (ast.lit_mach_int(?tm, ?i)) {
            // FIXME: the entire handling of mach types falls apart
            // if target int width is larger than host, at the moment;
            // re-do the mach-int types using 'big' when that works.
            auto t = T_int();
            alt (tm) {
G
Graydon Hoare 已提交
1907 1908 1909 1910 1911 1912 1913 1914 1915
                case (common.ty_u8) { t = T_i8(); }
                case (common.ty_u16) { t = T_i16(); }
                case (common.ty_u32) { t = T_i32(); }
                case (common.ty_u64) { t = T_i64(); }

                case (common.ty_i8) { t = T_i8(); }
                case (common.ty_i16) { t = T_i16(); }
                case (common.ty_i32) { t = T_i32(); }
                case (common.ty_i64) { t = T_i64(); }
1916
            }
1917
            ret C_integral(i, t);
1918
        }
1919
        case (ast.lit_char(?c)) {
1920
            ret C_integral(c as int, T_char());
1921 1922
        }
        case (ast.lit_bool(?b)) {
1923
            ret C_bool(b);
1924 1925
        }
        case (ast.lit_nil) {
1926
            ret C_nil();
1927 1928
        }
        case (ast.lit_str(?s)) {
1929
            ret C_str(cx, s);
1930 1931 1932 1933
        }
    }
}

1934
fn target_type(@crate_ctxt cx, @ty.t t) -> @ty.t {
1935
    alt (t.struct) {
1936 1937
        case (ty.ty_int) {
            auto tm = ty.ty_machine(cx.sess.get_targ_cfg().int_type);
1938 1939
            ret @rec(struct=tm with *t);
        }
1940 1941
        case (ty.ty_uint) {
            auto tm = ty.ty_machine(cx.sess.get_targ_cfg().uint_type);
1942 1943
            ret @rec(struct=tm with *t);
        }
1944
        case (_) { /* fall through */ }
1945 1946 1947 1948
    }
    ret t;
}

1949
fn node_ann_type(@crate_ctxt cx, &ast.ann a) -> @ty.t {
1950 1951
    alt (a) {
        case (ast.ann_none) {
1952
            cx.sess.bug("missing type annotation");
1953 1954
        }
        case (ast.ann_type(?t)) {
1955
            ret target_type(cx, t);
1956 1957 1958 1959
        }
    }
}

1960 1961 1962 1963
fn node_type(@crate_ctxt cx, &ast.ann a) -> TypeRef {
    ret type_of(cx, node_ann_type(cx, a));
}

1964 1965
fn trans_unary(@block_ctxt cx, ast.unop op,
               @ast.expr e, &ast.ann a) -> result {
1966 1967 1968

    auto sub = trans_expr(cx, e);

1969 1970
    alt (op) {
        case (ast.bitnot) {
1971
            sub = autoderef(sub.bcx, sub.val, ty.expr_ty(e));
1972
            ret res(sub.bcx, cx.build.Not(sub.val));
1973 1974
        }
        case (ast.not) {
1975
            sub = autoderef(sub.bcx, sub.val, ty.expr_ty(e));
1976
            ret res(sub.bcx, cx.build.Not(sub.val));
1977 1978
        }
        case (ast.neg) {
1979
            sub = autoderef(sub.bcx, sub.val, ty.expr_ty(e));
1980
            ret res(sub.bcx, cx.build.Neg(sub.val));
1981
        }
1982
        case (ast.box) {
1983
            auto e_ty = ty.expr_ty(e);
1984 1985 1986 1987
            auto e_val = sub.val;
            sub = trans_malloc(sub.bcx, node_ann_type(sub.bcx.fcx.ccx, a));
            auto box = sub.val;
            auto rc = sub.bcx.build.GEP(box,
1988 1989
                                        vec(C_int(0),
                                            C_int(abi.box_rc_field_refcnt)));
1990 1991 1992 1993
            auto body = sub.bcx.build.GEP(box,
                                          vec(C_int(0),
                                              C_int(abi.box_rc_field_body)));
            sub.bcx.build.Store(C_int(1), rc);
1994
            sub = copy_ty(sub.bcx, INIT, body, e_val, e_ty);
1995
            ret res(sub.bcx, box);
1996
        }
1997
        case (ast.deref) {
1998 1999 2000
            auto val = sub.bcx.build.GEP(sub.val,
                                         vec(C_int(0),
                                             C_int(abi.box_rc_field_body)));
2001
            auto e_ty = node_ann_type(sub.bcx.fcx.ccx, a);
2002 2003
            if (ty.type_is_scalar(e_ty) ||
                ty.type_is_nil(e_ty)) {
2004
                val = sub.bcx.build.Load(val);
2005
            }
2006
            ret res(sub.bcx, val);
2007
        }
G
Graydon Hoare 已提交
2008 2009 2010
        case (ast._mutable) {
            ret trans_expr(cx, e);
        }
2011 2012 2013 2014
    }
    fail;
}

2015 2016 2017 2018 2019 2020 2021
fn trans_compare(@block_ctxt cx, ast.binop op, @ty.t t,
                 ValueRef lhs, ValueRef rhs) -> result {

    if (ty.type_is_scalar(t)) {
        ret res(cx, trans_scalar_compare(cx, op, t, lhs, rhs));

    } else if (ty.type_is_structural(t)) {
2022 2023
        auto scx = new_sub_block_ctxt(cx, "structural compare start");
        auto next = new_sub_block_ctxt(cx, "structural compare end");
2024 2025 2026 2027 2028 2029
        cx.build.Br(scx.llbb);

        // Start with the assumptioin that our predicate holds.
        auto flag = scx.build.Alloca(T_i1());
        scx.build.Store(C_integral(1, T_i1()), flag);

2030 2031 2032 2033 2034 2035 2036 2037 2038 2039 2040 2041 2042 2043 2044 2045 2046 2047 2048 2049 2050 2051 2052 2053 2054 2055
        // Attempt to prove otherwise by assuming true, comparing each element
        // and writing 0 + early-exiting if any comparisons fail.

        fn inner(@block_ctxt next_cx,
                 ValueRef flag,
                 ast.binop op,
                 @block_ctxt cx,
                 ValueRef av,
                 ValueRef bv,
                 @ty.t t) -> result {
            // Compare av op bv
            auto cnt_cx = new_sub_block_ctxt(cx, "continue comparison");
            auto stop_cx = new_sub_block_ctxt(cx, "stop comparison");

            auto r = trans_compare(cx, op, t, av, bv);

            // if true, then carry on, else write 0 to flag, branch to 'next'.
            r.bcx.build.CondBr(r.val, cnt_cx.llbb, stop_cx.llbb);
            stop_cx.build.Store(C_integral(0, T_i1()), flag);
            stop_cx.build.Br(next_cx.llbb);

            ret res(cnt_cx, C_nil());
        }

        // FIXME: this is wrong for tag types; need to confirm discriminants
        // are equal before blindly walking over elements.
2056

2057 2058 2059
        auto r = iter_structural_ty_full(scx, lhs, rhs, t,
                                         bind inner(next, flag, op,
                                                    _, _, _, _));
2060

2061 2062
        r.bcx.build.Br(next.llbb);
        auto v = next.build.Load(flag);
2063 2064 2065 2066 2067 2068 2069 2070 2071 2072 2073 2074 2075 2076 2077 2078 2079 2080 2081 2082
        ret res(next, v);

    } else {
        // FIXME: compare vec, str, box?
        cx.fcx.ccx.sess.unimpl("type in trans_compare");
        ret res(cx, C_bool(false));
    }
}

fn trans_scalar_compare(@block_ctxt cx, ast.binop op, @ty.t t,
                        ValueRef lhs, ValueRef rhs) -> ValueRef {
    if (ty.type_is_fp(t)) {
        ret trans_fp_compare(cx, op, t, lhs, rhs);
    } else {
        ret trans_integral_compare(cx, op, t, lhs, rhs);
    }
}

fn trans_fp_compare(@block_ctxt cx, ast.binop op, @ty.t fptype,
                    ValueRef lhs, ValueRef rhs) -> ValueRef {
2083

2084 2085 2086 2087 2088 2089 2090 2091 2092 2093 2094 2095 2096 2097 2098 2099 2100
    auto cmp = lib.llvm.LLVMIntEQ;
    alt (op) {
        // FIXME: possibly use the unordered-or-< predicates here,
        // for now we're only going with ordered-and-< style (no NaNs).
        case (ast.eq) { cmp = lib.llvm.LLVMRealOEQ; }
        case (ast.ne) { cmp = lib.llvm.LLVMRealONE; }
        case (ast.lt) { cmp = lib.llvm.LLVMRealOLT; }
        case (ast.gt) { cmp = lib.llvm.LLVMRealOGT; }
        case (ast.le) { cmp = lib.llvm.LLVMRealOLE; }
        case (ast.ge) { cmp = lib.llvm.LLVMRealOGE; }
    }

    ret cx.build.FCmp(cmp, lhs, rhs);
}

fn trans_integral_compare(@block_ctxt cx, ast.binop op, @ty.t intype,
                          ValueRef lhs, ValueRef rhs) -> ValueRef {
2101 2102 2103 2104 2105
    auto cmp = lib.llvm.LLVMIntEQ;
    alt (op) {
        case (ast.eq) { cmp = lib.llvm.LLVMIntEQ; }
        case (ast.ne) { cmp = lib.llvm.LLVMIntNE; }

2106 2107 2108 2109 2110 2111 2112 2113 2114 2115 2116 2117 2118 2119 2120 2121 2122 2123 2124 2125 2126 2127 2128 2129 2130 2131 2132 2133
        case (ast.lt) {
            if (ty.type_is_signed(intype)) {
                cmp = lib.llvm.LLVMIntSLT;
            } else {
                cmp = lib.llvm.LLVMIntULT;
            }
        }
        case (ast.le) {
            if (ty.type_is_signed(intype)) {
                cmp = lib.llvm.LLVMIntSLE;
            } else {
                cmp = lib.llvm.LLVMIntULE;
            }
        }
        case (ast.gt) {
            if (ty.type_is_signed(intype)) {
                cmp = lib.llvm.LLVMIntSGT;
            } else {
                cmp = lib.llvm.LLVMIntUGT;
            }
        }
        case (ast.ge) {
            if (ty.type_is_signed(intype)) {
                cmp = lib.llvm.LLVMIntSGE;
            } else {
                cmp = lib.llvm.LLVMIntUGE;
            }
        }
2134 2135 2136 2137
    }
    ret cx.build.ICmp(cmp, lhs, rhs);
}

2138
fn trans_eager_binop(@block_ctxt cx, ast.binop op, @ty.t intype,
2139
                     ValueRef lhs, ValueRef rhs) -> result {
2140 2141

    alt (op) {
2142 2143
        case (ast.add) { ret res(cx, cx.build.Add(lhs, rhs)); }
        case (ast.sub) { ret res(cx, cx.build.Sub(lhs, rhs)); }
2144

2145
        case (ast.mul) { ret res(cx, cx.build.Mul(lhs, rhs)); }
2146 2147
        case (ast.div) {
            if (ty.type_is_signed(intype)) {
2148
                ret res(cx, cx.build.SDiv(lhs, rhs));
2149
            } else {
2150
                ret res(cx, cx.build.UDiv(lhs, rhs));
2151 2152 2153 2154
            }
        }
        case (ast.rem) {
            if (ty.type_is_signed(intype)) {
2155
                ret res(cx, cx.build.SRem(lhs, rhs));
2156
            } else {
2157
                ret res(cx, cx.build.URem(lhs, rhs));
2158 2159
            }
        }
2160

2161 2162 2163 2164 2165 2166
        case (ast.bitor) { ret res(cx, cx.build.Or(lhs, rhs)); }
        case (ast.bitand) { ret res(cx, cx.build.And(lhs, rhs)); }
        case (ast.bitxor) { ret res(cx, cx.build.Xor(lhs, rhs)); }
        case (ast.lsl) { ret res(cx, cx.build.Shl(lhs, rhs)); }
        case (ast.lsr) { ret res(cx, cx.build.LShr(lhs, rhs)); }
        case (ast.asr) { ret res(cx, cx.build.AShr(lhs, rhs)); }
2167
        case (_) {
2168
            ret trans_compare(cx, op, intype, lhs, rhs);
2169 2170 2171 2172 2173
        }
    }
    fail;
}

2174 2175 2176 2177 2178 2179 2180 2181 2182 2183 2184 2185 2186 2187 2188 2189 2190 2191 2192 2193
fn autoderef(@block_ctxt cx, ValueRef v, @ty.t t) -> result {
    let ValueRef v1 = v;
    let @ty.t t1 = t;

    while (true) {
        alt (t1.struct) {
            case (ty.ty_box(?inner)) {
                auto body = cx.build.GEP(v1,
                                         vec(C_int(0),
                                             C_int(abi.box_rc_field_body)));
                t1 = inner;
                v1 = load_scalar_or_boxed(cx, body, inner);
            }
            case (_) {
                ret res(cx, v1);
            }
        }
    }
}

2194 2195 2196 2197 2198 2199 2200 2201 2202 2203 2204 2205 2206 2207 2208
fn autoderefed_ty(@ty.t t) -> @ty.t {
    let @ty.t t1 = t;

    while (true) {
        alt (t1.struct) {
            case (ty.ty_box(?inner)) {
                t1 = inner;
            }
            case (_) {
                ret t1;
            }
        }
    }
}

2209 2210
fn trans_binary(@block_ctxt cx, ast.binop op,
                @ast.expr a, @ast.expr b) -> result {
2211

2212 2213 2214 2215 2216 2217
    // First couple cases are lazy:

    alt (op) {
        case (ast.and) {
            // Lazy-eval and
            auto lhs_res = trans_expr(cx, a);
2218
            lhs_res = autoderef(lhs_res.bcx, lhs_res.val, ty.expr_ty(a));
2219

2220
            auto rhs_cx = new_scope_block_ctxt(cx, "rhs");
2221
            auto rhs_res = trans_expr(rhs_cx, b);
2222
            rhs_res = autoderef(rhs_res.bcx, rhs_res.val, ty.expr_ty(b));
2223

2224
            auto lhs_false_cx = new_scope_block_ctxt(cx, "lhs false");
2225
            auto lhs_false_res = res(lhs_false_cx, C_bool(false));
2226 2227 2228

            lhs_res.bcx.build.CondBr(lhs_res.val,
                                     rhs_cx.llbb,
2229 2230 2231 2232
                                     lhs_false_cx.llbb);

            ret join_results(cx, T_bool(),
                             vec(lhs_false_res, rhs_res));
2233 2234 2235 2236 2237
        }

        case (ast.or) {
            // Lazy-eval or
            auto lhs_res = trans_expr(cx, a);
2238
            lhs_res = autoderef(lhs_res.bcx, lhs_res.val, ty.expr_ty(a));
2239

2240
            auto rhs_cx = new_scope_block_ctxt(cx, "rhs");
2241
            auto rhs_res = trans_expr(rhs_cx, b);
2242
            rhs_res = autoderef(rhs_res.bcx, rhs_res.val, ty.expr_ty(b));
2243

2244
            auto lhs_true_cx = new_scope_block_ctxt(cx, "lhs true");
2245
            auto lhs_true_res = res(lhs_true_cx, C_bool(true));
2246 2247

            lhs_res.bcx.build.CondBr(lhs_res.val,
2248
                                     lhs_true_cx.llbb,
2249
                                     rhs_cx.llbb);
2250 2251 2252

            ret join_results(cx, T_bool(),
                             vec(lhs_true_res, rhs_res));
2253
        }
2254 2255

        case (_) {
2256 2257
            // Remaining cases are eager:
            auto lhs = trans_expr(cx, a);
2258 2259
            auto lhty = ty.expr_ty(a);
            lhs = autoderef(lhs.bcx, lhs.val, lhty);
2260
            auto rhs = trans_expr(lhs.bcx, b);
2261 2262
            auto rhty = ty.expr_ty(b);
            rhs = autoderef(rhs.bcx, rhs.val, rhty);
2263 2264 2265
            ret trans_eager_binop(rhs.bcx, op,
                                  autoderefed_ty(lhty),
                                  lhs.val, rhs.val);
2266
        }
2267 2268 2269 2270
    }
    fail;
}

2271 2272 2273 2274 2275 2276 2277 2278 2279 2280 2281 2282 2283 2284 2285 2286 2287 2288 2289 2290 2291 2292 2293 2294 2295 2296 2297 2298 2299 2300 2301
fn join_results(@block_ctxt parent_cx,
                TypeRef t,
                vec[result] ins)
    -> result {

    let vec[result] live = vec();
    let vec[ValueRef] vals = vec();
    let vec[BasicBlockRef] bbs = vec();

    for (result r in ins) {
        if (! is_terminated(r.bcx)) {
            live += r;
            vals += r.val;
            bbs += r.bcx.llbb;
        }
    }

    alt (_vec.len[result](live)) {
        case (0u) {
            // No incoming edges are live, so we're in dead-code-land.
            // Arbitrarily pick the first dead edge, since the caller
            // is just going to propagate it outward.
            check (_vec.len[result](ins) >= 1u);
            ret ins.(0);
        }

        case (1u) {
            // Only one incoming edge is live, so we just feed that block
            // onward.
            ret live.(0);
        }
2302 2303

        case (_) { /* fall through */ }
2304 2305 2306 2307 2308 2309 2310 2311 2312 2313 2314
    }

    // We have >1 incoming edges. Make a join block and br+phi them into it.
    auto join_cx = new_sub_block_ctxt(parent_cx, "join");
    for (result r in live) {
        r.bcx.build.Br(join_cx.llbb);
    }
    auto phi = join_cx.build.Phi(t, vals, bbs);
    ret res(join_cx, phi);
}

G
Graydon Hoare 已提交
2315 2316 2317
fn trans_if(@block_ctxt cx, @ast.expr cond, &ast.block thn,
            &vec[tup(@ast.expr, ast.block)] elifs,
            &option.t[ast.block] els) -> result {
2318 2319 2320

    auto cond_res = trans_expr(cx, cond);

2321
    auto then_cx = new_scope_block_ctxt(cx, "then");
2322 2323
    auto then_res = trans_block(then_cx, thn);

2324
    auto else_cx = new_scope_block_ctxt(cx, "else");
2325
    auto else_res = res(else_cx, C_nil());
2326

G
Graydon Hoare 已提交
2327 2328 2329 2330 2331 2332 2333 2334 2335 2336 2337 2338 2339 2340 2341 2342 2343 2344
    auto num_elifs = _vec.len[tup(@ast.expr, ast.block)](elifs);
    if (num_elifs > 0u) {
        auto next_elif = elifs.(0u);
        auto next_elifthn = next_elif._0;
        auto next_elifcnd = next_elif._1;
        auto rest_elifs = _vec.shift[tup(@ast.expr, ast.block)](elifs);
        else_res = trans_if(else_cx, next_elifthn, next_elifcnd,
                            rest_elifs, els);
    }

    /* else: FIXME: rustboot has a problem here
       with preconditions inside an else block */
    if (num_elifs == 0u)  {
        alt (els) {
            case (some[ast.block](?eblk)) {
                else_res = trans_block(else_cx, eblk);
            }
            case (_) { /* fall through */ }
2345 2346 2347
        }
    }

2348
    cond_res.bcx.build.CondBr(cond_res.val,
2349 2350
                              then_cx.llbb,
                              else_cx.llbb);
2351 2352 2353 2354

    // FIXME: use inferred type when available.
    ret join_results(cx, T_nil(),
                     vec(then_res, else_res));
2355 2356
}

G
Graydon Hoare 已提交
2357 2358 2359 2360 2361 2362 2363 2364 2365 2366 2367 2368 2369 2370
fn trans_for(@block_ctxt cx,
             @ast.decl decl,
             @ast.expr seq,
             &ast.block body) -> result {

    fn inner(@block_ctxt cx,
             @ast.local local, ValueRef curr,
             @ty.t t, ast.block body) -> result {

        auto scope_cx = new_scope_block_ctxt(cx, "for loop scope");
        auto next_cx = new_sub_block_ctxt(cx, "next");

        cx.build.Br(scope_cx.llbb);
        auto local_res = alloc_local(scope_cx, local);
2371
        auto bcx = copy_ty(local_res.bcx, INIT, local_res.val, curr, t).bcx;
2372
        scope_cx.cleanups += clean(bind drop_slot(_, local_res.val, t));
2373
        bcx = trans_block(bcx, body).bcx;
G
Graydon Hoare 已提交
2374 2375 2376 2377 2378 2379 2380 2381 2382 2383 2384 2385 2386 2387 2388 2389 2390 2391
        bcx.build.Br(next_cx.llbb);
        ret res(next_cx, C_nil());
    }


    let @ast.local local;
    alt (decl.node) {
        case (ast.decl_local(?loc)) {
            local = loc;
        }
    }

    auto seq_ty = ty.expr_ty(seq);
    auto seq_res = trans_expr(cx, seq);
    ret iter_sequence(seq_res.bcx, seq_res.val, seq_ty,
                      bind inner(_, local, _, _, body));
}

2392 2393 2394 2395
fn trans_for_each(@block_ctxt cx,
                  @ast.decl decl,
                  @ast.expr seq,
                  &ast.block body) -> result {
2396 2397 2398 2399 2400 2401 2402 2403 2404 2405 2406 2407 2408 2409 2410 2411 2412 2413 2414 2415 2416 2417 2418 2419 2420 2421 2422

    /*
     * The translation is a little .. complex here. Code like:
     *
     *    let ty1 p = ...;
     *
     *    let ty1 q = ...;
     *
     *    foreach (ty v in foo(a,b)) { body(p,q,v) }
     *
     *
     * Turns into a something like so (C/Rust mishmash):
     *
     *    type env = { *ty1 p, *ty2 q, ... };
     *
     *    let env e = { &p, &q, ... };
     *
     *    fn foreach123_body(env* e, ty v) { body(*(e->p),*(e->q),v) }
     *
     *    foo([foreach123_body, env*], a, b);
     *
     */

    // Step 1: walk body and figure out which references it makes
    // escape. This could be determined upstream, and probably ought
    // to be so, eventualy. For first cut, skip this. Null env.

2423
    auto env_ty = T_opaque_closure_ptr(cx.fcx.ccx.tn);
2424 2425 2426 2427 2428 2429 2430 2431 2432 2433 2434 2435 2436 2437 2438 2439 2440 2441 2442 2443 2444 2445


    // Step 2: Declare foreach body function.

    // FIXME: possibly support alias-mode here?
    auto decl_ty = plain_ty(ty.ty_nil);
    alt (decl.node) {
        case (ast.decl_local(?local)) {
            decl_ty = node_ann_type(cx.fcx.ccx, local.ann);
        }
    }

    let str s =
        cx.fcx.ccx.names.next("_rust_foreach")
        + sep() + cx.fcx.ccx.path;

    // The 'env' arg entering the body function is a fake env member (as in
    // the env-part of the normal rust calling convention) that actually
    // points to a stack allocated env in this frame. We bundle that env
    // pointer along with the foreach-body-fn pointer into a 'normal' fn pair
    // and pass it in as a first class fn-arg to the iterator.

2446 2447 2448 2449
    auto iter_body_llty = type_of_fn_full(cx.fcx.ccx, ast.proto_fn,
                                          none[TypeRef],
                                          vec(rec(mode=ast.val, ty=decl_ty)),
                                          plain_ty(ty.ty_nil));
2450

2451 2452
    let ValueRef lliterbody = decl_fastcall_fn(cx.fcx.ccx.llmod,
                                               s, iter_body_llty);
2453 2454 2455 2456

    // FIXME: handle ty params properly.
    let vec[ast.ty_param] ty_params = vec();

2457
    auto fcx = new_fn_ctxt(cx.fcx.ccx, s, lliterbody);
2458 2459 2460 2461 2462 2463 2464
    auto bcx = new_top_block_ctxt(fcx);

    // FIXME: populate lllocals from llenv here.
    auto res = trans_block(bcx, body);
    res.bcx.build.RetVoid();


2465
    // Step 3: Call iter passing [lliterbody, llenv], plus other args.
2466 2467

    alt (seq.node) {
2468

2469
        case (ast.expr_call(?f, ?args, ?ann)) {
2470

2471 2472 2473 2474 2475 2476 2477
            auto pair = cx.build.Alloca(T_fn_pair(cx.fcx.ccx.tn,
                                                  iter_body_llty));
            auto code_cell = cx.build.GEP(pair,
                                          vec(C_int(0),
                                              C_int(abi.fn_field_code)));
            cx.build.Store(lliterbody, code_cell);

2478 2479
            // log "lliterbody: " + val_str(cx.fcx.ccx.tn, lliterbody);
            ret trans_call(cx, f,
2480
                           some[ValueRef](cx.build.Load(pair)),
2481 2482
                           args,
                           ann);
2483 2484
        }
    }
2485 2486 2487 2488
    fail;
}


2489 2490
fn trans_while(@block_ctxt cx, @ast.expr cond,
               &ast.block body) -> result {
2491

2492 2493
    auto cond_cx = new_scope_block_ctxt(cx, "while cond");
    auto body_cx = new_scope_block_ctxt(cx, "while loop body");
2494
    auto next_cx = new_sub_block_ctxt(cx, "next");
2495 2496

    auto body_res = trans_block(body_cx, body);
2497 2498 2499 2500 2501 2502 2503 2504
    auto cond_res = trans_expr(cond_cx, cond);

    body_res.bcx.build.Br(cond_cx.llbb);
    cond_res.bcx.build.CondBr(cond_res.val,
                              body_cx.llbb,
                              next_cx.llbb);

    cx.build.Br(cond_cx.llbb);
2505 2506 2507
    ret res(next_cx, C_nil());
}

2508 2509
fn trans_do_while(@block_ctxt cx, &ast.block body,
                  @ast.expr cond) -> result {
2510

2511
    auto body_cx = new_scope_block_ctxt(cx, "do-while loop body");
2512
    auto next_cx = new_sub_block_ctxt(cx, "next");
2513 2514

    auto body_res = trans_block(body_cx, body);
2515 2516 2517 2518 2519 2520
    auto cond_res = trans_expr(body_res.bcx, cond);

    cond_res.bcx.build.CondBr(cond_res.val,
                              body_cx.llbb,
                              next_cx.llbb);
    cx.build.Br(body_cx.llbb);
2521 2522 2523
    ret res(next_cx, body_res.val);
}

P
Patrick Walton 已提交
2524 2525 2526 2527 2528
// Pattern matching translation

// Returns a pointer to the union part of the LLVM representation of a tag
// type, cast to the appropriate type.
fn get_pat_union_ptr(@block_ctxt cx, vec[@ast.pat] subpats, ValueRef llval)
G
Graydon Hoare 已提交
2529
    -> ValueRef {
P
Patrick Walton 已提交
2530 2531 2532 2533 2534 2535 2536 2537 2538 2539 2540 2541 2542
    auto llblobptr = cx.build.GEP(llval, vec(C_int(0), C_int(1)));

    // Generate the union type.
    let vec[TypeRef] llsubpattys = vec();
    for (@ast.pat subpat in subpats) {
        llsubpattys += vec(type_of(cx.fcx.ccx, pat_ty(subpat)));
    }

    // Recursively check subpatterns.
    auto llunionty = T_struct(llsubpattys);
    ret cx.build.TruncOrBitCast(llblobptr, T_ptr(llunionty));
}

2543 2544
fn trans_pat_match(@block_ctxt cx, @ast.pat pat, ValueRef llval,
                   @block_ctxt next_cx) -> result {
P
Patrick Walton 已提交
2545 2546 2547
    alt (pat.node) {
        case (ast.pat_wild(_)) { ret res(cx, llval); }
        case (ast.pat_bind(_, _, _)) { ret res(cx, llval); }
2548 2549 2550

        case (ast.pat_lit(?lt, ?ann)) {
            auto lllit = trans_lit(cx.fcx.ccx, *lt, ann);
2551 2552
            auto lltype = ty.ann_to_type(ann);
            auto lleq = trans_compare(cx, ast.eq, lltype, llval, lllit);
2553

2554 2555
            auto matched_cx = new_sub_block_ctxt(lleq.bcx, "matched_cx");
            lleq.bcx.build.CondBr(lleq.val, matched_cx.llbb, next_cx.llbb);
2556 2557 2558
            ret res(matched_cx, llval);
        }

P
Patrick Walton 已提交
2559 2560 2561
        case (ast.pat_tag(?id, ?subpats, ?vdef_opt, ?ann)) {
            auto lltagptr = cx.build.GEP(llval, vec(C_int(0), C_int(0)));
            auto lltag = cx.build.Load(lltagptr);
2562

P
Patrick Walton 已提交
2563 2564 2565
            auto vdef = option.get[ast.variant_def](vdef_opt);
            auto variant_id = vdef._1;
            auto variant_tag = 0;
2566 2567

            auto variants = tag_variants(cx.fcx.ccx, vdef._0);
P
Patrick Walton 已提交
2568
            auto i = 0;
2569 2570
            for (ast.variant v in variants) {
                auto this_variant_id = v.id;
P
Patrick Walton 已提交
2571
                if (variant_id._0 == this_variant_id._0 &&
G
Graydon Hoare 已提交
2572
                    variant_id._1 == this_variant_id._1) {
P
Patrick Walton 已提交
2573 2574 2575 2576 2577 2578 2579 2580 2581 2582 2583 2584 2585 2586 2587 2588 2589 2590 2591
                    variant_tag = i;
                }
                i += 1;
            }

            auto matched_cx = new_sub_block_ctxt(cx, "matched_cx");

            auto lleq = cx.build.ICmp(lib.llvm.LLVMIntEQ, lltag,
                                      C_int(variant_tag));
            cx.build.CondBr(lleq, matched_cx.llbb, next_cx.llbb);

            if (_vec.len[@ast.pat](subpats) > 0u) {
                auto llunionptr = get_pat_union_ptr(matched_cx, subpats,
                                                    llval);
                auto i = 0;
                for (@ast.pat subpat in subpats) {
                    auto llsubvalptr = matched_cx.build.GEP(llunionptr,
                                                            vec(C_int(0),
                                                                C_int(i)));
2592
                    auto llsubval = load_scalar_or_boxed(matched_cx,
G
Graydon Hoare 已提交
2593 2594
                                                         llsubvalptr,
                                                         pat_ty(subpat));
P
Patrick Walton 已提交
2595 2596 2597 2598 2599 2600 2601 2602 2603 2604 2605 2606 2607
                    auto subpat_res = trans_pat_match(matched_cx, subpat,
                                                      llsubval, next_cx);
                    matched_cx = subpat_res.bcx;
                }
            }

            ret res(matched_cx, llval);
        }
    }

    fail;
}

2608 2609
fn trans_pat_binding(@block_ctxt cx, @ast.pat pat, ValueRef llval)
    -> result {
P
Patrick Walton 已提交
2610 2611
    alt (pat.node) {
        case (ast.pat_wild(_)) { ret res(cx, llval); }
2612
        case (ast.pat_lit(_, _)) { ret res(cx, llval); }
P
Patrick Walton 已提交
2613 2614 2615 2616 2617 2618 2619 2620 2621
        case (ast.pat_bind(?id, ?def_id, ?ann)) {
            auto ty = node_ann_type(cx.fcx.ccx, ann);
            auto llty = type_of(cx.fcx.ccx, ty);

            auto dst = cx.build.Alloca(llty);
            llvm.LLVMSetValueName(dst, _str.buf(id));
            cx.fcx.lllocals.insert(def_id, dst);
            cx.cleanups += clean(bind drop_slot(_, dst, ty));

2622
            ret copy_ty(cx, INIT, dst, llval, ty);
P
Patrick Walton 已提交
2623 2624 2625 2626 2627 2628 2629 2630 2631 2632 2633
        }
        case (ast.pat_tag(_, ?subpats, _, _)) {
            if (_vec.len[@ast.pat](subpats) == 0u) { ret res(cx, llval); }

            auto llunionptr = get_pat_union_ptr(cx, subpats, llval);

            auto this_cx = cx;
            auto i = 0;
            for (@ast.pat subpat in subpats) {
                auto llsubvalptr = this_cx.build.GEP(llunionptr,
                                                     vec(C_int(0), C_int(i)));
2634
                auto llsubval = load_scalar_or_boxed(this_cx, llsubvalptr,
G
Graydon Hoare 已提交
2635
                                                     pat_ty(subpat));
P
Patrick Walton 已提交
2636 2637 2638
                auto subpat_res = trans_pat_binding(this_cx, subpat,
                                                    llsubval);
                this_cx = subpat_res.bcx;
2639
                i += 1;
P
Patrick Walton 已提交
2640 2641 2642 2643 2644 2645 2646
            }

            ret res(this_cx, llval);
        }
    }
}

2647 2648
fn trans_alt(@block_ctxt cx, @ast.expr expr, vec[ast.arm] arms)
    -> result {
P
Patrick Walton 已提交
2649 2650 2651 2652 2653 2654 2655 2656 2657 2658 2659 2660 2661 2662 2663 2664 2665 2666 2667 2668 2669 2670 2671 2672 2673 2674 2675 2676 2677 2678 2679 2680 2681
    auto expr_res = trans_expr(cx, expr);

    auto last_cx = new_sub_block_ctxt(expr_res.bcx, "last");

    auto this_cx = expr_res.bcx;
    for (ast.arm arm in arms) {
        auto next_cx = new_sub_block_ctxt(expr_res.bcx, "next");
        auto match_res = trans_pat_match(this_cx, arm.pat, expr_res.val,
                                         next_cx);

        auto binding_cx = new_scope_block_ctxt(match_res.bcx, "binding");
        match_res.bcx.build.Br(binding_cx.llbb);

        auto binding_res = trans_pat_binding(binding_cx, arm.pat,
                                             expr_res.val);

        auto block_res = trans_block(binding_res.bcx, arm.block);
        if (!is_terminated(block_res.bcx)) {
            block_res.bcx.build.Br(last_cx.llbb);
        }

        this_cx = next_cx;
    }

    // FIXME: This is executed when none of the patterns match; it should fail
    // instead!
    this_cx.build.Br(last_cx.llbb);

    // FIXME: This is very wrong; we should phi together all the arm blocks,
    // since this is an expression.
    ret res(last_cx, C_nil());
}

2682
type generic_info = rec(@ty.t item_type,
2683 2684
                        vec[ValueRef] tydescs);

2685 2686
type lval_result = rec(result res,
                       bool is_mem,
2687
                       option.t[generic_info] generic,
2688 2689 2690 2691 2692
                       option.t[ValueRef] llobj);

fn lval_mem(@block_ctxt cx, ValueRef val) -> lval_result {
    ret rec(res=res(cx, val),
            is_mem=true,
2693
            generic=none[generic_info],
2694 2695 2696 2697 2698 2699
            llobj=none[ValueRef]);
}

fn lval_val(@block_ctxt cx, ValueRef val) -> lval_result {
    ret rec(res=res(cx, val),
            is_mem=false,
2700
            generic=none[generic_info],
2701 2702
            llobj=none[ValueRef]);
}
2703

2704 2705 2706 2707 2708 2709 2710 2711 2712 2713 2714 2715 2716 2717 2718 2719 2720 2721 2722 2723 2724 2725 2726 2727 2728 2729 2730 2731
fn lval_generic_fn(@block_ctxt cx,
                   ty.ty_params_and_ty tpt,
                   ast.def_id fn_id,
                   &ast.ann ann)
    -> lval_result {

    check (cx.fcx.ccx.fn_pairs.contains_key(fn_id));
    auto lv = lval_val(cx, cx.fcx.ccx.fn_pairs.get(fn_id));
    auto monoty = node_ann_type(cx.fcx.ccx, ann);
    auto tys = ty.resolve_ty_params(tpt, monoty);

    if (_vec.len[@ty.t](tys) != 0u) {
        auto bcx = cx;
        let vec[ValueRef] tydescs = vec();
        for (@ty.t t in tys) {
            auto td = get_tydesc(bcx, t);
            bcx = td.bcx;
            append[ValueRef](tydescs, td.val);
        }
        auto gen = rec( item_type = tpt._1,
                        tydescs = tydescs );
        lv = rec(res = res(bcx, lv.res.val),
                 generic = some[generic_info](gen)
                 with lv);
    }
    ret lv;
}

2732
fn trans_path(@block_ctxt cx, &ast.path p, &option.t[ast.def] dopt,
G
Graydon Hoare 已提交
2733
              &ast.ann ann) -> lval_result {
2734 2735 2736 2737 2738
    alt (dopt) {
        case (some[ast.def](?def)) {
            alt (def) {
                case (ast.def_arg(?did)) {
                    check (cx.fcx.llargs.contains_key(did));
2739
                    ret lval_mem(cx, cx.fcx.llargs.get(did));
2740 2741 2742
                }
                case (ast.def_local(?did)) {
                    check (cx.fcx.lllocals.contains_key(did));
2743
                    ret lval_mem(cx, cx.fcx.lllocals.get(did));
G
Graydon Hoare 已提交
2744
                }
P
Patrick Walton 已提交
2745 2746
                case (ast.def_binding(?did)) {
                    check (cx.fcx.lllocals.contains_key(did));
2747
                    ret lval_mem(cx, cx.fcx.lllocals.get(did));
P
Patrick Walton 已提交
2748
                }
2749 2750 2751 2752
                case (ast.def_obj_field(?did)) {
                    check (cx.fcx.llobjfields.contains_key(did));
                    ret lval_mem(cx, cx.fcx.llobjfields.get(did));
                }
2753
                case (ast.def_fn(?did)) {
2754
                    check (cx.fcx.ccx.items.contains_key(did));
2755
                    auto fn_item = cx.fcx.ccx.items.get(did);
2756
                    ret lval_generic_fn(cx, ty.item_ty(fn_item), did, ann);
2757
                }
2758
                case (ast.def_obj(?did)) {
2759 2760 2761
                    check (cx.fcx.ccx.items.contains_key(did));
                    auto fn_item = cx.fcx.ccx.items.get(did);
                    ret lval_generic_fn(cx, ty.item_ty(fn_item), did, ann);
2762
                }
2763
                case (ast.def_variant(?tid, ?vid)) {
2764
                    if (cx.fcx.ccx.fn_pairs.contains_key(vid)) {
2765 2766 2767
                        check (cx.fcx.ccx.items.contains_key(tid));
                        auto tag_item = cx.fcx.ccx.items.get(tid);
                        auto params = ty.item_ty(tag_item)._0;
2768
                        auto fty = plain_ty(ty.ty_nil);
2769 2770 2771 2772 2773 2774 2775 2776 2777 2778 2779
                        alt (tag_item.node) {
                            case (ast.item_tag(_, ?variants, _, _)) {
                                for (ast.variant v in variants) {
                                    if (v.id == vid) {
                                        fty = node_ann_type(cx.fcx.ccx,
                                                            v.ann);
                                    }
                                }
                            }
                        }
                        ret lval_generic_fn(cx, tup(params, fty), vid, ann);
2780
                    } else {
2781 2782 2783 2784 2785 2786 2787 2788 2789 2790 2791 2792 2793 2794 2795 2796 2797
                        // Nullary variant.
                        auto tag_ty = node_ann_type(cx.fcx.ccx, ann);
                        auto info = cx.fcx.ccx.tags.get(tag_ty);
                        check (info.lldiscrims.contains_key(vid));
                        auto lldiscrim_gv = info.lldiscrims.get(vid);
                        auto lldiscrim = cx.build.Load(lldiscrim_gv);

                        auto alloc_result = alloc_ty(cx, tag_ty);
                        auto lltagblob = alloc_result.val;
                        auto lltagptr = alloc_result.bcx.build.PointerCast(
                            lltagblob, T_ptr(type_of(cx.fcx.ccx, tag_ty)));

                        auto lldiscrimptr = alloc_result.bcx.build.GEP(
                            lltagptr, vec(C_int(0), C_int(0)));
                        alloc_result.bcx.build.Store(lldiscrim, lldiscrimptr);

                        ret lval_val(alloc_result.bcx, lltagptr);
2798
                    }
2799
                }
2800 2801 2802 2803
                case (ast.def_const(?did)) {
                    check (cx.fcx.ccx.consts.contains_key(did));
                    ret lval_mem(cx, cx.fcx.ccx.consts.get(did));
                }
2804 2805 2806 2807 2808 2809
                case (ast.def_native_fn(?did)) {
                    check (cx.fcx.ccx.native_items.contains_key(did));
                    auto fn_item = cx.fcx.ccx.native_items.get(did);
                    ret lval_generic_fn(cx, ty.native_item_ty(fn_item),
                                        did, ann);
                }
2810 2811
                case (_) {
                    cx.fcx.ccx.sess.unimpl("def variant in trans");
G
Graydon Hoare 已提交
2812 2813 2814
                }
            }
        }
2815
        case (none[ast.def]) {
2816
            cx.fcx.ccx.sess.err("unresolved expr_path in trans");
2817 2818 2819 2820 2821
        }
    }
    fail;
}

2822 2823
fn trans_field(@block_ctxt cx, &ast.span sp, @ast.expr base,
               &ast.ident field, &ast.ann ann) -> lval_result {
2824
    auto r = trans_expr(cx, base);
2825
    auto t = ty.expr_ty(base);
2826 2827
    r = autoderef(r.bcx, r.val, t);
    t = autoderefed_ty(t);
2828 2829 2830
    alt (t.struct) {
        case (ty.ty_tup(?fields)) {
            let uint ix = ty.field_num(cx.fcx.ccx.sess, sp, field);
2831
            auto v = GEP_tup_like(r.bcx, t, r.val, vec(0, ix as int));
2832
            ret lval_mem(v.bcx, v.val);
2833
        }
2834 2835
        case (ty.ty_rec(?fields)) {
            let uint ix = ty.field_idx(cx.fcx.ccx.sess, sp, field, fields);
2836
            auto v = GEP_tup_like(r.bcx, t, r.val, vec(0, ix as int));
2837
            ret lval_mem(v.bcx, v.val);
2838
        }
2839 2840
        case (ty.ty_obj(?methods)) {
            let uint ix = ty.method_idx(cx.fcx.ccx.sess, sp, field, methods);
2841 2842 2843 2844 2845 2846
            auto vtbl = r.bcx.build.GEP(r.val,
                                        vec(C_int(0),
                                            C_int(abi.obj_field_vtbl)));
            vtbl = r.bcx.build.Load(vtbl);
            auto v =  r.bcx.build.GEP(vtbl, vec(C_int(0),
                                                C_int(ix as int)));
2847 2848 2849

            auto lvo = lval_mem(r.bcx, v);
            ret rec(llobj = some[ValueRef](r.val) with lvo);
2850
        }
2851
        case (_) { cx.fcx.ccx.sess.unimpl("field variant in trans_field"); }
2852 2853 2854 2855
    }
    fail;
}

2856 2857
fn trans_index(@block_ctxt cx, &ast.span sp, @ast.expr base,
               @ast.expr idx, &ast.ann ann) -> lval_result {
2858

G
Graydon Hoare 已提交
2859
    auto lv = trans_expr(cx, base);
2860
    lv = autoderef(lv.bcx, lv.val, ty.expr_ty(base));
G
Graydon Hoare 已提交
2861 2862
    auto ix = trans_expr(lv.bcx, idx);
    auto v = lv.val;
2863
    auto bcx = ix.bcx;
2864 2865

    auto llunit_ty = node_type(cx.fcx.ccx, ann);
2866 2867 2868
    auto unit_sz = size_of(bcx, node_ann_type(cx.fcx.ccx, ann));
    bcx = unit_sz.bcx;
    auto scaled_ix = bcx.build.Mul(ix.val, unit_sz.val);
2869

2870 2871
    auto lim = bcx.build.GEP(v, vec(C_int(0), C_int(abi.vec_elt_fill)));
    lim = bcx.build.Load(lim);
2872

2873 2874
    auto bounds_check = bcx.build.ICmp(lib.llvm.LLVMIntULT,
                                       scaled_ix, lim);
2875

2876 2877 2878
    auto fail_cx = new_sub_block_ctxt(bcx, "fail");
    auto next_cx = new_sub_block_ctxt(bcx, "next");
    bcx.build.CondBr(bounds_check, next_cx.llbb, fail_cx.llbb);
2879 2880

    // fail: bad bounds check.
B
Brian Anderson 已提交
2881
    auto fail_res = trans_fail(fail_cx, sp, "bounds check");
2882 2883 2884 2885
    fail_res.bcx.build.Br(next_cx.llbb);

    auto body = next_cx.build.GEP(v, vec(C_int(0), C_int(abi.vec_elt_data)));
    auto elt = next_cx.build.GEP(body, vec(C_int(0), ix.val));
2886
    ret lval_mem(next_cx, elt);
2887 2888
}

2889 2890 2891 2892
// The additional bool returned indicates whether it's mem (that is
// represented as an alloca or heap, hence needs a 'load' to be used as an
// immediate).

2893
fn trans_lval(@block_ctxt cx, @ast.expr e) -> lval_result {
2894
    alt (e.node) {
2895 2896
        case (ast.expr_path(?p, ?dopt, ?ann)) {
            ret trans_path(cx, p, dopt, ann);
2897 2898 2899 2900
        }
        case (ast.expr_field(?base, ?ident, ?ann)) {
            ret trans_field(cx, e.span, base, ident, ann);
        }
2901 2902 2903
        case (ast.expr_index(?base, ?idx, ?ann)) {
            ret trans_index(cx, e.span, base, idx, ann);
        }
2904
        case (_) { cx.fcx.ccx.sess.unimpl("expr variant in trans_lval"); }
G
Graydon Hoare 已提交
2905 2906 2907 2908
    }
    fail;
}

2909
fn trans_cast(@block_ctxt cx, @ast.expr e, &ast.ann ann) -> result {
2910 2911 2912 2913
    auto e_res = trans_expr(cx, e);
    auto llsrctype = val_ty(e_res.val);
    auto t = node_ann_type(cx.fcx.ccx, ann);
    auto lldsttype = type_of(cx.fcx.ccx, t);
2914
    if (!ty.type_is_fp(t)) {
2915 2916
        if (llvm.LLVMGetIntTypeWidth(lldsttype) >
            llvm.LLVMGetIntTypeWidth(llsrctype)) {
2917
            if (ty.type_is_signed(t)) {
2918 2919 2920 2921 2922 2923 2924 2925 2926 2927 2928 2929 2930 2931 2932 2933 2934 2935 2936 2937 2938 2939
                // Widening signed cast.
                e_res.val =
                    e_res.bcx.build.SExtOrBitCast(e_res.val,
                                                  lldsttype);
            } else {
                // Widening unsigned cast.
                e_res.val =
                    e_res.bcx.build.ZExtOrBitCast(e_res.val,
                                                  lldsttype);
            }
        } else {
            // Narrowing cast.
            e_res.val =
                e_res.bcx.build.TruncOrBitCast(e_res.val,
                                               lldsttype);
        }
    } else {
        cx.fcx.ccx.sess.unimpl("fp cast");
    }
    ret e_res;
}

2940 2941 2942 2943 2944
fn trans_bind_thunk(@crate_ctxt cx,
                    @ty.t incoming_fty,
                    @ty.t outgoing_fty,
                    vec[option.t[@ast.expr]] args,
                    TypeRef llclosure_ty,
2945 2946
                    vec[@ty.t] bound_tys,
                    uint ty_param_count) -> ValueRef {
2947 2948 2949
    // Construct a thunk-call with signature incoming_fty, and that copies
    // args forward into a call to outgoing_fty.

2950
    let str s = cx.names.next("_rust_thunk") + sep() + cx.path;
2951 2952 2953 2954 2955 2956
    let TypeRef llthunk_ty = get_pair_fn_ty(type_of(cx, incoming_fty));
    let ValueRef llthunk = decl_fastcall_fn(cx.llmod, s, llthunk_ty);

    auto fcx = new_fn_ctxt(cx, s, llthunk);
    auto bcx = new_top_block_ctxt(fcx);

2957
    auto llclosure = bcx.build.PointerCast(fcx.llenv, llclosure_ty);
2958 2959 2960 2961 2962 2963 2964 2965 2966 2967 2968 2969 2970 2971 2972 2973 2974

    auto llbody = bcx.build.GEP(llclosure,
                                vec(C_int(0),
                                    C_int(abi.box_rc_field_body)));

    auto lltarget = bcx.build.GEP(llbody,
                                  vec(C_int(0),
                                      C_int(abi.closure_elt_target)));

    auto llbound = bcx.build.GEP(llbody,
                                 vec(C_int(0),
                                     C_int(abi.closure_elt_bindings)));

    auto lltargetclosure = bcx.build.GEP(lltarget,
                                         vec(C_int(0),
                                             C_int(abi.fn_field_box)));
    lltargetclosure = bcx.build.Load(lltargetclosure);
2975 2976 2977 2978 2979 2980 2981 2982 2983 2984

    auto outgoing_ret_ty = ty.ty_fn_ret(outgoing_fty);
    auto outgoing_arg_tys = ty.ty_fn_args(outgoing_fty);

    auto llretptr = fcx.llretptr;
    if (ty.type_has_dynamic_size(outgoing_ret_ty)) {
        llretptr = bcx.build.PointerCast(llretptr, T_typaram_ptr(cx.tn));
    }

    let vec[ValueRef] llargs = vec(llretptr,
2985
                                   fcx.lltaskptr,
2986
                                   lltargetclosure);
2987 2988 2989 2990 2991 2992 2993 2994 2995 2996 2997 2998 2999

    // Copy in the type parameters.
    let uint i = 0u;
    while (i < ty_param_count) {
        auto lltyparam_ptr =
            bcx.build.GEP(llbody, vec(C_int(0),
                                      C_int(abi.closure_elt_ty_params),
                                      C_int(i as int)));
        llargs += vec(bcx.build.Load(lltyparam_ptr));
        i += 1u;
    }

    let uint a = 2u + i;    // retptr, task ptr, env come first
3000
    let int b = 0;
3001
    let uint outgoing_arg_index = 0u;
3002 3003 3004 3005 3006 3007 3008 3009 3010 3011 3012 3013 3014 3015 3016 3017
    for (option.t[@ast.expr] arg in args) {
        alt (arg) {

            // Arg provided at binding time; thunk copies it from closure.
            case (some[@ast.expr](_)) {
                let ValueRef bound_arg = bcx.build.GEP(llbound,
                                                       vec(C_int(0),
                                                           C_int(b)));
                // FIXME: possibly support passing aliases someday.
                llargs += bcx.build.Load(bound_arg);
                b += 1;
            }

            // Arg will be provided when the thunk is invoked.
            case (none[@ast.expr]) {
                let ValueRef passed_arg = llvm.LLVMGetParam(llthunk, a);
3018 3019 3020 3021 3022 3023 3024
                if (ty.type_has_dynamic_size(outgoing_arg_tys.
                        (outgoing_arg_index).ty)) {
                    // Cast to a generic typaram pointer in order to make a
                    // type-compatible call.
                    passed_arg = bcx.build.PointerCast(passed_arg,
                                                       T_typaram_ptr(cx.tn));
                }
3025 3026 3027 3028
                llargs += passed_arg;
                a += 1u;
            }
        }
3029 3030

        outgoing_arg_index += 0u;
3031 3032 3033 3034 3035 3036 3037
    }

    // FIXME: turn this call + ret into a tail call.
    auto lltargetfn = bcx.build.GEP(lltarget,
                                    vec(C_int(0),
                                        C_int(abi.fn_field_code)));
    lltargetfn = bcx.build.Load(lltargetfn);
3038

3039
    auto r = bcx.build.FastCall(lltargetfn, llargs);
3040
    bcx.build.RetVoid();
3041 3042 3043 3044

    ret llthunk;
}

3045 3046 3047
fn trans_bind(@block_ctxt cx, @ast.expr f,
              vec[option.t[@ast.expr]] args,
              &ast.ann ann) -> result {
3048 3049 3050 3051
    auto f_res = trans_lval(cx, f);
    if (f_res.is_mem) {
        cx.fcx.ccx.sess.unimpl("re-binding existing function");
    } else {
3052 3053
        let vec[@ast.expr] bound = vec();

3054 3055 3056 3057 3058
        for (option.t[@ast.expr] argopt in args) {
            alt (argopt) {
                case (none[@ast.expr]) {
                }
                case (some[@ast.expr](?e)) {
3059
                    append[@ast.expr](bound, e);
3060 3061 3062
                }
            }
        }
3063 3064 3065 3066 3067 3068 3069 3070 3071 3072 3073 3074 3075 3076 3077 3078 3079

        // Figure out which tydescs we need to pass, if any.
        // FIXME: typestate botch
        let @ty.t outgoing_fty = ty.plain_ty(ty.ty_nil);
        let vec[ValueRef] lltydescs = vec();
        alt (f_res.generic) {
            case (none[generic_info]) {
                outgoing_fty = ty.expr_ty(f);
            }
            case (some[generic_info](?ginfo)) {
                outgoing_fty = ginfo.item_type;
                lltydescs = ginfo.tydescs;
            }
        }
        auto ty_param_count = _vec.len[ValueRef](lltydescs);

        if (_vec.len[@ast.expr](bound) == 0u && ty_param_count == 0u) {
3080 3081 3082 3083 3084 3085
            // Trivial 'binding': just return the static pair-ptr.
            ret f_res.res;
        } else {
            auto bcx = f_res.res.bcx;
            auto pair_t = node_type(cx.fcx.ccx, ann);
            auto pair_v = bcx.build.Alloca(pair_t);
3086 3087 3088 3089

            // Translate the bound expressions.
            let vec[@ty.t] bound_tys = vec();
            let vec[ValueRef] bound_vals = vec();
3090
            auto i = 0u;
3091 3092 3093
            for (@ast.expr e in bound) {
                auto arg = trans_expr(bcx, e);
                bcx = arg.bcx;
3094

3095 3096
                append[ValueRef](bound_vals, arg.val);
                append[@ty.t](bound_tys, ty.expr_ty(e));
3097 3098

                i += 1u;
3099 3100
            }

3101 3102 3103
            // Get the type of the bound function.
            let TypeRef lltarget_ty = type_of(bcx.fcx.ccx, outgoing_fty);

3104
            // Synthesize a closure type.
3105
            let @ty.t bindings_ty = plain_ty(ty.ty_tup(bound_tys));
3106
            let TypeRef llbindings_ty = type_of(bcx.fcx.ccx, bindings_ty);
3107 3108
            let TypeRef llclosure_ty = T_closure_ptr(cx.fcx.ccx.tn,
                                                     lltarget_ty,
3109 3110
                                                     llbindings_ty,
                                                     ty_param_count);
3111 3112 3113 3114 3115 3116 3117 3118 3119 3120 3121 3122 3123 3124 3125 3126 3127 3128 3129 3130

            // Malloc a box for the body.
            auto r = trans_malloc_inner(bcx, llclosure_ty);
            auto box = r.val;
            bcx = r.bcx;
            auto rc = bcx.build.GEP(box,
                                    vec(C_int(0),
                                        C_int(abi.box_rc_field_refcnt)));
            auto closure =
                bcx.build.GEP(box,
                              vec(C_int(0),
                                  C_int(abi.box_rc_field_body)));
            bcx.build.Store(C_int(1), rc);

            // Store bindings tydesc.
            auto bound_tydesc =
                bcx.build.GEP(closure,
                              vec(C_int(0),
                                  C_int(abi.closure_elt_tydesc)));
            auto bindings_tydesc = get_tydesc(bcx, bindings_ty);
3131 3132
            bcx = bindings_tydesc.bcx;
            bcx.build.Store(bindings_tydesc.val, bound_tydesc);
3133

3134 3135
            // Store thunk-target.
            auto bound_target =
3136 3137
                bcx.build.GEP(closure,
                              vec(C_int(0),
3138
                                  C_int(abi.closure_elt_target)));
3139 3140
            auto src = bcx.build.Load(f_res.res.val);
            bcx.build.Store(src, bound_target);
3141

3142
            // Copy expr values into boxed bindings.
3143
            i = 0u;
3144 3145 3146 3147
            auto bindings =
                bcx.build.GEP(closure,
                              vec(C_int(0),
                                  C_int(abi.closure_elt_bindings)));
3148 3149
            for (ValueRef v in bound_vals) {
                auto bound = bcx.build.GEP(bindings,
3150
                                           vec(C_int(0), C_int(i as int)));
3151
                bcx = copy_ty(r.bcx, INIT, bound, v, bound_tys.(i)).bcx;
3152 3153 3154 3155 3156 3157 3158 3159 3160 3161 3162 3163 3164 3165 3166 3167 3168 3169 3170 3171 3172
                i += 1u;
            }

            // If necessary, copy tydescs describing type parameters into the
            // appropriate slot in the closure.
            alt (f_res.generic) {
                case (none[generic_info]) { /* nothing to do */ }
                case (some[generic_info](?ginfo)) {
                    auto ty_params_slot =
                        bcx.build.GEP(closure,
                                      vec(C_int(0),
                                          C_int(abi.closure_elt_ty_params)));
                    auto i = 0;
                    for (ValueRef td in ginfo.tydescs) {
                        auto ty_param_slot = bcx.build.GEP(ty_params_slot,
                                                           vec(C_int(0),
                                                               C_int(i)));
                        bcx.build.Store(td, ty_param_slot);
                        i += 1;
                    }
                }
3173 3174
            }

3175 3176 3177 3178
            // Make thunk and store thunk-ptr in outer pair's code slot.
            auto pair_code = bcx.build.GEP(pair_v,
                                           vec(C_int(0),
                                               C_int(abi.fn_field_code)));
3179 3180

            let @ty.t pair_ty = node_ann_type(cx.fcx.ccx, ann);
3181
            let ValueRef llthunk =
3182 3183 3184
                trans_bind_thunk(cx.fcx.ccx, pair_ty, outgoing_fty,
                                 args, llclosure_ty, bound_tys,
                                 ty_param_count);
3185 3186 3187 3188 3189 3190 3191

            bcx.build.Store(llthunk, pair_code);

            // Store box ptr in outer pair's box slot.
            auto pair_box = bcx.build.GEP(pair_v,
                                          vec(C_int(0),
                                              C_int(abi.fn_field_box)));
3192 3193 3194 3195 3196
            bcx.build.Store
                (bcx.build.PointerCast
                 (box,
                  T_opaque_closure_ptr(bcx.fcx.ccx.tn)),
                 pair_box);
3197

3198 3199 3200
            find_scope_cx(cx).cleanups +=
                clean(bind drop_slot(_, pair_v, pair_ty));

3201 3202
            ret res(bcx, pair_v);
        }
3203 3204 3205
    }
}

3206 3207 3208 3209 3210 3211 3212 3213 3214 3215 3216
// NB: must keep 4 fns in sync:
//
//  - type_of_fn_full
//  - create_llargs_for_fn_args.
//  - new_fn_ctxt
//  - trans_args

fn trans_args(@block_ctxt cx,
              ValueRef llenv,
              option.t[ValueRef] llobj,
              option.t[generic_info] gen,
3217
              option.t[ValueRef] lliterbody,
3218 3219 3220 3221 3222 3223 3224 3225 3226 3227 3228 3229
              &vec[@ast.expr] es,
              @ty.t fn_ty)
    -> tup(@block_ctxt, vec[ValueRef], ValueRef) {

    let vec[ty.arg] args = ty.ty_fn_args(fn_ty);
    let vec[ValueRef] llargs = vec();
    let vec[ValueRef] lltydescs = vec();
    let @block_ctxt bcx = cx;


    // Arg 0: Output pointer.
    auto retty = ty.ty_fn_ret(fn_ty);
3230 3231 3232 3233
    auto llretslot_res = alloc_ty(bcx, retty);
    bcx = llretslot_res.bcx;
    auto llretslot = llretslot_res.val;

3234 3235 3236 3237 3238 3239 3240 3241 3242 3243
    alt (gen) {
        case (some[generic_info](?g)) {
            lltydescs = g.tydescs;
            args = ty.ty_fn_args(g.item_type);
            retty = ty.ty_fn_ret(g.item_type);
        }
        case (_) {
        }
    }
    if (ty.type_has_dynamic_size(retty)) {
3244 3245
        llargs += bcx.build.PointerCast(llretslot,
                                        T_typaram_ptr(cx.fcx.ccx.tn));
3246 3247 3248 3249 3250 3251 3252
    } else if (ty.count_ty_params(retty) != 0u) {
        // It's possible that the callee has some generic-ness somewhere in
        // its return value -- say a method signature within an obj or a fn
        // type deep in a structure -- which the caller has a concrete view
        // of. If so, cast the caller's view of the restlot to the callee's
        // view, for the sake of making a type-compatible call.
        llargs += cx.build.PointerCast(llretslot,
3253
                                       T_ptr(type_of(bcx.fcx.ccx, retty)));
3254 3255 3256 3257 3258 3259
    } else {
        llargs += llretslot;
    }


    // Arg 1: Task pointer.
3260
    llargs += bcx.fcx.lltaskptr;
3261 3262 3263 3264 3265 3266 3267

    // Arg 2: Env (closure-bindings / self-obj)
    alt (llobj) {
        case (some[ValueRef](?ob)) {
            // Every object is always found in memory,
            // and not-yet-loaded (as part of an lval x.y
            // doted method-call).
3268
            llargs += bcx.build.Load(ob);
3269 3270 3271 3272 3273 3274 3275 3276 3277
        }
        case (_) {
            llargs += llenv;
        }
    }

    // Args >3: ty_params ...
    llargs += lltydescs;

3278 3279 3280 3281 3282 3283 3284 3285
    // ... then possibly an lliterbody argument.
    alt (lliterbody) {
        case (none[ValueRef]) {}
        case (some[ValueRef](?lli)) {
            llargs += lli;
        }
    }

3286
    // ... then explicit args.
3287 3288 3289 3290 3291 3292

    // First we figure out the caller's view of the types of the arguments.
    // This will be needed if this is a generic call, because the callee has
    // to cast her view of the arguments to the caller's view.
    auto arg_tys = type_of_explicit_args(cx.fcx.ccx, args);

3293 3294 3295 3296 3297 3298 3299 3300 3301 3302 3303 3304 3305 3306 3307 3308 3309 3310 3311 3312 3313 3314 3315 3316 3317 3318 3319 3320 3321 3322 3323 3324 3325 3326 3327 3328 3329 3330 3331 3332 3333
    auto i = 0u;
    for (@ast.expr e in es) {
        auto mode = args.(i).mode;

        auto val;
        if (ty.type_is_structural(ty.expr_ty(e))) {
            auto re = trans_expr(bcx, e);
            val = re.val;
            bcx = re.bcx;
            if (mode == ast.val) {
                // Until here we've been treating structures by pointer;
                // we are now passing it as an arg, so need to load it.
                val = bcx.build.Load(val);
            }
        } else if (mode == ast.alias) {
            let lval_result lv;
            if (ty.is_lval(e)) {
                lv = trans_lval(bcx, e);
            } else {
                auto r = trans_expr(bcx, e);
                lv = lval_val(r.bcx, r.val);
            }
            bcx = lv.res.bcx;

            if (lv.is_mem) {
                val = lv.res.val;
            } else {
                // Non-mem but we're trying to alias; synthesize an
                // alloca, spill to it and pass its address.
                auto llty = val_ty(lv.res.val);
                auto llptr = lv.res.bcx.build.Alloca(llty);
                lv.res.bcx.build.Store(lv.res.val, llptr);
                val = llptr;
            }

        } else {
            auto re = trans_expr(bcx, e);
            val = re.val;
            bcx = re.bcx;
        }

3334 3335 3336
        if (ty.count_ty_params(args.(i).ty) > 0u) {
            auto lldestty = arg_tys.(i);
            val = bcx.build.PointerCast(val, lldestty);
3337 3338 3339 3340 3341 3342 3343 3344 3345
        }

        llargs += val;
        i += 1u;
    }

    ret tup(bcx, llargs, llretslot);
}

3346
fn trans_call(@block_ctxt cx, @ast.expr f,
3347 3348 3349
              option.t[ValueRef] lliterbody,
              vec[@ast.expr] args,
              &ast.ann ann) -> result {
3350
    auto f_res = trans_lval(cx, f);
3351
    auto faddr = f_res.res.val;
3352
    auto llenv = C_null(T_opaque_closure_ptr(cx.fcx.ccx.tn));
3353 3354 3355 3356 3357 3358 3359 3360 3361

    alt (f_res.llobj) {
        case (some[ValueRef](_)) {
            // It's a vtbl entry.
            faddr = f_res.res.bcx.build.Load(faddr);
        }
        case (none[ValueRef]) {
            // It's a closure.
            auto bcx = f_res.res.bcx;
3362 3363
            auto pair = faddr;
            faddr = bcx.build.GEP(pair, vec(C_int(0),
G
Graydon Hoare 已提交
3364
                                            C_int(abi.fn_field_code)));
3365
            faddr = bcx.build.Load(faddr);
3366

3367 3368 3369 3370
            auto llclosure = bcx.build.GEP(pair,
                                           vec(C_int(0),
                                               C_int(abi.fn_field_box)));
            llenv = bcx.build.Load(llclosure);
3371
        }
3372
    }
3373 3374
    auto fn_ty = ty.expr_ty(f);
    auto ret_ty = ty.ann_to_type(ann);
G
Graydon Hoare 已提交
3375
    auto args_res = trans_args(f_res.res.bcx,
3376
                               llenv, f_res.llobj,
3377
                               f_res.generic,
3378
                               lliterbody,
3379
                               args, fn_ty);
G
Graydon Hoare 已提交
3380

3381
    auto bcx = args_res._0;
3382 3383 3384
    auto llargs = args_res._1;
    auto llretslot = args_res._2;

3385 3386 3387 3388 3389 3390 3391 3392
    /*
    log "calling: " + val_str(cx.fcx.ccx.tn, faddr);

    for (ValueRef arg in llargs) {
        log "arg: " + val_str(cx.fcx.ccx.tn, arg);
    }
    */

3393 3394 3395 3396 3397 3398 3399 3400 3401
    bcx.build.FastCall(faddr, llargs);
    auto retval = C_nil();

    if (!ty.type_is_nil(ret_ty)) {
        retval = load_scalar_or_boxed(bcx, llretslot, ret_ty);
        // Retval doesn't correspond to anything really tangible in the frame,
        // but it's a ref all the same, so we put a note here to drop it when
        // we're done in this scope.
        find_scope_cx(cx).cleanups += clean(bind drop_ty(_, retval, ret_ty));
3402
    }
G
Graydon Hoare 已提交
3403

3404
    ret res(bcx, retval);
3405 3406
}

3407 3408
fn trans_tup(@block_ctxt cx, vec[ast.elt] elts,
             &ast.ann ann) -> result {
3409 3410 3411 3412 3413 3414 3415
    auto bcx = cx;
    auto t = node_ann_type(bcx.fcx.ccx, ann);
    auto llty = type_of(bcx.fcx.ccx, t);
    auto tup_res = alloc_ty(bcx, t);
    auto tup_val = tup_res.val;
    bcx = tup_res.bcx;

3416
    find_scope_cx(cx).cleanups += clean(bind drop_ty(_, tup_val, t));
G
Graydon Hoare 已提交
3417
    let int i = 0;
3418

3419
    for (ast.elt e in elts) {
3420 3421 3422 3423 3424 3425
        auto e_ty = ty.expr_ty(e.expr);
        auto src_res = trans_expr(bcx, e.expr);
        bcx = src_res.bcx;
        auto dst_res = GEP_tup_like(bcx, t, tup_val, vec(0, i));
        bcx = dst_res.bcx;
        bcx = copy_ty(src_res.bcx, INIT, dst_res.val, src_res.val, e_ty).bcx;
G
Graydon Hoare 已提交
3426 3427
        i += 1;
    }
3428
    ret res(bcx, tup_val);
G
Graydon Hoare 已提交
3429 3430
}

3431 3432
fn trans_vec(@block_ctxt cx, vec[@ast.expr] args,
             &ast.ann ann) -> result {
3433 3434 3435 3436
    auto t = node_ann_type(cx.fcx.ccx, ann);
    auto unit_ty = t;
    alt (t.struct) {
        case (ty.ty_vec(?t)) {
G
Graydon Hoare 已提交
3437 3438 3439 3440 3441 3442 3443 3444
            unit_ty = t;
        }
        case (_) {
            cx.fcx.ccx.sess.bug("non-vec type in trans_vec");
        }
    }

    auto llunit_ty = type_of(cx.fcx.ccx, unit_ty);
3445 3446 3447
    auto bcx = cx;
    auto unit_sz = size_of(bcx, unit_ty);
    bcx = unit_sz.bcx;
G
Graydon Hoare 已提交
3448
    auto data_sz = llvm.LLVMConstMul(C_int(_vec.len[@ast.expr](args) as int),
3449
                                     unit_sz.val);
G
Graydon Hoare 已提交
3450 3451

    // FIXME: pass tydesc properly.
3452
    auto sub = trans_upcall(bcx, "upcall_new_vec", vec(data_sz, C_int(0)));
3453
    bcx = sub.bcx;
G
Graydon Hoare 已提交
3454

3455
    auto llty = type_of(bcx.fcx.ccx, t);
3456
    auto vec_val = bcx.build.IntToPtr(sub.val, llty);
3457
    find_scope_cx(bcx).cleanups += clean(bind drop_ty(_, vec_val, t));
G
Graydon Hoare 已提交
3458

3459 3460 3461 3462 3463 3464
    auto body = bcx.build.GEP(vec_val, vec(C_int(0),
                                           C_int(abi.vec_elt_data)));

    auto pseudo_tup_ty =
        plain_ty(ty.ty_tup(_vec.init_elt[@ty.t](unit_ty,
                                                _vec.len[@ast.expr](args))));
G
Graydon Hoare 已提交
3465
    let int i = 0;
3466

G
Graydon Hoare 已提交
3467
    for (@ast.expr e in args) {
3468 3469 3470 3471 3472
        auto src_res = trans_expr(bcx, e);
        bcx = src_res.bcx;
        auto dst_res = GEP_tup_like(bcx, pseudo_tup_ty, body, vec(0, i));
        bcx = dst_res.bcx;
        bcx = copy_ty(bcx, INIT, dst_res.val, src_res.val, unit_ty).bcx;
G
Graydon Hoare 已提交
3473 3474
        i += 1;
    }
3475 3476 3477
    auto fill = bcx.build.GEP(vec_val,
                              vec(C_int(0), C_int(abi.vec_elt_fill)));
    bcx.build.Store(data_sz, fill);
3478

3479
    ret res(bcx, vec_val);
G
Graydon Hoare 已提交
3480 3481
}

3482
fn trans_rec(@block_ctxt cx, vec[ast.field] fields,
3483 3484
             option.t[@ast.expr] base, &ast.ann ann) -> result {

3485 3486 3487 3488 3489 3490 3491
    auto bcx = cx;
    auto t = node_ann_type(bcx.fcx.ccx, ann);
    auto llty = type_of(bcx.fcx.ccx, t);
    auto rec_res = alloc_ty(bcx, t);
    auto rec_val = rec_res.val;
    bcx = rec_res.bcx;

3492
    find_scope_cx(cx).cleanups += clean(bind drop_ty(_, rec_val, t));
3493
    let int i = 0;
3494

G
Graydon Hoare 已提交
3495 3496 3497 3498 3499 3500 3501 3502 3503 3504 3505 3506 3507 3508 3509 3510 3511 3512
    auto base_val = C_nil();

    alt (base) {
        case (none[@ast.expr]) { }
        case (some[@ast.expr](?bexp)) {
            auto base_res = trans_expr(bcx, bexp);
            bcx = base_res.bcx;
            base_val = base_res.val;
        }
    }

    let vec[ty.field] ty_fields = vec();
    alt (t.struct) {
        case (ty.ty_rec(?flds)) { ty_fields = flds; }
    }

    for (ty.field tf in ty_fields) {
        auto e_ty = tf.ty;
3513 3514
        auto dst_res = GEP_tup_like(bcx, t, rec_val, vec(0, i));
        bcx = dst_res.bcx;
G
Graydon Hoare 已提交
3515 3516 3517 3518 3519 3520 3521 3522 3523 3524 3525 3526 3527 3528 3529 3530 3531 3532

        auto expr_provided = false;
        auto src_res = res(bcx, C_nil());

        for (ast.field f in fields) {
            if (_str.eq(f.ident, tf.ident)) {
                expr_provided = true;
                src_res = trans_expr(bcx, f.expr);
            }
        }
        if (!expr_provided) {
            src_res = GEP_tup_like(bcx, t, base_val, vec(0, i));
            src_res = res(src_res.bcx,
                          load_scalar_or_boxed(bcx, src_res.val, e_ty));
        }

        bcx = src_res.bcx;
        bcx = copy_ty(bcx, INIT, dst_res.val, src_res.val, e_ty).bcx;
3533 3534
        i += 1;
    }
3535
    ret res(bcx, rec_val);
3536 3537
}

G
Graydon Hoare 已提交
3538

G
Graydon Hoare 已提交
3539

3540
fn trans_expr(@block_ctxt cx, @ast.expr e) -> result {
3541
    alt (e.node) {
3542
        case (ast.expr_lit(?lit, ?ann)) {
3543
            ret res(cx, trans_lit(cx.fcx.ccx, *lit, ann));
3544 3545
        }

3546
        case (ast.expr_unary(?op, ?x, ?ann)) {
3547
            ret trans_unary(cx, op, x, ann);
3548 3549
        }

P
Patrick Walton 已提交
3550
        case (ast.expr_binary(?op, ?x, ?y, _)) {
3551
            ret trans_binary(cx, op, x, y);
3552
        }
3553

G
Graydon Hoare 已提交
3554 3555
        case (ast.expr_if(?cond, ?thn, ?elifs, ?els, _)) {
            ret trans_if(cx, cond, thn, elifs, els);
3556 3557
        }

G
Graydon Hoare 已提交
3558 3559 3560 3561
        case (ast.expr_for(?decl, ?seq, ?body, _)) {
            ret trans_for(cx, decl, seq, body);
        }

3562 3563 3564 3565
        case (ast.expr_for_each(?decl, ?seq, ?body, _)) {
            ret trans_for_each(cx, decl, seq, body);
        }

3566
        case (ast.expr_while(?cond, ?body, _)) {
3567
            ret trans_while(cx, cond, body);
3568 3569
        }

3570
        case (ast.expr_do_while(?body, ?cond, _)) {
3571
            ret trans_do_while(cx, body, cond);
3572 3573
        }

P
Patrick Walton 已提交
3574 3575 3576 3577
        case (ast.expr_alt(?expr, ?arms, _)) {
            ret trans_alt(cx, expr, arms);
        }

P
Patrick Walton 已提交
3578
        case (ast.expr_block(?blk, _)) {
3579
            auto sub_cx = new_scope_block_ctxt(cx, "block-expr body");
3580
            auto next_cx = new_sub_block_ctxt(cx, "next");
3581 3582 3583 3584 3585 3586 3587
            auto sub = trans_block(sub_cx, blk);

            cx.build.Br(sub_cx.llbb);
            sub.bcx.build.Br(next_cx.llbb);

            ret res(next_cx, sub.val);
        }
3588

3589
        case (ast.expr_assign(?dst, ?src, ?ann)) {
3590
            auto lhs_res = trans_lval(cx, dst);
3591 3592
            check (lhs_res.is_mem);
            auto rhs_res = trans_expr(lhs_res.res.bcx, src);
3593
            auto t = node_ann_type(cx.fcx.ccx, ann);
G
Graydon Hoare 已提交
3594
            // FIXME: calculate copy init-ness in typestate.
3595 3596
            ret copy_ty(rhs_res.bcx, DROP_EXISTING,
                        lhs_res.res.val, rhs_res.val, t);
3597
        }
G
Graydon Hoare 已提交
3598

3599 3600 3601
        case (ast.expr_assign_op(?op, ?dst, ?src, ?ann)) {
            auto t = node_ann_type(cx.fcx.ccx, ann);
            auto lhs_res = trans_lval(cx, dst);
3602
            check (lhs_res.is_mem);
3603
            auto lhs_val = load_scalar_or_boxed(lhs_res.res.bcx,
G
Graydon Hoare 已提交
3604
                                                lhs_res.res.val, t);
3605
            auto rhs_res = trans_expr(lhs_res.res.bcx, src);
3606 3607
            auto v = trans_eager_binop(rhs_res.bcx, op, t,
                                       lhs_val, rhs_res.val);
3608
            // FIXME: calculate copy init-ness in typestate.
3609 3610
            ret copy_ty(v.bcx, DROP_EXISTING,
                        lhs_res.res.val, v.val, t);
3611 3612
        }

3613 3614 3615 3616
        case (ast.expr_bind(?f, ?args, ?ann)) {
            ret trans_bind(cx, f, args, ann);
        }

G
Graydon Hoare 已提交
3617
        case (ast.expr_call(?f, ?args, ?ann)) {
3618
            ret trans_call(cx, f, none[ValueRef], args, ann);
3619 3620
        }

3621
        case (ast.expr_cast(?e, _, ?ann)) {
3622
            ret trans_cast(cx, e, ann);
3623
        }
G
Graydon Hoare 已提交
3624

G
Graydon Hoare 已提交
3625 3626 3627 3628
        case (ast.expr_vec(?args, ?ann)) {
            ret trans_vec(cx, args, ann);
        }

G
Graydon Hoare 已提交
3629 3630 3631
        case (ast.expr_tup(?args, ?ann)) {
            ret trans_tup(cx, args, ann);
        }
G
Graydon Hoare 已提交
3632

3633 3634
        case (ast.expr_rec(?args, ?base, ?ann)) {
            ret trans_rec(cx, args, base, ann);
3635 3636
        }

3637 3638 3639 3640 3641 3642 3643 3644 3645 3646 3647 3648 3649 3650 3651 3652
        case (ast.expr_fail) {
            ret trans_fail(cx, e.span, "explicit failure");
        }

        case (ast.expr_log(?a)) {
            ret trans_log(cx, a);
        }

        case (ast.expr_check_expr(?a)) {
            ret trans_check_expr(cx, a);
        }

        case (ast.expr_ret(?e)) {
            ret trans_ret(cx, e);
        }

3653
        case (ast.expr_put(?e)) {
3654 3655 3656
            ret trans_put(cx, e);
        }

3657 3658 3659 3660
        case (ast.expr_be(?e)) {
            ret trans_be(cx, e);
        }

3661 3662
        // lval cases fall through to trans_lval and then
        // possibly load the result (if it's non-structural).
G
Graydon Hoare 已提交
3663

3664
        case (_) {
3665
            auto t = ty.expr_ty(e);
3666
            auto sub = trans_lval(cx, e);
3667
            ret res(sub.res.bcx,
3668
                    load_scalar_or_boxed(sub.res.bcx, sub.res.val, t));
3669
        }
3670
    }
3671
    cx.fcx.ccx.sess.unimpl("expr variant in trans_expr");
3672 3673 3674
    fail;
}

3675
// We pass structural values around the compiler "by pointer" and
3676 3677
// non-structural values (scalars and boxes) "by value". This function selects
// whether to load a pointer or pass it.
3678

3679 3680 3681 3682
fn load_scalar_or_boxed(@block_ctxt cx,
                        ValueRef v,
                        @ty.t t) -> ValueRef {
    if (ty.type_is_scalar(t) || ty.type_is_boxed(t)) {
3683
        ret cx.build.Load(v);
3684 3685
    } else {
        ret v;
3686 3687 3688
    }
}

3689
fn trans_log(@block_ctxt cx, @ast.expr e) -> result {
3690

3691
    auto sub = trans_expr(cx, e);
3692
    auto e_ty = ty.expr_ty(e);
3693
    alt (e_ty.struct) {
3694
        case (ty.ty_str) {
3695 3696 3697 3698
            auto v = sub.bcx.build.PtrToInt(sub.val, T_int());
            ret trans_upcall(sub.bcx,
                             "upcall_log_str",
                             vec(v));
3699 3700
        }
        case (_) {
3701 3702 3703
            ret trans_upcall(sub.bcx,
                             "upcall_log_int",
                             vec(sub.val));
3704 3705
        }
    }
3706
    fail;
3707 3708
}

3709
fn trans_check_expr(@block_ctxt cx, @ast.expr e) -> result {
3710 3711 3712
    auto cond_res = trans_expr(cx, e);

    // FIXME: need pretty-printer.
B
Brian Anderson 已提交
3713
    auto expr_str = "<expr>";
3714
    auto fail_cx = new_sub_block_ctxt(cx, "fail");
B
Brian Anderson 已提交
3715
    auto fail_res = trans_fail(fail_cx, e.span, expr_str);
3716

3717
    auto next_cx = new_sub_block_ctxt(cx, "next");
3718 3719 3720 3721 3722 3723 3724
    fail_res.bcx.build.Br(next_cx.llbb);
    cond_res.bcx.build.CondBr(cond_res.val,
                              next_cx.llbb,
                              fail_cx.llbb);
    ret res(next_cx, C_nil());
}

B
Brian Anderson 已提交
3725 3726 3727 3728 3729 3730 3731 3732 3733
fn trans_fail(@block_ctxt cx, common.span sp, str fail_str) -> result {
    auto V_fail_str = p2i(C_cstr(cx.fcx.ccx, fail_str));
    auto V_filename = p2i(C_cstr(cx.fcx.ccx, sp.filename));
    auto V_line = sp.lo.line as int;
    auto args = vec(V_fail_str, V_filename, C_int(V_line));

    ret trans_upcall(cx, "upcall_fail", args);
}

3734
fn trans_put(@block_ctxt cx, &option.t[@ast.expr] e) -> result {
3735 3736 3737 3738 3739 3740 3741 3742 3743 3744 3745 3746 3747 3748 3749 3750 3751 3752 3753 3754 3755 3756 3757 3758 3759 3760 3761 3762 3763
    auto llcallee = C_nil();
    auto llenv = C_nil();

    alt (cx.fcx.lliterbody) {
        case (some[ValueRef](?lli)) {
            auto slot = cx.build.Alloca(val_ty(lli));
            cx.build.Store(lli, slot);

            llcallee = cx.build.GEP(slot, vec(C_int(0),
                                              C_int(abi.fn_field_code)));
            llcallee = cx.build.Load(llcallee);

            llenv = cx.build.GEP(slot, vec(C_int(0),
                                           C_int(abi.fn_field_box)));
            llenv = cx.build.Load(llenv);
        }
    }
    auto bcx = cx;
    auto dummy_retslot = bcx.build.Alloca(T_nil());
    let vec[ValueRef] llargs = vec(dummy_retslot, cx.fcx.lltaskptr, llenv);
    alt (e) {
        case (none[@ast.expr]) { }
        case (some[@ast.expr](?x)) {
            auto r = trans_expr(bcx, x);
            llargs += r.val;
            bcx = r.bcx;
        }
    }
    ret res(bcx, bcx.build.FastCall(llcallee, llargs));
3764 3765
}

3766 3767 3768 3769
fn trans_ret(@block_ctxt cx, &option.t[@ast.expr] e) -> result {
    auto bcx = cx;
    auto val = C_nil();

3770 3771
    alt (e) {
        case (some[@ast.expr](?x)) {
3772
            auto t = ty.expr_ty(x);
3773 3774 3775
            auto r = trans_expr(cx, x);
            bcx = r.bcx;
            val = r.val;
3776
            bcx = copy_ty(bcx, INIT, cx.fcx.llretptr, val, t).bcx;
3777
        }
3778
        case (_) { /* fall through */  }
3779 3780
    }

3781 3782
    // Run all cleanups and back out.
    let bool more_cleanups = true;
3783
    auto cleanup_cx = cx;
3784
    while (more_cleanups) {
3785
        bcx = trans_block_cleanups(bcx, cleanup_cx);
3786
        alt (cleanup_cx.parent) {
3787
            case (parent_some(?b)) {
3788
                cleanup_cx = b;
3789 3790 3791 3792 3793 3794 3795
            }
            case (parent_none) {
                more_cleanups = false;
            }
        }
    }

3796 3797
    bcx.build.RetVoid();
    ret res(bcx, C_nil());
3798 3799
}

3800
fn trans_be(@block_ctxt cx, @ast.expr e) -> result {
3801
    // FIXME: This should be a typestate precondition
3802
    check (ast.is_call_expr(e));
3803 3804
    // FIXME: Turn this into a real tail call once
    // calling convention issues are settled
3805 3806 3807
    ret trans_ret(cx, some(e));
}

3808 3809 3810 3811 3812 3813 3814 3815 3816 3817 3818 3819 3820 3821
fn init_local(@block_ctxt cx, @ast.local local) -> result {

    // Make a note to drop this slot on the way out.
    check (cx.fcx.lllocals.contains_key(local.id));
    auto llptr = cx.fcx.lllocals.get(local.id);
    auto ty = node_ann_type(cx.fcx.ccx, local.ann);
    auto bcx = cx;

    find_scope_cx(cx).cleanups +=
        clean(bind drop_slot(_, llptr, ty));

    alt (local.init) {
        case (some[@ast.expr](?e)) {
            auto sub = trans_expr(bcx, e);
3822
            bcx = copy_ty(sub.bcx, INIT, llptr, sub.val, ty).bcx;
3823 3824 3825
        }
        case (_) {
            if (middle.ty.type_has_dynamic_size(ty)) {
3826 3827
                auto llsz = size_of(bcx, ty);
                bcx = call_bzero(llsz.bcx, llptr, llsz.val).bcx;
3828 3829 3830 3831 3832 3833 3834 3835 3836 3837 3838 3839 3840

            } else {
                auto llty = type_of(bcx.fcx.ccx, ty);
                auto null = lib.llvm.llvm.LLVMConstNull(llty);
                bcx.build.Store(null, llptr);
            }
        }
    }
    ret res(bcx, llptr);
}

fn trans_stmt(@block_ctxt cx, &ast.stmt s) -> result {
    auto bcx = cx;
3841
    alt (s.node) {
3842
        case (ast.stmt_expr(?e)) {
3843
            bcx = trans_expr(cx, e).bcx;
3844
        }
3845

3846 3847 3848
        case (ast.stmt_decl(?d)) {
            alt (d.node) {
                case (ast.decl_local(?local)) {
3849
                    bcx = init_local(bcx, local).bcx;
3850
                }
G
Graydon Hoare 已提交
3851 3852 3853
                case (ast.decl_item(?i)) {
                    trans_item(cx.fcx.ccx, *i);
                }
3854 3855
            }
        }
3856
        case (_) {
3857
            cx.fcx.ccx.sess.unimpl("stmt variant");
3858 3859
        }
    }
3860
    ret res(bcx, C_nil());
3861 3862
}

3863
fn new_builder(BasicBlockRef llbb) -> builder {
3864 3865 3866 3867 3868
    let BuilderRef llbuild = llvm.LLVMCreateBuilder();
    llvm.LLVMPositionBuilderAtEnd(llbuild, llbb);
    ret builder(llbuild);
}

3869 3870
// You probably don't want to use this one. See the
// next three functions instead.
3871
fn new_block_ctxt(@fn_ctxt cx, block_parent parent,
3872
                  block_kind kind,
3873
                  str name) -> @block_ctxt {
3874
    let vec[cleanup] cleanups = vec();
3875
    let BasicBlockRef llbb =
3876
        llvm.LLVMAppendBasicBlock(cx.llfn,
3877
                                  _str.buf(cx.ccx.names.next(name)));
3878

3879
    ret @rec(llbb=llbb,
3880
             build=new_builder(llbb),
3881
             parent=parent,
3882
             kind=kind,
3883
             mutable cleanups=cleanups,
3884 3885 3886
             fcx=cx);
}

3887 3888
// Use this when you're at the top block of a function or the like.
fn new_top_block_ctxt(@fn_ctxt fcx) -> @block_ctxt {
3889 3890 3891 3892 3893 3894 3895
    auto cx = new_block_ctxt(fcx, parent_none, SCOPE_BLOCK,
                             "function top level");

    // FIXME: hack to give us some spill room to make up for an LLVM
    // bug where it destroys its own callee-saves.
    cx.build.Alloca(T_array(T_int(), 10u));
    ret cx;
3896
}
3897

3898 3899
// Use this when you're at a curly-brace or similar lexical scope.
fn new_scope_block_ctxt(@block_ctxt bcx, str n) -> @block_ctxt {
3900
    ret new_block_ctxt(bcx.fcx, parent_some(bcx), SCOPE_BLOCK, n);
3901 3902
}

3903
// Use this when you're making a general CFG BB within a scope.
3904
fn new_sub_block_ctxt(@block_ctxt bcx, str n) -> @block_ctxt {
3905
    ret new_block_ctxt(bcx.fcx, parent_some(bcx), NON_SCOPE_BLOCK, n);
3906
}
3907

3908

3909 3910
fn trans_block_cleanups(@block_ctxt cx,
                        @block_ctxt cleanup_cx) -> @block_ctxt {
3911
    auto bcx = cx;
3912

3913
    if (cleanup_cx.kind != SCOPE_BLOCK) {
3914 3915 3916
        check (_vec.len[cleanup](cleanup_cx.cleanups) == 0u);
    }

3917 3918 3919 3920
    auto i = _vec.len[cleanup](cleanup_cx.cleanups);
    while (i > 0u) {
        i -= 1u;
        auto c = cleanup_cx.cleanups.(i);
3921
        alt (c) {
3922
            case (clean(?cfn)) {
3923
                bcx = cfn(bcx).bcx;
3924 3925 3926
            }
        }
    }
3927 3928 3929
    ret bcx;
}

3930 3931 3932 3933 3934 3935 3936 3937 3938 3939
iter block_locals(&ast.block b) -> @ast.local {
    // FIXME: putting from inside an iter block doesn't work, so we can't
    // use the index here.
    for (@ast.stmt s in b.node.stmts) {
        alt (s.node) {
            case (ast.stmt_decl(?d)) {
                alt (d.node) {
                    case (ast.decl_local(?local)) {
                        put local;
                    }
3940
                    case (_) { /* fall through */ }
3941 3942
                }
            }
3943
            case (_) { /* fall through */ }
3944 3945 3946 3947
        }
    }
}

3948
fn alloc_ty(@block_ctxt cx, @ty.t t) -> result {
3949
    auto val = C_int(0);
3950
    auto bcx = cx;
3951
    if (ty.type_has_dynamic_size(t)) {
3952 3953 3954
        auto n = size_of(bcx, t);
        bcx = n.bcx;
        val = bcx.build.ArrayAlloca(T_i8(), n.val);
3955
    } else {
3956
        val = bcx.build.Alloca(type_of(cx.fcx.ccx, t));
3957
    }
3958
    ret res(bcx, val);
3959 3960
}

3961 3962 3963 3964 3965 3966 3967
fn alloc_local(@block_ctxt cx, @ast.local local) -> result {
    auto t = node_ann_type(cx.fcx.ccx, local.ann);
    auto r = alloc_ty(cx, t);
    r.bcx.fcx.lllocals.insert(local.id, r.val);
    ret r;
}

3968
fn trans_block(@block_ctxt cx, &ast.block b) -> result {
3969 3970
    auto bcx = cx;

3971
    for each (@ast.local local in block_locals(b)) {
3972
        bcx = alloc_local(bcx, local).bcx;
3973
    }
3974
    auto r = res(bcx, C_nil());
3975

3976
    for (@ast.stmt s in b.node.stmts) {
3977 3978
        r = trans_stmt(bcx, *s);
        bcx = r.bcx;
3979 3980 3981 3982 3983
        // If we hit a terminator, control won't go any further so
        // we're in dead-code land. Stop here.
        if (is_terminated(bcx)) {
            ret r;
        }
3984
    }
3985

3986 3987 3988 3989 3990 3991 3992 3993 3994 3995 3996 3997 3998
    alt (b.node.expr) {
        case (some[@ast.expr](?e)) {
            r = trans_expr(bcx, e);
            bcx = r.bcx;
            if (is_terminated(bcx)) {
                ret r;
            }
        }
        case (none[@ast.expr]) {
            r = res(bcx, C_nil());
        }
    }

3999
    bcx = trans_block_cleanups(bcx, find_scope_cx(bcx));
4000
    ret res(bcx, r.val);
4001 4002
}

4003 4004 4005 4006 4007 4008 4009
// NB: must keep 4 fns in sync:
//
//  - type_of_fn_full
//  - create_llargs_for_fn_args.
//  - new_fn_ctxt
//  - trans_args

4010
fn new_fn_ctxt(@crate_ctxt cx,
4011
               str name,
4012
               ValueRef llfndecl) -> @fn_ctxt {
4013

4014 4015 4016
    let ValueRef llretptr = llvm.LLVMGetParam(llfndecl, 0u);
    let ValueRef lltaskptr = llvm.LLVMGetParam(llfndecl, 1u);
    let ValueRef llenv = llvm.LLVMGetParam(llfndecl, 2u);
4017 4018

    let hashmap[ast.def_id, ValueRef] llargs = new_def_hash[ValueRef]();
4019 4020
    let hashmap[ast.def_id, ValueRef] llobjfields = new_def_hash[ValueRef]();
    let hashmap[ast.def_id, ValueRef] lllocals = new_def_hash[ValueRef]();
4021
    let hashmap[ast.def_id, ValueRef] lltydescs = new_def_hash[ValueRef]();
4022

4023
    ret @rec(llfn=llfndecl,
4024
             lltaskptr=lltaskptr,
4025 4026
             llenv=llenv,
             llretptr=llretptr,
4027
             mutable llself=none[ValueRef],
4028
             mutable lliterbody=none[ValueRef],
4029
             llargs=llargs,
4030
             llobjfields=llobjfields,
4031
             lllocals=lllocals,
4032
             lltydescs=lltydescs,
4033
             ccx=cx);
4034 4035
}

4036 4037 4038 4039 4040 4041 4042
// NB: must keep 4 fns in sync:
//
//  - type_of_fn_full
//  - create_llargs_for_fn_args.
//  - new_fn_ctxt
//  - trans_args

4043
fn create_llargs_for_fn_args(&@fn_ctxt cx,
4044
                             ast.proto proto,
4045
                             option.t[TypeRef] ty_self,
4046
                             @ty.t ret_ty,
4047
                             &vec[ast.arg] args,
4048
                             &vec[ast.ty_param] ty_params) {
4049 4050 4051 4052 4053 4054 4055 4056 4057 4058

    alt (ty_self) {
        case (some[TypeRef](_)) {
            cx.llself = some[ValueRef](cx.llenv);
        }
        case (_) {
        }
    }

    auto arg_n = 3u;
4059

4060 4061 4062 4063 4064 4065 4066
    if (ty_self == none[TypeRef]) {
        for (ast.ty_param tp in ty_params) {
            auto llarg = llvm.LLVMGetParam(cx.llfn, arg_n);
            check (llarg as int != 0);
            cx.lltydescs.insert(tp.id, llarg);
            arg_n += 1u;
        }
4067 4068
    }

4069
    if (proto == ast.proto_iter) {
4070 4071 4072 4073 4074
        auto llarg = llvm.LLVMGetParam(cx.llfn, arg_n);
        check (llarg as int != 0);
        cx.lliterbody = some[ValueRef](llarg);
        arg_n += 1u;
    }
4075

4076 4077 4078 4079 4080 4081 4082 4083
    for (ast.arg arg in args) {
        auto llarg = llvm.LLVMGetParam(cx.llfn, arg_n);
        check (llarg as int != 0);
        cx.llargs.insert(arg.id, llarg);
        arg_n += 1u;
    }
}

4084 4085 4086 4087
// Recommended LLVM style, strange though this is, is to copy from args to
// allocas immediately upon entry; this permits us to GEP into structures we
// were passed and whatnot. Apparently mem2reg will mop up.

4088 4089 4090 4091
fn copy_args_to_allocas(@block_ctxt cx,
                        option.t[TypeRef] ty_self,
                        vec[ast.arg] args,
                        vec[ty.arg] arg_tys) {
4092 4093 4094

    let uint arg_n = 0u;

4095 4096 4097 4098 4099 4100 4101 4102 4103 4104 4105 4106 4107 4108
    alt (cx.fcx.llself) {
        case (some[ValueRef](?self_v)) {
            alt (ty_self) {
                case (some[TypeRef](?self_t)) {
                    auto alloca = cx.build.Alloca(self_t);
                    cx.build.Store(self_v, alloca);
                    cx.fcx.llself = some[ValueRef](alloca);
                }
            }
        }
        case (_) {
        }
    }

4109
    for (ast.arg aarg in args) {
4110 4111 4112 4113 4114 4115 4116 4117 4118
        if (aarg.mode != ast.alias) {
            auto arg_t = type_of_arg(cx.fcx.ccx, arg_tys.(arg_n));
            auto alloca = cx.build.Alloca(arg_t);
            auto argval = cx.fcx.llargs.get(aarg.id);
            cx.build.Store(argval, alloca);
            // Overwrite the llargs entry for this arg with its alloca.
            cx.fcx.llargs.insert(aarg.id, alloca);
        }

4119 4120 4121 4122
        arg_n += 1u;
    }
}

4123 4124 4125 4126 4127
fn is_terminated(@block_ctxt cx) -> bool {
    auto inst = llvm.LLVMGetLastInstruction(cx.llbb);
    ret llvm.LLVMIsATerminatorInst(inst) as int != 0;
}

4128 4129
fn arg_tys_of_fn(ast.ann ann) -> vec[ty.arg] {
    alt (ty.ann_to_type(ann).struct) {
4130
        case (ty.ty_fn(_, ?arg_tys, _)) {
4131 4132 4133 4134 4135 4136
            ret arg_tys;
        }
    }
    fail;
}

4137 4138
fn ret_ty_of_fn_ty(@ty.t t) -> @ty.t {
    alt (t.struct) {
4139
        case (ty.ty_fn(_, _, ?ret_ty)) {
4140 4141 4142 4143 4144 4145
            ret ret_ty;
        }
    }
    fail;
}

4146 4147 4148 4149 4150

fn ret_ty_of_fn(ast.ann ann) -> @ty.t {
    ret ret_ty_of_fn_ty(ty.ann_to_type(ann));
}

4151
fn populate_fn_ctxt_from_llself(@block_ctxt cx, ValueRef llself) {
4152 4153 4154 4155 4156 4157 4158

    let vec[TypeRef] llfield_tys = vec();

    for (ast.obj_field f in cx.fcx.ccx.obj_fields) {
        llfield_tys += node_type(cx.fcx.ccx, f.ann);
    }

4159
    auto n_typarams = _vec.len[ast.ty_param](cx.fcx.ccx.obj_typarams);
4160
    let TypeRef llobj_box_ty = T_obj_ptr(cx.fcx.ccx.tn, n_typarams,
4161
                                         T_struct(llfield_tys));
4162 4163 4164 4165 4166 4167 4168 4169 4170 4171

    auto box_cell =
        cx.build.GEP(llself,
                     vec(C_int(0),
                         C_int(abi.obj_field_box)));

    auto box_ptr = cx.build.Load(box_cell);

    box_ptr = cx.build.PointerCast(box_ptr, llobj_box_ty);

4172 4173 4174 4175 4176
    auto obj_typarams = cx.build.GEP(box_ptr,
                                     vec(C_int(0),
                                         C_int(abi.box_rc_field_body),
                                         C_int(abi.obj_body_elt_typarams)));

4177 4178 4179 4180 4181 4182
    auto obj_fields = cx.build.GEP(box_ptr,
                                   vec(C_int(0),
                                       C_int(abi.box_rc_field_body),
                                       C_int(abi.obj_body_elt_fields)));

    let int i = 0;
4183 4184 4185 4186 4187 4188 4189 4190 4191 4192 4193

    for (ast.ty_param p in cx.fcx.ccx.obj_typarams) {
        let ValueRef lltyparam = cx.build.GEP(obj_typarams,
                                              vec(C_int(0),
                                                  C_int(i)));
        lltyparam = cx.build.Load(lltyparam);
        cx.fcx.lltydescs.insert(p.id, lltyparam);
        i += 1;
    }

    i = 0;
4194 4195 4196 4197 4198 4199 4200 4201 4202
    for (ast.obj_field f in cx.fcx.ccx.obj_fields) {
        let ValueRef llfield = cx.build.GEP(obj_fields,
                                            vec(C_int(0),
                                                C_int(i)));
        cx.fcx.llobjfields.insert(f.id, llfield);
        i += 1;
    }
}

4203 4204 4205
fn trans_fn(@crate_ctxt cx, &ast._fn f, ast.def_id fid,
            option.t[TypeRef] ty_self,
            &vec[ast.ty_param] ty_params, &ast.ann ann) {
4206

4207 4208 4209
    auto llfndecl = cx.item_ids.get(fid);
    cx.item_names.insert(cx.path, llfndecl);

4210
    auto fcx = new_fn_ctxt(cx, cx.path, llfndecl);
4211
    create_llargs_for_fn_args(fcx, f.proto,
4212
                              ty_self, ret_ty_of_fn(ann),
4213
                              f.decl.inputs, ty_params);
4214
    auto bcx = new_top_block_ctxt(fcx);
4215

4216
    copy_args_to_allocas(bcx, ty_self, f.decl.inputs,
4217 4218 4219 4220
                         arg_tys_of_fn(ann));

    alt (fcx.llself) {
        case (some[ValueRef](?llself)) {
4221
            populate_fn_ctxt_from_llself(bcx, llself);
4222 4223 4224 4225
        }
        case (_) {
        }
    }
4226

4227 4228
    auto res = trans_block(bcx, f.body);
    if (!is_terminated(res.bcx)) {
4229 4230
        // FIXME: until LLVM has a unit type, we are moving around
        // C_nil values rather than their void type.
4231
        res.bcx.build.RetVoid();
4232
    }
4233 4234
}

4235 4236 4237
fn trans_vtbl(@crate_ctxt cx, TypeRef self_ty,
              &ast._obj ob,
              &vec[ast.ty_param] ty_params) -> ValueRef {
G
Graydon Hoare 已提交
4238
    let vec[ValueRef] methods = vec();
4239 4240 4241 4242 4243 4244 4245 4246 4247

    fn meth_lteq(&@ast.method a, &@ast.method b) -> bool {
        ret _str.lteq(a.node.ident, b.node.ident);
    }

    auto meths = std.sort.merge_sort[@ast.method](bind meth_lteq(_,_),
                                                  ob.methods);

    for (@ast.method m in meths) {
4248

4249 4250
        auto llfnty = T_nil();
        alt (node_ann_type(cx, m.node.ann).struct) {
4251 4252
            case (ty.ty_fn(?proto, ?inputs, ?output)) {
                llfnty = type_of_fn_full(cx, proto,
4253 4254 4255 4256 4257
                                         some[TypeRef](self_ty),
                                         inputs, output);
            }
        }

4258
        let @crate_ctxt mcx = @rec(path=cx.path + sep() + m.node.ident
4259 4260
                                   with *cx);

4261
        let str s = cx.names.next("_rust_method") + sep() + mcx.path;
4262 4263 4264
        let ValueRef llfn = decl_fastcall_fn(cx.llmod, s, llfnty);
        cx.item_ids.insert(m.node.id, llfn);

4265 4266
        trans_fn(mcx, m.node.meth, m.node.id, some[TypeRef](self_ty),
                 ty_params, m.node.ann);
4267
        methods += llfn;
G
Graydon Hoare 已提交
4268
    }
4269 4270 4271
    auto vtbl = C_struct(methods);
    auto gvar = llvm.LLVMAddGlobal(cx.llmod,
                                   val_ty(vtbl),
4272
                                   _str.buf("_rust_vtbl" + sep() + cx.path));
4273 4274
    llvm.LLVMSetInitializer(gvar, vtbl);
    llvm.LLVMSetGlobalConstant(gvar, True);
4275 4276
    llvm.LLVMSetLinkage(gvar, lib.llvm.LLVMPrivateLinkage
                        as llvm.Linkage);
4277
    ret gvar;
G
Graydon Hoare 已提交
4278 4279
}

4280 4281
fn trans_obj(@crate_ctxt cx, &ast._obj ob, ast.def_id oid,
             &vec[ast.ty_param] ty_params, &ast.ann ann) {
4282 4283 4284 4285

    auto llctor_decl = cx.item_ids.get(oid);
    cx.item_names.insert(cx.path, llctor_decl);

4286
    // Translate obj ctor args to function arguments.
4287 4288 4289 4290 4291 4292 4293 4294 4295
    let vec[ast.arg] fn_args = vec();
    for (ast.obj_field f in ob.fields) {
        fn_args += vec(rec(mode=ast.alias,
                           ty=f.ty,
                           ident=f.ident,
                           id=f.id));
    }

    auto fcx = new_fn_ctxt(cx, cx.path, llctor_decl);
4296
    create_llargs_for_fn_args(fcx, ast.proto_fn,
4297
                              none[TypeRef], ret_ty_of_fn(ann),
4298
                              fn_args, ty_params);
4299 4300 4301

    auto bcx = new_top_block_ctxt(fcx);

4302
    let vec[ty.arg] arg_tys = arg_tys_of_fn(ann);
4303
    copy_args_to_allocas(bcx, none[TypeRef], fn_args, arg_tys);
4304

4305
    auto llself_ty = type_of(cx, ret_ty_of_fn(ann));
4306
    auto pair = bcx.fcx.llretptr;
4307
    auto vtbl = trans_vtbl(cx, llself_ty, ob, ty_params);
4308 4309 4310
    auto pair_vtbl = bcx.build.GEP(pair,
                                   vec(C_int(0),
                                       C_int(abi.obj_field_vtbl)));
4311 4312 4313
    auto pair_box = bcx.build.GEP(pair,
                                  vec(C_int(0),
                                      C_int(abi.obj_field_box)));
4314
    bcx.build.Store(vtbl, pair_vtbl);
4315

4316
    let TypeRef llbox_ty = T_opaque_obj_ptr(cx.tn);
4317 4318 4319 4320

    if (_vec.len[ast.ty_param](ty_params) == 0u &&
        _vec.len[ty.arg](arg_tys) == 0u) {
        // Store null into pair, if no args or typarams.
4321 4322 4323 4324 4325 4326 4327
        bcx.build.Store(C_null(llbox_ty), pair_box);
    } else {
        // Malloc a box for the body and copy args in.
        let vec[@ty.t] obj_fields = vec();
        for (ty.arg a in arg_tys) {
            append[@ty.t](obj_fields, a.ty);
        }
4328 4329

        // Synthesize an obj body type.
4330 4331 4332 4333 4334 4335 4336
        auto tydesc_ty = plain_ty(ty.ty_type);
        let vec[@ty.t] tps = vec();
        for (ast.ty_param tp in ty_params) {
            append[@ty.t](tps, tydesc_ty);
        }

        let @ty.t typarams_ty = plain_ty(ty.ty_tup(tps));
4337
        let @ty.t fields_ty = plain_ty(ty.ty_tup(obj_fields));
4338 4339
        let @ty.t body_ty = plain_ty(ty.ty_tup(vec(tydesc_ty,
                                                   typarams_ty,
4340 4341 4342 4343
                                                   fields_ty)));
        let @ty.t boxed_body_ty = plain_ty(ty.ty_box(body_ty));

        let TypeRef llboxed_body_ty = type_of(cx, boxed_body_ty);
4344 4345

        // Malloc a box for the body.
4346 4347 4348 4349 4350 4351 4352 4353 4354
        auto box = trans_malloc_inner(bcx, llboxed_body_ty);
        bcx = box.bcx;
        auto rc = GEP_tup_like(bcx, boxed_body_ty, box.val,
                               vec(0, abi.box_rc_field_refcnt));
        bcx = rc.bcx;
        auto body = GEP_tup_like(bcx, boxed_body_ty, box.val,
                                 vec(0, abi.box_rc_field_body));
        bcx = body.bcx;
        bcx.build.Store(C_int(1), rc.val);
4355

4356 4357
        // Store body tydesc.
        auto body_tydesc =
4358 4359 4360
            GEP_tup_like(bcx, body_ty, body.val,
                         vec(0, abi.obj_body_elt_tydesc));
        bcx = body_tydesc.bcx;
4361

4362 4363 4364
        auto body_td = get_tydesc(bcx, body_ty);
        bcx = body_td.bcx;
        bcx.build.Store(body_td.val, body_tydesc.val);
4365

4366 4367 4368 4369 4370 4371 4372 4373 4374 4375 4376 4377 4378 4379 4380
        // Copy typarams into captured typarams.
        auto body_typarams =
            GEP_tup_like(bcx, body_ty, body.val,
                         vec(0, abi.obj_body_elt_typarams));
        bcx = body_typarams.bcx;
        let int i = 0;
        for (ast.ty_param tp in ty_params) {
            auto typaram = bcx.fcx.lltydescs.get(tp.id);
            auto capture = GEP_tup_like(bcx, typarams_ty, body_typarams.val,
                                        vec(0, i));
            bcx = capture.bcx;
            bcx = copy_ty(bcx, INIT, capture.val, typaram, tydesc_ty).bcx;
            i += 1;
        }

4381 4382
        // Copy args into body fields.
        auto body_fields =
4383 4384 4385
            GEP_tup_like(bcx, body_ty, body.val,
                         vec(0, abi.obj_body_elt_fields));
        bcx = body_fields.bcx;
4386

4387
        i = 0;
4388
        for (ast.obj_field f in ob.fields) {
4389 4390
            auto arg = bcx.fcx.llargs.get(f.id);
            arg = load_scalar_or_boxed(bcx, arg, arg_tys.(i).ty);
4391 4392 4393 4394
            auto field = GEP_tup_like(bcx, fields_ty, body_fields.val,
                                      vec(0, i));
            bcx = field.bcx;
            bcx = copy_ty(bcx, INIT, field.val, arg, arg_tys.(i).ty).bcx;
4395 4396 4397
            i += 1;
        }
        // Store box ptr in outer pair.
4398
        auto p = bcx.build.PointerCast(box.val, llbox_ty);
4399
        bcx.build.Store(p, pair_box);
4400
    }
4401
    bcx.build.RetVoid();
4402 4403
}

4404
fn trans_tag_variant(@crate_ctxt cx, ast.def_id tag_id,
4405 4406
                     &ast.variant variant, int index,
                     &vec[ast.ty_param] ty_params) {
4407
    if (_vec.len[ast.variant_arg](variant.args) == 0u) {
4408 4409 4410
        ret;    // nullary constructors are just constants
    }

4411 4412 4413 4414 4415 4416 4417 4418 4419 4420
    // Translate variant arguments to function arguments.
    let vec[ast.arg] fn_args = vec();
    auto i = 0u;
    for (ast.variant_arg varg in variant.args) {
        fn_args += vec(rec(mode=ast.alias,
                           ty=varg.ty,
                           ident="arg" + _uint.to_str(i, 10u),
                           id=varg.id));
    }

4421
    check (cx.item_ids.contains_key(variant.id));
4422 4423
    let ValueRef llfndecl = cx.item_ids.get(variant.id);

4424
    auto fcx = new_fn_ctxt(cx, cx.path, llfndecl);
4425
    create_llargs_for_fn_args(fcx, ast.proto_fn,
4426
                              none[TypeRef], ret_ty_of_fn(variant.ann),
4427
                              fn_args, ty_params);
4428

4429 4430 4431
    auto bcx = new_top_block_ctxt(fcx);

    auto arg_tys = arg_tys_of_fn(variant.ann);
4432
    copy_args_to_allocas(bcx, none[TypeRef], fn_args, arg_tys);
4433

4434 4435 4436
    // FIXME: This is wrong for generic tags. We should be dynamically
    // computing "size" below based on the tydescs passed in.
    auto info = cx.tags.get(mk_plain_tag(tag_id));
4437 4438 4439 4440 4441 4442

    auto lltagty = T_struct(vec(T_int(), T_array(T_i8(), info.size)));

    // FIXME: better name.
    llvm.LLVMAddTypeName(cx.llmod, _str.buf("tag"), lltagty);

4443 4444
    auto lldiscrimptr = bcx.build.GEP(fcx.llretptr,
                                      vec(C_int(0), C_int(0)));
4445 4446
    bcx.build.Store(C_int(index), lldiscrimptr);

4447 4448
    auto llblobptr = bcx.build.GEP(fcx.llretptr,
                                   vec(C_int(0), C_int(1)));
4449 4450 4451

    // First, generate the union type.
    let vec[TypeRef] llargtys = vec();
4452
    for (ty.arg arg in arg_tys) {
4453 4454 4455 4456 4457 4458 4459 4460
        llargtys += vec(type_of(cx, arg.ty));
    }

    auto llunionty = T_struct(llargtys);
    auto llunionptr = bcx.build.TruncOrBitCast(llblobptr, T_ptr(llunionty));

    i = 0u;
    for (ast.variant_arg va in variant.args) {
4461
        auto llargval = bcx.build.Load(fcx.llargs.get(va.id));
4462 4463 4464 4465 4466 4467 4468 4469
        auto lldestptr = bcx.build.GEP(llunionptr,
                                       vec(C_int(0), C_int(i as int)));

        bcx.build.Store(llargval, lldestptr);
        i += 1u;
    }

    bcx = trans_block_cleanups(bcx, find_scope_cx(bcx));
4470
    bcx.build.RetVoid();
4471 4472
}

4473 4474 4475 4476 4477 4478 4479 4480 4481 4482 4483 4484 4485 4486 4487 4488 4489 4490 4491 4492 4493 4494 4495 4496 4497 4498 4499 4500 4501 4502 4503
// FIXME: this should do some structural hash-consing to avoid
// duplicate constants. I think. Maybe LLVM has a magical mode
// that does so later on?

fn trans_const_expr(@crate_ctxt cx, @ast.expr e) -> ValueRef {
    alt (e.node) {
        case (ast.expr_lit(?lit, ?ann)) {
            ret trans_lit(cx, *lit, ann);
        }
    }
}

fn trans_const(@crate_ctxt cx, @ast.expr e,
               &ast.def_id cid, &ast.ann ann) {
    auto t = node_ann_type(cx, ann);
    auto v = trans_const_expr(cx, e);
    if (ty.type_is_scalar(t)) {
        // The scalars come back as 1st class LLVM vals
        // which we have to stick into global constants.
        auto g = llvm.LLVMAddGlobal(cx.llmod, val_ty(v),
                                    _str.buf(cx.names.next(cx.path)));
        llvm.LLVMSetInitializer(g, v);
        llvm.LLVMSetGlobalConstant(g, True);
        llvm.LLVMSetLinkage(g, lib.llvm.LLVMPrivateLinkage
                            as llvm.Linkage);
        cx.consts.insert(cid, g);
    } else {
        cx.consts.insert(cid, v);
    }
}

4504
fn trans_item(@crate_ctxt cx, &ast.item item) {
4505
    alt (item.node) {
4506
        case (ast.item_fn(?name, ?f, ?tps, ?fid, ?ann)) {
4507
            auto sub_cx = @rec(path=cx.path + sep() + name with *cx);
4508
            trans_fn(sub_cx, f, fid, none[TypeRef], tps, ann);
4509
        }
4510
        case (ast.item_obj(?name, ?ob, ?tps, ?oid, ?ann)) {
4511
            auto sub_cx = @rec(path=cx.path + sep() + name,
4512
                               obj_typarams=tps,
4513
                               obj_fields=ob.fields with *cx);
4514
            trans_obj(sub_cx, ob, oid, tps, ann);
4515
        }
4516
        case (ast.item_mod(?name, ?m, _)) {
4517
            auto sub_cx = @rec(path=cx.path + sep() + name with *cx);
4518
            trans_mod(sub_cx, m);
4519
        }
4520
        case (ast.item_tag(?name, ?variants, ?tps, ?tag_id)) {
4521
            auto sub_cx = @rec(path=cx.path + sep() + name with *cx);
4522
            auto i = 0;
4523
            for (ast.variant variant in variants) {
4524
                trans_tag_variant(sub_cx, tag_id, variant, i, tps);
4525
                i += 1;
4526 4527
            }
        }
4528
        case (ast.item_const(?name, _, ?expr, ?cid, ?ann)) {
4529
            auto sub_cx = @rec(path=cx.path + sep() + name with *cx);
4530 4531
            trans_const(sub_cx, expr, cid, ann);
        }
4532
        case (_) { /* fall through */ }
4533 4534 4535
    }
}

4536
fn trans_mod(@crate_ctxt cx, &ast._mod m) {
4537 4538
    for (@ast.item item in m.items) {
        trans_item(cx, *item);
4539 4540 4541
    }
}

4542 4543 4544 4545 4546 4547 4548 4549
fn get_pair_fn_ty(TypeRef llpairty) -> TypeRef {
    // Bit of a kludge: pick the fn typeref out of the pair.
    let vec[TypeRef] pair_tys = vec(T_nil(), T_nil());
    llvm.LLVMGetStructElementTypes(llpairty,
                                   _vec.buf[TypeRef](pair_tys));
    ret llvm.LLVMGetElementType(pair_tys.(0));
}

4550 4551 4552 4553 4554 4555 4556
fn decl_fn_and_pair(@crate_ctxt cx,
                    str kind,
                    str name,
                    &ast.ann ann,
                    ast.def_id id) {

    auto llpairty = node_type(cx, ann);
4557
    auto llfty = get_pair_fn_ty(llpairty);
4558 4559

    // Declare the function itself.
4560
    let str s = cx.names.next("_rust_" + kind) + sep() + name;
4561 4562 4563
    let ValueRef llfn = decl_fastcall_fn(cx.llmod, s, llfty);

    // Declare the global constant pair that points to it.
4564
    let str ps = cx.names.next("_rust_" + kind + "_pair") + sep() + name;
4565 4566 4567 4568 4569 4570

    register_fn_pair(cx, ps, llpairty, llfn, id);
}

fn register_fn_pair(@crate_ctxt cx, str ps, TypeRef llpairty, ValueRef llfn,
                    ast.def_id id) {
4571 4572 4573
    let ValueRef gvar = llvm.LLVMAddGlobal(cx.llmod, llpairty,
                                           _str.buf(ps));
    auto pair = C_struct(vec(llfn,
4574
                             C_null(T_opaque_closure_ptr(cx.tn))));
4575 4576 4577 4578 4579 4580 4581 4582 4583 4584 4585

    llvm.LLVMSetInitializer(gvar, pair);
    llvm.LLVMSetGlobalConstant(gvar, True);
    llvm.LLVMSetLinkage(gvar,
                        lib.llvm.LLVMPrivateLinkage
                        as llvm.Linkage);

    cx.item_ids.insert(id, llfn);
    cx.fn_pairs.insert(id, gvar);
}

4586 4587 4588 4589 4590 4591 4592 4593 4594 4595
fn native_fn_wrapper_type(@crate_ctxt cx, &ast.ann ann) -> TypeRef {
    auto x = node_ann_type(cx, ann);
    alt (x.struct) {
        case (ty.ty_native_fn(?abi, ?args, ?out)) {
            ret type_of_fn(cx, ast.proto_fn, args, out);
        }
    }
    fail;
}

4596 4597 4598 4599
fn decl_native_fn_and_pair(@crate_ctxt cx,
                           str name,
                           &ast.ann ann,
                           ast.def_id id) {
4600 4601 4602 4603
    // Declare the wrapper.
    auto wrapper_type = native_fn_wrapper_type(cx, ann);
    let str s = cx.names.next("_rust_wrapper") + sep() + name;
    let ValueRef wrapper_fn = decl_fastcall_fn(cx.llmod, s, wrapper_type);
4604

4605 4606 4607
    // Declare the global constant pair that points to it.
    auto wrapper_pair_type = T_fn_pair(cx.tn, wrapper_type);
    let str ps = cx.names.next("_rust_wrapper_pair") + sep() + name;
4608

4609 4610 4611 4612 4613
    register_fn_pair(cx, ps, wrapper_pair_type, wrapper_fn, id);

    // Declare the function itself.
    auto llfty = get_pair_fn_ty(node_type(cx, ann));
    decl_cdecl_fn(cx.llmod, name, llfty);
4614 4615
}

4616 4617 4618 4619 4620
fn collect_native_item(&@crate_ctxt cx, @ast.native_item i) -> @crate_ctxt {
    alt (i.node) {
        case (ast.native_item_fn(?name, _, _, ?fid, ?ann)) {
            cx.native_items.insert(fid, i);
            if (! cx.obj_methods.contains_key(fid)) {
4621
                decl_native_fn_and_pair(cx, name, ann, fid);
4622 4623 4624 4625 4626 4627
            }
        }
        case (_) { /* fall through */ }
    }
    ret cx;
}
4628

4629
fn collect_item(&@crate_ctxt cx, @ast.item i) -> @crate_ctxt {
4630

4631
    alt (i.node) {
4632
        case (ast.item_fn(?name, ?f, _, ?fid, ?ann)) {
4633
            cx.items.insert(fid, i);
4634 4635 4636
            if (! cx.obj_methods.contains_key(fid)) {
                decl_fn_and_pair(cx, "fn", name, ann, fid);
            }
4637 4638
        }

4639 4640
        case (ast.item_obj(?name, ?ob, _, ?oid, ?ann)) {
            cx.items.insert(oid, i);
4641 4642 4643 4644
            decl_fn_and_pair(cx, "obj_ctor", name, ann, oid);
            for (@ast.method m in ob.methods) {
                cx.obj_methods.insert(m.node.id, ());
            }
4645 4646
        }

4647 4648 4649 4650
        case (ast.item_const(?name, _, _, ?cid, _)) {
            cx.items.insert(cid, i);
        }

4651 4652 4653
        case (ast.item_mod(?name, ?m, ?mid)) {
            cx.items.insert(mid, i);
        }
4654

4655
        case (ast.item_tag(_, ?variants, ?tps, ?tag_id)) {
4656 4657
            auto vi = new_def_hash[uint]();
            auto navi = new_def_hash[uint]();
4658 4659 4660 4661 4662 4663 4664 4665

            auto info = @rec(
                th=mk_type_handle(),
                mutable size=0u,
                mutable lldiscrims=@new_def_hash[ValueRef]()
            );

            cx.tags.insert(mk_plain_tag(tag_id), info);
4666
            cx.items.insert(tag_id, i);
4667 4668
        }

4669
        case (_) { /* fall through */ }
4670 4671 4672 4673 4674
    }
    ret cx;
}


4675
fn collect_items(@crate_ctxt cx, @ast.crate crate) {
4676

4677 4678
    let fold.ast_fold[@crate_ctxt] fld =
        fold.new_identity_fold[@crate_ctxt]();
4679

4680 4681
    fld = @rec( update_env_for_item = bind collect_item(_,_),
                update_env_for_native_item = bind collect_native_item(_,_)
4682 4683
                with *fld );

4684
    fold.fold_crate[@crate_ctxt](cx, fld, crate);
4685 4686
}

4687 4688 4689 4690 4691 4692 4693 4694 4695 4696 4697 4698 4699 4700 4701 4702 4703 4704 4705 4706 4707 4708 4709 4710 4711 4712 4713 4714 4715 4716
fn collect_tag_ctor(&@crate_ctxt cx, @ast.item i) -> @crate_ctxt {

    alt (i.node) {

        case (ast.item_tag(_, ?variants, _, _)) {
            for (ast.variant variant in variants) {
                if (_vec.len[ast.variant_arg](variant.args) != 0u) {
                    decl_fn_and_pair(cx, "tag", variant.name,
                                     variant.ann, variant.id);
                }
            }
        }

        case (_) { /* fall through */ }
    }
    ret cx;
}

fn collect_tag_ctors(@crate_ctxt cx, @ast.crate crate) {

    let fold.ast_fold[@crate_ctxt] fld =
        fold.new_identity_fold[@crate_ctxt]();

    fld = @rec( update_env_for_item = bind collect_tag_ctor(_,_)
                with *fld );

    fold.fold_crate[@crate_ctxt](cx, fld, crate);
}


4717 4718 4719 4720 4721 4722
// The tag type resolution pass, which determines all the LLVM types that
// correspond to each tag type in the crate.

fn resolve_tag_types_for_item(&@crate_ctxt cx, @ast.item i) -> @crate_ctxt {
    alt (i.node) {
        case (ast.item_tag(_, ?variants, _, ?tag_id)) {
4723 4724
            auto max_align = 0u;
            auto max_size = 0u;
4725

4726
            auto info = cx.tags.get(mk_plain_tag(tag_id));
4727 4728

            for (ast.variant variant in variants) {
4729
                if (_vec.len[ast.variant_arg](variant.args) > 0u) {
4730
                    auto llvariantty = type_of_variant(cx, variant);
G
Graydon Hoare 已提交
4731 4732 4733 4734 4735 4736
                    auto align =
                        llvm.LLVMPreferredAlignmentOfType(cx.td.lltd,
                                                          llvariantty);
                    auto size =
                        llvm.LLVMStoreSizeOfType(cx.td.lltd,
                                                 llvariantty) as uint;
4737 4738
                    if (max_align < align) { max_align = align; }
                    if (max_size < size) { max_size = size; }
4739 4740 4741
                }
            }

4742
            info.size = max_size;
4743

4744 4745 4746
            // FIXME: alignment is wrong here, manually insert padding I
            // guess :(
            auto tag_ty = T_struct(vec(T_int(), T_array(T_i8(), max_size)));
4747
            auto th = info.th.llth;
4748 4749 4750 4751 4752 4753 4754 4755 4756 4757
            llvm.LLVMRefineType(llvm.LLVMResolveTypeHandle(th), tag_ty);
        }
        case (_) {
            // fall through
        }
    }

    ret cx;
}

4758 4759 4760 4761 4762 4763 4764 4765 4766 4767
fn resolve_tag_types(@crate_ctxt cx, @ast.crate crate) {
    let fold.ast_fold[@crate_ctxt] fld =
        fold.new_identity_fold[@crate_ctxt]();

    fld = @rec( update_env_for_item = bind resolve_tag_types_for_item(_,_)
                with *fld );

    fold.fold_crate[@crate_ctxt](cx, fld, crate);
}

4768 4769 4770 4771 4772
// The constant translation pass.

fn trans_constant(&@crate_ctxt cx, @ast.item it) -> @crate_ctxt {
    alt (it.node) {
        case (ast.item_tag(_, ?variants, _, ?tag_id)) {
4773
            auto info = cx.tags.get(mk_plain_tag(tag_id));
4774 4775 4776 4777 4778 4779 4780 4781

            auto tag_ty = llvm.LLVMResolveTypeHandle(info.th.llth);
            check (llvm.LLVMCountStructElementTypes(tag_ty) == 2u);
            auto elts = vec(0 as TypeRef, 0 as TypeRef);
            llvm.LLVMGetStructElementTypes(tag_ty, _vec.buf[TypeRef](elts));
            auto union_ty = elts.(1);

            auto i = 0u;
4782 4783 4784
            auto n_variants = _vec.len[ast.variant](variants);
            while (i < n_variants) {
                auto variant = variants.(i);
4785 4786 4787 4788 4789 4790 4791 4792 4793 4794 4795 4796 4797 4798

                auto discrim_val = C_int(i as int);

                // FIXME: better name.
                auto discrim_gvar = llvm.LLVMAddGlobal(cx.llmod, T_int(),
                    _str.buf("tag_discrim"));

                // FIXME: Eventually we do want to export these, but we need
                // to figure out what name they get first!
                llvm.LLVMSetInitializer(discrim_gvar, discrim_val);
                llvm.LLVMSetGlobalConstant(discrim_gvar, True);
                llvm.LLVMSetLinkage(discrim_gvar, lib.llvm.LLVMPrivateLinkage
                                    as llvm.Linkage);

4799
                info.lldiscrims.insert(variant.id, discrim_gvar);
4800 4801 4802 4803

                i += 1u;
            }
        }
4804 4805 4806 4807 4808 4809 4810 4811

        case (ast.item_const(?name, _, ?expr, ?cid, ?ann)) {
            // FIXME: The whole expr-translation system needs cloning to deal
            // with consts.
            auto v = C_int(1);
            cx.item_ids.insert(cid, v);
        }

4812 4813 4814 4815 4816 4817 4818 4819 4820 4821 4822 4823 4824 4825 4826 4827 4828
        case (_) {
            // empty
        }
    }

    ret cx;
}

fn trans_constants(@crate_ctxt cx, @ast.crate crate) {
    let fold.ast_fold[@crate_ctxt] fld =
        fold.new_identity_fold[@crate_ctxt]();

    fld = @rec(update_env_for_item = bind trans_constant(_,_) with *fld);

    fold.fold_crate[@crate_ctxt](cx, fld, crate);
}

4829 4830 4831 4832
fn p2i(ValueRef v) -> ValueRef {
    ret llvm.LLVMConstPtrToInt(v, T_int());
}

4833 4834 4835 4836
fn i2p(ValueRef v, TypeRef t) -> ValueRef {
    ret llvm.LLVMConstIntToPtr(v, t);
}

4837
fn trans_exit_task_glue(@crate_ctxt cx) {
4838 4839 4840 4841
    let vec[TypeRef] T_args = vec();
    let vec[ValueRef] V_args = vec();

    auto llfn = cx.glues.exit_task_glue;
4842
    let ValueRef lltaskptr = llvm.LLVMGetParam(llfn, 3u);
4843 4844
    auto fcx = @rec(llfn=llfn,
                    lltaskptr=lltaskptr,
4845
                    llenv=C_null(T_opaque_closure_ptr(cx.tn)),
4846
                    llretptr=C_null(T_ptr(T_nil())),
4847
                    mutable llself=none[ValueRef],
4848
                    mutable lliterbody=none[ValueRef],
4849
                    llargs=new_def_hash[ValueRef](),
4850
                    llobjfields=new_def_hash[ValueRef](),
4851
                    lllocals=new_def_hash[ValueRef](),
4852
                    lltydescs=new_def_hash[ValueRef](),
4853
                    ccx=cx);
4854

4855
    auto bcx = new_top_block_ctxt(fcx);
4856
    trans_upcall(bcx, "upcall_exit", V_args);
4857
    bcx.build.RetVoid();
4858 4859
}

4860
fn create_typedefs(@crate_ctxt cx) {
4861 4862 4863
    llvm.LLVMAddTypeName(cx.llmod, _str.buf("crate"), T_crate(cx.tn));
    llvm.LLVMAddTypeName(cx.llmod, _str.buf("task"), T_task(cx.tn));
    llvm.LLVMAddTypeName(cx.llmod, _str.buf("tydesc"), T_tydesc(cx.tn));
4864 4865
}

4866
fn create_crate_constant(@crate_ctxt cx) {
4867

4868
    let ValueRef crate_addr = p2i(cx.crate_ptr);
4869 4870 4871 4872 4873 4874 4875

    let ValueRef activate_glue_off =
        llvm.LLVMConstSub(p2i(cx.glues.activate_glue), crate_addr);

    let ValueRef yield_glue_off =
        llvm.LLVMConstSub(p2i(cx.glues.yield_glue), crate_addr);

4876 4877
    let ValueRef exit_task_glue_off =
        llvm.LLVMConstSub(p2i(cx.glues.exit_task_glue), crate_addr);
4878 4879 4880

    let ValueRef crate_val =
        C_struct(vec(C_null(T_int()),     // ptrdiff_t image_base_off
4881
                     p2i(cx.crate_ptr),   // uintptr_t self_addr
4882 4883 4884 4885 4886 4887 4888 4889 4890 4891 4892
                     C_null(T_int()),     // ptrdiff_t debug_abbrev_off
                     C_null(T_int()),     // size_t debug_abbrev_sz
                     C_null(T_int()),     // ptrdiff_t debug_info_off
                     C_null(T_int()),     // size_t debug_info_sz
                     activate_glue_off,   // size_t activate_glue_off
                     yield_glue_off,      // size_t yield_glue_off
                     C_null(T_int()),     // size_t unwind_glue_off
                     C_null(T_int()),     // size_t gc_glue_off
                     exit_task_glue_off,  // size_t main_exit_task_glue_off
                     C_null(T_int()),     // int n_rust_syms
                     C_null(T_int()),     // int n_c_syms
4893 4894
                     C_null(T_int()),     // int n_libs
                     C_int(abi.abi_x86_rustc_fastcall) // uintptr_t abi_tag
4895 4896
                     ));

4897
    llvm.LLVMSetInitializer(cx.crate_ptr, crate_val);
4898 4899
}

4900 4901 4902 4903 4904 4905 4906 4907 4908 4909 4910 4911 4912 4913 4914 4915 4916 4917 4918 4919 4920 4921 4922 4923
fn find_main_fn(@crate_ctxt cx) -> ValueRef {
    auto e = sep() + "main";
    let ValueRef v = C_nil();
    let uint n = 0u;
    for each (tup(str,ValueRef) i in cx.item_names.items()) {
        if (_str.ends_with(i._0, e)) {
            n += 1u;
            v = i._1;
        }
    }
    alt (n) {
        case (0u) {
            cx.sess.err("main fn not found");
        }
        case (1u) {
            ret v;
        }
        case (_) {
            cx.sess.err("multiple main fns found");
        }
    }
    fail;
}

4924
fn trans_main_fn(@crate_ctxt cx, ValueRef llcrate) {
4925 4926 4927
    auto T_main_args = vec(T_int(), T_int());
    auto T_rust_start_args = vec(T_int(), T_int(), T_int(), T_int());

4928 4929 4930 4931 4932 4933 4934
    auto main_name;
    if (_str.eq(std.os.target_os(), "win32")) {
        main_name = "WinMain@16";
    } else {
        main_name = "main";
    }

4935
    auto llmain =
4936
        decl_cdecl_fn(cx.llmod, main_name, T_fn(T_main_args, T_int()));
4937

4938 4939
    auto llrust_start = decl_cdecl_fn(cx.llmod, "rust_start",
                                      T_fn(T_rust_start_args, T_int()));
4940 4941 4942

    auto llargc = llvm.LLVMGetParam(llmain, 0u);
    auto llargv = llvm.LLVMGetParam(llmain, 1u);
4943
    auto llrust_main = find_main_fn(cx);
4944 4945 4946 4947 4948 4949 4950 4951 4952 4953 4954

    //
    // Emit the moral equivalent of:
    //
    // main(int argc, char **argv) {
    //     rust_start(&_rust.main, &crate, argc, argv);
    // }
    //

    let BasicBlockRef llbb =
        llvm.LLVMAppendBasicBlock(llmain, _str.buf(""));
4955
    auto b = new_builder(llbb);
4956 4957 4958 4959 4960 4961

    auto start_args = vec(p2i(llrust_main), p2i(llcrate), llargc, llargv);

    b.Ret(b.Call(llrust_start, start_args));
}

4962 4963
fn declare_intrinsics(ModuleRef llmod) -> hashmap[str,ValueRef] {

4964
    let vec[TypeRef] T_trap_args = vec();
4965 4966 4967 4968 4969 4970
    auto trap = decl_cdecl_fn(llmod, "llvm.trap",
                              T_fn(T_trap_args, T_void()));

    auto intrinsics = new_str_hash[ValueRef]();
    intrinsics.insert("llvm.trap", trap);
    ret intrinsics;
4971 4972
}

4973 4974 4975 4976 4977 4978 4979 4980 4981 4982 4983 4984 4985 4986 4987 4988 4989 4990

fn trace_str(@block_ctxt cx, str s) {
    trans_upcall(cx, "upcall_trace_str", vec(p2i(C_cstr(cx.fcx.ccx, s))));
}

fn trace_word(@block_ctxt cx, ValueRef v) {
    trans_upcall(cx, "upcall_trace_word", vec(v));
}

fn trace_ptr(@block_ctxt cx, ValueRef v) {
    trace_word(cx, cx.build.PtrToInt(v, T_int()));
}

fn trap(@block_ctxt bcx) {
    let vec[ValueRef] v = vec();
    bcx.build.Call(bcx.fcx.ccx.intrinsics.get("llvm.trap"), v);
}

4991 4992 4993 4994 4995 4996 4997 4998
fn check_module(ModuleRef llmod) {
    auto pm = mk_pass_manager();
    llvm.LLVMAddVerifierPass(pm.llpm);
    llvm.LLVMRunPassManager(pm.llpm, llmod);

    // TODO: run the linter here also, once there are llvm-c bindings for it.
}

4999 5000
fn make_no_op_type_glue(ModuleRef llmod, type_names tn) -> ValueRef {
    auto ty = T_fn(vec(T_taskptr(tn), T_ptr(T_i8())), T_void());
5001
    auto fun = decl_fastcall_fn(llmod, abi.no_op_type_glue_name(), ty);
5002 5003
    auto bb_name = _str.buf("_rust_no_op_type_glue_bb");
    auto llbb = llvm.LLVMAppendBasicBlock(fun, bb_name);
5004
    new_builder(llbb).RetVoid();
5005 5006 5007
    ret fun;
}

5008 5009 5010 5011 5012 5013 5014 5015 5016 5017 5018 5019 5020 5021 5022 5023 5024 5025 5026 5027 5028 5029 5030 5031 5032 5033 5034 5035 5036 5037 5038 5039 5040 5041 5042 5043 5044 5045 5046 5047 5048 5049 5050 5051 5052
fn make_memcpy_glue(ModuleRef llmod) -> ValueRef {

    // We're not using the LLVM memcpy intrinsic. It appears to call through
    // to the platform memcpy in some cases, which is not terribly safe to run
    // on a rust stack.

    auto p8 = T_ptr(T_i8());

    auto ty = T_fn(vec(p8, p8, T_int()), T_void());
    auto fun = decl_fastcall_fn(llmod, abi.memcpy_glue_name(), ty);

    auto initbb = llvm.LLVMAppendBasicBlock(fun, _str.buf("init"));
    auto hdrbb = llvm.LLVMAppendBasicBlock(fun, _str.buf("hdr"));
    auto loopbb = llvm.LLVMAppendBasicBlock(fun, _str.buf("loop"));
    auto endbb = llvm.LLVMAppendBasicBlock(fun, _str.buf("end"));

    auto dst = llvm.LLVMGetParam(fun, 0u);
    auto src = llvm.LLVMGetParam(fun, 1u);
    auto count = llvm.LLVMGetParam(fun, 2u);

    // Init block.
    auto ib = new_builder(initbb);
    auto ip = ib.Alloca(T_int());
    ib.Store(C_int(0), ip);
    ib.Br(hdrbb);

    // Loop-header block
    auto hb = new_builder(hdrbb);
    auto i = hb.Load(ip);
    hb.CondBr(hb.ICmp(lib.llvm.LLVMIntEQ, count, i), endbb, loopbb);

    // Loop-body block
    auto lb = new_builder(loopbb);
    i = lb.Load(ip);
    lb.Store(lb.Load(lb.GEP(src, vec(i))),
             lb.GEP(dst, vec(i)));
    lb.Store(lb.Add(i, C_int(1)), ip);
    lb.Br(hdrbb);

    // End block
    auto eb = new_builder(endbb);
    eb.RetVoid();
    ret fun;
}

5053 5054 5055 5056 5057 5058 5059 5060 5061 5062 5063 5064 5065 5066 5067 5068 5069 5070 5071 5072 5073 5074 5075 5076 5077 5078 5079 5080 5081 5082 5083 5084 5085 5086 5087 5088 5089 5090 5091 5092 5093
fn make_bzero_glue(ModuleRef llmod) -> ValueRef {

    // We're not using the LLVM memset intrinsic. Same as with memcpy.

    auto p8 = T_ptr(T_i8());

    auto ty = T_fn(vec(p8, T_int()), T_void());
    auto fun = decl_fastcall_fn(llmod, abi.bzero_glue_name(), ty);

    auto initbb = llvm.LLVMAppendBasicBlock(fun, _str.buf("init"));
    auto hdrbb = llvm.LLVMAppendBasicBlock(fun, _str.buf("hdr"));
    auto loopbb = llvm.LLVMAppendBasicBlock(fun, _str.buf("loop"));
    auto endbb = llvm.LLVMAppendBasicBlock(fun, _str.buf("end"));

    auto dst = llvm.LLVMGetParam(fun, 0u);
    auto count = llvm.LLVMGetParam(fun, 1u);

    // Init block.
    auto ib = new_builder(initbb);
    auto ip = ib.Alloca(T_int());
    ib.Store(C_int(0), ip);
    ib.Br(hdrbb);

    // Loop-header block
    auto hb = new_builder(hdrbb);
    auto i = hb.Load(ip);
    hb.CondBr(hb.ICmp(lib.llvm.LLVMIntEQ, count, i), endbb, loopbb);

    // Loop-body block
    auto lb = new_builder(loopbb);
    i = lb.Load(ip);
    lb.Store(C_integral(0, T_i8()), lb.GEP(dst, vec(i)));
    lb.Store(lb.Add(i, C_int(1)), ip);
    lb.Br(hdrbb);

    // End block
    auto eb = new_builder(endbb);
    eb.RetVoid();
    ret fun;
}

5094 5095 5096
fn make_glues(ModuleRef llmod, type_names tn) -> @glue_fns {
    ret @rec(activate_glue = decl_glue(llmod, tn, abi.activate_glue_name()),
             yield_glue = decl_glue(llmod, tn, abi.yield_glue_name()),
5097 5098 5099 5100 5101 5102 5103 5104 5105 5106 5107
             /*
              * Note: the signature passed to decl_cdecl_fn here looks unusual
              * because it is. It corresponds neither to an upcall signature
              * nor a normal rust-ABI signature. In fact it is a fake
              * signature, that exists solely to acquire the task pointer as
              * an argument to the upcall. It so happens that the runtime sets
              * up the task pointer as the sole incoming argument to the frame
              * that we return into when returning to the exit task glue. So
              * this is the signature required to retrieve it.
              */
             exit_task_glue = decl_cdecl_fn(llmod, abi.exit_task_glue_name(),
5108 5109 5110 5111
                                            T_fn(vec(T_int(),
                                                     T_int(),
                                                     T_int(),
                                                     T_taskptr(tn)),
5112
                                                 T_void())),
5113 5114

             upcall_glues =
5115
             _vec.init_fn[ValueRef](bind decl_upcall_glue(llmod, tn, _),
G
Graydon Hoare 已提交
5116
                                    abi.n_upcall_glues as uint),
5117
             no_op_type_glue = make_no_op_type_glue(llmod, tn),
5118 5119
             memcpy_glue = make_memcpy_glue(llmod),
             bzero_glue = make_bzero_glue(llmod));
5120 5121
}

5122 5123
fn trans_crate(session.session sess, @ast.crate crate, str output,
               bool shared) {
5124 5125 5126 5127
    auto llmod =
        llvm.LLVMModuleCreateWithNameInContext(_str.buf("rust_out"),
                                               llvm.LLVMGetGlobalContext());

5128 5129
    llvm.LLVMSetDataLayout(llmod, _str.buf(x86.get_data_layout()));
    llvm.LLVMSetTarget(llmod, _str.buf(x86.get_target_triple()));
5130
    auto td = mk_target_data(x86.get_data_layout());
5131
    auto tn = mk_type_names();
5132
    let ValueRef crate_ptr =
5133
        llvm.LLVMAddGlobal(llmod, T_crate(tn), _str.buf("rust_crate"));
5134

5135 5136
    llvm.LLVMSetModuleInlineAsm(llmod, _str.buf(x86.get_module_asm()));

5137
    auto intrinsics = declare_intrinsics(llmod);
5138

5139
    auto glues = make_glues(llmod, tn);
5140 5141
    auto hasher = ty.hash_ty;
    auto eqer = ty.eq_ty;
5142
    auto tags = map.mk_hashmap[@ty.t,@tag_info](hasher, eqer);
5143
    auto tydescs = map.mk_hashmap[@ty.t,ValueRef](hasher, eqer);
5144
    let vec[ast.ty_param] obj_typarams = vec();
5145
    let vec[ast.obj_field] obj_fields = vec();
5146

5147 5148
    auto cx = @rec(sess = sess,
                   llmod = llmod,
5149
                   td = td,
5150
                   tn = tn,
5151
                   crate_ptr = crate_ptr,
5152
                   upcalls = new_str_hash[ValueRef](),
5153
                   intrinsics = intrinsics,
5154 5155
                   item_names = new_str_hash[ValueRef](),
                   item_ids = new_def_hash[ValueRef](),
5156
                   items = new_def_hash[@ast.item](),
5157
                   native_items = new_def_hash[@ast.native_item](),
5158
                   tags = tags,
5159
                   fn_pairs = new_def_hash[ValueRef](),
5160
                   consts = new_def_hash[ValueRef](),
5161
                   obj_methods = new_def_hash[()](),
5162
                   tydescs = tydescs,
5163
                   obj_typarams = obj_typarams,
5164
                   obj_fields = obj_fields,
5165
                   glues = glues,
5166
                   names = namegen(0),
5167
                   path = "_rust");
5168

5169 5170
    create_typedefs(cx);

5171
    collect_items(cx, crate);
5172
    resolve_tag_types(cx, crate);
5173
    collect_tag_ctors(cx, crate);
5174
    trans_constants(cx, crate);
5175

5176
    trans_mod(cx, crate.node.module);
5177
    trans_exit_task_glue(cx);
5178
    create_crate_constant(cx);
5179
    if (!shared) {
5180
        trans_main_fn(cx, cx.crate_ptr);
5181
    }
5182

5183 5184
    check_module(llmod);

5185
    llvm.LLVMWriteBitcodeToFile(llmod, _str.buf(output));
5186 5187 5188 5189 5190 5191 5192 5193 5194 5195 5196 5197 5198
    llvm.LLVMDisposeModule(llmod);
}

//
// Local Variables:
// mode: rust
// fill-column: 78;
// indent-tabs-mode: nil
// c-basic-offset: 4
// buffer-file-coding-system: utf-8-unix
// compile-command: "make -k -C ../.. 2>&1 | sed -e 's/\\/x\\//x:\\//g'";
// End:
//