chrome/test/chromedriver/net/adb_client_socket.h

   1 // Copyright (c) 2013 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 CHROME_TEST_CHROMEDRIVER_NET_ADB_CLIENT_SOCKET_H_
   6 #define CHROME_TEST_CHROMEDRIVER_NET_ADB_CLIENT_SOCKET_H_
   7
   8 #include "base/callback.h"
   9 #include "net/base/io_buffer.h"
  10 #include "net/socket/stream_socket.h"
  11
  12 class AdbClientSocket {
  13  public:
  14   typedef base::Callback<void(int, const std::string&)> CommandCallback;
  15   typedef base::Callback<void(int result,
  16                               net::StreamSocket*)> SocketCallback;
  17
  18   static void AdbQuery(int port,
  19                        const std::string& query,
  20                        const CommandCallback& callback);
  21
  22   static void TransportQuery(int port,
  23                              const std::string& serial,
  24                              const std::string& socket_name,
  25                              const SocketCallback& callback);
  26
  27   static void HttpQuery(int port,
  28                         const std::string& serial,
  29                         const std::string& socket_name,
  30                         const std::string& request,
  31                         const CommandCallback& callback);
  32
  33   static void HttpQuery(int port,
  34                         const std::string& serial,
  35                         const std::string& socket_name,
  36                         const std::string& request,
  37                         const SocketCallback& callback);
  38
  39   explicit AdbClientSocket(int port);
  40   ~AdbClientSocket();
  41
  42  protected:
  43   void Connect(const net::CompletionCallback& callback);
  44
  45   void SendCommand(const std::string& command,
  46                    bool is_void,
  47                    bool has_length,
  48                    const CommandCallback& callback);
  49
  50   scoped_ptr<net::StreamSocket> socket_;
  51
  52  private:
  53   void ReadResponse(const CommandCallback& callback,
  54                     bool is_void,
  55                     bool has_length,
  56                     int result);
  57
  58   void OnResponseStatus(const CommandCallback& callback,
  59                         bool is_void,
  60                         bool has_length,
  61                         scoped_refptr<net::IOBuffer> response_buffer,
  62                         int result);
  63
  64   void OnResponseLength(const CommandCallback& callback,
  65                         const std::string& response,
  66                         scoped_refptr<net::IOBuffer> response_buffer,
  67                         int result);
  68
  69   void OnResponseData(const CommandCallback& callback,
  70                       const std::string& response,
  71                       scoped_refptr<net::IOBuffer> response_buffer,
  72                       int bytes_left,
  73                       int result);
  74
  75   std::string host_;
  76   int port_;
  77
  78   DISALLOW_COPY_AND_ASSIGN(AdbClientSocket);
  79 };
  80
  81 #endif  // CHROME_TEST_CHROMEDRIVER_NET_ADB_CLIENT_SOCKET_H_