authorgravatar for andrew@ziglang.orgAndrew Kelley <andrew@ziglang.org> 2018-07-28 01:26:11-04:00
committergravatar for andrew@ziglang.orgAndrew Kelley <andrew@ziglang.org> 2018-07-28 01:26:11-04:00
logc6f9a4c0445000e59bca16838cc413140d399f35
treef9ded327b406a8ed10fe23d894ef70393b48e81d
parent6fed777637daafc077ec2c079b0557f3c8bdce51

cancel detects suspend bit

1 files changed, 8 insertions(+), 1 deletions(-)

src/ir.cpp+8-1
......@@ -6672,6 +6672,7 @@ static IrInstruction *ir_gen_cancel(IrBuilder *irb, Scope *scope, AstNode *node)
66726672
66736673 IrBasicBlock *done_block = ir_create_basic_block(irb, scope, "CancelDone");
66746674 IrBasicBlock *not_canceled_block = ir_create_basic_block(irb, scope, "NotCanceled");
6675 IrBasicBlock *pre_return_block = ir_create_basic_block(irb, scope, "PreReturn");
66756676 IrBasicBlock *post_return_block = ir_create_basic_block(irb, scope, "PostReturn");
66766677 IrBasicBlock *do_cancel_block = ir_create_basic_block(irb, scope, "DoCancel");
66776678
......@@ -6684,6 +6685,7 @@ static IrInstruction *ir_gen_cancel(IrBuilder *irb, Scope *scope, AstNode *node)
66846685 IrInstruction *inverted_ptr_mask = ir_build_const_usize(irb, scope, node, 0x7); // 0b111
66856686 IrInstruction *ptr_mask = ir_build_un_op(irb, scope, node, IrUnOpBinNot, inverted_ptr_mask); // 0b111...000
66866687 IrInstruction *await_mask = ir_build_const_usize(irb, scope, node, 0x4); // 0b100
6688 IrInstruction *is_suspended_mask = ir_build_const_usize(irb, scope, node, 0x2); // 0b010
66876689
66886690 // TODO relies on Zig not re-ordering fields
66896691 IrInstruction *casted_target_inst = ir_build_ptr_cast(irb, scope, node, promise_T_type_val, target_inst);
......@@ -6704,13 +6706,18 @@ static IrInstruction *ir_gen_cancel(IrBuilder *irb, Scope *scope, AstNode *node)
67046706 ir_set_cursor_at_end_and_append_block(irb, not_canceled_block);
67056707 IrInstruction *awaiter_addr = ir_build_bin_op(irb, scope, node, IrBinOpBinAnd, prev_atomic_value, ptr_mask, false);
67066708 IrInstruction *is_returned_bool = ir_build_bin_op(irb, scope, node, IrBinOpCmpEq, awaiter_addr, ptr_mask, false);
6707 ir_build_cond_br(irb, scope, node, is_returned_bool, post_return_block, do_cancel_block, is_comptime);
6709 ir_build_cond_br(irb, scope, node, is_returned_bool, post_return_block, pre_return_block, is_comptime);
67086710
67096711 ir_set_cursor_at_end_and_append_block(irb, post_return_block);
67106712 IrInstruction *is_awaited_value = ir_build_bin_op(irb, scope, node, IrBinOpBinAnd, prev_atomic_value, await_mask, false);
67116713 IrInstruction *is_awaited_bool = ir_build_bin_op(irb, scope, node, IrBinOpCmpNotEq, is_awaited_value, zero, false);
67126714 ir_build_cond_br(irb, scope, node, is_awaited_bool, done_block, do_cancel_block, is_comptime);
67136715
6716 ir_set_cursor_at_end_and_append_block(irb, pre_return_block);
6717 IrInstruction *is_suspended_value = ir_build_bin_op(irb, scope, node, IrBinOpBinAnd, prev_atomic_value, is_suspended_mask, false);
6718 IrInstruction *is_suspended_bool = ir_build_bin_op(irb, scope, node, IrBinOpCmpNotEq, is_suspended_value, zero, false);
6719 ir_build_cond_br(irb, scope, node, is_suspended_bool, do_cancel_block, done_block, is_comptime);
6720
67146721 ir_set_cursor_at_end_and_append_block(irb, do_cancel_block);
67156722 ir_build_cancel(irb, scope, node, target_inst);
67166723 ir_build_br(irb, scope, node, done_block, is_comptime);