Index: blimp/common/create_blimp_message.h |
diff --git a/blimp/common/create_blimp_message.h b/blimp/common/create_blimp_message.h |
index e201459c67e15319c93539236fd77bfff8a8342e..e98bc4a96b7f68dcc2de3f34447d09028e5494bd 100644 |
--- a/blimp/common/create_blimp_message.h |
+++ b/blimp/common/create_blimp_message.h |
@@ -11,6 +11,7 @@ |
#include <string> |
#include "blimp/common/blimp_common_export.h" |
+#include "blimp/common/proto/protocol_control.pb.h" |
namespace blimp { |
@@ -24,7 +25,6 @@ class NavigationMessage; |
class RenderWidgetMessage; |
class SettingsMessage; |
class SizeMessage; |
-class StartConnectionMessage; |
class TabControlMessage; |
// Suite of helper methods to simplify the repetitive task of creating |
@@ -77,6 +77,9 @@ BLIMP_COMMON_EXPORT std::unique_ptr<BlimpMessage> CreateStartConnectionMessage( |
BLIMP_COMMON_EXPORT std::unique_ptr<BlimpMessage> CreateCheckpointAckMessage( |
int64_t checkpoint_id); |
+BLIMP_COMMON_EXPORT std::unique_ptr<BlimpMessage> CreateEndConnectionMessage( |
+ EndConnectionMessage::Reason reason); |
+ |
} // namespace blimp |
#endif // BLIMP_COMMON_CREATE_BLIMP_MESSAGE_H_ |