More plumbing for logging connection IP addresses

remoting/protocol/connection_to_client.cc

-// Copyright (c) 2011 The Chromium Authors. All rights reserved.
+// 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/connection_to_client.h"
 
 #include "base/bind.h"
 #include "base/location.h"
 #include "base/message_loop_proxy.h"
 #include "google/protobuf/message.h"
 #include "net/base/io_buffer.h"
 #include "remoting/protocol/host_control_dispatcher.h"
 #include "remoting/protocol/host_event_dispatcher.h"
 #include "remoting/protocol/host_stub.h"
 #include "remoting/protocol/input_stub.h"
 
 namespace remoting {
 namespace protocol {
 
 ConnectionToClient::ConnectionToClient(protocol::Session* session)
     : handler_(NULL),
       host_stub_(NULL),
       input_stub_(NULL),
       session_(session) {
   session_->SetStateChangeCallback(
       base::Bind(&ConnectionToClient::OnSessionStateChange,
                  base::Unretained(this)));
+  session_->SetRouteChangeCallback(
+      base::Bind(&ConnectionToClient::OnSessionRouteChange,
+                 base::Unretained(this)));
 }
 
 ConnectionToClient::~ConnectionToClient() {
   if (session_.get()) {
     base::MessageLoopProxy::current()->DeleteSoon(
         FROM_HERE, session_.release());
   }
 }
 
 void ConnectionToClient::SetEventHandler(EventHandler* event_handler) {
@@ skipping 86 matching lines @@
       CloseChannels();
       handler_->OnConnectionClosed(this);
       break;
 
     case Session::FAILED:
       CloseOnError();
       break;
   }
 }
 
+void ConnectionToClient::OnSessionRouteChange(
+    const std::string& channel_name, const net::IPEndPoint& end_point) {
+  handler_->OnClientIpAddress(this, channel_name, end_point);
+}
+
 void ConnectionToClient::OnChannelInitialized(bool successful) {
   DCHECK(CalledOnValidThread());
 
   if (!successful) {
     LOG(ERROR) << "Failed to connect a channel";
     CloseOnError();
     return;
   }
 
   NotifyIfChannelsReady();
@@ skipping 15 matching lines @@
 }
 
 void ConnectionToClient::CloseChannels() {
   control_dispatcher_.reset();
   event_dispatcher_.reset();
   video_writer_.reset();
 }
 
 }  // namespace protocol
 }  // namespace remoting