1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88
|
// Copyright 2013 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CHROME_TEST_CHROMEDRIVER_NET_ADB_CLIENT_SOCKET_H_
#define CHROME_TEST_CHROMEDRIVER_NET_ADB_CLIENT_SOCKET_H_
#include "base/functional/callback.h"
#include "net/base/completion_once_callback.h"
#include "net/base/io_buffer.h"
#include "net/socket/stream_socket.h"
class AdbClientSocket {
public:
typedef base::RepeatingCallback<void(int, const std::string&)>
CommandCallback;
typedef base::RepeatingCallback<void(int result, net::StreamSocket*)>
SocketCallback;
typedef base::RepeatingCallback<void(const std::string&)> ParserCallback;
static void AdbQuery(int port,
const std::string& query,
const CommandCallback& callback);
static void TransportQuery(int port,
const std::string& serial,
const std::string& socket_name,
const SocketCallback& callback);
static void HttpQuery(int port,
const std::string& serial,
const std::string& socket_name,
const std::string& request,
const CommandCallback& callback);
static void HttpQuery(int port,
const std::string& serial,
const std::string& socket_name,
const std::string& request,
const SocketCallback& callback);
static void SendFile(int port,
const std::string& serial,
const std::string& filename,
const std::string& content,
const CommandCallback& callback);
explicit AdbClientSocket(int port);
AdbClientSocket(const AdbClientSocket&) = delete;
AdbClientSocket& operator=(const AdbClientSocket&) = delete;
~AdbClientSocket();
protected:
void Connect(net::CompletionOnceCallback callback);
void SendCommand(const std::string& command,
bool has_output,
bool has_length,
const CommandCallback& response_callback);
std::unique_ptr<net::StreamSocket> socket_;
void ReadResponse(const CommandCallback& callback,
bool has_output,
bool has_length,
int result);
private:
static void ReadStatusOutput(const CommandCallback& response_callback,
scoped_refptr<net::IOBuffer> socket_buffer,
int socket_result);
void ReadUntilEOF(const ParserCallback& parse_output_callback,
const CommandCallback& response_callback,
scoped_refptr<net::GrowableIOBuffer> socket_buffer,
int socket_result);
static void ParseOutput(bool has_length,
const CommandCallback& response_callback,
const std::string& adb_output);
int port_;
friend class AdbClientSocketTest;
};
#endif // CHROME_TEST_CHROMEDRIVER_NET_ADB_CLIENT_SOCKET_H_
|