1 /*
2 * Copyright (c) 2024 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 <iostream>
17 #include "abc_class_processor.h"
18 #include "abc_field_processor.h"
19 #include "abc_literal_array_processor.h"
20 #include "abc_method_processor.h"
21 #include "mangling.h"
22
23 namespace panda::abc2program {
24
AbcClassProcessor(panda_file::File::EntityId entity_id,Abc2ProgramEntityContainer & entity_container)25 AbcClassProcessor::AbcClassProcessor(panda_file::File::EntityId entity_id, Abc2ProgramEntityContainer &entity_container)
26 : AbcFileEntityProcessor(entity_id, entity_container), record_(pandasm::Record("", LANG_ECMA))
27 {
28 class_data_accessor_ = std::make_unique<panda_file::ClassDataAccessor>(*file_, entity_id_);
29 }
30
FillProgramData()31 void AbcClassProcessor::FillProgramData()
32 {
33 program_->lang = LANG_ECMA;
34 FillRecord();
35 FillFunctions();
36 FillLiteralArrayTable();
37 }
38
FillRecord()39 void AbcClassProcessor::FillRecord()
40 {
41 FillRecordName();
42 record_.metadata->SetAccessFlags(class_data_accessor_->GetAccessFlags());
43 ASSERT(program_->record_table.count(record_.name) == 0);
44 FillRecordData();
45 program_->record_table.emplace(record_.name, std::move(record_));
46 }
47
FillRecordName()48 void AbcClassProcessor::FillRecordName()
49 {
50 record_.name = entity_container_.GetFullRecordNameById(entity_id_);
51 }
52
FillRecordData()53 void AbcClassProcessor::FillRecordData()
54 {
55 FillRecordMetaData();
56 FillFields();
57 FillRecordSourceFile();
58 }
59
FillRecordMetaData()60 void AbcClassProcessor::FillRecordMetaData()
61 {
62 FillRecordAttributes();
63 }
64
FillRecordAttributes()65 void AbcClassProcessor::FillRecordAttributes()
66 {
67 if (file_->IsExternal(entity_id_)) {
68 record_.metadata->SetAttribute(ABC_ATTR_EXTERNAL);
69 }
70 }
71
FillRecordAnnotations()72 void AbcClassProcessor::FillRecordAnnotations()
73 {
74 log::Unimplemented(__PRETTY_FUNCTION__);
75 }
76
FillRecordSourceFile()77 void AbcClassProcessor::FillRecordSourceFile()
78 {
79 std::optional<panda_file::File::EntityId> source_file_id = class_data_accessor_->GetSourceFileId();
80 if (source_file_id.has_value()) {
81 record_.source_file = GetStringById(source_file_id.value());
82 } else {
83 record_.source_file = "";
84 }
85 }
86
FillFields()87 void AbcClassProcessor::FillFields()
88 {
89 if (file_->IsExternal(entity_id_)) {
90 return;
91 }
92 class_data_accessor_->EnumerateFields([&](panda_file::FieldDataAccessor &fda) -> void {
93 panda_file::File::EntityId field_id = fda.GetFieldId();
94 AbcFieldProcessor field_processor(field_id, entity_container_, record_);
95 field_processor.FillProgramData();
96 });
97 }
98
FillFunctions()99 void AbcClassProcessor::FillFunctions()
100 {
101 if (file_->IsExternal(entity_id_)) {
102 return;
103 }
104 class_data_accessor_->EnumerateMethods([&](panda_file::MethodDataAccessor &mda) -> void {
105 panda_file::File::EntityId method_id = mda.GetMethodId();
106 AbcMethodProcessor method_processor(method_id, entity_container_);
107 method_processor.FillProgramData();
108 });
109 }
110
FillLiteralArrayTable()111 void AbcClassProcessor::FillLiteralArrayTable()
112 {
113 literal_data_accessor_ = std::make_unique<panda_file::LiteralDataAccessor>(*file_, file_->GetLiteralArraysId());
114 FillModuleLiteralArrays();
115 FillUnnestedLiteralArrays();
116 FillNestedLiteralArrays();
117 }
118
FillModuleLiteralArrays()119 void AbcClassProcessor::FillModuleLiteralArrays()
120 {
121 for (uint32_t module_literal_array_id : entity_container_.GetMouleLiteralArrayIdSet()) {
122 FillModuleLiteralArrayById(module_literal_array_id);
123 }
124
125 for (uint32_t module_literal_array_id : entity_container_.GetModuleRequestPhaseIdSet()) {
126 FillModuleRequestPhaseById(module_literal_array_id);
127 }
128 }
129
FillUnnestedLiteralArrays()130 void AbcClassProcessor::FillUnnestedLiteralArrays()
131 {
132 for (uint32_t unnested_literal_array_id : entity_container_.GetUnnestedLiteralArrayIdSet()) {
133 FillLiteralArrayById(unnested_literal_array_id);
134 }
135 }
136
FillNestedLiteralArrays()137 void AbcClassProcessor::FillNestedLiteralArrays()
138 {
139 auto &unprocessed_literal_array_id_set = entity_container_.GetUnprocessedNestedLiteralArrayIdSet();
140 while (!unprocessed_literal_array_id_set.empty()) {
141 auto id = unprocessed_literal_array_id_set.begin();
142 entity_container_.AddProcessedNestedLiteralArrayId(*id);
143 FillLiteralArrayById(*id);
144 unprocessed_literal_array_id_set.erase(id);
145 }
146 }
147
FillModuleLiteralArrayById(uint32_t module_literal_array_id)148 void AbcClassProcessor::FillModuleLiteralArrayById(uint32_t module_literal_array_id)
149 {
150 panda_file::File::EntityId entity_id(module_literal_array_id);
151 AbcModuleArrayProcessor module_array_processor(entity_id, entity_container_);
152 module_array_processor.FillProgramData();
153 }
154
FillLiteralArrayById(uint32_t literal_array_id)155 void AbcClassProcessor::FillLiteralArrayById(uint32_t literal_array_id)
156 {
157 panda_file::File::EntityId entity_id(literal_array_id);
158 AbcLiteralArrayProcessor literal_array_processor(entity_id, entity_container_, *literal_data_accessor_);
159 literal_array_processor.FillProgramData();
160 }
161
FillModuleRequestPhaseById(uint32_t module_literal_array_id)162 void AbcClassProcessor::FillModuleRequestPhaseById(uint32_t module_literal_array_id)
163 {
164 panda_file::File::EntityId entity_id(module_literal_array_id);
165 AbcLiteralArrayProcessor literal_array_processor(entity_id, entity_container_, *literal_data_accessor_);
166 literal_array_processor.FillModuleRequestPhase();
167 }
168
169 } // namespace panda::abc2program
170