77 lines
3.4 KiB
C#
77 lines
3.4 KiB
C#
// Copyright (c) .NET Foundation. All rights reserved.
|
|
// Licensed under the Apache License, Version 2.0. See License.txt in the project root for license information.
|
|
|
|
using System;
|
|
using Microsoft.AspNetCore.Antiforgery;
|
|
using Microsoft.AspNetCore.Antiforgery.Internal;
|
|
using Microsoft.Extensions.DependencyInjection.Extensions;
|
|
using Microsoft.Extensions.ObjectPool;
|
|
using Microsoft.Extensions.Options;
|
|
|
|
namespace Microsoft.Extensions.DependencyInjection
|
|
{
|
|
/// <summary>
|
|
/// Extension methods for setting up antiforgery services in an <see cref="IServiceCollection" />.
|
|
/// </summary>
|
|
public static class AntiforgeryServiceCollectionExtensions
|
|
{
|
|
/// <summary>
|
|
/// Adds antiforgery services to the specified <see cref="IServiceCollection" />.
|
|
/// </summary>
|
|
/// <param name="services">The <see cref="IServiceCollection" /> to add services to.</param>
|
|
/// <returns>The <see cref="IServiceCollection"/> so that additional calls can be chained.</returns>
|
|
public static IServiceCollection AddAntiforgery(this IServiceCollection services)
|
|
{
|
|
if (services == null)
|
|
{
|
|
throw new ArgumentNullException(nameof(services));
|
|
}
|
|
|
|
services.AddDataProtection();
|
|
|
|
// Don't overwrite any options setups that a user may have added.
|
|
services.TryAddEnumerable(
|
|
ServiceDescriptor.Transient<IConfigureOptions<AntiforgeryOptions>, AntiforgeryOptionsSetup>());
|
|
|
|
services.TryAddSingleton<IAntiforgery, DefaultAntiforgery>();
|
|
services.TryAddSingleton<IAntiforgeryTokenGenerator, DefaultAntiforgeryTokenGenerator>();
|
|
services.TryAddSingleton<IAntiforgeryTokenSerializer, DefaultAntiforgeryTokenSerializer>();
|
|
services.TryAddSingleton<IAntiforgeryTokenStore, DefaultAntiforgeryTokenStore>();
|
|
services.TryAddSingleton<IClaimUidExtractor, DefaultClaimUidExtractor>();
|
|
services.TryAddSingleton<IAntiforgeryAdditionalDataProvider, DefaultAntiforgeryAdditionalDataProvider>();
|
|
|
|
services.TryAddSingleton<ObjectPool<AntiforgerySerializationContext>>(serviceProvider =>
|
|
{
|
|
var provider = serviceProvider.GetRequiredService<ObjectPoolProvider>();
|
|
var policy = new AntiforgerySerializationContextPooledObjectPolicy();
|
|
return provider.Create(policy);
|
|
});
|
|
|
|
return services;
|
|
}
|
|
|
|
/// <summary>
|
|
/// Adds antiforgery services to the specified <see cref="IServiceCollection" />.
|
|
/// </summary>
|
|
/// <param name="services">The <see cref="IServiceCollection" /> to add services to.</param>
|
|
/// <param name="setupAction">An <see cref="Action{AntiforgeryOptions}"/> to configure the provided <see cref="AntiforgeryOptions"/>.</param>
|
|
/// <returns>The <see cref="IServiceCollection"/> so that additional calls can be chained.</returns>
|
|
public static IServiceCollection AddAntiforgery(this IServiceCollection services, Action<AntiforgeryOptions> setupAction)
|
|
{
|
|
if (services == null)
|
|
{
|
|
throw new ArgumentNullException(nameof(services));
|
|
}
|
|
|
|
if (setupAction == null)
|
|
{
|
|
throw new ArgumentNullException(nameof(setupAction));
|
|
}
|
|
|
|
services.AddAntiforgery();
|
|
services.Configure(setupAction);
|
|
return services;
|
|
}
|
|
}
|
|
}
|