1 /**
2 * Copyright (c) 2021 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16 #include "variable.h"
17
18 #include <binder/scope.h>
19
20 #include <utility>
21
22 namespace panda::es2panda::binder {
23
LocalVariable(Decl * decl,VariableFlags flags)24 LocalVariable::LocalVariable(Decl *decl, VariableFlags flags) : Variable(decl, flags)
25 {
26 if (decl->IsConstDecl()) {
27 flags_ |= VariableFlags::READONLY;
28 }
29 }
30
Name() const31 const util::StringView &Variable::Name() const
32 {
33 return decl_->Name();
34 }
35
Copy(ArenaAllocator * allocator,Decl * decl) const36 LocalVariable *LocalVariable::Copy(ArenaAllocator *allocator, Decl *decl) const
37 {
38 auto *var = allocator->New<LocalVariable>(decl, flags_);
39 CHECK_NOT_NULL(var);
40 var->vreg_ = vreg_;
41 return var;
42 }
43
SetLexical(Scope * scope,util::PatchFix * patchFixHelper)44 void LocalVariable::SetLexical(Scope *scope, util::PatchFix *patchFixHelper)
45 {
46 if (LexicalBound()) {
47 return;
48 }
49
50 VariableScope *varScope = scope->IsFunctionParamScope() ?
51 scope->AsFunctionParamScope()->GetFunctionScope() : scope->EnclosingVariableScope();
52 CHECK_NOT_NULL(varScope);
53 uint32_t slot = 0;
54 auto name = Declaration()->Name();
55
56 if (patchFixHelper && patchFixHelper->IsScopeValidToPatchLexical(varScope)) {
57 // get slot from symbol table for lexical variable, if not found, slot is set to UINT32_MAX
58 slot = patchFixHelper->GetSlotIdFromSymbolTable(std::string(name));
59 // Store the additional lexical variable into PatchEnv
60 if (patchFixHelper->IsAdditionalVarInPatch(slot)) {
61 patchFixHelper->AllocSlotfromPatchEnv(std::string(name));
62 } else {
63 // Just for restore 'newlexenv' instruction for func_main_0 in patch
64 varScope->RestoreFuncMain0LexEnv(patchFixHelper->GetEnvSizeOfFuncMain0());
65 }
66 } else {
67 if (decl_ && decl_->NeedSetInSendableEnv(varScope)) {
68 AddFlag(VariableFlags::IN_SENDABLE_ENV);
69 slot = varScope->NextSendableSlot();
70 } else {
71 slot = varScope->NextSlot();
72 }
73 }
74
75 BindLexEnvSlot(slot);
76 // gather lexical variables for debuginfo
77 varScope->AddLexicalVarNameAndType(slot, name,
78 static_cast<typename std::underlying_type<binder::DeclType>::type>(Declaration()->Type()));
79 }
80
SetLexical(Scope * scope,util::PatchFix * patchFixHelper)81 void GlobalVariable::SetLexical([[maybe_unused]] Scope *scope, [[maybe_unused]] util::PatchFix *patchFixHelper) {}
SetLexical(Scope * scope,util::PatchFix * patchFixHelper)82 void ModuleVariable::SetLexical([[maybe_unused]] Scope *scope, [[maybe_unused]] util::PatchFix *patchFixHelper) {}
SetLexical(Scope * scope,util::PatchFix * patchFixHelper)83 void EnumVariable::SetLexical([[maybe_unused]] Scope *scope, [[maybe_unused]] util::PatchFix *patchFixHelper) {}
SetLexical(Scope * scope,util::PatchFix * patchFixHelper)84 void NamespaceVariable::SetLexical([[maybe_unused]] Scope *scope, [[maybe_unused]] util::PatchFix *patchFixHelper) {}
SetLexical(Scope * scope,util::PatchFix * patchFixHelper)85 void ImportEqualsVariable::SetLexical([[maybe_unused]] Scope *scope, [[maybe_unused]] util::PatchFix *patchFixHelper) {}
SetLexical(Scope * scope,util::PatchFix * patchFixHelper)86 void EnumLiteralVariable::SetLexical([[maybe_unused]] Scope *scope, [[maybe_unused]] util::PatchFix *patchFixHelper) {}
87
ResetDecl(Decl * decl)88 void EnumVariable::ResetDecl(Decl *decl)
89 {
90 decl_ = decl;
91 }
92
93 } // namespace panda::es2panda::binder
94