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_errcode.h"
27 #include "softbus_log.h"
28 #include "softbus_tcp_socket.h"
29 #include "softbus_type_def.h"
30 #include "trans_pending_pkt.h"
31
32 typedef struct {
33 SoftBusMutex lock;
34 bool lockInit;
35 } SoftBusTcpListenerLock;
36 static SoftBusTcpListenerLock g_lock = {
37 .lockInit = false,
38 };
39
TdcLockInit(void)40 static void TdcLockInit(void)
41 {
42 if (g_lock.lockInit == false) {
43 if (SoftBusMutexInit(&g_lock.lock, NULL) != SOFTBUS_OK) {
44 SoftBusLog(SOFTBUS_LOG_TRAN, SOFTBUS_LOG_ERROR, "TDC lock init failed");
45 return;
46 }
47 g_lock.lockInit = true;
48 }
49 return;
50 }
OnConnectEvent(int events,int cfd,const char * ip)51 static int32_t OnConnectEvent(int events, int cfd, const char *ip)
52 {
53 (void)events;
54 (void)cfd;
55 (void)ip;
56 return SOFTBUS_OK;
57 }
58
OnDataEvent(int events,int32_t fd)59 static int32_t OnDataEvent(int events, int32_t fd)
60 {
61 TcpDirectChannelInfo channel;
62 if (TransTdcGetInfoByFd(fd, &channel) == NULL) {
63 SoftBusLog(SOFTBUS_LOG_TRAN, SOFTBUS_LOG_WARN, "can not match fd.[%d]", fd);
64 return SOFTBUS_ERR;
65 }
66
67 if (events == SOFTBUS_SOCKET_IN) {
68 int32_t channelId = channel.channelId;
69 int32_t ret = TransTdcRecvData(channelId);
70 if (ret == SOFTBUS_DATA_NOT_ENOUGH) {
71 return SOFTBUS_OK;
72 }
73 if (ret != SOFTBUS_OK) {
74 SoftBusLog(SOFTBUS_LOG_TRAN, SOFTBUS_LOG_ERROR, "client cId[%d] process data fail,", channelId);
75 TransDelDataBufNode(channelId);
76 TransTdcCloseChannel(channelId);
77 ClientTransTdcOnSessionClosed(channelId);
78 return SOFTBUS_ERR;
79 }
80 }
81 return SOFTBUS_OK;
82 }
83
84 static SoftbusBaseListener g_listener = {
85 .onConnectEvent = OnConnectEvent,
86 .onDataEvent = OnDataEvent,
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 if (SetSoftbusBaseListener(DIRECT_CHANNEL_CLIENT, &g_listener) != SOFTBUS_OK) {
101 SoftBusLog(SOFTBUS_LOG_TRAN, SOFTBUS_LOG_ERROR, "start sdk base listener failed.");
102 SoftBusMutexUnlock(&g_lock.lock);
103 return SOFTBUS_ERR;
104 }
105 if (StartBaseClient(DIRECT_CHANNEL_CLIENT) < SOFTBUS_OK) {
106 SoftBusLog(SOFTBUS_LOG_TRAN, SOFTBUS_LOG_ERROR, "client start 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 TcpShutDown(fd);
124 }
125