X-Git-Url: http://git.megacz.com/?a=blobdiff_plain;f=rts%2FException.cmm;h=a3f3dd0d8290aad57c8ad12b72bbbec46d0661c5;hb=e576ba5d31fbae54c43e88316fb0dbdba9cbd4ff;hp=f4327b9ce24aab743331294ba5708b6f81d692c8;hpb=b1953bbb1ed3cb16497e5447db7487f0c2d9e41a;p=ghc-hetmet.git diff --git a/rts/Exception.cmm b/rts/Exception.cmm index f4327b9..a3f3dd0 100644 --- a/rts/Exception.cmm +++ b/rts/Exception.cmm @@ -53,15 +53,56 @@ INFO_TABLE_RET( stg_unblockAsyncExceptionszh_ret, 0/*framesize*/, 0/*bitmap*/, RET_SMALL ) { + CInt r; + // Not true: see comments above // ASSERT(StgTSO_blocked_exceptions(CurrentTSO) != NULL); - foreign "C" awakenBlockedExceptionQueue(MyCapability() "ptr", - CurrentTSO "ptr") [R1]; - StgTSO_flags(CurrentTSO) = StgTSO_flags(CurrentTSO) & ~(TSO_BLOCKEX::I32|TSO_INTERRUPTIBLE::I32); + /* Eagerly raise a blocked exception, if there is one */ + if (StgTSO_blocked_exceptions(CurrentTSO) != END_TSO_QUEUE) { + /* + * We have to be very careful here, as in killThread#, since + * we are about to raise an async exception in the current + * thread, which might result in the thread being killed. + */ + +#ifndef REG_R1 + /* + * raiseAsync assumes that the stack is in ThreadRunGHC state, + * i.e. with a return address on the top. In unreg mode, the + * return value for IO is on top of the return address, so we + * need to make a small adjustment here. + */ + Sp_adj(1); +#endif + SAVE_THREAD_STATE(); + r = foreign "C" maybePerformBlockedException (MyCapability() "ptr", + CurrentTSO "ptr") [R1]; + + if (r != 0::CInt) { + if (StgTSO_what_next(CurrentTSO) == ThreadKilled::I16) { + R1 = ThreadFinished; + jump StgReturn; + } else { + LOAD_THREAD_STATE(); + ASSERT(StgTSO_what_next(CurrentTSO) == ThreadRunGHC::I16); + jump %ENTRY_CODE(Sp(0)); + } + } +#ifndef REG_R1 + /* + * Readjust stack in unregisterised mode if we didn't raise an + * exception, see above + */ + else { + Sp_adj(-1); + } +#endif + } + #ifdef REG_R1 Sp_adj(1); jump %ENTRY_CODE(Sp(0)); @@ -116,12 +157,38 @@ blockAsyncExceptionszh_fast unblockAsyncExceptionszh_fast { + CInt r; + /* Args: R1 :: IO a */ STK_CHK_GEN( WDS(2), R1_PTR, unblockAsyncExceptionszh_fast); - if (TO_W_(StgTSO_flags(CurrentTSO)) & TSO_BLOCKEX) { - foreign "C" awakenBlockedExceptionQueue(MyCapability() "ptr", - CurrentTSO "ptr") [R1]; + if ((TO_W_(StgTSO_flags(CurrentTSO)) & TSO_BLOCKEX) != 0) { + + StgTSO_flags(CurrentTSO) = StgTSO_flags(CurrentTSO) & + ~(TSO_BLOCKEX::I32|TSO_INTERRUPTIBLE::I32); + + /* Eagerly raise a blocked exception, if there is one */ + if (StgTSO_blocked_exceptions(CurrentTSO) != END_TSO_QUEUE) { + /* + * We have to be very careful here, as in killThread#, since + * we are about to raise an async exception in the current + * thread, which might result in the thread being killed. + */ + SAVE_THREAD_STATE(); + r = foreign "C" maybePerformBlockedException (MyCapability() "ptr", + CurrentTSO "ptr") [R1]; + + if (r != 0::CInt) { + if (StgTSO_what_next(CurrentTSO) == ThreadKilled::I16) { + R1 = ThreadFinished; + jump StgReturn; + } else { + LOAD_THREAD_STATE(); + ASSERT(StgTSO_what_next(CurrentTSO) == ThreadRunGHC::I16); + jump %ENTRY_CODE(Sp(0)); + } + } + } /* avoid growing the stack unnecessarily */ if (Sp(0) == stg_unblockAsyncExceptionszh_ret_info) { @@ -148,7 +215,7 @@ killThreadzh_fast target = R1; exception = R2; - STK_CHK_GEN( WDS(3), R1_PTR | R2_PTR, killThreadzh_fast); + STK_CHK_GEN( WDS(3), R1_PTR & R2_PTR, killThreadzh_fast); /* * We might have killed ourselves. In which case, better be *very* @@ -329,10 +396,13 @@ raisezh_fast /* ToDo: currently this is a hack. Would be much better if * the info was only displayed for an *uncaught* exception. */ - if (RtsFlags_ProfFlags_showCCSOnException(RtsFlags)) { + if (RtsFlags_ProfFlags_showCCSOnException(RtsFlags) != 0::I32) { foreign "C" fprintCCS_stderr(W_[CCCS] "ptr"); } #endif + + /* Inform the Hpc that an exception has been thrown */ + foreign "C" hs_hpc_raise_event(CurrentTSO "ptr"); retry_pop_stack: StgTSO_sp(CurrentTSO) = Sp; @@ -341,14 +411,27 @@ retry_pop_stack: if (frame_type == ATOMICALLY_FRAME) { /* The exception has reached the edge of a memory transaction. Check that * the transaction is valid. If not then perhaps the exception should - * not have been thrown: re-run the transaction */ - W_ trec; + * not have been thrown: re-run the transaction. "trec" will either be + * a top-level transaction running the atomic block, or a nested + * transaction running an invariant check. In the latter case we + * abort and de-allocate the top-level transaction that encloses it + * as well (we could just abandon its transaction record, but this makes + * sure it's marked as aborted and available for re-use). */ + W_ trec, outer; W_ r; trec = StgTSO_trec(CurrentTSO); r = foreign "C" stmValidateNestOfTransactions(trec "ptr"); + "ptr" outer = foreign "C" stmGetEnclosingTRec(trec "ptr") []; foreign "C" stmAbortTransaction(MyCapability() "ptr", trec "ptr"); + foreign "C" stmFreeAbortedTRec(MyCapability() "ptr", trec "ptr"); + + if (outer != NO_TREC) { + foreign "C" stmAbortTransaction(MyCapability() "ptr", outer "ptr"); + foreign "C" stmFreeAbortedTRec(MyCapability() "ptr", outer "ptr"); + } + StgTSO_trec(CurrentTSO) = NO_TREC; - if (r) { + if (r != 0) { // Transaction was valid: continue searching for a catch frame Sp = Sp + SIZEOF_StgAtomicallyFrame; goto retry_pop_stack; @@ -397,6 +480,9 @@ retry_pop_stack: * If exceptions were unblocked, arrange that they are unblocked * again after executing the handler by pushing an * unblockAsyncExceptions_ret stack frame. + * + * If we've reached an STM catch frame then roll back the nested + * transaction we were using. */ W_ frame; frame = Sp; @@ -407,6 +493,12 @@ retry_pop_stack: Sp(0) = stg_unblockAsyncExceptionszh_ret_info; } } else { + W_ trec, outer; + trec = StgTSO_trec(CurrentTSO); + "ptr" outer = foreign "C" stmGetEnclosingTRec(trec "ptr") []; + foreign "C" stmAbortTransaction(MyCapability() "ptr", trec "ptr") []; + foreign "C" stmFreeAbortedTRec(MyCapability() "ptr", trec "ptr") []; + StgTSO_trec(CurrentTSO) = outer; Sp = Sp + SIZEOF_StgCatchSTMFrame; }