| 1 | // | 
|---|
| 2 | // Cforall Version 1.0.0 Copyright (C) 2015 University of Waterloo | 
|---|
| 3 | // | 
|---|
| 4 | // The contents of this file are covered under the licence agreement in the | 
|---|
| 5 | // file "LICENCE" distributed with Cforall. | 
|---|
| 6 | // | 
|---|
| 7 | // ExceptVisitor.cc -- | 
|---|
| 8 | // | 
|---|
| 9 | // Author           : Andrew Beach | 
|---|
| 10 | // Created On       : Wed Jun 14 16:49:00 2017 | 
|---|
| 11 | // Last Modified By : Andrew Beach | 
|---|
| 12 | // Last Modified On : Thr Jun 22 15:57:00 2017 | 
|---|
| 13 | // Update Count     : 0 | 
|---|
| 14 | // | 
|---|
| 15 |  | 
|---|
| 16 | #include "ExceptTranslate.h" | 
|---|
| 17 | #include "Common/PassVisitor.h" | 
|---|
| 18 |  | 
|---|
| 19 | namespace ControlFlow { | 
|---|
| 20 |  | 
|---|
| 21 | // This (large) section could probably be moved out of the class | 
|---|
| 22 | // and be static helpers instead. | 
|---|
| 23 |  | 
|---|
| 24 | // Type(Qualifiers &, false, std::list<Attribute *> &) | 
|---|
| 25 |  | 
|---|
| 26 | // void (*function)() | 
|---|
| 27 | static FunctionType void_func_t(Type::Qualifiers(), false); | 
|---|
| 28 | // void (*function)(int, exception); | 
|---|
| 29 | static FunctionType catch_func_t(Type::Qualifiers(), false); | 
|---|
| 30 | // int (*function)(exception); | 
|---|
| 31 | static FunctionType match_func_t(Type::Qualifiers(), false); | 
|---|
| 32 | // bool (*function)(exception); | 
|---|
| 33 | static FunctionType handle_func_t(Type::Qualifiers(), false); | 
|---|
| 34 |  | 
|---|
| 35 | static void init_func_types() { | 
|---|
| 36 | static init_complete = false; | 
|---|
| 37 | if (init_complete) { | 
|---|
| 38 | return; | 
|---|
| 39 | } | 
|---|
| 40 | ObjectDecl index_obj( | 
|---|
| 41 | "index_t", | 
|---|
| 42 | Type::StorageClasses(), | 
|---|
| 43 | LinkageSpec::Cforall, | 
|---|
| 44 | /*bitfieldWidth*/ NULL, | 
|---|
| 45 | new BasicType(emptyQualifiers, BasicType::UnsignedInt), | 
|---|
| 46 | /*init*/ NULL | 
|---|
| 47 | ); | 
|---|
| 48 | ObjectDecl exception_obj( | 
|---|
| 49 | "exception_t", | 
|---|
| 50 | Type::StorageClasses(), | 
|---|
| 51 | LinkageSpec::Cforall, | 
|---|
| 52 | /*bitfieldWidth*/ NULL, | 
|---|
| 53 | new BasicType(emptyQualifiers, BasicType::UnsignedInt), | 
|---|
| 54 | /*init*/ NULL | 
|---|
| 55 | ); | 
|---|
| 56 | ObjectDecl bool_obj( | 
|---|
| 57 | "bool_t", | 
|---|
| 58 | Type::StorageClasses(), | 
|---|
| 59 | LinkageSpec::Cforall, | 
|---|
| 60 | /*bitfieldWidth*/ NULL, | 
|---|
| 61 | new BasicType(emptyQualifiers, BasicType::Bool), | 
|---|
| 62 | /*init*/ NULL | 
|---|
| 63 | ); | 
|---|
| 64 |  | 
|---|
| 65 | catch_func_t.get_parameters().push_back(index_obj.clone()); | 
|---|
| 66 | catch_func_t.get_parameters().push_back(exception_obj.clone()); | 
|---|
| 67 | match_func_t.get_returnVals().push_back(index_obj.clone()); | 
|---|
| 68 | match_func_t.get_parameters().push_back(exception_obj.clone()); | 
|---|
| 69 | handle_func_t.get_returnVals().push_back(bool_obj.clone()); | 
|---|
| 70 | handle_func_t.get_parameters().push_back(exception_obj.clone()); | 
|---|
| 71 |  | 
|---|
| 72 | init_complete = true; | 
|---|
| 73 | } | 
|---|
| 74 |  | 
|---|
| 75 | // Buricratic Helpers (Not having to do with the paritular operation.) | 
|---|
| 76 |  | 
|---|
| 77 | typedef std::list<CatchStmt*> CatchList; | 
|---|
| 78 |  | 
|---|
| 79 | void split( CatchList& allHandlers, CatchList& terHandlers, | 
|---|
| 80 | CatchList& resHandlers ) { | 
|---|
| 81 | while ( !allHandlers.empty() ) { | 
|---|
| 82 | Statement * stmt = allHandlers.front(); | 
|---|
| 83 | allHandlers.pop_front(); | 
|---|
| 84 | if (CaseStmt::Terminate == stmt->get_kind()) { | 
|---|
| 85 | terHandlers.push_back(stmt); | 
|---|
| 86 | } else { | 
|---|
| 87 | resHandlers.push_back(stmt); | 
|---|
| 88 | } | 
|---|
| 89 | } | 
|---|
| 90 | } | 
|---|
| 91 |  | 
|---|
| 92 | template<typename T> | 
|---|
| 93 | void free_all( std::list<T *> &list ) { | 
|---|
| 94 | std::list<T *>::iterator it; | 
|---|
| 95 | for ( it = list.begin() ; it != list.end() ; ++it ) { | 
|---|
| 96 | delete *it; | 
|---|
| 97 | } | 
|---|
| 98 | list.clear(); | 
|---|
| 99 | } | 
|---|
| 100 |  | 
|---|
| 101 | void appendDeclStmt( CompoundStmt * block, Declaration * item ) { | 
|---|
| 102 | block->push_back(new DeclStmt(no_labels, item)); | 
|---|
| 103 | } | 
|---|
| 104 |  | 
|---|
| 105 | Expression * nameOf( FunctionDecl * function ) { | 
|---|
| 106 | return new VariableExpr( function ); | 
|---|
| 107 | } | 
|---|
| 108 |  | 
|---|
| 109 | // ThrowStmt Mutation Helpers | 
|---|
| 110 |  | 
|---|
| 111 | Statement * create_terminate_throw( ThrowStmt *throwStmt ) { | 
|---|
| 112 | // __throw_terminate( EXPR ); | 
|---|
| 113 | ApplicationExpr * call = new ApplicationExpr( /* ... */ ); | 
|---|
| 114 | call->get_args.push_back( throwStmt->get_expr() ); | 
|---|
| 115 | Statement * result = new ExprStmt( throwStmt->get_labels(), call ); | 
|---|
| 116 | throwStmt->set_expr( nullptr ); | 
|---|
| 117 | delete throwStmt; | 
|---|
| 118 | return result; | 
|---|
| 119 | } | 
|---|
| 120 | Statement * create_terminate_rethrow( ThrowStmt *throwStmt ) { | 
|---|
| 121 | // __rethrow_terminate(); | 
|---|
| 122 | Statement * result = new ExprStmt( | 
|---|
| 123 | throwStmt->get_labels(), | 
|---|
| 124 | new ApplicationExpr( /* ... */ ); | 
|---|
| 125 | ); | 
|---|
| 126 | delete throwStmt; | 
|---|
| 127 | return result; | 
|---|
| 128 | } | 
|---|
| 129 | Statement * create_resume_throw( ThrowStmt *throwStmt ) { | 
|---|
| 130 | // __throw_resume( EXPR ); | 
|---|
| 131 | ApplicationExpr * call = new ApplicationExpr( /* ... */ ); | 
|---|
| 132 | call->get_args.push_back( throwStmt->get_expr() ); | 
|---|
| 133 | Statement * result = new ExprStmt( throwStmt->get_labels(), call ); | 
|---|
| 134 | throwStmt->set_expr( nullptr ); | 
|---|
| 135 | delete throwStmt; | 
|---|
| 136 | return result; | 
|---|
| 137 | } | 
|---|
| 138 | Statement * create_resume_rethrow( ThrowStmt *throwStmt ) { | 
|---|
| 139 | // return false; | 
|---|
| 140 | Statement * result = new ReturnStmt( | 
|---|
| 141 | throwStmt->get_labels(), | 
|---|
| 142 | new ConstantExpr( | 
|---|
| 143 | Constant( | 
|---|
| 144 | new BasicType( | 
|---|
| 145 | Type::Qualifiers(), | 
|---|
| 146 | BasicType::Bool | 
|---|
| 147 | ), | 
|---|
| 148 | "0") | 
|---|
| 149 | ) | 
|---|
| 150 | ); | 
|---|
| 151 | delete throwStmt; | 
|---|
| 152 | return result; | 
|---|
| 153 | } | 
|---|
| 154 |  | 
|---|
| 155 | // TryStmt Mutation Helpers | 
|---|
| 156 |  | 
|---|
| 157 | CompoundStmt * take_try_block( TryStmt *tryStmt ) { | 
|---|
| 158 | CompoundStmt * block = tryStmt->get_block(); | 
|---|
| 159 | tryStmt->set_block( nullptr ); | 
|---|
| 160 | return block; | 
|---|
| 161 | } | 
|---|
| 162 | FunctionDecl * create_try_wrapper( TryStmt *tryStmt ) { | 
|---|
| 163 | CompoundStmt * body = base_try->get_block(); | 
|---|
| 164 | base_try->set_block(nullptr); | 
|---|
| 165 |  | 
|---|
| 166 | return new FunctionDecl("try", Type::StorageClasses(), | 
|---|
| 167 | LinkageSpec::Cforall, void_func_t, body); | 
|---|
| 168 | } | 
|---|
| 169 |  | 
|---|
| 170 | FunctionDecl * create_terminate_catch( CatchList &handlers ) { | 
|---|
| 171 | std::list<CaseStmt *> handler_wrappers; | 
|---|
| 172 |  | 
|---|
| 173 | // Index 1..{number of handlers} | 
|---|
| 174 | int index = 0; | 
|---|
| 175 | CatchList::iterator it = handlers.begin(); | 
|---|
| 176 | for ( ; it != handlers.end() ; ++it ) { | 
|---|
| 177 | ++index; | 
|---|
| 178 | CatchStmt * handler = *it; | 
|---|
| 179 |  | 
|---|
| 180 | std::list<Statement *> core; | 
|---|
| 181 | if ( /*the exception is named*/ ) { | 
|---|
| 182 | ObjectDecl * local_except = /* Dynamic case, same */; | 
|---|
| 183 | core->push_back( new DeclStmt( noLabel, local_except ) ); | 
|---|
| 184 | } | 
|---|
| 185 | // Append the provided statement to the handler. | 
|---|
| 186 | core->push_back( cur_handler->get_body() ); | 
|---|
| 187 | // Append return onto the inner block? case stmt list? | 
|---|
| 188 | CaseStmt * wrapper = new CaseStmt( | 
|---|
| 189 | noLabels, | 
|---|
| 190 | new ConstantExpr( Constant::from_int( index ) ), | 
|---|
| 191 | core | 
|---|
| 192 | ); | 
|---|
| 193 | handler_wrappers.push_back(wrapper); | 
|---|
| 194 | } | 
|---|
| 195 | // TODO: Some sort of meaningful error on default perhaps? | 
|---|
| 196 |  | 
|---|
| 197 | SwitchStmt * handler_lookup = new SwitchStmt( | 
|---|
| 198 | noLabels, | 
|---|
| 199 | /*parameter 0: index*/, | 
|---|
| 200 | handler_wrappers, | 
|---|
| 201 | false | 
|---|
| 202 | ); | 
|---|
| 203 | CompoundStmt * body = new CompoundStmt( noLabels ); | 
|---|
| 204 | body->push_back( handler_lookup ); | 
|---|
| 205 |  | 
|---|
| 206 | return new FunctionDecl("catch", Type::StorageClasses(), | 
|---|
| 207 | LinkageSpec::Cforall, catch_func_t, body); | 
|---|
| 208 | } | 
|---|
| 209 |  | 
|---|
| 210 | // Create a single check from a moddified handler. | 
|---|
| 211 | CompoundStmt *create_single_matcher( CatchStmt * modded_handler ) { | 
|---|
| 212 | CompoundStmt * block = new CompoundStmt( noLables ); | 
|---|
| 213 |  | 
|---|
| 214 | appendDeclStmt( block, modded_handler->get_decl() ); | 
|---|
| 215 |  | 
|---|
| 216 | // TODO: This is not the actual check. | 
|---|
| 217 | LogicalExpr * cond = new ConstantExpr( Constant::from_bool( false ) ); | 
|---|
| 218 |  | 
|---|
| 219 | if ( modded_handler->get_cond() ) { | 
|---|
| 220 | cond = new LogicalExpr( cond, modded_handler->get_cond() )q | 
|---|
| 221 | } | 
|---|
| 222 | block->push_back( new IfStmt( noLabels, | 
|---|
| 223 | cond, modded_handler->get_body() ); | 
|---|
| 224 |  | 
|---|
| 225 | modded_handler->set_decl( nullptr ); | 
|---|
| 226 | modded_handler->set_cond( nullptr ); | 
|---|
| 227 | modded_handler->set_body( nullptr ); | 
|---|
| 228 | delete modded_handler; | 
|---|
| 229 | return block; | 
|---|
| 230 | } | 
|---|
| 231 |  | 
|---|
| 232 | FunctionDecl * create_terminate_match( CatchList &handlers ) { | 
|---|
| 233 | CompoundStmt * body = new CompoundStmt( noLabels ); | 
|---|
| 234 |  | 
|---|
| 235 | // Index 1..{number of handlers} | 
|---|
| 236 | int index = 0; | 
|---|
| 237 | CatchList::iterator it; | 
|---|
| 238 | for ( it = handlers.begin() ; it != handlers.end() ; ++it ) { | 
|---|
| 239 | ++index; | 
|---|
| 240 | CatchStmt * handler = *it; | 
|---|
| 241 |  | 
|---|
| 242 | // body should have been taken by create_terminate_catch. | 
|---|
| 243 | // assert( nullptr == handler->get_body() ); | 
|---|
| 244 | handler->set_body( new ReturnStmt( noLabels, | 
|---|
| 245 | new ConstantExpr( Constant::from_int( index ) ) ) ); | 
|---|
| 246 |  | 
|---|
| 247 | body->push_back( create_single_matcher( handler ) ); | 
|---|
| 248 | } | 
|---|
| 249 |  | 
|---|
| 250 | return new FunctionDecl("match", Type::StorageClasses(), | 
|---|
| 251 | LinkageSpec::Cforall, match_func_t, body); | 
|---|
| 252 | } | 
|---|
| 253 |  | 
|---|
| 254 | Statement * create_terminate_caller( | 
|---|
| 255 | FunctionDecl * try_wrapper, | 
|---|
| 256 | FunctionDecl * terminate_catch, | 
|---|
| 257 | FunctionDecl * terminate_match) { | 
|---|
| 258 |  | 
|---|
| 259 | ApplicationExpr * caller = new ApplicationExpr( /* ... */ ); | 
|---|
| 260 | std::list<Expression *>& args = caller.get_args(); | 
|---|
| 261 | args.push_back( nameOf( try_wrapper ) ); | 
|---|
| 262 | args.push_back( nameOf( terminate_catch ) ); | 
|---|
| 263 | args.push_back( nameOf( terminate_match ) ); | 
|---|
| 264 |  | 
|---|
| 265 | return new ExprStmt( noLabels, caller ); | 
|---|
| 266 | } | 
|---|
| 267 |  | 
|---|
| 268 | FunctionDecl * create_resume_handler( CatchList &handlers ) { | 
|---|
| 269 | CompoundStmt * body = new CompountStmt( noLabels ); | 
|---|
| 270 |  | 
|---|
| 271 | CatchList::iterator it; | 
|---|
| 272 | for ( it = handlers.begin() ; it != handlers.end() ; ++it ) { | 
|---|
| 273 | CatchStmt * handler = *it; | 
|---|
| 274 |  | 
|---|
| 275 | // Modifiy body. | 
|---|
| 276 | CompoundStmt * handling_code = | 
|---|
| 277 | dynamic_cast<CompoundStmt*>( handler->get_body() ); | 
|---|
| 278 | if ( ! handling_code ) { | 
|---|
| 279 | handling_code = new CompoundStmt( noLabels ); | 
|---|
| 280 | handling_code->push_back( handler->get_body() ); | 
|---|
| 281 | } | 
|---|
| 282 | handling_code->push_back( new ReturnStmt( noLabel, | 
|---|
| 283 | new ConstantExpr( Constant::from_bool( false ) ) ) ); | 
|---|
| 284 | handler->set_body( handling_code ); | 
|---|
| 285 |  | 
|---|
| 286 | // Create the handler. | 
|---|
| 287 | body->push_back( create_single_matcher( handler ) ); | 
|---|
| 288 | } | 
|---|
| 289 |  | 
|---|
| 290 | return new FunctionDecl("handle", Type::StorageClasses(), | 
|---|
| 291 | LinkageSpec::Cforall, handle_func_t, body); | 
|---|
| 292 | } | 
|---|
| 293 |  | 
|---|
| 294 | Statement * create_resume_wrapper( | 
|---|
| 295 | Statement * wraps, | 
|---|
| 296 | FunctionDecl * resume_handler ) { | 
|---|
| 297 | CompoundStmt * body = new CompoundStmt( noLabels ); | 
|---|
| 298 |  | 
|---|
| 299 | // struct node = {current top resume handler, call to resume_handler}; | 
|---|
| 300 | // __attribute__((cleanup( ... ))); | 
|---|
| 301 | // set top resume handler to node. | 
|---|
| 302 | // The wrapped statement. | 
|---|
| 303 |  | 
|---|
| 304 | ListInit * node_init; | 
|---|
| 305 | { | 
|---|
| 306 | std::list<Initializer*> field_inits; | 
|---|
| 307 | field_inits.push_back( new SingleInit( /* ... */ ) ); | 
|---|
| 308 | field_inits.push_back( new SingleInit( nameOf( resume_handler ) ) ); | 
|---|
| 309 | node_init = new ListInit( field_inits ); | 
|---|
| 310 | } | 
|---|
| 311 |  | 
|---|
| 312 | std::list< Attribute * > attributes; | 
|---|
| 313 | { | 
|---|
| 314 | std::list< Expression * > attr_params; | 
|---|
| 315 | attr_params.push_back( nameOf( /* ... deconstructor ... */ ) ); | 
|---|
| 316 | attrributes.push_back( new Attribute( "cleanup", attr_params ) ); | 
|---|
| 317 | } | 
|---|
| 318 |  | 
|---|
| 319 | appendDeclStmt( body, | 
|---|
| 320 | /**/ ObjectDecl( | 
|---|
| 321 | "resume_node", | 
|---|
| 322 | Type::StorageClasses(), | 
|---|
| 323 | LinkageSpec::Cforall, | 
|---|
| 324 | nullptr, | 
|---|
| 325 | /* Type* = resume_node */, | 
|---|
| 326 | node_init, | 
|---|
| 327 | attributes | 
|---|
| 328 | ) | 
|---|
| 329 | ); | 
|---|
| 330 | body->push_back( wraps ); | 
|---|
| 331 | return body; | 
|---|
| 332 | } | 
|---|
| 333 |  | 
|---|
| 334 | FunctionDecl * create_finally_wrapper( TryStmt * tryStmt ) { | 
|---|
| 335 | CompoundStmt * body = tryStmt->get_finally(); | 
|---|
| 336 | tryStmt->set_finally( nullptr ); | 
|---|
| 337 |  | 
|---|
| 338 | return new FunctionDecl("finally", Type::StorageClasses(), | 
|---|
| 339 | LinkageSpec::Cforall, void_func_t, body); | 
|---|
| 340 | } | 
|---|
| 341 |  | 
|---|
| 342 | ObjectDecl * create_finally_hook( FunctionDecl * finally_wrapper ) { | 
|---|
| 343 | // struct _cleanup_hook NAME __attribute__((cleanup( ... ))); | 
|---|
| 344 |  | 
|---|
| 345 | // Make Cleanup Attribute. | 
|---|
| 346 | std::list< Attribute * > attributes; | 
|---|
| 347 | { | 
|---|
| 348 | std::list< Expression * > attr_params; | 
|---|
| 349 | attr_params.push_back( nameOf( finally_wrapper ) ); | 
|---|
| 350 | attrributes.push_back( new Attribute( "cleanup", attr_params ) ); | 
|---|
| 351 | } | 
|---|
| 352 |  | 
|---|
| 353 | return ObjectDecl( /* ... */ | 
|---|
| 354 | const std::string &name "finally_hook", | 
|---|
| 355 | Type::StorageClasses(), | 
|---|
| 356 | LinkageSpec::Cforall, | 
|---|
| 357 | nullptr, | 
|---|
| 358 | /* ... Type * ... */, | 
|---|
| 359 | nullptr, | 
|---|
| 360 | attributes | 
|---|
| 361 | ); | 
|---|
| 362 | } | 
|---|
| 363 |  | 
|---|
| 364 |  | 
|---|
| 365 | class ExceptionMutatorCore : public WithScoping { | 
|---|
| 366 | enum Context { NoHandler, TerHandler, ResHandler }; | 
|---|
| 367 |  | 
|---|
| 368 | // Also need to handle goto, break & continue. | 
|---|
| 369 | // They need to be cut off in a ResHandler, until we enter another | 
|---|
| 370 | // loop, switch or the goto stays within the function. | 
|---|
| 371 |  | 
|---|
| 372 | Context curContext; | 
|---|
| 373 |  | 
|---|
| 374 | // We might not need this, but a unique base for each try block's | 
|---|
| 375 | // generated functions might be nice. | 
|---|
| 376 | //std::string curFunctionName; | 
|---|
| 377 | //unsigned int try_count = 0; | 
|---|
| 378 |  | 
|---|
| 379 |  | 
|---|
| 380 | public: | 
|---|
| 381 | ExceptionMutatorCore() : | 
|---|
| 382 | curContext(NoHandler) | 
|---|
| 383 | {} | 
|---|
| 384 |  | 
|---|
| 385 | void premutate( CatchStmt *tryStmt ); | 
|---|
| 386 | Statement * postmutate( ThrowStmt *throwStmt ); | 
|---|
| 387 | Statement * postmutate( TryStmt *tryStmt ); | 
|---|
| 388 | }; | 
|---|
| 389 |  | 
|---|
| 390 | Statement * ExceptionMutatorCore::postmutate( ThrowStmt *throwStmt ) { | 
|---|
| 391 | // Ignoring throwStmt->get_target() for now. | 
|---|
| 392 | if ( ThrowStmt::Terminate == throwStmt->get_kind() ) { | 
|---|
| 393 | if ( throwStmt->get_expr() ) { | 
|---|
| 394 | return create_terminate_throw( throwStmt ); | 
|---|
| 395 | } else if ( TerHandler == curContext ) { | 
|---|
| 396 | return create_terminate_rethrow( throwStmt ); | 
|---|
| 397 | } else { | 
|---|
| 398 | assertf(false, "Invalid throw in %s at %i\n", | 
|---|
| 399 | throwStmt->location.filename, | 
|---|
| 400 | throwStmt->location.linenumber); | 
|---|
| 401 | return nullptr; | 
|---|
| 402 | } | 
|---|
| 403 | } else { | 
|---|
| 404 | if ( throwStmt->get_expr() ) { | 
|---|
| 405 | return create_resume_throw( throwStmt ); | 
|---|
| 406 | } else if ( ResHandler == curContext ) { | 
|---|
| 407 | return create_resume_rethrow( throwStmt ); | 
|---|
| 408 | } else { | 
|---|
| 409 | assertf(false, "Invalid throwResume in %s at %i\n", | 
|---|
| 410 | throwStmt->location.filename, | 
|---|
| 411 | throwStmt->location.linenumber); | 
|---|
| 412 | return nullptr; | 
|---|
| 413 | } | 
|---|
| 414 | } | 
|---|
| 415 | } | 
|---|
| 416 |  | 
|---|
| 417 | Statement * ExceptionMutatorCore::postmutate( TryStmt *tryStmt ) { | 
|---|
| 418 | // Generate a prefix for the function names? | 
|---|
| 419 |  | 
|---|
| 420 | CompoundStmt * block = new CompoundStmt(); | 
|---|
| 421 | Statement * inner = take_try_block( tryStmt ); | 
|---|
| 422 |  | 
|---|
| 423 | if ( tryStmt->get_finally() ) { | 
|---|
| 424 | // Define the helper function. | 
|---|
| 425 | FunctionDecl * finally_block = | 
|---|
| 426 | create_finally_wrapper( tryStmt ); | 
|---|
| 427 | appendDeclStmt( block, finally_block ); | 
|---|
| 428 | // Create and add the finally cleanup hook. | 
|---|
| 429 | appendDeclStmt( block, create_finally_hook( finally_block ) ); | 
|---|
| 430 | } | 
|---|
| 431 |  | 
|---|
| 432 | StatementList termination_handlers; | 
|---|
| 433 | StatementList resumption_handlers; | 
|---|
| 434 | split( tryStmt->get_handlers(), | 
|---|
| 435 | termination_handlers, resumption_handlers ); | 
|---|
| 436 |  | 
|---|
| 437 | if ( resumeption_handlers.size() ) { | 
|---|
| 438 | // Define the helper function. | 
|---|
| 439 | FunctionDecl * resume_handler = | 
|---|
| 440 | create_resume_handler( resumption_handlers ); | 
|---|
| 441 | appendDeclStmt( block, resume_handler ); | 
|---|
| 442 | // Prepare hooks | 
|---|
| 443 | inner = create_resume_wrapper( inner, resume_handler ); | 
|---|
| 444 | } | 
|---|
| 445 |  | 
|---|
| 446 | if ( termination_handlers.size() ) { | 
|---|
| 447 | // Define the three helper functions. | 
|---|
| 448 | FunctionDecl * try_wrapper = create_try_wrapper( inner ); | 
|---|
| 449 | appendDeclStmt( block, try_wrapper ); | 
|---|
| 450 | FunctionDecl * terminate_catch = | 
|---|
| 451 | create_terminate_catch( termination_handlers ); | 
|---|
| 452 | appendDeclStmt( block, terminate_catch ); | 
|---|
| 453 | FunctionDecl * terminate_match = | 
|---|
| 454 | create_terminate_match( termination_handlers ); | 
|---|
| 455 | appendDeclStmt( block, terminate_match ); | 
|---|
| 456 | // Build the call to the try wrapper. | 
|---|
| 457 | inner = create_terminate_caller( | 
|---|
| 458 | try_wrapper, terminate_catch, terminate_match ); | 
|---|
| 459 | } | 
|---|
| 460 |  | 
|---|
| 461 | // Embed the try block. | 
|---|
| 462 | block->push_back( inner ); | 
|---|
| 463 |  | 
|---|
| 464 | free_all( termination_handlers ); | 
|---|
| 465 | free_all( resumption_handlers ); | 
|---|
| 466 |  | 
|---|
| 467 | return block; | 
|---|
| 468 | } | 
|---|
| 469 |  | 
|---|
| 470 | void ExceptionMutatorCore::premutate( CatchStmt *catchStmt ) { | 
|---|
| 471 | GuardValue( curContext ); | 
|---|
| 472 | if ( CatchStmt::Termination == catchStmt->get_kind() ) { | 
|---|
| 473 | curContext = TerHandler; | 
|---|
| 474 | } else { | 
|---|
| 475 | curContext = ResHandler; | 
|---|
| 476 | } | 
|---|
| 477 | } | 
|---|
| 478 |  | 
|---|
| 479 | void translateEHM( std::list< Declaration *> & translationUnit ) { | 
|---|
| 480 | PassVisitor<ExceptionMutatorCore> translator; | 
|---|
| 481 | for ( Declaration * decl : translationUnit ) { | 
|---|
| 482 | decl->mutate( translator ); | 
|---|
| 483 | } | 
|---|
| 484 | } | 
|---|
| 485 | } | 
|---|