net/socket/websocket_transport_connect_sub_job.h

   1 // Copyright 2014 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 #ifndef NET_SOCKET_WEBSOCKET_TRANSPORT_CONNECT_SUB_JOB_H_
   6 #define NET_SOCKET_WEBSOCKET_TRANSPORT_CONNECT_SUB_JOB_H_
   7
   8 #include "base/compiler_specific.h"
   9 #include "base/macros.h"
  10 #include "base/memory/scoped_ptr.h"
  11 #include "net/base/address_list.h"
  12 #include "net/base/load_states.h"
  13 #include "net/socket/websocket_endpoint_lock_manager.h"
  14 #include "net/socket/websocket_transport_client_socket_pool.h"
  15
  16 namespace net {
  17
  18 class BoundNetLog;
  19 class ClientSocketFactory;
  20 class IPEndPoint;
  21 class StreamSocket;
  22
  23 // Attempts to connect to a subset of the addresses required by a
  24 // WebSocketTransportConnectJob, specifically either the IPv4 or IPv6
  25 // addresses. Each address is tried in turn, and parent_job->OnSubJobComplete()
  26 // is called when the first address succeeds or the last address fails.
  27 class WebSocketTransportConnectSubJob
  28     : public WebSocketEndpointLockManager::Waiter {
  29  public:
  30   typedef WebSocketTransportConnectJob::SubJobType SubJobType;
  31
  32   WebSocketTransportConnectSubJob(const AddressList& addresses,
  33                                   WebSocketTransportConnectJob* parent_job,
  34                                   SubJobType type);
  35
  36   ~WebSocketTransportConnectSubJob() override;
  37
  38   // Start connecting.
  39   int Start();
  40
  41   bool started() { return next_state_ != STATE_NONE; }
  42
  43   LoadState GetLoadState() const;
  44
  45   SubJobType type() const { return type_; }
  46
  47   scoped_ptr<StreamSocket> PassSocket() { return transport_socket_.Pass(); }
  48
  49   // Implementation of WebSocketEndpointLockManager::EndpointWaiter.
  50   void GotEndpointLock() override;
  51
  52  private:
  53   enum State {
  54     STATE_NONE,
  55     STATE_OBTAIN_LOCK,
  56     STATE_OBTAIN_LOCK_COMPLETE,
  57     STATE_TRANSPORT_CONNECT,
  58     STATE_TRANSPORT_CONNECT_COMPLETE,
  59     STATE_DONE,
  60   };
  61
  62   ClientSocketFactory* client_socket_factory() const;
  63
  64   const BoundNetLog& net_log() const;
  65
  66   const IPEndPoint& CurrentAddress() const;
  67
  68   void OnIOComplete(int result);
  69   int DoLoop(int result);
  70   int DoEndpointLock();
  71   int DoEndpointLockComplete();
  72   int DoTransportConnect();
  73   int DoTransportConnectComplete(int result);
  74
  75   WebSocketTransportConnectJob* const parent_job_;
  76
  77   const AddressList addresses_;
  78   size_t current_address_index_;
  79
  80   State next_state_;
  81   const SubJobType type_;
  82
  83   scoped_ptr<StreamSocket> transport_socket_;
  84
  85   DISALLOW_COPY_AND_ASSIGN(WebSocketTransportConnectSubJob);
  86 };
  87
  88 }  // namespace net
  89
  90 #endif  // NET_SOCKET_WEBSOCKET_TRANSPORT_CONNECT_SUB_JOB_H_