Port QuicServer to Chrome network stack

net/quic/quic_connection_packet_writer_wrapper.cc

 // Copyright 2014 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
-#include "net/tools/quic/quic_packet_writer_wrapper.h"
+#include "net/quic/quic_connection_packet_writer_wrapper.h"
 
-#include "net/quic/quic_types.h"
+#include "base/location.h"
+#include "base/logging.h"
+#include "base/message_loop/message_loop.h"
+#include "base/metrics/sparse_histogram.h"
+#include "net/base/io_buffer.h"
+#include "net/base/net_errors.h"
+#include "net/quic/crypto/quic_random.h"
+#include "net/quic/quic_server_packet_writer.h"
 
 namespace net {
-namespace tools {
 
-QuicPacketWriterWrapper::QuicPacketWriterWrapper() {}
+QuicConnectionPacketWriterWrapper::QuicConnectionPacketWriterWrapper(
+    QuicServerPacketWriter* writer,
+    QuicConnection* connection)
+    : weak_factory_(this),
+      writer_(writer),
+      connection_(connection) {
+}
 
-QuicPacketWriterWrapper::QuicPacketWriterWrapper(QuicPacketWriter* writer)
-    : writer_(writer) {}
+QuicConnectionPacketWriterWrapper::~QuicConnectionPacketWriterWrapper() {
+}
 
-QuicPacketWriterWrapper::~QuicPacketWriterWrapper() {}
-
-WriteResult QuicPacketWriterWrapper::WritePacket(
+WriteResult QuicConnectionPacketWriterWrapper::WritePacket(
     const char* buffer,
     size_t buf_len,
-    const net::IPAddressNumber& self_address,
-    const net::IPEndPoint& peer_address) {
+    const IPAddressNumber& self_address,
+    const IPEndPoint& peer_address) {
+  writer_->SetWriteCallback(
+      base::Bind(&QuicConnectionPacketWriterWrapper::OnWriteComplete,
+                 weak_factory_.GetWeakPtr()));
   return writer_->WritePacket(buffer, buf_len, self_address, peer_address);
 }
 
-bool QuicPacketWriterWrapper::IsWriteBlockedDataBuffered() const {
+bool QuicConnectionPacketWriterWrapper::IsWriteBlockedDataBuffered() const {
   return writer_->IsWriteBlockedDataBuffered();
 }
 
-bool QuicPacketWriterWrapper::IsWriteBlocked() const {
+bool QuicConnectionPacketWriterWrapper::IsWriteBlocked() const {
   return writer_->IsWriteBlocked();
 }
 
-void QuicPacketWriterWrapper::SetWritable() {
+void QuicConnectionPacketWriterWrapper::SetWritable() {
   writer_->SetWritable();
 }
 
-void QuicPacketWriterWrapper::set_writer(QuicPacketWriter* writer) {
-  writer_.reset(writer);
+void QuicConnectionPacketWriterWrapper::OnWriteComplete(WriteResult wr) {
+  connection_->OnPacketSent(wr);
+  connection_->OnCanWrite();
 }
 
-QuicPacketWriter* QuicPacketWriterWrapper::release_writer() {
-  return writer_.release();
-}
-
-}  // namespace tools
 }  // namespace net