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
16 #include "client_trans_tcp_direct_listener.h"
17
18 #include <stdbool.h>
19 #include <unistd.h>
20
21 #include "client_trans_tcp_direct_callback.h"
22 #include "client_trans_tcp_direct_manager.h"
23 #include "client_trans_tcp_direct_message.h"
24 #include "softbus_adapter_thread.h"
25 #include "softbus_base_listener.h"
26 #include "softbus_def.h"
27 #include "softbus_errcode.h"
28 #include "softbus_log.h"
29 #include "softbus_socket.h"
30 #include "softbus_type_def.h"
31 #include "trans_pending_pkt.h"
32
33 typedef struct {
34 SoftBusMutex lock;
35 bool lockInit;
36 } SoftBusTcpListenerLock;
37
38 static SoftBusTcpListenerLock g_lock = {
39 .lockInit = false,
40 };
41
42
TdcLockInit(void)43 static void TdcLockInit(void)
44 {
45 if (g_lock.lockInit == false) {
46 if (SoftBusMutexInit(&g_lock.lock, NULL) != SOFTBUS_OK) {
47 SoftBusLog(SOFTBUS_LOG_TRAN, SOFTBUS_LOG_ERROR, "TDC lock init failed");
48 return;
49 }
50 g_lock.lockInit = true;
51 }
52 return;
53 }
ClientTdcOnConnectEvent(ListenerModule module,int cfd,const ConnectOption * clientAddr)54 NO_SANITIZE("cfi") static int32_t ClientTdcOnConnectEvent(ListenerModule module, int cfd,
55 const ConnectOption *clientAddr)
56 {
57 (void)module;
58 (void)cfd;
59 (void)clientAddr;
60 return SOFTBUS_OK;
61 }
62
ClientTdcOnDataEvent(ListenerModule module,int events,int32_t fd)63 NO_SANITIZE("cfi") static int32_t ClientTdcOnDataEvent(ListenerModule module, int events, int32_t fd)
64 {
65 (void)module;
66 TcpDirectChannelInfo channel;
67 if (TransTdcGetInfoByFd(fd, &channel) == NULL) {
68 SoftBusLog(SOFTBUS_LOG_TRAN, SOFTBUS_LOG_WARN, "can not match fd.[%d]", fd);
69 return SOFTBUS_ERR;
70 }
71
72 if (events == SOFTBUS_SOCKET_IN) {
73 int32_t channelId = channel.channelId;
74 int32_t ret = TransTdcRecvData(channelId);
75 if (ret == SOFTBUS_DATA_NOT_ENOUGH) {
76 return SOFTBUS_OK;
77 }
78 if (ret != SOFTBUS_OK) {
79 SoftBusLog(SOFTBUS_LOG_TRAN, SOFTBUS_LOG_ERROR, "client cId[%d] process data fail,", channelId);
80 TransDelDataBufNode(channelId);
81 TransTdcCloseChannel(channelId);
82 ClientTransTdcOnSessionClosed(channelId);
83 return SOFTBUS_ERR;
84 }
85 }
86 return SOFTBUS_OK;
87 }
88
TransTdcCreateListener(int32_t fd)89 int32_t TransTdcCreateListener(int32_t fd)
90 {
91 static bool isInitedFlag = false;
92 TdcLockInit();
93 if (SoftBusMutexLock(&g_lock.lock) != SOFTBUS_OK) {
94 SoftBusLog(SOFTBUS_LOG_TRAN, SOFTBUS_LOG_ERROR, "TransTdcCreateListener lock failed.");
95 return SOFTBUS_ERR;
96 }
97 if (isInitedFlag == false) {
98 isInitedFlag = true;
99
100 static SoftbusBaseListener listener = {
101 .onConnectEvent = ClientTdcOnConnectEvent,
102 .onDataEvent = ClientTdcOnDataEvent,
103 };
104
105 if (StartBaseClient(DIRECT_CHANNEL_CLIENT, &listener) != SOFTBUS_OK) {
106 SoftBusLog(SOFTBUS_LOG_TRAN, SOFTBUS_LOG_ERROR, "start sdk base listener failed.");
107 SoftBusMutexUnlock(&g_lock.lock);
108 return SOFTBUS_ERR;
109 }
110 SoftBusLog(SOFTBUS_LOG_TRAN, SOFTBUS_LOG_INFO, "create sdk listener success.");
111 }
112 SoftBusMutexUnlock(&g_lock.lock);
113
114 return AddTrigger(DIRECT_CHANNEL_CLIENT, fd, READ_TRIGGER);
115 }
116
TransTdcReleaseFd(int32_t fd)117 void TransTdcReleaseFd(int32_t fd)
118 {
119 if (fd < 0) {
120 return;
121 }
122 DelTrigger(DIRECT_CHANNEL_CLIENT, fd, READ_TRIGGER);
123 ConnShutdownSocket(fd);
124 }
125
TransTdcStopRead(int32_t fd)126 int32_t TransTdcStopRead(int32_t fd)
127 {
128 if (fd < 0) {
129 return SOFTBUS_OK;
130 }
131 return DelTrigger(DIRECT_CHANNEL_CLIENT, fd, READ_TRIGGER);
132 }
133