2015-12-08 22:57:38 -05:00
|
|
|
|
using System;
|
|
|
|
|
|
using System.Threading.Tasks;
|
2016-05-19 19:10:24 -04:00
|
|
|
|
using Microsoft.AspNetCore.Identity;
|
2017-03-08 21:45:08 -05:00
|
|
|
|
using Bit.Core.Models.Table;
|
2015-12-08 22:57:38 -05:00
|
|
|
|
using Bit.Core.Enums;
|
2017-01-13 21:34:54 -05:00
|
|
|
|
using OtpNet;
|
2015-12-08 22:57:38 -05:00
|
|
|
|
|
|
|
|
|
|
namespace Bit.Core.Identity
|
|
|
|
|
|
{
|
2016-05-19 19:10:24 -04:00
|
|
|
|
public class AuthenticatorTokenProvider : IUserTwoFactorTokenProvider<User>
|
2015-12-08 22:57:38 -05:00
|
|
|
|
{
|
|
|
|
|
|
public Task<bool> CanGenerateTwoFactorTokenAsync(UserManager<User> manager, User user)
|
|
|
|
|
|
{
|
2017-06-07 14:14:34 -04:00
|
|
|
|
var provider = user.GetTwoFactorProvider(TwoFactorProviderType.Authenticator);
|
|
|
|
|
|
|
|
|
|
|
|
var canGenerate = user.TwoFactorProviderIsEnabled(TwoFactorProviderType.Authenticator)
|
|
|
|
|
|
&& !string.IsNullOrWhiteSpace(provider.MetaData["Key"]);
|
2017-06-06 23:19:42 -04:00
|
|
|
|
|
2015-12-08 22:57:38 -05:00
|
|
|
|
return Task.FromResult(canGenerate);
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
public Task<string> GenerateAsync(string purpose, UserManager<User> manager, User user)
|
|
|
|
|
|
{
|
|
|
|
|
|
return Task.FromResult<string>(null);
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
public Task<bool> ValidateAsync(string purpose, string token, UserManager<User> manager, User user)
|
|
|
|
|
|
{
|
2017-06-07 14:14:34 -04:00
|
|
|
|
var provider = user.GetTwoFactorProvider(TwoFactorProviderType.Authenticator);
|
|
|
|
|
|
var otp = new Totp(Base32Encoding.ToBytes(provider.MetaData["Key"]));
|
2015-12-08 22:57:38 -05:00
|
|
|
|
|
|
|
|
|
|
long timeStepMatched;
|
2017-01-28 22:11:54 -05:00
|
|
|
|
var valid = otp.VerifyTotp(token, out timeStepMatched, new VerificationWindow(1, 1));
|
2015-12-08 22:57:38 -05:00
|
|
|
|
|
|
|
|
|
|
return Task.FromResult(valid);
|
|
|
|
|
|
}
|
|
|
|
|
|
}
|
|
|
|
|
|
}
|