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 |