Skip to content

Commit 6313291

Browse files
LFsWangLFsWang
committed
[Sema] NameAndTypeResolver
1 parent fcbb73d commit 6313291

File tree

13 files changed

+755
-4
lines changed

13 files changed

+755
-4
lines changed

‎include/soll/AST/Decl.h‎

Lines changed: 25 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@ class Decl : public ASTNode {
2727
std::string Name;
2828
Visibility Vis;
2929
std::string UniqueName;
30+
const ASTNode *Parent;
3031

3132
protected:
3233
friend class ASTReader;
@@ -35,10 +36,11 @@ class Decl : public ASTNode {
3536
Decl(SourceRange L,
3637
llvm::StringRef Name = llvm::StringRef::withNullAsEmpty(nullptr),
3738
Visibility Vis = Visibility::Default)
38-
: Location(L), Name(Name.str()), Vis(Vis), UniqueName(Name.str()) {}
39+
: Location(L), Name(Name.str()), Vis(Vis), UniqueName(Name.str()),
40+
Parent(nullptr) {}
3941

4042
Decl(SourceRange L, std::string Name, Visibility Vis = Visibility::Default)
41-
: Location(L), Name(Name), Vis(Vis), UniqueName(Name) {}
43+
: Location(L), Name(Name), Vis(Vis), UniqueName(Name), Parent(nullptr) {}
4244

4345
public:
4446
virtual void accept(DeclVisitor &visitor) = 0;
@@ -48,6 +50,15 @@ class Decl : public ASTNode {
4850
llvm::StringRef getUniqueName() const { return UniqueName; }
4951
void setUniqueName(llvm::StringRef NewName) { UniqueName = NewName.str(); }
5052
Visibility getVisibility() const { return Vis; }
53+
54+
void setScope(const ASTNode *D) { Parent = D; }
55+
/// @returns the scope this declaration resides in. Can be nullptr if it is
56+
/// the global scope. Available only after name and type resolution step.
57+
const ASTNode *scope() const { return Parent; }
58+
59+
bool isStructMember() const ;
60+
61+
bool isVisibleAsUnqualifiedName() const ;
5162
};
5263

5364
/**
@@ -414,6 +425,18 @@ class StructDecl : public Decl {
414425
Token getToken() const { return Tok; }
415426
TypePtr getType() const { return Ty; }
416427
TypePtr getConstructorType() const { return ConstructorTy; }
428+
std::vector<Decl *> getMembers() {
429+
std::vector<Decl *> Res;
430+
for (auto &M : Members)
431+
Res.push_back(M.get());
432+
return Res;
433+
}
434+
std::vector<const Decl *> getMembers() const {
435+
std::vector<const Decl *> Res;
436+
for (auto &M : Members)
437+
Res.push_back(M.get());
438+
return Res;
439+
}
417440
};
418441

419442
class ModifierInvocation {

‎include/soll/Sema/DeclarationContainer.h‎

Lines changed: 127 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,127 @@
1+
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
2+
#pragma once
3+
4+
#include "soll/AST/AST.h"
5+
#include "soll/AST/Decl.h"
6+
#include "soll/Basic/DiagnosticSema.h"
7+
#include "soll/Sema/Sema.h"
8+
9+
#include <iostream>
10+
#include <map>
11+
#include <memory>
12+
#include <string>
13+
#include <vector>
14+
using std::cerr;
15+
using std::endl;
16+
17+
namespace soll {
18+
19+
class DeclarationContainer {
20+
public:
21+
DeclarationContainer() = default;
22+
explicit DeclarationContainer(const ASTNode *EnclosingNode,
23+
DeclarationContainer *EnclosingContainer)
24+
: EnclosingNode(EnclosingNode), EnclosingContainer(EnclosingContainer) {
25+
if (EnclosingContainer)
26+
EnclosingContainer->InnerContainers.emplace_back(this);
27+
}
28+
29+
ASTNode const *enclosingNode() const { return EnclosingNode; }
30+
31+
void activateVariable(llvm::StringRef Name) {
32+
assert(InvisibleDeclarations.count(Name) && InvisibleDeclarations.at(Name).size() == 1);
33+
assert(Declarations.count(Name) == 0 || Declarations.at(Name).empty());
34+
35+
Declarations[Name].emplace_back(InvisibleDeclarations.at(Name).front());
36+
InvisibleDeclarations.erase(Name);
37+
}
38+
39+
bool isInvisible(llvm::StringRef Name) const {
40+
return InvisibleDeclarations.count(Name);
41+
}
42+
43+
bool registerDeclaration(const Decl *D, bool Invisible, bool Update) {
44+
return registerDeclaration(D, llvm::StringRef(), nullptr, Invisible,
45+
Update);
46+
}
47+
48+
bool registerDeclaration(const Decl *D, llvm::StringRef Name,
49+
const SourceRange *Loc, bool Invisible, bool Update) {
50+
if (Name.empty()) {
51+
Name = D->getName();
52+
}
53+
54+
if (Name.empty())
55+
return true;
56+
57+
if (Update) {
58+
assert(!dynamic_cast<const FunctionDecl *>(D) &&
59+
"Attempt to update function definition.");
60+
Declarations.erase(Name);
61+
InvisibleDeclarations.erase(Name);
62+
return true;
63+
} else {
64+
if (conflictingDeclaration(D, Name))
65+
return false;
66+
if (EnclosingContainer && D->isVisibleAsUnqualifiedName()) {
67+
// assert(false && "TODO: isVisibleAsUnqualifiedName for struct");
68+
cerr << "TODO: isVisibleAsUnqualifiedName for struct" << endl;
69+
// deal with homonymCandidates
70+
}
71+
}
72+
73+
std::vector<const Decl *> Decls =
74+
Invisible ? InvisibleDeclarations[Name] : Declarations[Name];
75+
if (find(Decls.begin(), Decls.end(), D) != Decls.end())
76+
Decls.emplace_back(D);
77+
78+
return true;
79+
}
80+
81+
const Decl *conflictingDeclaration(const Decl *D,
82+
llvm::StringRef Name) const {
83+
if (Name.empty())
84+
Name = D->getName();
85+
assert(!Name.empty());
86+
87+
std::vector<Decl const *> Decls;
88+
if (Declarations.count(Name))
89+
Decls += Declarations.at(Name);
90+
91+
if (InvisibleDeclarations.count(Name))
92+
Decls += InvisibleDeclarations.at(Name);
93+
94+
if (dynamic_cast<const FunctionDecl *>(D) ||
95+
dynamic_cast<const EventDecl *>(D) ||
96+
dynamic_cast<const MagicVariableDecl *>(D)) {
97+
for (const Decl *RegDecl : Decls) {
98+
if (dynamic_cast<const FunctionDecl *>(D) &&
99+
!dynamic_cast<const FunctionDecl *>(RegDecl))
100+
return RegDecl;
101+
if (dynamic_cast<const EventDecl *>(D) &&
102+
!dynamic_cast<const EventDecl *>(RegDecl))
103+
return RegDecl;
104+
if (dynamic_cast<const MagicVariableDecl *>(D) &&
105+
!dynamic_cast<const MagicVariableDecl *>(RegDecl))
106+
return RegDecl;
107+
}
108+
} else if (Decls.size() == 1 && Decls.front() == D) {
109+
return nullptr;
110+
} else if (!Decls.empty()) {
111+
return Decls.front();
112+
}
113+
114+
return nullptr;
115+
}
116+
117+
private:
118+
const ASTNode *EnclosingNode = nullptr;
119+
const DeclarationContainer *EnclosingContainer = nullptr;
120+
std::vector<const DeclarationContainer *> InnerContainers;
121+
std::map<llvm::StringRef, std::vector<const Decl *>> Declarations;
122+
std::map<llvm::StringRef, std::vector<const Decl *>> InvisibleDeclarations;
123+
// std::vector<std::tuple<std::string, langutil::SourceLocation const*>>
124+
// m_homonymCandidates;
125+
};
126+
127+
} // namespace soll

‎include/soll/Sema/GlobalContent.h‎

Lines changed: 98 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,98 @@
1+
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
2+
#pragma once
3+
4+
#include "soll/AST/AST.h"
5+
#include "soll/AST/Decl.h"
6+
#include "soll/Basic/DiagnosticSema.h"
7+
#include "soll/Sema/Sema.h"
8+
9+
#include <iostream>
10+
#include <map>
11+
#include <memory>
12+
#include <string>
13+
#include <vector>
14+
15+
using std::cerr;
16+
using std::endl;
17+
18+
namespace soll {
19+
/// Magic variables get negative ids for easy differentiation
20+
namespace {
21+
int magicVariableToID(std::string const &_name) {
22+
if (_name == "abi")
23+
return -1;
24+
else if (_name == "addmod")
25+
return -2;
26+
else if (_name == "assert")
27+
return -3;
28+
else if (_name == "block")
29+
return -4;
30+
else if (_name == "blockhash")
31+
return -5;
32+
else if (_name == "ecrecover")
33+
return -6;
34+
else if (_name == "gasleft")
35+
return -7;
36+
else if (_name == "keccak256")
37+
return -8;
38+
else if (_name == "msg")
39+
return -15;
40+
else if (_name == "mulmod")
41+
return -16;
42+
else if (_name == "now")
43+
return -17;
44+
else if (_name == "require")
45+
return -18;
46+
else if (_name == "revert")
47+
return -19;
48+
else if (_name == "ripemd160")
49+
return -20;
50+
else if (_name == "selfdestruct")
51+
return -21;
52+
else if (_name == "sha256")
53+
return -22;
54+
else if (_name == "sha3")
55+
return -23;
56+
else if (_name == "suicide")
57+
return -24;
58+
else if (_name == "super")
59+
return -25;
60+
else if (_name == "tx")
61+
return -26;
62+
else if (_name == "type")
63+
return -27;
64+
else if (_name == "this")
65+
return -28;
66+
else
67+
__builtin_unreachable();
68+
}
69+
} // namespace
70+
71+
class GlobalContext {
72+
public:
73+
void setCurrentContract(const ContractDecl *Cont) { CurrentContract = Cont; }
74+
void resetCurrentContract() { CurrentContract = nullptr; }
75+
const MagicVariableDecl *currentThis() {
76+
if (!ThisCache[CurrentContract]) {
77+
TypePtr Type = CurrentContract->getType();
78+
ThisCache[CurrentContract] = std::make_shared<MagicVariableDecl>(
79+
magicVariableToID("this"), "this", Type);
80+
}
81+
return ThisCache[CurrentContract].get();
82+
}
83+
84+
const MagicVariableDecl *currentSuper() {
85+
assert(false && "unimp currentSuper");
86+
return nullptr;
87+
}
88+
89+
GlobalContext() = default;
90+
GlobalContext(const GlobalContext &) = delete;
91+
~GlobalContext() = default;
92+
93+
private:
94+
const ContractDecl *CurrentContract = nullptr;
95+
std::map<const ContractDecl *, std::shared_ptr<MagicVariableDecl>> ThisCache;
96+
std::map<const ContractDecl *, std::shared_ptr<MagicVariableDecl>> SuperCache;
97+
};
98+
} // namespace soll

‎include/soll/Sema/NameAndTypeResolver.h‎

Lines changed: 40 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,40 @@
1+
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
2+
#pragma once
3+
4+
#include "soll/AST/AST.h"
5+
#include "soll/AST/Decl.h"
6+
#include "soll/Basic/DiagnosticSema.h"
7+
#include "soll/Sema/DeclarationContainer.h"
8+
#include "soll/Sema/Sema.h"
9+
10+
namespace soll {
11+
class NameAndTypeResolver {
12+
public:
13+
void setScope(const ASTNode *Node) { CurrentScope = Scopes[Node].get(); }
14+
bool resolveNamesAndTypesInternal(ASTNode *Node, bool ResolveCode);
15+
bool updateDeclaration(const Decl *D) {
16+
Scopes[nullptr]->registerDeclaration(D, false, true);
17+
return true;
18+
}
19+
20+
void activateVariable(llvm::StringRef Name) {
21+
cerr << "Find " << Name.str() << endl;
22+
if (CurrentScope == nullptr) {
23+
cerr << "jizz" << endl;
24+
return ;
25+
}
26+
if (CurrentScope->isInvisible(Name))
27+
CurrentScope->activateVariable(Name);
28+
}
29+
30+
NameAndTypeResolver(Sema &Action, GlobalContext &GC);
31+
void Register(SourceUnit &SU);
32+
bool Resolve(SourceUnit &SU);
33+
34+
private:
35+
Sema &Action;
36+
GlobalContext &GC;
37+
std::map<const ASTNode *, std::shared_ptr<DeclarationContainer>> Scopes;
38+
DeclarationContainer *CurrentScope = nullptr;
39+
};
40+
} // namespace soll

‎include/soll/Sema/ReferenceResolver.h‎

Lines changed: 37 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,37 @@
1+
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
2+
#pragma once
3+
4+
#include "soll/AST/AST.h"
5+
#include "soll/AST/Decl.h"
6+
#include "soll/Basic/DiagnosticSema.h"
7+
#include "soll/Sema/DeclarationContainer.h"
8+
#include "soll/Sema/NameAndTypeResolver.h"
9+
#include "soll/Sema/Sema.h"
10+
11+
namespace soll {
12+
class ReferencesResolver : public DeclVisitor, public StmtVisitor {
13+
public:
14+
ReferencesResolver(Sema &Action, NameAndTypeResolver &NTR,
15+
bool ResolveInsideCode);
16+
bool resolve(ASTNode *Node) {
17+
Status = true;
18+
if (auto D = dynamic_cast<Decl *>(Node))
19+
D->accept(*this);
20+
else if (auto S = dynamic_cast<Stmt *>(Node))
21+
S->accept(*this);
22+
return Status;
23+
}
24+
void visit(Block &B) override;
25+
void visit(ForStmt &F) override;
26+
void visit(DeclStmt &D) override;
27+
28+
void visit(FunctionDecl &FD) override;
29+
30+
private:
31+
Sema &Action;
32+
NameAndTypeResolver &Resolver;
33+
bool ResolveInsideCode;
34+
bool Status;
35+
};
36+
37+
} // namespace soll

0 commit comments

Comments
 (0)