* Plumb through node host to authentication provider, as some authentication providers need that info.

Author: stamhankar999

lib/cassandra/auth.rb

@@ -34,7 +34,7 @@ module Auth
     #
     # @see Cassandra::Auth::Providers
     class Provider
-      # @!method create_authenticator(authentication_class, protocol_version)
+      # @!method create_authenticator(authentication_class, host)
       #
       # Create a new authenticator object. This method will be called once per
       # connection that requires authentication. The auth provider can create
@@ -45,6 +45,8 @@ class Provider
       #
       # @param authentication_class [String] the authentication class used by
       #   the server.
+      # @param host [Cassandra::Host] the node to whom we're authenticating.
+      #
       # @return [Cassandra::Auth::Authenticator, nil] an object with an
       #   interface matching {Cassandra::Auth::Authenticator} or `nil` if the
       #   authentication class is not supported.

lib/cassandra/cluster/connector.rb

@@ -154,7 +154,7 @@ def do_connect(host)
                 supported_cql_versions.first :
                 '3.1.0'
 
-            startup_connection(connection, cql_version, compression)
+            startup_connection(host, connection, cql_version, compression)
           end
           f.fallback do |error|
             case error
@@ -200,7 +200,7 @@ def do_connect(host)
         end
       end
 
-      def startup_connection(connection, cql_version, compression)
+      def startup_connection(host, connection, cql_version, compression)
         connection.send_request(Protocol::StartupRequest.new(cql_version, compression),
                                 @execution_options.timeout).flat_map do |r|
           case r
@@ -213,12 +213,9 @@ def startup_connection(connection, cql_version, compression)
                 Ione::Future.failed(cannot_authenticate_error)
               end
             else
-              authenticator = @connection_options.create_authenticator(
-                r.authentication_class)
+              authenticator = @connection_options.create_authenticator(r.authentication_class, host)
               if authenticator
-                challenge_response_cycle(connection,
-                                         authenticator,
-                                         authenticator.initial_response)
+                challenge_response_cycle(connection, authenticator, authenticator.initial_response)
               else
                 Ione::Future.failed(cannot_authenticate_error)
               end

lib/cassandra/cluster/options.rb

@@ -80,8 +80,17 @@ def compression
         @compressor && @compressor.algorithm
       end
 
-      def create_authenticator(authentication_class)
-        @auth_provider && @auth_provider.create_authenticator(authentication_class)
+      def create_authenticator(authentication_class, host)
+        if @auth_provider
+          # Auth providers should take an auth-class and host, but they used to not, so for backward compatibility
+          # we figure out if this provider does, and if so send both args, otherwise just send the auth-class.
+
+          if @auth_provider.method(:create_authenticator).arity == 1
+            @auth_provider.create_authenticator(authentication_class)
+          else
+            @auth_provider.create_authenticator(authentication_class, host)
+          end
+        end
       end
 
       def connections_per_local_node