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_retransmission.h"
17
18 #include "nstackx_dfile_session.h"
19 #include "nstackx_dfile_log.h"
20
21 #define TAG "nStackXDFile"
22
23 #define NSTACKX_BACK_PRESSURE_BYPASS_TIME 10
SendBackPressureFrame(DFileTrans * dFileTrans)24 static void SendBackPressureFrame(DFileTrans *dFileTrans)
25 {
26 PeerInfo *peerInfo = (PeerInfo *)(dFileTrans->context);
27 uint8_t buf[NSTACKX_DEFAULT_FRAME_SIZE];
28 size_t frameLen = 0;
29 uint32_t recvListAllSize;
30 uint32_t recvInnerAllSize;
31 uint32_t allSize;
32 uint32_t blockFrameSize = sizeof(BlockFrame) + dFileTrans->fileManager->maxFrameLength;
33
34 if (!CapsRecvFeedback(dFileTrans->session)) {
35 return;
36 }
37
38 if (dFileTrans->backPressureBypassCnt <= NSTACKX_BACK_PRESSURE_BYPASS_TIME) {
39 dFileTrans->backPressureBypassCnt++;
40 return;
41 }
42
43 if (GetFileBlockListSize(&dFileTrans->fileManager->taskList, &recvListAllSize, &recvInnerAllSize) != NSTACKX_EOK) {
44 dFileTrans->fileManager->errCode = FILE_MANAGER_EMUTEX;
45 NotifyFileManagerMsg(dFileTrans->fileManager, FILE_MANAGER_INNER_ERROR);
46 DFILE_LOGE(TAG, "failed to get GetFileBlockListSize");
47 return;
48 }
49 allSize = recvListAllSize + recvInnerAllSize;
50 uint32_t recvListWindowSize = (dFileTrans->fileManager->iowMaxRate * DFILE_KILOBYTES / NSTACKX_MILLI_TICKS)
51 * NSTACKX_ACK_INTERVAL * FILE_RECV_LIST_SLOW_START_RATE;
52 if (((allSize >= (uint32_t)(dFileTrans->fileManager->iowCount * FILE_RECV_LIST_IO_WRITE_THRESHOLD)) ||
53 (allSize >= (uint32_t)(dFileTrans->fileManager->maxRecvBlockListSize * FILE_RECV_LIST_IO_WRITE_THRESHOLD)) ||
54 (allSize * blockFrameSize >= recvListWindowSize)) &&
55 allSize > 0) {
56 dFileTrans->fileManager->recvListOverIo = 1;
57 } else {
58 dFileTrans->fileManager->recvListOverIo = 0;
59 }
60
61 EncodeBackPressFrame(buf, NSTACKX_DEFAULT_FRAME_SIZE, &frameLen, dFileTrans->fileManager->recvListOverIo);
62 int32_t ret = DFileWriteHandle(buf, frameLen, peerInfo);
63 if (ret != (int32_t)frameLen && ret != NSTACKX_EAGAIN) {
64 DFILE_LOGE(TAG, "send back pressure frame failed");
65 }
66 if (dFileTrans->fileManager->recvListOverIo == 1) {
67 DFILE_LOGI(TAG, "socket %hhu send back pressure fileManager->recvListOverIo %hhu allSize %u iowCount %llu",
68 peerInfo->socketIndex, dFileTrans->fileManager->recvListOverIo, allSize,
69 dFileTrans->fileManager->iowCount);
70 }
71 if (dFileTrans->fileManager->recvListOverIo == 1) {
72 ClockGetTime(CLOCK_MONOTONIC, &dFileTrans->ts);
73 }
74 }
75
SendFileDataAck(DFileTrans * dFileTrans,DFileReceiveState * nextState)76 void SendFileDataAck(DFileTrans *dFileTrans, DFileReceiveState *nextState)
77 {
78 ClockGetTime(CLOCK_MONOTONIC, &dFileTrans->heartBeatTs);
79 *nextState = STATE_RECEIVE_FILE_DATA_ONGOING;
80
81 SendBackPressureFrame(dFileTrans);
82 }
83