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 #ifndef OHOS_ABILITY_RUNTIME_APP_EXIT_REASON_DATA_MANAGER_H 17 #define OHOS_ABILITY_RUNTIME_APP_EXIT_REASON_DATA_MANAGER_H 18 19 #include <mutex> 20 #include <string> 21 #include <vector> 22 23 #include "ability_util.h" 24 #include "distributed_kv_data_manager.h" 25 #include "exit_reason.h" 26 #include "singleton.h" 27 28 namespace OHOS { 29 namespace AbilityRuntime { 30 class AppExitReasonDataManager : public DelayedSingleton<AppExitReasonDataManager> { 31 public: 32 AppExitReasonDataManager(); 33 34 virtual ~AppExitReasonDataManager(); 35 36 int32_t SetAppExitReason(const std::string &bundleName, uint32_t accessTokenId, 37 const std::vector<std::string> &abilityList, const AAFwk::ExitReason &exitReason, 38 const AppExecFwk::RunningProcessInfo &processInfo, bool withKillMsg); 39 40 int32_t GetAppExitReason(const std::string &bundleName, uint32_t accessTokenId, const std::string &abilityName, 41 bool &isSetReason, AAFwk::ExitReason &exitReason, AppExecFwk::RunningProcessInfo &processInfo, 42 int64_t &time_stamp, bool &withKillMsg); 43 44 int32_t DeleteAppExitReason(const std::string &bundleName, int32_t uid, int32_t appIndex); 45 46 int32_t DeleteAppExitReason(const std::string &bundleName, uint32_t accessTokenId); 47 48 int32_t AddAbilityRecoverInfo(uint32_t accessTokenId, 49 const std::string &moduleName, const std::string &abilityName, const int &sessionId); 50 51 int32_t DeleteAbilityRecoverInfo( 52 uint32_t accessTokenId, const std::string &moduleName, const std::string &abilityName); 53 54 int32_t DeleteAllRecoverInfoByTokenId(uint32_t tokenId); 55 56 int32_t DeleteAbilityRecoverInfoBySessionId(const int32_t sessionId); 57 58 int32_t GetAbilityRecoverInfo(uint32_t accessTokenId, 59 const std::string &moduleName, const std::string &abilityName, bool &hasRecoverInfo); 60 61 uint32_t GetTokenIdBySessionID(const int32_t sessionId); 62 63 int32_t SetUIExtensionAbilityExitReason(const std::string &bundleName, 64 const std::vector<std::string> &extensionList, const AAFwk::ExitReason &exitReason, 65 const AppExecFwk::RunningProcessInfo &processInfo, bool withKillMsg); 66 67 bool GetUIExtensionAbilityExitReason(const std::string &keyEx, AAFwk::ExitReason &exitReason, 68 AppExecFwk::RunningProcessInfo &processInfo, int64_t &time_stamp, bool &withKillMsg); 69 70 int32_t RecordSignalReason(int32_t pid, int32_t uid, int32_t signal, std::string &bundleName); 71 72 int32_t GetRecordAppAbilityNames(const uint32_t accessTokenId, std::vector<std::string> &abilityLists); 73 74 void UpdateAppExitReason(uint32_t accessTokenId, const std::vector<std::string> &abilityList, 75 const AAFwk::ExitReason &exitReason, const AppExecFwk::RunningProcessInfo &processInfo, bool withKillMsg); 76 77 private: 78 DistributedKv::Status GetKvStore(); 79 bool CheckKvStore(); 80 DistributedKv::Value ConvertAppExitReasonInfoToValue( 81 const std::vector<std::string> &abilityList, const AAFwk::ExitReason &exitReason, 82 const AppExecFwk::RunningProcessInfo &processInfo, bool withKillMsg); 83 void ConvertAppExitReasonInfoFromValue(const DistributedKv::Value &value, AAFwk::ExitReason &exitReason, 84 int64_t &time_stamp, std::vector<std::string> &abilityList, AppExecFwk::RunningProcessInfo &processInfo, 85 bool &withKillMsg); 86 void ConvertReasonFromValue(const nlohmann::json &jsonObject, AAFwk::ExitReason &exitReason, bool &withKillMsg); 87 void ConvertAccessTokenIdFromValue(const DistributedKv::Value &value, uint32_t &accessTokenId); 88 void InnerDeleteAppExitReason(const std::string &keyName); 89 void InnerDeleteSessionId(const int32_t sessionId); 90 void InnerAddSessionId(const int32_t sessionId, uint32_t accessTokenId); 91 92 void UpdateAbilityRecoverInfo(uint32_t accessTokenId, 93 const std::vector<std::string> &recoverInfoList, const std::vector<int> &sessionIdList); 94 DistributedKv::Value ConvertAbilityRecoverInfoToValue( 95 const std::vector<std::string> &recoverInfoList, const std::vector<int> &sessionIdList); 96 void ConvertAbilityRecoverInfoFromValue( 97 const DistributedKv::Value &value, std::vector<std::string> &recoverInfoList, std::vector<int> &sessionIdList); 98 void InnerDeleteAbilityRecoverInfo(uint32_t accessTokenId); 99 DistributedKv::Key GetAbilityRecoverInfoKey(uint32_t accessTokenId); 100 DistributedKv::Value ConvertAppExitReasonInfoToValueOfExtensionName( 101 const std::string &extensionListName, const AAFwk::ExitReason &exitReason, 102 const AppExecFwk::RunningProcessInfo &processInfo, bool withKillMsg); 103 104 DistributedKv::Key GetSessionIdKey(const int32_t sessionId); 105 DistributedKv::Value ConvertAccessTokenIdToValue(uint32_t accessTokenId); 106 DistributedKv::Status RestoreKvStore(DistributedKv::Status status); 107 108 const DistributedKv::AppId appId_ { "app_exit_reason_storage" }; 109 const DistributedKv::StoreId storeId_ { "app_exit_reason_infos" }; 110 DistributedKv::DistributedKvDataManager dataManager_; 111 std::shared_ptr<DistributedKv::SingleKvStore> kvStorePtr_; 112 mutable std::mutex kvStorePtrMutex_; 113 }; 114 } // namespace AbilityRuntime 115 } // namespace OHOS 116 #endif // OHOS_ABILITY_RUNTIME_APP_EXIT_REASON_DATA_MANAGER_H