ivica | 7f6a6fc | 2015-09-08 02:40:29 -0700 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (c) 2015 The WebRTC project authors. All Rights Reserved. |
| 3 | * |
| 4 | * Use of this source code is governed by a BSD-style license |
| 5 | * that can be found in the LICENSE file in the root of the source |
| 6 | * tree. An additional intellectual property rights grant can be found |
| 7 | * in the file PATENTS. All contributing project authors may |
| 8 | * be found in the AUTHORS file in the root of the source tree. |
| 9 | */ |
| 10 | |
kwiberg | bfefb03 | 2016-05-01 14:53:46 -0700 | [diff] [blame] | 11 | #include <memory> |
| 12 | |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 13 | #include "modules/rtp_rtcp/include/rtp_rtcp_defines.h" |
| 14 | #include "modules/rtp_rtcp/source/byte_io.h" |
| 15 | #include "modules/rtp_rtcp/source/rtp_format.h" |
| 16 | #include "modules/rtp_rtcp/source/rtp_utility.h" |
| 17 | #include "rtc_base/checks.h" |
| 18 | #include "test/layer_filtering_transport.h" |
ivica | 7f6a6fc | 2015-09-08 02:40:29 -0700 | [diff] [blame] | 19 | |
| 20 | namespace webrtc { |
| 21 | namespace test { |
| 22 | |
| 23 | LayerFilteringTransport::LayerFilteringTransport( |
eladalon | 413ee9a | 2017-08-22 04:02:52 -0700 | [diff] [blame] | 24 | SingleThreadedTaskQueueForTesting* task_queue, |
ivica | 7f6a6fc | 2015-09-08 02:40:29 -0700 | [diff] [blame] | 25 | const FakeNetworkPipe::Config& config, |
stefan | f116bd0 | 2015-10-27 08:29:42 -0700 | [diff] [blame] | 26 | Call* send_call, |
ivica | 7f6a6fc | 2015-09-08 02:40:29 -0700 | [diff] [blame] | 27 | uint8_t vp8_video_payload_type, |
| 28 | uint8_t vp9_video_payload_type, |
sprang | ce4aef1 | 2015-11-02 07:23:20 -0800 | [diff] [blame] | 29 | int selected_tl, |
minyue | 20c84cc | 2017-04-10 16:57:57 -0700 | [diff] [blame] | 30 | int selected_sl, |
| 31 | const std::map<uint8_t, MediaType>& payload_type_map) |
eladalon | 413ee9a | 2017-08-22 04:02:52 -0700 | [diff] [blame] | 32 | : DirectTransport(task_queue, config, send_call, payload_type_map), |
ivica | 7f6a6fc | 2015-09-08 02:40:29 -0700 | [diff] [blame] | 33 | vp8_video_payload_type_(vp8_video_payload_type), |
| 34 | vp9_video_payload_type_(vp9_video_payload_type), |
sprang | ce4aef1 | 2015-11-02 07:23:20 -0800 | [diff] [blame] | 35 | selected_tl_(selected_tl), |
| 36 | selected_sl_(selected_sl), |
| 37 | discarded_last_packet_(false) {} |
ivica | e78e2c7 | 2015-10-07 23:44:29 -0700 | [diff] [blame] | 38 | |
sprang | ce4aef1 | 2015-11-02 07:23:20 -0800 | [diff] [blame] | 39 | bool LayerFilteringTransport::DiscardedLastPacket() const { |
| 40 | return discarded_last_packet_; |
ivica | e78e2c7 | 2015-10-07 23:44:29 -0700 | [diff] [blame] | 41 | } |
ivica | 7f6a6fc | 2015-09-08 02:40:29 -0700 | [diff] [blame] | 42 | |
stefan | 1d8a506 | 2015-10-02 03:39:33 -0700 | [diff] [blame] | 43 | bool LayerFilteringTransport::SendRtp(const uint8_t* packet, |
| 44 | size_t length, |
| 45 | const PacketOptions& options) { |
sprang | ce4aef1 | 2015-11-02 07:23:20 -0800 | [diff] [blame] | 46 | if (selected_tl_ == -1 && selected_sl_ == -1) { |
ivica | 7f6a6fc | 2015-09-08 02:40:29 -0700 | [diff] [blame] | 47 | // Nothing to change, forward the packet immediately. |
stefan | 1d8a506 | 2015-10-02 03:39:33 -0700 | [diff] [blame] | 48 | return test::DirectTransport::SendRtp(packet, length, options); |
ivica | 7f6a6fc | 2015-09-08 02:40:29 -0700 | [diff] [blame] | 49 | } |
| 50 | |
| 51 | bool set_marker_bit = false; |
sprang | ce4aef1 | 2015-11-02 07:23:20 -0800 | [diff] [blame] | 52 | RtpUtility::RtpHeaderParser parser(packet, length); |
ivica | 7f6a6fc | 2015-09-08 02:40:29 -0700 | [diff] [blame] | 53 | RTPHeader header; |
danilchap | f6975f4 | 2015-12-28 10:18:46 -0800 | [diff] [blame] | 54 | parser.Parse(&header); |
sprang | ce4aef1 | 2015-11-02 07:23:20 -0800 | [diff] [blame] | 55 | |
kwiberg | 352444f | 2016-11-28 15:58:53 -0800 | [diff] [blame] | 56 | RTC_DCHECK_LE(length, IP_PACKET_SIZE); |
sprang | ce4aef1 | 2015-11-02 07:23:20 -0800 | [diff] [blame] | 57 | uint8_t temp_buffer[IP_PACKET_SIZE]; |
| 58 | memcpy(temp_buffer, packet, length); |
ivica | 7f6a6fc | 2015-09-08 02:40:29 -0700 | [diff] [blame] | 59 | |
| 60 | if (header.payloadType == vp8_video_payload_type_ || |
| 61 | header.payloadType == vp9_video_payload_type_) { |
| 62 | const uint8_t* payload = packet + header.headerLength; |
henrikg | 91d6ede | 2015-09-17 00:24:34 -0700 | [diff] [blame] | 63 | RTC_DCHECK_GT(length, header.headerLength); |
ivica | 7f6a6fc | 2015-09-08 02:40:29 -0700 | [diff] [blame] | 64 | const size_t payload_length = length - header.headerLength; |
henrikg | 91d6ede | 2015-09-17 00:24:34 -0700 | [diff] [blame] | 65 | RTC_DCHECK_GT(payload_length, header.paddingLength); |
ivica | 7f6a6fc | 2015-09-08 02:40:29 -0700 | [diff] [blame] | 66 | const size_t payload_data_length = payload_length - header.paddingLength; |
| 67 | |
| 68 | const bool is_vp8 = header.payloadType == vp8_video_payload_type_; |
kwiberg | bfefb03 | 2016-05-01 14:53:46 -0700 | [diff] [blame] | 69 | std::unique_ptr<RtpDepacketizer> depacketizer( |
ivica | 7f6a6fc | 2015-09-08 02:40:29 -0700 | [diff] [blame] | 70 | RtpDepacketizer::Create(is_vp8 ? kRtpVideoVp8 : kRtpVideoVp9)); |
| 71 | RtpDepacketizer::ParsedPayload parsed_payload; |
| 72 | if (depacketizer->Parse(&parsed_payload, payload, payload_data_length)) { |
sprang | ce4aef1 | 2015-11-02 07:23:20 -0800 | [diff] [blame] | 73 | const int temporal_idx = static_cast<int>( |
ivica | 7f6a6fc | 2015-09-08 02:40:29 -0700 | [diff] [blame] | 74 | is_vp8 ? parsed_payload.type.Video.codecHeader.VP8.temporalIdx |
sprang | ce4aef1 | 2015-11-02 07:23:20 -0800 | [diff] [blame] | 75 | : parsed_payload.type.Video.codecHeader.VP9.temporal_idx); |
| 76 | const int spatial_idx = static_cast<int>( |
ivica | 7f6a6fc | 2015-09-08 02:40:29 -0700 | [diff] [blame] | 77 | is_vp8 ? kNoSpatialIdx |
sprang | ce4aef1 | 2015-11-02 07:23:20 -0800 | [diff] [blame] | 78 | : parsed_payload.type.Video.codecHeader.VP9.spatial_idx); |
| 79 | if (selected_sl_ >= 0 && spatial_idx == selected_sl_ && |
ivica | 7f6a6fc | 2015-09-08 02:40:29 -0700 | [diff] [blame] | 80 | parsed_payload.type.Video.codecHeader.VP9.end_of_frame) { |
| 81 | // This layer is now the last in the superframe. |
| 82 | set_marker_bit = true; |
sprang | ce4aef1 | 2015-11-02 07:23:20 -0800 | [diff] [blame] | 83 | } else if ((selected_tl_ >= 0 && temporal_idx != kNoTemporalIdx && |
| 84 | temporal_idx > selected_tl_) || |
| 85 | (selected_sl_ >= 0 && spatial_idx != kNoSpatialIdx && |
ilnik | 68af10d | 2017-03-02 04:59:33 -0800 | [diff] [blame] | 86 | spatial_idx > selected_sl_)) { |
sprang | ce4aef1 | 2015-11-02 07:23:20 -0800 | [diff] [blame] | 87 | // Truncate packet to a padding packet. |
| 88 | length = header.headerLength + 1; |
| 89 | temp_buffer[0] |= (1 << 5); // P = 1. |
| 90 | temp_buffer[1] &= 0x7F; // M = 0. |
| 91 | discarded_last_packet_ = true; |
| 92 | temp_buffer[header.headerLength] = 1; // One byte of padding. |
ivica | 7f6a6fc | 2015-09-08 02:40:29 -0700 | [diff] [blame] | 93 | } |
| 94 | } else { |
| 95 | RTC_NOTREACHED() << "Parse error"; |
| 96 | } |
| 97 | } |
| 98 | |
ivica | 7f6a6fc | 2015-09-08 02:40:29 -0700 | [diff] [blame] | 99 | // We are discarding some of the packets (specifically, whole layers), so |
| 100 | // make sure the marker bit is set properly, and that sequence numbers are |
| 101 | // continuous. |
ivica | e78e2c7 | 2015-10-07 23:44:29 -0700 | [diff] [blame] | 102 | if (set_marker_bit) |
ivica | 7f6a6fc | 2015-09-08 02:40:29 -0700 | [diff] [blame] | 103 | temp_buffer[1] |= kRtpMarkerBitMask; |
ivica | 7f6a6fc | 2015-09-08 02:40:29 -0700 | [diff] [blame] | 104 | |
stefan | 1d8a506 | 2015-10-02 03:39:33 -0700 | [diff] [blame] | 105 | return test::DirectTransport::SendRtp(temp_buffer, length, options); |
ivica | 7f6a6fc | 2015-09-08 02:40:29 -0700 | [diff] [blame] | 106 | } |
| 107 | |
| 108 | } // namespace test |
| 109 | } // namespace webrtc |