| Index: blimp/common/create_blimp_message.cc
|
| diff --git a/blimp/common/create_blimp_message.cc b/blimp/common/create_blimp_message.cc
|
| index a3af17d6c8baf4be6ef8b3e887a50f0474caea00..1ee3d1e3c1fbd928d8a82a0773522c6cf694ffd0 100644
|
| --- a/blimp/common/create_blimp_message.cc
|
| +++ b/blimp/common/create_blimp_message.cc
|
| @@ -66,7 +66,6 @@ std::unique_ptr<BlimpMessage> CreateBlimpMessage(
|
| }
|
|
|
| std::unique_ptr<BlimpMessage> CreateBlimpMessage(SizeMessage** size_message) {
|
| - DCHECK(size_message);
|
| TabControlMessage* control_message;
|
| std::unique_ptr<BlimpMessage> output = CreateBlimpMessage(&control_message);
|
| control_message->mutable_size();
|
| @@ -109,8 +108,6 @@ std::unique_ptr<BlimpMessage> CreateCheckpointAckMessage(
|
| std::unique_ptr<BlimpMessage> output(new BlimpMessage);
|
|
|
| ProtocolControlMessage* control_message = output->mutable_protocol_control();
|
| - control_message->mutable_checkpoint_ack();
|
| -
|
| CheckpointAckMessage* checkpoint_ack_message =
|
| control_message->mutable_checkpoint_ack();
|
| checkpoint_ack_message->set_checkpoint_id(checkpoint_id);
|
| @@ -118,4 +115,16 @@ std::unique_ptr<BlimpMessage> CreateCheckpointAckMessage(
|
| return output;
|
| }
|
|
|
| +std::unique_ptr<BlimpMessage> CreateEndConnectionMessage(
|
| + EndConnectionMessage::Reason reason) {
|
| + std::unique_ptr<BlimpMessage> output(new BlimpMessage);
|
| +
|
| + ProtocolControlMessage* control_message = output->mutable_protocol_control();
|
| + EndConnectionMessage* end_connection_message =
|
| + control_message->mutable_end_connection();
|
| + end_connection_message->set_reason(reason);
|
| +
|
| + return output;
|
| +}
|
| +
|
| } // namespace blimp
|
|
|