1 /*
2 * Copyright (c) 2021 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 #ifdef RELATIONAL_STORE
16 #include "runtime_config.h"
17
18 #include "db_constant.h"
19 #include "db_dfx_adapter.h"
20 #include "kvdb_manager.h"
21 #include "kv_store_errno.h"
22 #include "log_print.h"
23 #include "network_adapter.h"
24 #include "runtime_context.h"
25
26 namespace DistributedDB {
27 std::mutex RuntimeConfig::communicatorMutex_;
28 std::mutex RuntimeConfig::multiUserMutex_;
29 std::shared_ptr<IProcessCommunicator> RuntimeConfig::processCommunicator_ = nullptr;
30
31 // Used to set the process userid and appId
SetProcessLabel(const std::string & appId,const std::string & userId)32 DBStatus RuntimeConfig::SetProcessLabel(const std::string &appId, const std::string &userId)
33 {
34 if (appId.size() > DBConstant::MAX_APP_ID_LENGTH || appId.empty() ||
35 userId.size() > DBConstant::MAX_USER_ID_LENGTH || userId.empty()) {
36 LOGE("Invalid app or user info[%zu]-[%zu]", appId.length(), userId.length());
37 return INVALID_ARGS;
38 }
39
40 int errCode = KvDBManager::SetProcessLabel(appId, userId);
41 if (errCode != E_OK) {
42 LOGE("Failed to set the process label:%d", errCode);
43 return DB_ERROR;
44 }
45 return OK;
46 }
47
48 // Set process communicator.
SetProcessCommunicator(const std::shared_ptr<IProcessCommunicator> & inCommunicator)49 DBStatus RuntimeConfig::SetProcessCommunicator(const std::shared_ptr<IProcessCommunicator> &inCommunicator)
50 {
51 std::lock_guard<std::mutex> lock(communicatorMutex_);
52 if (processCommunicator_ != nullptr) {
53 LOGE("processCommunicator_ is not null!");
54 return DB_ERROR;
55 }
56
57 std::string processLabel = RuntimeContext::GetInstance()->GetProcessLabel();
58 if (processLabel.empty()) {
59 LOGE("ProcessLabel is not set!");
60 return DB_ERROR;
61 }
62
63 auto *adapter = new (std::nothrow) NetworkAdapter(processLabel, inCommunicator);
64 if (adapter == nullptr) {
65 LOGE("New NetworkAdapter failed!");
66 return DB_ERROR;
67 }
68 processCommunicator_ = inCommunicator;
69 if (RuntimeContext::GetInstance()->SetCommunicatorAdapter(adapter) != E_OK) {
70 LOGE("SetProcessCommunicator not support!");
71 delete adapter;
72 return DB_ERROR;
73 }
74 KvDBManager::RestoreSyncableKvStore();
75 return OK;
76 }
77
SetPermissionCheckCallback(const PermissionCheckCallbackV2 & callback)78 DBStatus RuntimeConfig::SetPermissionCheckCallback(const PermissionCheckCallbackV2 &callback)
79 {
80 int errCode = RuntimeContext::GetInstance()->SetPermissionCheckCallback(callback);
81 return TransferDBErrno(errCode);
82 }
83
SetPermissionCheckCallback(const PermissionCheckCallbackV3 & callback)84 DBStatus RuntimeConfig::SetPermissionCheckCallback(const PermissionCheckCallbackV3 &callback)
85 {
86 int errCode = RuntimeContext::GetInstance()->SetPermissionCheckCallback(callback);
87 return TransferDBErrno(errCode);
88 }
89
SetProcessSystemAPIAdapter(const std::shared_ptr<IProcessSystemApiAdapter> & adapter)90 DBStatus RuntimeConfig::SetProcessSystemAPIAdapter(const std::shared_ptr<IProcessSystemApiAdapter> &adapter)
91 {
92 return TransferDBErrno(RuntimeContext::GetInstance()->SetProcessSystemApiAdapter(adapter));
93 }
94
Dump(int fd,const std::vector<std::u16string> & args)95 void RuntimeConfig::Dump(int fd, const std::vector<std::u16string> &args)
96 {
97 DBDfxAdapter::Dump(fd, args);
98 }
99
SetSyncActivationCheckCallback(const SyncActivationCheckCallback & callback)100 DBStatus RuntimeConfig::SetSyncActivationCheckCallback(const SyncActivationCheckCallback &callback)
101 {
102 std::lock_guard<std::mutex> lock(multiUserMutex_);
103 int errCode = RuntimeContext::GetInstance()->SetSyncActivationCheckCallback(callback);
104 return TransferDBErrno(errCode);
105 }
106
NotifyUserChanged()107 DBStatus RuntimeConfig::NotifyUserChanged()
108 {
109 std::lock_guard<std::mutex> lock(multiUserMutex_);
110 int errCode = RuntimeContext::GetInstance()->NotifyUserChanged();
111 return TransferDBErrno(errCode);
112 }
113
IsProcessSystemApiAdapterValid()114 bool RuntimeConfig::IsProcessSystemApiAdapterValid()
115 {
116 return RuntimeContext::GetInstance()->IsProcessSystemApiAdapterValid();
117 }
118
SetSyncActivationCheckCallback(const SyncActivationCheckCallbackV2 & callback)119 DBStatus RuntimeConfig::SetSyncActivationCheckCallback(const SyncActivationCheckCallbackV2 &callback)
120 {
121 std::lock_guard<std::mutex> lock(multiUserMutex_);
122 int errCode = RuntimeContext::GetInstance()->SetSyncActivationCheckCallback(callback);
123 return TransferDBErrno(errCode);
124 }
125
SetPermissionConditionCallback(const PermissionConditionCallback & callback)126 DBStatus RuntimeConfig::SetPermissionConditionCallback(const PermissionConditionCallback &callback)
127 {
128 int errCode = RuntimeContext::GetInstance()->SetPermissionConditionCallback(callback);
129 return TransferDBErrno(errCode);
130 }
131
SetTranslateToDeviceIdCallback(const DistributedDB::TranslateToDeviceIdCallback & callback)132 void RuntimeConfig::SetTranslateToDeviceIdCallback(const DistributedDB::TranslateToDeviceIdCallback &callback)
133 {
134 RuntimeContext::GetInstance()->SetTranslateToDeviceIdCallback(callback);
135 }
136 } // namespace DistributedDB
137 #endif