mtcute/packages/wasm/tests/ige.test.ts

43 lines
1.3 KiB
TypeScript

import { beforeAll, describe, expect, it } from 'vitest'
import { getPlatform } from '@mtcute/core/platform.js'
import { __getWasm, ige256Decrypt, ige256Encrypt } from '../src/index.js'
import { initWasm } from './init.js'
const p = getPlatform()
beforeAll(async () => {
await initWasm()
})
describe('aes-ige', () => {
const key = p.hexDecode('5468697320697320616E20696D706C655468697320697320616E20696D706C65')
const iv = p.hexDecode('6D656E746174696F6E206F6620494745206D6F646520666F72204F70656E5353')
const data = p.hexDecode('99706487a1cde613bc6de0b6f24b1c7aa448c8b9c3403e3467a8cad89340f53b')
const dataEnc = p.hexDecode('792ea8ae577b1a66cb3bd92679b8030ca54ee631976bd3a04547fdcb4639fa69')
it('should correctly encrypt', () => {
const aes = ige256Encrypt(data, key, iv)
expect(p.hexEncode(aes)).toEqual(p.hexEncode(dataEnc))
})
it('should correctly decrypt', () => {
const aes = ige256Decrypt(dataEnc, key, iv)
expect(p.hexEncode(aes)).toEqual(p.hexEncode(data))
})
it('should not leak memory', () => {
const mem = __getWasm().memory.buffer
const memSize = mem.byteLength
for (let i = 0; i < 10000; i++) {
ige256Decrypt(ige256Encrypt(data, key, iv), key, iv)
}
expect(mem.byteLength).toEqual(memSize)
})
})