// Copyright (c) 2012 The Chromium Authors. All rights reserved. // Use of this source code is governed by a BSD-style license that can be // found in the LICENSE file. #include "remoting/protocol/protobuf_video_writer.h" #include "base/bind.h" #include "net/socket/stream_socket.h" #include "remoting/base/constants.h" #include "remoting/proto/video.pb.h" #include "remoting/protocol/channel_factory.h" #include "remoting/protocol/session.h" #include "remoting/protocol/util.h" namespace remoting { namespace protocol { ProtobufVideoWriter::ProtobufVideoWriter() : channel_factory_(NULL) { } ProtobufVideoWriter::~ProtobufVideoWriter() { Close(); } void ProtobufVideoWriter::Init(protocol::Session* session, const InitializedCallback& callback) { channel_factory_ = session->GetTransportChannelFactory(); initialized_callback_ = callback; channel_factory_->CreateStreamChannel( kVideoChannelName, base::Bind(&ProtobufVideoWriter::OnChannelReady, base::Unretained(this))); } void ProtobufVideoWriter::OnChannelReady(scoped_ptr<net::StreamSocket> socket) { if (!socket.get()) { initialized_callback_.Run(false); return; } DCHECK(!channel_.get()); channel_ = socket.Pass(); // TODO(sergeyu): Provide WriteFailedCallback for the buffered writer. buffered_writer_.Init( channel_.get(), BufferedSocketWriter::WriteFailedCallback()); initialized_callback_.Run(true); } void ProtobufVideoWriter::Close() { buffered_writer_.Close(); channel_.reset(); if (channel_factory_) { channel_factory_->CancelChannelCreation(kVideoChannelName); channel_factory_ = NULL; } } bool ProtobufVideoWriter::is_connected() { return channel_.get() != NULL; } void ProtobufVideoWriter::ProcessVideoPacket(scoped_ptr<VideoPacket> packet, const base::Closure& done) { // Older clients may expect deprecated flags field. Always set it to 7. // // TODO(sergeyu): Remove this field after M31 is released. packet->set_deprecated_flags(7); buffered_writer_.Write(SerializeAndFrameMessage(*packet), done); } } // namespace protocol } // namespace remoting