mirror of
https://github.com/laurent22/joplin.git
synced 2024-12-24 10:27:10 +02:00
273 lines
8.4 KiB
TypeScript
273 lines
8.4 KiB
TypeScript
const BaseCommand = require('./base-command').default;
|
|
import { _ } from '@joplin/lib/locale';
|
|
import EncryptionService from '@joplin/lib/services/e2ee/EncryptionService';
|
|
import DecryptionWorker from '@joplin/lib/services/DecryptionWorker';
|
|
import BaseItem from '@joplin/lib/models/BaseItem';
|
|
import Setting from '@joplin/lib/models/Setting';
|
|
import shim from '@joplin/lib/shim';
|
|
import * as pathUtils from '@joplin/lib/path-utils';
|
|
import { getEncryptionEnabled, localSyncInfo } from '@joplin/lib/services/synchronizer/syncInfoUtils';
|
|
import { generateMasterKeyAndEnableEncryption, loadMasterKeysFromSettings, masterPasswordIsValid, setupAndDisableEncryption } from '@joplin/lib/services/e2ee/utils';
|
|
const imageType = require('image-type');
|
|
const readChunk = require('read-chunk');
|
|
|
|
class Command extends BaseCommand {
|
|
usage() {
|
|
return 'e2ee <command> [path]';
|
|
}
|
|
|
|
description() {
|
|
return _('Manages E2EE configuration. Commands are `enable`, `disable`, `decrypt`, `status`, `decrypt-file`, and `target-status`.'); // `generate-ppk`
|
|
}
|
|
|
|
options() {
|
|
return [
|
|
// This is here mostly for testing - shouldn't be used
|
|
['-p, --password <password>', 'Use this password as master password (For security reasons, it is not recommended to use this option).'],
|
|
['-v, --verbose', 'More verbose output for the `target-status` command'],
|
|
['-o, --output <directory>', 'Output directory'],
|
|
['--retry-failed-items', 'Applies to `decrypt` command - retries decrypting items that previously could not be decrypted.'],
|
|
];
|
|
}
|
|
|
|
async action(args: any) {
|
|
const options = args.options;
|
|
|
|
const askForMasterKey = async (error: any) => {
|
|
const masterKeyId = error.masterKeyId;
|
|
const password = await this.prompt(_('Enter master password:'), { type: 'string', secure: true });
|
|
if (!password) {
|
|
this.stdout(_('Operation cancelled'));
|
|
return false;
|
|
}
|
|
|
|
const masterKey = localSyncInfo().masterKeys.find(mk => mk.id === masterKeyId);
|
|
if (!(await masterPasswordIsValid(password, masterKey))) {
|
|
this.stdout(_('Invalid password'));
|
|
return false;
|
|
}
|
|
|
|
Setting.setObjectValue('encryption.passwordCache', masterKeyId, password);
|
|
await loadMasterKeysFromSettings(EncryptionService.instance());
|
|
return true;
|
|
};
|
|
|
|
const startDecryption = async () => {
|
|
this.stdout(_('Starting decryption... Please wait as it may take several minutes depending on how much there is to decrypt.'));
|
|
while (true) {
|
|
try {
|
|
const result = await DecryptionWorker.instance().start();
|
|
|
|
if (result.error) throw result.error;
|
|
|
|
const line = [];
|
|
line.push(_('Decrypted items: %d', result.decryptedItemCount));
|
|
if (result.skippedItemCount) line.push(_('Skipped items: %d (use --retry-failed-items to retry decrypting them)', result.skippedItemCount));
|
|
this.stdout(line.join('\n'));
|
|
break;
|
|
} catch (error) {
|
|
if (error.code === 'masterKeyNotLoaded') {
|
|
const ok = await askForMasterKey(error);
|
|
if (!ok) return;
|
|
continue;
|
|
}
|
|
|
|
throw error;
|
|
}
|
|
}
|
|
|
|
this.stdout(_('Completed decryption.'));
|
|
};
|
|
|
|
if (args.command === 'enable') {
|
|
const argPassword = options.password ? options.password.toString() : '';
|
|
const password = argPassword ? argPassword : await this.prompt(_('Enter master password:'), { type: 'string', secure: true });
|
|
if (!password) {
|
|
this.stdout(_('Operation cancelled'));
|
|
return;
|
|
}
|
|
|
|
// If the password was passed via command line, we don't ask for
|
|
// confirmation. This is to allow setting up E2EE entirely from the
|
|
// command line.
|
|
if (!argPassword) {
|
|
const password2 = await this.prompt(_('Confirm password:'), { type: 'string', secure: true });
|
|
if (!password2) {
|
|
this.stdout(_('Operation cancelled'));
|
|
return;
|
|
}
|
|
if (password !== password2) {
|
|
this.stdout(_('Passwords do not match!'));
|
|
return;
|
|
}
|
|
}
|
|
|
|
await generateMasterKeyAndEnableEncryption(EncryptionService.instance(), password);
|
|
return;
|
|
}
|
|
|
|
if (args.command === 'disable') {
|
|
await setupAndDisableEncryption(EncryptionService.instance());
|
|
return;
|
|
}
|
|
|
|
if (args.command === 'decrypt') {
|
|
if (args.path) {
|
|
const plainText = await EncryptionService.instance().decryptString(args.path);
|
|
this.stdout(plainText);
|
|
} else {
|
|
if (args.options['retry-failed-items']) await DecryptionWorker.instance().clearDisabledItems();
|
|
await startDecryption();
|
|
}
|
|
|
|
return;
|
|
}
|
|
|
|
if (args.command === 'status') {
|
|
this.stdout(_('Encryption is: %s', getEncryptionEnabled() ? _('Enabled') : _('Disabled')));
|
|
return;
|
|
}
|
|
|
|
if (args.command === 'decrypt-file') {
|
|
while (true) {
|
|
try {
|
|
const outputDir = options.output ? options.output : require('os').tmpdir();
|
|
let outFile = `${outputDir}/${pathUtils.filename(args.path)}.${Date.now()}.bin`;
|
|
await EncryptionService.instance().decryptFile(args.path, outFile);
|
|
const buffer = await readChunk(outFile, 0, 64);
|
|
const detectedType = imageType(buffer);
|
|
|
|
if (detectedType) {
|
|
const newOutFile = `${outFile}.${detectedType.ext}`;
|
|
await shim.fsDriver().move(outFile, newOutFile);
|
|
outFile = newOutFile;
|
|
}
|
|
|
|
this.stdout(outFile);
|
|
break;
|
|
} catch (error) {
|
|
if (error.code === 'masterKeyNotLoaded') {
|
|
const ok = await askForMasterKey(error);
|
|
if (!ok) return;
|
|
continue;
|
|
}
|
|
|
|
throw error;
|
|
}
|
|
}
|
|
return;
|
|
}
|
|
|
|
// if (args.command === 'generate-ppk') {
|
|
// const syncInfo = localSyncInfo();
|
|
// if (syncInfo.ppk) throw new Error('This account already has a public-private key pair');
|
|
|
|
// const argPassword = options.password ? options.password.toString() : '';
|
|
// if (!argPassword) throw new Error('Password must be provided'); // TODO: should get from prompt
|
|
// const ppk = await generateKeyPair(EncryptionService.instance(), argPassword);
|
|
|
|
// syncInfo.ppk = ppk;
|
|
// saveLocalSyncInfo(syncInfo);
|
|
// await Setting.saveAll();
|
|
// }
|
|
|
|
if (args.command === 'target-status') {
|
|
const fs = require('fs-extra');
|
|
|
|
const targetPath = args.path;
|
|
if (!targetPath) throw new Error('Please specify the sync target path.');
|
|
|
|
const dirPaths = function(targetPath: string) {
|
|
const paths: string[] = [];
|
|
fs.readdirSync(targetPath).forEach((path: string) => {
|
|
paths.push(path);
|
|
});
|
|
return paths;
|
|
};
|
|
|
|
let itemCount = 0;
|
|
let resourceCount = 0;
|
|
let encryptedItemCount = 0;
|
|
let encryptedResourceCount = 0;
|
|
let otherItemCount = 0;
|
|
|
|
const encryptedPaths = [];
|
|
const decryptedPaths = [];
|
|
|
|
const paths = dirPaths(targetPath);
|
|
|
|
for (let i = 0; i < paths.length; i++) {
|
|
const path = paths[i];
|
|
const fullPath = `${targetPath}/${path}`;
|
|
const stat = await fs.stat(fullPath);
|
|
|
|
// this.stdout(fullPath);
|
|
|
|
if (path === '.resource') {
|
|
const resourcePaths = dirPaths(fullPath);
|
|
for (let j = 0; j < resourcePaths.length; j++) {
|
|
const resourcePath = resourcePaths[j];
|
|
resourceCount++;
|
|
const fullResourcePath = `${fullPath}/${resourcePath}`;
|
|
const isEncrypted = await EncryptionService.instance().fileIsEncrypted(fullResourcePath);
|
|
if (isEncrypted) {
|
|
encryptedResourceCount++;
|
|
encryptedPaths.push(fullResourcePath);
|
|
} else {
|
|
decryptedPaths.push(fullResourcePath);
|
|
}
|
|
}
|
|
} else if (stat.isDirectory()) {
|
|
continue;
|
|
} else {
|
|
const content = await fs.readFile(fullPath, 'utf8');
|
|
const item = await BaseItem.unserialize(content);
|
|
const ItemClass = BaseItem.itemClass(item);
|
|
|
|
if (!ItemClass.encryptionSupported()) {
|
|
otherItemCount++;
|
|
continue;
|
|
}
|
|
|
|
itemCount++;
|
|
|
|
const isEncrypted = await EncryptionService.instance().itemIsEncrypted(item);
|
|
|
|
if (isEncrypted) {
|
|
encryptedItemCount++;
|
|
encryptedPaths.push(fullPath);
|
|
} else {
|
|
decryptedPaths.push(fullPath);
|
|
}
|
|
}
|
|
}
|
|
|
|
this.stdout(`Encrypted items: ${encryptedItemCount}/${itemCount}`);
|
|
this.stdout(`Encrypted resources: ${encryptedResourceCount}/${resourceCount}`);
|
|
this.stdout(`Other items (never encrypted): ${otherItemCount}`);
|
|
|
|
if (options.verbose) {
|
|
this.stdout('');
|
|
this.stdout('# Encrypted paths');
|
|
this.stdout('');
|
|
for (let i = 0; i < encryptedPaths.length; i++) {
|
|
const path = encryptedPaths[i];
|
|
this.stdout(path);
|
|
}
|
|
|
|
this.stdout('');
|
|
this.stdout('# Decrypted paths');
|
|
this.stdout('');
|
|
for (let i = 0; i < decryptedPaths.length; i++) {
|
|
const path = decryptedPaths[i];
|
|
this.stdout(path);
|
|
}
|
|
}
|
|
|
|
return;
|
|
}
|
|
}
|
|
}
|
|
|
|
module.exports = Command;
|