Used oneof in blimp_message.proto

blimp/net/engine_authentication_handler.cc

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "blimp/net/engine_authentication_handler.h"
 
 #include <string>
 
 #include "base/callback_helpers.h"
 #include "base/logging.h"
@@ skipping 93 matching lines @@
 }
 
 void Authenticator::OnConnectionError(int error) {
   DVLOG(1) << "Connection error before authenticated "
            << net::ErrorToString(error);
   OnConnectionAuthenticated(false);
 }
 
 void Authenticator::ProcessMessage(std::unique_ptr<BlimpMessage> message,
                                    const net::CompletionCallback& callback) {
-  if (message->type() == BlimpMessage::PROTOCOL_CONTROL &&
-      message->protocol_control().type() ==
-          ProtocolControlMessage::START_CONNECTION) {
+  if (message->has_protocol_control() &&
+      message->protocol_control().has_start_connection()) {
     bool token_match =
         client_token_ ==
         message->protocol_control().start_connection().client_token();
     DVLOG(1) << "Authentication challenge received: "
              << message->protocol_control().start_connection().client_token()
              << ", and token "
              << (token_match ? " matches" : " does not match");
     OnConnectionAuthenticated(token_match);
   } else {
     DVLOG(1) << "Expected START_CONNECTION message, got " << *message
@@ skipping 18 matching lines @@
 
 void EngineAuthenticationHandler::HandleConnection(
     std::unique_ptr<BlimpConnection> connection) {
   // Authenticator manages its own lifetime.
   new Authenticator(std::move(connection),
                     connection_handler_weak_factory_.GetWeakPtr(),
                     client_token_);
 }
 
 }  // namespace blimp