Index: net/tools/quic/test_tools/quic_test_client.cc |
diff --git a/net/tools/quic/test_tools/quic_test_client.cc b/net/tools/quic/test_tools/quic_test_client.cc |
index c2a95aa102a98ade248edff5209ed3c5ee119a57..41846fd7d14d7c4cb611697c457829f999a3b216 100644 |
--- a/net/tools/quic/test_tools/quic_test_client.cc |
+++ b/net/tools/quic/test_tools/quic_test_client.cc |
@@ -321,10 +321,10 @@ ssize_t QuicTestClient::GetOrCreateStreamAndSendRequest( |
return 1; |
if (rv == QUIC_PENDING) { |
// May need to retry request if asynchronous rendezvous fails. |
- auto* new_headers = new BalsaHeaders; |
- new_headers->CopyFrom(*headers); |
- push_promise_data_to_resend_.reset( |
- new TestClientDataToResend(new_headers, body, fin, this, delegate)); |
+ std::unique_ptr<SpdyHeaderBlock> new_headers(new SpdyHeaderBlock( |
+ SpdyBalsaUtils::RequestHeadersToSpdyHeaders(*headers))); |
+ push_promise_data_to_resend_.reset(new TestClientDataToResend( |
+ std::move(new_headers), body, fin, this, delegate)); |
return 1; |
} |
} |
@@ -359,13 +359,14 @@ ssize_t QuicTestClient::GetOrCreateStreamAndSendRequest( |
ret = body.length(); |
} |
if (FLAGS_enable_quic_stateless_reject_support) { |
- BalsaHeaders* new_headers = nullptr; |
+ std::unique_ptr<SpdyHeaderBlock> new_headers; |
if (headers) { |
- new_headers = new BalsaHeaders; |
- new_headers->CopyFrom(*headers); |
+ new_headers.reset(new SpdyHeaderBlock( |
+ SpdyBalsaUtils::RequestHeadersToSpdyHeaders(*headers))); |
} |
- std::unique_ptr<QuicClient::QuicDataToResend> data_to_resend( |
- new TestClientDataToResend(new_headers, body, fin, this, delegate)); |
+ std::unique_ptr<QuicClientBase::QuicDataToResend> data_to_resend( |
+ new TestClientDataToResend(std::move(new_headers), body, fin, this, |
+ delegate)); |
client()->MaybeAddQuicDataToResend(std::move(data_to_resend)); |
} |
return ret; |
@@ -736,12 +737,11 @@ void QuicTestClient::WaitForWriteToFlush() { |
} |
void QuicTestClient::TestClientDataToResend::Resend() { |
- test_client_->GetOrCreateStreamAndSendRequest(headers_, body_, fin_, |
+ BalsaHeaders balsa_headers; |
+ SpdyBalsaUtils::SpdyHeadersToRequestHeaders(*headers_, &balsa_headers); |
+ test_client_->GetOrCreateStreamAndSendRequest(&balsa_headers, body_, fin_, |
delegate_); |
- if (headers_ != nullptr) { |
- delete headers_; |
- headers_ = nullptr; |
- } |
+ headers_.reset(); |
} |
// static |