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 "nstackx_dfile_session.h"
17
18 #include "nstackx_congestion.h"
19 #include "nstackx_dfile.h"
20 #include "nstackx_dfile_config.h"
21 #include "nstackx_dfile_frame.h"
22 #include "nstackx_dfile_send.h"
23 #include "nstackx_epoll.h"
24 #include "nstackx_error.h"
25 #include "nstackx_event.h"
26 #include "nstackx_dfile_log.h"
27 #ifdef MBEDTLS_INCLUDED
28 #include "nstackx_mbedtls.h"
29 #else
30 #include "nstackx_openssl.h"
31 #endif
32 #include "nstackx_timer.h"
33 #include "nstackx_dfile_control.h"
34 #include "nstackx_dfile_mp.h"
35 #include "securec.h"
36 #include "nstackx_util.h"
37 #include "nstackx_dfile_dfx.h"
38
39 #define TAG "nStackXDFile"
40
41 #define DEFAULT_WAIT_TIME_MS 1000
42 #define MULTI_THREADS_SOCKET_WAIT_TIME_MS 1
43 #define DEFAULT_NEGOTIATE_TIMEOUT 1000
44 #define TCP_NEGOTIATE_TIMEOUT 10000
45 #define MAX_SERVER_NEGOTIATE_VALID_TIMEOUT (600 * DEFAULT_NEGOTIATE_TIMEOUT)
46 #define MAX_NEGOTIATE_TIMEOUT_COUNT 3
47 #define MAX_UNPROCESSED_READ_EVENT_COUNT 3
48 #define MAX_RECVBUF_COUNT 130000
49 #define MAX_NOMEM_PRINT 10000
50
51 static void ReadEventHandle(void *arg);
52 static void ProcessSessionTrans(const DFileSession *session, uint16_t exceptTransId);
53 DFileEventFunc g_dfileEventFunc;
54
CreateQueueNode(const uint8_t * frame,size_t length,const struct sockaddr_in * peerAddr,uint8_t socketIndex)55 static QueueNode *CreateQueueNode(const uint8_t *frame, size_t length,
56 const struct sockaddr_in *peerAddr, uint8_t socketIndex)
57 {
58 QueueNode *queueNode = NULL;
59
60 if (frame == NULL || length == 0 || length > NSTACKX_MAX_FRAME_SIZE) {
61 return NULL;
62 }
63
64 queueNode = calloc(1, sizeof(QueueNode));
65 if (queueNode == NULL) {
66 return NULL;
67 }
68
69 queueNode->frame = calloc(1, length);
70 if (queueNode->frame == NULL) {
71 free(queueNode);
72 return NULL;
73 }
74 queueNode->length = length;
75 queueNode->sendLen = 0;
76
77 if (memcpy_s(queueNode->frame, length, frame, length) != EOK) {
78 DestroyQueueNode(queueNode);
79 return NULL;
80 }
81 if (peerAddr != NULL) {
82 if (memcpy_s(&queueNode->peerAddr, sizeof(struct sockaddr_in), peerAddr, sizeof(struct sockaddr_in)) != EOK) {
83 DestroyQueueNode(queueNode);
84 return NULL;
85 }
86 }
87 queueNode->socketIndex = socketIndex;
88
89 return queueNode;
90 }
91
DestroyQueueNode(QueueNode * queueNode)92 void DestroyQueueNode(QueueNode *queueNode)
93 {
94 if (queueNode != NULL) {
95 free(queueNode->frame);
96 free(queueNode);
97 }
98 }
99
NotifyMsgRecver(const DFileSession * session,DFileMsgType msgType,const DFileMsg * msg)100 void NotifyMsgRecver(const DFileSession *session, DFileMsgType msgType, const DFileMsg *msg)
101 {
102 if (session == NULL) {
103 DFILE_LOGI(TAG, "session is NULL");
104 return;
105 }
106
107 if (session->msgReceiver == NULL) {
108 DFILE_LOGI(TAG, "msgReceiver is NULL");
109 return;
110 }
111
112 session->msgReceiver(session->sessionId, msgType, msg);
113 }
114
CalculateSessionTransferRatePrepare(DFileSession * session)115 void CalculateSessionTransferRatePrepare(DFileSession *session)
116 {
117 #ifdef NSTACKX_SMALL_FILE_SUPPORT
118 if (session->sessionType == DFILE_SESSION_TYPE_CLIENT && (!ListIsEmpty(&session->pendingFileLists)
119 || !ListIsEmpty(&session->smallFileLists))) {
120 return;
121 }
122 #else
123 if (session->sessionType == DFILE_SESSION_TYPE_CLIENT && !ListIsEmpty(&session->pendingFileLists)) {
124 return;
125 }
126 #endif
127 if (!ListIsEmpty(&session->dFileTransChain)) {
128 return;
129 }
130 DFILE_LOGI(TAG, "begin to calculate transfer rate");
131 session->bytesTransferred = 0;
132 session->transCount = 0;
133 ClockGetTime(CLOCK_MONOTONIC, &session->startTs);
134 }
135
136 #ifdef NSTACKX_SMALL_FILE_SUPPORT
SendSmallList(DFileSession * session)137 static int32_t SendSmallList(DFileSession *session)
138 {
139 FileListInfo *fileListInfo = NULL;
140 DFileMsg data;
141 if (session->fileListProcessingCnt > 0 || session->fileListPendingCnt == 0) {
142 while (!ListIsEmpty(&session->smallFileLists)) {
143 fileListInfo = (FileListInfo *)ListPopFront(&session->smallFileLists);
144 session->smallListPendingCnt--;
145 if (fileListInfo == NULL) {
146 continue;
147 }
148 int32_t ret = DFileStartTrans(session, fileListInfo);
149 if (ret == NSTACKX_EOK) {
150 return NSTACKX_TRUE;
151 }
152 DFILE_LOGE(TAG, "DFileStartTrans fail, error: %d", ret);
153 (void)memset_s(&data, sizeof(data), 0, sizeof(data));
154 data.errorCode = ret;
155 data.fileList.files = (const char **)fileListInfo->files;
156 data.fileList.fileNum = fileListInfo->fileNum;
157 data.fileList.userData = fileListInfo->userData;
158 NotifyMsgRecver(session, DFILE_ON_FILE_SEND_FAIL, &data);
159 DestroyFileListInfo(fileListInfo);
160 }
161 }
162 return NSTACKX_FALSE;
163 }
164 #endif
165
SendPendingList(DFileSession * session)166 static void SendPendingList(DFileSession *session)
167 {
168 FileListInfo *fileListInfo = NULL;
169 DFileMsg data;
170 while (!ListIsEmpty(&session->pendingFileLists)) {
171 fileListInfo = (FileListInfo *)ListPopFront(&session->pendingFileLists);
172 session->fileListPendingCnt--;
173 if (fileListInfo == NULL) {
174 continue;
175 }
176 int32_t ret = DFileStartTrans(session, fileListInfo);
177 if (ret == NSTACKX_EOK) {
178 break;
179 }
180 DFILE_LOGE(TAG, "DFileStartTrans fail, error: %d", ret);
181 (void)memset_s(&data, sizeof(data), 0, sizeof(data));
182 data.errorCode = ret;
183 data.fileList.files = (const char **)fileListInfo->files;
184 data.fileList.fileNum = fileListInfo->fileNum;
185 data.fileList.userData = fileListInfo->userData;
186 NotifyMsgRecver(session, DFILE_ON_FILE_SEND_FAIL, &data);
187 DestroyFileListInfo(fileListInfo);
188 }
189 }
190
SendSmallAndPendingList(DFileSession * session)191 static void SendSmallAndPendingList(DFileSession *session)
192 {
193 #ifdef NSTACKX_SMALL_FILE_SUPPORT
194 DFILE_LOGI(TAG, "fileListPendingCnt %u fileListProcessingCnt %u smallListPendingCnt %u smallListProcessingCnt %u",
195 session->fileListPendingCnt, session->fileListProcessingCnt, session->smallListPendingCnt,
196 session->smallListProcessingCnt);
197 if (SendSmallList(session) != NSTACKX_TRUE) {
198 SendPendingList(session);
199 }
200 #else
201 DFILE_LOGI(TAG, "fileListPendingCnt %u fileListProcessingCnt %u",
202 session->fileListPendingCnt, session->fileListProcessingCnt);
203 SendPendingList(session);
204 #endif
205 }
206
NoticeSessionProgress(DFileSession * session)207 void NoticeSessionProgress(DFileSession *session)
208 {
209 DFileMsg data;
210 (void)memset_s(&data, sizeof(data), 0, sizeof(data));
211 if ((FileManagerGetTotalBytes(session->fileManager, &data.transferUpdate.totalBytes) == NSTACKX_EOK) &&
212 (FileManagerGetBytesTransferred(session->fileManager, &data.transferUpdate.bytesTransferred) == NSTACKX_EOK) &&
213 (data.transferUpdate.bytesTransferred <= data.transferUpdate.totalBytes) &&
214 (data.transferUpdate.bytesTransferred > 0)) {
215 NotifyMsgRecver(session, DFILE_ON_SESSION_IN_PROGRESS, &data);
216 }
217 }
218
UpdateMsgProcessInfo(const DFileSession * session,struct DFileTrans * dFileTrans,DFileTransMsgType msgType,DFileTransMsg * msg)219 static void UpdateMsgProcessInfo(const DFileSession *session, struct DFileTrans *dFileTrans,
220 DFileTransMsgType msgType, DFileTransMsg *msg)
221 {
222 uint64_t totalBytes = 0;
223 uint64_t bytesTrans = 0;
224
225 if (session == NULL || dFileTrans == NULL || msg == NULL ||
226 (msgType != DFILE_TRANS_MSG_FILE_RECEIVED && msgType != DFILE_TRANS_MSG_FILE_RECEIVE_FAIL &&
227 msgType != DFILE_TRANS_MSG_FILE_SEND_FAIL && msgType != DFILE_TRANS_MSG_FILE_SENT)) {
228 return;
229 }
230
231 msg->transferUpdate.transId = dFileTrans->transId;
232
233 if (msgType == DFILE_TRANS_MSG_FILE_RECEIVED || msgType == DFILE_TRANS_MSG_FILE_SENT) {
234 totalBytes = DFileTransGetTotalBytes(dFileTrans);
235 if (totalBytes > 0) {
236 bytesTrans = totalBytes;
237 goto L_END;
238 }
239 }
240
241 if (FileManagerGetTransUpdateInfo(session->fileManager, dFileTrans->transId, &totalBytes, &bytesTrans) !=
242 NSTACKX_EOK) {
243 return;
244 }
245
246 L_END:
247 msg->transferUpdate.totalBytes = totalBytes;
248 msg->transferUpdate.bytesTransferred = bytesTrans;
249 if (msgType == DFILE_TRANS_MSG_FILE_SENT) {
250 if (dFileTrans->fileList->vtransFlag) {
251 msg->fileList.transId = dFileTrans->fileList->vtransRealTransId;
252 msg->transferUpdate.transId = dFileTrans->fileList->vtransRealTransId;
253 }
254 NotifyMsgRecver(session, DFILE_ON_TRANS_IN_PROGRESS, msg);
255 }
256 }
257
WakeSendThread(DFileSession * session,uint8_t isSender,uint8_t socketIndex)258 static void WakeSendThread(DFileSession *session, uint8_t isSender, uint8_t socketIndex)
259 {
260 SemPost(&session->outboundQueueWait[socketIndex]);
261 if (isSender && session->clientSendThreadNum > 1) {
262 for (uint16_t i = 0; i < session->clientSendThreadNum - 1; i++) {
263 SemPost(&session->sendThreadPara[i].sendWait);
264 }
265 }
266 }
267
CalculateSessionTransferRate(DFileSession * session,uint64_t totalBytes,DFileTransMsgType msgType)268 static void CalculateSessionTransferRate(DFileSession *session, uint64_t totalBytes, DFileTransMsgType msgType)
269 {
270 if (msgType != DFILE_TRANS_MSG_FILE_SENT && msgType != DFILE_TRANS_MSG_END) {
271 return;
272 }
273 if (totalBytes <= UINT64_MAX - session->bytesTransferred) {
274 session->bytesTransferred += totalBytes;
275 } else {
276 session->bytesTransferred = UINT64_MAX;
277 }
278 session->transCount++;
279 if (!ListIsEmpty(&session->dFileTransChain)) {
280 return;
281 }
282 #ifdef NSTACKX_SMALL_FILE_SUPPORT
283 if (session->sessionType == DFILE_SESSION_TYPE_CLIENT && (!ListIsEmpty(&session->pendingFileLists)
284 || !ListIsEmpty(&session->smallFileLists))) {
285 return;
286 }
287 #else
288 if (session->sessionType == DFILE_SESSION_TYPE_CLIENT && !ListIsEmpty(&session->pendingFileLists)) {
289 return;
290 }
291 #endif
292 struct timespec endTs;
293 ClockGetTime(CLOCK_MONOTONIC, &endTs);
294
295 uint32_t spendTime = GetTimeDiffMs(&endTs, &session->startTs);
296 if (spendTime == 0) {
297 return;
298 }
299 const double rate = 1.0 * session->bytesTransferred / DFILE_MEGABYTES * MSEC_TICKS_PER_SEC / spendTime;
300 DFILE_LOGI(TAG, "Total %u trans, %llu bytes, used %u ms. rate %.2f MB/s",
301 session->transCount, session->bytesTransferred, spendTime, rate);
302 DFileMsg msgData;
303 (void)memset_s(&msgData, sizeof(msgData), 0, sizeof(msgData));
304 msgData.rate = (uint32_t)rate;
305 NotifyMsgRecver(session, DFILE_ON_SESSION_TRANSFER_RATE, &msgData);
306
307 TransferCompleteEvent(rate);
308 }
309
CheckTransDone(DFileSession * session,struct DFileTrans * dFileTrans,DFileTransMsgType msgType)310 static void CheckTransDone(DFileSession *session, struct DFileTrans *dFileTrans, DFileTransMsgType msgType)
311 {
312 if (msgType == DFILE_TRANS_MSG_FILE_RECEIVE_FAIL || msgType == DFILE_TRANS_MSG_FILE_SENT ||
313 msgType == DFILE_TRANS_MSG_FILE_SEND_FAIL || msgType == DFILE_TRANS_MSG_END) {
314 uint8_t flag = dFileTrans->fileList->smallFlag;
315 if (SetTransIdState(session, dFileTrans->transId, STATE_TRANS_DONE) != NSTACKX_EOK) {
316 DFILE_LOGE(TAG, "set trans id state fail");
317 }
318 if (((PeerInfo *)dFileTrans->context)->currentTransCount > 0) {
319 ((PeerInfo *)dFileTrans->context)->currentTransCount--;
320 }
321 ListRemoveNode(&dFileTrans->list);
322 uint64_t totalBytes = DFileTransGetTotalBytes(dFileTrans);
323 DFileTransDestroy(dFileTrans);
324 if (session->sessionType == DFILE_SESSION_TYPE_CLIENT) {
325 if (flag == NSTACKX_TRUE && session->smallListProcessingCnt > 0) {
326 session->smallListProcessingCnt--;
327 } else if (session->fileListProcessingCnt > 0) {
328 session->fileListProcessingCnt--;
329 }
330 SendSmallAndPendingList(session);
331 }
332 CalculateSessionTransferRate(session, totalBytes, msgType);
333 }
334 }
335
DTransMsgReceiver(struct DFileTrans * dFileTrans,DFileTransMsgType msgType,DFileTransMsg * msg)336 static void DTransMsgReceiver(struct DFileTrans *dFileTrans, DFileTransMsgType msgType,
337 DFileTransMsg *msg)
338 {
339 PeerInfo *peerInfo = dFileTrans->context;
340 DFileSession *session = peerInfo->session;
341
342 UpdateMsgProcessInfo(session, dFileTrans, msgType, msg);
343
344 switch (msgType) {
345 case DFILE_TRANS_MSG_FILE_SEND_DATA:
346 WakeSendThread(session, dFileTrans->isSender, peerInfo->socketIndex);
347 break;
348 case DFILE_TRANS_MSG_FILE_LIST_RECEIVED:
349 NotifyMsgRecver(session, DFILE_ON_FILE_LIST_RECEIVED, msg);
350 break;
351 case DFILE_TRANS_MSG_FILE_RECEIVED: /* Receiver receive all the file data */
352 NotifyMsgRecver(session, DFILE_ON_FILE_RECEIVE_SUCCESS, msg);
353 break;
354 case DFILE_TRANS_MSG_FILE_RECEIVED_TO_FAIL:
355 case DFILE_TRANS_MSG_FILE_RECEIVE_FAIL:
356 NotifyMsgRecver(session, DFILE_ON_FILE_RECEIVE_FAIL, msg);
357 break;
358 case DFILE_TRANS_MSG_FILE_SENT: /* Sender send TRANSFER DONE ACK frame and come to end */
359 NoticeSessionProgress(session);
360 NotifyMsgRecver(session, DFILE_ON_FILE_SEND_SUCCESS, msg);
361 break;
362 case DFILE_TRANS_MSG_FILE_SEND_FAIL:
363 NotifyMsgRecver(session, DFILE_ON_FILE_SEND_FAIL, msg);
364 break;
365 case DFILE_TRANS_MSG_IN_PROGRESS:
366 NotifyMsgRecver(session, DFILE_ON_TRANS_IN_PROGRESS, msg);
367 break;
368 case DFILE_TRANS_MSG_FILE_SEND_ACK:
369 ProcessSessionTrans(session, dFileTrans->transId);
370 break;
371 default:
372 DFILE_LOGI(TAG, "transId %u, recv DFileTrans event %d", dFileTrans->transId, msgType);
373 }
374
375 CheckTransDone(session, dFileTrans, msgType);
376 }
377
ServerSettingTimeoutHandle(void * data)378 static void ServerSettingTimeoutHandle(void *data)
379 {
380 PeerInfo *peerInfo = data;
381 ListRemoveNode(&peerInfo->list);
382 TimerDelete(peerInfo->settingTimer);
383 peerInfo->session->peerInfoCnt--;
384 free(peerInfo);
385 DFILE_LOGI(TAG, "DFileServer Setting Negotationion timeout");
386 }
387
ClientSettingTimeoutHandle(void * data)388 static void ClientSettingTimeoutHandle(void *data)
389 {
390 PeerInfo *peerInfo = data;
391 uint8_t cnt = peerInfo->settingTimeoutCnt++;
392 DFileMsg msgData;
393 uint32_t timeout = CapsTcp(peerInfo->session) ? TCP_NEGOTIATE_TIMEOUT : DEFAULT_NEGOTIATE_TIMEOUT;
394 (void)memset_s(&msgData, sizeof(msgData), 0, sizeof(msgData));
395 if (cnt >= MAX_NEGOTIATE_TIMEOUT_COUNT) {
396 TimerDelete(peerInfo->settingTimer);
397 peerInfo->settingTimer = NULL;
398 peerInfo->settingTimeoutCnt = 0;
399 msgData.errorCode = NSTACKX_EFAILED;
400 DFILE_LOGI(TAG, "DFileClient Setting Negotationion timeout");
401 NotifyMsgRecver(peerInfo->session, DFILE_ON_CONNECT_FAIL, &msgData);
402 } else {
403 DFileSessionSendSetting(peerInfo);
404 DFILE_LOGI(TAG, "Client Setting Negotationion timeout %u times", peerInfo->settingTimeoutCnt);
405 if (TimerSetTimeout(peerInfo->settingTimer, timeout, NSTACKX_FALSE) != NSTACKX_EOK) {
406 msgData.errorCode = NSTACKX_EFAILED;
407 NotifyMsgRecver(peerInfo->session, DFILE_ON_CONNECT_FAIL, &msgData);
408 DFILE_LOGE(TAG, "Timer setting timer fail");
409 }
410 }
411 }
412
SearchDFileTransNode(List * dFileTransChain,uint16_t transId)413 static DFileTrans *SearchDFileTransNode(List *dFileTransChain, uint16_t transId)
414 {
415 List *pos = NULL;
416 DFileTrans *trans = NULL;
417
418 if (dFileTransChain == NULL || transId == 0) {
419 return NULL;
420 }
421
422 LIST_FOR_EACH(pos, dFileTransChain) {
423 trans = (DFileTrans *)pos;
424 if (trans->transId == transId) {
425 return trans;
426 }
427 }
428 return NULL;
429 }
430
SearchPeerInfoNode(const DFileSession * session,const struct sockaddr_in * peerAddr)431 static PeerInfo *SearchPeerInfoNode(const DFileSession *session, const struct sockaddr_in *peerAddr)
432 {
433 List *pos = NULL;
434 PeerInfo *peerInfo = NULL;
435 LIST_FOR_EACH(pos, &session->peerInfoChain) {
436 peerInfo = (PeerInfo *)pos;
437 if (memcmp(&peerInfo->dstAddr, peerAddr, sizeof(struct sockaddr_in)) == 0 &&
438 peerInfo->session->sessionId == session->sessionId) {
439 return peerInfo;
440 }
441 }
442 return NULL;
443 }
444
DFileSessionSendSetting(PeerInfo * peerInfo)445 void DFileSessionSendSetting(PeerInfo *peerInfo)
446 {
447 uint32_t timeout = CapsTcp(peerInfo->session) ? TCP_NEGOTIATE_TIMEOUT : DEFAULT_NEGOTIATE_TIMEOUT;
448 uint8_t buf[NSTACKX_DEFAULT_FRAME_SIZE];
449 size_t frameLen = 0;
450 DFileMsg data;
451 (void)memset_s(&data, sizeof(data), 0, sizeof(data));
452 SettingFrame settingFramePara;
453 settingFramePara.connType = peerInfo->connType;
454 settingFramePara.mtu = peerInfo->localMtu;
455 settingFramePara.capability = peerInfo->session->capability & NSTACKX_CAPS_LINK_SEQUENCE;
456 settingFramePara.dataFrameSize = 0;
457 settingFramePara.capsCheck = NSTACKX_INTERNAL_CAPS_RECV_FEEDBACK;
458 if (peerInfo->session->fileManager->keyLen) {
459 DFileGetCipherCaps(peerInfo->session, &settingFramePara);
460 settingFramePara.deviceBits = DFileGetDeviceBits();
461 }
462 EncodeSettingFrame(buf, NSTACKX_DEFAULT_FRAME_SIZE, &frameLen, &settingFramePara);
463
464 if (peerInfo->settingTimer != NULL) {
465 int32_t ret = DFileWriteHandle(buf, frameLen, peerInfo);
466 if (ret != (int32_t)frameLen && ret != NSTACKX_EAGAIN) {
467 data.errorCode = NSTACKX_EFAILED;
468 NotifyMsgRecver(peerInfo->session, DFILE_ON_CONNECT_FAIL, &data);
469 }
470 return;
471 }
472
473 if (peerInfo->session->sessionType == DFILE_SESSION_TYPE_CLIENT) {
474 peerInfo->settingTimer = TimerStart(peerInfo->session->epollfd, timeout,
475 NSTACKX_FALSE, ClientSettingTimeoutHandle, peerInfo);
476 if (peerInfo->settingTimer == NULL) {
477 DFILE_LOGE(TAG, "setting timmer creat fail");
478 data.errorCode = NSTACKX_EFAILED;
479 NotifyMsgRecver(peerInfo->session, DFILE_ON_CONNECT_FAIL, &data);
480 return;
481 }
482 } else {
483 peerInfo->settingTimer = TimerStart(peerInfo->session->epollfd, MAX_SERVER_NEGOTIATE_VALID_TIMEOUT,
484 NSTACKX_FALSE, ServerSettingTimeoutHandle, peerInfo);
485 if (peerInfo->settingTimer == NULL) {
486 return;
487 }
488 }
489
490 int32_t ret = DFileWriteHandle(buf, frameLen, peerInfo);
491 if (ret != (int32_t)frameLen && ret != NSTACKX_EAGAIN) {
492 data.errorCode = NSTACKX_EFAILED;
493 NotifyMsgRecver(peerInfo->session, DFILE_ON_CONNECT_FAIL, &data);
494 }
495 }
496
SetDFileSessionConfig(DFileSession * session,DFileConfig * dFileConfig,uint16_t connType,PeerInfo * peerInfo)497 static void SetDFileSessionConfig(DFileSession *session, DFileConfig *dFileConfig, uint16_t connType,
498 PeerInfo *peerInfo)
499 {
500 peerInfo->maxSendRate = dFileConfig->sendRate;
501 (void)memset_s(peerInfo->integralLossRate, INTEGRAL_TIME * sizeof(double), 0, INTEGRAL_TIME * sizeof(double));
502 peerInfo->fastStartCounter = 0;
503 ClockGetTime(CLOCK_MONOTONIC, &peerInfo->measureBefore);
504 peerInfo->dataFrameSize = dFileConfig->dataFrameSize;
505 if (connType == CONNECT_TYPE_WLAN) {
506 peerInfo->sendRate = dFileConfig->sendRate / NSTACKX_WLAN_INIT_SPEED_DIVISOR;
507 } else {
508 peerInfo->sendRate = dFileConfig->sendRate / NSTACKX_P2P_INIT_SPEED_DIVISOR;
509 }
510
511 #ifndef NSTACKX_WITH_LITEOS
512 if ((!peerInfo->gotWifiRate) && (connType == CONNECT_TYPE_WLAN)) {
513 peerInfo->sendRate = (uint16_t)(NSTACKX_WLAN_INIT_RATE / MSEC_TICKS_PER_SEC * DATA_FRAME_SEND_INTERVAL_MS
514 / dFileConfig->dataFrameSize + NSTACKX_WLAN_COMPENSATION_RATE);
515 }
516 #endif
517 if (peerInfo->sendRate < NSTACKX_MIN_SENDRATE) {
518 peerInfo->sendRate = NSTACKX_MIN_SENDRATE;
519 }
520
521 if (FileManagerSetMaxFrameLength(session->fileManager, peerInfo->dataFrameSize) != NSTACKX_EOK) {
522 DFILE_LOGE(TAG, "failed to set max frame length");
523 }
524
525 DFILE_LOGI(TAG, "connType is %u set sendrate is %u maxSendRate is %u peerInfo->dataFrameSize is %u",
526 connType, peerInfo->sendRate, peerInfo->maxSendRate, peerInfo->dataFrameSize);
527 if (session->sessionType == DFILE_SESSION_TYPE_SERVER) {
528 if (FileManagerSetRecvParaWithConnType(session->fileManager, connType) != NSTACKX_EOK) {
529 DFILE_LOGE(TAG, "failed to set recv para");
530 }
531 }
532 }
533
DFileSessionSetPeerInfo(PeerInfo * peerInfo,SettingState state,SettingFrame * settingFrame)534 static void DFileSessionSetPeerInfo(PeerInfo *peerInfo, SettingState state, SettingFrame *settingFrame)
535 {
536 peerInfo->state = state;
537 peerInfo->mtu = settingFrame->mtu;
538 peerInfo->connType = settingFrame->connType;
539 uint16_t localMtu = peerInfo->localMtu;
540 peerInfo->mtuInuse = (localMtu < peerInfo->mtu) ? localMtu : peerInfo->mtu;
541 peerInfo->remoteSessionId = settingFrame->header.sessionId;
542 }
543
DFileSessionHandleClientSetting(DFileSession * session,DFileFrame * dFileFrame,struct sockaddr_in * peerAddr)544 static void DFileSessionHandleClientSetting(DFileSession *session, DFileFrame *dFileFrame, struct sockaddr_in *peerAddr)
545 {
546 List *pos = NULL;
547 SettingFrame hostSettingFrame;
548 (void)memset_s(&hostSettingFrame, sizeof(hostSettingFrame), 0, sizeof(hostSettingFrame));
549 DFILE_LOGI(TAG, "handle Setting Frame, DFileSessionType %u", session->sessionType);
550 /* unsupport version */
551 if (DecodeSettingFrame((SettingFrame *)dFileFrame, &hostSettingFrame) != NSTACKX_EOK || hostSettingFrame.mtu == 0) {
552 return;
553 }
554 PeerInfo *peerInfo = SearchPeerInfoNode(session, peerAddr);
555 if (peerInfo == NULL) {
556 DFILE_LOGE(TAG, "recv unknown peer setting, maybe be attacked");
557 return;
558 }
559 peerInfo->remoteDFileVersion = hostSettingFrame.dFileVersion;
560 TimerDelete(peerInfo->settingTimer);
561 peerInfo->settingTimer = NULL;
562 DFileSessionSetPeerInfo(peerInfo, SETTING_NEGOTIATED, &hostSettingFrame);
563 LIST_FOR_EACH(pos, &session->dFileTransChain) {
564 DFileTrans *trans = (DFileTrans *)pos;
565 if (DFileTransSetMtu(trans, peerInfo->mtuInuse) != NSTACKX_EOK) {
566 DFILE_LOGE(TAG, "set trans mtu failed");
567 }
568 }
569 DFileMsg data;
570 (void)memset_s(&data, sizeof(data), 0, sizeof(data));
571 data.errorCode = NSTACKX_EOK;
572 NotifyMsgRecver(peerInfo->session, DFILE_ON_CONNECT_SUCCESS, &data);
573
574 DFileConfig dFileConfig;
575 (void)memset_s(&dFileConfig, sizeof(dFileConfig), 0, sizeof(dFileConfig));
576 if (GetDFileConfig(&dFileConfig, peerInfo->mtuInuse, hostSettingFrame.connType) == NSTACKX_EOK) {
577 SetDFileSessionConfig(session, &dFileConfig, hostSettingFrame.connType, peerInfo);
578 }
579 if (hostSettingFrame.capability & NSTACKX_CAPS_LINK_SEQUENCE) {
580 DFILE_LOGI(TAG, "server replies not support Link Sequence");
581 } else {
582 DFILE_LOGI(TAG, "server replies using normal ACK");
583 session->capability &= ~NSTACKX_CAPS_LINK_SEQUENCE;
584 }
585 DFileChooseCipherType(&hostSettingFrame, session);
586 }
587
DFileGetMTU(SocketProtocol protocol)588 static uint16_t DFileGetMTU(SocketProtocol protocol)
589 {
590 /* for udp, return NSTACKX_DEFAULT_FRAME_SIZE, for D2D, need call D2D MTU interface */
591 uint16_t mtu = 0;
592 if (protocol == NSTACKX_PROTOCOL_UDP) {
593 mtu = NSTACKX_DEFAULT_FRAME_SIZE;
594 } else if (protocol == NSTACKX_PROTOCOL_D2D) {
595 DFILE_LOGE(TAG, "d2d not support");
596 } else if (protocol == NSTACKX_PROTOCOL_TCP) {
597 mtu = NSTACKX_DEFAULT_FRAME_SIZE;
598 }
599
600 return mtu;
601 }
602
CreatePeerInfo(DFileSession * session,const struct sockaddr_in * dstAddr,uint16_t peerMtu,uint16_t connType,uint8_t socketIndex)603 PeerInfo *CreatePeerInfo(DFileSession *session, const struct sockaddr_in *dstAddr, uint16_t peerMtu,
604 uint16_t connType, uint8_t socketIndex)
605 {
606 if (session->peerInfoCnt >= MAX_PEERINFO_SIZE) {
607 return NULL;
608 }
609 PeerInfo *peerInfo = calloc(1, sizeof(PeerInfo));
610 if (peerInfo == NULL) {
611 return NULL;
612 }
613 peerInfo->session = session;
614 peerInfo->dstAddr = *dstAddr;
615 peerInfo->connType = connType;
616 peerInfo->socketIndex = socketIndex;
617 peerInfo->localMtu = DFileGetMTU(session->protocol);
618 session->peerInfoCnt++;
619 peerInfo->gotWifiRate = 0;
620 peerInfo->ackInterval = NSTACKX_INIT_ACK_INTERVAL;
621 peerInfo->rateStateInterval = NSTACKX_INIT_RATE_STAT_INTERVAL;
622
623 if (GetInterfaceNameByIP(session->socket[socketIndex]->srcAddr.sin_addr.s_addr,
624 peerInfo->localInterfaceName, sizeof(peerInfo->localInterfaceName)) != NSTACKX_EOK) {
625 DFILE_LOGE(TAG, "GetInterfaceNameByIP failed %d", errno);
626 }
627
628 if (peerMtu == 0) {
629 return peerInfo;
630 }
631 peerInfo->mtu = peerMtu;
632 peerInfo->mtuInuse = (peerInfo->localMtu < peerInfo->mtu) ? peerInfo->localMtu : peerInfo->mtu;
633 return peerInfo;
634 }
635
HandleLinkSeqCap(DFileSession * session,SettingFrame * hostSettingFrame)636 static void HandleLinkSeqCap(DFileSession *session, SettingFrame *hostSettingFrame)
637 {
638 if (hostSettingFrame->capability & NSTACKX_CAPS_LINK_SEQUENCE) {
639 DFILE_LOGI(TAG, "client wants to enable Link Sequence");
640 } else {
641 DFILE_LOGI(TAG, "client wants to use normal ACK");
642 session->capability &= ~NSTACKX_CAPS_LINK_SEQUENCE;
643 }
644
645 if (hostSettingFrame->capability & NSTACKX_INTERNAL_CAPS_RECV_FEEDBACK) {
646 DFILE_LOGI(TAG, "client support recv feedback");
647 session->capsCheck |= NSTACKX_INTERNAL_CAPS_RECV_FEEDBACK;
648 } else {
649 DFILE_LOGI(TAG, "client do not support recv feedback");
650 session->capsCheck &= ~NSTACKX_INTERNAL_CAPS_RECV_FEEDBACK;
651 }
652 }
653
AllocPeerInfo(DFileSession * session,const struct sockaddr_in * peerAddr,const SettingFrame * frame,uint8_t socketIndex)654 static PeerInfo *AllocPeerInfo(DFileSession *session, const struct sockaddr_in *peerAddr, const SettingFrame *frame,
655 uint8_t socketIndex)
656 {
657 PeerInfo *peerInfo = NULL;
658
659 peerInfo = CreatePeerInfo(session, peerAddr, frame->mtu, frame->connType, socketIndex);
660 if (peerInfo == NULL) {
661 return NULL;
662 }
663
664 peerInfo->remoteSessionId = frame->header.sessionId;
665 peerInfo->state = SETTING_NEGOTIATING;
666 DFileSessionSendSetting(peerInfo);
667 if (peerInfo->settingTimer == NULL) {
668 free(peerInfo);
669 session->peerInfoCnt--;
670 return NULL;
671 }
672 ListInsertTail(&peerInfo->session->peerInfoChain, &peerInfo->list);
673 return peerInfo;
674 }
675
DFileSessionHandleServerSetting(DFileSession * session,DFileFrame * dFileFrame,struct sockaddr_in * peerAddr,uint8_t socketIndex)676 static void DFileSessionHandleServerSetting(DFileSession *session, DFileFrame *dFileFrame,
677 struct sockaddr_in *peerAddr, uint8_t socketIndex)
678 {
679 SettingFrame hostSettingFrame;
680 DFileConfig dFileConfig;
681 (void)memset_s(&hostSettingFrame, sizeof(hostSettingFrame), 0, sizeof(hostSettingFrame));
682 (void)memset_s(&dFileConfig, sizeof(dFileConfig), 0, sizeof(dFileConfig));
683 DFILE_LOGI(TAG, "handle Setting Frame, DFileSessionType %u", session->sessionType);
684 if (DecodeSettingFrame((SettingFrame *)dFileFrame, &hostSettingFrame) != NSTACKX_EOK || hostSettingFrame.mtu == 0) {
685 return;
686 }
687
688 PeerInfo *peerInfo = SearchPeerInfoNode(session, peerAddr);
689 if (peerInfo != NULL) {
690 if (peerInfo->settingTimeoutCnt >= MAX_NEGOTIATE_TIMEOUT_COUNT) {
691 DFILE_LOGE(TAG, "receive more than %d Setting for one peer, drop", MAX_NEGOTIATE_TIMEOUT_COUNT);
692 return;
693 } else {
694 DFileSessionSetPeerInfo(peerInfo, SETTING_NEGOTIATING, &hostSettingFrame);
695 DFileSessionSendSetting(peerInfo);
696 goto L_END;
697 }
698 }
699
700 peerInfo = AllocPeerInfo(session, peerAddr, &hostSettingFrame, socketIndex);
701 if (peerInfo == NULL) {
702 return;
703 }
704
705 L_END:
706 peerInfo->settingTimeoutCnt++;
707 DFILE_LOGI(TAG, "DFileServer response Setting Frame. count %u", peerInfo->settingTimeoutCnt);
708 peerInfo->remoteDFileVersion = hostSettingFrame.dFileVersion;
709 if (GetDFileConfig(&dFileConfig, peerInfo->mtuInuse, hostSettingFrame.connType) == NSTACKX_EOK) {
710 SetDFileSessionConfig(session, &dFileConfig, hostSettingFrame.connType, peerInfo);
711 }
712 HandleLinkSeqCap(session, &hostSettingFrame);
713 DFileChooseCipherType(&hostSettingFrame, session);
714 }
715
DFileSessionHandleSetting(DFileSession * session,DFileFrame * dFileFrame,struct sockaddr_in * peerAddr,uint8_t socketIndex)716 static void DFileSessionHandleSetting(DFileSession *session, DFileFrame *dFileFrame,
717 struct sockaddr_in *peerAddr, uint8_t socketIndex)
718 {
719 if (session->sessionType == DFILE_SESSION_TYPE_SERVER) {
720 DFileSessionHandleServerSetting(session, dFileFrame, peerAddr, socketIndex);
721 } else if (session->sessionType == DFILE_SESSION_TYPE_CLIENT) {
722 DFileSessionHandleClientSetting(session, dFileFrame, peerAddr);
723 } else {
724 return;
725 }
726 }
727
HandleWithoutSettingError(DFileSession * session,const struct sockaddr_in * peerAddr)728 static void HandleWithoutSettingError(DFileSession *session, const struct sockaddr_in *peerAddr)
729 {
730 List *pos = NULL;
731 DFileTrans *trans = NULL;
732
733 PeerInfo *peerInfo = SearchPeerInfoNode(session, peerAddr);
734 if (peerInfo == NULL) {
735 DFILE_LOGE(TAG, "recv unknown peer rst, maybe be attacked");
736 return;
737 }
738
739 LIST_FOR_EACH(pos, &session->dFileTransChain) {
740 trans = (DFileTrans *)pos;
741 /*
742 * when client recv peer RST NSTACKX_DFILE_WITHOUT_SETTING_ERROR, Set Trans MTU 0,
743 * and will reset after new setting negotion.
744 */
745 if (DFileTransSetMtu(trans, 0) != NSTACKX_EOK) {
746 DFILE_LOGE(TAG, "DFileTransSetMtu(trans, 0) failed %d", errno);
747 }
748 }
749
750 LIST_FOR_EACH(pos, &session->peerInfoChain) {
751 peerInfo = (PeerInfo *)pos;
752 if (memcmp(&peerInfo->dstAddr, peerAddr, sizeof(struct sockaddr_in)) == 0 &&
753 peerInfo->session->sessionId == session->sessionId) {
754 TimerDelete(peerInfo->settingTimer);
755 peerInfo->settingTimer = NULL;
756 peerInfo->state = SETTING_NEGOTIATING;
757 DFILE_LOGD(TAG, "Send Setting Frame");
758 DFileSessionSendSetting(peerInfo);
759 break;
760 }
761 }
762 }
763
DFileSessionHandleRst(DFileSession * session,DFileFrame * dFileFrame,struct sockaddr_in * peerAddr)764 static void DFileSessionHandleRst(DFileSession *session, DFileFrame *dFileFrame, struct sockaddr_in *peerAddr)
765 {
766 uint16_t errCode = 0;
767 if (DecodeRstFrame((RstFrame *)dFileFrame, &errCode, NULL, NULL) != NSTACKX_EOK) {
768 return;
769 }
770
771 uint16_t transId = ntohs(dFileFrame->header.transId);
772 DFILE_LOGD(TAG, "handle RST (%hu) frame, transId %hu", errCode, transId);
773
774 switch (errCode) {
775 case NSTACKX_DFILE_WITHOUT_SETTING_ERROR:
776 HandleWithoutSettingError(session, peerAddr);
777 break;
778 default:
779 DFILE_LOGE(TAG, "Unspported error code %hu", errCode);
780 break;
781 }
782 }
783
DFileSessionResolveBackPress(DFileSession * session,DataBackPressure backPress,uint32_t count)784 static void DFileSessionResolveBackPress(DFileSession *session, DataBackPressure backPress, uint32_t count)
785 {
786 uint32_t index;
787
788 if (PthreadMutexLock(&session->backPressLock) != 0) {
789 DFILE_LOGE(TAG, "pthread backPressLock mutex lock failed");
790 return;
791 }
792
793 if (backPress.recvListOverIo == 1) {
794 for (index = 0; index < count; index++) {
795 session->stopSendCnt[index]++;
796 }
797 } else {
798 for (index = 0; index < count; index++) {
799 session->stopSendCnt[index] = 0;
800 }
801 }
802
803 if (PthreadMutexUnlock(&session->backPressLock) != 0) {
804 DFILE_LOGE(TAG, "pthread backPressLock mutex unlock failed");
805 return;
806 }
807
808 return;
809 }
810
DFileSessionHandleBackPressure(DFileSession * session,const DFileFrame * dFileFrame,const struct sockaddr_in * peerAddr)811 static void DFileSessionHandleBackPressure(DFileSession *session, const DFileFrame *dFileFrame,
812 const struct sockaddr_in *peerAddr)
813 {
814 DataBackPressure backPress;
815 PeerInfo *peerInfo = SearchPeerInfoNode(session, peerAddr);
816 if (peerInfo == NULL) {
817 DFILE_LOGE(TAG, "can't get valid peerinfo");
818 return;
819 }
820
821 if (DecodeBackPressFrame((BackPressureFrame *)dFileFrame, &backPress) != NSTACKX_EOK) {
822 return;
823 }
824
825 DFileSessionResolveBackPress(session, backPress, session->clientSendThreadNum);
826
827 DFILE_LOGI(TAG, "handle back pressure recvListOverIo %u recvBufThreshold %u stopSendPeriod %u",
828 backPress.recvListOverIo, backPress.recvBufThreshold,
829 backPress.stopSendPeriod);
830
831 return;
832 }
833
CreateTrans(uint16_t transId,DFileSession * session,PeerInfo * peerInfo,uint8_t isSender)834 static DFileTrans *CreateTrans(uint16_t transId, DFileSession *session, PeerInfo *peerInfo, uint8_t isSender)
835 {
836 DFileTransPara transPara;
837
838 if (peerInfo == NULL) {
839 return NULL;
840 }
841 (void)memset_s(&transPara, sizeof(transPara), 0, sizeof(transPara));
842 transPara.isSender = isSender;
843 transPara.transId = transId; /* for receiver, use transId of sender */
844 transPara.fileManager = session->fileManager;
845 transPara.writeHandle = DFileWriteHandle;
846 transPara.msgReceiver = DTransMsgReceiver;
847 transPara.connType = peerInfo->connType;
848 transPara.context = peerInfo;
849 transPara.session = session;
850 transPara.onRenameFile = session->onRenameFile;
851
852 ClockGetTime(CLOCK_MONOTONIC, &peerInfo->startTime);
853 return DFileTransCreate(&transPara);
854 }
855
DFileSessionHandleFrame(DFileSession * session,DFileFrame * dFileFrame,struct sockaddr_in * peerAddr)856 static int32_t DFileSessionHandleFrame(DFileSession *session, DFileFrame *dFileFrame, struct sockaddr_in *peerAddr)
857 {
858 PeerInfo *peerInfo = SearchPeerInfoNode(session, peerAddr);
859 if (peerInfo == NULL) {
860 DFILE_LOGI(TAG, "can't find peerInfo");
861 return NSTACKX_EFAILED;
862 }
863
864 if (peerInfo->session->sessionType == DFILE_SESSION_TYPE_SERVER && peerInfo->state == SETTING_NEGOTIATING) {
865 peerInfo->state = SETTING_NEGOTIATED;
866 TimerDelete(peerInfo->settingTimer);
867 peerInfo->settingTimer = NULL;
868 }
869
870 uint16_t transId = ntohs(dFileFrame->header.transId);
871 if (transId == 0) {
872 DFILE_LOGE(TAG, "transId is 0");
873 return NSTACKX_EFAILED;
874 }
875
876 DFileTrans *trans = SearchDFileTransNode(&(session->dFileTransChain), transId);
877 if (trans == NULL) {
878 if (dFileFrame->header.type != NSTACKX_DFILE_FILE_HEADER_FRAME) {
879 /* Only HEADER frame can start dfile transfer (receiver) */
880 DFILE_LOGE(TAG, "trans %u is NULL && type is %u", transId, dFileFrame->header.type);
881 return NSTACKX_EFAILED;
882 }
883 if (IsTransIdDone(session, transId) == NSTACKX_EOK) {
884 return NSTACKX_EFAILED;
885 }
886 trans = CreateTrans(transId, session, peerInfo, NSTACKX_FALSE);
887 if (trans == NULL) {
888 DFileMsg data;
889 (void)memset_s(&data, sizeof(data), 0, sizeof(data));
890 data.errorCode = NSTACKX_ENOMEM;
891 NotifyMsgRecver(session, DFILE_ON_FATAL_ERROR, &data);
892 DFILE_LOGE(TAG, "trans is NULL");
893 return NSTACKX_EFAILED;
894 }
895 if (DFileTransSetMtu(trans, peerInfo->mtuInuse) != NSTACKX_EOK) {
896 DFILE_LOGE(TAG, "set trans mtu failed");
897 }
898 CalculateSessionTransferRatePrepare(session);
899 ListInsertTail(&(session->dFileTransChain), &(trans->list));
900 peerInfo->currentTransCount++;
901 }
902
903 return HandleDFileFrame(trans, dFileFrame);
904 }
905
DFileWriteHandle(const uint8_t * frame,size_t len,void * context)906 int32_t DFileWriteHandle(const uint8_t *frame, size_t len, void *context)
907 {
908 PeerInfo *peerInfo = context;
909 DFileSession *session = peerInfo->session;
910 QueueNode *queueNode = NULL;
911 struct sockaddr_in peerAddr;
912
913 peerAddr = peerInfo->dstAddr;
914 queueNode = CreateQueueNode(frame, len, &peerAddr, peerInfo->socketIndex);
915 if (queueNode == NULL) {
916 return NSTACKX_ENOMEM;
917 }
918
919 if (PthreadMutexLock(&session->outboundQueueLock) != 0) {
920 DFILE_LOGE(TAG, "pthread mutex lock failed");
921 DestroyQueueNode(queueNode);
922 return NSTACKX_EFAILED;
923 }
924 ListInsertTail(&session->outboundQueue, &queueNode->list);
925 session->outboundQueueSize++;
926 if (PthreadMutexUnlock(&session->outboundQueueLock) != 0) {
927 /* Don't need to free node, as it's mount to outboundQueue */
928 DFILE_LOGE(TAG, "pthread mutex unlock failed");
929 return NSTACKX_EFAILED;
930 }
931 SemPost(&session->outboundQueueWait[peerInfo->socketIndex]);
932 return (int32_t)len;
933 }
934
BindMainLoopToTargetCpu(void)935 static void BindMainLoopToTargetCpu(void)
936 {
937 int32_t cpu;
938 int32_t cpus = GetCpuNum();
939 if (cpus >= FIRST_CPU_NUM_LEVEL) {
940 return;
941 } else if (cpus >= THIRD_CPU_NUM_LEVEL) {
942 cpu = CPU_IDX_0;
943 } else {
944 return;
945 }
946 StartThreadBindCore(cpu);
947 }
948
GetEpollWaitTimeOut(DFileSession * session)949 static int64_t GetEpollWaitTimeOut(DFileSession *session)
950 {
951 int64_t minTimeout = DEFAULT_WAIT_TIME_MS;
952 if (session->mainLoopActiveReadFlag && session->inboundQueueSize) {
953 minTimeout = 0;
954 return minTimeout;
955 }
956 List *pos = NULL;
957 DFileTrans *trans = NULL;
958 int64_t timeout;
959 LIST_FOR_EACH(pos, &session->dFileTransChain) {
960 trans = (DFileTrans *)pos;
961 timeout = DFileTransGetTimeout(trans);
962 if (timeout >= 0 && timeout < minTimeout) {
963 minTimeout = timeout;
964 }
965 }
966 if (minTimeout > DEFAULT_WAIT_TIME_MS) {
967 minTimeout = DEFAULT_WAIT_TIME_MS;
968 }
969 return minTimeout;
970 }
971
ProcessSessionTrans(const DFileSession * session,uint16_t exceptTransId)972 static void ProcessSessionTrans(const DFileSession *session, uint16_t exceptTransId)
973 {
974 List *tmp = NULL;
975 List *pos = NULL;
976 LIST_FOR_EACH_SAFE(pos, tmp, &session->dFileTransChain) {
977 DFileTrans *trans = (DFileTrans *)pos;
978 if (trans->transId != exceptTransId) {
979 DFileTransProcess(trans);
980 }
981 }
982 }
983
NotifyBindPort(const DFileSession * session)984 static void NotifyBindPort(const DFileSession *session)
985 {
986 DFileMsg data;
987 int32_t socketNum;
988 (void)memset_s(&data, sizeof(data), 0, sizeof(data));
989 socketNum = 1;
990
991 for (int i = 0; i < socketNum; i++) {
992 data.sockAddr[i].sin_port = ntohs(session->socket[i]->srcAddr.sin_port);
993 data.sockAddr[i].sin_addr.s_addr = ntohl(session->socket[i]->srcAddr.sin_addr.s_addr);
994 }
995 NotifyMsgRecver(session, DFILE_ON_BIND, &data);
996 }
997
DFileMainLoop(void * arg)998 void *DFileMainLoop(void *arg)
999 {
1000 DFileSession *session = arg;
1001 int32_t ret = NSTACKX_EOK;
1002 DFileMsg msgData;
1003 (void)memset_s(&msgData, sizeof(msgData), 0, sizeof(msgData));
1004 uint8_t isBind = NSTACKX_FALSE;
1005 DFILE_LOGI(TAG, "main thread start");
1006 SetThreadName(DFFILE_MAIN_THREAD_NAME);
1007 SetMaximumPriorityForThread();
1008 SetTidToBindInfo(session, POS_MAIN_THERD_START);
1009 NotifyBindPort(session);
1010 while (!session->closeFlag) {
1011 int64_t minTimeout = GetEpollWaitTimeOut(session);
1012 ret = EpollLoop(session->epollfd, (int32_t)minTimeout);
1013 if (ret == NSTACKX_EFAILED) {
1014 DFILE_LOGE(TAG, "epoll wait failed");
1015 break;
1016 }
1017 if (isBind == NSTACKX_FALSE && session->transFlag == NSTACKX_TRUE) {
1018 BindMainLoopToTargetCpu();
1019 isBind = NSTACKX_TRUE;
1020 }
1021 ProcessSessionTrans(session, 0);
1022 if (session->mainLoopActiveReadFlag && session->inboundQueueSize) {
1023 session->partReadFlag = NSTACKX_TRUE;
1024 ReadEventHandle(session);
1025 session->partReadFlag = NSTACKX_FALSE;
1026 }
1027 }
1028
1029 if (ret == NSTACKX_EFAILED || DFileSessionCheckFatalFlag(session)) {
1030 msgData.errorCode = NSTACKX_EFAILED;
1031 NotifyMsgRecver(session, DFILE_ON_FATAL_ERROR, &msgData);
1032 }
1033
1034 /* Notify sender thread to terminate */
1035 PostOutboundQueueWait(session);
1036
1037 /* Unblock "select" and notify receiver thread to terminate */
1038 NotifyPipeEvent(session);
1039 return NULL;
1040 }
1041
AmendPeerInfoSendRate(PeerInfo * peerInfo)1042 static void AmendPeerInfoSendRate(PeerInfo *peerInfo)
1043 {
1044 peerInfo->amendSendRate = peerInfo->sendRate;
1045 DFILE_LOGI(TAG, "current: sendrate %u, realsendframerate %u---new amendSendRate %d",
1046 peerInfo->sendRate, peerInfo->sendFrameRate, peerInfo->amendSendRate);
1047 }
1048
DFileSendCalculateRate(DFileSession * session,PeerInfo * peerInfo)1049 static void DFileSendCalculateRate(DFileSession *session, PeerInfo *peerInfo)
1050 {
1051 uint64_t sendCount;
1052
1053 if (session->sessionType != DFILE_SESSION_TYPE_CLIENT) {
1054 return;
1055 }
1056
1057 struct timespec nowTime;
1058 ClockGetTime(CLOCK_MONOTONIC, &nowTime);
1059 uint64_t measureElapse = GetTimeDiffUs(&nowTime, &peerInfo->startTime);
1060 /* just calculate io rate */
1061 if (measureElapse > peerInfo->rateStateInterval) {
1062 /* ONLY PEER 0 calculate io rate */
1063 sendCount = (uint64_t)peerInfo->sendCount;
1064 if (peerInfo->socketIndex == 0) {
1065 session->fileManager->iorRate = (uint32_t)(session->fileManager->iorBytes *
1066 NSTACKX_MICRO_TICKS / measureElapse / DFILE_MEGABYTES);
1067 DFILE_LOGI(TAG, "IO read rate: %u MB/s send list full times %u", session->fileManager->iorRate,
1068 session->fileManager->sendListFullTimes);
1069 session->fileManager->sendListFullTimes = 0;
1070 session->fileManager->iorBytes = 0;
1071 }
1072
1073 peerInfo->sendFrameRate = (uint32_t)(sendCount * DATA_FRAME_SEND_INTERVAL_US / measureElapse);
1074 peerInfo->sendCountRateMB = (uint32_t)(sendCount * peerInfo->dataFrameSize *
1075 NSTACKX_MICRO_TICKS / measureElapse / DFILE_MEGABYTES);
1076 if (peerInfo->qdiscSearchNum != 0) {
1077 peerInfo->qdiscAveLeft = peerInfo->qdiscAveLeft / peerInfo->qdiscSearchNum;
1078 }
1079
1080 DFILE_LOGI(TAG, "framesize %u maxsendrate %u sendRate %u, amendSendRate %d sendCount %llu,"
1081 "measureElapse %llu sendFrameRate %u %uMB/s,"
1082 "total send block num %llu eAgainCount %u send list empty times %u sleep times %u, noPendingData %u,"
1083 "min qdisc %u max qdisc %u search num %u ave qdisc %u"
1084 "totalRecvBlocks %llu socket:%u "
1085 "overRun %llu maxRetryCountPerSec %u maxRetryCountLastSec %u wlanCatagory %u",
1086 peerInfo->dataFrameSize, peerInfo->maxSendRate, peerInfo->sendRate, peerInfo->amendSendRate,
1087 peerInfo->sendCount, measureElapse, peerInfo->sendFrameRate, peerInfo->sendCountRateMB,
1088 NSTACKX_ATOM_FETCH(&(session->totalSendBlocks)), peerInfo->eAgainCount,
1089 NSTACKX_ATOM_FETCH(&(session->sendBlockListEmptyTimes)), session->sleepTimes,
1090 NSTACKX_ATOM_FETCH(&(session->noPendingDataTimes)), peerInfo->qdiscMinLeft,
1091 peerInfo->qdiscMaxLeft, peerInfo->qdiscSearchNum, peerInfo->qdiscAveLeft,
1092 NSTACKX_ATOM_FETCH(&(session->totalRecvBlocks)), peerInfo->socketIndex,
1093 peerInfo->overRun, peerInfo->maxRetryCountPerSec, peerInfo->maxRetryCountLastSec, session->wlanCatagory);
1094 AmendPeerInfoSendRate(peerInfo);
1095 ClearSessionStats(session);
1096 ClearPeerinfoStats(peerInfo);
1097 ClockGetTime(CLOCK_MONOTONIC, &peerInfo->startTime);
1098 }
1099 }
1100
UpdateAllTransRetryCount(DFileSession * session,PeerInfo * peerInfo)1101 void UpdateAllTransRetryCount(DFileSession *session, PeerInfo *peerInfo)
1102 {
1103 List *pos = NULL;
1104 DFileTrans *trans = NULL;
1105 uint32_t allRetryCount = 0;
1106 LIST_FOR_EACH(pos, &session->dFileTransChain) { /* for client , only one peer */
1107 trans = (DFileTrans *)pos;
1108 allRetryCount += trans->retryCount;
1109 }
1110 peerInfo->allDtransRetryCount = allRetryCount;
1111 }
1112
DFileRecvCalculateRate(DFileSession * session,DFileFrame * dFileFrame,struct sockaddr_in * peerAddr)1113 static void DFileRecvCalculateRate(DFileSession *session, DFileFrame *dFileFrame, struct sockaddr_in *peerAddr)
1114 {
1115 struct timespec nowTime;
1116 uint64_t recvCount;
1117 uint32_t timeOut;
1118
1119 if (session->sessionType != DFILE_SESSION_TYPE_SERVER ||
1120 dFileFrame->header.type != NSTACKX_DFILE_FILE_DATA_FRAME) {
1121 return;
1122 }
1123
1124 PeerInfo *peerInfo = SearchPeerInfoNode(session, peerAddr);
1125 if (peerInfo == NULL) {
1126 return;
1127 }
1128
1129 timeOut = (peerInfo->connType == CONNECT_TYPE_P2P) ? NSTACKX_P2P_MAX_CONTROL_FRAME_TIMEOUT :
1130 NSTACKX_WLAN_MAX_CONTROL_FRAME_TIMEOUT;
1131
1132 peerInfo->recvCount++;
1133 ClockGetTime(CLOCK_MONOTONIC, &nowTime);
1134 uint64_t measureElapse = GetTimeDiffUs(&nowTime, &session->measureBefore);
1135 if (measureElapse > peerInfo->rateStateInterval) {
1136 session->fileManager->iowRate = (uint32_t)(session->fileManager->iowBytes *
1137 NSTACKX_MICRO_TICKS / measureElapse / DFILE_KILOBYTES);
1138 session->fileManager->iowCount = session->fileManager->iowBytes / peerInfo->dataFrameSize *
1139 (NSTACKX_MILLI_TICKS * timeOut - peerInfo->rateStateInterval) / measureElapse;
1140 if (session->fileManager->iowRate > session->fileManager->iowMaxRate) {
1141 session->fileManager->iowMaxRate = session->fileManager->iowRate;
1142 }
1143 DFILE_LOGI(TAG, "measureElapse %llu iowBytes %llu iowCount %llu IO write rate : %u KB/s", measureElapse,
1144 session->fileManager->iowBytes, session->fileManager->iowCount, session->fileManager->iowRate);
1145 session->fileManager->iowBytes = 0;
1146 ClockGetTime(CLOCK_MONOTONIC, &session->measureBefore);
1147 }
1148
1149 measureElapse = GetTimeDiffUs(&nowTime, &peerInfo->startTime);
1150 if (measureElapse > peerInfo->rateStateInterval) {
1151 recvCount = (uint64_t)peerInfo->recvCount;
1152 peerInfo->recvFrameRate = (uint32_t)(recvCount * DATA_FRAME_SEND_INTERVAL_US / measureElapse);
1153 peerInfo->recvCountRateMB = (uint32_t)(recvCount * peerInfo->dataFrameSize *
1154 NSTACKX_MICRO_TICKS / measureElapse / DFILE_MEGABYTES);
1155 peerInfo->recvCount = 0;
1156 ClockGetTime(CLOCK_MONOTONIC, &peerInfo->startTime);
1157 }
1158 }
1159
CheckElapseTime(const struct timespec * before,uint64_t overRun)1160 static uint64_t CheckElapseTime(const struct timespec *before, uint64_t overRun)
1161 {
1162 struct timespec now;
1163 ClockGetTime(CLOCK_MONOTONIC, &now);
1164 uint64_t elapseUs = GetTimeDiffUs(&now, before);
1165 elapseUs += overRun;
1166 if (elapseUs < DATA_FRAME_SEND_INTERVAL_US) {
1167 if (usleep((useconds_t)((uint64_t)DATA_FRAME_SEND_INTERVAL_US - elapseUs)) != NSTACKX_EOK) {
1168 DFILE_LOGE(TAG, "usleep(DATA_FRAME_SEND_INTERVAL_US - elapseUs) failed %d", errno);
1169 }
1170 return 0;
1171 } else {
1172 uint64_t delta = (elapseUs - DATA_FRAME_SEND_INTERVAL_US);
1173 return delta;
1174 }
1175 }
1176
BindClientSendThreadToTargetCpu(uint32_t idx)1177 static void BindClientSendThreadToTargetCpu(uint32_t idx)
1178 {
1179 int32_t cpu;
1180 int32_t cpus = GetCpuNum();
1181 if (cpus >= FIRST_CPU_NUM_LEVEL) {
1182 return;
1183 } else if (cpus >= SECOND_CPU_NUM_LEVEL) {
1184 cpu = CPU_IDX_1 + (int32_t)idx;
1185 } else if (cpus >= THIRD_CPU_NUM_LEVEL) {
1186 cpu = CPU_IDX_1;
1187 } else {
1188 return;
1189 }
1190 if (cpu > cpus) {
1191 cpu = cpus - 1;
1192 }
1193 StartThreadBindCore(cpu);
1194 }
1195
DFileSenderUpdateMeasureTime(DFileSession * session,uint8_t socketIndex)1196 static void DFileSenderUpdateMeasureTime(DFileSession *session, uint8_t socketIndex)
1197 {
1198 if (session->sessionType == DFILE_SESSION_TYPE_CLIENT) {
1199 PeerInfo *peerInfo = ClientGetPeerInfoBySocketIndex(socketIndex, session);
1200 if (peerInfo == NULL) {
1201 return;
1202 }
1203 ClockGetTime(CLOCK_MONOTONIC, &peerInfo->startTime);
1204 BindClientSendThreadToTargetCpu(0);
1205 }
1206 }
1207
TerminateMainThreadFatalInner(void * arg)1208 static void TerminateMainThreadFatalInner(void *arg)
1209 {
1210 DFileSession *session = (DFileSession *)arg;
1211 DFileSessionSetFatalFlag(session);
1212 }
1213
PostFatalEvent(DFileSession * session)1214 static void PostFatalEvent(DFileSession *session)
1215 {
1216 if (PostEvent(&session->eventNodeChain, session->epollfd, TerminateMainThreadFatalInner, session) != NSTACKX_EOK) {
1217 DFILE_LOGE(TAG, "PostEvent TerminateMainThreadFatalInner failed");
1218 DFileSessionSetFatalFlag(session);
1219 }
1220 }
1221
GetSocketWaitMs(uint32_t threadNum)1222 static uint32_t GetSocketWaitMs(uint32_t threadNum)
1223 {
1224 if (threadNum > 1) {
1225 return MULTI_THREADS_SOCKET_WAIT_TIME_MS;
1226 }
1227 return DEFAULT_WAIT_TIME_MS;
1228 }
1229
SetSendThreadName(uint32_t threadIdx)1230 static void SetSendThreadName(uint32_t threadIdx)
1231 {
1232 char name[MAX_THREAD_NAME_LEN] = {0};
1233 if (sprintf_s(name, sizeof(name), "%s%u", DFFILE_SEND_THREAD_NAME_PREFIX, threadIdx) < 0) {
1234 DFILE_LOGE(TAG, "sprintf send thead name failed");
1235 }
1236 SetThreadName(name);
1237 }
1238
1239 typedef struct {
1240 struct DFileSession *session;
1241 uint32_t threadIdx;
1242 }SendThreadCtx;
1243
DFileAddiSenderHandle(void * arg)1244 static void *DFileAddiSenderHandle(void *arg)
1245 {
1246 SendThreadCtx *sendThreadCtx = (SendThreadCtx *)arg;
1247 struct DFileSession *session = sendThreadCtx->session;
1248 uint32_t threadIdx = sendThreadCtx->threadIdx;
1249 free(sendThreadCtx);
1250 int32_t ret = NSTACKX_EOK;
1251 DFILE_LOGI(TAG, "send thread %u start", threadIdx);
1252 SetSendThreadName(threadIdx);
1253 SetTidToBindInfo(session, threadIdx + POS_SEND_THERD_START);
1254 List unsent;
1255 uint8_t canWrite = NSTACKX_FALSE;
1256 uint32_t socketWaitMs = GetSocketWaitMs(session->clientSendThreadNum);
1257 BindClientSendThreadToTargetCpu(threadIdx + 1);
1258 ListInitHead(&unsent);
1259 while (!session->addiSenderCloseFlag) {
1260 if (ListIsEmpty(&unsent) && !FileManagerHasPendingData(session->fileManager)) {
1261 NSTACKX_ATOM_FETCH_INC(&session->noPendingDataTimes);
1262 SemWait(&session->sendThreadPara[threadIdx].sendWait);
1263 if (session->addiSenderCloseFlag) {
1264 break;
1265 }
1266 }
1267 if (ret == NSTACKX_EAGAIN) {
1268 ret = WaitSocketEvent(NULL, session->socket[0]->sockfd, socketWaitMs, NULL, &canWrite);
1269 if (ret != NSTACKX_EOK || session->closeFlag) {
1270 break;
1271 }
1272 if (!canWrite) {
1273 ret = NSTACKX_EAGAIN;
1274 continue;
1275 }
1276 }
1277 ret = SendDataFrame(session, &unsent, threadIdx, 0);
1278 if ((ret < 0 && ret != NSTACKX_EAGAIN) || session->closeFlag) {
1279 break;
1280 }
1281 if (ret == NSTACKX_EAGAIN) {
1282 continue;
1283 }
1284 SemWait(&session->sendThreadPara[threadIdx].semNewCycle);
1285 }
1286 if (ret < 0 && ret != NSTACKX_EAGAIN) {
1287 PostFatalEvent(session);
1288 }
1289 DestroyIovList(&unsent, session, threadIdx);
1290 return NULL;
1291 }
1292
CloseAddiSendThread(struct DFileSession * session)1293 static void CloseAddiSendThread(struct DFileSession *session)
1294 {
1295 if (session->sessionType == DFILE_SESSION_TYPE_SERVER || session->clientSendThreadNum <= 1) {
1296 return;
1297 }
1298 session->addiSenderCloseFlag = NSTACKX_TRUE;
1299 for (uint16_t i = 0; i < session->clientSendThreadNum - 1; i++) {
1300 SemPost(&session->sendThreadPara[i].sendWait);
1301 SemPost(&session->sendThreadPara[i].semNewCycle);
1302 PthreadJoin(session->sendThreadPara[i].senderTid, NULL);
1303 SemDestroy(&session->sendThreadPara[i].sendWait);
1304 SemDestroy(&session->sendThreadPara[i].semNewCycle);
1305 }
1306 }
1307
ErrorHandle(struct DFileSession * session,uint16_t cnt)1308 static void ErrorHandle(struct DFileSession *session, uint16_t cnt)
1309 {
1310 while (cnt > 0) {
1311 SemPost(&session->sendThreadPara[cnt - 1].sendWait);
1312 SemPost(&session->sendThreadPara[cnt - 1].semNewCycle);
1313 PthreadJoin(session->sendThreadPara[cnt - 1].senderTid, NULL);
1314 SemDestroy(&session->sendThreadPara[cnt - 1].sendWait);
1315 SemDestroy(&session->sendThreadPara[cnt - 1].semNewCycle);
1316 cnt--;
1317 }
1318 }
1319
CreateAddiSendThread(struct DFileSession * session)1320 static int32_t CreateAddiSendThread(struct DFileSession *session)
1321 {
1322 uint16_t i;
1323 if (session->sessionType == DFILE_SESSION_TYPE_SERVER || session->clientSendThreadNum <= 1) {
1324 return NSTACKX_EOK;
1325 }
1326 session->addiSenderCloseFlag = NSTACKX_FALSE;
1327 for (i = 0; i < session->clientSendThreadNum - 1; i++) {
1328 SendThreadCtx *sendThreadCtx = (SendThreadCtx *)calloc(1, sizeof(SendThreadCtx));
1329 if (sendThreadCtx == NULL) {
1330 goto L_ERR_SENDER_THREAD;
1331 }
1332
1333 if (SemInit(&session->sendThreadPara[i].sendWait, 0, 0) != 0) {
1334 free(sendThreadCtx);
1335 goto L_ERR_SENDER_THREAD;
1336 }
1337
1338 if (SemInit(&session->sendThreadPara[i].semNewCycle, 0, 0) != 0) {
1339 SemDestroy(&session->sendThreadPara[i].sendWait);
1340 free(sendThreadCtx);
1341 goto L_ERR_SENDER_THREAD;
1342 }
1343
1344 sendThreadCtx->session = session;
1345 sendThreadCtx->threadIdx = i;
1346 if (PthreadCreate(&(session->sendThreadPara[i].senderTid), NULL, DFileAddiSenderHandle, sendThreadCtx)) {
1347 DFILE_LOGE(TAG, "Create sender thread failed");
1348 SemDestroy(&session->sendThreadPara[i].sendWait);
1349 SemDestroy(&session->sendThreadPara[i].semNewCycle);
1350 free(sendThreadCtx);
1351 goto L_ERR_SENDER_THREAD;
1352 }
1353 }
1354 return NSTACKX_EOK;
1355
1356 L_ERR_SENDER_THREAD:
1357 session->addiSenderCloseFlag = NSTACKX_TRUE;
1358 ErrorHandle(session, i);
1359 return NSTACKX_EFAILED;
1360 }
1361
UpdatePeerinfoQdiscInfo(PeerInfo * peerInfo,uint32_t qDiscLeft)1362 static void UpdatePeerinfoQdiscInfo(PeerInfo *peerInfo, uint32_t qDiscLeft)
1363 {
1364 if (peerInfo->qdiscMinLeft == 0) {
1365 peerInfo->qdiscMinLeft = (uint16_t)qDiscLeft;
1366 } else if (peerInfo->qdiscMinLeft > qDiscLeft) {
1367 peerInfo->qdiscMinLeft = (uint16_t)qDiscLeft;
1368 }
1369 if (peerInfo->qdiscMaxLeft < qDiscLeft) {
1370 peerInfo->qdiscMaxLeft = (uint16_t)qDiscLeft;
1371 }
1372 peerInfo->qdiscSearchNum++;
1373 peerInfo->qdiscAveLeft += qDiscLeft;
1374 }
1375
UpdatePeerinfoAmendSendrateByQdisc(PeerInfo * peerInfo)1376 static void UpdatePeerinfoAmendSendrateByQdisc(PeerInfo *peerInfo)
1377 {
1378 uint32_t qDiscLeft;
1379 uint32_t qDiscLeftSendRate;
1380
1381 if (GetQdiscLen(peerInfo->localInterfaceName, ROOT_QUEUE, &qDiscLeft) == NSTACKX_EOK) {
1382 if (peerInfo->mtuInuse == 0) {
1383 return;
1384 }
1385 uint32_t mtuNumInOneFrame = peerInfo->dataFrameSize / peerInfo->mtuInuse;
1386 if (mtuNumInOneFrame == 0) {
1387 mtuNumInOneFrame = 1;
1388 }
1389 UpdatePeerinfoQdiscInfo(peerInfo, qDiscLeft);
1390 qDiscLeftSendRate = qDiscLeft / mtuNumInOneFrame;
1391 if (peerInfo->sendRate >= qDiscLeftSendRate) {
1392 peerInfo->amendSendRate = (int32_t)qDiscLeftSendRate;
1393 } else {
1394 peerInfo->amendSendRate = peerInfo->sendRate;
1395 }
1396 }
1397 return;
1398 }
1399
UpdateInfoAfterSend(DFileSession * session,PeerInfo * peerInfo,int32_t curAmendSendRate,struct timespec * before,uint8_t socketIndex)1400 static void UpdateInfoAfterSend(DFileSession *session, PeerInfo *peerInfo, int32_t curAmendSendRate,
1401 struct timespec *before, uint8_t socketIndex)
1402 {
1403 session->cycleRunning[socketIndex] = NSTACKX_FALSE;
1404 if (curAmendSendRate > 0 && peerInfo->intervalSendCount >= (uint32_t)curAmendSendRate) {
1405 if (peerInfo->decreaseStatus == 1) {
1406 peerInfo->overRun = 0;
1407 peerInfo->decreaseStatus = 0;
1408 }
1409 peerInfo->overRun = CheckElapseTime(before, peerInfo->overRun);
1410 if (peerInfo->overRun == 0) {
1411 session->sleepTimes++;
1412 }
1413 }
1414 peerInfo->intervalSendCount = 0;
1415
1416 UpdatePeerinfoAmendSendrateByQdisc(peerInfo);
1417 }
1418
DFileSessionSendFrame(DFileSession * session,QueueNode ** preQueueNode,List * unsent,struct timespec * before,uint8_t socketIndex)1419 static int32_t DFileSessionSendFrame(DFileSession *session, QueueNode **preQueueNode, List *unsent,
1420 struct timespec *before, uint8_t socketIndex)
1421 {
1422 int32_t ret;
1423
1424 if (CapsTcp(session) && !ListIsEmpty(unsent)) {
1425 session->sendRemain = 1;
1426 ret = SendDataFrame(session, unsent, (uint32_t)(session->clientSendThreadNum - 1), socketIndex);
1427 session->sendRemain = 0;
1428 if ((ret == NSTACKX_EFAILED || ret == NSTACKX_EAGAIN) || session->closeFlag) {
1429 DFILE_LOGI(TAG, "ret is %d", ret);
1430 return ret;
1431 }
1432 }
1433
1434 ret = SendOutboundFrame(session, preQueueNode);
1435 if (session->sessionType != DFILE_SESSION_TYPE_CLIENT) {
1436 return ret;
1437 }
1438 PeerInfo *peerInfo = ClientGetPeerInfoBySocketIndex(socketIndex, session);
1439 if (peerInfo == NULL) {
1440 DFILE_LOGE(TAG, "can't get valid peerinfo");
1441 return NSTACKX_EFAILED;
1442 }
1443 if (peerInfo->mtuInuse == 0) {
1444 return ret;
1445 }
1446
1447 if (ret == NSTACKX_EOK) {
1448 if (!session->cycleRunning[socketIndex]) {
1449 session->cycleRunning[socketIndex] = NSTACKX_TRUE;
1450 }
1451 ret = SendDataFrame(session, unsent, (uint32_t)(session->clientSendThreadNum - 1), socketIndex);
1452 }
1453 int32_t curAmendSendRate = peerInfo->amendSendRate;
1454 DFileSendCalculateRate(session, peerInfo);
1455
1456 if ((ret == NSTACKX_EFAILED || ret == NSTACKX_EAGAIN) || session->closeFlag) {
1457 DFILE_LOGI(TAG, "ret is %d and peerInfo->intervalSendCount is %u", ret, peerInfo->intervalSendCount);
1458 return ret;
1459 }
1460
1461 UpdateInfoAfterSend(session, peerInfo, curAmendSendRate, before, socketIndex);
1462
1463 for (uint16_t i = 0; i < session->clientSendThreadNum - 1; i++) {
1464 SemPost(&session->sendThreadPara[i].semNewCycle);
1465 }
1466 return ret;
1467 }
1468
WaitNewSendData(const QueueNode * queueNode,const List * unsent,DFileSession * session,uint8_t socketIndex)1469 static void WaitNewSendData(const QueueNode *queueNode, const List *unsent, DFileSession *session, uint8_t socketIndex)
1470 {
1471 if (queueNode == NULL && ListIsEmpty(unsent) && !FileManagerHasPendingData(session->fileManager) &&
1472 !session->outboundQueueSize) {
1473 NSTACKX_ATOM_FETCH_INC(&session->noPendingDataTimes);
1474 SemWait(&session->outboundQueueWait[socketIndex]);
1475 }
1476 }
1477
DFileSenderPre(DFileSession * session,uint8_t socketIndex)1478 static void DFileSenderPre(DFileSession *session, uint8_t socketIndex)
1479 {
1480 SetSendThreadName((uint32_t)(session->clientSendThreadNum + socketIndex - 1));
1481 DFileSenderUpdateMeasureTime(session, socketIndex);
1482 SetMaximumPriorityForThread();
1483 uint32_t pos = (uint32_t)(session->clientSendThreadNum - 1 + socketIndex + POS_SEND_THERD_START);
1484 if (pos >= POS_SEND_THERD_START + NSTACKX_MAX_CLIENT_SEND_THREAD_NUM) {
1485 return;
1486 }
1487
1488 SetTidToBindInfo(session, pos);
1489 }
1490
DFileSenderClose(DFileSession * session,QueueNode * queueNode,List * unsent,void * arg)1491 void DFileSenderClose(DFileSession *session, QueueNode *queueNode, List *unsent, void *arg)
1492 {
1493 DFILE_LOGI(TAG, "DFileSendCalculateRate: total send block num %llu.", session->totalSendBlocks);
1494 CloseAddiSendThread(session);
1495 DestroyIovList(unsent, session, session->clientSendThreadNum - 1U);
1496 DestroyQueueNode(queueNode);
1497 free(arg);
1498 return;
1499 }
1500
DFileSenderHandle(void * arg)1501 void *DFileSenderHandle(void *arg)
1502 {
1503 DFileSession *session = ((SenderThreadPara*)arg)->session;
1504 uint8_t socketIndex = ((SenderThreadPara*)arg)->socketIndex;
1505 QueueNode *queueNode = NULL;
1506 List unsent;
1507 int32_t ret = NSTACKX_EOK;
1508 struct timespec before;
1509 uint8_t canWrite = NSTACKX_FALSE;
1510 uint32_t socketWaitMs = GetSocketWaitMs(session->clientSendThreadNum);
1511 uint8_t isBind = NSTACKX_FALSE;
1512
1513 if (CreateAddiSendThread(session) != NSTACKX_EOK) {
1514 PostFatalEvent(session);
1515 return NULL;
1516 }
1517 ListInitHead(&unsent);
1518 DFileSenderPre(session, socketIndex);
1519 while (!session->closeFlag) {
1520 WaitNewSendData(queueNode, &unsent, session, socketIndex);
1521 if (session->closeFlag) {
1522 break;
1523 }
1524 if (!session->cycleRunning[socketIndex]) {
1525 ClockGetTime(CLOCK_MONOTONIC, &before);
1526 }
1527 if (ret == NSTACKX_EAGAIN) {
1528 ret = WaitSocketEvent(NULL, session->socket[socketIndex]->sockfd, socketWaitMs, NULL, &canWrite);
1529 if (ret != NSTACKX_EOK || session->closeFlag) {
1530 break;
1531 }
1532 if (!canWrite) {
1533 ret = NSTACKX_EAGAIN;
1534 continue;
1535 }
1536 }
1537 if (session->sessionType == DFILE_SESSION_TYPE_CLIENT && isBind == NSTACKX_FALSE &&
1538 session->transFlag == NSTACKX_TRUE) {
1539 BindClientSendThreadToTargetCpu(0);
1540 isBind = NSTACKX_TRUE;
1541 }
1542 ret = DFileSessionSendFrame(session, &queueNode, &unsent, &before, socketIndex);
1543 if (ret < 0 && ret != NSTACKX_EAGAIN) {
1544 PeerShuttedEvent();
1545 PostFatalEvent(session);
1546 break;
1547 }
1548 }
1549 DFileSenderClose(session, queueNode, &unsent, arg);
1550 return NULL;
1551 }
1552
ClearDFileFrameList(List * head)1553 static void ClearDFileFrameList(List *head)
1554 {
1555 if (head == NULL || ListIsEmpty(head)) {
1556 return;
1557 }
1558 List *tmp = NULL;
1559 List *pos = NULL;
1560 LIST_FOR_EACH_SAFE(pos, tmp, head) {
1561 QueueNode *node = (QueueNode *)pos;
1562 ListRemoveNode(&node->list);
1563 DestroyQueueNode(node);
1564 }
1565 }
1566
CheckDfileType(uint8_t type)1567 static inline int32_t CheckDfileType(uint8_t type)
1568 {
1569 if (type >= NSTACKX_DFILE_TYPE_MAX || type < NSTACKX_DFILE_FILE_HEADER_FRAME) {
1570 return NSTACKX_EFAILED;
1571 }
1572 return NSTACKX_EOK;
1573 }
1574
CheckSessionType(DFileSessionType type)1575 static inline int32_t CheckSessionType(DFileSessionType type)
1576 {
1577 if (type < DFILE_SESSION_TYPE_CLIENT || type > DFILE_SESSION_TYPE_SERVER) {
1578 return NSTACKX_EFAILED;
1579 }
1580 return NSTACKX_EOK;
1581 }
1582
ProcessDFileFrameList(DFileSession * session,List * head)1583 static void ProcessDFileFrameList(DFileSession *session, List *head)
1584 {
1585 QueueNode *queueNode = NULL;
1586 DFileFrame *dFileFrame = NULL;
1587 struct sockaddr_in *peerAddr = NULL;
1588 int32_t handleFrameRet = NSTACKX_EOK;
1589 while (!ListIsEmpty(head) && !session->closeFlag) {
1590 queueNode = (QueueNode *)ListPopFront(head);
1591 if (queueNode == NULL) {
1592 continue;
1593 }
1594 dFileFrame = (DFileFrame *)(queueNode->frame);
1595 peerAddr = &queueNode->peerAddr;
1596 uint8_t type = dFileFrame->header.type;
1597
1598 if (ntohs(dFileFrame->header.length) > NSTACKX_MAX_FRAME_SIZE - sizeof(DFileFrameHeader)) {
1599 DFILE_LOGE(TAG, "header length %u is too big", ntohs(dFileFrame->header.length));
1600 DestroyQueueNode(queueNode);
1601 continue;
1602 }
1603 if (CheckDfileType(dFileFrame->header.type) != NSTACKX_EOK ||
1604 CheckSessionType(session->sessionType) != NSTACKX_EOK) {
1605 handleFrameRet = NSTACKX_EFAILED;
1606 } else if (dFileFrame->header.type == NSTACKX_DFILE_SETTING_FRAME) {
1607 DFileSessionHandleSetting(session, dFileFrame, peerAddr, queueNode->socketIndex);
1608 } else if (dFileFrame->header.type == NSTACKX_DFILE_RST_FRAME &&
1609 dFileFrame->header.transId == 0) {
1610 DFileSessionHandleRst(session, dFileFrame, peerAddr);
1611 } else if (dFileFrame->header.type == NSTACKX_DFILE_FILE_BACK_PRESSURE_FRAME) {
1612 DFileSessionHandleBackPressure(session, dFileFrame, peerAddr);
1613 } else {
1614 /*
1615 * For NSTACKX_DFILE_FILE_DATA_FRAME, "dFileFrame" may be free when handling failed, and become invalid.
1616 */
1617 handleFrameRet = DFileSessionHandleFrame(session, dFileFrame, peerAddr);
1618 }
1619
1620 if (handleFrameRet != NSTACKX_EOK || type != NSTACKX_DFILE_FILE_DATA_FRAME) {
1621 /* For FILE_DATA frame, the memory is passed to file manager. */
1622 free(queueNode->frame);
1623 queueNode->frame = NULL;
1624 }
1625 free(queueNode);
1626 }
1627 ClearDFileFrameList(head);
1628 }
1629
ReadEventHandle(void * arg)1630 static void ReadEventHandle(void *arg)
1631 {
1632 DFileSession *session = arg;
1633 List newHead;
1634 ListInitHead(&newHead);
1635 struct timespec before, now;
1636 if (!session->partReadFlag) {
1637 NSTACKX_ATOM_FETCH_DEC(&session->unprocessedReadEventCount);
1638 } else {
1639 ClockGetTime(CLOCK_MONOTONIC, &before);
1640 }
1641 while (session->inboundQueueSize && !session->closeFlag) {
1642 if (session->partReadFlag) {
1643 ClockGetTime(CLOCK_MONOTONIC, &now);
1644 if (GetTimeDiffMs(&now, &before) >= DEFAULT_WAIT_TIME_MS) {
1645 break;
1646 }
1647 }
1648 if (PthreadMutexLock(&session->inboundQueueLock) != 0) {
1649 DFILE_LOGE(TAG, "PthreadMutexLock error");
1650 return;
1651 }
1652 ListMove(&session->inboundQueue, &newHead);
1653 session->recvBlockNumInner += session->inboundQueueSize;
1654 session->inboundQueueSize = 0;
1655 if (PthreadMutexUnlock(&session->inboundQueueLock) != 0) {
1656 DFILE_LOGE(TAG, "PthreadMutexUnlock error");
1657 break;
1658 }
1659 ProcessDFileFrameList(session, &newHead);
1660 }
1661 ClearDFileFrameList(&newHead);
1662 }
1663
DFileAddInboundQueue(DFileSession * session,const uint8_t * frame,size_t frameLength,struct sockaddr_in * peerAddr,uint8_t socketIndex)1664 static int32_t DFileAddInboundQueue(DFileSession *session, const uint8_t *frame, size_t frameLength,
1665 struct sockaddr_in *peerAddr, uint8_t socketIndex)
1666 {
1667 if (session->inboundQueueSize > MAX_RECVBUF_COUNT) {
1668 if (session->inboundQueueSize % MAX_NOMEM_PRINT == 0) {
1669 DFILE_LOGI(TAG, "no mem inboundQueueSize:%llu", session->inboundQueueSize);
1670 }
1671 return NSTACKX_ENOMEM;
1672 }
1673 QueueNode *queueNode = CreateQueueNode(frame, frameLength, peerAddr, socketIndex);
1674 if (queueNode == NULL) {
1675 return NSTACKX_ENOMEM;
1676 }
1677
1678 if (PthreadMutexLock(&session->inboundQueueLock) != 0) {
1679 DestroyQueueNode(queueNode);
1680 return NSTACKX_EFAILED;
1681 }
1682 ListInsertTail(&session->inboundQueue, &queueNode->list);
1683 session->inboundQueueSize++;
1684 session->recvBlockNumDirect++;
1685
1686 if (PthreadMutexUnlock(&session->inboundQueueLock) != 0) {
1687 /* queue node is pushed to list, don't need destroy here. */
1688 return NSTACKX_EFAILED;
1689 }
1690
1691 return NSTACKX_EOK;
1692 }
1693
DFileReceiverUpdateSessionMeasureTime(DFileSession * session)1694 static void DFileReceiverUpdateSessionMeasureTime(DFileSession *session)
1695 {
1696 if (session->sessionType == DFILE_SESSION_TYPE_SERVER) {
1697 ClockGetTime(CLOCK_MONOTONIC, &session->measureBefore);
1698 }
1699 }
1700
BindServerRecvThreadToTargetCpu(DFileSession * session)1701 static void BindServerRecvThreadToTargetCpu(DFileSession *session)
1702 {
1703 int32_t cpu;
1704 int32_t cpus = GetCpuNum();
1705 if (session->sessionType != DFILE_SESSION_TYPE_SERVER) {
1706 cpu = CPU_IDX_2;
1707 StartThreadBindCore(cpu);
1708 return;
1709 }
1710 if (cpus >= FIRST_CPU_NUM_LEVEL) {
1711 return;
1712 } else if (cpus >= SECOND_CPU_NUM_LEVEL) {
1713 cpu = CPU_IDX_1;
1714 } else if (cpus >= THIRD_CPU_NUM_LEVEL) {
1715 cpu = CPU_IDX_0;
1716 } else {
1717 return;
1718 }
1719 StartThreadBindCore(cpu);
1720 }
1721
PostReadEventToMainLoop(DFileSession * session)1722 static void PostReadEventToMainLoop(DFileSession *session)
1723 {
1724 if (NSTACKX_ATOM_FETCH(&(session->unprocessedReadEventCount)) >= MAX_UNPROCESSED_READ_EVENT_COUNT) {
1725 return;
1726 }
1727 NSTACKX_ATOM_FETCH_INC(&session->unprocessedReadEventCount);
1728 if (PostEvent(&session->eventNodeChain, session->epollfd, ReadEventHandle, session) != NSTACKX_EOK) {
1729 DFILE_LOGE(TAG, "post read event failed");
1730 NSTACKX_ATOM_FETCH_DEC(&session->unprocessedReadEventCount);
1731 session->mainLoopActiveReadFlag = NSTACKX_TRUE;
1732 } else {
1733 session->mainLoopActiveReadFlag = NSTACKX_FALSE;
1734 }
1735 }
1736
DFileSessionHandleReadBuffer(DFileSession * session,const uint8_t * buf,size_t bufLen,struct sockaddr_in * peerAddr,uint8_t socketIndex)1737 int32_t DFileSessionHandleReadBuffer(DFileSession *session, const uint8_t *buf, size_t bufLen,
1738 struct sockaddr_in *peerAddr, uint8_t socketIndex)
1739 {
1740 DFileFrame *dFileFrame = NULL;
1741 if (DecodeDFileFrame(buf, bufLen, &dFileFrame) != NSTACKX_EOK) {
1742 /* discard packet with non-zero trans id during cancel stage */
1743 DFILE_LOGE(TAG, "drop malformed frame");
1744 return NSTACKX_EOK;
1745 }
1746
1747 int32_t ret = DFileAddInboundQueue(session, buf, bufLen, peerAddr, socketIndex);
1748 if (ret == NSTACKX_ENOMEM) {
1749 return NSTACKX_EOK;
1750 }
1751 if (ret != NSTACKX_EOK) {
1752 DFILE_LOGE(TAG, "frame add in bound queue failed :%d", ret);
1753 return NSTACKX_EFAILED;
1754 }
1755 PostReadEventToMainLoop(session);
1756 DFileRecvCalculateRate(session, dFileFrame, peerAddr);
1757 return NSTACKX_EOK;
1758 }
1759
DFileRecverPre(DFileSession * session)1760 static void DFileRecverPre(DFileSession *session)
1761 {
1762 SetThreadName(DFFILE_RECV_THREAD_NAME);
1763 DFileReceiverUpdateSessionMeasureTime(session);
1764 SetMaximumPriorityForThread();
1765 SetTidToBindInfo(session, POS_RECV_THERD_START);
1766 }
1767
RcverWaitSocket(DFileSession * session,uint8_t * canRead)1768 int32_t RcverWaitSocket(DFileSession *session, uint8_t *canRead)
1769 {
1770 if (session->acceptFlag == 0) {
1771 return WaitSocketEvent(session, session->socket[0]->sockfd, DEFAULT_WAIT_TIME_MS, canRead, NULL);
1772 } else {
1773 return WaitSocketEvent(session, session->acceptSocket->sockfd, DEFAULT_WAIT_TIME_MS, canRead, NULL);
1774 }
1775 }
1776
DFileSocketRecv(DFileSession * session)1777 int32_t DFileSocketRecv(DFileSession *session)
1778 {
1779 return DFileSocketRecvSP(session);
1780 }
1781
DFileAcceptSocket(DFileSession * session)1782 int32_t DFileAcceptSocket(DFileSession *session)
1783 {
1784 session->acceptSocket = AcceptSocket(session->socket[0]);
1785 if (session->acceptSocket == NULL) {
1786 DFILE_LOGI(TAG, "accept socket failed");
1787 return NSTACKX_EFAILED;
1788 } else {
1789 DFILE_LOGI(TAG, "accept socket success");
1790 session->acceptFlag = 1;
1791 SetTcpKeepAlive(session->acceptSocket->sockfd);
1792 }
1793
1794 AcceptSocketEvent();
1795
1796 return NSTACKX_EOK;
1797 }
1798
DFileReceiverHandle(void * arg)1799 void *DFileReceiverHandle(void *arg)
1800 {
1801 DFileSession *session = arg;
1802 uint8_t canRead = NSTACKX_FALSE;
1803 int32_t ret = NSTACKX_EAGAIN;
1804 uint8_t isBind = NSTACKX_FALSE;
1805
1806 DFILE_LOGI(TAG, "recv thread start");
1807 DFileRecverPre(session);
1808 while (!session->closeFlag) {
1809 if (ret == NSTACKX_EAGAIN || !canRead) {
1810 ret = RcverWaitSocket(session, &canRead);
1811 if (ret != NSTACKX_EOK || session->closeFlag) {
1812 break;
1813 }
1814 }
1815 if (!canRead) {
1816 continue;
1817 }
1818 if (isBind == NSTACKX_FALSE && session->transFlag == NSTACKX_TRUE) {
1819 BindServerRecvThreadToTargetCpu(session);
1820 isBind = NSTACKX_TRUE;
1821 }
1822
1823 ret = DFileSocketRecv(session);
1824 if (ret != NSTACKX_EAGAIN && ret != NSTACKX_EOK) {
1825 PeerShuttedEvent();
1826 break;
1827 }
1828 }
1829 DFILE_LOGI(TAG, "Total recv blocks: direct %llu inner %llu", session->recvBlockNumDirect,
1830 session->recvBlockNumInner);
1831 if (ret < 0 && ret != NSTACKX_EAGAIN && ret != NSTACKX_PEER_CLOSE) {
1832 PostFatalEvent(session);
1833 }
1834
1835 DFILE_LOGI(TAG, "session %u Exit receiver thread ret %d", session->sessionId, ret);
1836 return NULL;
1837 }
1838
DFileControlHandle(void * arg)1839 void *DFileControlHandle(void *arg)
1840 {
1841 SetThreadName(DFFILE_CONTROL_THREAD_NAME);
1842 DFileSession *session = arg;
1843 if (session->sessionType == DFILE_SESSION_TYPE_CLIENT) {
1844 DFileSenderControlHandle(session);
1845 } else {
1846 DFileReceiverControlHandle(session);
1847 }
1848 return NULL;
1849 }
1850
RealPathFileName(FileListInfo * fileListInfo)1851 static int32_t RealPathFileName(FileListInfo *fileListInfo)
1852 {
1853 uint32_t i;
1854 int32_t ret = NSTACKX_EOK;
1855 for (i = 0; i < fileListInfo->fileNum; i++) {
1856 char *tmpFileName = fileListInfo->files[i];
1857 char *tmpFileNameRes = realpath(tmpFileName, NULL);
1858 if (tmpFileNameRes == NULL) {
1859 ret = NSTACKX_EFAILED;
1860 break;
1861 }
1862 fileListInfo->files[i] = tmpFileNameRes;
1863 free(tmpFileName);
1864 }
1865 if (ret != NSTACKX_EOK) {
1866 DFILE_LOGE(TAG, "realpath failed");
1867 }
1868 return ret;
1869 }
1870
FreeTransFileListInfo(FileListInfo * fileListInfo)1871 static void FreeTransFileListInfo(FileListInfo *fileListInfo)
1872 {
1873 free(fileListInfo->files);
1874 fileListInfo->files = NULL;
1875 if (fileListInfo->remotePath != NULL) {
1876 free(fileListInfo->remotePath);
1877 fileListInfo->remotePath = NULL;
1878 }
1879 free(fileListInfo);
1880 }
1881
DFileStartTransInner(DFileSession * session,FileListInfo * fileListInfo)1882 static int32_t DFileStartTransInner(DFileSession *session, FileListInfo *fileListInfo)
1883 {
1884 uint16_t transId = session->lastDFileTransId + 1;
1885 if (transId == 0) { /* overflow */
1886 transId = 1;
1887 }
1888
1889 PeerInfo *peerInfo = TransSelectPeerInfo(session);
1890 DFileTrans *trans = CreateTrans(transId, session, peerInfo, NSTACKX_TRUE);
1891 if (trans == NULL) {
1892 DFILE_LOGE(TAG, "CreateTrans error");
1893 return NSTACKX_ENOMEM;
1894 }
1895
1896 if (DFileTransSetMtu(trans, peerInfo->mtuInuse) != NSTACKX_EOK) {
1897 DFILE_LOGE(TAG, "set trans mtu failed");
1898 }
1899 if (RealPathFileName(fileListInfo) != NSTACKX_EOK) {
1900 DFileTransDestroy(trans);
1901 return NSTACKX_EFAILED;
1902 }
1903 int32_t ret = DFileTransSendFiles(trans, fileListInfo);
1904 if (ret != NSTACKX_EOK) {
1905 DFileTransDestroy(trans);
1906 DFILE_LOGE(TAG, "DFileTransSendFiles fail, error: %d", ret);
1907 return ret;
1908 }
1909 ret = DFileTransAddExtraInfo(trans, fileListInfo->pathType, fileListInfo->noticeFileNameType,
1910 fileListInfo->userData);
1911 if (ret != NSTACKX_EOK) {
1912 DFILE_LOGE(TAG, "DFileTransAddExtraInfo fail");
1913 DFileTransDestroy(trans);
1914 return NSTACKX_EFAILED;
1915 }
1916 trans->fileList->tarFlag = fileListInfo->tarFlag;
1917 trans->fileList->smallFlag = fileListInfo->smallFlag;
1918 trans->fileList->tarFile = fileListInfo->tarFile;
1919 trans->fileList->noSyncFlag = fileListInfo->noSyncFlag;
1920
1921 fileListInfo->userData = NULL;
1922 ListInsertTail(&(session->dFileTransChain), &(trans->list));
1923 session->lastDFileTransId = transId;
1924 if (fileListInfo->smallFlag == NSTACKX_TRUE) {
1925 session->smallListProcessingCnt++;
1926 } else {
1927 session->fileListProcessingCnt++;
1928 }
1929 /* Elements in ctx->fileListInfo->files[] are reused by dFileTranns, so don't need to free. */
1930 FreeTransFileListInfo(fileListInfo);
1931 return NSTACKX_EOK;
1932 }
1933
DFileStartTrans(DFileSession * session,FileListInfo * fileListInfo)1934 int32_t DFileStartTrans(DFileSession *session, FileListInfo *fileListInfo)
1935 {
1936 if (PthreadMutexLock(&session->transIdLock) != 0) {
1937 DFILE_LOGE(TAG, "pthread mutex lock error");
1938 return NSTACKX_EFAILED;
1939 }
1940 int32_t ret = DFileStartTransInner(session, fileListInfo);
1941 if (PthreadMutexUnlock(&session->transIdLock) != 0) {
1942 DFILE_LOGE(TAG, "pthread mutex unlock error");
1943 }
1944 return ret;
1945 }
1946
TerminateMainThreadInner(void * arg)1947 void TerminateMainThreadInner(void *arg)
1948 {
1949 DFileSession *session = (DFileSession *)arg;
1950 DFileSessionSetTerminateFlag(session);
1951 }
1952
StartDFileThreadsInner(DFileSession * session)1953 int32_t StartDFileThreadsInner(DFileSession *session)
1954 {
1955 if (PthreadCreate(&(session->tid), NULL, DFileMainLoop, session)) {
1956 DFILE_LOGE(TAG, "Create mainloop thread failed");
1957 goto L_ERR_MAIN_LOOP_THREAD;
1958 }
1959
1960 if (CreateSenderThread(session) != NSTACKX_EOK) {
1961 goto L_ERR_SENDER_THREAD;
1962 }
1963
1964 if (PthreadCreate(&(session->receiverTid), NULL, DFileReceiverHandle, session)) {
1965 DFILE_LOGE(TAG, "Create receiver thread failed");
1966 goto L_ERR_RECEIVER_THREAD;
1967 }
1968
1969 if (PthreadCreate(&(session->controlTid), NULL, DFileControlHandle, session)) {
1970 DFILE_LOGE(TAG, "Create control thread failed");
1971 goto L_ERR_CONTROL_THREAD;
1972 }
1973 return NSTACKX_EOK;
1974 L_ERR_CONTROL_THREAD:
1975 DFileSessionSetTerminateFlag(session);
1976 PthreadJoin(session->controlTid, NULL);
1977 session->receiverTid = INVALID_TID;
1978 L_ERR_RECEIVER_THREAD:
1979 DFileSessionSetTerminateFlag(session);
1980 PthreadJoin(session->senderTid[0], NULL);
1981 session->senderTid[0] = INVALID_TID;
1982 PostOutboundQueueWait(session);
1983 L_ERR_SENDER_THREAD:
1984 DFileSessionSetTerminateFlag(session);
1985 if (PostEvent(&session->eventNodeChain, session->epollfd, TerminateMainThreadInner, session) != NSTACKX_EOK) {
1986 DFILE_LOGE(TAG, "post terminate thread failed");
1987 }
1988 PthreadJoin(session->tid, NULL);
1989 session->tid = INVALID_TID;
1990 L_ERR_MAIN_LOOP_THREAD:
1991 return NSTACKX_EFAILED;
1992 }
1993
FileManagerMsgHandle(FileManagerMsgType msgType,int32_t errCode,void * context)1994 static void FileManagerMsgHandle(FileManagerMsgType msgType, int32_t errCode, void *context)
1995 {
1996 DFileSession *session = context;
1997 if (msgType == FILE_MANAGER_INNER_ERROR) {
1998 DFILE_LOGE(TAG, "Session (%u) fatal error -- File Manager error: %d", session->sessionId, errCode);
1999 PostFatalEvent(session);
2000 }
2001
2002 if (msgType == FILE_MANAGER_IN_PROGRESS) {
2003 NoticeSessionProgress(session);
2004 }
2005 }
2006
CreateFileManager(DFileSession * session,const uint8_t * key,uint32_t keyLen,uint8_t isSender,uint16_t connType)2007 int32_t CreateFileManager(DFileSession *session, const uint8_t *key, uint32_t keyLen, uint8_t isSender,
2008 uint16_t connType)
2009 {
2010 FileManagerMsgPara msgPara;
2011 if (session == NULL) {
2012 DFILE_LOGE(TAG, "invalid input");
2013 return NSTACKX_EINVAL;
2014 }
2015 if (isSender && (connType != CONNECT_TYPE_P2P && connType != CONNECT_TYPE_WLAN)) {
2016 DFILE_LOGE(TAG, "connType for sender is illagal");
2017 return NSTACKX_EINVAL;
2018 }
2019 if (keyLen > 0) {
2020 if ((keyLen != AES_128_KEY_LENGTH && keyLen != CHACHA20_KEY_LENGTH) || key == NULL) {
2021 DFILE_LOGE(TAG, "error key or key len");
2022 return NSTACKX_EFAILED;
2023 }
2024 if (!IsCryptoIncluded()) {
2025 DFILE_LOGE(TAG, "crypto is not opened");
2026 return NSTACKX_EFAILED;
2027 }
2028 }
2029 msgPara.epollfd = session->epollfd;
2030 msgPara.eventNodeChain = &session->eventNodeChain;
2031 msgPara.msgReceiver = FileManagerMsgHandle;
2032 msgPara.context = session;
2033 session->fileManager = FileManagerCreate(isSender, &msgPara, key, keyLen, connType);
2034 if (session->fileManager == NULL) {
2035 DFILE_LOGE(TAG, "create filemanager failed");
2036 return NSTACKX_EFAILED;
2037 }
2038 return NSTACKX_EOK;
2039 }
2040
DestroyReceiverPipe(DFileSession * session)2041 void DestroyReceiverPipe(DFileSession *session)
2042 {
2043 if (session->receiverPipe[PIPE_OUT] != INVALID_PIPE_DESC) {
2044 CloseDesc(session->receiverPipe[PIPE_OUT]);
2045 session->receiverPipe[PIPE_OUT] = INVALID_PIPE_DESC;
2046 }
2047 if (session->receiverPipe[PIPE_IN] != INVALID_PIPE_DESC) {
2048 CloseDesc(session->receiverPipe[PIPE_IN]);
2049 session->receiverPipe[PIPE_IN] = INVALID_PIPE_DESC;
2050 }
2051 }
2052
DFileSetEvent(void * softObj,DFileEventFunc func)2053 void DFileSetEvent(void *softObj, DFileEventFunc func)
2054 {
2055 g_dfileEventFunc = func;
2056 (void)softObj;
2057 }
2058
NSTACKX_DFileAssembleFunc(void * softObj,const DFileEvent * info)2059 void NSTACKX_DFileAssembleFunc(void *softObj, const DFileEvent *info)
2060 {
2061 if (g_dfileEventFunc != NULL) {
2062 g_dfileEventFunc(softObj, info);
2063 }
2064 }
2065