Context Navigation

source: src/Concurrency/Keywords.cc @ 03bd407

ADTaaron-thesisarm-ehast-experimentalcleanup-dtorsdeferred_resndemanglerenumforall-pointer-decayjacob/cs343-translationjenkins-sandboxnew-astnew-ast-unique-exprnew-envno_listpersistent-indexerpthread-emulationqualifiedEnumwith_gc

Last change on this file since 03bd407 was 9a705dc8, checked in by Rob Schluntz <rschlunt@…>, 7 years ago
Implement concurrency keyword casts
Property mode set to `100644`
File size: 20.7 KB

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

Note: See TracBrowser for help on using the repository browser.

Download in other formats: