1 /*
2 * Copyright (c) 2023 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 "ecmascript/compiler/aot_snapshot/snapshot_global_data.h"
17
18 #include "ecmascript/compiler/aot_snapshot/aot_snapshot_constants.h"
19 #include "ecmascript/js_tagged_value.h"
20 #include "ecmascript/jspandafile/program_object.h"
21 #include "ecmascript/tagged_array-inl.h"
22
23 namespace panda::ecmascript::kungfu {
GetConstantPoolFromSnapshotData(JSThread * thread,const SnapshotGlobalData * globalData,uint32_t dataIdx,uint32_t cpArrayIdx)24 JSHandle<ConstantPool> ReviseData::GetConstantPoolFromSnapshotData(JSThread *thread,
25 const SnapshotGlobalData *globalData,
26 uint32_t dataIdx, uint32_t cpArrayIdx)
27 {
28 JSHandle<TaggedArray> data(thread, globalData->GetData());
29 auto cpArrayOffset = SnapshotGlobalData::Cast(SnapshotGlobalData::CP_TOP_ITEM::CP_ARRAY_ID);
30 JSHandle<TaggedArray> cpArr(thread, data->Get(dataIdx + cpArrayOffset));
31 return JSHandle<ConstantPool>(thread, cpArr->Get(cpArrayIdx));
32 }
33
Resolve(JSThread * thread,const SnapshotGlobalData * globalData,const CMap<std::pair<std::string,uint32_t>,uint32_t> & methodToEntryIndexMap)34 void ReviseData::Resolve(JSThread *thread, const SnapshotGlobalData *globalData,
35 const CMap<std::pair<std::string, uint32_t>, uint32_t> &methodToEntryIndexMap)
36 {
37 for (auto &item: data_) {
38 JSHandle<ConstantPool> newCP = GetConstantPoolFromSnapshotData(thread, globalData,
39 item.dataIdx_, item.cpArrayIdx_);
40
41 JSTaggedValue val = newCP->GetObjectFromCache(item.constpoolIdx_);
42 AOTLiteralInfo *aotLiteralInfo = AOTLiteralInfo::Cast(val.GetTaggedObject());
43 uint32_t aotLiteralInfoLen = aotLiteralInfo->GetCacheLength();
44 std::string name = globalData->GetFileNameByDataIdx(item.dataIdx_).c_str();
45 for (uint32_t i = 0; i < aotLiteralInfoLen; ++i) {
46 JSTaggedValue methodOffsetVal = aotLiteralInfo->GetObjectFromCache(i);
47 if (methodOffsetVal.GetInt() == -1) {
48 continue;
49 }
50 uint32_t methodOffset = static_cast<uint32_t>(methodOffsetVal.GetInt());
51 if (thread->GetEcmaVM()->GetJSOptions().IsEnableCompilerLogSnapshot()) {
52 LOG_COMPILER(INFO) << "[aot-snapshot] store AOT entry index of method (offset: "
53 << methodOffset << ") ";
54 }
55 AnFileInfo::FuncEntryIndexKey key = std::make_pair(name, methodOffset);
56 uint32_t entryIndex = methodToEntryIndexMap.at(key);
57 aotLiteralInfo->SetObjectToCache(thread, i, JSTaggedValue(entryIndex));
58 }
59 }
60 }
61
AddSnapshotCpArrayToData(JSThread * thread,CString fileName,uint32_t fileIndex,JSHandle<TaggedArray> snapshotCpArray)62 void SnapshotGlobalData::AddSnapshotCpArrayToData(JSThread *thread, CString fileName, uint32_t fileIndex,
63 JSHandle<TaggedArray> snapshotCpArray)
64 {
65 if (isFirstData_) {
66 isFirstData_ = false;
67 } else {
68 curDataIdx_ += AOTSnapshotConstants::SNAPSHOT_DATA_ITEM_SIZE;
69 }
70 // handle file info
71 JSHandle<EcmaString> nameStr = thread->GetEcmaVM()->GetFactory()->NewFromStdString(fileName.c_str());
72 auto fileInfo = thread->GetEcmaVM()->GetFactory()->NewTaggedArray(Cast(CP_PANDA_INFO_ITEM::COUNT));
73 fileInfo->Set(thread, Cast(CP_PANDA_INFO_ITEM::NAME_ID), nameStr);
74 fileInfo->Set(thread, Cast(CP_PANDA_INFO_ITEM::INDEX_ID), JSTaggedValue(fileIndex));
75
76 JSHandle<TaggedArray> dataHandle(thread, data_);
77 dataHandle->Set(thread, curDataIdx_ + Cast(CP_TOP_ITEM::PANDA_INFO_ID), fileInfo);
78
79 // handle constant pool
80 curSnapshotCpArray_ = snapshotCpArray.GetTaggedValue();
81 dataHandle->Set(thread, curDataIdx_ + Cast(CP_TOP_ITEM::CP_ARRAY_ID), curSnapshotCpArray_);
82 dataIdxToFileNameMap_[curDataIdx_] = fileName;
83 }
84
GetFileNameByDataIdx(uint32_t dataIdx) const85 CString SnapshotGlobalData::GetFileNameByDataIdx(uint32_t dataIdx) const
86 {
87 auto it = dataIdxToFileNameMap_.find(dataIdx);
88 if (it != dataIdxToFileNameMap_.end()) {
89 return it->second;
90 }
91 LOG_COMPILER(FATAL) << "Can't find snapshot data by index '" << dataIdx << "'";
92 UNREACHABLE();
93 }
94 } // namespace panda::ecmascript
95