mirror of
https://github.com/simple-login/app.git
synced 2024-09-28 20:51:29 +02:00
63 lines
2.1 KiB
Python
63 lines
2.1 KiB
Python
|
import uuid
|
||
|
import json
|
||
|
import secrets
|
||
|
import webauthn
|
||
|
from app.config import URL as SITE_URL
|
||
|
from urllib.parse import urlparse
|
||
|
|
||
|
from flask import render_template, flash, redirect, url_for, session
|
||
|
from flask_login import login_required, current_user
|
||
|
from flask_wtf import FlaskForm
|
||
|
from wtforms import HiddenField, validators
|
||
|
|
||
|
from app.dashboard.base import dashboard_bp
|
||
|
from app.extensions import db
|
||
|
from app.log import LOG
|
||
|
|
||
|
|
||
|
class FidoTokenForm(FlaskForm):
|
||
|
sk_assertion = HiddenField("sk_assertion", validators=[validators.DataRequired()])
|
||
|
|
||
|
|
||
|
@dashboard_bp.route("/fido_setup", methods=["GET", "POST"])
|
||
|
@login_required
|
||
|
def fido_setup():
|
||
|
if current_user.fido_uuid is not None:
|
||
|
flash("You have already registered your security key", "warning")
|
||
|
return redirect(url_for("dashboard.index"))
|
||
|
|
||
|
fido_token_form = FidoTokenForm()
|
||
|
|
||
|
# Prepare infomation for key registration process
|
||
|
rp_id = urlparse(SITE_URL).hostname
|
||
|
fido_uuid = str(uuid.uuid4())
|
||
|
challenge = secrets.token_urlsafe(32)
|
||
|
|
||
|
credential_create_options = webauthn.WebAuthnMakeCredentialOptions(
|
||
|
challenge, 'Simple Login', rp_id, fido_uuid,
|
||
|
current_user.email, current_user.name, False, attestation='none')
|
||
|
|
||
|
# Don't think this one should be used, but it's not configurable by arguments
|
||
|
# https://www.w3.org/TR/webauthn/#sctn-location-extension
|
||
|
registration_dict = credential_create_options.registration_dict
|
||
|
del registration_dict['extensions']['webauthn.loc']
|
||
|
|
||
|
session['fido_uuid'] = fido_uuid
|
||
|
session['fido_challenge'] = challenge.rstrip('=')
|
||
|
|
||
|
if fido_token_form.validate_on_submit():
|
||
|
sk_assertion = fido_token_form.sk_assertion.data
|
||
|
LOG.d(sk_assertion)
|
||
|
# if totp.verify(token):
|
||
|
# current_user.enable_otp = True
|
||
|
# db.session.commit()
|
||
|
# flash("Security key has been activated", "success")
|
||
|
# return redirect(url_for("dashboard.index"))
|
||
|
# else:
|
||
|
# flash("Incorrect challenge", "warning")
|
||
|
|
||
|
return render_template(
|
||
|
"dashboard/fido_setup.html", fido_token_form=fido_token_form,
|
||
|
credential_create_options=registration_dict
|
||
|
)
|