Index: net/quic/quic_connection_test.cc |
diff --git a/net/quic/quic_connection_test.cc b/net/quic/quic_connection_test.cc |
index 5a76ec4730660a743ca7ba5e37ac4da541c5fb18..a25065eb8463ad5b5eb06d5d631e116ac51c0f06 100644 |
--- a/net/quic/quic_connection_test.cc |
+++ b/net/quic/quic_connection_test.cc |
@@ -87,13 +87,20 @@ class TaggingEncrypter : public QuicEncrypter { |
return true; |
} |
- QuicData* EncryptPacket(QuicPacketSequenceNumber sequence_number, |
- StringPiece associated_data, |
- StringPiece plaintext) override { |
+ bool EncryptPacket(QuicPacketSequenceNumber sequence_number, |
+ StringPiece associated_data, |
+ StringPiece plaintext, |
+ char* output, |
+ size_t* output_length, |
+ size_t max_output_length) override { |
const size_t len = plaintext.size() + kTagSize; |
- uint8* buffer = new uint8[len]; |
- Encrypt(StringPiece(), associated_data, plaintext, buffer); |
- return new QuicData(reinterpret_cast<char*>(buffer), len, true); |
+ if (max_output_length < len) { |
+ return false; |
+ } |
+ Encrypt(StringPiece(), associated_data, plaintext, |
+ reinterpret_cast<unsigned char*>(output)); |
+ *output_length = len; |
+ return true; |
} |
size_t GetKeySize() const override { return 0; } |