make CallCredentials implementation agnostic

pull/18652/head
Jan Tattermusch 6 years ago
parent e3290aa0fe
commit a1a878b593
  1. 4
      src/csharp/Grpc.Core.Tests/FakeCredentials.cs
  2. 44
      src/csharp/Grpc.Core/CallCredentials.cs
  3. 38
      src/csharp/Grpc.Core/CallCredentialsConfiguratorBase.cs
  4. 49
      src/csharp/Grpc.Core/Internal/DefaultCallCredentialsConfigurator.cs
  5. 51
      src/csharp/Grpc.Core/Internal/MetadataCredentials.cs

@ -42,9 +42,9 @@ namespace Grpc.Core.Tests
internal class FakeCallCredentials : CallCredentials internal class FakeCallCredentials : CallCredentials
{ {
internal override CallCredentialsSafeHandle ToNativeCredentials() public override void InternalPopulateConfiguration(CallCredentialsConfiguratorBase configurator, object state)
{ {
return null; // not invoking the configurator on purpose
} }
} }
} }

@ -16,9 +16,8 @@
#endregion #endregion
using System;
using System.Collections.Generic; using System.Collections.Generic;
using System.Threading.Tasks; using System.Collections.ObjectModel;
using Grpc.Core.Internal; using Grpc.Core.Internal;
using Grpc.Core.Utils; using Grpc.Core.Utils;
@ -38,7 +37,7 @@ namespace Grpc.Core
/// <returns>The new <c>CompositeCallCredentials</c></returns> /// <returns>The new <c>CompositeCallCredentials</c></returns>
public static CallCredentials Compose(params CallCredentials[] credentials) public static CallCredentials Compose(params CallCredentials[] credentials)
{ {
return new CompositeCallCredentials(credentials); return new CompositeCallCredentialsConfiguration(credentials);
} }
/// <summary> /// <summary>
@ -48,13 +47,44 @@ namespace Grpc.Core
/// <param name="interceptor">authentication interceptor</param> /// <param name="interceptor">authentication interceptor</param>
public static CallCredentials FromInterceptor(AsyncAuthInterceptor interceptor) public static CallCredentials FromInterceptor(AsyncAuthInterceptor interceptor)
{ {
return new MetadataCredentials(interceptor); return new AsyncAuthInterceptorCredentialsConfiguration(interceptor);
} }
/// <summary> /// <summary>
/// Creates native object for the credentials. /// Populates this call credential instances.
/// You never need to invoke this, part of internal implementation.
/// </summary> /// </summary>
/// <returns>The native credentials.</returns> public abstract void InternalPopulateConfiguration(CallCredentialsConfiguratorBase configurator, object state);
internal abstract CallCredentialsSafeHandle ToNativeCredentials();
private class CompositeCallCredentialsConfiguration : CallCredentials
{
readonly IReadOnlyList<CallCredentials> credentials;
public CompositeCallCredentialsConfiguration(CallCredentials[] credentials)
{
GrpcPreconditions.CheckArgument(credentials.Length >= 2, "Composite credentials object can only be created from 2 or more credentials.");
this.credentials = new List<CallCredentials>(credentials).AsReadOnly();
}
public override void InternalPopulateConfiguration(CallCredentialsConfiguratorBase configurator, object state)
{
configurator.SetCompositeCredentials(state, credentials);
}
}
internal class AsyncAuthInterceptorCredentialsConfiguration : CallCredentials
{
readonly AsyncAuthInterceptor interceptor;
public AsyncAuthInterceptorCredentialsConfiguration(AsyncAuthInterceptor interceptor)
{
this.interceptor = GrpcPreconditions.CheckNotNull(interceptor);
}
public override void InternalPopulateConfiguration(CallCredentialsConfiguratorBase configurator, object state)
{
configurator.SetAsyncAuthInterceptorCredentials(state, interceptor);
}
}
} }
} }

@ -0,0 +1,38 @@
#region Copyright notice and license
// Copyright 2019 The gRPC Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#endregion
using System.Collections.Generic;
namespace Grpc.Core
{
/// <summary>
/// Base class for objects that can consume configuration from <c>CallCredentials</c> objects.
/// </summary>
public abstract class CallCredentialsConfiguratorBase
{
/// <summary>
/// Consumes configuration for composite call credentials.
/// </summary>
public abstract void SetCompositeCredentials(object state, IReadOnlyList<CallCredentials> credentials);
/// <summary>
/// Consumes configuration for call credentials created from <c>AsyncAuthInterceptor</c>
/// </summary>
public abstract void SetAsyncAuthInterceptorCredentials(object state, AsyncAuthInterceptor interceptor);
}
}

@ -18,39 +18,38 @@
using System; using System;
using System.Collections.Generic; using System.Collections.Generic;
using System.Threading.Tasks;
using Grpc.Core.Internal;
using Grpc.Core.Utils; using Grpc.Core.Utils;
namespace Grpc.Core.Internal namespace Grpc.Core.Internal
{ {
/// <summary> /// <summary>
/// Credentials that allow composing multiple credentials objects into one <see cref="CallCredentials"/> object. /// Creates native call credential objects from instances of <c>CallCredentials</c>.
/// </summary> /// </summary>
internal sealed class CompositeCallCredentials : CallCredentials internal class DefaultCallCredentialsConfigurator : CallCredentialsConfiguratorBase
{ {
readonly List<CallCredentials> credentials; CallCredentialsSafeHandle nativeCredentials;
/// <summary> public CallCredentialsSafeHandle NativeCredentials => nativeCredentials;
/// Initializes a new instance of <c>CompositeCallCredentials</c> class.
/// The resulting credentials object will be composite of all the credentials specified as parameters. public override void SetAsyncAuthInterceptorCredentials(object state, AsyncAuthInterceptor interceptor)
/// </summary>
/// <param name="credentials">credentials to compose</param>
public CompositeCallCredentials(params CallCredentials[] credentials)
{ {
GrpcPreconditions.CheckArgument(credentials.Length >= 2, "Composite credentials object can only be created from 2 or more credentials."); GrpcPreconditions.CheckState(nativeCredentials == null);
this.credentials = new List<CallCredentials>(credentials);
var plugin = new NativeMetadataCredentialsPlugin(interceptor);
nativeCredentials = plugin.Credentials;
} }
internal override CallCredentialsSafeHandle ToNativeCredentials() public override void SetCompositeCredentials(object state, IReadOnlyList<CallCredentials> credentials)
{ {
return ToNativeRecursive(0); GrpcPreconditions.CheckState(nativeCredentials == null);
GrpcPreconditions.CheckArgument(credentials.Count >= 2);
nativeCredentials = CompositeToNativeRecursive(credentials, 0);
} }
// Recursive descent makes managing lifetime of intermediate CredentialSafeHandle instances easier. // Recursive descent makes managing lifetime of intermediate CredentialSafeHandle instances easier.
// In practice, we won't usually see composites from more than two credentials anyway. // In practice, we won't usually see composites from more than two credentials anyway.
private CallCredentialsSafeHandle ToNativeRecursive(int startIndex) private CallCredentialsSafeHandle CompositeToNativeRecursive(IReadOnlyList<CallCredentials> credentials, int startIndex)
{ {
if (startIndex == credentials.Count - 1) if (startIndex == credentials.Count - 1)
{ {
@ -58,7 +57,7 @@ namespace Grpc.Core.Internal
} }
using (var cred1 = credentials[startIndex].ToNativeCredentials()) using (var cred1 = credentials[startIndex].ToNativeCredentials())
using (var cred2 = ToNativeRecursive(startIndex + 1)) using (var cred2 = CompositeToNativeRecursive(credentials, startIndex + 1))
{ {
var nativeComposite = CallCredentialsSafeHandle.CreateComposite(cred1, cred2); var nativeComposite = CallCredentialsSafeHandle.CreateComposite(cred1, cred2);
if (nativeComposite.IsInvalid) if (nativeComposite.IsInvalid)
@ -69,4 +68,18 @@ namespace Grpc.Core.Internal
} }
} }
} }
internal static class CallCredentialsExtensions
{
/// <summary>
/// Creates native object for the credentials.
/// </summary>
/// <returns>The native credentials.</returns>
public static CallCredentialsSafeHandle ToNativeCredentials(this CallCredentials credentials)
{
var configurator = new DefaultCallCredentialsConfigurator();
credentials.InternalPopulateConfiguration(configurator, credentials);
return configurator.NativeCredentials;
}
}
} }

@ -1,51 +0,0 @@
#region Copyright notice and license
// Copyright 2019 The gRPC Authors
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
#endregion
using System;
using System.Collections.Generic;
using System.Threading.Tasks;
using Grpc.Core.Internal;
using Grpc.Core.Utils;
namespace Grpc.Core.Internal
{
/// <summary>
/// Client-side credentials that delegate metadata based auth to an interceptor.
/// The interceptor is automatically invoked for each remote call that uses <c>MetadataCredentials.</c>
/// </summary>
internal sealed class MetadataCredentials : CallCredentials
{
readonly AsyncAuthInterceptor interceptor;
/// <summary>
/// Initializes a new instance of <c>MetadataCredentials</c> class.
/// </summary>
/// <param name="interceptor">authentication interceptor</param>
public MetadataCredentials(AsyncAuthInterceptor interceptor)
{
this.interceptor = GrpcPreconditions.CheckNotNull(interceptor);
}
internal override CallCredentialsSafeHandle ToNativeCredentials()
{
NativeMetadataCredentialsPlugin plugin = new NativeMetadataCredentialsPlugin(interceptor);
return plugin.Credentials;
}
}
}
Loading…
Cancel
Save