OpenHarmony-v6.0-Release/s

/*
 * Copyright (c) Huawei Technologies Co., Ltd. 2023. All rights reserved.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#include "app_start_filter.h"

namespace SysTuning {
namespace TraceStreamer {
constexpr uint32_t INVALID_DATA = 2;
constexpr uint32_t MIN_VECTOR_SIZE = 2;
constexpr uint32_t VALID_DATA_COUNT = 6;
APPStartupFilter::APPStartupFilter(TraceDataCache *dataCache, const TraceStreamerFilters *filter)
    : FilterBase(dataCache, filter), mAPPStartupData_(0)
{
}

APPStartupFilter::~APPStartupFilter() = default;

void APPStartupFilter::FilterAllAPPStartupData()
{
    ParserAppStartup();
    ParserSoInitalization();
}

bool APPStartupFilter::CaclRsDataByPid(appMap &mAPPStartupData)
{
    auto frameSliceData = traceDataCache_->GetFrameSliceData();
    auto sliceData = traceDataCache_->GetConstInternalSlicesData();
    for (const auto &item : mAPPStartupData) {
        if (item.second.empty()) {
            continue;
        }
        auto itorSecond = item.second.begin();
        if (itorSecond->second->ipid_ == INVALID_UINT32) {
            continue;
        }
        for (int m = 0; m < frameSliceData->Ipids().size(); m++) {
            if (itorSecond->second->ipid_ == frameSliceData->Ipids()[m] && !frameSliceData->Types()[m] &&
                frameSliceData->Flags()[m] != INVALID_DATA) {
                auto startTime = sliceData.TimeStampData()[frameSliceData->CallStackIds()[m]];
                auto callId = sliceData.CallIds()[frameSliceData->CallStackIds()[m]];
                auto endTime = startTime + frameSliceData->Durs()[m];
                if (frameSliceData->Durs()[m] == INVALID_UINT64) {
                    endTime = INVALID_UINT64;
                }
                mAPPStartupData[item.first].emplace(FIRST_FRAME_APP_PHASE,
                                                    std::make_unique<APPStartupData>(callId, itorSecond->second->ipid_,
                                                                                     itorSecond->second->tid_,
                                                                                     startTime, endTime));
                auto dstId = frameSliceData->Dsts()[m];
                if (dstId == INVALID_UINT64) {
                    continue;
                }
                callId = sliceData.CallIds()[frameSliceData->CallStackIds()[dstId]];
                startTime = frameSliceData->TimeStampData()[dstId];
                endTime = startTime + frameSliceData->Durs()[dstId];
                mAPPStartupData[item.first].emplace(FIRST_FRAME_RENDER_PHASE,
                                                    std::make_unique<APPStartupData>(callId, itorSecond->second->ipid_,
                                                                                     itorSecond->second->tid_,
                                                                                     startTime, endTime));
                break;
            }
        }
    }
    return false;
}

void APPStartupFilter::UpdatePidByNameIndex(const appMap &mAPPStartupData)
{
    auto threadData = traceDataCache_->GetConstThreadData();
    for (const auto &item : mAPPStartupData) {
        auto ipid = INVALID_UINT32;
        auto tid = INVALID_UINT32;
        if (item.second.count(UI_ABILITY_LAUNCHING)) {
            ipid = item.second.at(UI_ABILITY_LAUNCHING)->ipid_;
            tid = item.second.at(UI_ABILITY_LAUNCHING)->tid_;
        } else {
            for (int i = 0; i < threadData.size(); ++i) {
                if (item.first == threadData[i].nameIndex_) {
                    ipid = threadData[i].internalPid_;
                    tid = threadData[i].tid_;
                }
            }
        }
        for (const auto &itemSecond : item.second) {
            itemSecond.second->ipid_ = ipid;
            itemSecond.second->tid_ = tid;
        }
    }
}

void APPStartupFilter::AppendAssociatedData(DataIndex packedNameIndex,
                                            const std::map<uint32_t, std::unique_ptr<APPStartupData>> &stagesData)
{
    for (auto itorSecond = stagesData.begin(); itorSecond != stagesData.end(); ++itorSecond) {
        auto item = itorSecond;
        auto endTime = INVALID_UINT64;
        if (item->first < VALID_DATA_COUNT) {
            int num = item->first + 1;
            if ((++item) != stagesData.end() && num == item->first) {
                endTime = (item)->second->startTime_;
            }
        } else {
            endTime = itorSecond->second->endTime_;
        }
        AppStartupRow appStartupRow = {itorSecond->second->ipid_,
                                       itorSecond->second->tid_,
                                       itorSecond->second->callid_,
                                       itorSecond->second->startTime_,
                                       endTime,
                                       itorSecond->first,
                                       packedNameIndex};
        traceDataCache_->GetAppStartupData()->AppendNewData(appStartupRow);
    }
}

// AppenAllData
void APPStartupFilter::AppenAllData(const appMap &mAPPStartupData)
{
    for (auto itor = mAPPStartupData.begin(); itor != mAPPStartupData.end(); ++itor) {
        if (!(itor->second).count(UI_ABILITY_LAUNCHING)) {
            continue;
        }
        AppendAssociatedData(itor->first, itor->second);
    }
}

bool APPStartupFilter::UpdateAPPStartupData(uint32_t row, const std::string &nameString, uint32_t startIndex)
{
    auto sliceData = traceDataCache_->GetConstInternalSlicesData();
    auto vNameString = SplitStringToVec(nameString, "##");
    TS_CHECK_TRUE_RET(vNameString.size() >= MIN_VECTOR_SIZE, false);
    auto dataIndex = traceDataCache_->GetDataIndex(vNameString[1].c_str());
    // for update the last Item(valid val)
    if (!procTouchItems_.empty()) {
        auto lastProcTouchItem = std::move(procTouchItems_.back());
        mAPPStartupData_[dataIndex].insert(std::make_pair(PROCESS_TOUCH, std::move(lastProcTouchItem)));
        procTouchItems_.clear();
    }
    if (!startUIAbilityBySCBItems_.empty()) {
        auto lastStartUIAbilityBySCBItem = std::move(startUIAbilityBySCBItems_.back());
        mAPPStartupData_[dataIndex].insert(
            std::make_pair(START_UI_ABILITY_BY_SCB, std::move(lastStartUIAbilityBySCBItem)));
        startUIAbilityBySCBItems_.clear();
    }
    if (!loadAbilityItems_.empty()) {
        auto lastLoadAbilityItem = std::move(loadAbilityItems_.back());
        mAPPStartupData_[dataIndex].insert(std::make_pair(LOAD_ABILITY, std::move(lastLoadAbilityItem)));
        loadAbilityItems_.clear();
    }
    auto callId = sliceData.CallIds()[row];
    auto startTime = sliceData.TimeStampData()[row];
    mAPPStartupData_[dataIndex].insert(
        std::make_pair(startIndex, std::make_unique<APPStartupData>(callId, INVALID_UINT32, INVALID_UINT32, startTime,
                                                                    INVALID_UINT64)));
    return true;
}

void APPStartupFilter::ParserAppStartup()
{
    auto sliceData = traceDataCache_->GetConstInternalSlicesData();
    std::string mainThreadName = "";
    for (auto i = 0; i < sliceData.NamesData().size(); i++) {
        auto &nameString = traceDataCache_->GetDataFromDict(sliceData.NamesData()[i]);
        auto callId = sliceData.CallIds()[i];
        auto startTime = sliceData.TimeStampData()[i];
        if (streamFilters_->configFilter_->GetAppStartupConfig().CheckIfPhase1(nameString)) {
            procTouchItems_.emplace_back(
                std::make_unique<APPStartupData>(callId, INVALID_UINT32, INVALID_UINT32, startTime, INVALID_UINT64));
        } else if (streamFilters_->configFilter_->GetAppStartupConfig().CheckIfPhase2(nameString)) {
            startUIAbilityBySCBItems_.emplace_back(
                std::make_unique<APPStartupData>(callId, INVALID_UINT32, INVALID_UINT32, startTime, INVALID_UINT64));
        } else if (streamFilters_->configFilter_->GetAppStartupConfig().CheckIfPhase3(nameString)) {
            loadAbilityItems_.emplace_back(
                std::make_unique<APPStartupData>(callId, INVALID_UINT32, INVALID_UINT32, startTime, INVALID_UINT64));
        } else if (streamFilters_->configFilter_->GetAppStartupConfig().CheckIfPhase4(nameString)) {
            UpdateAPPStartupData(i, nameString, APPLICATION_LAUNCHING);
        } else if (streamFilters_->configFilter_->GetAppStartupConfig().CheckIfPhase5(nameString)) {
            ProcAbilityLaunchData(nameString, i);
        } else if (streamFilters_->configFilter_->GetAppStartupConfig().CheckIfPhase6(nameString)) {
            ProcForegroundData(i);
        }
    }
    for (auto &item : mAPPStartupDataWithPid_) {
        UpdatePidByNameIndex(item.second);
        CaclRsDataByPid(item.second);
        AppenAllData(item.second);
    }
    return;
}

bool APPStartupFilter::ProcAbilityLaunchData(const std::string &nameString, uint64_t raw)
{
    auto sliceData = traceDataCache_->GetConstInternalSlicesData();
    auto vNameString = SplitStringToVec(nameString, "##");
    TS_CHECK_TRUE_RET(vNameString.size() >= MIN_VECTOR_SIZE, false);
    auto dataIndex = traceDataCache_->GetDataIndex(vNameString[1].c_str());
    uint32_t callId = sliceData.CallIds()[raw];
    uint64_t startTime = sliceData.TimeStampData()[raw];
    if (dataIndex == INVALID_DATAINDEX) {
        return false;
    }
    auto thread = traceDataCache_->GetThreadData(sliceData.CallIds()[raw]);
    thread->nameIndex_ = dataIndex;
    mAPPStartupData_[dataIndex].insert(std::make_pair(
        UI_ABILITY_LAUNCHING,
        std::make_unique<APPStartupData>(callId, thread->internalPid_, thread->tid_, startTime, INVALID_UINT64)));
    mAPPStartupDataWithPid_.insert(std::make_pair(thread->internalPid_, std::move(mAPPStartupData_)));
    return true;
}

void APPStartupFilter::ProcForegroundData(uint64_t raw)
{
    auto sliceData = traceDataCache_->GetConstInternalSlicesData();
    uint32_t callId = sliceData.CallIds()[raw];
    uint64_t startTime = sliceData.TimeStampData()[raw];
    auto threadData = traceDataCache_->GetConstThreadData();
    // callid is thread table->itid
    auto nameindex = threadData[callId].nameIndex_;
    auto ipid = threadData[callId].internalPid_;
    auto tid = threadData[callId].tid_;
    if (mAPPStartupDataWithPid_.count(ipid) && mAPPStartupDataWithPid_[ipid].count(nameindex) &&
        !mAPPStartupDataWithPid_[ipid][nameindex].count(UI_ABILITY_ONFOREGROUND)) {
        mAPPStartupDataWithPid_[ipid][nameindex].insert(std::make_pair(
            UI_ABILITY_ONFOREGROUND, std::make_unique<APPStartupData>(callId, ipid, tid, startTime, INVALID_UINT64)));
    }
}

void APPStartupFilter::CalcDepthByTimeStamp(std::map<uint32_t, std::map<uint64_t, uint32_t>>::iterator it,
                                            uint32_t &depth,
                                            uint64_t endTime,
                                            uint64_t startTime)
{
    if (it->second.empty()) {
        it->second.insert(std::make_pair(endTime, 0));
        return;
    }
    auto itor = it->second.begin();
    if (itor->first > startTime) {
        depth = it->second.size();
        it->second.insert(std::make_pair(endTime, it->second.size()));
        return;
    }
    depth = itor->second;
    for (auto itorSecond = itor; itorSecond != it->second.end(); ++itorSecond) {
        if (itorSecond->first < startTime && depth > itorSecond->second) {
            depth = itorSecond->second;
            itor = itorSecond;
        }
    }
    it->second.erase(itor);
    it->second.insert(std::make_pair(endTime, depth));
}

void APPStartupFilter::ParserSoInitalization()
{
    std::map<uint32_t, std::map<uint64_t, uint32_t>> mMaxTimeAndDepthWithPid;
    auto sliceData = traceDataCache_->GetConstInternalSlicesData();
    auto threadData = traceDataCache_->GetConstThreadData();
    std::string nameString = "";
    for (auto i = 0; i < sliceData.NamesData().size(); i++) {
        nameString = traceDataCache_->GetDataFromDict(sliceData.NamesData()[i]);
        if (nameString.find(dlopenCmd_) != std::string::npos) {
            uint64_t startTime = sliceData.TimeStampData()[i];
            uint64_t endTime = startTime + sliceData.DursData()[i];
            uint32_t depth = 0;
            auto callId = sliceData.CallIds()[i];
            auto pid = threadData[callId].internalPid_;
            auto tid = threadData[callId].tid_;
            auto it = mMaxTimeAndDepthWithPid.find(pid);
            SoStaticInitalizationRow soStaticInitalizationRow;
            if (it == mMaxTimeAndDepthWithPid.end()) {
                mMaxTimeAndDepthWithPid.insert(std::make_pair(pid, std::map<uint64_t, uint32_t>{{endTime, 0}}));
                soStaticInitalizationRow.ipid = pid;
                soStaticInitalizationRow.tid = tid;
                soStaticInitalizationRow.callId = callId;
                soStaticInitalizationRow.startTime = startTime;
                soStaticInitalizationRow.endTime = endTime;
                soStaticInitalizationRow.soName = sliceData.NamesData()[i];
                soStaticInitalizationRow.depth = depth;
                traceDataCache_->GetSoStaticInitalizationData()->AppendNewData(soStaticInitalizationRow);
                continue;
            } else {
                CalcDepthByTimeStamp(it, depth, endTime, startTime);
                soStaticInitalizationRow.ipid = threadData[callId].internalPid_;
                soStaticInitalizationRow.tid = threadData[callId].tid_;
                soStaticInitalizationRow.callId = callId;
                soStaticInitalizationRow.startTime = startTime;
                soStaticInitalizationRow.endTime = endTime;
                soStaticInitalizationRow.soName = sliceData.NamesData()[i];
                soStaticInitalizationRow.depth = depth;
                traceDataCache_->GetSoStaticInitalizationData()->AppendNewData(soStaticInitalizationRow);
            }
        }
    }
}
} // namespace TraceStreamer
} // namespace SysTuning