Page MenuHomePhorge
Files F32934030

D7293.1768243282.diff
No OneTemporary
Actions

Size
2 KB
Referenced Files
None
Subscribers
None

D7293.1768243282.diff
View Options

diff --git a/web/media/encryption-utils.js b/web/media/encryption-utils.js
--- a/web/media/encryption-utils.js
+++ b/web/media/encryption-utils.js
@@ -2,16 +2,91 @@
import invariant from 'invariant';
-import { hexToUintArray } from 'lib/media/data-utils.js';
+import { hexToUintArray, uintArrayToHexString } from 'lib/media/data-utils.js';
import { fileInfoFromData } from 'lib/media/file-utils.js';
-import type { MediaMissionFailure } from 'lib/types/media-types.js';
+import type {
+ MediaMissionFailure,
+ MediaMissionStep,
+} from 'lib/types/media-types.js';
import { getMessageForException } from 'lib/utils/errors.js';
-import { unpad } from 'lib/utils/pkcs7-padding.js';
+import { calculatePaddedLength, pad, unpad } from 'lib/utils/pkcs7-padding.js';
import * as AES from './aes-crypto-utils.js';
const PADDING_THRESHOLD = 5000000; // 5MB
+type EncryptFileResult = {
+ +success: true,
+ +file: File,
+ +uri: string,
+ +encryptionKey: string,
+};
+
+async function encryptFile(input: File): Promise<{
+ steps: $ReadOnlyArray<MediaMissionStep>,
+ result: EncryptFileResult | MediaMissionFailure,
+}> {
+ const steps = [];
+ let success = true,
+ exceptionMessage;
+
+ // Step 1: Read the file into an ArrayBuffer
+ let data;
+ const arrayBufferStart = Date.now();
+ try {
+ const inputBuffer = await input.arrayBuffer();
+ data = new Uint8Array(inputBuffer);
+ } catch (e) {
+ success = false;
+ exceptionMessage = getMessageForException(e);
+ }
+ steps.push({
+ step: 'array_buffer_from_blob',
+ success,
+ exceptionMessage,
+ time: Date.now() - arrayBufferStart,
+ });
+ if (!success || !data) {
+ return { steps, result: { success: false, reason: 'array_buffer_failed' } };
+ }
+
+ // Step 2: Encrypt the data
+ const startEncrypt = Date.now();
+ const paddedLength = calculatePaddedLength(data.length);
+ const shouldPad = paddedLength <= PADDING_THRESHOLD;
+ let key, encryptedData;
+ try {
+ const plaintextData = shouldPad ? pad(data) : data;
+ key = await AES.generateKey();
+ encryptedData = await AES.encrypt(key, plaintextData);
+ } catch (e) {
+ success = false;
+ exceptionMessage = getMessageForException(e);
+ }
+ steps.push({
+ step: 'encrypt_data',
+ dataSize: encryptedData?.byteLength ?? -1,
+ isPadded: shouldPad,
+ time: Date.now() - startEncrypt,
+ success,
+ exceptionMessage,
+ });
+ if (!success || !encryptedData || !key) {
+ return { steps, result: { success: false, reason: 'encryption_failed' } };
+ }
+
+ const output = new File([encryptedData], input.name, { type: input.type });
+ return {
+ steps,
+ result: {
+ success: true,
+ file: output,
+ uri: URL.createObjectURL(output),
+ encryptionKey: uintArrayToHexString(key),
+ },
+ };
+}
+
type DecryptFileStep =
| {
+step: 'fetch_buffer',
@@ -146,4 +221,4 @@
return { steps, result: { success: true, uri: objectURL } };
}
-export { decryptMedia };
+export { encryptFile, decryptMedia };

File Metadata

Mime Type
text/plain
Expires
Mon, Jan 12, 6:41 PM (5 h, 23 m)
Storage Engine
blob
Storage Format
Raw Data
Storage Handle
5924382
Default Alt Text
D7293.1768243282.diff (2 KB)

Event Timeline