| OLD | NEW |
| (Empty) |
| 1 // Copyright 2015 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 "blimp/net/client_connection_manager.h" | |
| 6 | |
| 7 #include <utility> | |
| 8 | |
| 9 #include "base/logging.h" | |
| 10 #include "base/memory/ptr_util.h" | |
| 11 #include "blimp/common/create_blimp_message.h" | |
| 12 #include "blimp/common/proto/blimp_message.pb.h" | |
| 13 #include "blimp/common/protocol_version.h" | |
| 14 #include "blimp/net/blimp_connection.h" | |
| 15 #include "blimp/net/blimp_message_processor.h" | |
| 16 #include "blimp/net/blimp_transport.h" | |
| 17 #include "blimp/net/browser_connection_handler.h" | |
| 18 #include "blimp/net/connection_handler.h" | |
| 19 #include "blimp/net/message_port.h" | |
| 20 #include "net/base/net_errors.h" | |
| 21 | |
| 22 namespace blimp { | |
| 23 | |
| 24 ClientConnectionManager::ClientConnectionManager( | |
| 25 ConnectionHandler* connection_handler) | |
| 26 : connection_handler_(connection_handler), weak_factory_(this) { | |
| 27 DCHECK(connection_handler_); | |
| 28 } | |
| 29 | |
| 30 ClientConnectionManager::~ClientConnectionManager() {} | |
| 31 | |
| 32 void ClientConnectionManager::AddTransport( | |
| 33 std::unique_ptr<BlimpTransport> transport) { | |
| 34 DCHECK(transport); | |
| 35 transports_.push_back(std::move(transport)); | |
| 36 } | |
| 37 | |
| 38 void ClientConnectionManager::Connect() { | |
| 39 // A |transport| added first is used first. When it fails to connect, | |
| 40 // the next transport is used. | |
| 41 DCHECK(!transports_.empty()); | |
| 42 Connect(0); | |
| 43 } | |
| 44 | |
| 45 void ClientConnectionManager::Connect(int transport_index) { | |
| 46 DVLOG(1) << "ClientConnectionManager::Connect(" << transport_index << ")"; | |
| 47 if (static_cast<size_t>(transport_index) < transports_.size()) { | |
| 48 transports_[transport_index]->Connect( | |
| 49 base::Bind(&ClientConnectionManager::OnConnectResult, | |
| 50 base::Unretained(this), transport_index)); | |
| 51 } else { | |
| 52 // TODO(haibinlu): add an error reporting path out for this. | |
| 53 LOG(WARNING) << "All transports failed to connect"; | |
| 54 } | |
| 55 } | |
| 56 | |
| 57 void ClientConnectionManager::OnConnectResult(int transport_index, int result) { | |
| 58 DCHECK_NE(result, net::ERR_IO_PENDING); | |
| 59 const auto& transport = transports_[transport_index]; | |
| 60 DVLOG(1) << "OnConnectResult; result = " << result; | |
| 61 if (result == net::OK) { | |
| 62 std::unique_ptr<BlimpConnection> connection = transport->MakeConnection(); | |
| 63 connection->AddConnectionErrorObserver(this); | |
| 64 SendAuthenticationMessage(std::move(connection)); | |
| 65 } else { | |
| 66 DVLOG(1) << "Transport " << transport->GetName() | |
| 67 << " failed to connect:" << net::ErrorToString(result); | |
| 68 Connect(transport_index + 1); | |
| 69 } | |
| 70 } | |
| 71 | |
| 72 void ClientConnectionManager::SendAuthenticationMessage( | |
| 73 std::unique_ptr<BlimpConnection> connection) { | |
| 74 DVLOG(1) << "Sending authentication message."; | |
| 75 connection->GetOutgoingMessageProcessor()->ProcessMessage( | |
| 76 CreateStartConnectionMessage(client_auth_token_, kProtocolVersion), | |
| 77 base::Bind(&ClientConnectionManager::OnAuthenticationMessageSent, | |
| 78 weak_factory_.GetWeakPtr(), | |
| 79 base::Passed(std::move(connection)))); | |
| 80 } | |
| 81 | |
| 82 void ClientConnectionManager::OnAuthenticationMessageSent( | |
| 83 std::unique_ptr<BlimpConnection> connection, | |
| 84 int result) { | |
| 85 DVLOG(1) << "AuthenticationMessageSent, result=" << result; | |
| 86 if (result != net::OK) { | |
| 87 // If a write error occurred, just throw away |connection|. | |
| 88 // We don't need to propagate the error code here because the connection | |
| 89 // will already have done so via the ErrorObserver object. | |
| 90 return; | |
| 91 } | |
| 92 connection_handler_->HandleConnection(std::move(connection)); | |
| 93 } | |
| 94 | |
| 95 void ClientConnectionManager::OnConnectionError(int error) { | |
| 96 // TODO(kmarshall): implement reconnection logic. | |
| 97 VLOG(0) << "Connection dropped, error=" << error; | |
| 98 } | |
| 99 | |
| 100 } // namespace blimp | |
| OLD | NEW |