| [ba912706] | 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 | } | 
|---|