1 /*
2 * Copyright (c) 2023-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 #include "soft_bus_socket_listener.h"
17
18 #include "accesstoken_log.h"
19 #include "constant.h"
20 #include "remote_command_manager.h"
21 #include "socket.h"
22 #include "soft_bus_manager.h"
23
24 namespace OHOS {
25 namespace Security {
26 namespace AccessToken {
27 namespace {
28 static constexpr OHOS::HiviewDFX::HiLogLabel LABEL = {LOG_CORE, SECURITY_DOMAIN_ACCESSTOKEN, "SoftBusSocketListener"};
29 }
30 namespace {
31 static const int32_t MAX_ONBYTES_RECEIVED_DATA_LEN = 1024 * 1024 * 10;
32 } // namespace
33
34 std::mutex SoftBusSocketListener::socketMutex_;
35 std::map<int32_t, std::string> SoftBusSocketListener::socketBindMap_;
36
OnBind(int32_t socket,PeerSocketInfo info)37 void SoftBusSocketListener::OnBind(int32_t socket, PeerSocketInfo info)
38 {
39 ACCESSTOKEN_LOG_INFO(LABEL, "Socket fd is %{public}d.", socket);
40
41 if (socket <= Constant::INVALID_SOCKET_FD) {
42 ACCESSTOKEN_LOG_ERROR(LABEL, "Socket fb invalid.");
43 return;
44 }
45
46 std::string peerNetworkId(info.networkId);
47 std::lock_guard<std::mutex> guard(socketMutex_);
48 auto iter = socketBindMap_.find(socket);
49 if (iter == socketBindMap_.end()) {
50 socketBindMap_.insert(std::pair<int32_t, std::string>(socket, peerNetworkId));
51 } else {
52 iter->second = peerNetworkId;
53 }
54 }
55
OnShutdown(int32_t socket,ShutdownReason reason)56 void SoftBusSocketListener::OnShutdown(int32_t socket, ShutdownReason reason)
57 {
58 ACCESSTOKEN_LOG_INFO(LABEL, "Socket fd %{public}d shutdown because %{public}u.", socket, reason);
59
60 if (socket <= Constant::INVALID_SOCKET_FD) {
61 ACCESSTOKEN_LOG_ERROR(LABEL, "Socket fb invalid.");
62 return;
63 }
64
65 // clear sessionId state
66 std::lock_guard<std::mutex> guard(socketMutex_);
67 auto iter = socketBindMap_.find(socket);
68 if (iter != socketBindMap_.end()) {
69 socketBindMap_.erase(iter);
70 }
71 }
72
GetNetworkIdBySocket(const int32_t socket,std::string & networkId)73 bool SoftBusSocketListener::GetNetworkIdBySocket(const int32_t socket, std::string& networkId)
74 {
75 if (socket <= Constant::INVALID_SOCKET_FD) {
76 ACCESSTOKEN_LOG_ERROR(LABEL, "Socket fb invalid.");
77 return false;
78 }
79
80 std::lock_guard<std::mutex> guard(socketMutex_);
81 auto iter = socketBindMap_.find(socket);
82 if (iter != socketBindMap_.end()) {
83 networkId = iter->second;
84 return true;
85 }
86 return false;
87 }
88
OnClientBytes(int32_t socket,const void * data,uint32_t dataLen)89 void SoftBusSocketListener::OnClientBytes(int32_t socket, const void *data, uint32_t dataLen)
90 {
91 ACCESSTOKEN_LOG_INFO(LABEL, "Socket fd %{public}d, recv len %{public}d.", socket, dataLen);
92
93 if ((socket <= Constant::INVALID_SOCKET_FD) || (data == nullptr) ||
94 (dataLen == 0) || (dataLen > MAX_ONBYTES_RECEIVED_DATA_LEN)) {
95 ACCESSTOKEN_LOG_ERROR(LABEL, "Params invalid.");
96 return;
97 }
98
99 std::string networkId;
100 if (!GetNetworkIdBySocket(socket, networkId)) {
101 ACCESSTOKEN_LOG_ERROR(LABEL, "Socket invalid, bind service first.");
102 return;
103 }
104
105 // channel create in SoftBusDeviceConnectionListener::OnDeviceOnline->RemoteCommandManager::NotifyDeviceOnline
106 auto channel = RemoteCommandManager::GetInstance().GetExecutorChannel(networkId);
107 if (channel == nullptr) {
108 ACCESSTOKEN_LOG_ERROR(LABEL, "GetExecutorChannel failed");
109 return;
110 }
111 channel->HandleDataReceived(socket, static_cast<unsigned char *>(const_cast<void *>(data)), dataLen);
112 }
113
OnServiceBytes(int32_t socket,const void * data,uint32_t dataLen)114 void SoftBusSocketListener::OnServiceBytes(int32_t socket, const void *data, uint32_t dataLen)
115 {
116 ACCESSTOKEN_LOG_INFO(LABEL, "Socket fd %{public}d, recv len %{public}d.", socket, dataLen);
117
118 if ((socket <= Constant::INVALID_SOCKET_FD) || (data == nullptr) ||
119 (dataLen == 0) || (dataLen > MAX_ONBYTES_RECEIVED_DATA_LEN)) {
120 ACCESSTOKEN_LOG_ERROR(LABEL, "Params invalid.");
121 return;
122 }
123
124 std::string networkId;
125 if (SoftBusManager::GetInstance().GetNetworkIdBySocket(socket, networkId)) {
126 // channel create in SoftBusDeviceConnectionListener::OnDeviceOnline->RemoteCommandManager::NotifyDeviceOnline
127 auto channel = RemoteCommandManager::GetInstance().GetExecutorChannel(networkId);
128 if (channel == nullptr) {
129 ACCESSTOKEN_LOG_ERROR(LABEL, "GetExecutorChannel failed");
130 return;
131 }
132 channel->HandleDataReceived(socket, static_cast<unsigned char *>(const_cast<void *>(data)), dataLen);
133 } else {
134 ACCESSTOKEN_LOG_ERROR(LABEL, "Unkonow socket.");
135 }
136 }
137
CleanUpAllBindSocket()138 void SoftBusSocketListener::CleanUpAllBindSocket()
139 {
140 std::lock_guard<std::mutex> guard(socketMutex_);
141 for (auto it = socketBindMap_.begin(); it != socketBindMap_.end();) {
142 ::Shutdown(it->first);
143 it = socketBindMap_.erase(it);
144 }
145 }
146 } // namespace AccessToken
147 } // namespace Security
148 } // namespace OHOS
149