| 1 | #include <stdio.h> | 
|---|
| 2 | #include "except.h" | 
|---|
| 3 |  | 
|---|
| 4 | #define EXCEPTION 2 | 
|---|
| 5 |  | 
|---|
| 6 | struct type_raii_t { | 
|---|
| 7 | char * msg; | 
|---|
| 8 | }; | 
|---|
| 9 |  | 
|---|
| 10 | //Dtor function to test clean up routines | 
|---|
| 11 | void dtor( struct type_raii_t * this ) { | 
|---|
| 12 | printf("%s\n", this->msg); | 
|---|
| 13 | } | 
|---|
| 14 |  | 
|---|
| 15 | //Type macro use to make scope unwinding easier to see. | 
|---|
| 16 | #define raii_t __attribute__(( cleanup(dtor) )) struct type_raii_t | 
|---|
| 17 |  | 
|---|
| 18 | //Leaf functions that raises exception | 
|---|
| 19 | void bar() { | 
|---|
| 20 | raii_t a = { "Bar dtor" }; | 
|---|
| 21 |  | 
|---|
| 22 | throw( EXCEPTION ); | 
|---|
| 23 | } | 
|---|
| 24 |  | 
|---|
| 25 | //Matcher function which will check if the exception was correctly caught | 
|---|
| 26 | extern int this_exception; | 
|---|
| 27 | _Unwind_Reason_Code foo_try_match() { | 
|---|
| 28 | return this_exception == 2 ? _URC_HANDLER_FOUND : _URC_CONTINUE_UNWIND; | 
|---|
| 29 | } | 
|---|
| 30 |  | 
|---|
| 31 | //Try statements are hoisted out see comments for details | 
|---|
| 32 | //With this could probably be unique and simply linked from | 
|---|
| 33 | //libcfa but there is one problem left, see the exception table | 
|---|
| 34 | //for details | 
|---|
| 35 | void try( void (*try_block)(), void (*catch_block)() ) | 
|---|
| 36 | { | 
|---|
| 37 | //Setup statments | 
|---|
| 38 | //These 2 statments won't actually result in any code, | 
|---|
| 39 | //they only setup global tables. | 
|---|
| 40 | //However, they clobber gcc cancellation support from gcc. | 
|---|
| 41 | //We can replace the personality routine but replacing the exception | 
|---|
| 42 | //table gcc generates is not really doable, it generates labels based | 
|---|
| 43 | //on how the assembly works. | 
|---|
| 44 | //Setup the personality routine | 
|---|
| 45 | asm volatile (".cfi_personality 0x3,__gcfa_personality_v0"); | 
|---|
| 46 | //Setup the exception table | 
|---|
| 47 | asm volatile (".cfi_lsda 0x3, .LLSDACFA2"); | 
|---|
| 48 |  | 
|---|
| 49 | //Label which defines the start of the area for which the handler is setup | 
|---|
| 50 | asm volatile (".TRYSTART:"); | 
|---|
| 51 |  | 
|---|
| 52 | //The actual statements of the try blocks | 
|---|
| 53 | try_block(); | 
|---|
| 54 |  | 
|---|
| 55 | //asm statement to prevent deadcode removal | 
|---|
| 56 | asm volatile goto ("" : : : : CATCH ); | 
|---|
| 57 |  | 
|---|
| 58 | //Normal return | 
|---|
| 59 | return; | 
|---|
| 60 |  | 
|---|
| 61 | //Exceptionnal path | 
|---|
| 62 | CATCH : __attribute__(( unused )); | 
|---|
| 63 | //Label which defines the end of the area for which the handler is setup | 
|---|
| 64 | asm volatile (".TRYEND:"); | 
|---|
| 65 | //Label which defines the start of the exception landing pad | 
|---|
| 66 | //basically what will be called when the exception is caught | 
|---|
| 67 | //Note, if multiple handlers are given, the multiplexing should be done | 
|---|
| 68 | //by the generated code, not the exception runtime | 
|---|
| 69 | asm volatile (".CATCH:"); | 
|---|
| 70 |  | 
|---|
| 71 | //Exception handler | 
|---|
| 72 | catch_block(); | 
|---|
| 73 | } | 
|---|
| 74 |  | 
|---|
| 75 | //Exception table data we need to generate | 
|---|
| 76 | //While this is almost generic, the custom data refers to | 
|---|
| 77 | //foo_try_match try match, which is no way generic | 
|---|
| 78 | //Some more works need to be done if we want to have a single | 
|---|
| 79 | //call to the try routine | 
|---|
| 80 | asm ( | 
|---|
| 81 | //HEADER | 
|---|
| 82 | ".LFECFA1:\n" | 
|---|
| 83 | "       .globl  __gcfa_personality_v0\n" | 
|---|
| 84 | "       .section        .gcc_except_table,\"a\",@progbits\n" | 
|---|
| 85 | ".LLSDACFA2:\n"                                                 //TABLE header | 
|---|
| 86 | "       .byte   0xff\n" | 
|---|
| 87 | "       .byte   0xff\n" | 
|---|
| 88 | "       .byte   0x1\n" | 
|---|
| 89 | "       .uleb128 .LLSDACSECFA2-.LLSDACSBCFA2\n"         //BODY length | 
|---|
| 90 | //Body uses language specific data and therefore could be modified arbitrarily | 
|---|
| 91 | ".LLSDACSBCFA2:\n"                                              //BODY start | 
|---|
| 92 | "       .uleb128 .TRYSTART-try\n"                               //Handled area start  (relative to start of function) | 
|---|
| 93 | "       .uleb128 .TRYEND-.TRYSTART\n"                           //Handled area length | 
|---|
| 94 | "       .uleb128 .CATCH-try\n"                          //Hanlder landing pad adress  (relative to start of function) | 
|---|
| 95 | "       .uleb128 1\n"                                           //Action code, gcc seems to use always 0 | 
|---|
| 96 | //Beyond this point we don't match gcc data' | 
|---|
| 97 | "       .uleb128 foo_try_match-try\n"                   //Handler routine to check if the exception is matched | 
|---|
| 98 | ".LLSDACSECFA2:\n"                                              //BODY end | 
|---|
| 99 | "       .text\n"                                                        //TABLE footer | 
|---|
| 100 | "       .size   try, .-try\n" | 
|---|
| 101 | "       .ident  \"GCC: (Ubuntu 6.2.0-3ubuntu11~16.04) 6.2.0 20160901\"\n" | 
|---|
| 102 | "       .section        .note.GNU-stack,\"x\",@progbits\n" | 
|---|
| 103 | ); | 
|---|
| 104 |  | 
|---|
| 105 | void foo() { | 
|---|
| 106 | raii_t a = { "Foo dtor" }; | 
|---|
| 107 |  | 
|---|
| 108 | //Since try will clobber the gcc exception table assembly, | 
|---|
| 109 | //We need to nest this to have gcc regenerate the data | 
|---|
| 110 | void foo_try_block() { | 
|---|
| 111 | raii_t b = { "Foo try dtor" }; | 
|---|
| 112 |  | 
|---|
| 113 | bar(); | 
|---|
| 114 |  | 
|---|
| 115 | printf("Called bar successfully\n"); | 
|---|
| 116 | } | 
|---|
| 117 |  | 
|---|
| 118 | void foo_catch_block() { | 
|---|
| 119 | printf("Exception caught\n"); | 
|---|
| 120 | } | 
|---|
| 121 |  | 
|---|
| 122 | //Actual call to the try block | 
|---|
| 123 | try( foo_try_block, foo_catch_block ); | 
|---|
| 124 |  | 
|---|
| 125 | printf( "Foo exited normally\n" ); | 
|---|
| 126 | } | 
|---|
| 127 |  | 
|---|
| 128 | int main() { | 
|---|
| 129 | raii_t a = { "Main dtor" }; | 
|---|
| 130 |  | 
|---|
| 131 | foo(); | 
|---|
| 132 |  | 
|---|
| 133 | printf("End of program reached\n"); | 
|---|
| 134 | } | 
|---|