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 16 #ifndef NET_ACCESS_POLICY_RDB_H 17 #define NET_ACCESS_POLICY_RDB_H 18 19 #include "rdb_errno.h" 20 #include "rdb_helper.h" 21 #include "rdb_open_callback.h" 22 #include "rdb_store.h" 23 #include "result_set.h" 24 #include "rdb_sql_utils.h" 25 #include "net_access_policy.h" 26 #include "rdb_predicates.h" 27 28 namespace OHOS { 29 namespace NetManagerStandard { 30 namespace NetAccessPolicyRdbFiledConst { 31 const std::string FILED_UID = "uid"; 32 const std::string FILED_WIFI_POLICY = "wifiPolicy"; 33 const std::string FILED_CELLULAR_POLICY = "cellularPolicy"; 34 const std::string FILED_SET_FROM_CONFIG_FLAG = "setFromConfigFlag"; 35 const std::string FILED_IS_BROKER = "isBroker"; 36 37 constexpr int32_t FILED_COLUMN_INDEX_ZERO = 0; 38 constexpr int32_t FILED_COLUMN_INDEX_ONE = 1; 39 constexpr int32_t FILED_COLUMN_INDEX_TWO = 2; 40 constexpr int32_t FILED_COLUMN_INDEX_THREE = 3; 41 } 42 43 constexpr const char* DATABASE_NAME = "/data/service/el1/public/netmanager/net_uid_access_policy.db"; 44 constexpr const char* DATABASE_BACK_NAME = "/data/service/el1/public/netmanager/net_uid_access_policy_back.db"; 45 constexpr const char* POLICY_DATABASE_BACKUP_FILE = 46 "/data/service/el1/public/netmanager/net_uid_access_policy_backup.txt"; 47 48 typedef struct NetAccessPolicyData { 49 int32_t uid; 50 int32_t wifiPolicy; 51 int32_t cellularPolicy; 52 int32_t setFromConfigFlag; 53 }NetAccessPolicyData; 54 55 class NetAccessPolicyRDB { 56 public: 57 static NetAccessPolicyRDB& GetInstance(); 58 NetAccessPolicyRDB() = default; 59 ~NetAccessPolicyRDB() = default; 60 int32_t InitRdbStore(); 61 int32_t GetRdbStore(); 62 int32_t GetBackUpRdbStore(); 63 int32_t InsertData(NetAccessPolicyData policy); 64 int32_t DeleteByUid(const int32_t uid); 65 int32_t UpdateByUid(int32_t uid, NetAccessPolicyData policy); 66 int32_t BackUpPolicyDB(const std::string &sourceDB, const std::string &targetDB); 67 int32_t InitRdbStoreBackupDB(); 68 int32_t InsertDataToBackDB(NetAccessPolicyData policy); 69 int32_t DeleteByUidToBackDB(const int32_t uid); 70 int32_t UpdateByUidToBackDB(int32_t uid, NetAccessPolicyData policy); 71 IsRdbNull()72 bool IsRdbNull() 73 { 74 return rdbStore_ == nullptr; 75 } 76 std::vector<NetAccessPolicyData> QueryAll(); 77 int32_t QueryByUid(int uid, NetAccessPolicyData &uidPolicy); 78 79 class RdbDataOpenCallback : public NativeRdb::RdbOpenCallback { 80 public: 81 int32_t OnCreate(NativeRdb::RdbStore &rdbStore) override; 82 int32_t OnUpgrade(NativeRdb::RdbStore &rdbStore, int32_t oldVersion, int32_t newVersion) override; 83 private: 84 void UpgradeDbVersionTo(NativeRdb::RdbStore &store, int newVersion); 85 void AddIsBroker(NativeRdb::RdbStore &store, int newVersion); 86 }; 87 friend class RdbDataOpenCallback; 88 89 private: 90 std::shared_ptr<NativeRdb::RdbStore> rdbStore_{nullptr}; 91 std::shared_ptr<NativeRdb::RdbStore> backupRdbStore_{nullptr}; 92 }; 93 } // namespace NetManagerStandard 94 } // namespace OHOS 95 #endif // NET_ACCESS_POLICY_RDB_H 96