summaryrefslogtreecommitdiff
path: root/app/models/u2f_registration.rb
blob: 1a389081913548419ec358e0f3953cb9b58d3c4d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
# frozen_string_literal: true

# Registration information for U2F (universal 2nd factor) devices, like Yubikeys

class U2fRegistration < ApplicationRecord
  belongs_to :user
  after_commit :schedule_webauthn_migration, on: :create
  after_commit :update_webauthn_registration, on: :update, if: :counter_changed?

  def schedule_webauthn_migration
    BackgroundMigrationWorker.perform_async('MigrateU2fWebauthn', [id, id])
  end

  def update_webauthn_registration
    # When we update the sign count of this registration
    # we need to update the sign count of the corresponding webauthn registration
    # as well if it exists already
    WebauthnRegistration.find_by_credential_xid(webauthn_credential_xid)&.update_attribute(:counter, counter)
  end

  def self.register(user, app_id, params, challenges)
    u2f = U2F::U2F.new(app_id)
    registration = self.new

    begin
      response = U2F::RegisterResponse.load_from_json(params[:device_response])
      registration_data = u2f.register!(challenges, response)
      registration.update(certificate: registration_data.certificate,
                          key_handle: registration_data.key_handle,
                          public_key: registration_data.public_key,
                          counter: registration_data.counter,
                          user: user,
                          name: params[:name])
    rescue JSON::ParserError, NoMethodError, ArgumentError
      registration.errors.add(:base, _('Your U2F device did not send a valid JSON response.'))
    rescue U2F::Error => e
      registration.errors.add(:base, e.message)
    end

    registration
  end

  def self.authenticate(user, app_id, json_response, challenges)
    response = U2F::SignResponse.load_from_json(json_response)
    registration = user.u2f_registrations.find_by_key_handle(response.key_handle)
    u2f = U2F::U2F.new(app_id)

    if registration
      u2f.authenticate!(challenges, response, Base64.decode64(registration.public_key), registration.counter)
      registration.update(counter: response.counter)
      true
    end
  rescue JSON::ParserError, NoMethodError, ArgumentError, U2F::Error
    false
  end

  private

  def webauthn_credential_xid
    # To find the corresponding webauthn registration, we use that
    # the key handle of the u2f reg corresponds to the credential xid of the webauthn reg
    # (with some base64 back and forth)
    Base64.strict_encode64(Base64.urlsafe_decode64(key_handle))
  end
end