Skip to content

Commit e5a22ef

Browse files
kornilova203kornilova203
committed
Generate types for anonymous declarations
1 parent 78584f0 commit e5a22ef

File tree

12 files changed

+235
-111
lines changed

12 files changed

+235
-111
lines changed

‎bindgen/TypeTranslator.cpp

Lines changed: 97 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
#include "TypeTranslator.h"
22
#include "Utils.h"
33
#include "ir/types/FunctionPointerType.h"
4-
#include "ir/types/PointerType.h"
4+
#include "clang/AST/RecordLayout.h"
55

66
TypeTranslator::TypeTranslator(clang::ASTContext *ctx_, IR &ir)
77
: ctx(ctx_), ir(ir), typeMap() {
@@ -106,13 +106,19 @@ TypeTranslator::translateStructOrUnionOrEnum(const clang::QualType &qtpe) {
106106
std::shared_ptr<Type>
107107
TypeTranslator::translateStructOrUnion(const clang::QualType &qtpe) {
108108
if (qtpe->hasUnnamedOrLocalType()) {
109-
// TODO: Verify that the local part is not a problem
110-
uint64_t sizeInBits = ctx->getTypeSize(qtpe);
111-
assert(sizeInBits % 8 == 0);
112-
return std::make_shared<ArrayType>(
113-
std::make_shared<PrimitiveType>("Byte"), sizeInBits / 8);
109+
if (qtpe->isStructureType()) {
110+
std::string name =
111+
"anonymous_" + std::to_string(anonymousStructId++);
112+
clang::RecordDecl *record = qtpe->getAsStructureType()->getDecl();
113+
return addStructDefinition(record, name);
114+
} else if (qtpe->isUnionType()) {
115+
std::string name =
116+
"anonymous_" + std::to_string(anonymousUnionId++);
117+
clang::RecordDecl *record = qtpe->getAsUnionType()->getDecl();
118+
return addUnionDefinition(record, name);
119+
}
120+
return nullptr;
114121
}
115-
116122
return translateStructOrUnionOrEnum(qtpe);
117123
}
118124

@@ -144,14 +150,11 @@ std::shared_ptr<Type> TypeTranslator::translate(const clang::QualType &qtpe) {
144150
return translatePointer(
145151
tpe->getAs<clang::PointerType>()->getPointeeType());
146152

147-
} else if (qtpe->isStructureType()) {
148-
return translateStructOrUnion(qtpe);
149-
150-
} else if (qtpe->isUnionType()) {
153+
} else if (qtpe->isStructureType() || qtpe->isUnionType()) {
151154
return translateStructOrUnion(qtpe);
152155

153156
} else if (qtpe->isEnumeralType()) {
154-
return translateStructOrUnionOrEnum(qtpe);
157+
return translateEnum(qtpe);
155158

156159
} else if (qtpe->isConstantArrayType()) {
157160
return translateConstantArray(ctx->getAsConstantArrayType(qtpe));
@@ -176,3 +179,85 @@ std::string TypeTranslator::getTypeFromTypeMap(std::string cType) {
176179
}
177180
return "";
178181
}
182+
183+
std::shared_ptr<Location> TypeTranslator::getLocation(clang::Decl *decl) {
184+
clang::SourceManager &sm = ctx->getSourceManager();
185+
std::string filename = std::string(sm.getFilename(decl->getLocation()));
186+
char *p = realpath(filename.c_str(), nullptr);
187+
std::string path;
188+
if (p) {
189+
path = p;
190+
delete[] p;
191+
} else {
192+
// TODO: check when it happens
193+
path = "";
194+
}
195+
196+
unsigned lineNumber = sm.getSpellingLineNumber(decl->getLocation());
197+
return std::make_shared<Location>(path, lineNumber);
198+
}
199+
200+
std::shared_ptr<TypeDef>
201+
TypeTranslator::addUnionDefinition(clang::RecordDecl *record,
202+
std::string name) {
203+
std::vector<std::shared_ptr<Field>> fields;
204+
205+
for (const clang::FieldDecl *field : record->fields()) {
206+
std::string fname = field->getNameAsString();
207+
std::shared_ptr<Type> ftype = translate(field->getType());
208+
209+
fields.push_back(std::make_shared<Field>(fname, ftype));
210+
}
211+
212+
uint64_t sizeInBits = ctx->getTypeSize(record->getTypeForDecl());
213+
assert(sizeInBits % 8 == 0);
214+
215+
return ir.addUnion(name, std::move(fields), sizeInBits / 8,
216+
getLocation(record));
217+
}
218+
219+
std::shared_ptr<TypeDef>
220+
TypeTranslator::addStructDefinition(clang::RecordDecl *record,
221+
std::string name) {
222+
std::string newName = "struct_" + name;
223+
224+
if (record->hasAttr<clang::PackedAttr>()) {
225+
llvm::errs() << "Warning: struct " << name << " is packed. "
226+
<< "Packed structs are not supported by Scala Native. "
227+
<< "Access to fields will not work correctly.\n";
228+
llvm::errs().flush();
229+
}
230+
231+
std::vector<std::shared_ptr<Field>> fields;
232+
const clang::ASTRecordLayout &recordLayout =
233+
ctx->getASTRecordLayout(record);
234+
235+
bool isBitFieldStruct = false;
236+
for (const clang::FieldDecl *field : record->fields()) {
237+
if (field->isBitField()) {
238+
isBitFieldStruct = true;
239+
}
240+
std::shared_ptr<Type> ftype = translate(field->getType());
241+
uint64_t recordOffsetInBits =
242+
recordLayout.getFieldOffset(field->getFieldIndex());
243+
fields.push_back(std::make_shared<Field>(field->getNameAsString(),
244+
ftype, recordOffsetInBits));
245+
}
246+
247+
uint64_t sizeInBits = ctx->getTypeSize(record->getTypeForDecl());
248+
assert(sizeInBits % 8 == 0);
249+
250+
return ir.addStruct(name, std::move(fields), sizeInBits / 8,
251+
getLocation(record),
252+
record->hasAttr<clang::PackedAttr>(), isBitFieldStruct);
253+
}
254+
255+
std::shared_ptr<Type>
256+
TypeTranslator::translateEnum(const clang::QualType &type) {
257+
if (type->hasUnnamedOrLocalType()) {
258+
clang::EnumDecl *enumDecl = type->getAs<clang::EnumType>()->getDecl();
259+
return std::make_shared<PrimitiveType>(getTypeFromTypeMap(
260+
enumDecl->getIntegerType().getUnqualifiedType().getAsString()));
261+
}
262+
return translateStructOrUnionOrEnum(type);
263+
}

‎bindgen/TypeTranslator.h

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,9 +16,19 @@ class TypeTranslator {
1616

1717
std::string getTypeFromTypeMap(std::string cType);
1818

19+
std::shared_ptr<TypeDef> addUnionDefinition(clang::RecordDecl *record,
20+
std::string name);
21+
22+
std::shared_ptr<TypeDef> addStructDefinition(clang::RecordDecl *record,
23+
std::string name);
24+
25+
std::shared_ptr<Location> getLocation(clang::Decl *decl);
26+
1927
private:
2028
clang::ASTContext *ctx;
2129
IR &ir;
30+
int anonymousStructId = 0;
31+
int anonymousUnionId = 0;
2232

2333
/**
2434
* Primitive types
@@ -36,4 +46,6 @@ class TypeTranslator {
3646

3747
std::shared_ptr<Type>
3848
translateConstantArray(const clang::ConstantArrayType *ar);
49+
50+
std::shared_ptr<Type> translateEnum(const clang::QualType &type);
3951
};

‎bindgen/ir/IR.cpp

Lines changed: 11 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -33,9 +33,10 @@ void IR::addEnum(std::string name, const std::string &type,
3333
}
3434
}
3535

36-
void IR::addStruct(std::string name, std::vector<std::shared_ptr<Field>> fields,
37-
uint64_t typeSize, std::shared_ptr<Location> location,
38-
bool isPacked, bool isBitField) {
36+
std::shared_ptr<TypeDef>
37+
IR::addStruct(std::string name, std::vector<std::shared_ptr<Field>> fields,
38+
uint64_t typeSize, std::shared_ptr<Location> location,
39+
bool isPacked, bool isBitField) {
3940
std::shared_ptr<Struct> s =
4041
std::make_shared<Struct>(name, std::move(fields), typeSize,
4142
std::move(location), isPacked, isBitField);
@@ -44,22 +45,27 @@ void IR::addStruct(std::string name, std::vector<std::shared_ptr<Field>> fields,
4445
if (typeDef) {
4546
/* the struct type used to be opaque type, typeDef contains nullptr */
4647
typeDef.get()->setType(s);
48+
return typeDef;
4749
} else {
4850
typeDefs.push_back(s->generateTypeDef());
51+
return typeDefs.back();
4952
}
5053
}
5154

52-
void IR::addUnion(std::string name, std::vector<std::shared_ptr<Field>> fields,
53-
uint64_t maxSize, std::shared_ptr<Location> location) {
55+
std::shared_ptr<TypeDef>
56+
IR::addUnion(std::string name, std::vector<std::shared_ptr<Field>> fields,
57+
uint64_t maxSize, std::shared_ptr<Location> location) {
5458
std::shared_ptr<Union> u = std::make_shared<Union>(
5559
name, std::move(fields), maxSize, std::move(location));
5660
unions.push_back(u);
5761
std::shared_ptr<TypeDef> typeDef = getTypeDefWithName("union_" + name);
5862
if (typeDef) {
5963
/* the union type used to be opaque type, typeDef contains nullptr */
6064
typeDef.get()->setType(u);
65+
return typeDef;
6166
} else {
6267
typeDefs.push_back(u->generateTypeDef());
68+
return typeDefs.back();
6369
}
6470
}
6571

‎bindgen/ir/IR.h

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -35,12 +35,14 @@ class IR {
3535
std::vector<Enumerator> enumerators,
3636
std::shared_ptr<Location> location);
3737

38-
void addStruct(std::string name, std::vector<std::shared_ptr<Field>> fields,
39-
uint64_t typeSize, std::shared_ptr<Location> location,
40-
bool isPacked, bool isBitField);
41-
42-
void addUnion(std::string name, std::vector<std::shared_ptr<Field>> fields,
43-
uint64_t maxSize, std::shared_ptr<Location> location);
38+
std::shared_ptr<TypeDef>
39+
addStruct(std::string name, std::vector<std::shared_ptr<Field>> fields,
40+
uint64_t typeSize, std::shared_ptr<Location> location,
41+
bool isPacked, bool isBitField);
42+
43+
std::shared_ptr<TypeDef>
44+
addUnion(std::string name, std::vector<std::shared_ptr<Field>> fields,
45+
uint64_t maxSize, std::shared_ptr<Location> location);
4446

4547
void addLiteralDefine(std::string name, std::string literal,
4648
std::shared_ptr<Type> type);

‎bindgen/visitor/TreeVisitor.cpp

Lines changed: 5 additions & 73 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,4 @@
11
#include "TreeVisitor.h"
2-
#include "clang/AST/RecordLayout.h"
32

43
bool TreeVisitor::VisitFunctionDecl(clang::FunctionDecl *func) {
54
if (!astContext->getSourceManager().isInMainFile(func->getLocation())) {
@@ -37,7 +36,7 @@ bool TreeVisitor::VisitTypedefDecl(clang::TypedefDecl *tpdef) {
3736
std::shared_ptr<Type> type =
3837
typeTranslator.translate(tpdef->getUnderlyingType());
3938
if (type) {
40-
ir.addTypeDef(name, type, getLocation(tpdef));
39+
ir.addTypeDef(name, type, typeTranslator.getLocation(tpdef));
4140
}
4241
return true;
4342
}
@@ -59,7 +58,8 @@ bool TreeVisitor::VisitEnumDecl(clang::EnumDecl *enumdecl) {
5958
std::string scalaType = typeTranslator.getTypeFromTypeMap(
6059
enumdecl->getIntegerType().getUnqualifiedType().getAsString());
6160

62-
ir.addEnum(name, scalaType, std::move(enumerators), getLocation(enumdecl));
61+
ir.addEnum(name, scalaType, std::move(enumerators),
62+
typeTranslator.getLocation(enumdecl));
6363

6464
return true;
6565
}
@@ -76,68 +76,17 @@ bool TreeVisitor::VisitRecordDecl(clang::RecordDecl *record) {
7676

7777
if (record->isUnion() && record->isThisDeclarationADefinition() &&
7878
!record->isAnonymousStructOrUnion() && !name.empty()) {
79-
handleUnion(record, name);
79+
typeTranslator.addUnionDefinition(record, name);
8080
return true;
8181

8282
} else if (record->isStruct() && record->isThisDeclarationADefinition() &&
8383
!record->isAnonymousStructOrUnion() && !name.empty()) {
84-
handleStruct(record, name);
84+
typeTranslator.addStructDefinition(record, name);
8585
return true;
8686
}
8787
return false;
8888
}
8989

90-
void TreeVisitor::handleUnion(clang::RecordDecl *record, std::string name) {
91-
std::vector<std::shared_ptr<Field>> fields;
92-
93-
for (const clang::FieldDecl *field : record->fields()) {
94-
std::string fname = field->getNameAsString();
95-
std::shared_ptr<Type> ftype =
96-
typeTranslator.translate(field->getType());
97-
98-
fields.push_back(std::make_shared<Field>(fname, ftype));
99-
}
100-
101-
uint64_t sizeInBits = astContext->getTypeSize(record->getTypeForDecl());
102-
assert(sizeInBits % 8 == 0);
103-
104-
ir.addUnion(name, std::move(fields), sizeInBits / 8, getLocation(record));
105-
}
106-
107-
void TreeVisitor::handleStruct(clang::RecordDecl *record, std::string name) {
108-
std::string newName = "struct_" + name;
109-
110-
if (record->hasAttr<clang::PackedAttr>()) {
111-
llvm::errs() << "Warning: struct " << name << " is packed. "
112-
<< "Packed structs are not supported by Scala Native. "
113-
<< "Access to fields will not work correctly.\n";
114-
llvm::errs().flush();
115-
}
116-
117-
std::vector<std::shared_ptr<Field>> fields;
118-
const clang::ASTRecordLayout &recordLayout =
119-
astContext->getASTRecordLayout(record);
120-
121-
bool isBitFieldStruct = false;
122-
for (const clang::FieldDecl *field : record->fields()) {
123-
if (field->isBitField()) {
124-
isBitFieldStruct = true;
125-
}
126-
std::shared_ptr<Type> ftype =
127-
typeTranslator.translate(field->getType());
128-
uint64_t recordOffsetInBits =
129-
recordLayout.getFieldOffset(field->getFieldIndex());
130-
fields.push_back(std::make_shared<Field>(field->getNameAsString(),
131-
ftype, recordOffsetInBits));
132-
}
133-
134-
uint64_t sizeInBits = astContext->getTypeSize(record->getTypeForDecl());
135-
assert(sizeInBits % 8 == 0);
136-
137-
ir.addStruct(name, std::move(fields), sizeInBits / 8, getLocation(record),
138-
record->hasAttr<clang::PackedAttr>(), isBitFieldStruct);
139-
}
140-
14190
bool TreeVisitor::VisitVarDecl(clang::VarDecl *varDecl) {
14291
if (!astContext->getSourceManager().isInMainFile(varDecl->getLocation())) {
14392
/* include variables only from the original header */
@@ -157,20 +106,3 @@ bool TreeVisitor::VisitVarDecl(clang::VarDecl *varDecl) {
157106
}
158107
return true;
159108
}
160-
161-
std::shared_ptr<Location> TreeVisitor::getLocation(clang::Decl *decl) {
162-
clang::SourceManager &sm = astContext->getSourceManager();
163-
std::string filename = std::string(sm.getFilename(decl->getLocation()));
164-
char *p = realpath(filename.c_str(), nullptr);
165-
std::string path;
166-
if (p) {
167-
path = p;
168-
delete[] p;
169-
} else {
170-
// TODO: check when it happens
171-
path = "";
172-
}
173-
174-
unsigned lineNumber = sm.getSpellingLineNumber(decl->getLocation());
175-
return std::make_shared<Location>(path, lineNumber);
176-
}

‎bindgen/visitor/TreeVisitor.h

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -11,12 +11,6 @@ class TreeVisitor : public clang::RecursiveASTVisitor<TreeVisitor> {
1111
TypeTranslator typeTranslator;
1212
IR &ir;
1313

14-
void handleUnion(clang::RecordDecl *record, std::string name);
15-
16-
void handleStruct(clang::RecordDecl *record, std::string name);
17-
18-
std::shared_ptr<Location> getLocation(clang::Decl *decl);
19-
2014
public:
2115
TreeVisitor(clang::CompilerInstance *CI, IR &ir)
2216
: astContext(&(CI->getASTContext())), typeTranslator(astContext, ir),

‎tests/samples/AnonymousTypes.h

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,17 @@
1+
/* structs */
2+
3+
void foo(struct { char a; } * s);
4+
5+
struct StructWithAnonymousStruct {
6+
struct {
7+
union {
8+
long a;
9+
} * innerUnion;
10+
} * innerStruct;
11+
12+
enum { A, B, C } innerEnum;
13+
};
14+
15+
struct {
16+
int result;
17+
} * bar();

0 commit comments

Comments
 (0)