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

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

18
import util.common;
19
import util.common.istr;
20
import util.common.new_def_hash;
21
import util.common.new_str_hash;
22 23 24

import lib.llvm.llvm;
import lib.llvm.builder;
25
import lib.llvm.target_data;
26
import lib.llvm.type_handle;
27
import lib.llvm.mk_pass_manager;
28
import lib.llvm.mk_target_data;
29
import lib.llvm.mk_type_handle;
30 31 32
import lib.llvm.llvm.ModuleRef;
import lib.llvm.llvm.ValueRef;
import lib.llvm.llvm.TypeRef;
33
import lib.llvm.llvm.TypeHandleRef;
34 35
import lib.llvm.llvm.BuilderRef;
import lib.llvm.llvm.BasicBlockRef;
36

37 38
import lib.llvm.False;
import lib.llvm.True;
39

40 41 42 43 44 45 46
state obj namegen(mutable int i) {
    fn next(str prefix) -> str {
        i += 1;
        ret prefix + istr(i);
    }
}

47 48
type glue_fns = rec(ValueRef activate_glue,
                    ValueRef yield_glue,
49
                    ValueRef exit_task_glue,
50 51
                    vec[ValueRef] upcall_glues);

52
tag arity { nullary; n_ary; }
53
type tag_info = rec(type_handle th,
54 55
                    mutable vec[tup(ast.def_id,arity)] variants,
                    mutable uint size);
56

57
type ty_info = rec(ValueRef take_glue, ValueRef drop_glue);
58

59
state type crate_ctxt = rec(session.session sess,
60
                            ModuleRef llmod,
61
                            target_data td,
62
                            hashmap[str, ValueRef] upcalls,
63
                            hashmap[str, ValueRef] intrinsics,
64 65
                            hashmap[str, ValueRef] item_names,
                            hashmap[ast.def_id, ValueRef] item_ids,
66
                            hashmap[ast.def_id, @ast.item] items,
67
                            hashmap[ast.def_id, @tag_info] tags,
68
                            hashmap[@typeck.ty, @ty_info] types,
69 70 71
                            @glue_fns glues,
                            namegen names,
                            str path);
72

73 74
state type fn_ctxt = rec(ValueRef llfn,
                         ValueRef lltaskptr,
75
                         hashmap[ast.def_id, ValueRef] llargs,
76
                         hashmap[ast.def_id, ValueRef] lllocals,
77
                         @crate_ctxt ccx);
78

79
tag cleanup {
80
    clean(fn(@block_ctxt cx) -> result);
81 82 83 84
}

state type block_ctxt = rec(BasicBlockRef llbb,
                            builder build,
85
                            block_parent parent,
86
                            bool is_scope,
87 88 89
                            mutable vec[cleanup] cleanups,
                            @fn_ctxt fcx);

90 91 92 93 94 95 96 97
// 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);
}

98

99 100 101 102 103 104 105 106
state type result = rec(mutable @block_ctxt bcx,
                        mutable ValueRef val);

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

107 108 109 110 111 112 113 114 115 116 117
fn ty_str(TypeRef t) -> str {
    ret lib.llvm.type_to_str(t);
}

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

fn val_str(ValueRef v) -> str {
    ret ty_str(val_ty(v));
}
118 119 120 121


// LLVM type constructors.

122 123 124 125 126 127 128 129 130 131 132
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.
133 134 135
    ret llvm.LLVMVoidType();
}

136 137 138 139 140
fn T_nil() -> TypeRef {
    // NB: See above in T_void().
    ret llvm.LLVMInt1Type();
}

141 142 143 144
fn T_i1() -> TypeRef {
    ret llvm.LLVMInt1Type();
}

145 146 147 148 149 150 151 152 153
fn T_i8() -> TypeRef {
    ret llvm.LLVMInt8Type();
}

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

fn T_i32() -> TypeRef {
154 155 156
    ret llvm.LLVMInt32Type();
}

157 158 159 160
fn T_i64() -> TypeRef {
    ret llvm.LLVMInt64Type();
}

161 162 163 164 165 166 167 168
fn T_f32() -> TypeRef {
    ret llvm.LLVMFloatType();
}

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

169 170 171 172
fn T_bool() -> TypeRef {
    ret T_i1();
}

173 174 175 176 177
fn T_int() -> TypeRef {
    // FIXME: switch on target type.
    ret T_i32();
}

178 179 180 181
fn T_char() -> TypeRef {
    ret T_i32();
}

182 183 184 185
fn T_fn(vec[TypeRef] inputs, TypeRef output) -> TypeRef {
    ret llvm.LLVMFunctionType(output,
                              _vec.buf[TypeRef](inputs),
                              _vec.len[TypeRef](inputs),
186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204
                              False);
}

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

fn T_task() -> TypeRef {
    ret T_struct(vec(T_int(),      // Refcount
205 206 207 208 209 210 211 212 213 214
                     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
                     ));
}

215 216 217 218
fn T_array(TypeRef t, uint n) -> TypeRef {
    ret llvm.LLVMArrayType(t, n);
}

219 220 221 222 223
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
224 225 226
                     ));
}

227 228
fn T_str() -> TypeRef {
    ret T_vec(T_i8());
229 230
}

231 232 233 234
fn T_box(TypeRef t) -> TypeRef {
    ret T_struct(vec(T_int(), t));
}

235 236 237 238 239 240 241 242 243 244 245 246 247 248
fn T_crate() -> TypeRef {
    ret 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
249
                     T_int()       // int n_libs
250 251 252
                     ));
}

253 254 255 256 257 258
fn T_double() -> TypeRef {
    ret llvm.LLVMDoubleType();
}

fn T_taskptr() -> TypeRef {
    ret T_ptr(T_task());
259 260
}

261
fn type_of(@crate_ctxt cx, @typeck.ty t) -> TypeRef {
262 263
    let TypeRef llty = type_of_inner(cx, t);
    check (llty as int != 0);
264
    llvm.LLVMAddTypeName(cx.llmod, _str.buf(typeck.ty_to_str(t)), llty);
265 266 267
    ret llty;
}

268
fn type_of_inner(@crate_ctxt cx, @typeck.ty t) -> TypeRef {
269 270 271 272 273 274
    alt (t.struct) {
        case (typeck.ty_nil) { ret T_nil(); }
        case (typeck.ty_bool) { ret T_bool(); }
        case (typeck.ty_int) { ret T_int(); }
        case (typeck.ty_uint) { ret T_int(); }
        case (typeck.ty_machine(?tm)) {
275 276 277 278 279 280 281 282 283 284 285 286 287
            alt (tm) {
                case (common.ty_i8) { ret T_i8(); }
                case (common.ty_u8) { ret T_i8(); }
                case (common.ty_i16) { ret T_i16(); }
                case (common.ty_u16) { ret T_i16(); }
                case (common.ty_i32) { ret T_i32(); }
                case (common.ty_u32) { ret T_i32(); }
                case (common.ty_i64) { ret T_i64(); }
                case (common.ty_u64) { ret T_i64(); }
                case (common.ty_f32) { ret T_f32(); }
                case (common.ty_f64) { ret T_f64(); }
            }
        }
288
        case (typeck.ty_char) { ret T_char(); }
289
        case (typeck.ty_str) { ret T_ptr(T_str()); }
290 291 292
        case (typeck.ty_tag(?tag_id)) {
            ret llvm.LLVMResolveTypeHandle(cx.tags.get(tag_id).th.llth);
        }
293
        case (typeck.ty_box(?t)) {
294 295
            ret T_ptr(T_box(type_of(cx, t)));
        }
296
        case (typeck.ty_vec(?t)) {
297
            ret T_ptr(T_vec(type_of(cx, t)));
298
        }
299
        case (typeck.ty_tup(?elts)) {
300
            let vec[TypeRef] tys = vec();
301 302
            for (@typeck.ty elt in elts) {
                tys += type_of(cx, elt);
303 304 305
            }
            ret T_struct(tys);
        }
306 307 308 309 310 311 312
        case (typeck.ty_rec(?fields)) {
            let vec[TypeRef] tys = vec();
            for (typeck.field f in fields) {
                tys += type_of(cx, f.ty);
            }
            ret T_struct(tys);
        }
313
        case (typeck.ty_fn(?args, ?out)) {
314
            let vec[TypeRef] atys = vec(T_taskptr());
315 316 317 318 319 320
            for (typeck.arg arg in args) {
                let TypeRef t = type_of(cx, arg.ty);
                alt (arg.mode) {
                    case (ast.alias) {
                        t = T_ptr(t);
                    }
321
                    case (_) { /* fall through */  }
322 323 324 325 326
                }
                atys += t;
            }
            ret T_fn(atys, type_of(cx, out));
        }
327
        case (typeck.ty_var(_)) {
328
            // FIXME: implement.
329 330
            log "ty_var in trans.type_of";
            ret T_i8();
331 332 333 334 335
        }
    }
    fail;
}

336 337 338 339 340 341 342 343
fn type_of_arg(@crate_ctxt cx, &typeck.arg arg) -> TypeRef {
    auto ty = type_of(cx, arg.ty);
    if (arg.mode == ast.alias) {
        ty = T_ptr(ty);
    }
    ret ty;
}

344 345 346 347 348 349 350 351 352 353 354 355 356 357 358 359
// 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 {
            auto v = vec(c);
            result += _str.from_bytes(v);
        }
    }
    ret result;
}

360 361 362 363 364 365
// LLVM constant constructors.

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

366
fn C_integral(int i, TypeRef t) -> ValueRef {
367 368 369 370 371 372
    // 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);
    //
373 374 375
    ret llvm.LLVMConstIntOfString(t, _str.buf(istr(i)), 10);
}

376 377 378 379 380
fn C_nil() -> ValueRef {
    // NB: See comment above in T_void().
    ret C_integral(0, T_i1());
}

381 382
fn C_bool(bool b) -> ValueRef {
    if (b) {
383
        ret C_integral(1, T_bool());
384
    } else {
385
        ret C_integral(0, T_bool());
386 387 388
    }
}

389 390
fn C_int(int i) -> ValueRef {
    ret C_integral(i, T_int());
391 392
}

393
fn C_str(@crate_ctxt cx, str s) -> ValueRef {
394
    auto sc = llvm.LLVMConstString(_str.buf(s), _str.byte_len(s), False);
395
    auto g = llvm.LLVMAddGlobal(cx.llmod, val_ty(sc),
396 397
                                _str.buf(cx.names.next("str")));
    llvm.LLVMSetInitializer(g, sc);
398
    llvm.LLVMSetGlobalConstant(g, True);
399
    ret g;
400 401
}

402 403 404 405 406 407 408 409 410 411 412
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));
}

413 414 415 416 417 418
fn C_struct(vec[ValueRef] elts) -> ValueRef {
    ret llvm.LLVMConstStruct(_vec.buf[ValueRef](elts),
                             _vec.len[ValueRef](elts),
                             False);
}

419
fn C_tydesc(TypeRef t) -> ValueRef {
420 421 422 423 424 425 426 427 428 429
    ret C_struct(vec(C_null(T_ptr(T_opaque())),        // first_param
                     llvm.LLVMSizeOf(t),               // size
                     llvm.LLVMAlignOf(t),              // align
                     C_null(T_ptr(T_opaque())),        // copy_glue_off
                     C_null(T_ptr(T_opaque())),        // drop_glue_off
                     C_null(T_ptr(T_opaque())),        // free_glue_off
                     C_null(T_ptr(T_opaque())),        // sever_glue_off
                     C_null(T_ptr(T_opaque())),        // mark_glue_off
                     C_null(T_ptr(T_opaque())),        // obj_drop_glue_off
                     C_null(T_ptr(T_opaque()))));      // is_stateful
430 431
}

432
fn decl_fn(ModuleRef llmod, str name, uint cc, TypeRef llty) -> ValueRef {
433 434
    let ValueRef llfn =
        llvm.LLVMAddFunction(llmod, _str.buf(name), llty);
435
    llvm.LLVMSetFunctionCallConv(llfn, cc);
436 437 438
    ret llfn;
}

439 440
fn decl_cdecl_fn(ModuleRef llmod, str name, TypeRef llty) -> ValueRef {
    ret decl_fn(llmod, name, lib.llvm.LLVMCCallConv, llty);
441 442
}

443 444
fn decl_fastcall_fn(ModuleRef llmod, str name, TypeRef llty) -> ValueRef {
    ret decl_fn(llmod, name, lib.llvm.LLVMFastCallConv, llty);
445 446
}

447
fn decl_glue(ModuleRef llmod, str s) -> ValueRef {
448
    ret decl_cdecl_fn(llmod, s, T_fn(vec(T_taskptr()), T_void()));
449 450
}

451
fn decl_upcall(ModuleRef llmod, uint _n) -> ValueRef {
452 453 454 455
    // 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.
456
    let int n = _n as int;
457
    let str s = abi.upcall_glue_name(n);
458
    let vec[TypeRef] args =
459 460
        vec(T_taskptr(), // taskptr
            T_int())     // callee
461 462
        + _vec.init_elt[TypeRef](T_int(), n as uint);

463
    ret decl_fastcall_fn(llmod, s, T_fn(args, T_int()));
464 465
}

466
fn get_upcall(@crate_ctxt cx, str name, int n_args) -> ValueRef {
467 468 469
    if (cx.upcalls.contains_key(name)) {
        ret cx.upcalls.get(name);
    }
470
    auto inputs = vec(T_taskptr());
471
    inputs += _vec.init_elt[TypeRef](T_int(), n_args as uint);
472
    auto output = T_int();
473
    auto f = decl_cdecl_fn(cx.llmod, name, T_fn(inputs, output));
474 475 476 477
    cx.upcalls.insert(name, f);
    ret f;
}

478
fn trans_upcall(@block_ctxt cx, str name, vec[ValueRef] args) -> result {
479
    let int n = _vec.len[ValueRef](args) as int;
480
    let ValueRef llupcall = get_upcall(cx.fcx.ccx, name, n);
481 482
    llupcall = llvm.LLVMConstPointerCast(llupcall, T_int());

483
    let ValueRef llglue = cx.fcx.ccx.glues.upcall_glues.(n);
484 485 486 487
    let vec[ValueRef] call_args = vec(cx.fcx.lltaskptr, llupcall);
    for (ValueRef a in args) {
        call_args += cx.build.ZExtOrBitCast(a, T_int());
    }
488
    ret res(cx, cx.build.FastCall(llglue, call_args));
489 490
}

491 492 493
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)));
494 495
}

496 497 498 499 500 501 502 503 504 505 506 507 508 509
fn find_scope_cx(@block_ctxt cx) -> @block_ctxt {
    if (cx.is_scope) {
        ret cx;
    }
    alt (cx.parent) {
        case (parent_some(?b)) {
            be find_scope_cx(b);
        }
        case (parent_none) {
            fail;
        }
    }
}

510
fn trans_malloc(@block_ctxt cx, @typeck.ty t) -> result {
511
    auto scope_cx = find_scope_cx(cx);
512 513 514 515 516 517 518
    auto ptr_ty = type_of(cx.fcx.ccx, t);
    auto body_ty = lib.llvm.llvm.LLVMGetElementType(ptr_ty);
    // FIXME: need a table to collect tydesc globals.
    auto tydesc = C_int(0);
    auto sz = cx.build.IntCast(lib.llvm.llvm.LLVMSizeOf(body_ty), T_int());
    auto sub = trans_upcall(cx, "upcall_malloc", vec(sz, tydesc));
    sub.val = sub.bcx.build.IntToPtr(sub.val, ptr_ty);
519
    scope_cx.cleanups += clean(bind drop_ty(_, sub.val, t));
520 521 522 523
    ret sub;
}


524 525 526 527 528 529 530 531 532 533
// Glue and referent count twiddling

fn get_ty_info(@crate_ctxt cx, @typeck.ty ty) -> @ty_info {
    if (!cx.types.contains_key(ty)) {
        make_ty_info(cx, ty);
    }
    ret cx.types.get(ty);
}

fn make_ty_info(@crate_ctxt cx, @typeck.ty ty) {
534 535 536 537 538
    auto tg = make_take_glue;
    auto take_glue = make_generic_glue(cx, ty, "take", tg);
    auto dg = make_drop_glue;
    auto drop_glue = make_generic_glue(cx, ty, "drop", dg);
    cx.types.insert(ty, @rec(take_glue=take_glue, drop_glue=drop_glue));
539 540
}

541 542
fn make_generic_glue(@crate_ctxt cx, @typeck.ty t, str name,
                     val_and_ty_fn helper) -> ValueRef {
543 544 545 546 547 548 549 550
    auto arg_t;
    if (typeck.type_is_structural(t)) {
        arg_t = T_ptr(type_of(cx, t));
    } else {
        arg_t = type_of(cx, t);
    }
    auto llfnty = T_fn(vec(T_taskptr(), arg_t), T_void());

551
    auto fn_name = cx.names.next("_rust_" + name) + "." + typeck.ty_to_str(t);
552 553 554 555 556 557 558 559
    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);

    auto llval = llvm.LLVMGetParam(llfn, 1u);

560
    auto res = helper(bcx, llval, t);
561

562
    res.bcx.build.RetVoid();
563 564 565
    ret llfn;
}

566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599
fn make_take_glue(@block_ctxt cx, ValueRef v, @typeck.ty t) -> result {
    if (typeck.type_is_boxed(t)) {
        ret incr_refcnt_of_boxed(cx, v);

    } else if (typeck.type_is_binding(t)) {
        cx.fcx.ccx.sess.unimpl("binding type in trans.incr_all_refcnts");

    } else if (typeck.type_is_structural(t)) {
        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());
}

fn make_drop_glue(@block_ctxt cx, ValueRef v, @typeck.ty t) -> result {
600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637 638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657
    alt (t.struct) {
        case (typeck.ty_str) {
            ret decr_refcnt_and_if_zero(cx, v,
                                        bind trans_non_gc_free(_, v),
                                        "free string",
                                        T_int(), C_int(0));
        }

        case (typeck.ty_vec(_)) {
            fn hit_zero(@block_ctxt cx, ValueRef v,
                        @typeck.ty t) -> result {
                auto res = iter_sequence(cx, v, t, bind drop_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, t),
                                        "free vector",
                                        T_int(), C_int(0));
        }

        case (typeck.ty_box(?body_ty)) {
            fn hit_zero(@block_ctxt cx, ValueRef v,
                        @typeck.ty body_ty) -> result {
                auto body = cx.build.GEP(v,
                                         vec(C_int(0),
                                             C_int(abi.box_rc_field_body)));

                auto body_val = load_non_structural(cx, body, body_ty);
                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));
        }

        case (_) {
            if (typeck.type_is_structural(t)) {
                ret iter_structural_ty(cx, v, t,
                                       bind drop_ty(_, _, _));

            } else if (typeck.type_is_binding(t)) {
                cx.fcx.ccx.sess.unimpl("binding type in " +
                                       "trans.make_drop_glue_inner");

            } else if (typeck.type_is_scalar(t) ||
                       typeck.type_is_nil(t)) {
                ret res(cx, C_nil());
            }
        }
    }
    cx.fcx.ccx.sess.bug("bad type in trans.make_drop_glue_inner");
    fail;
}

658 659
fn decr_refcnt_and_if_zero(@block_ctxt cx,
                           ValueRef box_ptr,
660
                           fn(@block_ctxt cx) -> result inner,
661
                           str inner_name,
662
                           TypeRef t_else, ValueRef v_else) -> result {
663

664
    auto load_rc_cx = new_sub_block_ctxt(cx, "load rc");
665 666 667 668
    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");

669 670
    auto null_test = cx.build.IsNull(box_ptr);
    cx.build.CondBr(null_test, next_cx.llbb, load_rc_cx.llbb);
671

672 673 674 675 676 677 678 679 680 681

    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);
682 683 684 685

    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);
686 687 688 689
    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);
690

691
    auto phi = next_cx.build.Phi(t_else,
692
                                 vec(v_else, v_else, v_else, inner_res.val),
693
                                 vec(cx.llbb,
694
                                     load_rc_cx.llbb,
695
                                     rc_adj_cx.llbb,
696 697
                                     inner_res.bcx.llbb));

698
    ret res(next_cx, phi);
699 700
}

701 702 703 704 705 706 707 708 709 710 711 712 713
fn type_of_variant(@crate_ctxt cx, &ast.variant v) -> TypeRef {
    let vec[TypeRef] lltys = vec();
    alt (typeck.ann_to_type(v.ann).struct) {
        case (typeck.ty_fn(?args, _)) {
            for (typeck.arg arg in args) {
                lltys += vec(type_of(cx, arg.ty));
            }
        }
        case (_) { fail; }
    }
    ret T_struct(lltys);
}

714 715 716 717 718 719 720 721 722 723 724 725 726
type val_and_ty_fn =
    fn(@block_ctxt cx, ValueRef v, @typeck.ty t) -> result;

// Iterates through the elements of a tup, rec or tag.
fn iter_structural_ty(@block_ctxt cx,
                      ValueRef v,
                      @typeck.ty t,
                      val_and_ty_fn f)
    -> result {
    let result r = res(cx, C_nil());
    alt (t.struct) {
        case (typeck.ty_tup(?args)) {
            let int i = 0;
727
            for (@typeck.ty arg in args) {
728
                auto elt = r.bcx.build.GEP(v, vec(C_int(0), C_int(i)));
729 730 731
                r = f(r.bcx,
                      load_non_structural(r.bcx, elt, arg),
                      arg);
732 733 734
                i += 1;
            }
        }
735 736 737 738
        case (typeck.ty_rec(?fields)) {
            let int i = 0;
            for (typeck.field fld in fields) {
                auto llfld = r.bcx.build.GEP(v, vec(C_int(0), C_int(i)));
739 740 741
                r = f(r.bcx,
                      load_non_structural(r.bcx, llfld, fld.ty),
                      fld.ty);
742 743 744
                i += 1;
            }
        }
745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764 765 766
        case (typeck.ty_tag(?tid)) {
            check (cx.fcx.ccx.tags.contains_key(tid));
            auto info = cx.fcx.ccx.tags.get(tid);
            auto n_variants = _vec.len[tup(ast.def_id,arity)](info.variants);

            // Look up the tag in the typechecked AST.
            check (cx.fcx.ccx.items.contains_key(tid));
            auto tag_item = cx.fcx.ccx.items.get(tid);
            let vec[ast.variant] variants = vec();  // FIXME: typestate bug
            alt (tag_item.node) {
                case (ast.item_tag(_, ?vs, _, _)) {
                    variants = vs;
                }
                case (_) {
                    log "trans: ty_tag doesn't actually refer to a tag";
                    fail;
                }
            }

            auto lldiscrim_ptr = cx.build.GEP(v, vec(C_int(0), C_int(0)));
            auto llunion_ptr = cx.build.GEP(v, vec(C_int(0), C_int(1)));
            auto lldiscrim = cx.build.Load(lldiscrim_ptr);
G
Graydon Hoare 已提交
767

768 769 770 771 772
            auto unr_cx = new_sub_block_ctxt(cx, "tag-iter-unr");
            unr_cx.build.Unreachable();

            auto llswitch = cx.build.Switch(lldiscrim, unr_cx.llbb,
                                            n_variants);
G
Graydon Hoare 已提交
773

774 775 776 777 778 779 780 781 782 783 784 785
            auto next_cx = new_sub_block_ctxt(cx, "tag-iter-next");

            auto i = 0u;
            for (tup(ast.def_id,arity) variant in info.variants) {
                auto variant_cx = new_sub_block_ctxt(cx, "tag-iter-variant-" +
                                                     _uint.to_str(i, 10u));
                llvm.LLVMAddCase(llswitch, C_int(i as int), variant_cx.llbb);

                alt (variant._1) {
                    case (n_ary) {
                        let vec[ValueRef] vals = vec(C_int(0), C_int(1),
                                                     C_int(i as int));
786
                        auto llvar = variant_cx.build.GEP(v, vals);
787 788
                        auto llvarty = type_of_variant(cx.fcx.ccx,
                                                       variants.(i));
789

790
                        auto fn_ty = typeck.ann_to_type(variants.(i).ann);
791 792
                        alt (fn_ty.struct) {
                            case (typeck.ty_fn(?args, _)) {
793 794 795 796
                                auto llvarp = variant_cx.build.
                                    TruncOrBitCast(llunion_ptr,
                                                   T_ptr(llvarty));

797 798
                                auto j = 0u;
                                for (typeck.arg a in args) {
799 800
                                    auto llfldp = variant_cx.build.GEP(llvarp,
                                        vec(C_int(0), C_int(j as int)));
801 802
                                    auto llfld =
                                        load_non_structural(variant_cx,
803
                                                            llfldp, a.ty);
804

805 806 807 808 809 810 811 812 813
                                    auto res = f(variant_cx, llfld, a.ty);
                                    variant_cx = res.bcx;
                                    j += 1u;
                                }
                            }
                            case (_) { fail; }
                        }

                        variant_cx.build.Br(next_cx.llbb);
814 815 816 817 818 819 820 821 822 823 824 825
                    }
                    case (nullary) {
                        // Nothing to do.
                        variant_cx.build.Br(next_cx.llbb);
                    }
                }

                i += 1u;
            }

            ret res(next_cx, C_nil());
        }
826 827 828
        case (_) {
            cx.fcx.ccx.sess.unimpl("type in iter_structural_ty");
        }
829
    }
830
    ret r;
831 832
}

833 834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855
// Iterates through the elements of a vec or str.
fn iter_sequence(@block_ctxt cx,
                 ValueRef v,
                 @typeck.ty ty,
                 val_and_ty_fn f) -> result {

    fn iter_sequence_body(@block_ctxt cx,
                          ValueRef v,
                          @typeck.ty elt_ty,
                          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)));
        auto len = cx.build.Load(lenptr);
        if (trailing_null) {
            len = cx.build.Sub(len, C_int(1));
        }

        auto r = res(cx, C_nil());

856 857
        auto cond_cx = new_scope_block_ctxt(cx, "sequence-iter cond");
        auto body_cx = new_scope_block_ctxt(cx, "sequence-iter body");
858 859 860 861 862 863 864
        auto next_cx = new_sub_block_ctxt(cx, "next");

        auto ix = cond_cx.build.Phi(T_int(), vec(C_int(0)), vec(cx.llbb));
        auto end_test = cond_cx.build.ICmp(lib.llvm.LLVMIntEQ, ix, len);
        cond_cx.build.CondBr(end_test, body_cx.llbb, next_cx.llbb);

        auto elt = body_cx.build.GEP(p0, vec(ix));
865 866 867
        auto body_res = f(body_cx,
                          load_non_structural(body_cx, elt, elt_ty),
                          elt_ty);
868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883
        auto next_ix = body_res.bcx.build.Add(ix, C_int(1));
        cond_cx.build.AddIncomingToPhi(ix, vec(next_ix),
                                       vec(body_res.bcx.llbb));

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

    alt (ty.struct) {
        case (typeck.ty_vec(?et)) {
            ret iter_sequence_body(cx, v, et, f, false);
        }
        case (typeck.ty_str) {
            auto et = typeck.plain_ty(typeck.ty_machine(common.ty_u8));
            ret iter_sequence_body(cx, v, et, f, false);
        }
884
        case (_) { fail; }
885
    }
886 887 888 889 890 891 892
    cx.fcx.ccx.sess.bug("bad type in trans.iter_sequence");
    fail;
}

fn incr_all_refcnts(@block_ctxt cx,
                    ValueRef v,
                    @typeck.ty t) -> result {
893 894
    cx.build.FastCall(get_ty_info(cx.fcx.ccx, t).take_glue,
                      vec(cx.fcx.lltaskptr, v));
895
    ret res(cx, C_nil());
896 897
}

898 899 900
fn drop_slot(@block_ctxt cx,
             ValueRef slot,
             @typeck.ty t) -> result {
901
    be drop_ty(cx, load_non_structural(cx, slot, t), t);
902 903
}

904 905 906
fn drop_ty(@block_ctxt cx,
           ValueRef v,
           @typeck.ty t) -> result {
907 908 909
    cx.build.FastCall(get_ty_info(cx.fcx.ccx, t).drop_glue,
                      vec(cx.fcx.lltaskptr, v));
    ret res(cx, C_nil());
910 911 912 913 914 915
}

fn build_memcpy(@block_ctxt cx,
                ValueRef dst,
                ValueRef src,
                TypeRef llty) -> result {
916 917 918
    // FIXME: switch to the 64-bit variant when on such a platform.
    check (cx.fcx.ccx.intrinsics.contains_key("llvm.memcpy.p0i8.p0i8.i32"));
    auto memcpy = cx.fcx.ccx.intrinsics.get("llvm.memcpy.p0i8.p0i8.i32");
919 920
    auto src_ptr = cx.build.PointerCast(src, T_ptr(T_i8()));
    auto dst_ptr = cx.build.PointerCast(dst, T_ptr(T_i8()));
G
Graydon Hoare 已提交
921 922
    auto size = cx.build.IntCast(lib.llvm.llvm.LLVMSizeOf(llty),
                                 T_i32());
923 924 925 926 927 928
    auto align = cx.build.IntCast(C_int(1), T_i32());

    // FIXME: align seems like it should be
    //   lib.llvm.llvm.LLVMAlignOf(llty);
    // but this makes it upset because it's not a constant.

929 930 931 932 933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949
    auto volatile = C_integral(0, T_i1());
    ret res(cx, cx.build.Call(memcpy,
                              vec(dst_ptr, src_ptr,
                                  size, align, volatile)));
}

fn copy_ty(@block_ctxt cx,
           bool is_init,
           ValueRef dst,
           ValueRef src,
           @typeck.ty t) -> result {
    if (typeck.type_is_scalar(t)) {
        ret res(cx, cx.build.Store(src, dst));

    } else if (typeck.type_is_nil(t)) {
        ret res(cx, C_nil());

    } else if (typeck.type_is_binding(t)) {
        cx.fcx.ccx.sess.unimpl("binding type in trans.copy_ty");

    } else if (typeck.type_is_boxed(t)) {
950
        auto r = incr_all_refcnts(cx, src, t);
951
        if (! is_init) {
952
            r = drop_ty(r.bcx, r.bcx.build.Load(dst), t);
953 954 955 956 957 958 959 960
        }
        ret res(r.bcx, r.bcx.build.Store(src, dst));

    } else if (typeck.type_is_structural(t)) {
        auto r = incr_all_refcnts(cx, src, t);
        if (! is_init) {
            r = drop_ty(r.bcx, dst, t);
        }
961 962 963 964
        // In this one surprising case, we do a load/store on
        // structure types. This results in a memcpy. Usually
        // we talk about structures by pointers in this file.
        ret res(r.bcx, r.bcx.build.Store(r.bcx.build.Load(src), dst));
965 966 967 968
    }

    cx.fcx.ccx.sess.bug("unexpected type in trans.copy_ty: " +
                        typeck.ty_to_str(t));
969 970 971
    fail;
}

972
impure fn trans_lit(@block_ctxt cx, &ast.lit lit, &ast.ann ann) -> result {
973
    alt (lit.node) {
974
        case (ast.lit_int(?i)) {
975
            ret res(cx, C_int(i));
976 977
        }
        case (ast.lit_uint(?u)) {
978
            ret res(cx, C_int(u as int));
979
        }
980 981 982 983 984 985 986 987 988 989 990 991 992 993 994 995 996 997 998 999 1000
        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) {
                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(); }
                case (_) {
                    cx.fcx.ccx.sess.bug("bad mach int literal type");
                }
            }
            ret res(cx, C_integral(i, t));
        }
1001
        case (ast.lit_char(?c)) {
1002
            ret res(cx, C_integral(c as int, T_char()));
1003 1004
        }
        case (ast.lit_bool(?b)) {
1005 1006 1007 1008
            ret res(cx, C_bool(b));
        }
        case (ast.lit_nil) {
            ret res(cx, C_nil());
1009 1010 1011
        }
        case (ast.lit_str(?s)) {
            auto len = (_str.byte_len(s) as int) + 1;
1012
            auto sub = trans_upcall(cx, "upcall_new_str",
1013
                                    vec(p2i(C_str(cx.fcx.ccx, s)),
1014 1015
                                        C_int(len)));
            sub.val = sub.bcx.build.IntToPtr(sub.val,
1016
                                             T_ptr(T_str()));
1017
            auto t = node_ann_type(cx.fcx.ccx, ann);
1018
            find_scope_cx(cx).cleanups +=
1019
                clean(bind drop_ty(_, sub.val, t));
1020
            ret sub;
1021 1022 1023 1024
        }
    }
}

1025 1026 1027 1028 1029 1030 1031 1032 1033 1034
fn target_type(@crate_ctxt cx, @typeck.ty t) -> @typeck.ty {
    alt (t.struct) {
        case (typeck.ty_int) {
            auto tm = typeck.ty_machine(cx.sess.get_targ_cfg().int_type);
            ret @rec(struct=tm with *t);
        }
        case (typeck.ty_uint) {
            auto tm = typeck.ty_machine(cx.sess.get_targ_cfg().uint_type);
            ret @rec(struct=tm with *t);
        }
1035
        case (_) { /* fall through */ }
1036 1037 1038 1039 1040
    }
    ret t;
}

fn node_ann_type(@crate_ctxt cx, &ast.ann a) -> @typeck.ty {
1041 1042
    alt (a) {
        case (ast.ann_none) {
1043
            cx.sess.bug("missing type annotation");
1044 1045
        }
        case (ast.ann_type(?t)) {
1046
            ret target_type(cx, t);
1047 1048 1049 1050
        }
    }
}

1051 1052 1053 1054
fn node_type(@crate_ctxt cx, &ast.ann a) -> TypeRef {
    ret type_of(cx, node_ann_type(cx, a));
}

1055
impure fn trans_unary(@block_ctxt cx, ast.unop op,
1056
                      @ast.expr e, &ast.ann a) -> result {
1057 1058 1059

    auto sub = trans_expr(cx, e);

1060 1061
    alt (op) {
        case (ast.bitnot) {
1062 1063
            sub.val = cx.build.Not(sub.val);
            ret sub;
1064 1065
        }
        case (ast.not) {
1066 1067
            sub.val = cx.build.Not(sub.val);
            ret sub;
1068 1069 1070
        }
        case (ast.neg) {
            // FIXME: switch by signedness.
1071 1072
            sub.val = cx.build.Neg(sub.val);
            ret sub;
1073
        }
1074
        case (ast.box) {
1075 1076 1077 1078 1079
            auto e_ty = typeck.expr_ty(e);
            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,
1080 1081
                                        vec(C_int(0),
                                            C_int(abi.box_rc_field_refcnt)));
1082 1083 1084 1085 1086 1087
            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);
            sub = copy_ty(sub.bcx, true, body, e_val, e_ty);
            ret res(sub.bcx, box);
1088
        }
1089 1090 1091 1092 1093 1094 1095 1096 1097 1098
        case (ast.deref) {
            sub.val = sub.bcx.build.GEP(sub.val,
                                        vec(C_int(0),
                                            C_int(abi.box_rc_field_body)));
            auto e_ty = node_ann_type(sub.bcx.fcx.ccx, a);
            if (typeck.type_is_scalar(e_ty) ||
                typeck.type_is_nil(e_ty)) {
                sub.val = sub.bcx.build.Load(sub.val);
            }
            ret sub;
1099
        }
1100 1101 1102 1103
    }
    fail;
}

1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122 1123 1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139
fn trans_eager_binop(@block_ctxt cx, ast.binop op,
                     ValueRef lhs, ValueRef rhs) -> ValueRef {

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

        // FIXME: switch by signedness.
        case (ast.mul) { ret cx.build.Mul(lhs, rhs); }
        case (ast.div) { ret cx.build.SDiv(lhs, rhs); }
        case (ast.rem) { ret cx.build.SRem(lhs, rhs); }

        case (ast.bitor) { ret cx.build.Or(lhs, rhs); }
        case (ast.bitand) { ret cx.build.And(lhs, rhs); }
        case (ast.bitxor) { ret cx.build.Xor(lhs, rhs); }
        case (ast.lsl) { ret cx.build.Shl(lhs, rhs); }
        case (ast.lsr) { ret cx.build.LShr(lhs, rhs); }
        case (ast.asr) { ret cx.build.AShr(lhs, rhs); }
        case (_) {
            auto cmp = lib.llvm.LLVMIntEQ;
            alt (op) {
                case (ast.eq) { cmp = lib.llvm.LLVMIntEQ; }
                case (ast.ne) { cmp = lib.llvm.LLVMIntNE; }

                // FIXME: switch by signedness.
                case (ast.lt) { cmp = lib.llvm.LLVMIntSLT; }
                case (ast.le) { cmp = lib.llvm.LLVMIntSLE; }
                case (ast.ge) { cmp = lib.llvm.LLVMIntSGE; }
                case (ast.gt) { cmp = lib.llvm.LLVMIntSGT; }
            }
            ret cx.build.ICmp(cmp, lhs, rhs);
        }
    }
    fail;
}

1140
impure fn trans_binary(@block_ctxt cx, ast.binop op,
1141
                       @ast.expr a, @ast.expr b) -> result {
1142

1143 1144 1145 1146 1147 1148 1149
    // First couple cases are lazy:

    alt (op) {
        case (ast.and) {
            // Lazy-eval and
            auto lhs_res = trans_expr(cx, a);

1150
            auto rhs_cx = new_scope_block_ctxt(cx, "rhs");
1151 1152
            auto rhs_res = trans_expr(rhs_cx, b);

1153
            auto lhs_false_cx = new_scope_block_ctxt(cx, "lhs false");
1154
            auto lhs_false_res = res(lhs_false_cx, C_bool(false));
1155 1156 1157

            lhs_res.bcx.build.CondBr(lhs_res.val,
                                     rhs_cx.llbb,
1158 1159 1160 1161
                                     lhs_false_cx.llbb);

            ret join_results(cx, T_bool(),
                             vec(lhs_false_res, rhs_res));
1162 1163 1164 1165 1166 1167
        }

        case (ast.or) {
            // Lazy-eval or
            auto lhs_res = trans_expr(cx, a);

1168
            auto rhs_cx = new_scope_block_ctxt(cx, "rhs");
1169 1170
            auto rhs_res = trans_expr(rhs_cx, b);

1171
            auto lhs_true_cx = new_scope_block_ctxt(cx, "lhs true");
1172
            auto lhs_true_res = res(lhs_true_cx, C_bool(true));
1173 1174

            lhs_res.bcx.build.CondBr(lhs_res.val,
1175
                                     lhs_true_cx.llbb,
1176
                                     rhs_cx.llbb);
1177 1178 1179

            ret join_results(cx, T_bool(),
                             vec(lhs_true_res, rhs_res));
1180
        }
1181 1182

        case (_) {
1183 1184 1185 1186 1187
            // Remaining cases are eager:
            auto lhs = trans_expr(cx, a);
            auto sub = trans_expr(lhs.bcx, b);
            ret res(sub.bcx, trans_eager_binop(sub.bcx, op,
                                               lhs.val, sub.val));
1188
        }
1189 1190 1191 1192
    }
    fail;
}

1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223
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);
        }
1224 1225

        case (_) { /* fall through */ }
1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236
    }

    // 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);
}

1237
impure fn trans_if(@block_ctxt cx, @ast.expr cond,
1238
                   &ast.block thn, &option.t[ast.block] els) -> result {
1239 1240 1241

    auto cond_res = trans_expr(cx, cond);

1242
    auto then_cx = new_scope_block_ctxt(cx, "then");
1243 1244
    auto then_res = trans_block(then_cx, thn);

1245
    auto else_cx = new_scope_block_ctxt(cx, "else");
1246
    auto else_res = res(else_cx, C_nil());
1247 1248 1249

    alt (els) {
        case (some[ast.block](?eblk)) {
1250
            else_res = trans_block(else_cx, eblk);
1251
        }
1252
        case (_) { /* fall through */ }
1253 1254
    }

1255
    cond_res.bcx.build.CondBr(cond_res.val,
1256 1257
                              then_cx.llbb,
                              else_cx.llbb);
1258 1259 1260 1261

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

1264
impure fn trans_while(@block_ctxt cx, @ast.expr cond,
1265 1266
                      &ast.block body) -> result {

1267 1268
    auto cond_cx = new_scope_block_ctxt(cx, "while cond");
    auto body_cx = new_scope_block_ctxt(cx, "while loop body");
1269
    auto next_cx = new_sub_block_ctxt(cx, "next");
1270 1271

    auto body_res = trans_block(body_cx, body);
1272 1273 1274 1275 1276 1277 1278 1279
    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);
1280 1281 1282 1283
    ret res(next_cx, C_nil());
}

impure fn trans_do_while(@block_ctxt cx, &ast.block body,
1284
                         @ast.expr cond) -> result {
1285

1286
    auto body_cx = new_scope_block_ctxt(cx, "do-while loop body");
1287
    auto next_cx = new_sub_block_ctxt(cx, "next");
1288 1289

    auto body_res = trans_block(body_cx, body);
1290 1291 1292 1293 1294 1295
    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);
1296 1297 1298
    ret res(next_cx, body_res.val);
}

1299 1300 1301
// 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).
1302

1303 1304 1305 1306 1307 1308 1309 1310
fn trans_name(@block_ctxt cx, &ast.name n, &option.t[ast.def] dopt)
    -> tup(result, bool) {
    alt (dopt) {
        case (some[ast.def](?def)) {
            alt (def) {
                case (ast.def_arg(?did)) {
                    check (cx.fcx.llargs.contains_key(did));
                    ret tup(res(cx, cx.fcx.llargs.get(did)),
1311
                            true);
1312 1313 1314 1315 1316
                }
                case (ast.def_local(?did)) {
                    check (cx.fcx.lllocals.contains_key(did));
                    ret tup(res(cx, cx.fcx.lllocals.get(did)),
                            true);
G
Graydon Hoare 已提交
1317
                }
1318
                case (ast.def_fn(?did)) {
1319 1320
                    check (cx.fcx.ccx.item_ids.contains_key(did));
                    ret tup(res(cx, cx.fcx.ccx.item_ids.get(did)),
1321 1322
                            false);
                }
1323 1324
                case (ast.def_variant(?tid, ?vid)) {
                    check (cx.fcx.ccx.tags.contains_key(tid));
1325 1326 1327
                    check (cx.fcx.ccx.item_ids.contains_key(vid));
                    ret tup(res(cx, cx.fcx.ccx.item_ids.get(vid)),
                            false);
1328
                }
1329 1330
                case (_) {
                    cx.fcx.ccx.sess.unimpl("def variant in trans");
G
Graydon Hoare 已提交
1331 1332 1333
                }
            }
        }
1334 1335 1336 1337 1338 1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351
        case (none[ast.def]) {
            cx.fcx.ccx.sess.err("unresolved expr_name in trans");
        }
    }
    fail;
}

fn trans_field(@block_ctxt cx, &ast.span sp, @ast.expr base,
               &ast.ident field, &ast.ann ann) -> tup(result, bool) {
    auto lv = trans_lval(cx, base);
    auto r = lv._0;
    auto ty = typeck.expr_ty(base);
    alt (ty.struct) {
        case (typeck.ty_tup(?fields)) {
            let uint ix = typeck.field_num(cx.fcx.ccx.sess, sp, field);
            auto v = r.bcx.build.GEP(r.val, vec(C_int(0), C_int(ix as int)));
            ret tup(res(r.bcx, v), lv._1);
        }
1352 1353 1354 1355 1356 1357
        case (typeck.ty_rec(?fields)) {
            let uint ix = typeck.field_idx(cx.fcx.ccx.sess, sp,
                                           field, fields);
            auto v = r.bcx.build.GEP(r.val, vec(C_int(0), C_int(ix as int)));
            ret tup(res(r.bcx, v), lv._1);
        }
1358
        case (_) { cx.fcx.ccx.sess.unimpl("field variant in trans_field"); }
1359 1360 1361 1362
    }
    fail;
}

1363 1364 1365
fn trans_index(@block_ctxt cx, &ast.span sp, @ast.expr base,
               @ast.expr idx, &ast.ann ann) -> tup(result, bool) {

G
Graydon Hoare 已提交
1366 1367 1368
    auto lv = trans_expr(cx, base);
    auto ix = trans_expr(lv.bcx, idx);
    auto v = lv.val;
1369 1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387 1388 1389 1390 1391 1392 1393

    auto llunit_ty = node_type(cx.fcx.ccx, ann);
    auto unit_sz = ix.bcx.build.IntCast(lib.llvm.llvm.LLVMSizeOf(llunit_ty),
                                      T_int());
    auto scaled_ix = ix.bcx.build.Mul(ix.val, unit_sz);

    auto lim = ix.bcx.build.GEP(v, vec(C_int(0), C_int(abi.vec_elt_fill)));
    lim = ix.bcx.build.Load(lim);
    auto bounds_check = ix.bcx.build.ICmp(lib.llvm.LLVMIntULT,
                                          scaled_ix, lim);

    auto fail_cx = new_sub_block_ctxt(ix.bcx, "fail");
    auto next_cx = new_sub_block_ctxt(ix.bcx, "next");
    ix.bcx.build.CondBr(bounds_check, next_cx.llbb, fail_cx.llbb);

    // fail: bad bounds check.
    auto V_expr_str = p2i(C_str(cx.fcx.ccx, "out-of-bounds access"));
    auto V_filename = p2i(C_str(cx.fcx.ccx, sp.filename));
    auto V_line = sp.lo.line as int;
    auto args = vec(V_expr_str, V_filename, C_int(V_line));
    auto fail_res = trans_upcall(fail_cx, "upcall_fail", args);
    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));
G
Graydon Hoare 已提交
1394
    ret tup(res(next_cx, elt), true);
1395 1396
}

1397 1398 1399 1400
// 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).

1401 1402 1403 1404 1405 1406 1407 1408
fn trans_lval(@block_ctxt cx, @ast.expr e) -> tup(result, bool) {
    alt (e.node) {
        case (ast.expr_name(?n, ?dopt, _)) {
            ret trans_name(cx, n, dopt);
        }
        case (ast.expr_field(?base, ?ident, ?ann)) {
            ret trans_field(cx, e.span, base, ident, ann);
        }
1409 1410 1411
        case (ast.expr_index(?base, ?idx, ?ann)) {
            ret trans_index(cx, e.span, base, idx, ann);
        }
1412
        case (_) { cx.fcx.ccx.sess.unimpl("expr variant in trans_lval"); }
G
Graydon Hoare 已提交
1413 1414 1415 1416
    }
    fail;
}

1417
impure fn trans_cast(@block_ctxt cx, @ast.expr e, &ast.ann ann) -> result {
1418 1419 1420 1421 1422 1423 1424 1425 1426 1427 1428 1429 1430 1431 1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447
    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);
    if (!typeck.type_is_fp(t)) {
        if (llvm.LLVMGetIntTypeWidth(lldsttype) >
            llvm.LLVMGetIntTypeWidth(llsrctype)) {
            if (typeck.type_is_signed(t)) {
                // 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;
}

1448

1449
impure fn trans_args(@block_ctxt cx, &vec[@ast.expr] es, @typeck.ty fn_ty)
1450 1451 1452 1453
    -> tup(@block_ctxt, vec[ValueRef]) {
    let vec[ValueRef] vs = vec(cx.fcx.lltaskptr);
    let @block_ctxt bcx = cx;

1454 1455 1456 1457 1458 1459 1460
    let vec[typeck.arg] args = vec();   // FIXME: typestate bug
    alt (fn_ty.struct) {
        case (typeck.ty_fn(?a, _)) { args = a; }
        case (_) { fail; }
    }

    auto i = 0u;
1461
    for (@ast.expr e in es) {
1462 1463 1464
        auto mode = args.(i).mode;

        auto re;
1465
        if (typeck.type_is_structural(typeck.expr_ty(e))) {
1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492
            re = trans_expr(bcx, e);
            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.
                re.val = re.bcx.build.Load(re.val);
            }
        } else {
            if (mode == ast.alias) {
                let tup(result, bool /* is a pointer? */) pair;
                if (typeck.is_lval(e)) {
                    pair = trans_lval(bcx, e);
                } else {
                    pair = tup(trans_expr(bcx, e), false);
                }

                if (!pair._1) {
                    // Have to synthesize a pointer here...
                    auto llty = val_ty(pair._0.val);
                    auto llptr = pair._0.bcx.build.Alloca(llty);
                    pair._0.bcx.build.Store(pair._0.val, llptr);
                    re = res(pair._0.bcx, llptr);
                } else {
                    re = pair._0;
                }
            } else {
                re = trans_expr(bcx, e);
            }
1493
        }
1494 1495 1496 1497 1498

        vs += re.val;
        bcx = re.bcx;

        i += 1u;
1499 1500 1501 1502 1503
    }

    ret tup(bcx, vs);
}

1504
impure fn trans_call(@block_ctxt cx, @ast.expr f,
G
Graydon Hoare 已提交
1505
                     vec[@ast.expr] args, &ast.ann ann) -> result {
1506 1507
    auto f_res = trans_lval(cx, f);
    check (! f_res._1);
1508
    auto fn_ty = typeck.expr_ty(f);
G
Graydon Hoare 已提交
1509
    auto ret_ty = typeck.ann_to_type(ann);
1510
    auto args_res = trans_args(f_res._0.bcx, args, fn_ty);
G
Graydon Hoare 已提交
1511 1512
    auto retval = args_res._0.build.FastCall(f_res._0.val, args_res._1);

G
Graydon Hoare 已提交
1513 1514 1515 1516 1517 1518 1519 1520 1521
    // Structured returns come back as first-class values. This is nice for
    // LLVM but wrong for us; we treat structured values by pointer in
    // most of our code here. So spill it to an alloca.
    if (typeck.type_is_structural(ret_ty)) {
        auto local = args_res._0.build.Alloca(type_of(cx.fcx.ccx, ret_ty));
        args_res._0.build.Store(retval, local);
        retval = local;
    }

G
Graydon Hoare 已提交
1522 1523 1524 1525 1526 1527
    // 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));

    ret res(args_res._0, retval);
1528 1529
}

1530
impure fn trans_tup(@block_ctxt cx, vec[ast.elt] elts,
G
Graydon Hoare 已提交
1531
                    &ast.ann ann) -> result {
1532 1533 1534 1535
    auto ty = node_ann_type(cx.fcx.ccx, ann);
    auto llty = type_of(cx.fcx.ccx, ty);
    auto tup_val = cx.build.Alloca(llty);
    find_scope_cx(cx).cleanups += clean(bind drop_ty(_, tup_val, ty));
G
Graydon Hoare 已提交
1536 1537
    let int i = 0;
    auto r = res(cx, C_nil());
1538 1539 1540
    for (ast.elt e in elts) {
        auto t = typeck.expr_ty(e.expr);
        auto src_res = trans_expr(r.bcx, e.expr);
G
Graydon Hoare 已提交
1541 1542 1543 1544 1545 1546 1547
        auto dst_elt = r.bcx.build.GEP(tup_val, vec(C_int(0), C_int(i)));
        r = copy_ty(src_res.bcx, true, dst_elt, src_res.val, t);
        i += 1;
    }
    ret res(r.bcx, tup_val);
}

G
Graydon Hoare 已提交
1548 1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579 1580 1581 1582 1583 1584 1585
impure fn trans_vec(@block_ctxt cx, vec[@ast.expr] args,
                    &ast.ann ann) -> result {
    auto ty = node_ann_type(cx.fcx.ccx, ann);
    auto unit_ty = ty;
    alt (ty.struct) {
        case (typeck.ty_vec(?t)) {
            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);
    auto unit_sz = llvm.LLVMConstIntCast(llvm.LLVMSizeOf(llunit_ty),
                                         T_int(), False);
    auto data_sz = llvm.LLVMConstMul(C_int(_vec.len[@ast.expr](args) as int),
                                     unit_sz);

    // FIXME: pass tydesc properly.
    auto sub = trans_upcall(cx, "upcall_new_vec", vec(data_sz, C_int(0)));

    auto llty = type_of(cx.fcx.ccx, ty);
    auto vec_val = sub.bcx.build.IntToPtr(sub.val, llty);
    find_scope_cx(cx).cleanups += clean(bind drop_ty(_, vec_val, ty));

    auto body = sub.bcx.build.GEP(vec_val, vec(C_int(0),
                                               C_int(abi.vec_elt_data)));
    let int i = 0;
    for (@ast.expr e in args) {
        auto src_res = trans_expr(sub.bcx, e);
        auto dst_elt = sub.bcx.build.GEP(body, vec(C_int(0), C_int(i)));
        sub = copy_ty(src_res.bcx, true, dst_elt, src_res.val, unit_ty);
        i += 1;
    }
    ret res(sub.bcx, vec_val);
}

1586
impure fn trans_rec(@block_ctxt cx, vec[ast.field] fields,
1587
                    &ast.ann ann) -> result {
1588 1589 1590 1591
    auto ty = node_ann_type(cx.fcx.ccx, ann);
    auto llty = type_of(cx.fcx.ccx, ty);
    auto rec_val = cx.build.Alloca(llty);
    find_scope_cx(cx).cleanups += clean(bind drop_ty(_, rec_val, ty));
1592 1593
    let int i = 0;
    auto r = res(cx, C_nil());
1594 1595 1596 1597
    for (ast.field f in fields) {
        auto t = typeck.expr_ty(f.expr);
        auto src_res = trans_expr(r.bcx, f.expr);
        auto dst_elt = r.bcx.build.GEP(rec_val, vec(C_int(0), C_int(i)));
1598 1599 1600 1601
        // FIXME: calculate copy init-ness in typestate.
        r = copy_ty(src_res.bcx, true, dst_elt, src_res.val, t);
        i += 1;
    }
1602
    ret res(r.bcx, rec_val);
1603 1604
}

G
Graydon Hoare 已提交
1605

G
Graydon Hoare 已提交
1606

1607
impure fn trans_expr(@block_ctxt cx, @ast.expr e) -> result {
1608
    alt (e.node) {
1609 1610
        case (ast.expr_lit(?lit, ?ann)) {
            ret trans_lit(cx, *lit, ann);
1611 1612
        }

1613
        case (ast.expr_unary(?op, ?x, ?ann)) {
1614
            ret trans_unary(cx, op, x, ann);
1615 1616
        }

P
Patrick Walton 已提交
1617
        case (ast.expr_binary(?op, ?x, ?y, _)) {
1618
            ret trans_binary(cx, op, x, y);
1619
        }
1620

P
Patrick Walton 已提交
1621
        case (ast.expr_if(?cond, ?thn, ?els, _)) {
1622
            ret trans_if(cx, cond, thn, els);
1623 1624
        }

1625
        case (ast.expr_while(?cond, ?body, _)) {
1626
            ret trans_while(cx, cond, body);
1627 1628
        }

1629
        case (ast.expr_do_while(?body, ?cond, _)) {
1630
            ret trans_do_while(cx, body, cond);
1631 1632
        }

P
Patrick Walton 已提交
1633
        case (ast.expr_block(?blk, _)) {
1634
            auto sub_cx = new_scope_block_ctxt(cx, "block-expr body");
1635
            auto next_cx = new_sub_block_ctxt(cx, "next");
1636 1637 1638 1639 1640 1641 1642
            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);
        }
1643

1644
        case (ast.expr_assign(?dst, ?src, ?ann)) {
1645
            auto lhs_res = trans_lval(cx, dst);
1646
            check (lhs_res._1);
1647
            auto rhs_res = trans_expr(lhs_res._0.bcx, src);
1648
            auto t = node_ann_type(cx.fcx.ccx, ann);
G
Graydon Hoare 已提交
1649
            // FIXME: calculate copy init-ness in typestate.
1650
            ret copy_ty(rhs_res.bcx, false, lhs_res._0.val, rhs_res.val, t);
1651
        }
G
Graydon Hoare 已提交
1652

1653 1654 1655 1656 1657 1658 1659 1660 1661 1662 1663 1664
        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);
            check (lhs_res._1);
            auto lhs_val = load_non_structural(lhs_res._0.bcx,
                                               lhs_res._0.val, t);
            auto rhs_res = trans_expr(lhs_res._0.bcx, src);
            auto v = trans_eager_binop(rhs_res.bcx, op, lhs_val, rhs_res.val);
            // FIXME: calculate copy init-ness in typestate.
            ret copy_ty(rhs_res.bcx, false, lhs_res._0.val, v, t);
        }

G
Graydon Hoare 已提交
1665 1666
        case (ast.expr_call(?f, ?args, ?ann)) {
            ret trans_call(cx, f, args, ann);
1667 1668
        }

1669
        case (ast.expr_cast(?e, _, ?ann)) {
1670
            ret trans_cast(cx, e, ann);
1671
        }
G
Graydon Hoare 已提交
1672

G
Graydon Hoare 已提交
1673 1674 1675 1676
        case (ast.expr_vec(?args, ?ann)) {
            ret trans_vec(cx, args, ann);
        }

G
Graydon Hoare 已提交
1677 1678 1679
        case (ast.expr_tup(?args, ?ann)) {
            ret trans_tup(cx, args, ann);
        }
G
Graydon Hoare 已提交
1680

1681 1682 1683 1684
        case (ast.expr_rec(?args, ?ann)) {
            ret trans_rec(cx, args, ann);
        }

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

1688 1689 1690
        case (_) {
            auto t = typeck.expr_ty(e);
            auto sub = trans_lval(cx, e);
1691 1692
            ret res(sub._0.bcx,
                    load_non_structural(sub._0.bcx, sub._0.val, t));
1693
        }
1694
    }
1695
    cx.fcx.ccx.sess.unimpl("expr variant in trans_expr");
1696 1697 1698
    fail;
}

1699 1700 1701 1702 1703 1704 1705 1706 1707 1708 1709 1710 1711 1712
// We pass structural values around the compiler "by pointer" and
// non-structural values "by value". This function selects whether
// to load a pointer or pass it.

fn load_non_structural(@block_ctxt cx,
                       ValueRef v,
                       @typeck.ty t) -> ValueRef {
    if (typeck.type_is_structural(t)) {
        ret v;
    } else {
        ret cx.build.Load(v);
    }
}

1713
impure fn trans_log(@block_ctxt cx, @ast.expr e) -> result {
1714

1715 1716 1717 1718 1719 1720 1721 1722
    auto sub = trans_expr(cx, e);
    auto e_ty = typeck.expr_ty(e);
    alt (e_ty.struct) {
        case (typeck.ty_str) {
            auto v = sub.bcx.build.PtrToInt(sub.val, T_int());
            ret trans_upcall(sub.bcx,
                             "upcall_log_str",
                             vec(v));
1723 1724
        }
        case (_) {
1725 1726 1727
            ret trans_upcall(sub.bcx,
                             "upcall_log_int",
                             vec(sub.val));
1728 1729
        }
    }
1730
    fail;
1731 1732
}

1733
impure fn trans_check_expr(@block_ctxt cx, @ast.expr e) -> result {
1734 1735 1736
    auto cond_res = trans_expr(cx, e);

    // FIXME: need pretty-printer.
1737 1738
    auto V_expr_str = p2i(C_str(cx.fcx.ccx, "<expr>"));
    auto V_filename = p2i(C_str(cx.fcx.ccx, e.span.filename));
1739 1740 1741
    auto V_line = e.span.lo.line as int;
    auto args = vec(V_expr_str, V_filename, C_int(V_line));

1742
    auto fail_cx = new_sub_block_ctxt(cx, "fail");
1743 1744
    auto fail_res = trans_upcall(fail_cx, "upcall_fail", args);

1745
    auto next_cx = new_sub_block_ctxt(cx, "next");
1746 1747 1748 1749 1750 1751 1752
    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());
}

1753
impure fn trans_ret(@block_ctxt cx, &option.t[@ast.expr] e) -> result {
1754 1755 1756
    auto r = res(cx, C_nil());
    alt (e) {
        case (some[@ast.expr](?x)) {
G
Graydon Hoare 已提交
1757
            auto t = typeck.expr_ty(x);
1758
            r = trans_expr(cx, x);
1759 1760 1761 1762 1763

            // A return is an implicit copy into a newborn anonymous
            // 'return value' in the caller frame.
            r.bcx = incr_all_refcnts(r.bcx, r.val, t).bcx;

G
Graydon Hoare 已提交
1764 1765 1766 1767 1768 1769 1770
            if (typeck.type_is_structural(t)) {
                // We usually treat structurals by-pointer; in particular,
                // trans_expr will have given us a structure pointer. But in
                // this case we're about to return. LLVM wants a first-class
                // value here (which makes sense; the frame is going away!)
                r.val = r.bcx.build.Load(r.val);
            }
1771
        }
1772
        case (_) { /* fall through */  }
1773 1774
    }

1775 1776
    // Run all cleanups and back out.
    let bool more_cleanups = true;
1777
    auto cleanup_cx = cx;
1778
    while (more_cleanups) {
1779 1780
        r.bcx = trans_block_cleanups(r.bcx, cleanup_cx);
        alt (cleanup_cx.parent) {
1781
            case (parent_some(?b)) {
1782
                cleanup_cx = b;
1783 1784 1785 1786 1787 1788 1789
            }
            case (parent_none) {
                more_cleanups = false;
            }
        }
    }

1790
    alt (e) {
G
Graydon Hoare 已提交
1791
        case (some[@ast.expr](_)) {
1792 1793 1794
            r.val = r.bcx.build.Ret(r.val);
            ret r;
        }
1795
        case (_) { /* fall through */  }
1796 1797 1798 1799 1800
    }

    // FIXME: until LLVM has a unit type, we are moving around
    // C_nil values rather than their void type.
    r.val = r.bcx.build.Ret(C_nil());
1801 1802 1803
    ret r;
}

1804
impure fn trans_stmt(@block_ctxt cx, &ast.stmt s) -> result {
1805
    auto sub = res(cx, C_nil());
1806
    alt (s.node) {
1807
        case (ast.stmt_log(?a)) {
1808
            sub.bcx = trans_log(cx, a).bcx;
1809 1810
        }

1811
        case (ast.stmt_check_expr(?a)) {
1812
            sub.bcx = trans_check_expr(cx, a).bcx;
1813 1814
        }

1815 1816 1817 1818
        case (ast.stmt_ret(?e)) {
            sub.bcx = trans_ret(cx, e).bcx;
        }

1819
        case (ast.stmt_expr(?e)) {
1820
            sub.bcx = trans_expr(cx, e).bcx;
1821
        }
1822

1823 1824 1825
        case (ast.stmt_decl(?d)) {
            alt (d.node) {
                case (ast.decl_local(?local)) {
1826 1827 1828 1829 1830

                    // 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);
1831
                    find_scope_cx(cx).cleanups +=
1832 1833
                        clean(bind drop_slot(_, llptr, ty));

1834 1835
                    alt (local.init) {
                        case (some[@ast.expr](?e)) {
1836
                            sub = trans_expr(cx, e);
1837
                            sub = copy_ty(sub.bcx, true, llptr, sub.val, ty);
1838
                        }
1839 1840 1841 1842 1843
                        case (_) {
                            auto llty = type_of(cx.fcx.ccx, ty);
                            auto null = lib.llvm.llvm.LLVMConstNull(llty);
                            sub = res(cx, cx.build.Store(null, llptr));
                        }
1844 1845 1846 1847
                    }
                }
            }
        }
1848
        case (_) {
1849
            cx.fcx.ccx.sess.unimpl("stmt variant");
1850 1851
        }
    }
1852
    ret sub;
1853 1854
}

1855
fn new_builder(BasicBlockRef llbb, str name) -> builder {
1856 1857 1858 1859 1860
    let BuilderRef llbuild = llvm.LLVMCreateBuilder();
    llvm.LLVMPositionBuilderAtEnd(llbuild, llbb);
    ret builder(llbuild);
}

1861 1862
// You probably don't want to use this one. See the
// next three functions instead.
1863
fn new_block_ctxt(@fn_ctxt cx, block_parent parent,
1864
                  bool is_scope,
1865
                  str name) -> @block_ctxt {
1866
    let vec[cleanup] cleanups = vec();
1867
    let BasicBlockRef llbb =
1868
        llvm.LLVMAppendBasicBlock(cx.llfn,
1869
                                  _str.buf(cx.ccx.names.next(name)));
1870

1871
    ret @rec(llbb=llbb,
1872 1873
             build=new_builder(llbb, name),
             parent=parent,
1874
             is_scope=is_scope,
1875
             mutable cleanups=cleanups,
1876 1877 1878
             fcx=cx);
}

1879 1880
// 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 {
1881 1882
    ret new_block_ctxt(fcx, parent_none, true, "function top level");
}
1883

1884 1885 1886
// 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 {
    ret new_block_ctxt(bcx.fcx, parent_some(bcx), true, n);
1887 1888
}

1889
// Use this when you're making a general CFG BB within a scope.
1890
fn new_sub_block_ctxt(@block_ctxt bcx, str n) -> @block_ctxt {
1891
    ret new_block_ctxt(bcx.fcx, parent_some(bcx), false, n);
1892
}
1893

1894

1895 1896
fn trans_block_cleanups(@block_ctxt cx,
                        @block_ctxt cleanup_cx) -> @block_ctxt {
1897
    auto bcx = cx;
1898 1899 1900 1901 1902

    if (!cleanup_cx.is_scope) {
        check (_vec.len[cleanup](cleanup_cx.cleanups) == 0u);
    }

1903
    for (cleanup c in cleanup_cx.cleanups) {
1904
        alt (c) {
1905
            case (clean(?cfn)) {
1906
                bcx = cfn(bcx).bcx;
1907 1908 1909
            }
        }
    }
1910 1911 1912
    ret bcx;
}

1913 1914 1915 1916 1917 1918 1919 1920 1921 1922
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;
                    }
1923
                    case (_) { /* fall through */ }
1924 1925
                }
            }
1926
            case (_) { /* fall through */ }
1927 1928 1929 1930
        }
    }
}

1931
impure fn trans_block(@block_ctxt cx, &ast.block b) -> result {
1932 1933
    auto bcx = cx;

1934
    for each (@ast.local local in block_locals(b)) {
1935
        auto ty = node_type(cx.fcx.ccx, local.ann);
1936 1937 1938
        auto val = bcx.build.Alloca(ty);
        cx.fcx.lllocals.insert(local.id, val);
    }
1939
    auto r = res(bcx, C_nil());
1940

1941
    for (@ast.stmt s in b.node.stmts) {
1942 1943
        r = trans_stmt(bcx, *s);
        bcx = r.bcx;
1944 1945 1946 1947 1948
        // 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;
        }
1949
    }
1950

1951 1952 1953 1954 1955 1956 1957 1958 1959 1960 1961 1962 1963
    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());
        }
    }

1964
    bcx = trans_block_cleanups(bcx, find_scope_cx(bcx));
1965
    ret res(bcx, r.val);
1966 1967
}

1968
fn new_fn_ctxt(@crate_ctxt cx,
1969
               str name,
1970
               ValueRef llfndecl) -> @fn_ctxt {
1971

1972
    let ValueRef lltaskptr = llvm.LLVMGetParam(llfndecl, 0u);
1973

1974
    let hashmap[ast.def_id, ValueRef] lllocals = new_def_hash[ValueRef]();
1975 1976
    let hashmap[ast.def_id, ValueRef] llargs = new_def_hash[ValueRef]();

1977
    ret @rec(llfn=llfndecl,
1978
             lltaskptr=lltaskptr,
1979
             llargs=llargs,
1980
             lllocals=lllocals,
1981
             ccx=cx);
1982 1983
}

1984

1985 1986 1987 1988 1989 1990 1991 1992 1993 1994 1995
fn create_llargs_for_fn_args(@fn_ctxt cx, vec[ast.arg] args) {
    let uint arg_n = 1u;
    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;
    }
}


1996 1997 1998 1999
// 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.

2000 2001
fn copy_args_to_allocas(@block_ctxt cx, vec[ast.arg] args,
                        vec[typeck.arg] arg_tys) {
2002 2003 2004

    let uint arg_n = 0u;

2005
    for (ast.arg aarg in args) {
2006 2007 2008 2009 2010 2011 2012 2013 2014
        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);
        }

2015 2016 2017 2018
        arg_n += 1u;
    }
}

2019 2020 2021 2022 2023
fn is_terminated(@block_ctxt cx) -> bool {
    auto inst = llvm.LLVMGetLastInstruction(cx.llbb);
    ret llvm.LLVMIsATerminatorInst(inst) as int != 0;
}

2024 2025 2026 2027 2028 2029 2030 2031 2032
fn arg_tys_of_fn(ast.ann ann) -> vec[typeck.arg] {
    alt (typeck.ann_to_type(ann).struct) {
        case (typeck.ty_fn(?arg_tys, _)) {
            ret arg_tys;
        }
    }
    fail;
}

2033 2034
impure fn trans_fn(@crate_ctxt cx, &ast._fn f, ast.def_id fid,
                   &ast.ann ann) {
2035

2036 2037 2038
    auto llfndecl = cx.item_ids.get(fid);
    cx.item_names.insert(cx.path, llfndecl);

2039 2040 2041
    auto fcx = new_fn_ctxt(cx, cx.path, llfndecl);
    create_llargs_for_fn_args(fcx, f.inputs);

2042
    auto bcx = new_top_block_ctxt(fcx);
2043

2044
    copy_args_to_allocas(bcx, f.inputs, arg_tys_of_fn(ann));
2045

2046 2047
    auto res = trans_block(bcx, f.body);
    if (!is_terminated(res.bcx)) {
2048 2049 2050
        // FIXME: until LLVM has a unit type, we are moving around
        // C_nil values rather than their void type.
        res.bcx.build.Ret(C_nil());
2051
    }
2052 2053
}

2054
fn trans_tag_variant(@crate_ctxt cx, ast.def_id tag_id,
2055
                     &ast.variant variant, int index) {
2056
    if (_vec.len[ast.variant_arg](variant.args) == 0u) {
2057 2058 2059
        ret;    // nullary constructors are just constants
    }

2060 2061 2062 2063 2064 2065 2066 2067 2068 2069 2070 2071 2072 2073 2074 2075 2076
    // 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));
    }

    auto var_ty = typeck.ann_to_type(variant.ann);
    auto llfnty = type_of(cx, var_ty);

    let str s = cx.names.next("_rust_tag") + "." + cx.path;
    let ValueRef llfn = decl_fastcall_fn(cx.llmod, s, llfnty);
    cx.item_ids.insert(variant.id, llfn);

2077 2078 2079
    let ValueRef llfndecl = cx.item_ids.get(variant.id);
    cx.item_names.insert(cx.path, llfndecl);

2080 2081 2082
    auto fcx = new_fn_ctxt(cx, cx.path, llfndecl);
    create_llargs_for_fn_args(fcx, fn_args);

2083 2084 2085 2086 2087 2088 2089 2090 2091 2092 2093 2094 2095 2096 2097 2098 2099 2100 2101 2102 2103 2104 2105 2106 2107 2108 2109 2110 2111
    auto bcx = new_top_block_ctxt(fcx);

    auto arg_tys = arg_tys_of_fn(variant.ann);
    copy_args_to_allocas(bcx, fn_args, arg_tys);

    auto info = cx.tags.get(tag_id);

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

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

    auto lltagptr = bcx.build.Alloca(lltagty);
    auto lldiscrimptr = bcx.build.GEP(lltagptr, vec(C_int(0), C_int(0)));
    bcx.build.Store(C_int(index), lldiscrimptr);

    auto llblobptr = bcx.build.GEP(lltagptr, vec(C_int(0), C_int(1)));

    // First, generate the union type.
    let vec[TypeRef] llargtys = vec();
    for (typeck.arg arg in arg_tys) {
        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) {
2112
        auto llargval = bcx.build.Load(fcx.llargs.get(va.id));
2113 2114 2115 2116 2117 2118 2119 2120 2121 2122
        auto lldestptr = bcx.build.GEP(llunionptr,
                                       vec(C_int(0), C_int(i as int)));

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

    auto lltagval = bcx.build.Load(lltagptr);
    bcx = trans_block_cleanups(bcx, find_scope_cx(bcx));
    bcx.build.Ret(lltagval);
2123 2124
}

2125
impure fn trans_item(@crate_ctxt cx, &ast.item item) {
2126
    alt (item.node) {
2127
        case (ast.item_fn(?name, ?f, _, ?fid, ?ann)) {
2128
            auto sub_cx = @rec(path=cx.path + "." + name with *cx);
2129
            trans_fn(sub_cx, f, fid, ann);
2130
        }
2131 2132
        case (ast.item_mod(?name, ?m, _)) {
            auto sub_cx = @rec(path=cx.path + "." + name with *cx);
2133
            trans_mod(sub_cx, m);
2134
        }
2135 2136
        case (ast.item_tag(?name, ?variants, _, ?tag_id)) {
            auto sub_cx = @rec(path=cx.path + "." + name with *cx);
2137
            auto i = 0;
2138
            for (ast.variant variant in variants) {
2139 2140
                trans_tag_variant(sub_cx, tag_id, variant, i);
                i += 1;
2141 2142
            }
        }
2143
        case (_) { /* fall through */ }
2144 2145 2146
    }
}

2147
impure fn trans_mod(@crate_ctxt cx, &ast._mod m) {
2148 2149
    for (@ast.item item in m.items) {
        trans_item(cx, *item);
2150 2151 2152
    }
}

2153

2154
fn collect_item(&@crate_ctxt cx, @ast.item i) -> @crate_ctxt {
2155
    alt (i.node) {
2156 2157
        case (ast.item_fn(?name, ?f, _, ?fid, ?ann)) {
            // TODO: type-params
2158 2159
            cx.items.insert(fid, i);
            auto llty = node_type(cx, ann);
2160
            let str s = cx.names.next("_rust_fn") + "." + name;
2161
            let ValueRef llfn = decl_fastcall_fn(cx.llmod, s, llty);
2162
            cx.item_ids.insert(fid, llfn);
2163 2164
        }

2165 2166 2167 2168
        case (ast.item_const(?name, _, _, ?cid, _)) {
            cx.items.insert(cid, i);
        }

2169 2170 2171
        case (ast.item_mod(?name, ?m, ?mid)) {
            cx.items.insert(mid, i);
        }
2172 2173 2174 2175

        case (ast.item_tag(_, ?variants, _, ?tag_id)) {
            auto vi = new_def_hash[uint]();
            auto navi = new_def_hash[uint]();
2176 2177
            let vec[tup(ast.def_id,arity)] variant_info = vec();
            cx.tags.insert(tag_id, @rec(th=mk_type_handle(),
2178 2179
                                        mutable variants=variant_info,
                                        mutable size=0u));
2180
            cx.items.insert(tag_id, i);
2181 2182
        }

2183
        case (_) { /* fall through */ }
2184 2185 2186 2187 2188
    }
    ret cx;
}


2189
fn collect_items(@crate_ctxt cx, @ast.crate crate) {
2190

2191 2192
    let fold.ast_fold[@crate_ctxt] fld =
        fold.new_identity_fold[@crate_ctxt]();
2193 2194 2195 2196

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

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

2200 2201 2202 2203 2204 2205
// 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)) {
2206 2207
            auto max_align = 0u;
            auto max_size = 0u;
2208 2209 2210 2211 2212 2213

            auto info = cx.tags.get(tag_id);
            let vec[tup(ast.def_id,arity)] variant_info = vec();

            for (ast.variant variant in variants) {
                auto arity_info;
2214
                if (_vec.len[ast.variant_arg](variant.args) > 0u) {
2215 2216 2217 2218 2219 2220 2221
                    auto llvariantty = type_of_variant(cx, variant);
                    auto align = llvm.LLVMPreferredAlignmentOfType(cx.td.lltd,
                                                                 llvariantty);
                    auto size = llvm.LLVMStoreSizeOfType(cx.td.lltd,
                                                         llvariantty) as uint;
                    if (max_align < align) { max_align = align; }
                    if (max_size < size) { max_size = size; }
2222 2223 2224 2225 2226 2227 2228 2229 2230 2231

                    arity_info = n_ary;
                } else {
                    arity_info = nullary;
                }

                variant_info += vec(tup(variant.id, arity_info));
            }

            info.variants = variant_info;
2232
            info.size = max_size;
2233

2234 2235 2236
            // FIXME: alignment is wrong here, manually insert padding I
            // guess :(
            auto tag_ty = T_struct(vec(T_int(), T_array(T_i8(), max_size)));
2237 2238 2239 2240 2241 2242 2243 2244 2245 2246 2247
            auto th = cx.tags.get(tag_id).th.llth;
            llvm.LLVMRefineType(llvm.LLVMResolveTypeHandle(th), tag_ty);
        }
        case (_) {
            // fall through
        }
    }

    ret cx;
}

2248 2249 2250 2251 2252 2253 2254 2255 2256 2257
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);
}

2258 2259 2260 2261 2262 2263 2264 2265 2266 2267 2268 2269 2270 2271 2272 2273 2274 2275 2276
// 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)) {
            auto info = cx.tags.get(tag_id);

            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;
            while (i < _vec.len[tup(ast.def_id,arity)](info.variants)) {
                auto variant_info = info.variants.(i);
                alt (variant_info._1) {
                    case (nullary) {
                        // Nullary tags become constants.
2277
                        auto union_val = C_zero_byte_arr(info.size as uint);
2278
                        auto val = C_struct(vec(C_int(i as int), union_val));
G
Graydon Hoare 已提交
2279

2280 2281 2282 2283 2284 2285 2286 2287 2288 2289 2290 2291 2292 2293 2294 2295
                        // FIXME: better name
                        auto gvar = llvm.LLVMAddGlobal(cx.llmod, val_ty(val),
                                                       _str.buf("tag"));
                        llvm.LLVMSetInitializer(gvar, val);
                        llvm.LLVMSetGlobalConstant(gvar, True);
                        cx.item_ids.insert(variant_info._0, gvar);
                    }
                    case (n_ary) {
                        // N-ary tags are treated as functions and generated
                        // later.
                    }
                }

                i += 1u;
            }
        }
2296 2297 2298 2299 2300 2301 2302 2303

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

2304 2305 2306 2307 2308 2309 2310 2311 2312 2313 2314 2315 2316 2317 2318 2319 2320
        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);
}

2321 2322 2323 2324
fn p2i(ValueRef v) -> ValueRef {
    ret llvm.LLVMConstPtrToInt(v, T_int());
}

2325
fn trans_exit_task_glue(@crate_ctxt cx) {
2326 2327 2328 2329 2330 2331 2332
    let vec[TypeRef] T_args = vec();
    let vec[ValueRef] V_args = vec();

    auto llfn = cx.glues.exit_task_glue;
    let ValueRef lltaskptr = llvm.LLVMGetParam(llfn, 0u);
    auto fcx = @rec(llfn=llfn,
                    lltaskptr=lltaskptr,
2333
                    llargs=new_def_hash[ValueRef](),
2334
                    lllocals=new_def_hash[ValueRef](),
2335
                    ccx=cx);
2336

2337
    auto bcx = new_top_block_ctxt(fcx);
2338
    trans_upcall(bcx, "upcall_exit", V_args);
2339
    bcx.build.RetVoid();
2340 2341
}

2342 2343 2344 2345 2346
fn create_typedefs(@crate_ctxt cx) {
    llvm.LLVMAddTypeName(cx.llmod, _str.buf("rust_crate"), T_crate());
    llvm.LLVMAddTypeName(cx.llmod, _str.buf("rust_task"), T_task());
}

2347
fn crate_constant(@crate_ctxt cx) -> ValueRef {
2348 2349

    let ValueRef crate_ptr =
2350
        llvm.LLVMAddGlobal(cx.llmod, T_crate(),
2351 2352 2353 2354 2355 2356 2357 2358 2359 2360
                           _str.buf("rust_crate"));

    let ValueRef crate_addr = p2i(crate_ptr);

    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);

2361 2362
    let ValueRef exit_task_glue_off =
        llvm.LLVMConstSub(p2i(cx.glues.exit_task_glue), crate_addr);
2363 2364 2365

    let ValueRef crate_val =
        C_struct(vec(C_null(T_int()),     // ptrdiff_t image_base_off
2366
                     p2i(crate_ptr),      // uintptr_t self_addr
2367 2368 2369 2370 2371 2372 2373 2374 2375 2376 2377 2378 2379 2380 2381 2382 2383 2384
                     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
                     C_null(T_int())      // int n_libs
                     ));

    llvm.LLVMSetInitializer(crate_ptr, crate_val);
    ret crate_ptr;
}

2385
fn trans_main_fn(@crate_ctxt cx, ValueRef llcrate) {
2386 2387 2388
    auto T_main_args = vec(T_int(), T_int());
    auto T_rust_start_args = vec(T_int(), T_int(), T_int(), T_int());

2389 2390 2391 2392 2393 2394 2395
    auto main_name;
    if (_str.eq(std.os.target_os(), "win32")) {
        main_name = "WinMain@16";
    } else {
        main_name = "main";
    }

2396
    auto llmain =
2397
        decl_cdecl_fn(cx.llmod, main_name, T_fn(T_main_args, T_int()));
2398

2399 2400
    auto llrust_start = decl_cdecl_fn(cx.llmod, "rust_start",
                                      T_fn(T_rust_start_args, T_int()));
2401 2402 2403

    auto llargc = llvm.LLVMGetParam(llmain, 0u);
    auto llargv = llvm.LLVMGetParam(llmain, 1u);
2404 2405
    check (cx.item_names.contains_key("_rust.main"));
    auto llrust_main = cx.item_names.get("_rust.main");
2406 2407 2408 2409 2410 2411 2412 2413 2414 2415 2416

    //
    // 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(""));
2417
    auto b = new_builder(llbb, "");
2418 2419 2420 2421 2422 2423 2424

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

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

}

2425 2426
fn declare_intrinsics(ModuleRef llmod) -> hashmap[str,ValueRef] {

2427
    let vec[TypeRef] T_trap_args = vec();
2428 2429 2430 2431
    let vec[TypeRef] T_memcpy32_args = vec(T_ptr(T_i8()), T_ptr(T_i8()),
                                           T_i32(), T_i32(), T_i1());
    let vec[TypeRef] T_memcpy64_args = vec(T_ptr(T_i8()), T_ptr(T_i8()),
                                           T_i32(), T_i32(), T_i1());
2432 2433
    auto trap = decl_cdecl_fn(llmod, "llvm.trap",
                              T_fn(T_trap_args, T_void()));
2434 2435 2436 2437
    auto memcpy32 = decl_cdecl_fn(llmod, "llvm.memcpy.p0i8.p0i8.i32",
                                  T_fn(T_memcpy32_args, T_void()));
    auto memcpy64 = decl_cdecl_fn(llmod, "llvm.memcpy.p0i8.p0i8.i64",
                                  T_fn(T_memcpy64_args, T_void()));
2438 2439 2440

    auto intrinsics = new_str_hash[ValueRef]();
    intrinsics.insert("llvm.trap", trap);
2441 2442
    intrinsics.insert("llvm.memcpy.p0i8.p0i8.i32", memcpy32);
    intrinsics.insert("llvm.memcpy.p0i8.p0i8.i64", memcpy64);
2443
    ret intrinsics;
2444 2445
}

2446 2447 2448 2449 2450 2451 2452 2453
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.
}

2454
fn trans_crate(session.session sess, @ast.crate crate, str output) {
2455 2456 2457 2458
    auto llmod =
        llvm.LLVMModuleCreateWithNameInContext(_str.buf("rust_out"),
                                               llvm.LLVMGetGlobalContext());

2459 2460
    llvm.LLVMSetDataLayout(llmod, _str.buf(x86.get_data_layout()));
    llvm.LLVMSetTarget(llmod, _str.buf(x86.get_target_triple()));
2461
    auto td = mk_target_data(x86.get_data_layout());
2462

2463 2464
    llvm.LLVMSetModuleInlineAsm(llmod, _str.buf(x86.get_module_asm()));

2465
    auto intrinsics = declare_intrinsics(llmod);
2466

2467 2468 2469
    auto glues = @rec(activate_glue = decl_glue(llmod,
                                                abi.activate_glue_name()),
                      yield_glue = decl_glue(llmod, abi.yield_glue_name()),
2470 2471 2472 2473 2474 2475 2476 2477 2478 2479 2480 2481 2482 2483
                      /*
                       * 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(),
2484
                                    T_fn(vec(T_taskptr()), T_void())),
2485

2486 2487 2488 2489
                      upcall_glues =
                      _vec.init_fn[ValueRef](bind decl_upcall(llmod, _),
                                             abi.n_upcall_glues as uint));

2490 2491 2492 2493
    auto hasher = typeck.hash_ty;
    auto eqer = typeck.eq_ty;
    auto types = map.mk_hashmap[@typeck.ty,@ty_info](hasher, eqer);

2494 2495
    auto cx = @rec(sess = sess,
                   llmod = llmod,
2496
                   td = td,
2497
                   upcalls = new_str_hash[ValueRef](),
2498
                   intrinsics = intrinsics,
2499 2500
                   item_names = new_str_hash[ValueRef](),
                   item_ids = new_def_hash[ValueRef](),
2501
                   items = new_def_hash[@ast.item](),
2502
                   tags = new_def_hash[@tag_info](),
2503
                   types = types,
2504
                   glues = glues,
2505
                   names = namegen(0),
2506
                   path = "_rust");
2507

2508 2509
    create_typedefs(cx);

2510
    collect_items(cx, crate);
2511
    resolve_tag_types(cx, crate);
2512
    trans_constants(cx, crate);
2513

2514
    trans_mod(cx, crate.node.module);
2515
    trans_exit_task_glue(cx);
2516 2517
    trans_main_fn(cx, crate_constant(cx));

2518 2519
    check_module(llmod);

2520
    llvm.LLVMWriteBitcodeToFile(llmod, _str.buf(output));
2521 2522 2523 2524 2525 2526 2527 2528 2529 2530 2531 2532 2533
    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:
//