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