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 "stream_packetizer.h"
17
18 #include "stream_packet_header.h"
19
20 namespace Communication {
21 namespace SoftBus {
CalculateHeaderSize() const22 ssize_t StreamPacketizer::CalculateHeaderSize() const
23 {
24 auto total = StreamPacketHeader::STREAM_HEADER_SIZE;
25
26 return total;
27 }
28
CalculateExtSize(ssize_t extSize) const29 ssize_t StreamPacketizer::CalculateExtSize(ssize_t extSize) const
30 {
31 auto total = 0;
32
33 if (extSize > 0) {
34 total += TwoLevelsTlv::HEADER_LEN + TwoLevelsTlv::NUMS_LEN;
35 total += TwoLevelsTlv::HEADER_LEN + TwoLevelsTlv::NUMS_LEN;
36 total += Align(extSize, SHIFT);
37 total += TwoLevelsTlv::CHECK_SUM_LEN;
38 }
39
40 return total;
41 }
42
PacketizeStream()43 std::unique_ptr<char[]> StreamPacketizer::PacketizeStream()
44 {
45 dataSize_ = originData_->GetBufferLen();
46 hdrSize_ = CalculateHeaderSize();
47 extSize_ = CalculateExtSize(originData_->GetExtBufferLen());
48 auto data = std::make_unique<char[]>(hdrSize_ + extSize_ + dataSize_);
49
50 auto streamPktHeader = StreamPacketHeader(streamType_, extSize_ > 0, extSize_ + dataSize_,
51 originData_->GetStreamFrameInfo());
52 streamPktHeader.Packetize(data.get(), hdrSize_, 0);
53
54 TwoLevelsTlv tlv(originData_->GetExtBuffer(), originData_->GetExtBufferLen());
55 if (tlv.Packetize(data.get(), extSize_, hdrSize_) != 0) {
56 TRANS_LOGE(TRANS_STREAM, "packetize tlv failed");
57 return nullptr;
58 }
59
60 TRANS_LOGD(TRANS_STREAM,
61 "streamPktHeader version=%{public}d, subVersion=%{public}d, extFlag=%{public}d, streamType=%{public}d, "
62 "marker=%{public}d, flag=%{public}d, streamId=%{public}d(%{public}x), timestamp=%{public}u(%{public}x), "
63 "dataLen=%{public}u(%{public}x), seqNum=%{public}d(%{public}x), subSeqNum=%{public}d(%{public}x), "
64 "dataSize=%{public}zd, extSize=%{public}zd",
65 streamPktHeader.GetVersion(), streamPktHeader.GetSubVersion(), streamPktHeader.GetExtFlag(),
66 streamPktHeader.GetStreamType(), streamPktHeader.GetMarker(), streamPktHeader.GetFlag(),
67 streamPktHeader.GetStreamId(), streamPktHeader.GetStreamId(), streamPktHeader.GetTimestamp(),
68 streamPktHeader.GetTimestamp(), streamPktHeader.GetDataLen(), streamPktHeader.GetDataLen(),
69 streamPktHeader.GetSeqNum(), streamPktHeader.GetSeqNum(), streamPktHeader.GetSubSeqNum(),
70 streamPktHeader.GetSubSeqNum(), dataSize_, extSize_);
71
72 TRANS_LOGD(TRANS_STREAM,
73 "TLV version=%{public}d, num=%{public}d, extSize=%{public}zd, extLen=%{public}zd, checksum=%{public}u",
74 tlv.GetVersion(), tlv.GetTlvNums(), extSize_, tlv.GetExtLen(), tlv.GetCheckSum());
75
76 auto ret = memcpy_s(data.get() + hdrSize_ + extSize_, dataSize_, originData_->GetBuffer().get(),
77 originData_->GetBufferLen());
78 if (ret != 0) {
79 TRANS_LOGE(TRANS_STREAM, "Failed to memcpy data! ret=%{public}d", ret);
80 }
81
82 return data;
83 }
84 } // namespace SoftBus
85 } // namespace Communication
86