feat: fix mmkv parser, support for ios ekey mmkv

.gitattributes

@@ -0,0 +1 @@
+*.mmkv binary

src/features/settings/panels/QMCv2/ImportFileModal.tsx

@@ -52,7 +52,7 @@ export function ImportFileModal({ onClose, show }: ImportFileModalProps) {
           alert(`不是支持的 SQLite 数据库文件。\n表名：${qmc2Keys}`);
           return;
         }
-      } else if (/MMKVStreamEncryptId/i.test(file.name)) {
+      } else if (/MMKVStreamEncryptId|filenameEkeyMap/i.test(file.name)) {
         const fileBuffer = await file.arrayBuffer();
         const map = MMKVParser.toStringMap(new DataView(fileBuffer));
         qmc2Keys = Array.from(map.entries(), ([name, key]) => ({ name: getFileName(name), key }));

src/util/MMKVParser.ts

@@ -1,28 +1,29 @@
+import { formatHex } from './formatHex';
+
 const textDecoder = new TextDecoder('utf-8', { ignoreBOM: true });
 
 export class MMKVParser {
-  private offset = 8;
+  private offset = 4;
   private length: number;
 
   constructor(private view: DataView) {
     const payloadLength = view.getUint32(0, true);
     this.length = 4 + payloadLength;
+
+    // skip unused str
+    this.readInt();
   }
 
   toString() {
-    const offset = this.offset.toString(16).padStart(8, '0');
-    const length = this.length.toString(16).padStart(8, '0');
-    return `<MMKVParser offset=0x${offset} len=0x${length}>`;
+    const offset = formatHex(this.offset, 8);
+    const length = formatHex(this.length, 8);
+    return `<MMKVParser offset=${offset} len=${length}>`;
   }
 
   get eof() {
     return this.offset >= this.length;
   }
 
-  peek() {
-    return this.view.getUint8(this.offset);
-  }
-
   public readByte() {
     return this.view.getUint8(this.offset++);
   }
@@ -77,7 +78,9 @@ export class MMKVParser {
     const newOffset = this.offset + containerLen;
     const result = this.readString();
     if (newOffset !== this.offset) {
-      throw new Error('readVariantString failed: offset does not match');
+      const expected = formatHex(newOffset);
+      const actual = formatHex(this.offset);
+      throw new Error(`readVariantString failed: offset does mismatch (expect: ${expected}, actual: ${actual})`);
     }
     return result;
   }

src/util/__tests__/MMKVParser.test.ts

@@ -0,0 +1,19 @@
+import { MMKVParser } from '../MMKVParser';
+import { readFileSync } from 'node:fs';
+
+test('parse mmkv file as expected', () => {
+  const buff = readFileSync(__dirname + '/__fixture__/test.mmkv');
+  const view = new DataView(buff.buffer.slice(buff.byteOffset, buff.byteOffset + buff.byteLength));
+  expect(MMKVParser.toStringMap(view)).toEqual(
+    new Map([
+      ['key', 'value'],
+      [
+        'Lorem Ipsum',
+        'Lorem ipsum dolor sit amet, consectetur adipiscing elit. ' +
+          'Vestibulum congue volutpat metus non molestie. Quisque id est sapien. ' +
+          'Fusce eget tristique sem. Donec tellus lacus, viverra sed lectus eget, elementum ultrices dolor. ' +
+          'Integer non urna justo.',
+      ],
+    ])
+  );
+});

src/util/formatHex.ts

@@ -0,0 +1,3 @@
+export function formatHex(value: number, len = 8) {
+  return '0x' + (value | 0).toString(16).padStart(len, '0');
+}