feat: CRP-2957 Add PocketIC master public keys to TS ic_vetkeys library (#251)

Author: randombit

frontend/ic_vetkeys/src/utils/utils.test.ts

@@ -5,6 +5,7 @@ import {
     IbeCiphertext,
     MasterPublicKey,
     MasterPublicKeyId,
+    PocketIcMasterPublicKeyId,
     IbeSeed,
     TransportSecretKey,
     VetKey,
@@ -122,6 +123,64 @@ test("MasterPublicKey derivation using prod key", () => {
     );
 });
 
+test("MasterPublicKey derivation using PocketIC key_1", () => {
+    const canisterId = hexToBytes("ffffffffff9000030101");
+
+    const masterKey = MasterPublicKey.pocketicKey(
+        PocketIcMasterPublicKeyId.KEY_1,
+    );
+    const canisterKey = masterKey.deriveCanisterKey(canisterId);
+
+    const derivedKey = canisterKey.deriveSubKey(
+        new TextEncoder().encode("Test Derivation For PocketIC VetKD key_1"),
+    );
+
+    assertEqual(
+        bytesToHex(derivedKey.publicKeyBytes()),
+        "899a951f6ec2f9a96759c554a6cb01fb1cb20b2f2f96a2d2c869221c04d3349c3be8d49c3257312aed031f430f15f7ef0f4d43adf11251015d70dd91ac07df50fb70818ece721a1d6a314204acddde55542902f5d0d95e2406a5ab1fad18349d",
+    );
+});
+
+test("MasterPublicKey derivation using PocketIC test_key_1", () => {
+    const canisterId = hexToBytes("ffffffffff9000030101");
+
+    const masterKey = MasterPublicKey.pocketicKey(
+        PocketIcMasterPublicKeyId.TEST_KEY_1,
+    );
+    const canisterKey = masterKey.deriveCanisterKey(canisterId);
+
+    const derivedKey = canisterKey.deriveSubKey(
+        new TextEncoder().encode(
+            "Test Derivation For PocketIC VetKD test_key_1",
+        ),
+    );
+
+    assertEqual(
+        bytesToHex(derivedKey.publicKeyBytes()),
+        "a60993fc46593728bd9b0a4ffb1fb9a662dd89b29c99fde36e403c311c8992e6eeb097b31174dd43f74e73fe10c190271193a4345490f64a41ce778a2f6e7c16804919e843ac72ff65bab959c53fa839c9fb3cb263e41498d17fb82704fe18bc",
+    );
+});
+
+test("MasterPublicKey derivation using PocketIC dfx_test_key", () => {
+    const canisterId = hexToBytes("ffffffffff9000030101");
+
+    const masterKey = MasterPublicKey.pocketicKey(
+        PocketIcMasterPublicKeyId.DFX_TEST_KEY,
+    );
+    const canisterKey = masterKey.deriveCanisterKey(canisterId);
+
+    const derivedKey = canisterKey.deriveSubKey(
+        new TextEncoder().encode(
+            "Test Derivation For PocketIC VetKD dfx_test_key",
+        ),
+    );
+
+    assertEqual(
+        bytesToHex(derivedKey.publicKeyBytes()),
+        "800424bea66b95b715f86a9bed06b1f60df98206a57235c3e0f2da4d485dc1c93c56eef54155d559ef45c757fb0444920620b932652f1d683fdbc57db98b5aeb8ba664a5e040cbdf4d685e4e236a7193d1bd5b0927204fab05fff4f61f26b358",
+    );
+});
+
 test("DerivedPublicKey subderivation", () => {
     const canisterKey = DerivedPublicKey.deserialize(
         hexToBytes(

frontend/ic_vetkeys/src/utils/utils.ts

@@ -120,6 +120,15 @@ export enum MasterPublicKeyId {
     TEST_KEY_1 = "test_key_1",
 }
 
+/**
+ * Enumeration identifying possible PocketIC test keys
+ */
+export enum PocketIcMasterPublicKeyId {
+    KEY_1 = "key_1",
+    TEST_KEY_1 = "test_key_1",
+    DFX_TEST_KEY = "dfx_test_key",
+}
+
 /**
  * @internal helper to perform hex decoding
  */
@@ -208,6 +217,39 @@ export class MasterPublicKey {
         }
     }
 
+    /**
+     * Return the hardcoded master public key used in PocketIC
+     *
+     * This allows performing public key derivation offline
+     */
+    static pocketicKey(
+        keyId: PocketIcMasterPublicKeyId = PocketIcMasterPublicKeyId.KEY_1,
+    ): MasterPublicKey {
+        if (keyId == PocketIcMasterPublicKeyId.KEY_1) {
+            return MasterPublicKey.deserialize(
+                hexToBytes(
+                    "8c800b5cff00463d26e8167369168827f1e48f4d8d60f71dd6a295580f65275b5f5f8e6a792c876b2c72492136530d0710a27522ee63977a76216c3cef9e70bfcb45b88736fc62142e7e0737848ce06cbb1f45a4a6a349b142ae5cf7853561e0",
+                ),
+            );
+        } else if (keyId == PocketIcMasterPublicKeyId.TEST_KEY_1) {
+            return MasterPublicKey.deserialize(
+                hexToBytes(
+                    "9069b82c7aae418cef27678291e7f2cb1a008a500eceba7199bffca12421b07c158987c6a22618af3d1958738b2835691028801f7663d311799733286c557c8979184bb62cb559a4d582fca7d2e48b860f08ed6641aef66a059ec891889a6218",
+                ),
+            );
+        } else if (keyId == PocketIcMasterPublicKeyId.DFX_TEST_KEY) {
+            return MasterPublicKey.deserialize(
+                hexToBytes(
+                    "b181c14cf9d04ba45d782c0067a44b0aaa9fc2acf94f1a875f0dae801af4f80339a7e6bf8b09fcf993824c8df3080b3f1409b688ca08cbd44d2cb28db9899f4aa3b5f06b9174240448e10be2f01f9f80079ea5431ce2d11d1c8d1c775333315f",
+                ),
+            );
+        } else {
+            throw new Error(
+                "Unknown PocketIcMasterPublicKeyId value for pocketicKey",
+            );
+        }
+    }
+
     /**
      * @internal constructor
      */