443 lines
16 KiB
TypeScript
443 lines
16 KiB
TypeScript
/*
|
|
Copyright 2019, 2020 The Matrix.org Foundation C.I.C.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
import { ICryptoCallbacks, IDeviceTrustLevel, ISecretStorageKeyInfo } from 'matrix-js-sdk/src/matrix';
|
|
import { MatrixClient } from 'matrix-js-sdk/src/client';
|
|
import Modal from './Modal';
|
|
import * as sdk from './index';
|
|
import {MatrixClientPeg} from './MatrixClientPeg';
|
|
import { deriveKey } from 'matrix-js-sdk/src/crypto/key_passphrase';
|
|
import { decodeRecoveryKey } from 'matrix-js-sdk/src/crypto/recoverykey';
|
|
import { _t } from './languageHandler';
|
|
import { encodeBase64 } from "matrix-js-sdk/src/crypto/olmlib";
|
|
import { isSecureBackupRequired } from './utils/WellKnownUtils';
|
|
import AccessSecretStorageDialog from './components/views/dialogs/security/AccessSecretStorageDialog';
|
|
import RestoreKeyBackupDialog from './components/views/dialogs/security/RestoreKeyBackupDialog';
|
|
import SettingsStore from "./settings/SettingsStore";
|
|
import SecurityCustomisations from "./customisations/Security";
|
|
|
|
// This stores the secret storage private keys in memory for the JS SDK. This is
|
|
// only meant to act as a cache to avoid prompting the user multiple times
|
|
// during the same single operation. Use `accessSecretStorage` below to scope a
|
|
// single secret storage operation, as it will clear the cached keys once the
|
|
// operation ends.
|
|
let secretStorageKeys: Record<string, Uint8Array> = {};
|
|
let secretStorageKeyInfo: Record<string, ISecretStorageKeyInfo> = {};
|
|
let secretStorageBeingAccessed = false;
|
|
|
|
let nonInteractive = false;
|
|
|
|
let dehydrationCache: {
|
|
key?: Uint8Array,
|
|
keyInfo?: ISecretStorageKeyInfo,
|
|
} = {};
|
|
|
|
function isCachingAllowed(): boolean {
|
|
return secretStorageBeingAccessed;
|
|
}
|
|
|
|
/**
|
|
* This can be used by other components to check if secret storage access is in
|
|
* progress, so that we can e.g. avoid intermittently showing toasts during
|
|
* secret storage setup.
|
|
*
|
|
* @returns {bool}
|
|
*/
|
|
export function isSecretStorageBeingAccessed(): boolean {
|
|
return secretStorageBeingAccessed;
|
|
}
|
|
|
|
export class AccessCancelledError extends Error {
|
|
constructor() {
|
|
super("Secret storage access canceled");
|
|
}
|
|
}
|
|
|
|
async function confirmToDismiss(): Promise<boolean> {
|
|
const QuestionDialog = sdk.getComponent("dialogs.QuestionDialog");
|
|
const [sure] = await Modal.createDialog(QuestionDialog, {
|
|
title: _t("Cancel entering passphrase?"),
|
|
description: _t("Are you sure you want to cancel entering passphrase?"),
|
|
danger: false,
|
|
button: _t("Go Back"),
|
|
cancelButton: _t("Cancel"),
|
|
}).finished;
|
|
return !sure;
|
|
}
|
|
|
|
function makeInputToKey(
|
|
keyInfo: ISecretStorageKeyInfo,
|
|
): (keyParams: { passphrase: string, recoveryKey: string }) => Promise<Uint8Array> {
|
|
return async ({ passphrase, recoveryKey }) => {
|
|
if (passphrase) {
|
|
return deriveKey(
|
|
passphrase,
|
|
keyInfo.passphrase.salt,
|
|
keyInfo.passphrase.iterations,
|
|
);
|
|
} else {
|
|
return decodeRecoveryKey(recoveryKey);
|
|
}
|
|
};
|
|
}
|
|
|
|
async function getSecretStorageKey(
|
|
{ keys: keyInfos }: { keys: Record<string, ISecretStorageKeyInfo> },
|
|
ssssItemName,
|
|
): Promise<[string, Uint8Array]> {
|
|
const keyInfoEntries = Object.entries(keyInfos);
|
|
if (keyInfoEntries.length > 1) {
|
|
throw new Error("Multiple storage key requests not implemented");
|
|
}
|
|
const [keyId, keyInfo] = keyInfoEntries[0];
|
|
|
|
// Check the in-memory cache
|
|
if (isCachingAllowed() && secretStorageKeys[keyId]) {
|
|
return [keyId, secretStorageKeys[keyId]];
|
|
}
|
|
|
|
if (dehydrationCache.key) {
|
|
if (await MatrixClientPeg.get().checkSecretStorageKey(dehydrationCache.key, keyInfo)) {
|
|
cacheSecretStorageKey(keyId, keyInfo, dehydrationCache.key);
|
|
return [keyId, dehydrationCache.key];
|
|
}
|
|
}
|
|
|
|
const keyFromCustomisations = SecurityCustomisations.getSecretStorageKey?.();
|
|
if (keyFromCustomisations) {
|
|
console.log("Using key from security customisations (secret storage)")
|
|
cacheSecretStorageKey(keyId, keyInfo, keyFromCustomisations);
|
|
return [keyId, keyFromCustomisations];
|
|
}
|
|
|
|
if (nonInteractive) {
|
|
throw new Error("Could not unlock non-interactively");
|
|
}
|
|
|
|
const inputToKey = makeInputToKey(keyInfo);
|
|
const { finished } = Modal.createTrackedDialog("Access Secret Storage dialog", "",
|
|
AccessSecretStorageDialog,
|
|
/* props= */
|
|
{
|
|
keyInfo,
|
|
checkPrivateKey: async (input) => {
|
|
const key = await inputToKey(input);
|
|
return await MatrixClientPeg.get().checkSecretStorageKey(key, keyInfo);
|
|
},
|
|
},
|
|
/* className= */ null,
|
|
/* isPriorityModal= */ false,
|
|
/* isStaticModal= */ false,
|
|
/* options= */ {
|
|
onBeforeClose: async (reason) => {
|
|
if (reason === "backgroundClick") {
|
|
return confirmToDismiss();
|
|
}
|
|
return true;
|
|
},
|
|
},
|
|
);
|
|
const [input] = await finished;
|
|
if (!input) {
|
|
throw new AccessCancelledError();
|
|
}
|
|
const key = await inputToKey(input);
|
|
|
|
// Save to cache to avoid future prompts in the current session
|
|
cacheSecretStorageKey(keyId, keyInfo, key);
|
|
|
|
return [keyId, key];
|
|
}
|
|
|
|
export async function getDehydrationKey(
|
|
keyInfo: ISecretStorageKeyInfo,
|
|
checkFunc: (Uint8Array) => void,
|
|
): Promise<Uint8Array> {
|
|
const keyFromCustomisations = SecurityCustomisations.getSecretStorageKey?.();
|
|
if (keyFromCustomisations) {
|
|
console.log("Using key from security customisations (dehydration)")
|
|
return keyFromCustomisations;
|
|
}
|
|
|
|
const inputToKey = makeInputToKey(keyInfo);
|
|
const { finished } = Modal.createTrackedDialog("Access Secret Storage dialog", "",
|
|
AccessSecretStorageDialog,
|
|
/* props= */
|
|
{
|
|
keyInfo,
|
|
checkPrivateKey: async (input) => {
|
|
const key = await inputToKey(input);
|
|
try {
|
|
checkFunc(key);
|
|
return true;
|
|
} catch (e) {
|
|
return false;
|
|
}
|
|
},
|
|
},
|
|
/* className= */ null,
|
|
/* isPriorityModal= */ false,
|
|
/* isStaticModal= */ false,
|
|
/* options= */ {
|
|
onBeforeClose: async (reason) => {
|
|
if (reason === "backgroundClick") {
|
|
return confirmToDismiss();
|
|
}
|
|
return true;
|
|
},
|
|
},
|
|
);
|
|
const [input] = await finished;
|
|
if (!input) {
|
|
throw new AccessCancelledError();
|
|
}
|
|
const key = await inputToKey(input);
|
|
|
|
// need to copy the key because rehydration (unpickling) will clobber it
|
|
dehydrationCache = {key: new Uint8Array(key), keyInfo};
|
|
|
|
return key;
|
|
}
|
|
|
|
function cacheSecretStorageKey(
|
|
keyId: string,
|
|
keyInfo: ISecretStorageKeyInfo,
|
|
key: Uint8Array,
|
|
): void {
|
|
if (isCachingAllowed()) {
|
|
secretStorageKeys[keyId] = key;
|
|
secretStorageKeyInfo[keyId] = keyInfo;
|
|
}
|
|
}
|
|
|
|
async function onSecretRequested(
|
|
userId: string,
|
|
deviceId: string,
|
|
requestId: string,
|
|
name: string,
|
|
deviceTrust: IDeviceTrustLevel,
|
|
): Promise<string> {
|
|
console.log("onSecretRequested", userId, deviceId, requestId, name, deviceTrust);
|
|
const client = MatrixClientPeg.get();
|
|
if (userId !== client.getUserId()) {
|
|
return;
|
|
}
|
|
if (!deviceTrust || !deviceTrust.isVerified()) {
|
|
console.log(`Ignoring secret request from untrusted device ${deviceId}`);
|
|
return;
|
|
}
|
|
if (
|
|
name === "m.cross_signing.master" ||
|
|
name === "m.cross_signing.self_signing" ||
|
|
name === "m.cross_signing.user_signing"
|
|
) {
|
|
const callbacks = client.getCrossSigningCacheCallbacks();
|
|
if (!callbacks.getCrossSigningKeyCache) return;
|
|
const keyId = name.replace("m.cross_signing.", "");
|
|
const key = await callbacks.getCrossSigningKeyCache(keyId);
|
|
if (!key) {
|
|
console.log(
|
|
`${keyId} requested by ${deviceId}, but not found in cache`,
|
|
);
|
|
}
|
|
return key && encodeBase64(key);
|
|
} else if (name === "m.megolm_backup.v1") {
|
|
const key = await client._crypto.getSessionBackupPrivateKey();
|
|
if (!key) {
|
|
console.log(
|
|
`session backup key requested by ${deviceId}, but not found in cache`,
|
|
);
|
|
}
|
|
return key && encodeBase64(key);
|
|
}
|
|
console.warn("onSecretRequested didn't recognise the secret named ", name);
|
|
}
|
|
|
|
export const crossSigningCallbacks: ICryptoCallbacks = {
|
|
getSecretStorageKey,
|
|
cacheSecretStorageKey,
|
|
onSecretRequested,
|
|
getDehydrationKey,
|
|
};
|
|
|
|
export async function promptForBackupPassphrase(): Promise<Uint8Array> {
|
|
let key;
|
|
|
|
const { finished } = Modal.createTrackedDialog('Restore Backup', '', RestoreKeyBackupDialog, {
|
|
showSummary: false, keyCallback: k => key = k,
|
|
}, null, /* priority = */ false, /* static = */ true);
|
|
|
|
const success = await finished;
|
|
if (!success) throw new Error("Key backup prompt cancelled");
|
|
|
|
return key;
|
|
}
|
|
|
|
/**
|
|
* This helper should be used whenever you need to access secret storage. It
|
|
* ensures that secret storage (and also cross-signing since they each depend on
|
|
* each other in a cycle of sorts) have been bootstrapped before running the
|
|
* provided function.
|
|
*
|
|
* Bootstrapping secret storage may take one of these paths:
|
|
* 1. Create secret storage from a passphrase and store cross-signing keys
|
|
* in secret storage.
|
|
* 2. Access existing secret storage by requesting passphrase and accessing
|
|
* cross-signing keys as needed.
|
|
* 3. All keys are loaded and there's nothing to do.
|
|
*
|
|
* Additionally, the secret storage keys are cached during the scope of this function
|
|
* to ensure the user is prompted only once for their secret storage
|
|
* passphrase. The cache is then cleared once the provided function completes.
|
|
*
|
|
* @param {Function} [func] An operation to perform once secret storage has been
|
|
* bootstrapped. Optional.
|
|
* @param {bool} [forceReset] Reset secret storage even if it's already set up
|
|
*/
|
|
export async function accessSecretStorage(func = async () => { }, forceReset = false) {
|
|
const cli = MatrixClientPeg.get();
|
|
secretStorageBeingAccessed = true;
|
|
try {
|
|
if (!await cli.hasSecretStorageKey() || forceReset) {
|
|
// This dialog calls bootstrap itself after guiding the user through
|
|
// passphrase creation.
|
|
const { finished } = Modal.createTrackedDialogAsync('Create Secret Storage dialog', '',
|
|
import("./async-components/views/dialogs/security/CreateSecretStorageDialog"),
|
|
{
|
|
forceReset,
|
|
},
|
|
null,
|
|
/* priority = */ false,
|
|
/* static = */ true,
|
|
/* options = */ {
|
|
onBeforeClose: async (reason) => {
|
|
// If Secure Backup is required, you cannot leave the modal.
|
|
if (reason === "backgroundClick") {
|
|
return !isSecureBackupRequired();
|
|
}
|
|
return true;
|
|
},
|
|
},
|
|
);
|
|
const [confirmed] = await finished;
|
|
if (!confirmed) {
|
|
throw new Error("Secret storage creation canceled");
|
|
}
|
|
} else {
|
|
const InteractiveAuthDialog = sdk.getComponent("dialogs.InteractiveAuthDialog");
|
|
await cli.bootstrapCrossSigning({
|
|
authUploadDeviceSigningKeys: async (makeRequest) => {
|
|
const { finished } = Modal.createTrackedDialog(
|
|
'Cross-signing keys dialog', '', InteractiveAuthDialog,
|
|
{
|
|
title: _t("Setting up keys"),
|
|
matrixClient: cli,
|
|
makeRequest,
|
|
},
|
|
);
|
|
const [confirmed] = await finished;
|
|
if (!confirmed) {
|
|
throw new Error("Cross-signing key upload auth canceled");
|
|
}
|
|
},
|
|
});
|
|
await cli.bootstrapSecretStorage({
|
|
getKeyBackupPassphrase: promptForBackupPassphrase,
|
|
});
|
|
|
|
const keyId = Object.keys(secretStorageKeys)[0];
|
|
if (keyId && SettingsStore.getValue("feature_dehydration")) {
|
|
let dehydrationKeyInfo = {};
|
|
if (secretStorageKeyInfo[keyId] && secretStorageKeyInfo[keyId].passphrase) {
|
|
dehydrationKeyInfo = { passphrase: secretStorageKeyInfo[keyId].passphrase };
|
|
}
|
|
console.log("Setting dehydration key");
|
|
await cli.setDehydrationKey(secretStorageKeys[keyId], dehydrationKeyInfo, "Backup device");
|
|
} else if (!keyId) {
|
|
console.warn("Not setting dehydration key: no SSSS key found");
|
|
} else {
|
|
console.log("Not setting dehydration key: feature disabled");
|
|
}
|
|
}
|
|
|
|
// `return await` needed here to ensure `finally` block runs after the
|
|
// inner operation completes.
|
|
return await func();
|
|
} catch (e) {
|
|
SecurityCustomisations.catchAccessSecretStorageError?.(e);
|
|
console.error(e);
|
|
} finally {
|
|
// Clear secret storage key cache now that work is complete
|
|
secretStorageBeingAccessed = false;
|
|
if (!isCachingAllowed()) {
|
|
secretStorageKeys = {};
|
|
secretStorageKeyInfo = {};
|
|
}
|
|
}
|
|
}
|
|
|
|
// FIXME: this function name is a bit of a mouthful
|
|
export async function tryToUnlockSecretStorageWithDehydrationKey(
|
|
client: MatrixClient,
|
|
): Promise<void> {
|
|
const key = dehydrationCache.key;
|
|
let restoringBackup = false;
|
|
if (key && await client.isSecretStorageReady()) {
|
|
console.log("Trying to set up cross-signing using dehydration key");
|
|
secretStorageBeingAccessed = true;
|
|
nonInteractive = true;
|
|
try {
|
|
await client.checkOwnCrossSigningTrust();
|
|
|
|
// we also need to set a new dehydrated device to replace the
|
|
// device we rehydrated
|
|
let dehydrationKeyInfo = {};
|
|
if (dehydrationCache.keyInfo && dehydrationCache.keyInfo.passphrase) {
|
|
dehydrationKeyInfo = { passphrase: dehydrationCache.keyInfo.passphrase };
|
|
}
|
|
await client.setDehydrationKey(key, dehydrationKeyInfo, "Backup device");
|
|
|
|
// and restore from backup
|
|
const backupInfo = await client.getKeyBackupVersion();
|
|
if (backupInfo) {
|
|
restoringBackup = true;
|
|
// don't await, because this can take a long time
|
|
client.restoreKeyBackupWithSecretStorage(backupInfo)
|
|
.finally(() => {
|
|
secretStorageBeingAccessed = false;
|
|
nonInteractive = false;
|
|
if (!isCachingAllowed()) {
|
|
secretStorageKeys = {};
|
|
secretStorageKeyInfo = {};
|
|
}
|
|
});
|
|
}
|
|
} finally {
|
|
dehydrationCache = {};
|
|
// the secret storage cache is needed for restoring from backup, so
|
|
// don't clear it yet if we're restoring from backup
|
|
if (!restoringBackup) {
|
|
secretStorageBeingAccessed = false;
|
|
nonInteractive = false;
|
|
if (!isCachingAllowed()) {
|
|
secretStorageKeys = {};
|
|
secretStorageKeyInfo = {};
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|