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
|
// Copyright 2012 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
//
// A toy client, which connects to a specified port and sends QUIC
// request to that endpoint.
#ifndef NET_TOOLS_QUIC_QUIC_CLIENT_MESSAGE_LOOP_NETWORK_HELPER_H_
#define NET_TOOLS_QUIC_QUIC_CLIENT_MESSAGE_LOOP_NETWORK_HELPER_H_
#include <stddef.h>
#include <memory>
#include "base/command_line.h"
#include "net/base/ip_address.h"
#include "net/base/ip_endpoint.h"
#include "net/http/http_response_headers.h"
#include "net/quic/platform/impl/quic_chromium_clock.h"
#include "net/quic/quic_chromium_packet_reader.h"
#include "net/third_party/quiche/src/quiche/quic/core/http/quic_spdy_stream.h"
#include "net/third_party/quiche/src/quiche/quic/core/quic_config.h"
#include "net/third_party/quiche/src/quiche/quic/tools/quic_spdy_client_base.h"
namespace net {
class UDPClientSocket;
// An implementation of the quic::QuicClientBase::NetworkHelper based off
// the chromium epoll server.
class QuicClientMessageLooplNetworkHelper
: public quic::QuicClientBase::NetworkHelper,
public QuicChromiumPacketReader::Visitor {
public:
// Create a quic client, which will have events managed by an externally owned
// EpollServer.
QuicClientMessageLooplNetworkHelper(quic::QuicChromiumClock* clock,
quic::QuicClientBase* client);
QuicClientMessageLooplNetworkHelper(
const QuicClientMessageLooplNetworkHelper&) = delete;
QuicClientMessageLooplNetworkHelper& operator=(
const QuicClientMessageLooplNetworkHelper&) = delete;
~QuicClientMessageLooplNetworkHelper() override;
// QuicChromiumPacketReader::Visitor
bool OnReadError(int result, const DatagramClientSocket* socket) override;
bool OnPacket(const quic::QuicReceivedPacket& packet,
const quic::QuicSocketAddress& local_address,
const quic::QuicSocketAddress& peer_address) override;
// From NetworkHelper.
void RunEventLoop() override;
bool CreateUDPSocketAndBind(quic::QuicSocketAddress server_address,
quiche::QuicheIpAddress bind_to_address,
int bind_to_port) override;
void CleanUpAllUDPSockets() override;
quic::QuicSocketAddress GetLatestClientAddress() const override;
quic::QuicPacketWriter* CreateQuicPacketWriter() override;
private:
void StartPacketReaderIfNotStarted();
// Address of the client if the client is connected to the server.
quic::QuicSocketAddress client_address_;
// UDP socket connected to the server.
std::unique_ptr<UDPClientSocket> socket_;
std::unique_ptr<QuicChromiumPacketReader> packet_reader_;
bool packet_reader_started_ = false;
quic::QuicChromiumClock* clock_;
quic::QuicClientBase* client_;
};
} // namespace net
#endif // NET_TOOLS_QUIC_QUIC_CLIENT_MESSAGE_LOOP_NETWORK_HELPER_H_
|