Index: blimp/common/create_blimp_message.cc |
diff --git a/blimp/common/create_blimp_message.cc b/blimp/common/create_blimp_message.cc |
index 460560c6851c6513ff1ebf32150a5e3de6f8ebd6..14595f3cb44658bb1e132788818faf52acdcba2a 100644 |
--- a/blimp/common/create_blimp_message.cc |
+++ b/blimp/common/create_blimp_message.cc |
@@ -65,7 +65,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(); |
@@ -101,8 +100,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); |
@@ -110,4 +107,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 |