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_channel_manager.h"
17 
18 #include "client_trans_auth_manager.h"
19 #include "client_trans_proxy_manager.h"
20 #include "client_trans_session_callback.h"
21 #include "client_trans_statistics.h"
22 #include "client_trans_tcp_direct_manager.h"
23 #include "client_trans_tcp_direct_message.h"
24 #include "client_trans_udp_manager.h"
25 #include "softbus_errcode.h"
26 #include "trans_log.h"
27 
ClientTransChannelInit(void)28 int32_t ClientTransChannelInit(void)
29 {
30     IClientSessionCallBack *cb = GetClientSessionCb();
31     TRANS_CHECK_AND_RETURN_RET_LOGE(cb != NULL, SOFTBUS_NO_INIT, TRANS_SDK, "get client session Cb failed.");
32 
33     int32_t ret = TransTdcManagerInit(cb);
34     TRANS_CHECK_AND_RETURN_RET_LOGE(ret == SOFTBUS_OK, ret, TRANS_SDK, "trans tcp manager init failed.");
35 
36     ret = ClientTransAuthInit(cb);
37     TRANS_CHECK_AND_RETURN_RET_LOGE(ret == SOFTBUS_OK, ret, TRANS_SDK, "client trans auth init failed.");
38 
39     ret = ClientTransStatisticsInit();
40     TRANS_CHECK_AND_RETURN_RET_LOGE(ret == SOFTBUS_OK, ret, TRANS_SDK, "client trans statistics init failed.");
41 
42     ret = ClientTransProxyInit(cb);
43     TRANS_CHECK_AND_RETURN_RET_LOGE(ret == SOFTBUS_OK, ret, TRANS_SDK, "client trans proxy init failed.");
44 
45     ret = ClientTransUdpMgrInit(cb);
46     TRANS_CHECK_AND_RETURN_RET_LOGE(ret == SOFTBUS_OK, ret, TRANS_SDK, "client trans udp mgr init failed.");
47     return SOFTBUS_OK;
48 }
49 
ClientTransChannelDeinit(void)50 void ClientTransChannelDeinit(void)
51 {
52     TransTdcManagerDeinit();
53     ClientTransUdpMgrDeinit();
54     ClientTransProxyDeinit();
55     ClientTransStatisticsDeinit();
56 }
57 
ClientTransCloseChannel(int32_t channelId,int32_t type)58 int32_t ClientTransCloseChannel(int32_t channelId, int32_t type)
59 {
60     if (channelId < 0) {
61         TRANS_LOGW(TRANS_SDK, "Invalid param");
62         return SOFTBUS_INVALID_PARAM;
63     }
64     DeleteSocketResourceByChannelId(channelId, type);
65     int32_t ret = SOFTBUS_OK;
66     switch (type) {
67         case CHANNEL_TYPE_PROXY:
68             ClientTransProxyCloseChannel(channelId);
69             break;
70         case CHANNEL_TYPE_TCP_DIRECT:
71             TransDelDataBufNode(channelId);
72             TransTdcCloseChannel(channelId);
73             break;
74         case CHANNEL_TYPE_UDP:
75             ret = ClientTransCloseUdpChannel(channelId, SHUTDOWN_REASON_LOCAL);
76             break;
77         case CHANNEL_TYPE_AUTH:
78             ClientTransAuthCloseChannel(channelId, SHUTDOWN_REASON_LOCAL);
79             break;
80         default:
81             TRANS_LOGE(TRANS_SDK, "Invalid type");
82             return SOFTBUS_TRANS_INVALID_CHANNEL_TYPE;
83     }
84     return ret;
85 }
86 
ClientTransChannelSendBytes(int32_t channelId,int32_t channelType,const void * data,uint32_t len)87 int32_t ClientTransChannelSendBytes(int32_t channelId, int32_t channelType, const void *data, uint32_t len)
88 {
89     if ((data == NULL) || (len == 0)) {
90         TRANS_LOGW(TRANS_BYTES, "Invalid param");
91         return SOFTBUS_INVALID_PARAM;
92     }
93 
94     int32_t ret = SOFTBUS_OK;
95     switch (channelType) {
96         case CHANNEL_TYPE_AUTH:
97             ret = TransAuthChannelSendBytes(channelId, data, len);
98             break;
99         case CHANNEL_TYPE_PROXY:
100             ret = TransProxyChannelSendBytes(channelId, data, len);
101             break;
102         case CHANNEL_TYPE_TCP_DIRECT:
103             ret = TransTdcSendBytes(channelId, data, len);
104             break;
105         default:
106             TRANS_LOGE(TRANS_SDK, "Invalid channelType=%{public}d", channelType);
107             return SOFTBUS_TRANS_INVALID_CHANNEL_TYPE;
108     }
109     return ret;
110 }
111 
ClientTransChannelSendMessage(int32_t channelId,int32_t channelType,const void * data,uint32_t len)112 int32_t ClientTransChannelSendMessage(int32_t channelId, int32_t channelType, const void *data, uint32_t len)
113 {
114     if ((data == NULL) || (len == 0)) {
115         TRANS_LOGW(TRANS_MSG, "Invalid param.");
116         return SOFTBUS_INVALID_PARAM;
117     }
118 
119     int32_t ret = SOFTBUS_OK;
120     switch (channelType) {
121         case CHANNEL_TYPE_AUTH:
122             ret = TransAuthChannelSendMessage(channelId, data, len);
123             break;
124         case CHANNEL_TYPE_PROXY:
125             ret = TransProxyChannelSendMessage(channelId, data, len);
126             break;
127         case CHANNEL_TYPE_TCP_DIRECT:
128             ret = TransTdcSendMessage(channelId, data, len);
129             break;
130         default:
131             TRANS_LOGE(TRANS_MSG, "Invalid channelType=%{public}d", channelType);
132             return SOFTBUS_TRANS_CHANNEL_TYPE_INVALID;
133     }
134     return ret;
135 }
136 
ClientTransChannelSendStream(int32_t channelId,int32_t channelType,const StreamData * data,const StreamData * ext,const StreamFrameInfo * param)137 int32_t ClientTransChannelSendStream(int32_t channelId, int32_t channelType, const StreamData *data,
138     const StreamData *ext, const StreamFrameInfo *param)
139 {
140     if ((data == NULL) || (ext == NULL) || (param == NULL)) {
141         TRANS_LOGW(TRANS_STREAM, "Invalid param");
142         return SOFTBUS_INVALID_PARAM;
143     }
144 
145     int32_t ret = SOFTBUS_OK;
146     switch (channelType) {
147         case CHANNEL_TYPE_UDP:
148             ret = TransUdpChannelSendStream(channelId, data, ext, param);
149             break;
150         default:
151             TRANS_LOGE(TRANS_STREAM, "unsupport channelType=%{public}d.", channelType);
152             return SOFTBUS_TRANS_CHANNEL_TYPE_INVALID;
153     }
154     return ret;
155 }
156 
ClientTransChannelSendFile(int32_t channelId,int32_t channelType,const char * sFileList[],const char * dFileList[],uint32_t fileCnt)157 int32_t ClientTransChannelSendFile(int32_t channelId, int32_t channelType, const char *sFileList[],
158     const char *dFileList[], uint32_t fileCnt)
159 {
160     int32_t ret = SOFTBUS_OK;
161     switch (channelType) {
162         case CHANNEL_TYPE_UDP:
163             ret = TransUdpChannelSendFile(channelId, sFileList, dFileList, fileCnt);
164             break;
165         case CHANNEL_TYPE_PROXY:
166             ret = TransProxyChannelSendFile(channelId, sFileList, dFileList, fileCnt);
167             break;
168         default:
169             TRANS_LOGE(TRANS_FILE, "unsupport channelType=%{public}d.", channelType);
170             return SOFTBUS_TRANS_CHANNEL_TYPE_INVALID;
171     }
172     return ret;
173 }
174 
DeleteFileListener(const char * sessionName)175 void DeleteFileListener(const char *sessionName)
176 {
177     TransUdpDeleteFileListener(sessionName);
178 }
179 
ClientGetSessionKey(int32_t channelId,char * key,unsigned int len)180 int32_t ClientGetSessionKey(int32_t channelId, char *key, unsigned int len)
181 {
182     return TransTdcGetSessionKey(channelId, key, len);
183 }
184 
ClientGetHandle(int32_t channelId,int * handle)185 int32_t ClientGetHandle(int32_t channelId, int *handle)
186 {
187     return TransTdcGetHandle(channelId, handle);
188 }
189 
ClientDisableSessionListener(int32_t channelId)190 int32_t ClientDisableSessionListener(int32_t channelId)
191 {
192     return TransDisableSessionListener(channelId);
193 }