Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Pull to-device event list out of returned tuple #25

Merged
merged 7 commits into from
Aug 10, 2023
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion examples/bot.ts
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@ const dmTarget = creds?.['dmTarget'] ?? "@admin:localhost";
const homeserverUrl = creds?.['homeserverUrl'] ?? "http://localhost:8008";
const accessToken = creds?.['accessToken'] ?? 'YOUR_TOKEN';
const storage = new SimpleFsStorageProvider("./examples/storage/bot.json");
const crypto = new RustSdkCryptoStorageProvider("./examples/storage/bot_sled", StoreType.Sled);
const crypto = new RustSdkCryptoStorageProvider("./examples/storage/bot_sqlite", StoreType.Sqlite);

const client = new MatrixClient(homeserverUrl, accessToken, storage, crypto);
AutojoinRoomsMixin.setupOnClient(client);
Expand Down
2 changes: 1 addition & 1 deletion examples/encryption_appservice.ts
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ try {
const dmTarget = creds?.['dmTarget'] ?? "@admin:localhost";
const homeserverUrl = creds?.['homeserverUrl'] ?? "http://localhost:8008";
const storage = new SimpleFsStorageProvider("./examples/storage/encryption_appservice.json");
const crypto = new RustSdkAppserviceCryptoStorageProvider("./examples/storage/encryption_appservice_sled", StoreType.Sled);
const crypto = new RustSdkAppserviceCryptoStorageProvider("./examples/storage/encryption_appservice_sqlite", StoreType.Sqlite);
const worksImage = fs.readFileSync("./examples/static/it-works.png");

const registration: IAppserviceRegistration = {
Expand Down
2 changes: 1 addition & 1 deletion examples/encryption_bot.ts
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@ const dmTarget = creds?.['dmTarget'] ?? "@admin:localhost";
const homeserverUrl = creds?.['homeserverUrl'] ?? "http://localhost:8008";
const accessToken = creds?.['accessToken'] ?? 'YOUR_TOKEN';
const storage = new SimpleFsStorageProvider("./examples/storage/encryption_bot.json");
const crypto = new RustSdkCryptoStorageProvider("./examples/storage/encryption_bot_sled", StoreType.Sled);
const crypto = new RustSdkCryptoStorageProvider("./examples/storage/encryption_bot_sqlite", StoreType.Sqlite);
const worksImage = fs.readFileSync("./examples/static/it-works.png");

const client = new MatrixClient(homeserverUrl, accessToken, storage, crypto);
Expand Down
2 changes: 1 addition & 1 deletion package.json
Original file line number Diff line number Diff line change
Expand Up @@ -51,7 +51,7 @@
"tsconfig.json"
],
"dependencies": {
"@matrix-org/matrix-sdk-crypto-nodejs": "0.1.0-beta.6",
"@matrix-org/matrix-sdk-crypto-nodejs": "0.1.0-beta.9",
"@types/express": "^4.17.13",
"another-json": "^0.2.0",
"async-lock": "^1.3.2",
Expand Down
9 changes: 5 additions & 4 deletions src/e2ee/CryptoClient.ts
Original file line number Diff line number Diff line change
Expand Up @@ -168,12 +168,13 @@ export class CryptoClient {
leftDeviceLists.map(u => new UserId(u)));

await this.engine.lock.acquire(SYNC_LOCK_NAME, async () => {
const syncResp = await this.engine.machine.receiveSyncChanges(deviceMessages, deviceLists, otkCounts, unusedFallbackKeyAlgs);
const decryptedToDeviceMessages = JSON.parse(syncResp);
if (Array.isArray(decryptedToDeviceMessages)) {
for (const msg of decryptedToDeviceMessages) {
const syncResp = JSON.parse(await this.engine.machine.receiveSyncChanges(deviceMessages, deviceLists, otkCounts, unusedFallbackKeyAlgs));
AndrewFerr marked this conversation as resolved.
Show resolved Hide resolved
if (Array.isArray(syncResp) && syncResp.length === 2 && Array.isArray(syncResp[0])) {
for (const msg of syncResp[0] as IToDeviceMessage[]) {
this.client.emit("to_device.decrypted", msg);
}
} else {
LogService.error("CryptoClient", "OlmMachine.receiveSyncChanges did not return an expected value of [to-device events, room key changes]");
}

await this.engine.run();
Expand Down
4 changes: 2 additions & 2 deletions src/storage/RustSdkCryptoStorageProvider.ts
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,7 @@ export class RustSdkCryptoStorageProvider implements ICryptoStorageProvider {
*/
public constructor(
public readonly storagePath: string,
public readonly storageType: RustSdkCryptoStoreType = RustSdkCryptoStoreType.Sled,
public readonly storageType: RustSdkCryptoStoreType = RustSdkCryptoStoreType.Sqlite,
) {
this.storagePath = path.resolve(this.storagePath);
mkdirp.sync(storagePath);
Expand Down Expand Up @@ -69,7 +69,7 @@ export class RustSdkAppserviceCryptoStorageProvider extends RustSdkCryptoStorage
* @param baseStoragePath The *directory* to persist database details to.
* @param storageType The storage type to use. Must be supported by the rust-sdk.
*/
public constructor(private baseStoragePath: string, storageType: RustSdkCryptoStoreType = RustSdkCryptoStoreType.Sled) {
public constructor(private baseStoragePath: string, storageType: RustSdkCryptoStoreType = RustSdkCryptoStoreType.Sqlite) {
super(path.join(baseStoragePath, "_default"), storageType);
}

Expand Down
10 changes: 4 additions & 6 deletions test/MatrixClientTest.ts
Original file line number Diff line number Diff line change
@@ -1,6 +1,5 @@
import * as tmp from "tmp";
import * as simple from "simple-mock";
import { StoreType } from "@matrix-org/matrix-sdk-crypto-nodejs";

import {
EventKind,
Expand Down Expand Up @@ -48,13 +47,13 @@ describe('MatrixClient', () => {
expect(client.accessToken).toEqual(accessToken);
});

it('should create a crypto client when requested', () => {
it('should create a crypto client when requested', () => testCryptoStores(async (cryptoStoreType) => {
const homeserverUrl = "https://example.org";
const accessToken = "example_token";

const client = new MatrixClient(homeserverUrl, accessToken, null, new RustSdkCryptoStorageProvider(tmp.dirSync().name, StoreType.Sled));
const client = new MatrixClient(homeserverUrl, accessToken, null, new RustSdkCryptoStorageProvider(tmp.dirSync().name, cryptoStoreType));
expect(client.crypto).toBeDefined();
});
}));

it('should NOT create a crypto client when requested', () => {
const homeserverUrl = "https://example.org";
Expand Down Expand Up @@ -1401,8 +1400,7 @@ describe('MatrixClient', () => {
describe('processSync', () => {
interface ProcessSyncClient {
userId: string;

processSync(raw: any): Promise<any>;
processSync(raw: any): MatrixClient["processSync"];
}

it('should process non-room account data', async () => {
Expand Down
4 changes: 2 additions & 2 deletions test/TestUtils.ts
Original file line number Diff line number Diff line change
Expand Up @@ -40,14 +40,14 @@ export function createTestClient(
const http = new HttpBackend();
const hsUrl = "https://localhost";
const accessToken = "s3cret";
const client = new MatrixClient(hsUrl, accessToken, storage, cryptoStoreType !== undefined ? new RustSdkCryptoStorageProvider(tmp.dirSync().name, cryptoStoreType) : null);
const client = new MatrixClient(hsUrl, accessToken, storage, (cryptoStoreType !== undefined) ? new RustSdkCryptoStorageProvider(tmp.dirSync().name, cryptoStoreType) : null);
(<any>client).userId = userId; // private member access
setRequestFn(http.requestFn);

return { http, hsUrl, accessToken, client };
}

const CRYPTO_STORE_TYPES = [StoreType.Sled, StoreType.Sqlite];
const CRYPTO_STORE_TYPES: StoreType[] = [StoreType.Sqlite];

export async function testCryptoStores(fn: (StoreType) => Promise<void>): Promise<void> {
for (const st of CRYPTO_STORE_TYPES) {
Expand Down
2 changes: 1 addition & 1 deletion test/appservice/IntentTest.ts
Original file line number Diff line number Diff line change
Expand Up @@ -1137,7 +1137,7 @@ describe('Intent', () => {

beforeEach(() => {
storage = new MemoryStorageProvider();
cryptoStorage = new RustSdkAppserviceCryptoStorageProvider(tmp.dirSync().name, StoreType.Sled);
cryptoStorage = new RustSdkAppserviceCryptoStorageProvider(tmp.dirSync().name, StoreType.Sqlite);
options = {
homeserverUrl: hsUrl,
storage: storage,
Expand Down
67 changes: 66 additions & 1 deletion test/encryption/CryptoClientTest.ts
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import * as simple from "simple-mock";
import HttpBackend from 'matrix-mock-request';

import { EncryptedFile, MatrixClient, MembershipEvent, OTKAlgorithm, RoomEncryptionAlgorithm } from "../../src";
import { EncryptedFile, EncryptionAlgorithm, IOlmEncrypted, IToDeviceMessage, MatrixClient, MembershipEvent, OTKAlgorithm, RoomEncryptionAlgorithm } from "../../src";
import { createTestClient, testCryptoStores, TEST_DEVICE_ID } from "../TestUtils";

export function bindNullEngine(http: HttpBackend) {
Expand Down Expand Up @@ -85,6 +85,71 @@ describe('CryptoClient', () => {
}));
});

describe('processSync', () => {
/**
* Helper class to be able to call {@link MatrixClient#processSync}, which is otherwise private.
*/
interface ProcessSyncClient {
processSync: MatrixClient["processSync"];
}

it('should process encrypted to-device messages', () => testCryptoStores(async (cryptoStoreType) => {
const userId = "@alice:example.org";
const { client, http } = createTestClient(null, userId, cryptoStoreType);
const psClient = <ProcessSyncClient>(<any>client);

await client.cryptoStore.setDeviceId(TEST_DEVICE_ID);

const toDeviceMessage: IToDeviceMessage<IOlmEncrypted> = {
type: "m.room.encrypted",
sender: userId,
content: {
algorithm: EncryptionAlgorithm.OlmV1Curve25519AesSha2,
sender_key: "sender_curve25519_key",
ciphertext: {
["device_curve25519_key"]: {
type: 0,
body: "encrypted_payload_base_64",
},
},
},
};
const sync = {
to_device: { events: [toDeviceMessage] },
device_unused_fallback_key_types: [OTKAlgorithm.Signed],
device_one_time_keys_count: {
[OTKAlgorithm.Signed]: 12,
[OTKAlgorithm.Unsigned]: 14,
},
device_lists: {
changed: ["@bob:example.org"],
left: ["@charlie:example.org"],
},
};

const toDeviceSpy = simple.stub().callFn((ev) => {
for (const prop in toDeviceMessage) {
expect(ev).toHaveProperty(prop);
}
});
client.on("to_device.decrypted", toDeviceSpy);

bindNullEngine(http);
await Promise.all([
client.crypto.prepare([]),
http.flushAllExpected(),
]);

bindNullEngine(http);
await Promise.all([
psClient.processSync(sync),
http.flushAllExpected(),
]);

expect(toDeviceSpy.callCount).toBe(1);
}));
});

describe('isRoomEncrypted', () => {
it('should fail when the crypto has not been prepared', () => testCryptoStores(async (cryptoStoreType) => {
const userId = "@alice:example.org";
Expand Down
8 changes: 4 additions & 4 deletions yarn.lock
Original file line number Diff line number Diff line change
Expand Up @@ -584,10 +584,10 @@
"@jridgewell/resolve-uri" "^3.0.3"
"@jridgewell/sourcemap-codec" "^1.4.10"

"@matrix-org/matrix-sdk-crypto-nodejs@0.1.0-beta.6":
version "0.1.0-beta.6"
resolved "https://registry.yarnpkg.com/@matrix-org/matrix-sdk-crypto-nodejs/-/matrix-sdk-crypto-nodejs-0.1.0-beta.6.tgz#0ecae51103ee3c107af0d6d0738f33eb7cc9857e"
integrity sha512-JXyrHuCVMydUGgSetWsfqbbvHj3aUMOX5TUghlMtLFromyEu7wIsNgYt7PjJ+k3WdF4GVABRy4P6GNjaEMy2uA==
"@matrix-org/matrix-sdk-crypto-nodejs@0.1.0-beta.9":
version "0.1.0-beta.9"
resolved "https://registry.yarnpkg.com/@matrix-org/matrix-sdk-crypto-nodejs/-/matrix-sdk-crypto-nodejs-0.1.0-beta.9.tgz#dc21f3b0f4b35b73befc64a257b8e519afbcbed0"
integrity sha512-ee2YlBoXPLgp1aav9MqREJKvWJfURn9Jcs46FyWT4NXEl37KQDNC8CWWnqgqsHkLfBxxSxfq9kMA/mWQZF7QJw==
dependencies:
https-proxy-agent "^5.0.1"
node-downloader-helper "^2.1.5"
Expand Down
Loading