| OLD | NEW |
| 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "remoting/protocol/rtp_reader.h" | 5 #include "remoting/protocol/rtp_reader.h" |
| 6 | 6 |
| 7 #include "net/base/completion_callback.h" | 7 #include "net/base/completion_callback.h" |
| 8 #include "net/base/io_buffer.h" | 8 #include "net/base/io_buffer.h" |
| 9 | 9 |
| 10 namespace remoting { | 10 namespace remoting { |
| 11 namespace protocol { | 11 namespace protocol { |
| 12 | 12 |
| 13 RtpPacket::RtpPacket() {} | 13 RtpPacket::RtpPacket() {} |
| 14 RtpPacket::~RtpPacket() {} | 14 RtpPacket::~RtpPacket() {} |
| 15 | 15 |
| 16 // RtpReader class. | 16 // RtpReader class. |
| 17 RtpReader::RtpReader() {} | 17 RtpReader::RtpReader() {} |
| 18 RtpReader::~RtpReader() {} | 18 RtpReader::~RtpReader() {} |
| 19 | 19 |
| 20 void RtpReader::Init(net::Socket* socket, | 20 void RtpReader::Init(net::Socket* socket, |
| 21 OnMessageCallback* on_message_callback) { | 21 OnMessageCallback* on_message_callback) { |
| 22 on_message_callback_.reset(on_message_callback); | 22 on_message_callback_.reset(on_message_callback); |
| 23 SocketReaderBase::Init(socket); | 23 SocketReaderBase::Init(socket); |
| 24 } | 24 } |
| 25 | 25 |
| 26 void RtpReader::OnDataReceived(net::IOBuffer* buffer, int data_size) { | 26 void RtpReader::OnDataReceived(net::IOBuffer* buffer, int data_size) { |
| 27 RtpPacket packet; | 27 RtpPacket packet; |
| 28 int header_size = UnpackRtpHeader(reinterpret_cast<uint8*>(buffer->data()), | 28 int header_size = UnpackRtpHeader(reinterpret_cast<uint8*>(buffer->data()), |
| 29 data_size, &packet.header); | 29 data_size, packet.mutable_header()); |
| 30 if (header_size < 0) { | 30 if (header_size < 0) { |
| 31 LOG(WARNING) << "Received invalid RTP packet."; | 31 LOG(WARNING) << "Received invalid RTP packet."; |
| 32 return; | 32 return; |
| 33 } | 33 } |
| 34 packet.data = buffer; | 34 packet.mutable_payload()->Append( |
| 35 packet.payload = buffer->data() + header_size; | 35 buffer, buffer->data() + header_size, data_size - header_size); |
| 36 packet.payload_size = data_size - header_size; | |
| 37 | 36 |
| 38 on_message_callback_->Run(packet); | 37 on_message_callback_->Run(packet); |
| 39 } | 38 } |
| 40 | 39 |
| 41 } // namespace protocol | 40 } // namespace protocol |
| 42 } // namespace remoting | 41 } // namespace remoting |
| OLD | NEW |