Upstream/bitwarden-server
1
0
Fork 0
mirror of https://github.com/bitwarden/server.git synced 2024-12-28 17:57:37 +01:00
Code Issues Projects Releases Wiki Activity

token providers cant inject userservice

Browse Source
This commit is contained in:
Kyle Spearrin 2018-08-28 22:21:13 -04:00
parent cf73b168ee
commit db3316dd60
4 changed files with 39 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
src/Core/Identity/AuthenticatorTokenProvider.cs
View File

@ -5,16 +5,17 @@ using Bit.Core.Models.Table;
using Bit.Core.Enums;
using OtpNet;
using Bit.Core.Services;
using Microsoft.Extensions.DependencyInjection;
namespace Bit.Core.Identity
{
public class AuthenticatorTokenProvider : IUserTwoFactorTokenProvider<User>
{
private readonly IUserService _userService;
private readonly IServiceProvider _serviceProvider;
public AuthenticatorTokenProvider(IUserService userService)
public AuthenticatorTokenProvider(IServiceProvider serviceProvider)
{
_userService = userService;
_serviceProvider = serviceProvider;
}
public async Task<bool> CanGenerateTwoFactorTokenAsync(UserManager<User> manager, User user)
@ -24,7 +25,8 @@ namespace Bit.Core.Identity
{
return false;
}
return await user.TwoFactorProviderIsEnabledAsync(TwoFactorProviderType.Authenticator, _userService);
return await user.TwoFactorProviderIsEnabledAsync(TwoFactorProviderType.Authenticator,
_serviceProvider.GetRequiredService<IUserService>());
}
public Task<string> GenerateAsync(string purpose, UserManager<User> manager, User user)

19
src/Core/Identity/DuoWebTokenProvider.cs
View File

@ -4,26 +4,29 @@ using Bit.Core.Models.Table;
using Bit.Core.Enums;
using Bit.Core.Utilities.Duo;
using Bit.Core.Models;
using System;
using Microsoft.Extensions.DependencyInjection;
using Bit.Core.Services;
namespace Bit.Core.Identity
{
public class DuoWebTokenProvider : IUserTwoFactorTokenProvider<User>
{
private readonly IUserService _userService;
private readonly IServiceProvider _serviceProvider;
private readonly GlobalSettings _globalSettings;
public DuoWebTokenProvider(
IUserService userService,
IServiceProvider serviceProvider,
GlobalSettings globalSettings)
{
_userService = userService;
_serviceProvider = serviceProvider;
_globalSettings = globalSettings;
}
public async Task<bool> CanGenerateTwoFactorTokenAsync(UserManager<User> manager, User user)
{
if(!(await _userService.CanAccessPremium(user)))
var userService = _serviceProvider.GetRequiredService<IUserService>();
if(!(await userService.CanAccessPremium(user)))
{
return false;
}
@ -34,12 +37,13 @@ namespace Bit.Core.Identity
return false;
}
return await user.TwoFactorProviderIsEnabledAsync(TwoFactorProviderType.Duo, _userService);
return await user.TwoFactorProviderIsEnabledAsync(TwoFactorProviderType.Duo, userService);
}
public async Task<string> GenerateAsync(string purpose, UserManager<User> manager, User user)
{
if(!(await _userService.CanAccessPremium(user)))
var userService = _serviceProvider.GetRequiredService<IUserService>();
if(!(await userService.CanAccessPremium(user)))
{
return null;
}
@ -57,7 +61,8 @@ namespace Bit.Core.Identity
public async Task<bool> ValidateAsync(string purpose, string token, UserManager<User> manager, User user)
{
if(!(await _userService.CanAccessPremium(user)))
var userService = _serviceProvider.GetRequiredService<IUserService>();
if(!(await userService.CanAccessPremium(user)))
{
return false;
}

18
src/Core/Identity/U2fTokenProvider.cs
View File

@ -12,28 +12,30 @@ using U2F.Core.Models;
using U2F.Core.Exceptions;
using System;
using Bit.Core.Services;
using Microsoft.Extensions.DependencyInjection;
namespace Bit.Core.Identity
{
public class U2fTokenProvider : IUserTwoFactorTokenProvider<User>
{
private readonly IUserService _userService;
private readonly IServiceProvider _serviceProvider;
private readonly IU2fRepository _u2fRepository;
private readonly GlobalSettings _globalSettings;
public U2fTokenProvider(
IUserService userService,
IServiceProvider serviceProvider,
IU2fRepository u2fRepository,
GlobalSettings globalSettings)
{
_userService = userService;
_serviceProvider = serviceProvider;
_u2fRepository = u2fRepository;
_globalSettings = globalSettings;
}
public async Task<bool> CanGenerateTwoFactorTokenAsync(UserManager<User> manager, User user)
{
if(!(await _userService.CanAccessPremium(user)))
var userService = _serviceProvider.GetRequiredService<IUserService>();
if(!(await userService.CanAccessPremium(user)))
{
return false;
}
@ -44,12 +46,13 @@ namespace Bit.Core.Identity
return false;
}
return await user.TwoFactorProviderIsEnabledAsync(TwoFactorProviderType.U2f, _userService);
return await user.TwoFactorProviderIsEnabledAsync(TwoFactorProviderType.U2f, userService);
}
public async Task<string> GenerateAsync(string purpose, UserManager<User> manager, User user)
{
if(!(await _userService.CanAccessPremium(user)))
var userService = _serviceProvider.GetRequiredService<IUserService>();
if(!(await userService.CanAccessPremium(user)))
{
return null;
}
@ -108,7 +111,8 @@ namespace Bit.Core.Identity
public async Task<bool> ValidateAsync(string purpose, string token, UserManager<User> manager, User user)
{
if(!(await _userService.CanAccessPremium(user)) || string.IsNullOrWhiteSpace(token))
var userService = _serviceProvider.GetRequiredService<IUserService>();
if(!(await userService.CanAccessPremium(user)) || string.IsNullOrWhiteSpace(token))
{
return false;
}

16
src/Core/Identity/YubicoOtpTokenProvider.cs
View File

@ -5,25 +5,28 @@ using Bit.Core.Enums;
using YubicoDotNetClient;
using System.Linq;
using Bit.Core.Services;
using System;
using Microsoft.Extensions.DependencyInjection;
namespace Bit.Core.Identity
{
public class YubicoOtpTokenProvider : IUserTwoFactorTokenProvider<User>
{
private readonly IUserService _userService;
private readonly IServiceProvider _serviceProvider;
private readonly GlobalSettings _globalSettings;
public YubicoOtpTokenProvider(
IUserService userService,
IServiceProvider serviceProvider,
GlobalSettings globalSettings)
{
_userService = userService;
_serviceProvider = serviceProvider;
_globalSettings = globalSettings;
}
public async Task<bool> CanGenerateTwoFactorTokenAsync(UserManager<User> manager, User user)
{
if(!(await _userService.CanAccessPremium(user)))
var userService = _serviceProvider.GetRequiredService<IUserService>();
if(!(await userService.CanAccessPremium(user)))
{
return false;
}
@ -34,7 +37,7 @@ namespace Bit.Core.Identity
return false;
}
return await user.TwoFactorProviderIsEnabledAsync(TwoFactorProviderType.YubiKey, _userService);
return await user.TwoFactorProviderIsEnabledAsync(TwoFactorProviderType.YubiKey, userService);
}
public Task<string> GenerateAsync(string purpose, UserManager<User> manager, User user)
@ -44,7 +47,8 @@ namespace Bit.Core.Identity
public async Task<bool> ValidateAsync(string purpose, string token, UserManager<User> manager, User user)
{
if(!(await _userService.CanAccessPremium(user)))
var userService = _serviceProvider.GetRequiredService<IUserService>();
if(!(await userService.CanAccessPremium(user)))
{
return false;
}