trans.rs 99.7 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.ty;
15
import back.x86;
16 17
import back.abi;

18
import middle.ty.pat_ty;
P
Patrick Walton 已提交
19

20
import util.common;
21
import util.common.append;
22
import util.common.istr;
23
import util.common.new_def_hash;
24
import util.common.new_str_hash;
25 26 27

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

40 41
import lib.llvm.False;
import lib.llvm.True;
42

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

50 51
type glue_fns = rec(ValueRef activate_glue,
                    ValueRef yield_glue,
52
                    ValueRef exit_task_glue,
53 54
                    vec[ValueRef] upcall_glues,
                    ValueRef no_op_type_glue);
55

56
tag arity { nullary; n_ary; }
57
type tag_info = rec(type_handle th,
58 59
                    mutable vec[tup(ast.def_id,arity)] variants,
                    mutable uint size);
60

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

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

82
tag cleanup {
83
    clean(fn(@block_ctxt cx) -> result);
84 85 86 87
}

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

93 94 95 96 97 98 99 100
// 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);
}

101

102 103 104 105 106 107 108 109
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);
}

110 111 112 113 114 115 116 117 118 119 120
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));
}
121 122 123 124


// LLVM type constructors.

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

139 140 141 142 143
fn T_nil() -> TypeRef {
    // NB: See above in T_void().
    ret llvm.LLVMInt1Type();
}

144 145 146 147
fn T_i1() -> TypeRef {
    ret llvm.LLVMInt1Type();
}

148 149 150 151 152 153 154 155 156
fn T_i8() -> TypeRef {
    ret llvm.LLVMInt8Type();
}

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

fn T_i32() -> TypeRef {
157 158 159
    ret llvm.LLVMInt32Type();
}

160 161 162 163
fn T_i64() -> TypeRef {
    ret llvm.LLVMInt64Type();
}

164 165 166 167 168 169 170 171
fn T_f32() -> TypeRef {
    ret llvm.LLVMFloatType();
}

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

172 173 174 175
fn T_bool() -> TypeRef {
    ret T_i1();
}

176 177 178 179 180
fn T_int() -> TypeRef {
    // FIXME: switch on target type.
    ret T_i32();
}

181 182 183 184
fn T_char() -> TypeRef {
    ret T_i32();
}

185 186 187 188
fn T_fn(vec[TypeRef] inputs, TypeRef output) -> TypeRef {
    ret llvm.LLVMFunctionType(output,
                              _vec.buf[TypeRef](inputs),
                              _vec.len[TypeRef](inputs),
189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207
                              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
208 209 210 211 212 213 214 215 216 217
                     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
                     ));
}

218 219 220 221 222 223
fn T_tydesc() -> TypeRef {
    auto pvoid = T_ptr(T_i8());
    auto glue_fn_ty = T_ptr(T_fn(vec(T_taskptr(), pvoid), T_void()));
    ret T_struct(vec(pvoid,             // first_param
                     T_int(),           // size
                     T_int(),           // align
224
                     glue_fn_ty,        // take_glue_off
225 226 227 228 229 230 231 232
                     glue_fn_ty,        // drop_glue_off
                     glue_fn_ty,        // free_glue_off
                     glue_fn_ty,        // sever_glue_off
                     glue_fn_ty,        // mark_glue_off
                     glue_fn_ty,        // obj_drop_glue_off
                     glue_fn_ty));      // is_stateful
}

233 234 235 236
fn T_array(TypeRef t, uint n) -> TypeRef {
    ret llvm.LLVMArrayType(t, n);
}

237 238 239 240 241
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
242 243 244
                     ));
}

245 246
fn T_str() -> TypeRef {
    ret T_vec(T_i8());
247 248
}

249 250 251 252
fn T_box(TypeRef t) -> TypeRef {
    ret T_struct(vec(T_int(), t));
}

253 254 255 256 257 258 259 260 261 262 263 264 265 266
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
267
                     T_int()       // int n_libs
268 269 270
                     ));
}

271 272 273 274 275 276
fn T_double() -> TypeRef {
    ret llvm.LLVMDoubleType();
}

fn T_taskptr() -> TypeRef {
    ret T_ptr(T_task());
277 278
}

279
fn type_of(@crate_ctxt cx, @ty.t t) -> TypeRef {
280 281
    let TypeRef llty = type_of_inner(cx, t);
    check (llty as int != 0);
282
    llvm.LLVMAddTypeName(cx.llmod, _str.buf(ty.ty_to_str(t)), llty);
283 284 285
    ret llty;
}

286
fn type_of_fn(@crate_ctxt cx, vec[ty.arg] inputs, @ty.t output) -> TypeRef {
287
    let vec[TypeRef] atys = vec(T_taskptr());
288

289 290
    auto fn_ty = ty.plain_ty(ty.ty_fn(inputs, output));
    auto ty_param_count = ty.count_ty_params(fn_ty);
291 292
    auto i = 0u;
    while (i < ty_param_count) {
293
        atys += T_ptr(T_tydesc());
294 295 296
        i += 1u;
    }

297
    for (ty.arg arg in inputs) {
298 299 300 301 302 303 304 305 306 307 308
        let TypeRef t = type_of(cx, arg.ty);
        alt (arg.mode) {
            case (ast.alias) {
                t = T_ptr(t);
            }
            case (_) { /* fall through */  }
        }
        atys += t;
    }

    auto ret_ty;
309
    if (ty.type_is_nil(output)) {
310 311 312 313 314 315 316 317
        ret_ty = llvm.LLVMVoidType();
    } else {
        ret_ty = type_of(cx, output);
    }

    ret T_fn(atys, ret_ty);
}

318
fn type_of_inner(@crate_ctxt cx, @ty.t t) -> TypeRef {
319
    alt (t.struct) {
320 321 322 323 324
        case (ty.ty_nil) { ret T_nil(); }
        case (ty.ty_bool) { ret T_bool(); }
        case (ty.ty_int) { ret T_int(); }
        case (ty.ty_uint) { ret T_int(); }
        case (ty.ty_machine(?tm)) {
325 326 327 328 329 330 331 332 333 334 335 336 337
            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(); }
            }
        }
338 339 340
        case (ty.ty_char) { ret T_char(); }
        case (ty.ty_str) { ret T_ptr(T_str()); }
        case (ty.ty_tag(?tag_id)) {
341 342
            ret llvm.LLVMResolveTypeHandle(cx.tags.get(tag_id).th.llth);
        }
343
        case (ty.ty_box(?t)) {
344 345
            ret T_ptr(T_box(type_of(cx, t)));
        }
346
        case (ty.ty_vec(?t)) {
347
            ret T_ptr(T_vec(type_of(cx, t)));
348
        }
349
        case (ty.ty_tup(?elts)) {
350
            let vec[TypeRef] tys = vec();
351
            for (@ty.t elt in elts) {
352
                tys += type_of(cx, elt);
353 354 355
            }
            ret T_struct(tys);
        }
356
        case (ty.ty_rec(?fields)) {
357
            let vec[TypeRef] tys = vec();
358
            for (ty.field f in fields) {
359 360 361 362
                tys += type_of(cx, f.ty);
            }
            ret T_struct(tys);
        }
363
        case (ty.ty_fn(?args, ?out)) {
364 365
            ret type_of_fn(cx, args, out);
        }
366
        case (ty.ty_obj(?meths)) {
367
            let vec[TypeRef] mtys = vec();
368
            for (ty.method m in meths) {
369 370
                let TypeRef mty = type_of_fn(cx, m.inputs, m.output);
                mtys += T_ptr(mty);
371
            }
372 373 374
            let TypeRef vtbl = T_struct(mtys);
            let TypeRef pair =
                T_struct(vec(T_ptr(vtbl),
375
                             T_ptr(T_box(T_nil()))));
376
            ret pair;
377
        }
378
        case (ty.ty_var(_)) {
379
            log "ty_var in trans.type_of";
380 381
            fail;
        }
382
        case (ty.ty_param(_)) {
383
            ret T_ptr(T_i8());
384 385 386 387 388
        }
    }
    fail;
}

389
fn type_of_arg(@crate_ctxt cx, &ty.arg arg) -> TypeRef {
390 391 392 393 394 395 396
    auto ty = type_of(cx, arg.ty);
    if (arg.mode == ast.alias) {
        ty = T_ptr(ty);
    }
    ret ty;
}

397 398 399 400 401 402 403 404 405
// 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 {
406 407 408 409 410 411 412 413 414 415 416 417 418 419
            if (c == (',' as u8)) {
                result += "_";
            } else {
                if (c == ('{' as u8) || c == ('(' as u8)) {
                    result += "_of_";
                } else {
                    if (c != 10u8 && c != ('}' as u8) && c != (')' as u8) &&
                            c != (' ' as u8) && c != ('\t' as u8) &&
                            c != (';' as u8)) {
                        auto v = vec(c);
                        result += _str.from_bytes(v);
                    }
                }
            }
420 421 422 423 424
        }
    }
    ret result;
}

425 426 427 428 429 430
// LLVM constant constructors.

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

431
fn C_integral(int i, TypeRef t) -> ValueRef {
432 433 434 435 436 437
    // 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);
    //
438 439 440
    ret llvm.LLVMConstIntOfString(t, _str.buf(istr(i)), 10);
}

441 442 443 444 445
fn C_nil() -> ValueRef {
    // NB: See comment above in T_void().
    ret C_integral(0, T_i1());
}

446 447
fn C_bool(bool b) -> ValueRef {
    if (b) {
448
        ret C_integral(1, T_bool());
449
    } else {
450
        ret C_integral(0, T_bool());
451 452 453
    }
}

454 455
fn C_int(int i) -> ValueRef {
    ret C_integral(i, T_int());
456 457
}

458
fn C_str(@crate_ctxt cx, str s) -> ValueRef {
459
    auto sc = llvm.LLVMConstString(_str.buf(s), _str.byte_len(s), False);
460
    auto g = llvm.LLVMAddGlobal(cx.llmod, val_ty(sc),
461 462
                                _str.buf(cx.names.next("str")));
    llvm.LLVMSetInitializer(g, sc);
463
    llvm.LLVMSetGlobalConstant(g, True);
464
    ret g;
465 466
}

467 468 469 470 471 472 473 474 475 476 477
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));
}

478 479 480 481 482 483
fn C_struct(vec[ValueRef] elts) -> ValueRef {
    ret llvm.LLVMConstStruct(_vec.buf[ValueRef](elts),
                             _vec.len[ValueRef](elts),
                             False);
}

484
fn decl_fn(ModuleRef llmod, str name, uint cc, TypeRef llty) -> ValueRef {
485 486
    let ValueRef llfn =
        llvm.LLVMAddFunction(llmod, _str.buf(name), llty);
487
    llvm.LLVMSetFunctionCallConv(llfn, cc);
488 489 490
    ret llfn;
}

491 492
fn decl_cdecl_fn(ModuleRef llmod, str name, TypeRef llty) -> ValueRef {
    ret decl_fn(llmod, name, lib.llvm.LLVMCCallConv, llty);
493 494
}

495 496
fn decl_fastcall_fn(ModuleRef llmod, str name, TypeRef llty) -> ValueRef {
    ret decl_fn(llmod, name, lib.llvm.LLVMFastCallConv, llty);
497 498
}

499
fn decl_glue(ModuleRef llmod, str s) -> ValueRef {
500
    ret decl_cdecl_fn(llmod, s, T_fn(vec(T_taskptr()), T_void()));
501 502
}

503
fn decl_upcall(ModuleRef llmod, uint _n) -> ValueRef {
504 505 506 507
    // 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.
508
    let int n = _n as int;
509
    let str s = abi.upcall_glue_name(n);
510
    let vec[TypeRef] args =
511 512
        vec(T_taskptr(), // taskptr
            T_int())     // callee
513 514
        + _vec.init_elt[TypeRef](T_int(), n as uint);

515
    ret decl_fastcall_fn(llmod, s, T_fn(args, T_int()));
516 517
}

518
fn get_upcall(@crate_ctxt cx, str name, int n_args) -> ValueRef {
519 520 521
    if (cx.upcalls.contains_key(name)) {
        ret cx.upcalls.get(name);
    }
522
    auto inputs = vec(T_taskptr());
523
    inputs += _vec.init_elt[TypeRef](T_int(), n_args as uint);
524
    auto output = T_int();
525
    auto f = decl_cdecl_fn(cx.llmod, name, T_fn(inputs, output));
526 527 528 529
    cx.upcalls.insert(name, f);
    ret f;
}

530
fn trans_upcall(@block_ctxt cx, str name, vec[ValueRef] args) -> result {
531
    let int n = _vec.len[ValueRef](args) as int;
532
    let ValueRef llupcall = get_upcall(cx.fcx.ccx, name, n);
533 534
    llupcall = llvm.LLVMConstPointerCast(llupcall, T_int());

535
    let ValueRef llglue = cx.fcx.ccx.glues.upcall_glues.(n);
536 537 538 539
    let vec[ValueRef] call_args = vec(cx.fcx.lltaskptr, llupcall);
    for (ValueRef a in args) {
        call_args += cx.build.ZExtOrBitCast(a, T_int());
    }
540
    ret res(cx, cx.build.FastCall(llglue, call_args));
541 542
}

543 544 545
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)));
546 547
}

548 549 550 551 552 553 554 555 556 557 558 559 560 561
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;
        }
    }
}

562 563 564 565 566 567 568 569
fn size_of(TypeRef t) -> ValueRef {
    ret llvm.LLVMConstIntCast(lib.llvm.llvm.LLVMSizeOf(t), T_int(), False);
}

fn align_of(TypeRef t) -> ValueRef {
    ret llvm.LLVMConstIntCast(lib.llvm.llvm.LLVMAlignOf(t), T_int(), False);
}

570 571
fn trans_malloc_inner(@block_ctxt cx, TypeRef llptr_ty) -> result {
    auto llbody_ty = lib.llvm.llvm.LLVMGetElementType(llptr_ty);
572 573
    // FIXME: need a table to collect tydesc globals.
    auto tydesc = C_int(0);
574
    auto sz = size_of(llbody_ty);
575
    auto sub = trans_upcall(cx, "upcall_malloc", vec(sz, tydesc));
576 577 578 579 580 581 582 583
    sub.val = sub.bcx.build.IntToPtr(sub.val, llptr_ty);
    ret sub;
}

fn trans_malloc(@block_ctxt cx, @ty.t t) -> result {
    auto scope_cx = find_scope_cx(cx);
    auto llptr_ty = type_of(cx.fcx.ccx, t);
    auto sub = trans_malloc_inner(cx, llptr_ty);
584
    scope_cx.cleanups += clean(bind drop_ty(_, sub.val, t));
585 586 587 588
    ret sub;
}


589 590 591 592 593
// Type descriptor and type glue stuff

// Given a type and a field index into its corresponding type descriptor,
// returns an LLVM ValueRef of that field from the tydesc, generating the
// tydesc if necessary.
594 595
fn field_of_tydesc(@block_ctxt cx, @ty.t t, int field) -> ValueRef {
    auto tydesc = get_tydesc(cx, t);
596 597
    ret cx.build.GEP(tydesc, vec(C_int(0), C_int(field)));
}
598

599
fn get_tydesc(&@block_ctxt cx, @ty.t t) -> ValueRef {
600
    // Is the supplied type a type param? If so, return the passed-in tydesc.
601
    alt (ty.type_param(t)) {
602 603
        case (some[ast.def_id](?id)) { ret cx.fcx.lltydescs.get(id); }
        case (none[ast.def_id])      { /* fall through */ }
604
    }
605 606

    // Does it contain a type param? If so, generate a derived tydesc.
607
    if (ty.count_ty_params(t) > 0u) {
608 609 610 611 612
        log "TODO: trans.get_tydesc(): generate a derived type descriptor";
        fail;
    }

    // Otherwise, generate a tydesc if necessary, and return it.
613 614
    if (!cx.fcx.ccx.tydescs.contains_key(t)) {
        make_tydesc(cx.fcx.ccx, t);
615
    }
616
    ret cx.fcx.ccx.tydescs.get(t);
617 618
}

619
fn make_tydesc(@crate_ctxt cx, @ty.t t) {
620
    auto tg = make_take_glue;
621
    auto take_glue = make_generic_glue(cx, t, "take", tg);
622
    auto dg = make_drop_glue;
623
    auto drop_glue = make_generic_glue(cx, t, "drop", dg);
624

625
    auto llty = type_of(cx, t);
626 627 628
    auto pvoid = T_ptr(T_i8());
    auto glue_fn_ty = T_ptr(T_fn(vec(T_taskptr(), pvoid), T_void()));
    auto tydesc = C_struct(vec(C_null(pvoid),
629 630
                               size_of(llty),
                               align_of(llty),
631
                               take_glue,             // take_glue_off
632 633 634 635 636 637 638
                               drop_glue,             // drop_glue_off
                               C_null(glue_fn_ty),    // free_glue_off
                               C_null(glue_fn_ty),    // sever_glue_off
                               C_null(glue_fn_ty),    // mark_glue_off
                               C_null(glue_fn_ty),    // obj_drop_glue_off
                               C_null(glue_fn_ty)));  // is_stateful

639
    auto name = sanitize(cx.names.next("tydesc_" + ty.ty_to_str(t)));
640 641 642
    auto gvar = llvm.LLVMAddGlobal(cx.llmod, val_ty(tydesc), _str.buf(name));
    llvm.LLVMSetInitializer(gvar, tydesc);
    llvm.LLVMSetGlobalConstant(gvar, True);
643
    cx.tydescs.insert(t, gvar);
644 645
}

646
fn make_generic_glue(@crate_ctxt cx, @ty.t t, str name,
647
                     val_and_ty_fn helper) -> ValueRef {
648
    auto llfnty = T_fn(vec(T_taskptr(), T_ptr(T_i8())), T_void());
649

650
    auto fn_name = cx.names.next("_rust_" + name) + "." + ty.ty_to_str(t);
651 652 653 654 655 656
    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);

657
    auto re;
658
    if (!ty.type_is_scalar(t)) {
659
        auto llty;
660
        if (ty.type_is_structural(t)) {
661 662 663 664
            llty = T_ptr(type_of(cx, t));
        } else {
            llty = type_of(cx, t);
        }
665

666 667
        auto llrawptr = llvm.LLVMGetParam(llfn, 1u);
        auto llval = bcx.build.BitCast(llrawptr, llty);
G
Graydon Hoare 已提交
668

669 670 671 672
        re = helper(bcx, llval, t);
    } else {
        re = res(bcx, C_nil());
    }
673

674
    re.bcx.build.RetVoid();
675 676 677
    ret llfn;
}

678 679
fn make_take_glue(@block_ctxt cx, ValueRef v, @ty.t t) -> result {
    if (ty.type_is_boxed(t)) {
680 681
        ret incr_refcnt_of_boxed(cx, v);

682
    } else if (ty.type_is_structural(t)) {
683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702 703 704 705 706 707
        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());
}

708
fn make_drop_glue(@block_ctxt cx, ValueRef v, @ty.t t) -> result {
709
    alt (t.struct) {
710
        case (ty.ty_str) {
711 712 713 714 715 716
            ret decr_refcnt_and_if_zero(cx, v,
                                        bind trans_non_gc_free(_, v),
                                        "free string",
                                        T_int(), C_int(0));
        }

717 718
        case (ty.ty_vec(_)) {
            fn hit_zero(@block_ctxt cx, ValueRef v, @ty.t t) -> result {
719 720 721 722 723 724 725 726 727 728
                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));
        }

729 730
        case (ty.ty_box(?body_ty)) {
            fn hit_zero(@block_ctxt cx, ValueRef v, @ty.t body_ty) -> result {
731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746
                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 (_) {
747
            if (ty.type_is_structural(t)) {
748 749 750
                ret iter_structural_ty(cx, v, t,
                                       bind drop_ty(_, _, _));

751 752
            } else if (ty.type_is_scalar(t) ||
                       ty.type_is_nil(t)) {
753 754 755 756
                ret res(cx, C_nil());
            }
        }
    }
757
    cx.fcx.ccx.sess.bug("bad type in trans.make_drop_glue_inner: " +
758
                        ty.ty_to_str(t));
759 760 761
    fail;
}

762 763
fn decr_refcnt_and_if_zero(@block_ctxt cx,
                           ValueRef box_ptr,
764
                           fn(@block_ctxt cx) -> result inner,
765
                           str inner_name,
766
                           TypeRef t_else, ValueRef v_else) -> result {
767

768
    auto load_rc_cx = new_sub_block_ctxt(cx, "load rc");
769 770 771 772
    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");

773 774
    auto null_test = cx.build.IsNull(box_ptr);
    cx.build.CondBr(null_test, next_cx.llbb, load_rc_cx.llbb);
775

776 777 778 779 780 781 782 783 784 785

    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);
786 787 788 789

    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);
790 791 792 793
    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);
794

795
    auto phi = next_cx.build.Phi(t_else,
796
                                 vec(v_else, v_else, v_else, inner_res.val),
797
                                 vec(cx.llbb,
798
                                     load_rc_cx.llbb,
799
                                     rc_adj_cx.llbb,
800 801
                                     inner_res.bcx.llbb));

802
    ret res(next_cx, phi);
803 804
}

805 806
fn type_of_variant(@crate_ctxt cx, &ast.variant v) -> TypeRef {
    let vec[TypeRef] lltys = vec();
807 808 809
    alt (ty.ann_to_type(v.ann).struct) {
        case (ty.ty_fn(?args, _)) {
            for (ty.arg arg in args) {
810 811 812 813 814 815 816 817
                lltys += vec(type_of(cx, arg.ty));
            }
        }
        case (_) { fail; }
    }
    ret T_struct(lltys);
}

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

820
// Iterates through the elements of a structural type.
821 822
fn iter_structural_ty(@block_ctxt cx,
                      ValueRef v,
823
                      @ty.t t,
824 825 826
                      val_and_ty_fn f)
    -> result {
    let result r = res(cx, C_nil());
827

828 829 830 831
    fn iter_boxpp(@block_ctxt cx,
                  ValueRef box_cell,
                  val_and_ty_fn f) -> result {
        auto box_ptr = cx.build.Load(box_cell);
832 833
        auto tnil = ty.plain_ty(ty.ty_nil);
        auto tbox = ty.plain_ty(ty.ty_box(tnil));
834 835 836 837 838 839 840 841 842 843 844

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

        auto r = f(inner_cx, box_ptr, tbox);
        r.bcx.build.Br(next_cx.llbb);
        ret res(next_cx, r.val);
    }

845
    alt (t.struct) {
846
        case (ty.ty_tup(?args)) {
847
            let int i = 0;
848
            for (@ty.t arg in args) {
849
                auto elt = r.bcx.build.GEP(v, vec(C_int(0), C_int(i)));
850 851 852
                r = f(r.bcx,
                      load_non_structural(r.bcx, elt, arg),
                      arg);
853 854 855
                i += 1;
            }
        }
856
        case (ty.ty_rec(?fields)) {
857
            let int i = 0;
858
            for (ty.field fld in fields) {
859
                auto llfld = r.bcx.build.GEP(v, vec(C_int(0), C_int(i)));
860 861 862
                r = f(r.bcx,
                      load_non_structural(r.bcx, llfld, fld.ty),
                      fld.ty);
863 864 865
                i += 1;
            }
        }
866
        case (ty.ty_tag(?tid)) {
867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887
            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 已提交
888

889 890 891 892 893
            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 已提交
894

895 896 897 898 899 900 901 902 903 904 905 906
            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));
907
                        auto llvar = variant_cx.build.GEP(v, vals);
908 909
                        auto llvarty = type_of_variant(cx.fcx.ccx,
                                                       variants.(i));
910

911
                        auto fn_ty = ty.ann_to_type(variants.(i).ann);
912
                        alt (fn_ty.struct) {
913
                            case (ty.ty_fn(?args, _)) {
914 915 916 917
                                auto llvarp = variant_cx.build.
                                    TruncOrBitCast(llunion_ptr,
                                                   T_ptr(llvarty));

918
                                auto j = 0u;
919
                                for (ty.arg a in args) {
920 921
                                    auto llfldp = variant_cx.build.GEP(llvarp,
                                        vec(C_int(0), C_int(j as int)));
922 923
                                    auto llfld =
                                        load_non_structural(variant_cx,
924
                                                            llfldp, a.ty);
925

926 927 928 929 930 931 932 933 934
                                    auto res = f(variant_cx, llfld, a.ty);
                                    variant_cx = res.bcx;
                                    j += 1u;
                                }
                            }
                            case (_) { fail; }
                        }

                        variant_cx.build.Br(next_cx.llbb);
935 936 937 938 939 940 941 942 943 944 945 946
                    }
                    case (nullary) {
                        // Nothing to do.
                        variant_cx.build.Br(next_cx.llbb);
                    }
                }

                i += 1u;
            }

            ret res(next_cx, C_nil());
        }
947
        case (ty.ty_fn(_,_)) {
948 949 950 951
            auto box_cell =
                cx.build.GEP(v,
                             vec(C_int(0),
                                 C_int(abi.fn_field_box)));
952
            ret iter_boxpp(cx, box_cell, f);
953
        }
954
        case (ty.ty_obj(_)) {
955 956 957 958
            auto box_cell =
                cx.build.GEP(v,
                             vec(C_int(0),
                                 C_int(abi.obj_field_box)));
959
            ret iter_boxpp(cx, box_cell, f);
960
        }
961 962 963
        case (_) {
            cx.fcx.ccx.sess.unimpl("type in iter_structural_ty");
        }
964
    }
965
    ret r;
966 967
}

968 969 970
// Iterates through the elements of a vec or str.
fn iter_sequence(@block_ctxt cx,
                 ValueRef v,
971
                 @ty.t t,
972 973 974 975
                 val_and_ty_fn f) -> result {

    fn iter_sequence_body(@block_ctxt cx,
                          ValueRef v,
976
                          @ty.t elt_ty,
977 978 979 980 981 982 983
                          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)));
984 985

        auto llunit_ty = type_of(cx.fcx.ccx, elt_ty);
986
        auto unit_sz = size_of(llunit_ty);
987

988 989
        auto len = cx.build.Load(lenptr);
        if (trailing_null) {
990
            len = cx.build.Sub(len, unit_sz);
991 992 993 994
        }

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

995 996
        auto cond_cx = new_scope_block_ctxt(cx, "sequence-iter cond");
        auto body_cx = new_scope_block_ctxt(cx, "sequence-iter body");
997 998
        auto next_cx = new_sub_block_ctxt(cx, "next");

999 1000
        cx.build.Br(cond_cx.llbb);

1001
        auto ix = cond_cx.build.Phi(T_int(), vec(C_int(0)), vec(cx.llbb));
1002 1003 1004 1005 1006
        auto scaled_ix = cond_cx.build.Phi(T_int(),
                                           vec(C_int(0)), vec(cx.llbb));

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

1009
        auto elt = body_cx.build.GEP(p0, vec(C_int(0), ix));
1010 1011 1012
        auto body_res = f(body_cx,
                          load_non_structural(body_cx, elt, elt_ty),
                          elt_ty);
1013
        auto next_ix = body_res.bcx.build.Add(ix, C_int(1));
1014 1015
        auto next_scaled_ix = body_res.bcx.build.Add(scaled_ix, unit_sz);

1016 1017 1018
        cond_cx.build.AddIncomingToPhi(ix, vec(next_ix),
                                       vec(body_res.bcx.llbb));

1019 1020 1021
        cond_cx.build.AddIncomingToPhi(scaled_ix, vec(next_scaled_ix),
                                       vec(body_res.bcx.llbb));

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

1026 1027
    alt (t.struct) {
        case (ty.ty_vec(?et)) {
1028 1029
            ret iter_sequence_body(cx, v, et, f, false);
        }
1030 1031
        case (ty.ty_str) {
            auto et = ty.plain_ty(ty.ty_machine(common.ty_u8));
1032
            ret iter_sequence_body(cx, v, et, f, true);
1033
        }
1034
        case (_) { fail; }
1035
    }
1036 1037 1038 1039 1040 1041
    cx.fcx.ccx.sess.bug("bad type in trans.iter_sequence");
    fail;
}

fn incr_all_refcnts(@block_ctxt cx,
                    ValueRef v,
1042
                    @ty.t t) -> result {
1043

1044
    if (!ty.type_is_scalar(t)) {
1045
        auto llrawptr = cx.build.BitCast(v, T_ptr(T_i8()));
1046
        auto llfnptr = field_of_tydesc(cx, t, abi.tydesc_field_take_glue_off);
1047 1048
        auto llfn = cx.build.Load(llfnptr);
        cx.build.FastCall(llfn, vec(cx.fcx.lltaskptr, llrawptr));
1049
    }
1050
    ret res(cx, C_nil());
1051 1052
}

1053 1054
fn drop_slot(@block_ctxt cx,
             ValueRef slot,
1055
             @ty.t t) -> result {
1056 1057 1058 1059 1060 1061 1062
    auto llptr = load_non_structural(cx, slot, t);
    auto re = drop_ty(cx, llptr, t);

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

1065 1066
fn drop_ty(@block_ctxt cx,
           ValueRef v,
1067
           @ty.t t) -> result {
1068

1069
    if (!ty.type_is_scalar(t)) {
1070
        auto llrawptr = cx.build.BitCast(v, T_ptr(T_i8()));
1071 1072 1073
        auto llfnptr = field_of_tydesc(cx, t, abi.tydesc_field_drop_glue_off);
        auto llfn = cx.build.Load(llfnptr);
        cx.build.FastCall(llfn, vec(cx.fcx.lltaskptr, llrawptr));
1074
    }
1075
    ret res(cx, C_nil());
1076 1077 1078 1079 1080 1081
}

fn build_memcpy(@block_ctxt cx,
                ValueRef dst,
                ValueRef src,
                TypeRef llty) -> result {
1082 1083 1084
    // 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");
1085 1086
    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 已提交
1087 1088
    auto size = cx.build.IntCast(lib.llvm.llvm.LLVMSizeOf(llty),
                                 T_i32());
1089 1090 1091 1092 1093 1094
    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.

P
Patrick Walton 已提交
1095
    log "building memcpy";
1096 1097 1098 1099 1100 1101 1102 1103 1104 1105
    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,
1106 1107
           @ty.t t) -> result {
    if (ty.type_is_scalar(t)) {
1108 1109
        ret res(cx, cx.build.Store(src, dst));

1110
    } else if (ty.type_is_nil(t)) {
1111 1112
        ret res(cx, C_nil());

1113
    } else if (ty.type_is_boxed(t)) {
1114
        auto r = incr_all_refcnts(cx, src, t);
1115
        if (! is_init) {
1116
            r = drop_ty(r.bcx, r.bcx.build.Load(dst), t);
1117 1118 1119
        }
        ret res(r.bcx, r.bcx.build.Store(src, dst));

1120
    } else if (ty.type_is_structural(t)) {
1121 1122 1123 1124
        auto r = incr_all_refcnts(cx, src, t);
        if (! is_init) {
            r = drop_ty(r.bcx, dst, t);
        }
1125 1126 1127 1128
        // 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));
1129 1130 1131
    }

    cx.fcx.ccx.sess.bug("unexpected type in trans.copy_ty: " +
1132
                        ty.ty_to_str(t));
1133 1134 1135
    fail;
}

1136
impure fn trans_lit(@block_ctxt cx, &ast.lit lit, &ast.ann ann) -> result {
1137
    alt (lit.node) {
1138
        case (ast.lit_int(?i)) {
1139
            ret res(cx, C_int(i));
1140 1141
        }
        case (ast.lit_uint(?u)) {
1142
            ret res(cx, C_int(u as int));
1143
        }
1144 1145 1146 1147 1148 1149 1150 1151 1152 1153 1154 1155 1156 1157 1158 1159 1160 1161 1162 1163 1164
        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));
        }
1165
        case (ast.lit_char(?c)) {
1166
            ret res(cx, C_integral(c as int, T_char()));
1167 1168
        }
        case (ast.lit_bool(?b)) {
1169 1170 1171 1172
            ret res(cx, C_bool(b));
        }
        case (ast.lit_nil) {
            ret res(cx, C_nil());
1173 1174 1175
        }
        case (ast.lit_str(?s)) {
            auto len = (_str.byte_len(s) as int) + 1;
1176
            auto sub = trans_upcall(cx, "upcall_new_str",
1177
                                    vec(p2i(C_str(cx.fcx.ccx, s)),
1178 1179
                                        C_int(len)));
            sub.val = sub.bcx.build.IntToPtr(sub.val,
1180
                                             T_ptr(T_str()));
1181
            auto t = node_ann_type(cx.fcx.ccx, ann);
1182
            find_scope_cx(cx).cleanups +=
1183
                clean(bind drop_ty(_, sub.val, t));
1184
            ret sub;
1185 1186 1187 1188
        }
    }
}

1189
fn target_type(@crate_ctxt cx, @ty.t t) -> @ty.t {
1190
    alt (t.struct) {
1191 1192
        case (ty.ty_int) {
            auto tm = ty.ty_machine(cx.sess.get_targ_cfg().int_type);
1193 1194
            ret @rec(struct=tm with *t);
        }
1195 1196
        case (ty.ty_uint) {
            auto tm = ty.ty_machine(cx.sess.get_targ_cfg().uint_type);
1197 1198
            ret @rec(struct=tm with *t);
        }
1199
        case (_) { /* fall through */ }
1200 1201 1202 1203
    }
    ret t;
}

1204
fn node_ann_type(@crate_ctxt cx, &ast.ann a) -> @ty.t {
1205 1206
    alt (a) {
        case (ast.ann_none) {
1207
            cx.sess.bug("missing type annotation");
1208 1209
        }
        case (ast.ann_type(?t)) {
1210
            ret target_type(cx, t);
1211 1212 1213 1214
        }
    }
}

1215 1216 1217 1218
fn node_type(@crate_ctxt cx, &ast.ann a) -> TypeRef {
    ret type_of(cx, node_ann_type(cx, a));
}

1219
impure fn trans_unary(@block_ctxt cx, ast.unop op,
1220
                      @ast.expr e, &ast.ann a) -> result {
1221 1222 1223

    auto sub = trans_expr(cx, e);

1224 1225
    alt (op) {
        case (ast.bitnot) {
1226 1227
            sub.val = cx.build.Not(sub.val);
            ret sub;
1228 1229
        }
        case (ast.not) {
1230 1231
            sub.val = cx.build.Not(sub.val);
            ret sub;
1232 1233 1234
        }
        case (ast.neg) {
            // FIXME: switch by signedness.
1235 1236
            sub.val = cx.build.Neg(sub.val);
            ret sub;
1237
        }
1238
        case (ast.box) {
1239
            auto e_ty = ty.expr_ty(e);
1240 1241 1242 1243
            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,
1244 1245
                                        vec(C_int(0),
                                            C_int(abi.box_rc_field_refcnt)));
1246 1247 1248 1249 1250 1251
            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);
1252
        }
1253 1254 1255 1256 1257
        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);
1258 1259
            if (ty.type_is_scalar(e_ty) ||
                ty.type_is_nil(e_ty)) {
1260 1261 1262
                sub.val = sub.bcx.build.Load(sub.val);
            }
            ret sub;
1263
        }
1264 1265 1266 1267
    }
    fail;
}

1268 1269 1270 1271 1272 1273 1274 1275 1276 1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303
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;
}

1304
impure fn trans_binary(@block_ctxt cx, ast.binop op,
1305
                       @ast.expr a, @ast.expr b) -> result {
1306

1307 1308 1309 1310 1311 1312 1313
    // First couple cases are lazy:

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

1314
            auto rhs_cx = new_scope_block_ctxt(cx, "rhs");
1315 1316
            auto rhs_res = trans_expr(rhs_cx, b);

1317
            auto lhs_false_cx = new_scope_block_ctxt(cx, "lhs false");
1318
            auto lhs_false_res = res(lhs_false_cx, C_bool(false));
1319 1320 1321

            lhs_res.bcx.build.CondBr(lhs_res.val,
                                     rhs_cx.llbb,
1322 1323 1324 1325
                                     lhs_false_cx.llbb);

            ret join_results(cx, T_bool(),
                             vec(lhs_false_res, rhs_res));
1326 1327 1328 1329 1330 1331
        }

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

1332
            auto rhs_cx = new_scope_block_ctxt(cx, "rhs");
1333 1334
            auto rhs_res = trans_expr(rhs_cx, b);

1335
            auto lhs_true_cx = new_scope_block_ctxt(cx, "lhs true");
1336
            auto lhs_true_res = res(lhs_true_cx, C_bool(true));
1337 1338

            lhs_res.bcx.build.CondBr(lhs_res.val,
1339
                                     lhs_true_cx.llbb,
1340
                                     rhs_cx.llbb);
1341 1342 1343

            ret join_results(cx, T_bool(),
                             vec(lhs_true_res, rhs_res));
1344
        }
1345 1346

        case (_) {
1347 1348 1349 1350 1351
            // 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));
1352
        }
1353 1354 1355 1356
    }
    fail;
}

1357 1358 1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387
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);
        }
1388 1389

        case (_) { /* fall through */ }
1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400
    }

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

1401
impure fn trans_if(@block_ctxt cx, @ast.expr cond,
1402
                   &ast.block thn, &option.t[ast.block] els) -> result {
1403 1404 1405

    auto cond_res = trans_expr(cx, cond);

1406
    auto then_cx = new_scope_block_ctxt(cx, "then");
1407 1408
    auto then_res = trans_block(then_cx, thn);

1409
    auto else_cx = new_scope_block_ctxt(cx, "else");
1410
    auto else_res = res(else_cx, C_nil());
1411 1412 1413

    alt (els) {
        case (some[ast.block](?eblk)) {
1414
            else_res = trans_block(else_cx, eblk);
1415
        }
1416
        case (_) { /* fall through */ }
1417 1418
    }

1419
    cond_res.bcx.build.CondBr(cond_res.val,
1420 1421
                              then_cx.llbb,
                              else_cx.llbb);
1422 1423 1424 1425

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

1428
impure fn trans_while(@block_ctxt cx, @ast.expr cond,
1429 1430
                      &ast.block body) -> result {

1431 1432
    auto cond_cx = new_scope_block_ctxt(cx, "while cond");
    auto body_cx = new_scope_block_ctxt(cx, "while loop body");
1433
    auto next_cx = new_sub_block_ctxt(cx, "next");
1434 1435

    auto body_res = trans_block(body_cx, body);
1436 1437 1438 1439 1440 1441 1442 1443
    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);
1444 1445 1446 1447
    ret res(next_cx, C_nil());
}

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

1450
    auto body_cx = new_scope_block_ctxt(cx, "do-while loop body");
1451
    auto next_cx = new_sub_block_ctxt(cx, "next");
1452 1453

    auto body_res = trans_block(body_cx, body);
1454 1455 1456 1457 1458 1459
    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);
1460 1461 1462
    ret res(next_cx, body_res.val);
}

P
Patrick Walton 已提交
1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489
// Pattern matching translation

// Returns a pointer to the union part of the LLVM representation of a tag
// type, cast to the appropriate type.
fn get_pat_union_ptr(@block_ctxt cx, vec[@ast.pat] subpats, ValueRef llval)
        -> ValueRef {
    auto llblobptr = cx.build.GEP(llval, vec(C_int(0), C_int(1)));

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

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

impure fn trans_pat_match(@block_ctxt cx, @ast.pat pat, ValueRef llval,
                          @block_ctxt next_cx) -> result {
    alt (pat.node) {
        case (ast.pat_wild(_)) { ret res(cx, llval); }
        case (ast.pat_bind(_, _, _)) { ret res(cx, llval); }
        case (ast.pat_tag(?id, ?subpats, ?vdef_opt, ?ann)) {
            auto lltagptr = cx.build.GEP(llval, vec(C_int(0), C_int(0)));
            auto lltag = cx.build.Load(lltagptr);
1490

P
Patrick Walton 已提交
1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564
            auto vdef = option.get[ast.variant_def](vdef_opt);
            auto variant_id = vdef._1;
            auto tinfo = cx.fcx.ccx.tags.get(vdef._0);
            auto variant_tag = 0;
            auto i = 0;
            for (tup(ast.def_id,arity) vinfo in tinfo.variants) {
                auto this_variant_id = vinfo._0;
                if (variant_id._0 == this_variant_id._0 &&
                        variant_id._1 == this_variant_id._1) {
                    variant_tag = i;
                }
                i += 1;
            }

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

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

            if (_vec.len[@ast.pat](subpats) > 0u) {
                auto llunionptr = get_pat_union_ptr(matched_cx, subpats,
                                                    llval);
                auto i = 0;
                for (@ast.pat subpat in subpats) {
                    auto llsubvalptr = matched_cx.build.GEP(llunionptr,
                                                            vec(C_int(0),
                                                                C_int(i)));
                    auto llsubval = load_non_structural(matched_cx,
                                                        llsubvalptr,
                                                        pat_ty(subpat));
                    auto subpat_res = trans_pat_match(matched_cx, subpat,
                                                      llsubval, next_cx);
                    matched_cx = subpat_res.bcx;
                }
            }

            ret res(matched_cx, llval);
        }
    }

    fail;
}

impure fn trans_pat_binding(@block_ctxt cx, @ast.pat pat, ValueRef llval)
        -> result {
    alt (pat.node) {
        case (ast.pat_wild(_)) { ret res(cx, llval); }
        case (ast.pat_bind(?id, ?def_id, ?ann)) {
            auto ty = node_ann_type(cx.fcx.ccx, ann);
            auto llty = type_of(cx.fcx.ccx, ty);

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

            ret copy_ty(cx, true, dst, llval, ty);
        }
        case (ast.pat_tag(_, ?subpats, _, _)) {
            if (_vec.len[@ast.pat](subpats) == 0u) { ret res(cx, llval); }

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

            auto this_cx = cx;
            auto i = 0;
            for (@ast.pat subpat in subpats) {
                auto llsubvalptr = this_cx.build.GEP(llunionptr,
                                                     vec(C_int(0), C_int(i)));
                auto llsubval = load_non_structural(this_cx, llsubvalptr,
                                                    pat_ty(subpat));
                auto subpat_res = trans_pat_binding(this_cx, subpat,
                                                    llsubval);
                this_cx = subpat_res.bcx;
1565
                i += 1;
P
Patrick Walton 已提交
1566 1567 1568 1569 1570 1571 1572 1573 1574 1575 1576 1577 1578 1579 1580 1581 1582 1583 1584 1585 1586 1587 1588 1589 1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605 1606 1607
            }

            ret res(this_cx, llval);
        }
    }
}

impure fn trans_alt(@block_ctxt cx, @ast.expr expr, vec[ast.arm] arms)
        -> result {
    auto expr_res = trans_expr(cx, expr);

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

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

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

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

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

        this_cx = next_cx;
    }

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

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

1608 1609 1610
// 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).
1611

1612 1613 1614 1615 1616 1617 1618 1619
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)),
1620
                            true);
1621 1622 1623 1624 1625
                }
                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 已提交
1626
                }
P
Patrick Walton 已提交
1627 1628 1629 1630
                case (ast.def_binding(?did)) {
                    check (cx.fcx.lllocals.contains_key(did));
                    ret tup(res(cx, cx.fcx.lllocals.get(did)), true);
                }
1631
                case (ast.def_fn(?did)) {
1632 1633
                    check (cx.fcx.ccx.item_ids.contains_key(did));
                    ret tup(res(cx, cx.fcx.ccx.item_ids.get(did)),
1634 1635
                            false);
                }
1636 1637 1638 1639 1640
                case (ast.def_obj(?did)) {
                    check (cx.fcx.ccx.item_ids.contains_key(did));
                    ret tup(res(cx, cx.fcx.ccx.item_ids.get(did)),
                            false);
                }
1641 1642
                case (ast.def_variant(?tid, ?vid)) {
                    check (cx.fcx.ccx.tags.contains_key(tid));
1643 1644 1645
                    check (cx.fcx.ccx.item_ids.contains_key(vid));
                    ret tup(res(cx, cx.fcx.ccx.item_ids.get(vid)),
                            false);
1646
                }
1647 1648
                case (_) {
                    cx.fcx.ccx.sess.unimpl("def variant in trans");
G
Graydon Hoare 已提交
1649 1650 1651
                }
            }
        }
1652 1653 1654 1655 1656 1657 1658
        case (none[ast.def]) {
            cx.fcx.ccx.sess.err("unresolved expr_name in trans");
        }
    }
    fail;
}

1659 1660
impure fn trans_field(@block_ctxt cx, &ast.span sp, @ast.expr base,
                      &ast.ident field, &ast.ann ann) -> tup(result, bool) {
1661 1662
    auto lv = trans_lval(cx, base);
    auto r = lv._0;
1663 1664 1665 1666
    auto t = ty.expr_ty(base);
    alt (t.struct) {
        case (ty.ty_tup(?fields)) {
            let uint ix = ty.field_num(cx.fcx.ccx.sess, sp, field);
1667
            auto v = r.bcx.build.GEP(r.val, vec(C_int(0), C_int(ix as int)));
1668
            ret tup(res(r.bcx, v), true);
1669
        }
1670 1671
        case (ty.ty_rec(?fields)) {
            let uint ix = ty.field_idx(cx.fcx.ccx.sess, sp, field, fields);
1672
            auto v = r.bcx.build.GEP(r.val, vec(C_int(0), C_int(ix as int)));
1673 1674
            ret tup(res(r.bcx, v), true);
        }
1675 1676
        case (ty.ty_obj(?methods)) {
            let uint ix = ty.method_idx(cx.fcx.ccx.sess, sp, field, methods);
1677 1678 1679 1680 1681 1682 1683
            auto vtbl = r.bcx.build.GEP(r.val,
                                        vec(C_int(0),
                                            C_int(abi.obj_field_vtbl)));
            vtbl = r.bcx.build.Load(vtbl);
            auto v =  r.bcx.build.GEP(vtbl, vec(C_int(0),
                                                C_int(ix as int)));
            ret tup(res(r.bcx, v), true);
1684
        }
1685
        case (_) { cx.fcx.ccx.sess.unimpl("field variant in trans_field"); }
1686 1687 1688 1689
    }
    fail;
}

1690 1691
impure fn trans_index(@block_ctxt cx, &ast.span sp, @ast.expr base,
                      @ast.expr idx, &ast.ann ann) -> tup(result, bool) {
1692

G
Graydon Hoare 已提交
1693 1694 1695
    auto lv = trans_expr(cx, base);
    auto ix = trans_expr(lv.bcx, idx);
    auto v = lv.val;
1696 1697

    auto llunit_ty = node_type(cx.fcx.ccx, ann);
1698
    auto unit_sz = size_of(llunit_ty);
1699 1700 1701 1702
    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);
1703

1704 1705 1706 1707 1708 1709 1710 1711 1712 1713 1714 1715 1716 1717 1718 1719 1720
    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 已提交
1721
    ret tup(res(next_cx, elt), true);
1722 1723
}

1724 1725 1726 1727
// 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).

1728
impure fn trans_lval(@block_ctxt cx, @ast.expr e) -> tup(result, bool) {
1729 1730 1731 1732 1733 1734 1735
    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);
        }
1736 1737 1738
        case (ast.expr_index(?base, ?idx, ?ann)) {
            ret trans_index(cx, e.span, base, idx, ann);
        }
1739
        case (_) { cx.fcx.ccx.sess.unimpl("expr variant in trans_lval"); }
G
Graydon Hoare 已提交
1740 1741 1742 1743
    }
    fail;
}

1744
impure fn trans_cast(@block_ctxt cx, @ast.expr e, &ast.ann ann) -> result {
1745 1746 1747 1748
    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);
1749
    if (!ty.type_is_fp(t)) {
1750 1751
        if (llvm.LLVMGetIntTypeWidth(lldsttype) >
            llvm.LLVMGetIntTypeWidth(llsrctype)) {
1752
            if (ty.type_is_signed(t)) {
1753 1754 1755 1756 1757 1758 1759 1760 1761 1762 1763 1764 1765 1766 1767 1768 1769 1770 1771 1772 1773 1774
                // 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;
}

1775

1776
impure fn trans_args(@block_ctxt cx, &vec[@ast.expr] es, @ty.t fn_ty)
1777 1778 1779 1780
    -> tup(@block_ctxt, vec[ValueRef]) {
    let vec[ValueRef] vs = vec(cx.fcx.lltaskptr);
    let @block_ctxt bcx = cx;

1781
    let vec[ty.arg] args = vec();   // FIXME: typestate bug
1782
    alt (fn_ty.struct) {
1783
        case (ty.ty_fn(?a, _)) { args = a; }
1784 1785 1786 1787
        case (_) { fail; }
    }

    auto i = 0u;
1788
    for (@ast.expr e in es) {
1789 1790 1791
        auto mode = args.(i).mode;

        auto re;
1792
        if (ty.type_is_structural(ty.expr_ty(e))) {
1793 1794 1795 1796 1797 1798 1799 1800 1801
            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;
1802
                if (ty.is_lval(e)) {
1803 1804 1805 1806 1807 1808 1809 1810 1811 1812 1813 1814 1815 1816 1817 1818 1819
                    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);
            }
1820
        }
1821 1822 1823 1824 1825

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

        i += 1u;
1826 1827 1828 1829 1830
    }

    ret tup(bcx, vs);
}

1831
impure fn trans_call(@block_ctxt cx, @ast.expr f,
G
Graydon Hoare 已提交
1832
                     vec[@ast.expr] args, &ast.ann ann) -> result {
1833
    auto f_res = trans_lval(cx, f);
1834 1835 1836 1837
    auto faddr = f_res._0.val;
    if (f_res._1) {
        faddr = f_res._0.bcx.build.Load(faddr);
    }
1838 1839
    auto fn_ty = ty.expr_ty(f);
    auto ret_ty = ty.ann_to_type(ann);
1840
    auto args_res = trans_args(f_res._0.bcx, args, fn_ty);
G
Graydon Hoare 已提交
1841

1842
    auto real_retval = args_res._0.build.FastCall(faddr, args_res._1);
1843
    auto retval;
1844
    if (ty.type_is_nil(ret_ty)) {
1845 1846 1847 1848
        retval = C_nil();
    } else {
        retval = real_retval;
    }
G
Graydon Hoare 已提交
1849

G
Graydon Hoare 已提交
1850 1851 1852
    // 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.
1853
    if (ty.type_is_structural(ret_ty)) {
G
Graydon Hoare 已提交
1854 1855 1856 1857 1858
        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 已提交
1859 1860 1861 1862 1863 1864
    // 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);
1865 1866
}

1867
impure fn trans_tup(@block_ctxt cx, vec[ast.elt] elts,
G
Graydon Hoare 已提交
1868
                    &ast.ann ann) -> result {
1869 1870
    auto t = node_ann_type(cx.fcx.ccx, ann);
    auto llty = type_of(cx.fcx.ccx, t);
1871
    auto tup_val = cx.build.Alloca(llty);
1872
    find_scope_cx(cx).cleanups += clean(bind drop_ty(_, tup_val, t));
G
Graydon Hoare 已提交
1873 1874
    let int i = 0;
    auto r = res(cx, C_nil());
1875
    for (ast.elt e in elts) {
1876
        auto t = ty.expr_ty(e.expr);
1877
        auto src_res = trans_expr(r.bcx, e.expr);
G
Graydon Hoare 已提交
1878 1879 1880 1881 1882 1883 1884
        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 已提交
1885 1886
impure fn trans_vec(@block_ctxt cx, vec[@ast.expr] args,
                    &ast.ann ann) -> result {
1887 1888 1889 1890
    auto t = node_ann_type(cx.fcx.ccx, ann);
    auto unit_ty = t;
    alt (t.struct) {
        case (ty.ty_vec(?t)) {
G
Graydon Hoare 已提交
1891 1892 1893 1894 1895 1896 1897 1898
            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);
1899
    auto unit_sz = size_of(llunit_ty);
G
Graydon Hoare 已提交
1900 1901 1902 1903 1904 1905
    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)));

1906
    auto llty = type_of(cx.fcx.ccx, t);
G
Graydon Hoare 已提交
1907
    auto vec_val = sub.bcx.build.IntToPtr(sub.val, llty);
1908
    find_scope_cx(cx).cleanups += clean(bind drop_ty(_, vec_val, t));
G
Graydon Hoare 已提交
1909 1910 1911 1912 1913 1914 1915 1916 1917 1918

    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;
    }
1919 1920 1921 1922
    auto fill = sub.bcx.build.GEP(vec_val,
                                  vec(C_int(0), C_int(abi.vec_elt_fill)));
    sub.bcx.build.Store(data_sz, fill);

G
Graydon Hoare 已提交
1923 1924 1925
    ret res(sub.bcx, vec_val);
}

1926
impure fn trans_rec(@block_ctxt cx, vec[ast.field] fields,
1927
                    &ast.ann ann) -> result {
1928 1929
    auto t = node_ann_type(cx.fcx.ccx, ann);
    auto llty = type_of(cx.fcx.ccx, t);
1930
    auto rec_val = cx.build.Alloca(llty);
1931
    find_scope_cx(cx).cleanups += clean(bind drop_ty(_, rec_val, t));
1932 1933
    let int i = 0;
    auto r = res(cx, C_nil());
1934
    for (ast.field f in fields) {
1935
        auto t = ty.expr_ty(f.expr);
1936 1937
        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)));
1938 1939 1940 1941
        // FIXME: calculate copy init-ness in typestate.
        r = copy_ty(src_res.bcx, true, dst_elt, src_res.val, t);
        i += 1;
    }
1942
    ret res(r.bcx, rec_val);
1943 1944
}

G
Graydon Hoare 已提交
1945

G
Graydon Hoare 已提交
1946

1947
impure fn trans_expr(@block_ctxt cx, @ast.expr e) -> result {
1948
    alt (e.node) {
1949 1950
        case (ast.expr_lit(?lit, ?ann)) {
            ret trans_lit(cx, *lit, ann);
1951 1952
        }

1953
        case (ast.expr_unary(?op, ?x, ?ann)) {
1954
            ret trans_unary(cx, op, x, ann);
1955 1956
        }

P
Patrick Walton 已提交
1957
        case (ast.expr_binary(?op, ?x, ?y, _)) {
1958
            ret trans_binary(cx, op, x, y);
1959
        }
1960

P
Patrick Walton 已提交
1961
        case (ast.expr_if(?cond, ?thn, ?els, _)) {
1962
            ret trans_if(cx, cond, thn, els);
1963 1964
        }

1965
        case (ast.expr_while(?cond, ?body, _)) {
1966
            ret trans_while(cx, cond, body);
1967 1968
        }

1969
        case (ast.expr_do_while(?body, ?cond, _)) {
1970
            ret trans_do_while(cx, body, cond);
1971 1972
        }

P
Patrick Walton 已提交
1973 1974 1975 1976
        case (ast.expr_alt(?expr, ?arms, _)) {
            ret trans_alt(cx, expr, arms);
        }

P
Patrick Walton 已提交
1977
        case (ast.expr_block(?blk, _)) {
1978
            auto sub_cx = new_scope_block_ctxt(cx, "block-expr body");
1979
            auto next_cx = new_sub_block_ctxt(cx, "next");
1980 1981 1982 1983 1984 1985 1986
            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);
        }
1987

1988
        case (ast.expr_assign(?dst, ?src, ?ann)) {
1989
            auto lhs_res = trans_lval(cx, dst);
1990
            check (lhs_res._1);
1991
            auto rhs_res = trans_expr(lhs_res._0.bcx, src);
1992
            auto t = node_ann_type(cx.fcx.ccx, ann);
G
Graydon Hoare 已提交
1993
            // FIXME: calculate copy init-ness in typestate.
1994
            ret copy_ty(rhs_res.bcx, false, lhs_res._0.val, rhs_res.val, t);
1995
        }
G
Graydon Hoare 已提交
1996

1997 1998 1999 2000 2001 2002 2003 2004 2005 2006 2007 2008
        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 已提交
2009 2010
        case (ast.expr_call(?f, ?args, ?ann)) {
            ret trans_call(cx, f, args, ann);
2011 2012
        }

2013
        case (ast.expr_cast(?e, _, ?ann)) {
2014
            ret trans_cast(cx, e, ann);
2015
        }
G
Graydon Hoare 已提交
2016

G
Graydon Hoare 已提交
2017 2018 2019 2020
        case (ast.expr_vec(?args, ?ann)) {
            ret trans_vec(cx, args, ann);
        }

G
Graydon Hoare 已提交
2021 2022 2023
        case (ast.expr_tup(?args, ?ann)) {
            ret trans_tup(cx, args, ann);
        }
G
Graydon Hoare 已提交
2024

2025 2026 2027 2028
        case (ast.expr_rec(?args, ?ann)) {
            ret trans_rec(cx, args, ann);
        }

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

2032
        case (_) {
2033
            auto t = ty.expr_ty(e);
2034
            auto sub = trans_lval(cx, e);
2035 2036
            ret res(sub._0.bcx,
                    load_non_structural(sub._0.bcx, sub._0.val, t));
2037
        }
2038
    }
2039
    cx.fcx.ccx.sess.unimpl("expr variant in trans_expr");
2040 2041 2042
    fail;
}

2043 2044 2045 2046 2047 2048
// 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,
2049 2050
                       @ty.t t) -> ValueRef {
    if (ty.type_is_structural(t)) {
2051 2052 2053 2054 2055 2056
        ret v;
    } else {
        ret cx.build.Load(v);
    }
}

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

2059
    auto sub = trans_expr(cx, e);
2060
    auto e_ty = ty.expr_ty(e);
2061
    alt (e_ty.struct) {
2062
        case (ty.ty_str) {
2063 2064 2065 2066
            auto v = sub.bcx.build.PtrToInt(sub.val, T_int());
            ret trans_upcall(sub.bcx,
                             "upcall_log_str",
                             vec(v));
2067 2068
        }
        case (_) {
2069 2070 2071
            ret trans_upcall(sub.bcx,
                             "upcall_log_int",
                             vec(sub.val));
2072 2073
        }
    }
2074
    fail;
2075 2076
}

2077
impure fn trans_check_expr(@block_ctxt cx, @ast.expr e) -> result {
2078 2079 2080
    auto cond_res = trans_expr(cx, e);

    // FIXME: need pretty-printer.
2081 2082
    auto V_expr_str = p2i(C_str(cx.fcx.ccx, "<expr>"));
    auto V_filename = p2i(C_str(cx.fcx.ccx, e.span.filename));
2083 2084 2085
    auto V_line = e.span.lo.line as int;
    auto args = vec(V_expr_str, V_filename, C_int(V_line));

2086
    auto fail_cx = new_sub_block_ctxt(cx, "fail");
2087 2088
    auto fail_res = trans_upcall(fail_cx, "upcall_fail", args);

2089
    auto next_cx = new_sub_block_ctxt(cx, "next");
2090 2091 2092 2093 2094 2095 2096
    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());
}

2097
impure fn trans_ret(@block_ctxt cx, &option.t[@ast.expr] e) -> result {
2098 2099 2100
    auto r = res(cx, C_nil());
    alt (e) {
        case (some[@ast.expr](?x)) {
2101
            auto t = ty.expr_ty(x);
2102
            r = trans_expr(cx, x);
2103 2104 2105 2106 2107

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

2108
            if (ty.type_is_structural(t)) {
G
Graydon Hoare 已提交
2109 2110 2111 2112 2113 2114
                // 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);
            }
2115
        }
2116
        case (_) { /* fall through */  }
2117 2118
    }

2119 2120
    // Run all cleanups and back out.
    let bool more_cleanups = true;
2121
    auto cleanup_cx = cx;
2122
    while (more_cleanups) {
2123 2124
        r.bcx = trans_block_cleanups(r.bcx, cleanup_cx);
        alt (cleanup_cx.parent) {
2125
            case (parent_some(?b)) {
2126
                cleanup_cx = b;
2127 2128 2129 2130 2131 2132 2133
            }
            case (parent_none) {
                more_cleanups = false;
            }
        }
    }

2134
    alt (e) {
2135
        case (some[@ast.expr](?ex)) {
2136
            if (ty.type_is_nil(ty.expr_ty(ex))) {
2137 2138 2139 2140 2141
                r.bcx.build.RetVoid();
                r.val = C_nil();
            } else {
                r.val = r.bcx.build.Ret(r.val);
            }
2142 2143
            ret r;
        }
2144
        case (_) { /* fall through */  }
2145 2146 2147 2148
    }

    // FIXME: until LLVM has a unit type, we are moving around
    // C_nil values rather than their void type.
2149 2150
    r.bcx.build.RetVoid();
    r.val = C_nil();
2151 2152 2153
    ret r;
}

2154
impure fn trans_stmt(@block_ctxt cx, &ast.stmt s) -> result {
2155
    auto sub = res(cx, C_nil());
2156
    alt (s.node) {
2157
        case (ast.stmt_log(?a)) {
2158
            sub.bcx = trans_log(cx, a).bcx;
2159 2160
        }

2161
        case (ast.stmt_check_expr(?a)) {
2162
            sub.bcx = trans_check_expr(cx, a).bcx;
2163 2164
        }

2165 2166 2167 2168
        case (ast.stmt_ret(?e)) {
            sub.bcx = trans_ret(cx, e).bcx;
        }

2169
        case (ast.stmt_expr(?e)) {
2170
            sub.bcx = trans_expr(cx, e).bcx;
2171
        }
2172

2173 2174 2175
        case (ast.stmt_decl(?d)) {
            alt (d.node) {
                case (ast.decl_local(?local)) {
2176 2177 2178 2179 2180

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

2184 2185
                    alt (local.init) {
                        case (some[@ast.expr](?e)) {
2186
                            sub = trans_expr(cx, e);
2187
                            sub = copy_ty(sub.bcx, true, llptr, sub.val, ty);
2188
                        }
2189 2190 2191 2192 2193
                        case (_) {
                            auto llty = type_of(cx.fcx.ccx, ty);
                            auto null = lib.llvm.llvm.LLVMConstNull(llty);
                            sub = res(cx, cx.build.Store(null, llptr));
                        }
2194 2195 2196 2197
                    }
                }
            }
        }
2198
        case (_) {
2199
            cx.fcx.ccx.sess.unimpl("stmt variant");
2200 2201
        }
    }
2202
    ret sub;
2203 2204
}

2205
fn new_builder(BasicBlockRef llbb) -> builder {
2206 2207 2208 2209 2210
    let BuilderRef llbuild = llvm.LLVMCreateBuilder();
    llvm.LLVMPositionBuilderAtEnd(llbuild, llbb);
    ret builder(llbuild);
}

2211 2212
// You probably don't want to use this one. See the
// next three functions instead.
2213
fn new_block_ctxt(@fn_ctxt cx, block_parent parent,
2214
                  bool is_scope,
2215
                  str name) -> @block_ctxt {
2216
    let vec[cleanup] cleanups = vec();
2217
    let BasicBlockRef llbb =
2218
        llvm.LLVMAppendBasicBlock(cx.llfn,
2219
                                  _str.buf(cx.ccx.names.next(name)));
2220

2221
    ret @rec(llbb=llbb,
2222
             build=new_builder(llbb),
2223
             parent=parent,
2224
             is_scope=is_scope,
2225
             mutable cleanups=cleanups,
2226 2227 2228
             fcx=cx);
}

2229 2230
// 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 {
2231 2232
    ret new_block_ctxt(fcx, parent_none, true, "function top level");
}
2233

2234 2235 2236
// 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);
2237 2238
}

2239
// Use this when you're making a general CFG BB within a scope.
2240
fn new_sub_block_ctxt(@block_ctxt bcx, str n) -> @block_ctxt {
2241
    ret new_block_ctxt(bcx.fcx, parent_some(bcx), false, n);
2242
}
2243

2244

2245 2246
fn trans_block_cleanups(@block_ctxt cx,
                        @block_ctxt cleanup_cx) -> @block_ctxt {
2247
    auto bcx = cx;
2248 2249 2250 2251 2252

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

2253
    for (cleanup c in cleanup_cx.cleanups) {
2254
        alt (c) {
2255
            case (clean(?cfn)) {
2256
                bcx = cfn(bcx).bcx;
2257 2258 2259
            }
        }
    }
2260 2261 2262
    ret bcx;
}

2263 2264 2265 2266 2267 2268 2269 2270 2271 2272
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;
                    }
2273
                    case (_) { /* fall through */ }
2274 2275
                }
            }
2276
            case (_) { /* fall through */ }
2277 2278 2279 2280
        }
    }
}

2281
impure fn trans_block(@block_ctxt cx, &ast.block b) -> result {
2282 2283
    auto bcx = cx;

2284
    for each (@ast.local local in block_locals(b)) {
2285
        auto ty = node_type(cx.fcx.ccx, local.ann);
2286 2287 2288
        auto val = bcx.build.Alloca(ty);
        cx.fcx.lllocals.insert(local.id, val);
    }
2289
    auto r = res(bcx, C_nil());
2290

2291
    for (@ast.stmt s in b.node.stmts) {
2292 2293
        r = trans_stmt(bcx, *s);
        bcx = r.bcx;
2294 2295 2296 2297 2298
        // 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;
        }
2299
    }
2300

2301 2302 2303 2304 2305 2306 2307 2308 2309 2310 2311 2312 2313
    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());
        }
    }

2314
    bcx = trans_block_cleanups(bcx, find_scope_cx(bcx));
2315
    ret res(bcx, r.val);
2316 2317
}

2318
fn new_fn_ctxt(@crate_ctxt cx,
2319
               str name,
2320
               ValueRef llfndecl) -> @fn_ctxt {
2321

2322
    let ValueRef lltaskptr = llvm.LLVMGetParam(llfndecl, 0u);
2323

2324
    let hashmap[ast.def_id, ValueRef] lllocals = new_def_hash[ValueRef]();
2325
    let hashmap[ast.def_id, ValueRef] llargs = new_def_hash[ValueRef]();
2326
    let hashmap[ast.def_id, ValueRef] lltydescs = new_def_hash[ValueRef]();
2327

2328
    ret @rec(llfn=llfndecl,
2329
             lltaskptr=lltaskptr,
2330
             llargs=llargs,
2331
             lllocals=lllocals,
2332
             lltydescs=lltydescs,
2333
             ccx=cx);
2334 2335
}

2336

2337 2338
fn create_llargs_for_fn_args(&@fn_ctxt cx, &vec[ast.arg] args,
                             &vec[ast.ty_param] ty_params) {
2339
    let uint arg_n = 1u;
2340 2341 2342 2343 2344 2345 2346
    for (ast.ty_param tp in ty_params) {
        auto llarg = llvm.LLVMGetParam(cx.llfn, arg_n);
        check (llarg as int != 0);
        cx.lltydescs.insert(tp.id, llarg);
        arg_n += 1u;
    }

2347 2348 2349 2350 2351 2352 2353 2354 2355
    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;
    }
}


2356 2357 2358 2359
// 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.

2360
fn copy_args_to_allocas(@block_ctxt cx, vec[ast.arg] args,
2361
                        vec[ty.arg] arg_tys) {
2362 2363 2364

    let uint arg_n = 0u;

2365
    for (ast.arg aarg in args) {
2366 2367 2368 2369 2370 2371 2372 2373 2374
        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);
        }

2375 2376 2377 2378
        arg_n += 1u;
    }
}

2379 2380 2381 2382 2383
fn is_terminated(@block_ctxt cx) -> bool {
    auto inst = llvm.LLVMGetLastInstruction(cx.llbb);
    ret llvm.LLVMIsATerminatorInst(inst) as int != 0;
}

2384 2385 2386
fn arg_tys_of_fn(ast.ann ann) -> vec[ty.arg] {
    alt (ty.ann_to_type(ann).struct) {
        case (ty.ty_fn(?arg_tys, _)) {
2387 2388 2389 2390 2391 2392
            ret arg_tys;
        }
    }
    fail;
}

2393 2394 2395
fn ret_ty_of_fn(ast.ann ann) -> @ty.t {
    alt (ty.ann_to_type(ann).struct) {
        case (ty.ty_fn(_, ?ret_ty)) {
2396 2397 2398 2399 2400 2401
            ret ret_ty;
        }
    }
    fail;
}

2402
impure fn trans_fn(@crate_ctxt cx, &ast._fn f, ast.def_id fid,
2403
                   &vec[ast.ty_param] ty_params, &ast.ann ann) {
2404

2405 2406 2407
    auto llfndecl = cx.item_ids.get(fid);
    cx.item_names.insert(cx.path, llfndecl);

2408
    auto fcx = new_fn_ctxt(cx, cx.path, llfndecl);
2409
    create_llargs_for_fn_args(fcx, f.inputs, ty_params);
2410

2411
    auto bcx = new_top_block_ctxt(fcx);
2412

2413
    copy_args_to_allocas(bcx, f.inputs, arg_tys_of_fn(ann));
2414

2415 2416
    auto res = trans_block(bcx, f.body);
    if (!is_terminated(res.bcx)) {
2417 2418
        // FIXME: until LLVM has a unit type, we are moving around
        // C_nil values rather than their void type.
2419
        res.bcx.build.RetVoid();
2420
    }
2421 2422
}

2423 2424
impure fn trans_vtbl(@crate_ctxt cx, &ast._obj ob,
                     &vec[ast.ty_param] ty_params) -> ValueRef {
G
Graydon Hoare 已提交
2425 2426
    let vec[ValueRef] methods = vec();
    for (@ast.method m in ob.methods) {
2427 2428 2429 2430 2431 2432

        auto llfnty = node_type(cx, m.node.ann);
        let str s = cx.names.next("_rust_method") + "." + cx.path;
        let ValueRef llfn = decl_fastcall_fn(cx.llmod, s, llfnty);
        cx.item_ids.insert(m.node.id, llfn);

2433
        trans_fn(cx, m.node.meth, m.node.id, ty_params, m.node.ann);
2434
        methods += llfn;
G
Graydon Hoare 已提交
2435
    }
2436 2437 2438 2439 2440 2441 2442
    auto vtbl = C_struct(methods);
    auto gvar = llvm.LLVMAddGlobal(cx.llmod,
                                   val_ty(vtbl),
                                   _str.buf("_rust_vtbl" + "." + cx.path));
    llvm.LLVMSetInitializer(gvar, vtbl);
    llvm.LLVMSetGlobalConstant(gvar, True);
    ret gvar;
G
Graydon Hoare 已提交
2443 2444
}

2445
impure fn trans_obj(@crate_ctxt cx, &ast._obj ob, ast.def_id oid,
2446
                    &vec[ast.ty_param] ty_params, &ast.ann ann) {
2447 2448 2449 2450

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

2451
    // Translate obj ctor args to function arguments.
2452 2453 2454 2455 2456 2457 2458 2459 2460
    let vec[ast.arg] fn_args = vec();
    for (ast.obj_field f in ob.fields) {
        fn_args += vec(rec(mode=ast.alias,
                           ty=f.ty,
                           ident=f.ident,
                           id=f.id));
    }

    auto fcx = new_fn_ctxt(cx, cx.path, llctor_decl);
2461
    create_llargs_for_fn_args(fcx, fn_args, ty_params);
2462 2463 2464

    auto bcx = new_top_block_ctxt(fcx);

2465 2466
    let vec[ty.arg] arg_tys = arg_tys_of_fn(ann);
    copy_args_to_allocas(bcx, fn_args, arg_tys);
2467 2468

    auto pair = bcx.build.Alloca(type_of(cx, ret_ty_of_fn(ann)));
2469
    auto vtbl = trans_vtbl(cx, ob, ty_params);
2470 2471 2472
    auto pair_vtbl = bcx.build.GEP(pair,
                                   vec(C_int(0),
                                       C_int(abi.obj_field_vtbl)));
2473 2474 2475
    auto pair_box = bcx.build.GEP(pair,
                                  vec(C_int(0),
                                      C_int(abi.obj_field_box)));
2476
    bcx.build.Store(vtbl, pair_vtbl);
2477

2478 2479 2480 2481 2482 2483 2484 2485 2486 2487 2488 2489 2490 2491 2492 2493 2494 2495 2496 2497 2498 2499 2500 2501 2502
    let TypeRef llbox_ty = T_ptr(T_box(T_nil()));
    if (_vec.len[ty.arg](arg_tys) == 0u) {
        // Store null into pair, if no args.
        bcx.build.Store(C_null(llbox_ty), pair_box);
    } else {
        // Malloc a box for the body and copy args in.
        let vec[@ty.t] obj_fields = vec();
        for (ty.arg a in arg_tys) {
            append[@ty.t](obj_fields, a.ty);
        }
        // Synthesize an obj body:
        let @ty.t fields_ty = ty.plain_ty(ty.ty_tup(obj_fields));
        let TypeRef llfields_ty = type_of(bcx.fcx.ccx, fields_ty);
        let TypeRef llobj_body_ty =
            T_ptr(T_box(T_struct(vec(T_tydesc(),
                                     llfields_ty))));
        auto r = trans_malloc_inner(bcx, llobj_body_ty);
        auto box = r.val;
        auto rc = r.bcx.build.GEP(box,
                                  vec(C_int(0),
                                      C_int(abi.box_rc_field_refcnt)));
        auto body = r.bcx.build.GEP(box,
                                    vec(C_int(0),
                                        C_int(abi.box_rc_field_body)));
        r.bcx.build.Store(C_int(1), rc);
2503

2504 2505 2506 2507 2508
        // FIXME: Copy args into body

        auto p = r.bcx.build.PointerCast(box, llbox_ty);
        r.bcx.build.Store(p, pair_box);
    }
2509
    bcx.build.Ret(bcx.build.Load(pair));
2510 2511
}

2512
fn trans_tag_variant(@crate_ctxt cx, ast.def_id tag_id,
2513 2514
                     &ast.variant variant, int index,
                     &vec[ast.ty_param] ty_params) {
2515
    if (_vec.len[ast.variant_arg](variant.args) == 0u) {
2516 2517 2518
        ret;    // nullary constructors are just constants
    }

2519 2520 2521 2522 2523 2524 2525 2526 2527 2528
    // 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));
    }

2529
    auto var_ty = ty.ann_to_type(variant.ann);
2530 2531 2532 2533 2534 2535
    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);

2536 2537 2538
    let ValueRef llfndecl = cx.item_ids.get(variant.id);
    cx.item_names.insert(cx.path, llfndecl);

2539
    auto fcx = new_fn_ctxt(cx, cx.path, llfndecl);
2540
    create_llargs_for_fn_args(fcx, fn_args, ty_params);
2541

2542 2543 2544 2545 2546 2547 2548 2549 2550 2551 2552 2553 2554 2555 2556 2557 2558 2559 2560 2561
    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();
2562
    for (ty.arg arg in arg_tys) {
2563 2564 2565 2566 2567 2568 2569 2570
        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) {
2571
        auto llargval = bcx.build.Load(fcx.llargs.get(va.id));
2572 2573 2574 2575 2576 2577 2578 2579 2580 2581
        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);
2582 2583
}

2584
impure fn trans_item(@crate_ctxt cx, &ast.item item) {
2585
    alt (item.node) {
2586
        case (ast.item_fn(?name, ?f, ?tps, ?fid, ?ann)) {
2587
            auto sub_cx = @rec(path=cx.path + "." + name with *cx);
2588
            trans_fn(sub_cx, f, fid, tps, ann);
2589
        }
2590
        case (ast.item_obj(?name, ?ob, ?tps, ?oid, ?ann)) {
2591
            auto sub_cx = @rec(path=cx.path + "." + name with *cx);
2592
            trans_obj(sub_cx, ob, oid, tps, ann);
2593
        }
2594 2595
        case (ast.item_mod(?name, ?m, _)) {
            auto sub_cx = @rec(path=cx.path + "." + name with *cx);
2596
            trans_mod(sub_cx, m);
2597
        }
2598
        case (ast.item_tag(?name, ?variants, ?tps, ?tag_id)) {
2599
            auto sub_cx = @rec(path=cx.path + "." + name with *cx);
2600
            auto i = 0;
2601
            for (ast.variant variant in variants) {
2602
                trans_tag_variant(sub_cx, tag_id, variant, i, tps);
2603
                i += 1;
2604 2605
            }
        }
2606
        case (_) { /* fall through */ }
2607 2608 2609
    }
}

2610
impure fn trans_mod(@crate_ctxt cx, &ast._mod m) {
2611 2612
    for (@ast.item item in m.items) {
        trans_item(cx, *item);
2613 2614 2615
    }
}

2616

2617
fn collect_item(&@crate_ctxt cx, @ast.item i) -> @crate_ctxt {
2618
    alt (i.node) {
2619 2620
        case (ast.item_fn(?name, ?f, _, ?fid, ?ann)) {
            // TODO: type-params
2621 2622
            cx.items.insert(fid, i);
            auto llty = node_type(cx, ann);
2623
            let str s = cx.names.next("_rust_fn") + "." + name;
2624
            let ValueRef llfn = decl_fastcall_fn(cx.llmod, s, llty);
2625
            cx.item_ids.insert(fid, llfn);
2626 2627
        }

2628 2629 2630 2631 2632 2633 2634 2635 2636
        case (ast.item_obj(?name, ?ob, _, ?oid, ?ann)) {
            // TODO: type-params
            cx.items.insert(oid, i);
            auto llty = node_type(cx, ann);
            let str s = cx.names.next("_rust_obj_ctor") + "." + name;
            let ValueRef llfn = decl_fastcall_fn(cx.llmod, s, llty);
            cx.item_ids.insert(oid, llfn);
        }

2637 2638 2639 2640
        case (ast.item_const(?name, _, _, ?cid, _)) {
            cx.items.insert(cid, i);
        }

2641 2642 2643
        case (ast.item_mod(?name, ?m, ?mid)) {
            cx.items.insert(mid, i);
        }
2644 2645 2646 2647

        case (ast.item_tag(_, ?variants, _, ?tag_id)) {
            auto vi = new_def_hash[uint]();
            auto navi = new_def_hash[uint]();
2648 2649
            let vec[tup(ast.def_id,arity)] variant_info = vec();
            cx.tags.insert(tag_id, @rec(th=mk_type_handle(),
2650 2651
                                        mutable variants=variant_info,
                                        mutable size=0u));
2652
            cx.items.insert(tag_id, i);
2653 2654
        }

2655
        case (_) { /* fall through */ }
2656 2657 2658 2659 2660
    }
    ret cx;
}


2661
fn collect_items(@crate_ctxt cx, @ast.crate crate) {
2662

2663 2664
    let fold.ast_fold[@crate_ctxt] fld =
        fold.new_identity_fold[@crate_ctxt]();
2665 2666 2667 2668

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

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

2672 2673 2674 2675 2676 2677
// 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)) {
2678 2679
            auto max_align = 0u;
            auto max_size = 0u;
2680 2681 2682 2683 2684 2685

            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;
2686
                if (_vec.len[ast.variant_arg](variant.args) > 0u) {
2687 2688 2689 2690 2691 2692 2693
                    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; }
2694 2695 2696 2697 2698 2699 2700 2701 2702 2703

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

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

            info.variants = variant_info;
2704
            info.size = max_size;
2705

2706 2707 2708
            // FIXME: alignment is wrong here, manually insert padding I
            // guess :(
            auto tag_ty = T_struct(vec(T_int(), T_array(T_i8(), max_size)));
2709 2710 2711 2712 2713 2714 2715 2716 2717 2718 2719
            auto th = cx.tags.get(tag_id).th.llth;
            llvm.LLVMRefineType(llvm.LLVMResolveTypeHandle(th), tag_ty);
        }
        case (_) {
            // fall through
        }
    }

    ret cx;
}

2720 2721 2722 2723 2724 2725 2726 2727 2728 2729
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);
}

2730 2731 2732 2733 2734 2735 2736 2737 2738 2739 2740 2741 2742 2743 2744 2745 2746 2747 2748
// 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.
2749
                        auto union_val = C_zero_byte_arr(info.size as uint);
2750
                        auto val = C_struct(vec(C_int(i as int), union_val));
G
Graydon Hoare 已提交
2751

2752 2753 2754 2755 2756 2757 2758 2759 2760 2761 2762 2763 2764 2765 2766 2767
                        // 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;
            }
        }
2768 2769 2770 2771 2772 2773 2774 2775

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

2776 2777 2778 2779 2780 2781 2782 2783 2784 2785 2786 2787 2788 2789 2790 2791 2792
        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);
}

2793 2794 2795 2796
fn p2i(ValueRef v) -> ValueRef {
    ret llvm.LLVMConstPtrToInt(v, T_int());
}

2797
fn trans_exit_task_glue(@crate_ctxt cx) {
2798 2799 2800 2801 2802 2803 2804
    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,
2805
                    llargs=new_def_hash[ValueRef](),
2806
                    lllocals=new_def_hash[ValueRef](),
2807
                    lltydescs=new_def_hash[ValueRef](),
2808
                    ccx=cx);
2809

2810
    auto bcx = new_top_block_ctxt(fcx);
2811
    trans_upcall(bcx, "upcall_exit", V_args);
2812
    bcx.build.RetVoid();
2813 2814
}

2815 2816 2817
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());
2818
    llvm.LLVMAddTypeName(cx.llmod, _str.buf("rust_tydesc"), T_tydesc());
2819 2820
}

2821
fn crate_constant(@crate_ctxt cx) -> ValueRef {
2822 2823

    let ValueRef crate_ptr =
2824
        llvm.LLVMAddGlobal(cx.llmod, T_crate(),
2825 2826 2827 2828 2829 2830 2831 2832 2833 2834
                           _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);

2835 2836
    let ValueRef exit_task_glue_off =
        llvm.LLVMConstSub(p2i(cx.glues.exit_task_glue), crate_addr);
2837 2838 2839

    let ValueRef crate_val =
        C_struct(vec(C_null(T_int()),     // ptrdiff_t image_base_off
2840
                     p2i(crate_ptr),      // uintptr_t self_addr
2841 2842 2843 2844 2845 2846 2847 2848 2849 2850 2851 2852 2853 2854 2855 2856 2857 2858
                     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;
}

2859
fn trans_main_fn(@crate_ctxt cx, ValueRef llcrate) {
2860 2861 2862
    auto T_main_args = vec(T_int(), T_int());
    auto T_rust_start_args = vec(T_int(), T_int(), T_int(), T_int());

2863 2864 2865 2866 2867 2868 2869
    auto main_name;
    if (_str.eq(std.os.target_os(), "win32")) {
        main_name = "WinMain@16";
    } else {
        main_name = "main";
    }

2870
    auto llmain =
2871
        decl_cdecl_fn(cx.llmod, main_name, T_fn(T_main_args, T_int()));
2872

2873 2874
    auto llrust_start = decl_cdecl_fn(cx.llmod, "rust_start",
                                      T_fn(T_rust_start_args, T_int()));
2875 2876 2877

    auto llargc = llvm.LLVMGetParam(llmain, 0u);
    auto llargv = llvm.LLVMGetParam(llmain, 1u);
2878 2879
    check (cx.item_names.contains_key("_rust.main"));
    auto llrust_main = cx.item_names.get("_rust.main");
2880 2881 2882 2883 2884 2885 2886 2887 2888 2889 2890

    //
    // 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(""));
2891
    auto b = new_builder(llbb);
2892 2893 2894 2895 2896 2897

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

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

2898 2899
fn declare_intrinsics(ModuleRef llmod) -> hashmap[str,ValueRef] {

2900
    let vec[TypeRef] T_trap_args = vec();
2901 2902 2903 2904
    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());
2905 2906
    auto trap = decl_cdecl_fn(llmod, "llvm.trap",
                              T_fn(T_trap_args, T_void()));
2907 2908 2909 2910
    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()));
2911 2912 2913

    auto intrinsics = new_str_hash[ValueRef]();
    intrinsics.insert("llvm.trap", trap);
2914 2915
    intrinsics.insert("llvm.memcpy.p0i8.p0i8.i32", memcpy32);
    intrinsics.insert("llvm.memcpy.p0i8.p0i8.i64", memcpy64);
2916
    ret intrinsics;
2917 2918
}

2919 2920 2921 2922 2923 2924 2925 2926
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.
}

2927 2928 2929 2930 2931
fn make_no_op_type_glue(ModuleRef llmod) -> ValueRef {
    auto ty = T_fn(vec(T_taskptr(), T_ptr(T_i8())), T_void());
    auto fun = decl_fastcall_fn(llmod, "_rust_no_op_type_glue", ty);
    auto bb_name = _str.buf("_rust_no_op_type_glue_bb");
    auto llbb = llvm.LLVMAppendBasicBlock(fun, bb_name);
2932
    new_builder(llbb).RetVoid();
2933 2934 2935 2936 2937 2938 2939 2940 2941 2942 2943 2944 2945 2946 2947 2948 2949 2950 2951 2952 2953 2954 2955 2956 2957
    ret fun;
}

fn make_glues(ModuleRef llmod) -> @glue_fns {
    ret @rec(activate_glue = decl_glue(llmod, abi.activate_glue_name()),
             yield_glue = decl_glue(llmod, abi.yield_glue_name()),
             /*
              * 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(),
                                            T_fn(vec(T_taskptr()), T_void())),

             upcall_glues =
              _vec.init_fn[ValueRef](bind decl_upcall(llmod, _),
                                     abi.n_upcall_glues as uint),
             no_op_type_glue = make_no_op_type_glue(llmod));
}

2958
fn trans_crate(session.session sess, @ast.crate crate, str output) {
2959 2960 2961 2962
    auto llmod =
        llvm.LLVMModuleCreateWithNameInContext(_str.buf("rust_out"),
                                               llvm.LLVMGetGlobalContext());

2963 2964
    llvm.LLVMSetDataLayout(llmod, _str.buf(x86.get_data_layout()));
    llvm.LLVMSetTarget(llmod, _str.buf(x86.get_target_triple()));
2965
    auto td = mk_target_data(x86.get_data_layout());
2966

2967 2968
    llvm.LLVMSetModuleInlineAsm(llmod, _str.buf(x86.get_module_asm()));

2969
    auto intrinsics = declare_intrinsics(llmod);
2970

2971
    auto glues = make_glues(llmod);
2972 2973 2974
    auto hasher = ty.hash_ty;
    auto eqer = ty.eq_ty;
    auto tydescs = map.mk_hashmap[@ty.t,ValueRef](hasher, eqer);
2975

2976 2977
    auto cx = @rec(sess = sess,
                   llmod = llmod,
2978
                   td = td,
2979
                   upcalls = new_str_hash[ValueRef](),
2980
                   intrinsics = intrinsics,
2981 2982
                   item_names = new_str_hash[ValueRef](),
                   item_ids = new_def_hash[ValueRef](),
2983
                   items = new_def_hash[@ast.item](),
2984
                   tags = new_def_hash[@tag_info](),
2985
                   tydescs = tydescs,
2986
                   glues = glues,
2987
                   names = namegen(0),
2988
                   path = "_rust");
2989

2990 2991
    create_typedefs(cx);

2992
    collect_items(cx, crate);
2993
    resolve_tag_types(cx, crate);
2994
    trans_constants(cx, crate);
2995

2996
    trans_mod(cx, crate.node.module);
2997
    trans_exit_task_glue(cx);
2998 2999
    trans_main_fn(cx, crate_constant(cx));

3000 3001
    check_module(llmod);

3002
    llvm.LLVMWriteBitcodeToFile(llmod, _str.buf(output));
3003 3004 3005 3006 3007 3008 3009 3010 3011 3012 3013 3014 3015
    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:
//