OLD | NEW |
1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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/protobuf_video_writer.h" | 5 #include "remoting/protocol/protobuf_video_writer.h" |
6 | 6 |
| 7 #include "base/bind.h" |
7 #include "base/task.h" | 8 #include "base/task.h" |
| 9 #include "net/socket/stream_socket.h" |
| 10 #include "remoting/base/constants.h" |
8 #include "remoting/proto/video.pb.h" | 11 #include "remoting/proto/video.pb.h" |
9 #include "remoting/protocol/rtp_writer.h" | 12 #include "remoting/protocol/rtp_writer.h" |
10 #include "remoting/protocol/session.h" | 13 #include "remoting/protocol/session.h" |
11 #include "remoting/protocol/util.h" | 14 #include "remoting/protocol/util.h" |
12 | 15 |
13 namespace remoting { | 16 namespace remoting { |
14 namespace protocol { | 17 namespace protocol { |
15 | 18 |
16 ProtobufVideoWriter::ProtobufVideoWriter() { } | 19 ProtobufVideoWriter::ProtobufVideoWriter() { } |
17 | 20 |
18 ProtobufVideoWriter::~ProtobufVideoWriter() { } | 21 ProtobufVideoWriter::~ProtobufVideoWriter() { } |
19 | 22 |
20 void ProtobufVideoWriter::Init(protocol::Session* session) { | 23 void ProtobufVideoWriter::Init(protocol::Session* session, |
| 24 const InitializedCallback& callback) { |
| 25 initialized_callback_ = callback; |
| 26 |
| 27 session->CreateStreamChannel( |
| 28 kVideoChannelName, |
| 29 base::Bind(&ProtobufVideoWriter::OnChannelReady, base::Unretained(this))); |
| 30 } |
| 31 |
| 32 void ProtobufVideoWriter::OnChannelReady(const std::string& name, |
| 33 net::StreamSocket* socket) { |
| 34 DCHECK_EQ(name, std::string(kVideoChannelName)); |
| 35 if (!socket) { |
| 36 initialized_callback_.Run(false); |
| 37 return; |
| 38 } |
| 39 |
| 40 DCHECK(!channel_.get()); |
| 41 channel_.reset(socket); |
21 buffered_writer_ = new BufferedSocketWriter(); | 42 buffered_writer_ = new BufferedSocketWriter(); |
22 // TODO(sergeyu): Provide WriteFailedCallback for the buffered writer. | 43 // TODO(sergeyu): Provide WriteFailedCallback for the buffered writer. |
23 buffered_writer_->Init(session->video_channel(), NULL); | 44 buffered_writer_->Init(socket, NULL); |
| 45 |
| 46 initialized_callback_.Run(true); |
24 } | 47 } |
25 | 48 |
26 void ProtobufVideoWriter::Close() { | 49 void ProtobufVideoWriter::Close() { |
27 buffered_writer_->Close(); | 50 buffered_writer_->Close(); |
| 51 channel_.reset(); |
28 } | 52 } |
29 | 53 |
30 void ProtobufVideoWriter::ProcessVideoPacket(const VideoPacket* packet, | 54 void ProtobufVideoWriter::ProcessVideoPacket(const VideoPacket* packet, |
31 Task* done) { | 55 Task* done) { |
32 buffered_writer_->Write(SerializeAndFrameMessage(*packet), done); | 56 buffered_writer_->Write(SerializeAndFrameMessage(*packet), done); |
33 } | 57 } |
34 | 58 |
35 int ProtobufVideoWriter::GetPendingPackets() { | 59 int ProtobufVideoWriter::GetPendingPackets() { |
36 return buffered_writer_->GetBufferChunks(); | 60 return buffered_writer_->GetBufferChunks(); |
37 } | 61 } |
38 | 62 |
39 } // namespace protocol | 63 } // namespace protocol |
40 } // namespace remoting | 64 } // namespace remoting |
OLD | NEW |