зеркало из https://github.com/mozilla/gecko-dev.git
151 строка
3.6 KiB
C++
151 строка
3.6 KiB
C++
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
#ifndef mozilla_dom_U2FHIDTokenManager_h
|
|
#define mozilla_dom_U2FHIDTokenManager_h
|
|
|
|
#include "mozilla/dom/U2FTokenTransport.h"
|
|
#include "u2f-hid-rs/src/u2fhid-capi.h"
|
|
|
|
/*
|
|
* U2FHIDTokenManager is a Rust implementation of a secure token manager
|
|
* for the U2F and WebAuthn APIs, talking to HIDs.
|
|
*/
|
|
|
|
namespace mozilla {
|
|
namespace dom {
|
|
|
|
class U2FAppIds {
|
|
public:
|
|
explicit U2FAppIds(const nsTArray<nsTArray<uint8_t>>& aApplications)
|
|
{
|
|
mAppIds = rust_u2f_app_ids_new();
|
|
|
|
for (auto& app_id: aApplications) {
|
|
rust_u2f_app_ids_add(mAppIds, app_id.Elements(), app_id.Length());
|
|
}
|
|
}
|
|
|
|
rust_u2f_app_ids* Get() { return mAppIds; }
|
|
|
|
~U2FAppIds() { rust_u2f_app_ids_free(mAppIds); }
|
|
|
|
private:
|
|
rust_u2f_app_ids* mAppIds;
|
|
};
|
|
|
|
class U2FKeyHandles {
|
|
public:
|
|
explicit U2FKeyHandles(const nsTArray<WebAuthnScopedCredential>& aCredentials)
|
|
{
|
|
mKeyHandles = rust_u2f_khs_new();
|
|
|
|
for (auto& cred: aCredentials) {
|
|
rust_u2f_khs_add(mKeyHandles,
|
|
cred.id().Elements(),
|
|
cred.id().Length(),
|
|
cred.transports());
|
|
}
|
|
}
|
|
|
|
rust_u2f_key_handles* Get() { return mKeyHandles; }
|
|
|
|
~U2FKeyHandles() { rust_u2f_khs_free(mKeyHandles); }
|
|
|
|
private:
|
|
rust_u2f_key_handles* mKeyHandles;
|
|
};
|
|
|
|
class U2FResult {
|
|
public:
|
|
explicit U2FResult(uint64_t aTransactionId, rust_u2f_result* aResult)
|
|
: mTransactionId(aTransactionId)
|
|
, mResult(aResult)
|
|
{ }
|
|
|
|
~U2FResult() { rust_u2f_res_free(mResult); }
|
|
|
|
uint64_t GetTransactionId() { return mTransactionId; }
|
|
|
|
bool CopyRegistration(nsTArray<uint8_t>& aBuffer)
|
|
{
|
|
return CopyBuffer(U2F_RESBUF_ID_REGISTRATION, aBuffer);
|
|
}
|
|
|
|
bool CopyKeyHandle(nsTArray<uint8_t>& aBuffer)
|
|
{
|
|
return CopyBuffer(U2F_RESBUF_ID_KEYHANDLE, aBuffer);
|
|
}
|
|
|
|
bool CopySignature(nsTArray<uint8_t>& aBuffer)
|
|
{
|
|
return CopyBuffer(U2F_RESBUF_ID_SIGNATURE, aBuffer);
|
|
}
|
|
|
|
bool CopyAppId(nsTArray<uint8_t>& aBuffer)
|
|
{
|
|
return CopyBuffer(U2F_RESBUF_ID_APPID, aBuffer);
|
|
}
|
|
|
|
private:
|
|
bool CopyBuffer(uint8_t aResBufID, nsTArray<uint8_t>& aBuffer) {
|
|
if (!mResult) {
|
|
return false;
|
|
}
|
|
|
|
size_t len;
|
|
if (!rust_u2f_resbuf_length(mResult, aResBufID, &len)) {
|
|
return false;
|
|
}
|
|
|
|
if (!aBuffer.SetLength(len, fallible)) {
|
|
return false;
|
|
}
|
|
|
|
return rust_u2f_resbuf_copy(mResult, aResBufID, aBuffer.Elements());
|
|
}
|
|
|
|
uint64_t mTransactionId;
|
|
rust_u2f_result* mResult;
|
|
};
|
|
|
|
class U2FHIDTokenManager final : public U2FTokenTransport
|
|
{
|
|
public:
|
|
explicit U2FHIDTokenManager();
|
|
|
|
RefPtr<U2FRegisterPromise>
|
|
Register(const WebAuthnMakeCredentialInfo& aInfo) override;
|
|
|
|
RefPtr<U2FSignPromise>
|
|
Sign(const WebAuthnGetAssertionInfo& aInfo) override;
|
|
|
|
void Cancel() override;
|
|
void Drop() override;
|
|
|
|
void HandleRegisterResult(UniquePtr<U2FResult>&& aResult);
|
|
void HandleSignResult(UniquePtr<U2FResult>&& aResult);
|
|
|
|
private:
|
|
~U2FHIDTokenManager() { }
|
|
|
|
void ClearPromises() {
|
|
mRegisterPromise.RejectIfExists(NS_ERROR_DOM_UNKNOWN_ERR, __func__);
|
|
mSignPromise.RejectIfExists(NS_ERROR_DOM_UNKNOWN_ERR, __func__);
|
|
}
|
|
|
|
rust_u2f_manager* mU2FManager;
|
|
uint64_t mTransactionId;
|
|
nsTArray<uint8_t> mCurrentAppId;
|
|
MozPromiseHolder<U2FRegisterPromise> mRegisterPromise;
|
|
MozPromiseHolder<U2FSignPromise> mSignPromise;
|
|
};
|
|
|
|
} // namespace dom
|
|
} // namespace mozilla
|
|
|
|
#endif // mozilla_dom_U2FHIDTokenManager_h
|