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 | // Mangler.cc --
|
---|
8 | //
|
---|
9 | // Author : Richard C. Bilson
|
---|
10 | // Created On : Sun May 17 21:40:29 2015
|
---|
11 | // Last Modified By : Peter A. Buhr
|
---|
12 | // Last Modified On : Sat Feb 15 13:55:12 2020
|
---|
13 | // Update Count : 33
|
---|
14 | //
|
---|
15 | #include "Mangler.h"
|
---|
16 |
|
---|
17 | #include <algorithm> // for copy, transform
|
---|
18 | #include <cassert> // for assert, assertf
|
---|
19 | #include <functional> // for const_mem_fun_t, mem_fun
|
---|
20 | #include <iterator> // for ostream_iterator, back_insert_ite...
|
---|
21 | #include <list> // for _List_iterator, list, _List_const...
|
---|
22 | #include <string> // for string, char_traits, operator<<
|
---|
23 |
|
---|
24 | #include "CodeGen/OperatorTable.h" // for OperatorInfo, operatorLookup
|
---|
25 | #include "Common/PassVisitor.h"
|
---|
26 | #include "Common/SemanticError.h" // for SemanticError
|
---|
27 | #include "Common/utility.h" // for toString
|
---|
28 | #include "ResolvExpr/TypeEnvironment.h" // for TypeEnvironment
|
---|
29 | #include "SynTree/LinkageSpec.h" // for Spec, isOverridable, AutoGen, Int...
|
---|
30 | #include "SynTree/Declaration.h" // for TypeDecl, DeclarationWithType
|
---|
31 | #include "SynTree/Expression.h" // for TypeExpr, Expression, operator<<
|
---|
32 | #include "SynTree/Type.h" // for Type, ReferenceToType, Type::Fora...
|
---|
33 |
|
---|
34 | #include "AST/Pass.hpp"
|
---|
35 |
|
---|
36 | namespace SymTab {
|
---|
37 | namespace Mangler {
|
---|
38 | namespace {
|
---|
39 | /// Mangles names to a unique C identifier
|
---|
40 | struct Mangler_old : public WithShortCircuiting, public WithVisitorRef<Mangler_old>, public WithGuards {
|
---|
41 | Mangler_old( bool mangleOverridable, bool typeMode, bool mangleGenericParams );
|
---|
42 | Mangler_old( const Mangler_old & ) = delete;
|
---|
43 |
|
---|
44 | void previsit( const BaseSyntaxNode * ) { visit_children = false; }
|
---|
45 |
|
---|
46 | void postvisit( const ObjectDecl * declaration );
|
---|
47 | void postvisit( const FunctionDecl * declaration );
|
---|
48 | void postvisit( const TypeDecl * declaration );
|
---|
49 |
|
---|
50 | void postvisit( const VoidType * voidType );
|
---|
51 | void postvisit( const BasicType * basicType );
|
---|
52 | void postvisit( const PointerType * pointerType );
|
---|
53 | void postvisit( const ArrayType * arrayType );
|
---|
54 | void postvisit( const ReferenceType * refType );
|
---|
55 | void postvisit( const FunctionType * functionType );
|
---|
56 | void postvisit( const StructInstType * aggregateUseType );
|
---|
57 | void postvisit( const UnionInstType * aggregateUseType );
|
---|
58 | void postvisit( const EnumInstType * aggregateUseType );
|
---|
59 | void postvisit( const TypeInstType * aggregateUseType );
|
---|
60 | void postvisit( const TraitInstType * inst );
|
---|
61 | void postvisit( const TupleType * tupleType );
|
---|
62 | void postvisit( const VarArgsType * varArgsType );
|
---|
63 | void postvisit( const ZeroType * zeroType );
|
---|
64 | void postvisit( const OneType * oneType );
|
---|
65 | void postvisit( const QualifiedType * qualType );
|
---|
66 |
|
---|
67 | std::string get_mangleName() { return mangleName.str(); }
|
---|
68 | private:
|
---|
69 | std::ostringstream mangleName; ///< Mangled name being constructed
|
---|
70 | typedef std::map< std::string, std::pair< int, int > > VarMapType;
|
---|
71 | VarMapType varNums; ///< Map of type variables to indices
|
---|
72 | int nextVarNum; ///< Next type variable index
|
---|
73 | bool isTopLevel; ///< Is the Mangler at the top level
|
---|
74 | bool mangleOverridable; ///< Specially mangle overridable built-in methods
|
---|
75 | bool typeMode; ///< Produce a unique mangled name for a type
|
---|
76 | bool mangleGenericParams; ///< Include generic parameters in name mangling if true
|
---|
77 | bool inFunctionType = false; ///< Include type qualifiers if false.
|
---|
78 | bool inQualifiedType = false; ///< Add start/end delimiters around qualified type
|
---|
79 |
|
---|
80 | public:
|
---|
81 | Mangler_old( bool mangleOverridable, bool typeMode, bool mangleGenericParams,
|
---|
82 | int nextVarNum, const VarMapType& varNums );
|
---|
83 |
|
---|
84 | private:
|
---|
85 | void mangleDecl( const DeclarationWithType * declaration );
|
---|
86 | void mangleRef( const ReferenceToType * refType, std::string prefix );
|
---|
87 |
|
---|
88 | void printQualifiers( const Type *type );
|
---|
89 | }; // Mangler_old
|
---|
90 | } // namespace
|
---|
91 |
|
---|
92 | std::string mangle( const BaseSyntaxNode * decl, bool mangleOverridable, bool typeMode, bool mangleGenericParams ) {
|
---|
93 | PassVisitor<Mangler_old> mangler( mangleOverridable, typeMode, mangleGenericParams );
|
---|
94 | maybeAccept( decl, mangler );
|
---|
95 | return mangler.pass.get_mangleName();
|
---|
96 | }
|
---|
97 |
|
---|
98 | std::string mangleType( const Type * ty ) {
|
---|
99 | PassVisitor<Mangler_old> mangler( false, true, true );
|
---|
100 | maybeAccept( ty, mangler );
|
---|
101 | return mangler.pass.get_mangleName();
|
---|
102 | }
|
---|
103 |
|
---|
104 | std::string mangleConcrete( const Type * ty ) {
|
---|
105 | PassVisitor<Mangler_old> mangler( false, false, false );
|
---|
106 | maybeAccept( ty, mangler );
|
---|
107 | return mangler.pass.get_mangleName();
|
---|
108 | }
|
---|
109 |
|
---|
110 | namespace {
|
---|
111 | Mangler_old::Mangler_old( bool mangleOverridable, bool typeMode, bool mangleGenericParams )
|
---|
112 | : nextVarNum( 0 ), isTopLevel( true ),
|
---|
113 | mangleOverridable( mangleOverridable ), typeMode( typeMode ),
|
---|
114 | mangleGenericParams( mangleGenericParams ) {}
|
---|
115 |
|
---|
116 | Mangler_old::Mangler_old( bool mangleOverridable, bool typeMode, bool mangleGenericParams,
|
---|
117 | int nextVarNum, const VarMapType& varNums )
|
---|
118 | : varNums( varNums ), nextVarNum( nextVarNum ), isTopLevel( false ),
|
---|
119 | mangleOverridable( mangleOverridable ), typeMode( typeMode ),
|
---|
120 | mangleGenericParams( mangleGenericParams ) {}
|
---|
121 |
|
---|
122 | void Mangler_old::mangleDecl( const DeclarationWithType * declaration ) {
|
---|
123 | bool wasTopLevel = isTopLevel;
|
---|
124 | if ( isTopLevel ) {
|
---|
125 | varNums.clear();
|
---|
126 | nextVarNum = 0;
|
---|
127 | isTopLevel = false;
|
---|
128 | } // if
|
---|
129 | mangleName << Encoding::manglePrefix;
|
---|
130 | const CodeGen::OperatorInfo * opInfo = CodeGen::operatorLookup( declaration->get_name() );
|
---|
131 | if ( opInfo ) {
|
---|
132 | mangleName << opInfo->outputName.size() << opInfo->outputName;
|
---|
133 | } else {
|
---|
134 | mangleName << declaration->name.size() << declaration->name;
|
---|
135 | } // if
|
---|
136 | maybeAccept( declaration->get_type(), *visitor );
|
---|
137 | if ( mangleOverridable && LinkageSpec::isOverridable( declaration->get_linkage() ) ) {
|
---|
138 | // want to be able to override autogenerated and intrinsic routines,
|
---|
139 | // so they need a different name mangling
|
---|
140 | if ( declaration->get_linkage() == LinkageSpec::AutoGen ) {
|
---|
141 | mangleName << Encoding::autogen;
|
---|
142 | } else if ( declaration->get_linkage() == LinkageSpec::Intrinsic ) {
|
---|
143 | mangleName << Encoding::intrinsic;
|
---|
144 | } else {
|
---|
145 | // if we add another kind of overridable function, this has to change
|
---|
146 | assert( false && "unknown overrideable linkage" );
|
---|
147 | } // if
|
---|
148 | }
|
---|
149 | isTopLevel = wasTopLevel;
|
---|
150 | }
|
---|
151 |
|
---|
152 | void Mangler_old::postvisit( const ObjectDecl * declaration ) {
|
---|
153 | mangleDecl( declaration );
|
---|
154 | }
|
---|
155 |
|
---|
156 | void Mangler_old::postvisit( const FunctionDecl * declaration ) {
|
---|
157 | mangleDecl( declaration );
|
---|
158 | }
|
---|
159 |
|
---|
160 | void Mangler_old::postvisit( const VoidType * voidType ) {
|
---|
161 | printQualifiers( voidType );
|
---|
162 | mangleName << Encoding::void_t;
|
---|
163 | }
|
---|
164 |
|
---|
165 | void Mangler_old::postvisit( const BasicType * basicType ) {
|
---|
166 | printQualifiers( basicType );
|
---|
167 | assertf( basicType->kind < BasicType::NUMBER_OF_BASIC_TYPES, "Unhandled basic type: %d", basicType->kind );
|
---|
168 | mangleName << Encoding::basicTypes[ basicType->kind ];
|
---|
169 | }
|
---|
170 |
|
---|
171 | void Mangler_old::postvisit( const PointerType * pointerType ) {
|
---|
172 | printQualifiers( pointerType );
|
---|
173 | // mangle void (*f)() and void f() to the same name to prevent overloading on functions and function pointers
|
---|
174 | if ( ! dynamic_cast<FunctionType *>( pointerType->base ) ) mangleName << Encoding::pointer;
|
---|
175 | maybeAccept( pointerType->base, *visitor );
|
---|
176 | }
|
---|
177 |
|
---|
178 | void Mangler_old::postvisit( const ArrayType * arrayType ) {
|
---|
179 | // TODO: encode dimension
|
---|
180 | printQualifiers( arrayType );
|
---|
181 | mangleName << Encoding::array << "0";
|
---|
182 | maybeAccept( arrayType->base, *visitor );
|
---|
183 | }
|
---|
184 |
|
---|
185 | void Mangler_old::postvisit( const ReferenceType * refType ) {
|
---|
186 | // don't print prefix (e.g. 'R') for reference types so that references and non-references do not overload.
|
---|
187 | // Further, do not print the qualifiers for a reference type (but do run printQualifers because of TypeDecls, etc.),
|
---|
188 | // by pretending every reference type is a function parameter.
|
---|
189 | GuardValue( inFunctionType );
|
---|
190 | inFunctionType = true;
|
---|
191 | printQualifiers( refType );
|
---|
192 | maybeAccept( refType->base, *visitor );
|
---|
193 | }
|
---|
194 |
|
---|
195 | namespace {
|
---|
196 | inline std::list< Type* > getTypes( const std::list< DeclarationWithType* > decls ) {
|
---|
197 | std::list< Type* > ret;
|
---|
198 | std::transform( decls.begin(), decls.end(), std::back_inserter( ret ),
|
---|
199 | std::mem_fun( &DeclarationWithType::get_type ) );
|
---|
200 | return ret;
|
---|
201 | }
|
---|
202 | }
|
---|
203 |
|
---|
204 | void Mangler_old::postvisit( const FunctionType * functionType ) {
|
---|
205 | printQualifiers( functionType );
|
---|
206 | mangleName << Encoding::function;
|
---|
207 | // turn on inFunctionType so that printQualifiers does not print most qualifiers for function parameters,
|
---|
208 | // since qualifiers on outermost parameter type do not differentiate function types, e.g.,
|
---|
209 | // void (*)(const int) and void (*)(int) are the same type, but void (*)(const int *) and void (*)(int *) are different
|
---|
210 | GuardValue( inFunctionType );
|
---|
211 | inFunctionType = true;
|
---|
212 | std::list< Type* > returnTypes = getTypes( functionType->returnVals );
|
---|
213 | if (returnTypes.empty()) mangleName << Encoding::void_t;
|
---|
214 | else acceptAll( returnTypes, *visitor );
|
---|
215 | mangleName << "_";
|
---|
216 | std::list< Type* > paramTypes = getTypes( functionType->parameters );
|
---|
217 | acceptAll( paramTypes, *visitor );
|
---|
218 | mangleName << "_";
|
---|
219 | }
|
---|
220 |
|
---|
221 | void Mangler_old::mangleRef( const ReferenceToType * refType, std::string prefix ) {
|
---|
222 | printQualifiers( refType );
|
---|
223 |
|
---|
224 | mangleName << prefix << refType->name.length() << refType->name;
|
---|
225 |
|
---|
226 | if ( mangleGenericParams ) {
|
---|
227 | const std::list< Expression* > & params = refType->parameters;
|
---|
228 | if ( ! params.empty() ) {
|
---|
229 | mangleName << "_";
|
---|
230 | for ( const Expression * param : params ) {
|
---|
231 | const TypeExpr * paramType = dynamic_cast< const TypeExpr * >( param );
|
---|
232 | assertf(paramType, "Aggregate parameters should be type expressions: %s", toCString(param));
|
---|
233 | maybeAccept( paramType->type, *visitor );
|
---|
234 | }
|
---|
235 | mangleName << "_";
|
---|
236 | }
|
---|
237 | }
|
---|
238 | }
|
---|
239 |
|
---|
240 | void Mangler_old::postvisit( const StructInstType * aggregateUseType ) {
|
---|
241 | mangleRef( aggregateUseType, Encoding::struct_t );
|
---|
242 | }
|
---|
243 |
|
---|
244 | void Mangler_old::postvisit( const UnionInstType * aggregateUseType ) {
|
---|
245 | mangleRef( aggregateUseType, Encoding::union_t );
|
---|
246 | }
|
---|
247 |
|
---|
248 | void Mangler_old::postvisit( const EnumInstType * aggregateUseType ) {
|
---|
249 | mangleRef( aggregateUseType, Encoding::enum_t );
|
---|
250 | }
|
---|
251 |
|
---|
252 | void Mangler_old::postvisit( const TypeInstType * typeInst ) {
|
---|
253 | VarMapType::iterator varNum = varNums.find( typeInst->get_name() );
|
---|
254 | if ( varNum == varNums.end() ) {
|
---|
255 | mangleRef( typeInst, Encoding::type );
|
---|
256 | } else {
|
---|
257 | printQualifiers( typeInst );
|
---|
258 | // Note: Can't use name here, since type variable names do not actually disambiguate a function, e.g.
|
---|
259 | // forall(dtype T) void f(T);
|
---|
260 | // forall(dtype S) void f(S);
|
---|
261 | // are equivalent and should mangle the same way. This is accomplished by numbering the type variables when they
|
---|
262 | // are first found and prefixing with the appropriate encoding for the type class.
|
---|
263 | assertf( varNum->second.second < TypeDecl::NUMBER_OF_KINDS, "Unhandled type variable kind: %d", varNum->second.second );
|
---|
264 | mangleName << Encoding::typeVariables[varNum->second.second] << varNum->second.first;
|
---|
265 | } // if
|
---|
266 | }
|
---|
267 |
|
---|
268 | void Mangler_old::postvisit( const TraitInstType * inst ) {
|
---|
269 | printQualifiers( inst );
|
---|
270 | mangleName << inst->name.size() << inst->name;
|
---|
271 | }
|
---|
272 |
|
---|
273 | void Mangler_old::postvisit( const TupleType * tupleType ) {
|
---|
274 | printQualifiers( tupleType );
|
---|
275 | mangleName << Encoding::tuple << tupleType->types.size();
|
---|
276 | acceptAll( tupleType->types, *visitor );
|
---|
277 | }
|
---|
278 |
|
---|
279 | void Mangler_old::postvisit( const VarArgsType * varArgsType ) {
|
---|
280 | printQualifiers( varArgsType );
|
---|
281 | static const std::string vargs = "__builtin_va_list";
|
---|
282 | mangleName << Encoding::type << vargs.size() << vargs;
|
---|
283 | }
|
---|
284 |
|
---|
285 | void Mangler_old::postvisit( const ZeroType * ) {
|
---|
286 | mangleName << Encoding::zero;
|
---|
287 | }
|
---|
288 |
|
---|
289 | void Mangler_old::postvisit( const OneType * ) {
|
---|
290 | mangleName << Encoding::one;
|
---|
291 | }
|
---|
292 |
|
---|
293 | void Mangler_old::postvisit( const QualifiedType * qualType ) {
|
---|
294 | bool inqual = inQualifiedType;
|
---|
295 | if (! inqual ) {
|
---|
296 | // N marks the start of a qualified type
|
---|
297 | inQualifiedType = true;
|
---|
298 | mangleName << Encoding::qualifiedTypeStart;
|
---|
299 | }
|
---|
300 | maybeAccept( qualType->parent, *visitor );
|
---|
301 | maybeAccept( qualType->child, *visitor );
|
---|
302 | if ( ! inqual ) {
|
---|
303 | // E marks the end of a qualified type
|
---|
304 | inQualifiedType = false;
|
---|
305 | mangleName << Encoding::qualifiedTypeEnd;
|
---|
306 | }
|
---|
307 | }
|
---|
308 |
|
---|
309 | void Mangler_old::postvisit( const TypeDecl * decl ) {
|
---|
310 | // TODO: is there any case where mangling a TypeDecl makes sense? If so, this code needs to be
|
---|
311 | // fixed to ensure that two TypeDecls mangle to the same name when they are the same type and vice versa.
|
---|
312 | // Note: The current scheme may already work correctly for this case, I have not thought about this deeply
|
---|
313 | // and the case has not yet come up in practice. Alternatively, if not then this code can be removed
|
---|
314 | // aside from the assert false.
|
---|
315 | assertf(false, "Mangler_old should not visit typedecl: %s", toCString(decl));
|
---|
316 | assertf( decl->kind < TypeDecl::NUMBER_OF_KINDS, "Unhandled type variable kind: %d", decl->kind );
|
---|
317 | mangleName << Encoding::typeVariables[ decl->kind ] << ( decl->name.length() ) << decl->name;
|
---|
318 | }
|
---|
319 |
|
---|
320 | __attribute__((unused)) void printVarMap( const std::map< std::string, std::pair< int, int > > &varMap, std::ostream &os ) {
|
---|
321 | for ( std::map< std::string, std::pair< int, int > >::const_iterator i = varMap.begin(); i != varMap.end(); ++i ) {
|
---|
322 | os << i->first << "(" << i->second.first << "/" << i->second.second << ")" << std::endl;
|
---|
323 | } // for
|
---|
324 | }
|
---|
325 |
|
---|
326 | void Mangler_old::printQualifiers( const Type * type ) {
|
---|
327 | // skip if not including qualifiers
|
---|
328 | if ( typeMode ) return;
|
---|
329 | if ( ! type->forall.empty() ) {
|
---|
330 | std::list< std::string > assertionNames;
|
---|
331 | int dcount = 0, fcount = 0, vcount = 0, acount = 0;
|
---|
332 | mangleName << Encoding::forall;
|
---|
333 | for ( const TypeDecl * i : type->forall ) {
|
---|
334 | switch ( i->kind ) {
|
---|
335 | case TypeDecl::Dtype:
|
---|
336 | dcount++;
|
---|
337 | break;
|
---|
338 | case TypeDecl::Ftype:
|
---|
339 | fcount++;
|
---|
340 | break;
|
---|
341 | case TypeDecl::Ttype:
|
---|
342 | vcount++;
|
---|
343 | break;
|
---|
344 | default:
|
---|
345 | assert( false );
|
---|
346 | } // switch
|
---|
347 | varNums[ i->name ] = std::make_pair( nextVarNum, (int)i->kind );
|
---|
348 | for ( const DeclarationWithType * assert : i->assertions ) {
|
---|
349 | PassVisitor<Mangler_old> sub_mangler(
|
---|
350 | mangleOverridable, typeMode, mangleGenericParams, nextVarNum, varNums );
|
---|
351 | assert->accept( sub_mangler );
|
---|
352 | assertionNames.push_back( sub_mangler.pass.get_mangleName() );
|
---|
353 | acount++;
|
---|
354 | } // for
|
---|
355 | } // for
|
---|
356 | mangleName << dcount << "_" << fcount << "_" << vcount << "_" << acount << "_";
|
---|
357 | std::copy( assertionNames.begin(), assertionNames.end(), std::ostream_iterator< std::string >( mangleName, "" ) );
|
---|
358 | mangleName << "_";
|
---|
359 | } // if
|
---|
360 | if ( ! inFunctionType ) {
|
---|
361 | // these qualifiers do not distinguish the outermost type of a function parameter
|
---|
362 | if ( type->get_const() ) {
|
---|
363 | mangleName << Encoding::qualifiers.at(Type::Const);
|
---|
364 | } // if
|
---|
365 | if ( type->get_volatile() ) {
|
---|
366 | mangleName << Encoding::qualifiers.at(Type::Volatile);
|
---|
367 | } // if
|
---|
368 | // Removed due to restrict not affecting function compatibility in GCC
|
---|
369 | // if ( type->get_isRestrict() ) {
|
---|
370 | // mangleName << "E";
|
---|
371 | // } // if
|
---|
372 | if ( type->get_atomic() ) {
|
---|
373 | mangleName << Encoding::qualifiers.at(Type::Atomic);
|
---|
374 | } // if
|
---|
375 | }
|
---|
376 | if ( type->get_mutex() ) {
|
---|
377 | mangleName << Encoding::qualifiers.at(Type::Mutex);
|
---|
378 | } // if
|
---|
379 | if ( inFunctionType ) {
|
---|
380 | // turn off inFunctionType so that types can be differentiated for nested qualifiers
|
---|
381 | GuardValue( inFunctionType );
|
---|
382 | inFunctionType = false;
|
---|
383 | }
|
---|
384 | }
|
---|
385 | } // namespace
|
---|
386 | } // namespace Mangler
|
---|
387 | } // namespace SymTab
|
---|
388 |
|
---|
389 | namespace Mangle {
|
---|
390 | namespace {
|
---|
391 | /// Mangles names to a unique C identifier
|
---|
392 | struct Mangler_new : public ast::WithShortCircuiting, public ast::WithVisitorRef<Mangler_new>, public ast::WithGuards {
|
---|
393 | Mangler_new( Mangle::Mode mode );
|
---|
394 | Mangler_new( const Mangler_new & ) = delete;
|
---|
395 |
|
---|
396 | void previsit( const ast::Node * ) { visit_children = false; }
|
---|
397 |
|
---|
398 | void postvisit( const ast::ObjectDecl * declaration );
|
---|
399 | void postvisit( const ast::FunctionDecl * declaration );
|
---|
400 | void postvisit( const ast::TypeDecl * declaration );
|
---|
401 |
|
---|
402 | void postvisit( const ast::VoidType * voidType );
|
---|
403 | void postvisit( const ast::BasicType * basicType );
|
---|
404 | void postvisit( const ast::PointerType * pointerType );
|
---|
405 | void postvisit( const ast::ArrayType * arrayType );
|
---|
406 | void postvisit( const ast::ReferenceType * refType );
|
---|
407 | void postvisit( const ast::FunctionType * functionType );
|
---|
408 | void postvisit( const ast::StructInstType * aggregateUseType );
|
---|
409 | void postvisit( const ast::UnionInstType * aggregateUseType );
|
---|
410 | void postvisit( const ast::EnumInstType * aggregateUseType );
|
---|
411 | void postvisit( const ast::TypeInstType * aggregateUseType );
|
---|
412 | void postvisit( const ast::TraitInstType * inst );
|
---|
413 | void postvisit( const ast::TupleType * tupleType );
|
---|
414 | void postvisit( const ast::VarArgsType * varArgsType );
|
---|
415 | void postvisit( const ast::ZeroType * zeroType );
|
---|
416 | void postvisit( const ast::OneType * oneType );
|
---|
417 | void postvisit( const ast::QualifiedType * qualType );
|
---|
418 |
|
---|
419 | std::string get_mangleName() { return mangleName.str(); }
|
---|
420 | private:
|
---|
421 | std::ostringstream mangleName; ///< Mangled name being constructed
|
---|
422 | typedef std::map< std::string, std::pair< int, int > > VarMapType;
|
---|
423 | VarMapType varNums; ///< Map of type variables to indices
|
---|
424 | int nextVarNum; ///< Next type variable index
|
---|
425 | bool isTopLevel; ///< Is the Mangler at the top level
|
---|
426 | bool mangleOverridable; ///< Specially mangle overridable built-in methods
|
---|
427 | bool typeMode; ///< Produce a unique mangled name for a type
|
---|
428 | bool mangleGenericParams; ///< Include generic parameters in name mangling if true
|
---|
429 | bool inFunctionType = false; ///< Include type qualifiers if false.
|
---|
430 | bool inQualifiedType = false; ///< Add start/end delimiters around qualified type
|
---|
431 |
|
---|
432 | private:
|
---|
433 | Mangler_new( bool mangleOverridable, bool typeMode, bool mangleGenericParams,
|
---|
434 | int nextVarNum, const VarMapType& varNums );
|
---|
435 | friend class ast::Pass<Mangler_new>;
|
---|
436 |
|
---|
437 | private:
|
---|
438 | void mangleDecl( const ast::DeclWithType *declaration );
|
---|
439 | void mangleRef( const ast::ReferenceToType *refType, std::string prefix );
|
---|
440 |
|
---|
441 | void printQualifiers( const ast::Type *type );
|
---|
442 | }; // Mangler_new
|
---|
443 | } // namespace
|
---|
444 |
|
---|
445 |
|
---|
446 | std::string mangle( const ast::Node * decl, Mangle::Mode mode ) {
|
---|
447 | ast::Pass<Mangler_new> mangler( mode );
|
---|
448 | maybeAccept( decl, mangler );
|
---|
449 | return mangler.core.get_mangleName();
|
---|
450 | }
|
---|
451 |
|
---|
452 | namespace {
|
---|
453 | Mangler_new::Mangler_new( Mangle::Mode mode )
|
---|
454 | : nextVarNum( 0 ), isTopLevel( true ),
|
---|
455 | mangleOverridable ( ! mode.no_overrideable ),
|
---|
456 | typeMode ( mode.type ),
|
---|
457 | mangleGenericParams( ! mode.no_generic_params ) {}
|
---|
458 |
|
---|
459 | Mangler_new::Mangler_new( bool mangleOverridable, bool typeMode, bool mangleGenericParams,
|
---|
460 | int nextVarNum, const VarMapType& varNums )
|
---|
461 | : varNums( varNums ), nextVarNum( nextVarNum ), isTopLevel( false ),
|
---|
462 | mangleOverridable( mangleOverridable ), typeMode( typeMode ),
|
---|
463 | mangleGenericParams( mangleGenericParams ) {}
|
---|
464 |
|
---|
465 | void Mangler_new::mangleDecl( const ast::DeclWithType * decl ) {
|
---|
466 | bool wasTopLevel = isTopLevel;
|
---|
467 | if ( isTopLevel ) {
|
---|
468 | varNums.clear();
|
---|
469 | nextVarNum = 0;
|
---|
470 | isTopLevel = false;
|
---|
471 | } // if
|
---|
472 | mangleName << Encoding::manglePrefix;
|
---|
473 | const CodeGen::OperatorInfo * opInfo = CodeGen::operatorLookup( decl->name );
|
---|
474 | if ( opInfo ) {
|
---|
475 | mangleName << opInfo->outputName.size() << opInfo->outputName;
|
---|
476 | } else {
|
---|
477 | mangleName << decl->name.size() << decl->name;
|
---|
478 | } // if
|
---|
479 | maybeAccept( decl->get_type(), *visitor );
|
---|
480 | if ( mangleOverridable && decl->linkage.is_overrideable ) {
|
---|
481 | // want to be able to override autogenerated and intrinsic routines,
|
---|
482 | // so they need a different name mangling
|
---|
483 | if ( decl->linkage == ast::Linkage::AutoGen ) {
|
---|
484 | mangleName << Encoding::autogen;
|
---|
485 | } else if ( decl->linkage == ast::Linkage::Intrinsic ) {
|
---|
486 | mangleName << Encoding::intrinsic;
|
---|
487 | } else {
|
---|
488 | // if we add another kind of overridable function, this has to change
|
---|
489 | assert( false && "unknown overrideable linkage" );
|
---|
490 | } // if
|
---|
491 | }
|
---|
492 | isTopLevel = wasTopLevel;
|
---|
493 | }
|
---|
494 |
|
---|
495 | void Mangler_new::postvisit( const ast::ObjectDecl * decl ) {
|
---|
496 | mangleDecl( decl );
|
---|
497 | }
|
---|
498 |
|
---|
499 | void Mangler_new::postvisit( const ast::FunctionDecl * decl ) {
|
---|
500 | mangleDecl( decl );
|
---|
501 | }
|
---|
502 |
|
---|
503 | void Mangler_new::postvisit( const ast::VoidType * voidType ) {
|
---|
504 | printQualifiers( voidType );
|
---|
505 | mangleName << Encoding::void_t;
|
---|
506 | }
|
---|
507 |
|
---|
508 | void Mangler_new::postvisit( const ast::BasicType * basicType ) {
|
---|
509 | printQualifiers( basicType );
|
---|
510 | assertf( basicType->kind < ast::BasicType::NUMBER_OF_BASIC_TYPES, "Unhandled basic type: %d", basicType->kind );
|
---|
511 | mangleName << Encoding::basicTypes[ basicType->kind ];
|
---|
512 | }
|
---|
513 |
|
---|
514 | void Mangler_new::postvisit( const ast::PointerType * pointerType ) {
|
---|
515 | printQualifiers( pointerType );
|
---|
516 | // mangle void (*f)() and void f() to the same name to prevent overloading on functions and function pointers
|
---|
517 | if ( ! pointerType->base.as<ast::FunctionType>() ) mangleName << Encoding::pointer;
|
---|
518 | maybe_accept( pointerType->base.get(), *visitor );
|
---|
519 | }
|
---|
520 |
|
---|
521 | void Mangler_new::postvisit( const ast::ArrayType * arrayType ) {
|
---|
522 | // TODO: encode dimension
|
---|
523 | printQualifiers( arrayType );
|
---|
524 | mangleName << Encoding::array << "0";
|
---|
525 | maybeAccept( arrayType->base.get(), *visitor );
|
---|
526 | }
|
---|
527 |
|
---|
528 | void Mangler_new::postvisit( const ast::ReferenceType * refType ) {
|
---|
529 | // don't print prefix (e.g. 'R') for reference types so that references and non-references do not overload.
|
---|
530 | // Further, do not print the qualifiers for a reference type (but do run printQualifers because of TypeDecls, etc.),
|
---|
531 | // by pretending every reference type is a function parameter.
|
---|
532 | GuardValue( inFunctionType );
|
---|
533 | inFunctionType = true;
|
---|
534 | printQualifiers( refType );
|
---|
535 | maybeAccept( refType->base.get(), *visitor );
|
---|
536 | }
|
---|
537 |
|
---|
538 | inline std::vector< ast::ptr< ast::Type > > getTypes( const std::vector< ast::ptr< ast::DeclWithType > > & decls ) {
|
---|
539 | std::vector< ast::ptr< ast::Type > > ret;
|
---|
540 | std::transform( decls.begin(), decls.end(), std::back_inserter( ret ),
|
---|
541 | std::mem_fun( &ast::DeclWithType::get_type ) );
|
---|
542 | return ret;
|
---|
543 | }
|
---|
544 |
|
---|
545 | void Mangler_new::postvisit( const ast::FunctionType * functionType ) {
|
---|
546 | printQualifiers( functionType );
|
---|
547 | mangleName << Encoding::function;
|
---|
548 | // turn on inFunctionType so that printQualifiers does not print most qualifiers for function parameters,
|
---|
549 | // since qualifiers on outermost parameter type do not differentiate function types, e.g.,
|
---|
550 | // void (*)(const int) and void (*)(int) are the same type, but void (*)(const int *) and void (*)(int *) are different
|
---|
551 | GuardValue( inFunctionType );
|
---|
552 | inFunctionType = true;
|
---|
553 | std::vector< ast::ptr< ast::Type > > returnTypes = getTypes( functionType->returns );
|
---|
554 | if (returnTypes.empty()) mangleName << Encoding::void_t;
|
---|
555 | else accept_each( returnTypes, *visitor );
|
---|
556 | mangleName << "_";
|
---|
557 | std::vector< ast::ptr< ast::Type > > paramTypes = getTypes( functionType->params );
|
---|
558 | accept_each( paramTypes, *visitor );
|
---|
559 | mangleName << "_";
|
---|
560 | }
|
---|
561 |
|
---|
562 | void Mangler_new::mangleRef( const ast::ReferenceToType * refType, std::string prefix ) {
|
---|
563 | printQualifiers( refType );
|
---|
564 |
|
---|
565 | mangleName << prefix << refType->name.length() << refType->name;
|
---|
566 |
|
---|
567 | if ( mangleGenericParams ) {
|
---|
568 | if ( ! refType->params.empty() ) {
|
---|
569 | mangleName << "_";
|
---|
570 | for ( const ast::Expr * param : refType->params ) {
|
---|
571 | auto paramType = dynamic_cast< const ast::TypeExpr * >( param );
|
---|
572 | assertf(paramType, "Aggregate parameters should be type expressions: %s", toCString(param));
|
---|
573 | maybeAccept( paramType->type.get(), *visitor );
|
---|
574 | }
|
---|
575 | mangleName << "_";
|
---|
576 | }
|
---|
577 | }
|
---|
578 | }
|
---|
579 |
|
---|
580 | void Mangler_new::postvisit( const ast::StructInstType * aggregateUseType ) {
|
---|
581 | mangleRef( aggregateUseType, Encoding::struct_t );
|
---|
582 | }
|
---|
583 |
|
---|
584 | void Mangler_new::postvisit( const ast::UnionInstType * aggregateUseType ) {
|
---|
585 | mangleRef( aggregateUseType, Encoding::union_t );
|
---|
586 | }
|
---|
587 |
|
---|
588 | void Mangler_new::postvisit( const ast::EnumInstType * aggregateUseType ) {
|
---|
589 | mangleRef( aggregateUseType, Encoding::enum_t );
|
---|
590 | }
|
---|
591 |
|
---|
592 | void Mangler_new::postvisit( const ast::TypeInstType * typeInst ) {
|
---|
593 | VarMapType::iterator varNum = varNums.find( typeInst->name );
|
---|
594 | if ( varNum == varNums.end() ) {
|
---|
595 | mangleRef( typeInst, Encoding::type );
|
---|
596 | } else {
|
---|
597 | printQualifiers( typeInst );
|
---|
598 | // Note: Can't use name here, since type variable names do not actually disambiguate a function, e.g.
|
---|
599 | // forall(dtype T) void f(T);
|
---|
600 | // forall(dtype S) void f(S);
|
---|
601 | // are equivalent and should mangle the same way. This is accomplished by numbering the type variables when they
|
---|
602 | // are first found and prefixing with the appropriate encoding for the type class.
|
---|
603 | assertf( varNum->second.second < TypeDecl::NUMBER_OF_KINDS, "Unhandled type variable kind: %d", varNum->second.second );
|
---|
604 | mangleName << Encoding::typeVariables[varNum->second.second] << varNum->second.first;
|
---|
605 | } // if
|
---|
606 | }
|
---|
607 |
|
---|
608 | void Mangler_new::postvisit( const ast::TraitInstType * inst ) {
|
---|
609 | printQualifiers( inst );
|
---|
610 | mangleName << inst->name.size() << inst->name;
|
---|
611 | }
|
---|
612 |
|
---|
613 | void Mangler_new::postvisit( const ast::TupleType * tupleType ) {
|
---|
614 | printQualifiers( tupleType );
|
---|
615 | mangleName << Encoding::tuple << tupleType->types.size();
|
---|
616 | accept_each( tupleType->types, *visitor );
|
---|
617 | }
|
---|
618 |
|
---|
619 | void Mangler_new::postvisit( const ast::VarArgsType * varArgsType ) {
|
---|
620 | printQualifiers( varArgsType );
|
---|
621 | static const std::string vargs = "__builtin_va_list";
|
---|
622 | mangleName << Encoding::type << vargs.size() << vargs;
|
---|
623 | }
|
---|
624 |
|
---|
625 | void Mangler_new::postvisit( const ast::ZeroType * ) {
|
---|
626 | mangleName << Encoding::zero;
|
---|
627 | }
|
---|
628 |
|
---|
629 | void Mangler_new::postvisit( const ast::OneType * ) {
|
---|
630 | mangleName << Encoding::one;
|
---|
631 | }
|
---|
632 |
|
---|
633 | void Mangler_new::postvisit( const ast::QualifiedType * qualType ) {
|
---|
634 | bool inqual = inQualifiedType;
|
---|
635 | if (! inqual ) {
|
---|
636 | // N marks the start of a qualified type
|
---|
637 | inQualifiedType = true;
|
---|
638 | mangleName << Encoding::qualifiedTypeStart;
|
---|
639 | }
|
---|
640 | maybeAccept( qualType->parent.get(), *visitor );
|
---|
641 | maybeAccept( qualType->child.get(), *visitor );
|
---|
642 | if ( ! inqual ) {
|
---|
643 | // E marks the end of a qualified type
|
---|
644 | inQualifiedType = false;
|
---|
645 | mangleName << Encoding::qualifiedTypeEnd;
|
---|
646 | }
|
---|
647 | }
|
---|
648 |
|
---|
649 | void Mangler_new::postvisit( const ast::TypeDecl * decl ) {
|
---|
650 | // TODO: is there any case where mangling a TypeDecl makes sense? If so, this code needs to be
|
---|
651 | // fixed to ensure that two TypeDecls mangle to the same name when they are the same type and vice versa.
|
---|
652 | // Note: The current scheme may already work correctly for this case, I have not thought about this deeply
|
---|
653 | // and the case has not yet come up in practice. Alternatively, if not then this code can be removed
|
---|
654 | // aside from the assert false.
|
---|
655 | assertf(false, "Mangler_new should not visit typedecl: %s", toCString(decl));
|
---|
656 | assertf( decl->kind < ast::TypeDecl::Kind::NUMBER_OF_KINDS, "Unhandled type variable kind: %d", decl->kind );
|
---|
657 | mangleName << Encoding::typeVariables[ decl->kind ] << ( decl->name.length() ) << decl->name;
|
---|
658 | }
|
---|
659 |
|
---|
660 | __attribute__((unused)) void printVarMap( const std::map< std::string, std::pair< int, int > > &varMap, std::ostream &os ) {
|
---|
661 | for ( std::map< std::string, std::pair< int, int > >::const_iterator i = varMap.begin(); i != varMap.end(); ++i ) {
|
---|
662 | os << i->first << "(" << i->second.first << "/" << i->second.second << ")" << std::endl;
|
---|
663 | } // for
|
---|
664 | }
|
---|
665 |
|
---|
666 | void Mangler_new::printQualifiers( const ast::Type * type ) {
|
---|
667 | // skip if not including qualifiers
|
---|
668 | if ( typeMode ) return;
|
---|
669 | if ( auto ptype = dynamic_cast< const ast::ParameterizedType * >(type) ) {
|
---|
670 | if ( ! ptype->forall.empty() ) {
|
---|
671 | std::list< std::string > assertionNames;
|
---|
672 | int dcount = 0, fcount = 0, vcount = 0, acount = 0;
|
---|
673 | mangleName << Encoding::forall;
|
---|
674 | for ( const ast::TypeDecl * decl : ptype->forall ) {
|
---|
675 | switch ( decl->kind ) {
|
---|
676 | case ast::TypeDecl::Kind::Dtype:
|
---|
677 | dcount++;
|
---|
678 | break;
|
---|
679 | case ast::TypeDecl::Kind::Ftype:
|
---|
680 | fcount++;
|
---|
681 | break;
|
---|
682 | case ast::TypeDecl::Kind::Ttype:
|
---|
683 | vcount++;
|
---|
684 | break;
|
---|
685 | default:
|
---|
686 | assert( false );
|
---|
687 | } // switch
|
---|
688 | varNums[ decl->name ] = std::make_pair( nextVarNum, (int)decl->kind );
|
---|
689 | for ( const ast::DeclWithType * assert : decl->assertions ) {
|
---|
690 | ast::Pass<Mangler_new> sub_mangler(
|
---|
691 | mangleOverridable, typeMode, mangleGenericParams, nextVarNum, varNums );
|
---|
692 | assert->accept( sub_mangler );
|
---|
693 | assertionNames.push_back( sub_mangler.core.get_mangleName() );
|
---|
694 | acount++;
|
---|
695 | } // for
|
---|
696 | } // for
|
---|
697 | mangleName << dcount << "_" << fcount << "_" << vcount << "_" << acount << "_";
|
---|
698 | std::copy( assertionNames.begin(), assertionNames.end(), std::ostream_iterator< std::string >( mangleName, "" ) );
|
---|
699 | mangleName << "_";
|
---|
700 | } // if
|
---|
701 | } // if
|
---|
702 | if ( ! inFunctionType ) {
|
---|
703 | // these qualifiers do not distinguish the outermost type of a function parameter
|
---|
704 | if ( type->is_const() ) {
|
---|
705 | mangleName << Encoding::qualifiers.at(Type::Const);
|
---|
706 | } // if
|
---|
707 | if ( type->is_volatile() ) {
|
---|
708 | mangleName << Encoding::qualifiers.at(Type::Volatile);
|
---|
709 | } // if
|
---|
710 | // Removed due to restrict not affecting function compatibility in GCC
|
---|
711 | // if ( type->get_isRestrict() ) {
|
---|
712 | // mangleName << "E";
|
---|
713 | // } // if
|
---|
714 | if ( type->is_atomic() ) {
|
---|
715 | mangleName << Encoding::qualifiers.at(Type::Atomic);
|
---|
716 | } // if
|
---|
717 | }
|
---|
718 | if ( type->is_mutex() ) {
|
---|
719 | mangleName << Encoding::qualifiers.at(Type::Mutex);
|
---|
720 | } // if
|
---|
721 | if ( inFunctionType ) {
|
---|
722 | // turn off inFunctionType so that types can be differentiated for nested qualifiers
|
---|
723 | GuardValue( inFunctionType );
|
---|
724 | inFunctionType = false;
|
---|
725 | }
|
---|
726 | }
|
---|
727 | } // namespace
|
---|
728 | } // namespace Mangle
|
---|
729 |
|
---|
730 | // Local Variables: //
|
---|
731 | // tab-width: 4 //
|
---|
732 | // mode: c++ //
|
---|
733 | // compile-command: "make install" //
|
---|
734 | // End: //
|
---|