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