Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(6)

Side by Side Diff: media/cast/transport/rtp_sender/rtp_sender.cc

Issue 388663003: Cast: Reshuffle files under media/cast (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: missing includes Created 6 years, 5 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
OLDNEW
(Empty)
1 // Copyright 2013 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "media/cast/transport/rtp_sender/rtp_sender.h"
6
7 #include "base/big_endian.h"
8 #include "base/logging.h"
9 #include "base/rand_util.h"
10 #include "media/cast/transport/cast_transport_defines.h"
11 #include "media/cast/transport/pacing/paced_sender.h"
12
13 namespace media {
14 namespace cast {
15 namespace transport {
16
17 namespace {
18
19 // If there is only one referecne to the packet then copy the
20 // reference and return.
21 // Otherwise return a deep copy of the packet.
22 PacketRef FastCopyPacket(const PacketRef& packet) {
23 if (packet->HasOneRef())
24 return packet;
25 return make_scoped_refptr(new base::RefCountedData<Packet>(packet->data));
26 }
27
28 } // namespace
29
30 RtpSender::RtpSender(
31 base::TickClock* clock,
32 const scoped_refptr<base::SingleThreadTaskRunner>& transport_task_runner,
33 PacedSender* const transport)
34 : clock_(clock),
35 transport_(transport),
36 transport_task_runner_(transport_task_runner),
37 weak_factory_(this) {
38 // Randomly set sequence number start value.
39 config_.sequence_number = base::RandInt(0, 65535);
40 }
41
42 RtpSender::~RtpSender() {}
43
44 bool RtpSender::Initialize(const CastTransportRtpConfig& config) {
45 storage_.reset(new PacketStorage(config.stored_frames));
46 if (!storage_->IsValid()) {
47 return false;
48 }
49 config_.ssrc = config.ssrc;
50 config_.payload_type = config.rtp_payload_type;
51 packetizer_.reset(new RtpPacketizer(transport_, storage_.get(), config_));
52 return true;
53 }
54
55 void RtpSender::SendFrame(const EncodedFrame& frame) {
56 DCHECK(packetizer_);
57 packetizer_->SendFrameAsPackets(frame);
58 }
59
60 void RtpSender::ResendPackets(
61 const MissingFramesAndPacketsMap& missing_frames_and_packets,
62 bool cancel_rtx_if_not_in_list,
63 base::TimeDelta dedupe_window) {
64 DCHECK(storage_);
65 // Iterate over all frames in the list.
66 for (MissingFramesAndPacketsMap::const_iterator it =
67 missing_frames_and_packets.begin();
68 it != missing_frames_and_packets.end();
69 ++it) {
70 SendPacketVector packets_to_resend;
71 uint8 frame_id = it->first;
72 // Set of packets that the receiver wants us to re-send.
73 // If empty, we need to re-send all packets for this frame.
74 const PacketIdSet& missing_packet_set = it->second;
75
76 bool resend_all = missing_packet_set.find(kRtcpCastAllPacketsLost) !=
77 missing_packet_set.end();
78 bool resend_last = missing_packet_set.find(kRtcpCastLastPacket) !=
79 missing_packet_set.end();
80
81 const SendPacketVector* stored_packets = storage_->GetFrame8(frame_id);
82 if (!stored_packets)
83 continue;
84
85 for (SendPacketVector::const_iterator it = stored_packets->begin();
86 it != stored_packets->end(); ++it) {
87 const PacketKey& packet_key = it->first;
88 const uint16 packet_id = packet_key.second.second;
89
90 // Should we resend the packet?
91 bool resend = resend_all;
92
93 // Should we resend it because it's in the missing_packet_set?
94 if (!resend &&
95 missing_packet_set.find(packet_id) != missing_packet_set.end()) {
96 resend = true;
97 }
98
99 // If we were asked to resend the last packet, check if it's the
100 // last packet.
101 if (!resend && resend_last && (it + 1) == stored_packets->end()) {
102 resend = true;
103 }
104
105 if (resend) {
106 // Resend packet to the network.
107 VLOG(3) << "Resend " << static_cast<int>(frame_id) << ":"
108 << packet_id;
109 // Set a unique incremental sequence number for every packet.
110 PacketRef packet_copy = FastCopyPacket(it->second);
111 UpdateSequenceNumber(&packet_copy->data);
112 packets_to_resend.push_back(std::make_pair(packet_key, packet_copy));
113 } else if (cancel_rtx_if_not_in_list) {
114 transport_->CancelSendingPacket(it->first);
115 }
116 }
117 transport_->ResendPackets(packets_to_resend, dedupe_window);
118 }
119 }
120
121 void RtpSender::UpdateSequenceNumber(Packet* packet) {
122 // TODO(miu): This is an abstraction violation. This needs to be a part of
123 // the overall packet (de)serialization consolidation.
124 static const int kByteOffsetToSequenceNumber = 2;
125 base::BigEndianWriter big_endian_writer(
126 reinterpret_cast<char*>((&packet->front()) + kByteOffsetToSequenceNumber),
127 sizeof(uint16));
128 big_endian_writer.WriteU16(packetizer_->NextSequenceNumber());
129 }
130
131 } // namespace transport
132 } // namespace cast
133 } // namespace media
OLDNEW
« no previous file with comments | « media/cast/transport/rtp_sender/rtp_sender.h ('k') | media/cast/transport/transport/udp_transport.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698