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 #define MLOG_TAG "MediaLibraryRestore"
16
17 #include "acl.h"
18 #include "medialibrary_restore.h"
19 #include "dfx_utils.h"
20 #include "medialibrary_data_manager.h"
21 #include "medialibrary_rdb_utils.h"
22 #include "medialibrary_tracer.h"
23 #include "media_file_utils.h"
24 #include "media_log.h"
25 #include "parameter.h"
26 #include "parameters.h"
27 #include "post_event_utils.h"
28 #ifdef CLOUD_SYNC_MANAGER
29 #include "cloud_sync_manager.h"
30 #endif
31
32 namespace OHOS {
33 namespace Media {
34 namespace {
35 const std::string SWITCH_STATUS_KEY = "persist.multimedia.medialibrary.rdb_switch_status";
36 const std::string CLOUD_STOP_FLAG_K = "persist.kernel.medialibrarydata.stopflag";
37 const std::string CLOUD_STOP_FLAG_V = "1";
38 constexpr int PARAMETER_E_OK = 0;
39 constexpr int WAIT_SECONDS = 120;
40 constexpr int SLEEP_SECONDS = 60;
41 constexpr int MAX_RETRY_TIMES = 2;
42 enum HA_SWITCH_STATUS : uint32_t {
43 HA_SWITCH_READY = 0,
44 HA_SWITCHING = 1
45 };
46 } // namespace
47
GetInstance()48 MediaLibraryRestore &MediaLibraryRestore::GetInstance()
49 {
50 static MediaLibraryRestore instance;
51 return instance;
52 }
53
SaveHAModeSwitchStatusToPara(const int64_t & status)54 void MediaLibraryRestore::SaveHAModeSwitchStatusToPara(const int64_t &status)
55 {
56 int ret = SetParameter(SWITCH_STATUS_KEY.c_str(), std::to_string(status).c_str());
57 CHECK_AND_RETURN_LOG((ret == PARAMETER_E_OK), "MediaLibraryRestore SetParameter switch error");
58 }
59
CheckRestore(const int32_t & errCode)60 void MediaLibraryRestore::CheckRestore(const int32_t &errCode)
61 {
62 if (errCode != NativeRdb::E_SQLITE_CORRUPT) {
63 return;
64 }
65 MEDIA_INFO_LOG("Restore is called");
66 CHECK_AND_RETURN_LOG((!isRestoring_), "RdbStore is restoring");
67
68 std::string date = DfxUtils::GetCurrentDateMillisecond();
69 VariantMap map = {{KEY_DB_CORRUPT, std::move(date)}};
70 PostEventUtils::GetInstance().PostErrorProcess(ErrType::DB_CORRUPT_ERR, map);
71
72 isRestoring_ = true;
73 std::thread([] {
74 MEDIA_INFO_LOG("MediaLibraryRestore::Restore [start]");
75 auto rdb = MediaLibraryDataManager::GetInstance()->rdbStore_;
76 if (rdb == nullptr) {
77 MediaLibraryRestore::GetInstance().isRestoring_ = false;
78 MEDIA_ERR_LOG("Restore rdbStore is nullptr");
79 return;
80 }
81
82 int retryTimes = MAX_RETRY_TIMES;
83 int errCode = NativeRdb::E_OK;
84 do {
85 MediaLibraryTracer tracer;
86 tracer.Start("MediaLibraryRestore::Restore");
87 errCode = rdb->Restore("");
88 MEDIA_INFO_LOG("MediaLibraryRestore::Restore errCode = %{public}d", errCode);
89 if (errCode == NativeRdb::E_SQLITE_BUSY) {
90 retryTimes--;
91 continue;
92 }
93 break;
94 } while (retryTimes > 0);
95 MEDIA_INFO_LOG("MediaLibraryRestore::Restore [end]. errCode = %{public}d", errCode);
96 MediaLibraryRestore::GetInstance().isRestoring_ = false;
97 }).detach();
98 }
99
100 #ifdef CLOUD_SYNC_MANAGER
StopCloudSync()101 void MediaLibraryRestore::StopCloudSync()
102 {
103 MediaLibraryTracer tracer;
104 tracer.Start("MediaLibraryRestore::StopCloudSync");
105 FileManagement::CloudSync::CloudSyncManager::GetInstance().StopSync(BUNDLE_NAME, true);
106 uint32_t times = 0;
107 int ret = WaitParameter(CLOUD_STOP_FLAG_K.c_str(), CLOUD_STOP_FLAG_V.c_str(), WAIT_SECONDS);
108 if (ret == PARAMETER_E_OK) {
109 MEDIA_INFO_LOG("StopCloudSync success end");
110 return;
111 }
112 isBackuping_ = false;
113 MEDIA_INFO_LOG("StopCloudSync timeout error, set backup false");
114 return;
115 }
116 #endif
117
CheckBackup()118 void MediaLibraryRestore::CheckBackup()
119 {
120 MEDIA_INFO_LOG("CheckBackup is called");
121 CHECK_AND_RETURN_LOG((!isRestoring_), "CheckBackup: is restoring, return");
122
123 auto rdb = MediaLibraryDataManager::GetInstance()->rdbStore_;
124 CHECK_AND_RETURN_LOG((rdb != nullptr), "CheckBackup: rdbStore is nullptr");
125 if (!rdb->IsSlaveDiffFromMaster()) {
126 MEDIA_INFO_LOG("CheckBackup: isSlaveDiffFromMaster [false], return");
127 return;
128 }
129 MEDIA_INFO_LOG("CheckBackup: isSlaveDiffFromMaster [true]");
130 DoRdbBackup();
131 }
132
ResetHAModeSwitchStatus()133 void MediaLibraryRestore::ResetHAModeSwitchStatus()
134 {
135 auto switchStatus = HA_SWITCH_READY;
136 SaveHAModeSwitchStatusToPara(std::move(switchStatus));
137 #ifdef CLOUD_SYNC_MANAGER
138 auto ret = FileManagement::CloudSync::CloudSyncManager::GetInstance().StartSync(BUNDLE_NAME);
139 MEDIA_INFO_LOG("ResetHAModeSwitchStatus::StartSync [%{public}d]", ret);
140 ret = FileManagement::CloudSync::CloudSyncManager::GetInstance().DownloadThumb();
141 MEDIA_INFO_LOG("ResetHAModeSwitchStatus::DownloadThumb [%{public}d]", ret);
142 #endif
143 }
144
DoRdbBackup()145 void MediaLibraryRestore::DoRdbBackup()
146 {
147 CHECK_AND_RETURN_LOG((!isBackuping_.load()), "DoRdbBackup: is backuping, return");
148 isBackuping_ = true;
149
150 std::thread([] {
151 MEDIA_INFO_LOG("DoRdbBackup: Backup [start]");
152 {
153 MEDIA_INFO_LOG("DoRdbBackup: wait_for [start]");
154 std::unique_lock<std::mutex> lock(MediaLibraryRestore::GetInstance().mutex_);
155 MediaLibraryRestore::GetInstance().cv_.wait_for(lock, std::chrono::seconds(SLEEP_SECONDS),
156 [] { return !MediaLibraryRestore::GetInstance().IsBackuping(); });
157 MEDIA_INFO_LOG("DoRdbBackup: wait_for [end]");
158 }
159 CHECK_AND_RETURN_LOG((MediaLibraryRestore::GetInstance().IsBackuping()),
160 "DoRdbBackup: after sleep isbackuping fasle, return");
161 auto currentTime = MediaFileUtils::UTCTimeSeconds();
162 MediaLibraryRestore::GetInstance().SaveHAModeSwitchStatusToPara(currentTime);
163 #ifdef CLOUD_SYNC_MANAGER
164 MediaLibraryRestore::GetInstance().StopCloudSync();
165 #endif
166 auto rdb = MediaLibraryDataManager::GetInstance()->rdbStore_;
167 if (rdb == nullptr || !MediaLibraryRestore::GetInstance().IsBackuping()) {
168 MediaLibraryRestore::GetInstance().ResetHAModeSwitchStatus();
169 MediaLibraryRestore::GetInstance().isBackuping_ = false;
170 MEDIA_ERR_LOG("DoRdbBackup: rdb is nullptr or interrupt or isbackuping false");
171 return;
172 }
173 MediaLibraryTracer tracer;
174 tracer.Start("MediaLibraryRestore::DoRdbBackup Backup");
175 MediaLibraryRestore::GetInstance().isDoingBackup_ = true;
176 if (!MediaLibraryRdbUtils::ExecuteDatabaseQuickCheck(rdb)
177 || !MediaLibraryRestore::GetInstance().IsBackuping()) {
178 MediaLibraryRestore::GetInstance().ResetHAModeSwitchStatus();
179 MediaLibraryRestore::GetInstance().isBackuping_ = false;
180 MEDIA_ERR_LOG("DoRdbBackup: QuickCheck fail");
181 return;
182 }
183 MEDIA_INFO_LOG("DoRdbBackup: Backup [start]");
184 int errCode = rdb->Backup("", false);
185 MediaLibraryRestore::GetInstance().isDoingBackup_ = false;
186 if (errCode == NativeRdb::E_OK) {
187 Acl::AclSetSlaveDatabase();
188 }
189 MEDIA_INFO_LOG("DoRdbBackup: Backup [end]. errCode = %{public}d", errCode);
190 MediaLibraryRestore::GetInstance().ResetHAModeSwitchStatus();
191 if (errCode != NativeRdb::E_CANCEL) {
192 MediaLibraryRestore::GetInstance().isBackuping_ = false;
193 }
194 }).detach();
195 }
196
InterruptBackup()197 void MediaLibraryRestore::InterruptBackup()
198 {
199 MEDIA_INFO_LOG("InterruptBackup [start]");
200 if (!isBackuping_.load()) {
201 MEDIA_INFO_LOG("rdb is not backuping, no need to interrupt");
202 return;
203 }
204 auto rdb = MediaLibraryDataManager::GetInstance()->rdbStore_;
205 CHECK_AND_RETURN_LOG((rdb != nullptr), "[InterruptBackup] rdbStore is nullptr");
206 int errCode = rdb->InterruptBackup();
207 isBackuping_ = false;
208 ResetHAModeSwitchStatus();
209 cv_.notify_all();
210 MEDIA_INFO_LOG("InterruptBackup [end]. errCode = %{public}d", errCode);
211 }
212
CheckResultSet(const std::shared_ptr<NativeRdb::ResultSet> & resultSet)213 void MediaLibraryRestore::CheckResultSet(const std::shared_ptr<NativeRdb::ResultSet> &resultSet)
214 {
215 MediaLibraryTracer tracer;
216 tracer.Start("CheckResultSet");
217 if (resultSet == nullptr) {
218 return;
219 }
220 int count;
221 int errCode = resultSet->GetRowCount(count);
222 CheckRestore(errCode);
223 }
224
IsRestoring() const225 bool MediaLibraryRestore::IsRestoring() const
226 {
227 return isRestoring_;
228 }
229
IsBackuping() const230 bool MediaLibraryRestore::IsBackuping() const
231 {
232 return isBackuping_.load();
233 }
234
IsRealBackuping() const235 bool MediaLibraryRestore::IsRealBackuping() const
236 {
237 return isDoingBackup_;
238 }
239 } // namespace Media
240 } // namespace OHOS
241