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 "rdb_global_params_callback.h"
17
18 #include "data_storage_errors.h"
19 #include "data_storage_log_wrapper.h"
20 #include "parser_util.h"
21 #include "global_params_data.h"
22 #include "rdb_errno.h"
23 #include "rdb_store.h"
24 #include "string"
25 #include "values_bucket.h"
26
27 namespace OHOS {
28 namespace Telephony {
RdbGlobalParamsCallback(const std::vector<std::string> & createTableVec)29 RdbGlobalParamsCallback::RdbGlobalParamsCallback(
30 const std::vector<std::string> &createTableVec) : RdbBaseCallBack(createTableVec) {}
31
OnUpgrade(NativeRdb::RdbStore & rdbStore,int oldVersion,int newVersion)32 int RdbGlobalParamsCallback::OnUpgrade(NativeRdb::RdbStore &rdbStore, int oldVersion, int newVersion)
33 {
34 DATA_STORAGE_LOGI(
35 "RdbGlobalParamsCallback::OnUpgrade##oldVersion = %d, newVersion = %d\n", oldVersion, newVersion);
36 return NativeRdb::E_OK;
37 }
38
OnDowngrade(NativeRdb::RdbStore & rdbStore,int currentVersion,int targetVersion)39 int RdbGlobalParamsCallback::OnDowngrade(NativeRdb::RdbStore &rdbStore, int currentVersion, int targetVersion)
40 {
41 DATA_STORAGE_LOGI(
42 "RdbGlobalParamsCallback::OnDowngrade##currentVersion = %d, "
43 "targetVersion = %d\n",
44 currentVersion, targetVersion);
45 return NativeRdb::E_OK;
46 }
47
OnCreate(NativeRdb::RdbStore & rdbStore)48 int RdbGlobalParamsCallback::OnCreate(NativeRdb::RdbStore &rdbStore)
49 {
50 DATA_STORAGE_LOGI("RdbGlobalParamsCallback::OnCreate");
51 RdbBaseCallBack::OnCreate(rdbStore);
52 InitData(rdbStore, TABLE_NUMBER_MATCH, false);
53 InitData(rdbStore, TABLE_ECC_DATA, false);
54 return NativeRdb::E_OK;
55 }
56
OnOpen(NativeRdb::RdbStore & rdbStore)57 int RdbGlobalParamsCallback::OnOpen(NativeRdb::RdbStore &rdbStore)
58 {
59 DATA_STORAGE_LOGD("RdbGlobalParamsCallback::OnOpen");
60 InitData(rdbStore, TABLE_NUMBER_MATCH, true);
61 InitData(rdbStore, TABLE_ECC_DATA, true);
62 return NativeRdb::E_OK;
63 }
64
InitEccData(NativeRdb::RdbStore & rdbStore,const std::string & tableName,const bool hashCheck)65 void RdbGlobalParamsCallback::InitEccData(NativeRdb::RdbStore &rdbStore, const std::string &tableName,
66 const bool hashCheck)
67 {
68 DATA_STORAGE_LOGD("InitData start");
69 ParserUtil util;
70 std::vector<EccNum> vec;
71 int ret = util.ParserEccDataJson(vec, hashCheck);
72 if (ret != DATA_STORAGE_SUCCESS) {
73 if (ret != FILE_HASH_NO_CHANGE) {
74 DATA_STORAGE_LOGE("ParserEccDataJson fail ret = %{public}d", ret);
75 }
76 util.ClearTempDigest(ECC_DATA_HASH);
77 return;
78 }
79 ClearData(rdbStore, tableName);
80 ret = rdbStore.BeginTransaction();
81 if (ret != NativeRdb::E_OK) {
82 DATA_STORAGE_LOGE("BeginTransaction error!");
83 util.ClearTempDigest(ECC_DATA_HASH);
84 return;
85 }
86 DATA_STORAGE_LOGD("InitData size = %{public}zu", vec.size());
87 for (size_t i = 0; i < vec.size(); i++) {
88 NativeRdb::ValuesBucket value;
89 util.ParserEccDataToValuesBucket(value, vec[i]);
90 int64_t id;
91 rdbStore.Insert(id, tableName, value);
92 }
93 if (rdbStore.Commit() == NativeRdb::E_OK) {
94 util.RefreshDigest(ECC_DATA_HASH);
95 } else {
96 util.ClearTempDigest(ECC_DATA_HASH);
97 }
98 DATA_STORAGE_LOGD("InitData end");
99 }
100
InitNumMatchData(NativeRdb::RdbStore & rdbStore,const std::string & tableName,const bool hashCheck)101 void RdbGlobalParamsCallback::InitNumMatchData(NativeRdb::RdbStore &rdbStore, const std::string &tableName,
102 const bool hashCheck)
103 {
104 DATA_STORAGE_LOGD("InitData start");
105 ParserUtil util;
106 std::vector<NumMatch> vec;
107 int resultCode = util.ParserNumMatchJson(vec, hashCheck);
108 if (resultCode != DATA_STORAGE_SUCCESS) {
109 if (resultCode != FILE_HASH_NO_CHANGE) {
110 DATA_STORAGE_LOGE("ParserNumMatchJson fail resultCode = %{public}d", resultCode);
111 }
112 util.ClearTempDigest(NUM_MATCH_HASH);
113 return;
114 }
115 ClearData(rdbStore, tableName);
116 resultCode = rdbStore.BeginTransaction();
117 if (resultCode != NativeRdb::E_OK) {
118 DATA_STORAGE_LOGE("BeginTransaction error!");
119 util.ClearTempDigest(NUM_MATCH_HASH);
120 return;
121 }
122 DATA_STORAGE_LOGD("InitData size = %{public}zu", vec.size());
123 for (size_t i = 0; i < vec.size(); i++) {
124 NativeRdb::ValuesBucket value;
125 util.ParserNumMatchToValuesBucket(value, vec[i]);
126 int64_t id;
127 rdbStore.Insert(id, tableName, value);
128 }
129 if (rdbStore.Commit() == NativeRdb::E_OK) {
130 util.RefreshDigest(NUM_MATCH_HASH);
131 } else {
132 util.ClearTempDigest(NUM_MATCH_HASH);
133 }
134 DATA_STORAGE_LOGD("InitData end");
135 }
136
InitData(NativeRdb::RdbStore & rdbStore,const std::string & tableName,const bool hashCheck)137 void RdbGlobalParamsCallback::InitData(NativeRdb::RdbStore &rdbStore, const std::string &tableName,
138 const bool hashCheck)
139 {
140 DATA_STORAGE_LOGD("InitData start");
141 if (tableName.compare(TABLE_NUMBER_MATCH) == 0) {
142 InitNumMatchData(rdbStore, tableName, hashCheck);
143 } else if (tableName.compare(TABLE_ECC_DATA) == 0) {
144 InitEccData(rdbStore, tableName, hashCheck);
145 } else {
146 DATA_STORAGE_LOGE("RdbGlobalParamsCallback::InitData failed: tableName %s invalid\n", tableName.c_str());
147 }
148 }
149
ClearData(NativeRdb::RdbStore & rdbStore,const std::string & tableName)150 int RdbGlobalParamsCallback::ClearData(NativeRdb::RdbStore &rdbStore, const std::string &tableName)
151 {
152 std::string sql;
153 sql.append("delete from ").append(tableName);
154 return rdbStore.ExecuteSql(sql);
155 }
156 } // namespace Telephony
157 } // namespace OHOS
158