bitwarden/server
1
0
Fork 0
mirror of https://github.com/bitwarden/server synced 2025-12-31 23:53:17 +00:00
Code Issues 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
Download Patch File Download Diff File Expand all files Collapse all files

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;
}