elliot-sawyer /
totp-authenticator
| 1 | <?php |
||
| 2 | |||
| 3 | |||
| 4 | namespace ElliotSawyer\TOTPAuthenticator; |
||
| 5 | |||
| 6 | use Firesphere\BootstrapMFA\Providers\BootstrapMFAProvider; |
||
| 7 | use Firesphere\BootstrapMFA\Providers\MFAProvider; |
||
| 8 | use lfkeitel\phptotp\Base32; |
||
| 9 | use lfkeitel\phptotp\Totp; |
||
| 10 | use SilverStripe\Core\Injector\Injector; |
||
| 11 | use SilverStripe\ORM\ValidationException; |
||
| 12 | use SilverStripe\ORM\ValidationResult; |
||
| 13 | use SilverStripe\Security\Member; |
||
| 14 | use SilverStripe\Security\PasswordEncryptor_NotFoundException; |
||
| 15 | use ElliotSawyer\TOTPAuthenticator\TOTPAuthenticator; |
||
| 16 | |||
| 17 | /** |
||
| 18 | * Class TOTPProvider |
||
| 19 | * @package ElliotSawyer\TOTPAuthenticator |
||
| 20 | */ |
||
| 21 | class TOTPProvider extends BootstrapMFAProvider implements MFAProvider |
||
| 22 | { |
||
| 23 | /** |
||
| 24 | * @param string $token |
||
| 25 | * @param null $result |
||
|
0 ignored issues
–
show
Documentation
Bug
introduced
by
Loading history...
|
|||
| 26 | * @return bool|Member |
||
| 27 | * @throws ValidationException |
||
| 28 | * @throws PasswordEncryptor_NotFoundException |
||
| 29 | * @throws \Exception |
||
| 30 | */ |
||
| 31 | public function verifyToken($token, &$result = null) |
||
| 32 | { |
||
| 33 | if (!$result) { |
||
|
0 ignored issues
–
show
|
|||
| 34 | $result = Injector::inst()->get(ValidationResult::class); |
||
| 35 | } |
||
| 36 | $member = $this->getMember(); |
||
| 37 | if ($member && $member->ID) { |
||
| 38 | if (!$token) { |
||
| 39 | $result->addError(_t(self::class . '.INVALIDORMISSINGTOKEN', 'Invalid or missing second factor token')); |
||
| 40 | } else { |
||
| 41 | $secret = Base32::decode($member->TOTPSecret); |
||
| 42 | $algorithm = TOTPAuthenticator::get_algorithm(); |
||
| 43 | |||
| 44 | $totp = new Totp($algorithm); |
||
| 45 | $key = $totp->GenerateToken($secret); |
||
| 46 | $user_submitted_key = $token; |
||
| 47 | if ($user_submitted_key !== $key) { |
||
| 48 | $result->addError( |
||
| 49 | _t(self::class . '.INVALIDORMISSINGTOKEN', 'Invalid or missing second factor token') |
||
| 50 | ); |
||
| 51 | } else { |
||
| 52 | return $this->member; |
||
| 53 | } |
||
| 54 | } |
||
| 55 | } |
||
| 56 | |||
| 57 | return parent::verifyToken($token, $result); |
||
| 58 | } |
||
| 59 | } |
||
| 60 |