| 1 | //
 | 
|---|
| 2 | // Cforall Version 1.0.0 Copyright (C) 2016 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 | // Keywords.cc --
 | 
|---|
| 8 | //
 | 
|---|
| 9 | // Author           : Thierry Delisle
 | 
|---|
| 10 | // Created On       : Mon Mar 13 12:41:22 2017
 | 
|---|
| 11 | // Last Modified By :
 | 
|---|
| 12 | // Last Modified On :
 | 
|---|
| 13 | // Update Count     : 5
 | 
|---|
| 14 | //
 | 
|---|
| 15 | 
 | 
|---|
| 16 | #include "Concurrency/Keywords.h"
 | 
|---|
| 17 | 
 | 
|---|
| 18 | #include <cassert>                 // for assert
 | 
|---|
| 19 | #include <string>                  // for string, operator==
 | 
|---|
| 20 | 
 | 
|---|
| 21 | #include "Common/PassVisitor.h"    // for PassVisitor
 | 
|---|
| 22 | #include "Common/SemanticError.h"  // for SemanticError
 | 
|---|
| 23 | #include "Common/utility.h"        // for deleteAll, map_range
 | 
|---|
| 24 | #include "CodeGen/OperatorTable.h" // for isConstructor
 | 
|---|
| 25 | #include "InitTweak/InitTweak.h"   // for getPointerBase
 | 
|---|
| 26 | #include "Parser/LinkageSpec.h"    // for Cforall
 | 
|---|
| 27 | #include "SymTab/AddVisit.h"       // for acceptAndAdd
 | 
|---|
| 28 | #include "SynTree/Constant.h"      // for Constant
 | 
|---|
| 29 | #include "SynTree/Declaration.h"   // for StructDecl, FunctionDecl, ObjectDecl
 | 
|---|
| 30 | #include "SynTree/Expression.h"    // for VariableExpr, ConstantExpr, Untype...
 | 
|---|
| 31 | #include "SynTree/Initializer.h"   // for SingleInit, ListInit, Initializer ...
 | 
|---|
| 32 | #include "SynTree/Label.h"         // for Label
 | 
|---|
| 33 | #include "SynTree/Statement.h"     // for CompoundStmt, DeclStmt, ExprStmt
 | 
|---|
| 34 | #include "SynTree/Type.h"          // for StructInstType, Type, PointerType
 | 
|---|
| 35 | #include "SynTree/Visitor.h"       // for Visitor, acceptAll
 | 
|---|
| 36 | 
 | 
|---|
| 37 | class Attribute;
 | 
|---|
| 38 | 
 | 
|---|
| 39 | namespace Concurrency {
 | 
|---|
| 40 | 
 | 
|---|
| 41 |         namespace {
 | 
|---|
| 42 |                 const std::list<Label> noLabels;
 | 
|---|
| 43 |                 const std::list< Attribute * > noAttributes;
 | 
|---|
| 44 |                 Type::StorageClasses noStorage;
 | 
|---|
| 45 |                 Type::Qualifiers noQualifiers;
 | 
|---|
| 46 |         }
 | 
|---|
| 47 | 
 | 
|---|
| 48 |         //=============================================================================================
 | 
|---|
| 49 |         // Pass declarations
 | 
|---|
| 50 |         //=============================================================================================
 | 
|---|
| 51 | 
 | 
|---|
| 52 |         //-----------------------------------------------------------------------------
 | 
|---|
| 53 |         //Handles sue type declarations :
 | 
|---|
| 54 |         // sue MyType {                             struct MyType {
 | 
|---|
| 55 |         //      int data;                                  int data;
 | 
|---|
| 56 |         //      a_struct_t more_data;                      a_struct_t more_data;
 | 
|---|
| 57 |         //                                =>             NewField_t newField;
 | 
|---|
| 58 |         // };                                        };
 | 
|---|
| 59 |         //                                           static inline NewField_t * getter_name( MyType * this ) { return &this->newField; }
 | 
|---|
| 60 |         //
 | 
|---|
| 61 |         class ConcurrentSueKeyword : public WithDeclsToAdd {
 | 
|---|
| 62 |           public:
 | 
|---|
| 63 | 
 | 
|---|
| 64 |                 ConcurrentSueKeyword( std::string&& type_name, std::string&& field_name, std::string&& getter_name, std::string&& context_error, bool needs_main ) :
 | 
|---|
| 65 |                   type_name( type_name ), field_name( field_name ), getter_name( getter_name ), context_error( context_error ), needs_main( needs_main ) {}
 | 
|---|
| 66 | 
 | 
|---|
| 67 |                 virtual ~ConcurrentSueKeyword() {}
 | 
|---|
| 68 | 
 | 
|---|
| 69 |                 void postvisit( StructDecl * decl );
 | 
|---|
| 70 | 
 | 
|---|
| 71 |                 void handle( StructDecl * );
 | 
|---|
| 72 |                 FunctionDecl * forwardDeclare( StructDecl * );
 | 
|---|
| 73 |                 ObjectDecl * addField( StructDecl * );
 | 
|---|
| 74 |                 void addRoutines( ObjectDecl *, FunctionDecl * );
 | 
|---|
| 75 | 
 | 
|---|
| 76 |                 virtual bool is_target( StructDecl * decl ) = 0;
 | 
|---|
| 77 | 
 | 
|---|
| 78 |           private:
 | 
|---|
| 79 |                 const std::string type_name;
 | 
|---|
| 80 |                 const std::string field_name;
 | 
|---|
| 81 |                 const std::string getter_name;
 | 
|---|
| 82 |                 const std::string context_error;
 | 
|---|
| 83 |                 bool needs_main;
 | 
|---|
| 84 | 
 | 
|---|
| 85 |                 StructDecl* type_decl = nullptr;
 | 
|---|
| 86 |         };
 | 
|---|
| 87 | 
 | 
|---|
| 88 | 
 | 
|---|
| 89 |         //-----------------------------------------------------------------------------
 | 
|---|
| 90 |         //Handles thread type declarations :
 | 
|---|
| 91 |         // thread Mythread {                         struct MyThread {
 | 
|---|
| 92 |         //      int data;                                  int data;
 | 
|---|
| 93 |         //      a_struct_t more_data;                      a_struct_t more_data;
 | 
|---|
| 94 |         //                                =>             thread_desc __thrd_d;
 | 
|---|
| 95 |         // };                                        };
 | 
|---|
| 96 |         //                                           static inline thread_desc * get_thread( MyThread * this ) { return &this->__thrd_d; }
 | 
|---|
| 97 |         //
 | 
|---|
| 98 |         class ThreadKeyword final : public ConcurrentSueKeyword {
 | 
|---|
| 99 |           public:
 | 
|---|
| 100 | 
 | 
|---|
| 101 |                 ThreadKeyword() : ConcurrentSueKeyword(
 | 
|---|
| 102 |                         "thread_desc",
 | 
|---|
| 103 |                         "__thrd",
 | 
|---|
| 104 |                         "get_thread",
 | 
|---|
| 105 |                         "thread keyword requires threads to be in scope, add #include <thread>",
 | 
|---|
| 106 |                         true
 | 
|---|
| 107 |                 )
 | 
|---|
| 108 |                 {}
 | 
|---|
| 109 | 
 | 
|---|
| 110 |                 virtual ~ThreadKeyword() {}
 | 
|---|
| 111 | 
 | 
|---|
| 112 |                 virtual bool is_target( StructDecl * decl ) override final { return decl->is_thread(); }
 | 
|---|
| 113 | 
 | 
|---|
| 114 |                 static void implement( std::list< Declaration * > & translationUnit ) {
 | 
|---|
| 115 |                         PassVisitor< ThreadKeyword > impl;
 | 
|---|
| 116 |                         acceptAll( translationUnit, impl );
 | 
|---|
| 117 |                 }
 | 
|---|
| 118 |         };
 | 
|---|
| 119 | 
 | 
|---|
| 120 |         //-----------------------------------------------------------------------------
 | 
|---|
| 121 |         //Handles coroutine type declarations :
 | 
|---|
| 122 |         // coroutine MyCoroutine {                   struct MyCoroutine {
 | 
|---|
| 123 |         //      int data;                                  int data;
 | 
|---|
| 124 |         //      a_struct_t more_data;                      a_struct_t more_data;
 | 
|---|
| 125 |         //                                =>             coroutine_desc __cor_d;
 | 
|---|
| 126 |         // };                                        };
 | 
|---|
| 127 |         //                                           static inline coroutine_desc * get_coroutine( MyCoroutine * this ) { return &this->__cor_d; }
 | 
|---|
| 128 |         //
 | 
|---|
| 129 |         class CoroutineKeyword final : public ConcurrentSueKeyword {
 | 
|---|
| 130 |           public:
 | 
|---|
| 131 | 
 | 
|---|
| 132 |                 CoroutineKeyword() : ConcurrentSueKeyword(
 | 
|---|
| 133 |                         "coroutine_desc",
 | 
|---|
| 134 |                         "__cor",
 | 
|---|
| 135 |                         "get_coroutine",
 | 
|---|
| 136 |                         "coroutine keyword requires coroutines to be in scope, add #include <coroutine>",
 | 
|---|
| 137 |                         true
 | 
|---|
| 138 |                 )
 | 
|---|
| 139 |                 {}
 | 
|---|
| 140 | 
 | 
|---|
| 141 |                 virtual ~CoroutineKeyword() {}
 | 
|---|
| 142 | 
 | 
|---|
| 143 |                 virtual bool is_target( StructDecl * decl ) override final { return decl->is_coroutine(); }
 | 
|---|
| 144 | 
 | 
|---|
| 145 |                 static void implement( std::list< Declaration * > & translationUnit ) {
 | 
|---|
| 146 |                         PassVisitor< CoroutineKeyword > impl;
 | 
|---|
| 147 |                         acceptAll( translationUnit, impl );
 | 
|---|
| 148 |                 }
 | 
|---|
| 149 |         };
 | 
|---|
| 150 | 
 | 
|---|
| 151 |         //-----------------------------------------------------------------------------
 | 
|---|
| 152 |         //Handles monitor type declarations :
 | 
|---|
| 153 |         // monitor MyMonitor {                       struct MyMonitor {
 | 
|---|
| 154 |         //      int data;                                  int data;
 | 
|---|
| 155 |         //      a_struct_t more_data;                      a_struct_t more_data;
 | 
|---|
| 156 |         //                                =>             monitor_desc __mon_d;
 | 
|---|
| 157 |         // };                                        };
 | 
|---|
| 158 |         //                                           static inline monitor_desc * get_coroutine( MyMonitor * this ) { return &this->__cor_d; }
 | 
|---|
| 159 |         //
 | 
|---|
| 160 |         class MonitorKeyword final : public ConcurrentSueKeyword {
 | 
|---|
| 161 |           public:
 | 
|---|
| 162 | 
 | 
|---|
| 163 |                 MonitorKeyword() : ConcurrentSueKeyword(
 | 
|---|
| 164 |                         "monitor_desc",
 | 
|---|
| 165 |                         "__mon",
 | 
|---|
| 166 |                         "get_monitor",
 | 
|---|
| 167 |                         "monitor keyword requires monitors to be in scope, add #include <monitor>",
 | 
|---|
| 168 |                         false
 | 
|---|
| 169 |                 )
 | 
|---|
| 170 |                 {}
 | 
|---|
| 171 | 
 | 
|---|
| 172 |                 virtual ~MonitorKeyword() {}
 | 
|---|
| 173 | 
 | 
|---|
| 174 |                 virtual bool is_target( StructDecl * decl ) override final { return decl->is_monitor(); }
 | 
|---|
| 175 | 
 | 
|---|
| 176 |                 static void implement( std::list< Declaration * > & translationUnit ) {
 | 
|---|
| 177 |                         PassVisitor< MonitorKeyword > impl;
 | 
|---|
| 178 |                         acceptAll( translationUnit, impl );
 | 
|---|
| 179 |                 }
 | 
|---|
| 180 |         };
 | 
|---|
| 181 | 
 | 
|---|
| 182 |         //-----------------------------------------------------------------------------
 | 
|---|
| 183 |         //Handles mutex routines definitions :
 | 
|---|
| 184 |         // void foo( A * mutex a, B * mutex b,  int i ) {                  void foo( A * a, B * b,  int i ) {
 | 
|---|
| 185 |         //                                                                       monitor_desc * __monitors[] = { get_monitor(a), get_monitor(b) };
 | 
|---|
| 186 |         //                                                                       monitor_guard_t __guard = { __monitors, 2 };
 | 
|---|
| 187 |         //    /*Some code*/                                       =>           /*Some code*/
 | 
|---|
| 188 |         // }                                                               }
 | 
|---|
| 189 |         //
 | 
|---|
| 190 |         class MutexKeyword final {
 | 
|---|
| 191 |           public:
 | 
|---|
| 192 | 
 | 
|---|
| 193 |                 void postvisit( FunctionDecl * decl );
 | 
|---|
| 194 |                 void postvisit(   StructDecl * decl );
 | 
|---|
| 195 | 
 | 
|---|
| 196 |                 std::list<DeclarationWithType*> findMutexArgs( FunctionDecl* );
 | 
|---|
| 197 |                 void validate( DeclarationWithType * );
 | 
|---|
| 198 |                 void addDtorStatments( FunctionDecl* func, CompoundStmt *, const std::list<DeclarationWithType * > &);
 | 
|---|
| 199 |                 void addStatments( FunctionDecl* func, CompoundStmt *, const std::list<DeclarationWithType * > &);
 | 
|---|
| 200 | 
 | 
|---|
| 201 |                 static void implement( std::list< Declaration * > & translationUnit ) {
 | 
|---|
| 202 |                         PassVisitor< MutexKeyword > impl;
 | 
|---|
| 203 |                         acceptAll( translationUnit, impl );
 | 
|---|
| 204 |                 }
 | 
|---|
| 205 | 
 | 
|---|
| 206 |           private:
 | 
|---|
| 207 |                 StructDecl* monitor_decl = nullptr;
 | 
|---|
| 208 |                 StructDecl* guard_decl = nullptr;
 | 
|---|
| 209 |                 StructDecl* dtor_guard_decl = nullptr;
 | 
|---|
| 210 | 
 | 
|---|
| 211 |                 static std::unique_ptr< Type > generic_func;
 | 
|---|
| 212 |         };
 | 
|---|
| 213 | 
 | 
|---|
| 214 |         std::unique_ptr< Type > MutexKeyword::generic_func = std::unique_ptr< Type >(
 | 
|---|
| 215 |                 new FunctionType(
 | 
|---|
| 216 |                         noQualifiers,
 | 
|---|
| 217 |                         true
 | 
|---|
| 218 |                 )
 | 
|---|
| 219 |         );
 | 
|---|
| 220 | 
 | 
|---|
| 221 |         //-----------------------------------------------------------------------------
 | 
|---|
| 222 |         //Handles mutex routines definitions :
 | 
|---|
| 223 |         // void foo( A * mutex a, B * mutex b,  int i ) {                  void foo( A * a, B * b,  int i ) {
 | 
|---|
| 224 |         //                                                                       monitor_desc * __monitors[] = { get_monitor(a), get_monitor(b) };
 | 
|---|
| 225 |         //                                                                       monitor_guard_t __guard = { __monitors, 2 };
 | 
|---|
| 226 |         //    /*Some code*/                                       =>           /*Some code*/
 | 
|---|
| 227 |         // }                                                               }
 | 
|---|
| 228 |         //
 | 
|---|
| 229 |         class ThreadStarter final {
 | 
|---|
| 230 |           public:
 | 
|---|
| 231 | 
 | 
|---|
| 232 |                 void postvisit( FunctionDecl * decl );
 | 
|---|
| 233 |                 void previsit ( StructDecl   * decl );
 | 
|---|
| 234 | 
 | 
|---|
| 235 |                 void addStartStatement( FunctionDecl * decl, DeclarationWithType * param );
 | 
|---|
| 236 | 
 | 
|---|
| 237 |                 static void implement( std::list< Declaration * > & translationUnit ) {
 | 
|---|
| 238 |                         PassVisitor< ThreadStarter > impl;
 | 
|---|
| 239 |                         acceptAll( translationUnit, impl );
 | 
|---|
| 240 |                 }
 | 
|---|
| 241 | 
 | 
|---|
| 242 |           private :
 | 
|---|
| 243 |                 bool thread_ctor_seen = false;
 | 
|---|
| 244 |                 StructDecl * thread_decl = nullptr;
 | 
|---|
| 245 |         };
 | 
|---|
| 246 | 
 | 
|---|
| 247 |         //=============================================================================================
 | 
|---|
| 248 |         // General entry routine
 | 
|---|
| 249 |         //=============================================================================================
 | 
|---|
| 250 |         void applyKeywords( std::list< Declaration * > & translationUnit ) {
 | 
|---|
| 251 |                 ThreadKeyword   ::implement( translationUnit );
 | 
|---|
| 252 |                 CoroutineKeyword        ::implement( translationUnit );
 | 
|---|
| 253 |                 MonitorKeyword  ::implement( translationUnit );
 | 
|---|
| 254 |         }
 | 
|---|
| 255 | 
 | 
|---|
| 256 |         void implementMutexFuncs( std::list< Declaration * > & translationUnit ) {
 | 
|---|
| 257 |                 MutexKeyword    ::implement( translationUnit );
 | 
|---|
| 258 |         }
 | 
|---|
| 259 | 
 | 
|---|
| 260 |         void implementThreadStarter( std::list< Declaration * > & translationUnit ) {
 | 
|---|
| 261 |                 ThreadStarter   ::implement( translationUnit );
 | 
|---|
| 262 |         }
 | 
|---|
| 263 | 
 | 
|---|
| 264 |         //=============================================================================================
 | 
|---|
| 265 |         // Generic keyword implementation
 | 
|---|
| 266 |         //=============================================================================================
 | 
|---|
| 267 |         void ConcurrentSueKeyword::postvisit(StructDecl * decl) {
 | 
|---|
| 268 |                 if( decl->name == type_name && decl->body ) {
 | 
|---|
| 269 |                         assert( !type_decl );
 | 
|---|
| 270 |                         type_decl = decl;
 | 
|---|
| 271 |                 }
 | 
|---|
| 272 |                 else if ( is_target(decl) ) {
 | 
|---|
| 273 |                         handle( decl );
 | 
|---|
| 274 |                 }
 | 
|---|
| 275 | 
 | 
|---|
| 276 |         }
 | 
|---|
| 277 | 
 | 
|---|
| 278 |         void ConcurrentSueKeyword::handle( StructDecl * decl ) {
 | 
|---|
| 279 |                 if( ! decl->body ) return;
 | 
|---|
| 280 | 
 | 
|---|
| 281 |                 if( !type_decl ) throw SemanticError( context_error, decl );
 | 
|---|
| 282 | 
 | 
|---|
| 283 |                 FunctionDecl * func = forwardDeclare( decl );
 | 
|---|
| 284 |                 ObjectDecl * field = addField( decl );
 | 
|---|
| 285 |                 addRoutines( field, func );
 | 
|---|
| 286 |         }
 | 
|---|
| 287 | 
 | 
|---|
| 288 |         FunctionDecl * ConcurrentSueKeyword::forwardDeclare( StructDecl * decl ) {
 | 
|---|
| 289 | 
 | 
|---|
| 290 |                 StructDecl * forward = decl->clone();
 | 
|---|
| 291 |                 forward->set_body( false );
 | 
|---|
| 292 |                 deleteAll( forward->get_members() );
 | 
|---|
| 293 |                 forward->get_members().clear();
 | 
|---|
| 294 | 
 | 
|---|
| 295 |                 FunctionType * get_type = new FunctionType( noQualifiers, false );
 | 
|---|
| 296 |                 ObjectDecl * this_decl = new ObjectDecl(
 | 
|---|
| 297 |                         "this",
 | 
|---|
| 298 |                         noStorage,
 | 
|---|
| 299 |                         LinkageSpec::Cforall,
 | 
|---|
| 300 |                         nullptr,
 | 
|---|
| 301 |                         new ReferenceType(
 | 
|---|
| 302 |                                 noQualifiers,
 | 
|---|
| 303 |                                 new StructInstType(
 | 
|---|
| 304 |                                         noQualifiers,
 | 
|---|
| 305 |                                         decl
 | 
|---|
| 306 |                                 )
 | 
|---|
| 307 |                         ),
 | 
|---|
| 308 |                         nullptr
 | 
|---|
| 309 |                 );
 | 
|---|
| 310 | 
 | 
|---|
| 311 |                 get_type->get_parameters().push_back( this_decl );
 | 
|---|
| 312 |                 get_type->get_returnVals().push_back(
 | 
|---|
| 313 |                         new ObjectDecl(
 | 
|---|
| 314 |                                 "ret",
 | 
|---|
| 315 |                                 noStorage,
 | 
|---|
| 316 |                                 LinkageSpec::Cforall,
 | 
|---|
| 317 |                                 nullptr,
 | 
|---|
| 318 |                                 new PointerType(
 | 
|---|
| 319 |                                         noQualifiers,
 | 
|---|
| 320 |                                         new StructInstType(
 | 
|---|
| 321 |                                                 noQualifiers,
 | 
|---|
| 322 |                                                 type_decl
 | 
|---|
| 323 |                                         )
 | 
|---|
| 324 |                                 ),
 | 
|---|
| 325 |                                 nullptr
 | 
|---|
| 326 |                         )
 | 
|---|
| 327 |                 );
 | 
|---|
| 328 | 
 | 
|---|
| 329 |                 FunctionDecl * get_decl = new FunctionDecl(
 | 
|---|
| 330 |                         getter_name,
 | 
|---|
| 331 |                         Type::Static,
 | 
|---|
| 332 |                         LinkageSpec::Cforall,
 | 
|---|
| 333 |                         get_type,
 | 
|---|
| 334 |                         nullptr,
 | 
|---|
| 335 |                         noAttributes,
 | 
|---|
| 336 |                         Type::Inline
 | 
|---|
| 337 |                 );
 | 
|---|
| 338 | 
 | 
|---|
| 339 |                 FunctionDecl * main_decl = nullptr;
 | 
|---|
| 340 | 
 | 
|---|
| 341 |                 if( needs_main ) {
 | 
|---|
| 342 |                         FunctionType * main_type = new FunctionType( noQualifiers, false );
 | 
|---|
| 343 | 
 | 
|---|
| 344 |                         main_type->get_parameters().push_back( this_decl->clone() );
 | 
|---|
| 345 | 
 | 
|---|
| 346 |                         main_decl = new FunctionDecl(
 | 
|---|
| 347 |                                 "main",
 | 
|---|
| 348 |                                 noStorage,
 | 
|---|
| 349 |                                 LinkageSpec::Cforall,
 | 
|---|
| 350 |                                 main_type,
 | 
|---|
| 351 |                                 nullptr
 | 
|---|
| 352 |                         );
 | 
|---|
| 353 |                 }
 | 
|---|
| 354 | 
 | 
|---|
| 355 |                 declsToAddBefore.push_back( forward );
 | 
|---|
| 356 |                 if( needs_main ) declsToAddBefore.push_back( main_decl );
 | 
|---|
| 357 |                 declsToAddBefore.push_back( get_decl );
 | 
|---|
| 358 | 
 | 
|---|
| 359 |                 return get_decl;
 | 
|---|
| 360 |         }
 | 
|---|
| 361 | 
 | 
|---|
| 362 |         ObjectDecl * ConcurrentSueKeyword::addField( StructDecl * decl ) {
 | 
|---|
| 363 |                 ObjectDecl * field = new ObjectDecl(
 | 
|---|
| 364 |                         field_name,
 | 
|---|
| 365 |                         noStorage,
 | 
|---|
| 366 |                         LinkageSpec::Cforall,
 | 
|---|
| 367 |                         nullptr,
 | 
|---|
| 368 |                         new StructInstType(
 | 
|---|
| 369 |                                 noQualifiers,
 | 
|---|
| 370 |                                 type_decl
 | 
|---|
| 371 |                         ),
 | 
|---|
| 372 |                         nullptr
 | 
|---|
| 373 |                 );
 | 
|---|
| 374 | 
 | 
|---|
| 375 |                 decl->get_members().push_back( field );
 | 
|---|
| 376 | 
 | 
|---|
| 377 |                 return field;
 | 
|---|
| 378 |         }
 | 
|---|
| 379 | 
 | 
|---|
| 380 |         void ConcurrentSueKeyword::addRoutines( ObjectDecl * field, FunctionDecl * func ) {
 | 
|---|
| 381 |                 CompoundStmt * statement = new CompoundStmt( noLabels );
 | 
|---|
| 382 |                 statement->push_back(
 | 
|---|
| 383 |                         new ReturnStmt(
 | 
|---|
| 384 |                                 noLabels,
 | 
|---|
| 385 |                                 new AddressExpr(
 | 
|---|
| 386 |                                         new MemberExpr(
 | 
|---|
| 387 |                                                 field,
 | 
|---|
| 388 |                                                 UntypedExpr::createDeref( new VariableExpr( func->get_functionType()->get_parameters().front() ) )
 | 
|---|
| 389 |                                         )
 | 
|---|
| 390 |                                 )
 | 
|---|
| 391 |                         )
 | 
|---|
| 392 |                 );
 | 
|---|
| 393 | 
 | 
|---|
| 394 |                 FunctionDecl * get_decl = func->clone();
 | 
|---|
| 395 | 
 | 
|---|
| 396 |                 get_decl->set_statements( statement );
 | 
|---|
| 397 | 
 | 
|---|
| 398 |                 declsToAddAfter.push_back( get_decl );
 | 
|---|
| 399 | 
 | 
|---|
| 400 |                 // get_decl->fixUniqueId();
 | 
|---|
| 401 |         }
 | 
|---|
| 402 | 
 | 
|---|
| 403 |         //=============================================================================================
 | 
|---|
| 404 |         // Mutex keyword implementation
 | 
|---|
| 405 |         //=============================================================================================
 | 
|---|
| 406 | 
 | 
|---|
| 407 |         void MutexKeyword::postvisit(FunctionDecl* decl) {
 | 
|---|
| 408 | 
 | 
|---|
| 409 |                 std::list<DeclarationWithType*> mutexArgs = findMutexArgs( decl );
 | 
|---|
| 410 |                 if( mutexArgs.empty() ) return;
 | 
|---|
| 411 | 
 | 
|---|
| 412 |                 if( CodeGen::isConstructor(decl->name) ) throw SemanticError( "constructors cannot have mutex parameters", decl );
 | 
|---|
| 413 | 
 | 
|---|
| 414 |                 bool isDtor = CodeGen::isDestructor( decl->name );
 | 
|---|
| 415 | 
 | 
|---|
| 416 |                 if( isDtor && mutexArgs.size() != 1 ) throw SemanticError( "destructors can only have 1 mutex argument", decl );
 | 
|---|
| 417 | 
 | 
|---|
| 418 |                 for(auto arg : mutexArgs) {
 | 
|---|
| 419 |                         validate( arg );
 | 
|---|
| 420 |                 }
 | 
|---|
| 421 | 
 | 
|---|
| 422 |                 CompoundStmt* body = decl->get_statements();
 | 
|---|
| 423 |                 if( ! body ) return;
 | 
|---|
| 424 | 
 | 
|---|
| 425 |                 if( !monitor_decl ) throw SemanticError( "mutex keyword requires monitors to be in scope, add #include <monitor>", decl );
 | 
|---|
| 426 |                 if( !guard_decl ) throw SemanticError( "mutex keyword requires monitors to be in scope, add #include <monitor>", decl );
 | 
|---|
| 427 |                 if( !dtor_guard_decl ) throw SemanticError( "mutex keyword requires monitors to be in scope, add #include <monitor>", decl );
 | 
|---|
| 428 | 
 | 
|---|
| 429 |                 if( isDtor ) {
 | 
|---|
| 430 |                         addDtorStatments( decl, body, mutexArgs );
 | 
|---|
| 431 |                 }
 | 
|---|
| 432 |                 else {
 | 
|---|
| 433 |                         addStatments( decl, body, mutexArgs );
 | 
|---|
| 434 |                 }
 | 
|---|
| 435 |         }
 | 
|---|
| 436 | 
 | 
|---|
| 437 |         void MutexKeyword::postvisit(StructDecl* decl) {
 | 
|---|
| 438 | 
 | 
|---|
| 439 |                 if( decl->name == "monitor_desc" ) {
 | 
|---|
| 440 |                         assert( !monitor_decl );
 | 
|---|
| 441 |                         monitor_decl = decl;
 | 
|---|
| 442 |                 }
 | 
|---|
| 443 |                 else if( decl->name == "monitor_guard_t" ) {
 | 
|---|
| 444 |                         assert( !guard_decl );
 | 
|---|
| 445 |                         guard_decl = decl;
 | 
|---|
| 446 |                 }
 | 
|---|
| 447 |                 else if( decl->name == "monitor_dtor_guard_t" ) {
 | 
|---|
| 448 |                         assert( !dtor_guard_decl );
 | 
|---|
| 449 |                         dtor_guard_decl = decl;
 | 
|---|
| 450 |                 }
 | 
|---|
| 451 |         }
 | 
|---|
| 452 | 
 | 
|---|
| 453 |         std::list<DeclarationWithType*> MutexKeyword::findMutexArgs( FunctionDecl* decl ) {
 | 
|---|
| 454 |                 std::list<DeclarationWithType*> mutexArgs;
 | 
|---|
| 455 | 
 | 
|---|
| 456 |                 for( auto arg : decl->get_functionType()->get_parameters()) {
 | 
|---|
| 457 |                         //Find mutex arguments
 | 
|---|
| 458 |                         Type* ty = arg->get_type();
 | 
|---|
| 459 |                         if( ! ty->get_mutex() ) continue;
 | 
|---|
| 460 | 
 | 
|---|
| 461 |                         //Append it to the list
 | 
|---|
| 462 |                         mutexArgs.push_back( arg );
 | 
|---|
| 463 |                 }
 | 
|---|
| 464 | 
 | 
|---|
| 465 |                 return mutexArgs;
 | 
|---|
| 466 |         }
 | 
|---|
| 467 | 
 | 
|---|
| 468 |         void MutexKeyword::validate( DeclarationWithType * arg ) {
 | 
|---|
| 469 |                 Type* ty = arg->get_type();
 | 
|---|
| 470 | 
 | 
|---|
| 471 |                 //Makes sure it's not a copy
 | 
|---|
| 472 |                 ReferenceType* rty = dynamic_cast< ReferenceType * >( ty );
 | 
|---|
| 473 |                 if( ! rty ) throw SemanticError( "Mutex argument must be of reference type ", arg );
 | 
|---|
| 474 | 
 | 
|---|
| 475 |                 //Make sure the we are pointing directly to a type
 | 
|---|
| 476 |                 Type* base = rty->get_base();
 | 
|---|
| 477 |                 if( dynamic_cast< ReferenceType * >( base ) ) throw SemanticError( "Mutex argument have exactly one level of indirection ", arg );
 | 
|---|
| 478 |                 if( dynamic_cast< PointerType * >( base ) ) throw SemanticError( "Mutex argument have exactly one level of indirection ", arg );
 | 
|---|
| 479 | 
 | 
|---|
| 480 |                 //Make sure that typed isn't mutex
 | 
|---|
| 481 |                 if( base->get_mutex() ) throw SemanticError( "mutex keyword may only appear once per argument ", arg );
 | 
|---|
| 482 |         }
 | 
|---|
| 483 | 
 | 
|---|
| 484 |         void MutexKeyword::addDtorStatments( FunctionDecl* func, CompoundStmt * body, const std::list<DeclarationWithType * > & args ) {
 | 
|---|
| 485 |                 Type * arg_type = args.front()->get_type()->clone();
 | 
|---|
| 486 |                 arg_type->set_mutex( false );
 | 
|---|
| 487 | 
 | 
|---|
| 488 |                 ObjectDecl * monitors = new ObjectDecl(
 | 
|---|
| 489 |                         "__monitor",
 | 
|---|
| 490 |                         noStorage,
 | 
|---|
| 491 |                         LinkageSpec::Cforall,
 | 
|---|
| 492 |                         nullptr,
 | 
|---|
| 493 |                         new PointerType(
 | 
|---|
| 494 |                                 noQualifiers,
 | 
|---|
| 495 |                                 new StructInstType(
 | 
|---|
| 496 |                                         noQualifiers,
 | 
|---|
| 497 |                                         monitor_decl
 | 
|---|
| 498 |                                 )
 | 
|---|
| 499 |                         ),
 | 
|---|
| 500 |                         new SingleInit( new UntypedExpr(
 | 
|---|
| 501 |                                 new NameExpr( "get_monitor" ),
 | 
|---|
| 502 |                                 {  new CastExpr( new VariableExpr( args.front() ), arg_type ) }
 | 
|---|
| 503 |                         ))
 | 
|---|
| 504 |                 );
 | 
|---|
| 505 | 
 | 
|---|
| 506 |                 assert(generic_func);
 | 
|---|
| 507 | 
 | 
|---|
| 508 |                 //in reverse order :
 | 
|---|
| 509 |                 // monitor_guard_t __guard = { __monitors, #, func };
 | 
|---|
| 510 |                 body->push_front(
 | 
|---|
| 511 |                         new DeclStmt( noLabels, new ObjectDecl(
 | 
|---|
| 512 |                                 "__guard",
 | 
|---|
| 513 |                                 noStorage,
 | 
|---|
| 514 |                                 LinkageSpec::Cforall,
 | 
|---|
| 515 |                                 nullptr,
 | 
|---|
| 516 |                                 new StructInstType(
 | 
|---|
| 517 |                                         noQualifiers,
 | 
|---|
| 518 |                                         dtor_guard_decl
 | 
|---|
| 519 |                                 ),
 | 
|---|
| 520 |                                 new ListInit(
 | 
|---|
| 521 |                                         {
 | 
|---|
| 522 |                                                 new SingleInit( new AddressExpr( new VariableExpr( monitors ) ) ),
 | 
|---|
| 523 |                                                 new SingleInit( new CastExpr( new VariableExpr( func ), generic_func->clone() ) )
 | 
|---|
| 524 |                                         },
 | 
|---|
| 525 |                                         noDesignators,
 | 
|---|
| 526 |                                         true
 | 
|---|
| 527 |                                 )
 | 
|---|
| 528 |                         ))
 | 
|---|
| 529 |                 );
 | 
|---|
| 530 | 
 | 
|---|
| 531 |                 //monitor_desc * __monitors[] = { get_monitor(a), get_monitor(b) };
 | 
|---|
| 532 |                 body->push_front( new DeclStmt( noLabels, monitors) );
 | 
|---|
| 533 |         }
 | 
|---|
| 534 | 
 | 
|---|
| 535 |         void MutexKeyword::addStatments( FunctionDecl* func, CompoundStmt * body, const std::list<DeclarationWithType * > & args ) {
 | 
|---|
| 536 |                 ObjectDecl * monitors = new ObjectDecl(
 | 
|---|
| 537 |                         "__monitors",
 | 
|---|
| 538 |                         noStorage,
 | 
|---|
| 539 |                         LinkageSpec::Cforall,
 | 
|---|
| 540 |                         nullptr,
 | 
|---|
| 541 |                         new ArrayType(
 | 
|---|
| 542 |                                 noQualifiers,
 | 
|---|
| 543 |                                 new PointerType(
 | 
|---|
| 544 |                                         noQualifiers,
 | 
|---|
| 545 |                                         new StructInstType(
 | 
|---|
| 546 |                                                 noQualifiers,
 | 
|---|
| 547 |                                                 monitor_decl
 | 
|---|
| 548 |                                         )
 | 
|---|
| 549 |                                 ),
 | 
|---|
| 550 |                                 new ConstantExpr( Constant::from_ulong( args.size() ) ),
 | 
|---|
| 551 |                                 false,
 | 
|---|
| 552 |                                 false
 | 
|---|
| 553 |                         ),
 | 
|---|
| 554 |                         new ListInit(
 | 
|---|
| 555 |                                 map_range < std::list<Initializer*> > ( args, [](DeclarationWithType * var ){
 | 
|---|
| 556 |                                         Type * type = var->get_type()->clone();
 | 
|---|
| 557 |                                         type->set_mutex( false );
 | 
|---|
| 558 |                                         return new SingleInit( new UntypedExpr(
 | 
|---|
| 559 |                                                 new NameExpr( "get_monitor" ),
 | 
|---|
| 560 |                                                 {  new CastExpr( new VariableExpr( var ), type ) }
 | 
|---|
| 561 |                                         ) );
 | 
|---|
| 562 |                                 })
 | 
|---|
| 563 |                         )
 | 
|---|
| 564 |                 );
 | 
|---|
| 565 | 
 | 
|---|
| 566 |                 assert(generic_func);
 | 
|---|
| 567 | 
 | 
|---|
| 568 |                 //in reverse order :
 | 
|---|
| 569 |                 // monitor_guard_t __guard = { __monitors, #, func };
 | 
|---|
| 570 |                 body->push_front(
 | 
|---|
| 571 |                         new DeclStmt( noLabels, new ObjectDecl(
 | 
|---|
| 572 |                                 "__guard",
 | 
|---|
| 573 |                                 noStorage,
 | 
|---|
| 574 |                                 LinkageSpec::Cforall,
 | 
|---|
| 575 |                                 nullptr,
 | 
|---|
| 576 |                                 new StructInstType(
 | 
|---|
| 577 |                                         noQualifiers,
 | 
|---|
| 578 |                                         guard_decl
 | 
|---|
| 579 |                                 ),
 | 
|---|
| 580 |                                 new ListInit(
 | 
|---|
| 581 |                                         {
 | 
|---|
| 582 |                                                 new SingleInit( new VariableExpr( monitors ) ),
 | 
|---|
| 583 |                                                 new SingleInit( new ConstantExpr( Constant::from_ulong( args.size() ) ) ),
 | 
|---|
| 584 |                                                 new SingleInit( new CastExpr( new VariableExpr( func ), generic_func->clone() ) )
 | 
|---|
| 585 |                                         },
 | 
|---|
| 586 |                                         noDesignators,
 | 
|---|
| 587 |                                         true
 | 
|---|
| 588 |                                 )
 | 
|---|
| 589 |                         ))
 | 
|---|
| 590 |                 );
 | 
|---|
| 591 | 
 | 
|---|
| 592 |                 //monitor_desc * __monitors[] = { get_monitor(a), get_monitor(b) };
 | 
|---|
| 593 |                 body->push_front( new DeclStmt( noLabels, monitors) );
 | 
|---|
| 594 |         }
 | 
|---|
| 595 | 
 | 
|---|
| 596 |         //=============================================================================================
 | 
|---|
| 597 |         // General entry routine
 | 
|---|
| 598 |         //=============================================================================================
 | 
|---|
| 599 |         void ThreadStarter::previsit( StructDecl * decl ) {
 | 
|---|
| 600 |                 if( decl->name == "thread_desc" && decl->body ) {
 | 
|---|
| 601 |                         assert( !thread_decl );
 | 
|---|
| 602 |                         thread_decl = decl;
 | 
|---|
| 603 |                 }
 | 
|---|
| 604 |         }
 | 
|---|
| 605 | 
 | 
|---|
| 606 |         void ThreadStarter::postvisit(FunctionDecl * decl) {
 | 
|---|
| 607 |                 if( ! CodeGen::isConstructor(decl->name) ) return;
 | 
|---|
| 608 | 
 | 
|---|
| 609 |                 Type * typeof_this = InitTweak::getTypeofThis(decl->type);
 | 
|---|
| 610 |                 StructInstType * ctored_type = dynamic_cast< StructInstType * >( typeof_this );
 | 
|---|
| 611 |                 if( ctored_type && ctored_type->baseStruct == thread_decl ) {
 | 
|---|
| 612 |                         thread_ctor_seen = true;
 | 
|---|
| 613 |                 }
 | 
|---|
| 614 | 
 | 
|---|
| 615 |                 DeclarationWithType * param = decl->get_functionType()->get_parameters().front();
 | 
|---|
| 616 |                 auto type  = dynamic_cast< StructInstType * >( InitTweak::getPointerBase( param->get_type() ) );
 | 
|---|
| 617 |                 if( type && type->get_baseStruct()->is_thread() ) {
 | 
|---|
| 618 |                         if( !thread_decl || !thread_ctor_seen ) {
 | 
|---|
| 619 |                                 throw SemanticError("thread keyword requires threads to be in scope, add #include <thread>");
 | 
|---|
| 620 |                         }
 | 
|---|
| 621 | 
 | 
|---|
| 622 |                         addStartStatement( decl, param );
 | 
|---|
| 623 |                 }
 | 
|---|
| 624 |         }
 | 
|---|
| 625 | 
 | 
|---|
| 626 |         void ThreadStarter::addStartStatement( FunctionDecl * decl, DeclarationWithType * param ) {
 | 
|---|
| 627 |                 CompoundStmt * stmt = decl->get_statements();
 | 
|---|
| 628 | 
 | 
|---|
| 629 |                 if( ! stmt ) return;
 | 
|---|
| 630 | 
 | 
|---|
| 631 |                 stmt->push_back(
 | 
|---|
| 632 |                         new ExprStmt(
 | 
|---|
| 633 |                                 noLabels,
 | 
|---|
| 634 |                                 new UntypedExpr(
 | 
|---|
| 635 |                                         new NameExpr( "__thrd_start" ),
 | 
|---|
| 636 |                                         { new VariableExpr( param ) }
 | 
|---|
| 637 |                                 )
 | 
|---|
| 638 |                         )
 | 
|---|
| 639 |                 );
 | 
|---|
| 640 |         }
 | 
|---|
| 641 | };
 | 
|---|
| 642 | 
 | 
|---|
| 643 | // Local Variables: //
 | 
|---|
| 644 | // mode: c //
 | 
|---|
| 645 | // tab-width: 4 //
 | 
|---|
| 646 | // End: //
 | 
|---|