@lfdecentralizedtrust/zeto-contracts 0.5.0 → 0.5.1

package/ignition/modules/zeto_anon_enc_nullifier.ts

@@ -20,6 +20,7 @@ import {
   DepositVerifierModule,
   WithdrawNullifierVerifierModule,
   BatchWithdrawNullifierVerifierModule,
+  ZetoLockableLibModule,
 } from "./lib/deps";
 
 const VerifierModule = buildModule("Groth16Verifier_AnonEncNullifier", (m) => {
@@ -63,6 +64,7 @@ const BatchLockVerifierModule = buildModule(
 );
 
 export default buildModule("Zeto_AnonEncNullifier", (m) => {
+  const { zetoLockableLib } = m.useModule(ZetoLockableLibModule);
   const { smtLib, poseidon2, poseidon3 } = m.useModule(SmtLibModule);
   const { verifier } = m.useModule(VerifierModule);
   const { verifier: lockVerifier } = m.useModule(LockVerifierModule);
@@ -87,5 +89,6 @@ export default buildModule("Zeto_AnonEncNullifier", (m) => {
     smtLib,
     poseidon2,
     poseidon3,
+    zetoLockableLib,
   };
 });

package/ignition/modules/zeto_anon_enc_nullifier_kyc.ts

@@ -20,6 +20,7 @@ import {
   DepositKycVerifierModule,
   WithdrawNullifierVerifierModule,
   BatchWithdrawNullifierVerifierModule,
+  ZetoLockableLibModule,
 } from "./lib/deps";
 
 const VerifierModule = buildModule(
@@ -39,6 +40,7 @@ const BatchVerifierModule = buildModule(
 );
 
 export default buildModule("Zeto_AnonEncNullifierKyc", (m) => {
+  const { zetoLockableLib } = m.useModule(ZetoLockableLibModule);
   const { smtLib, poseidon2, poseidon3 } = m.useModule(SmtLibModule);
   const { verifier } = m.useModule(VerifierModule);
   const { verifier: batchVerifier } = m.useModule(BatchVerifierModule);
@@ -59,5 +61,6 @@ export default buildModule("Zeto_AnonEncNullifierKyc", (m) => {
     smtLib,
     poseidon2,
     poseidon3,
+    zetoLockableLib,
   };
 });

package/ignition/modules/zeto_anon_enc_nullifier_non_repudiation.ts

@@ -20,6 +20,7 @@ import {
   DepositVerifierModule,
   WithdrawNullifierVerifierModule,
   BatchWithdrawNullifierVerifierModule,
+  ZetoLockableLibModule,
 } from "./lib/deps";
 
 const VerifierModule = buildModule(
@@ -45,6 +46,7 @@ const BatchVerifierModule = buildModule(
 );
 
 export default buildModule("Zeto_AnonEncNullifierNonRepudiation", (m) => {
+  const { zetoLockableLib } = m.useModule(ZetoLockableLibModule);
   const { smtLib, poseidon2, poseidon3 } = m.useModule(SmtLibModule);
   const { verifier } = m.useModule(VerifierModule);
   const { verifier: batchVerifier } = m.useModule(BatchVerifierModule);
@@ -64,5 +66,6 @@ export default buildModule("Zeto_AnonEncNullifierNonRepudiation", (m) => {
     smtLib,
     poseidon2,
     poseidon3,
+    zetoLockableLib,
   };
 });

package/ignition/modules/zeto_anon_nullifier.ts

@@ -20,6 +20,7 @@ import {
   DepositVerifierModule,
   WithdrawNullifierVerifierModule,
   BatchWithdrawNullifierVerifierModule,
+  ZetoLockableLibModule,
 } from "./lib/deps";
 
 const VerifierModule = buildModule(
@@ -88,6 +89,7 @@ const BatchLockVerifierModule = buildModule(
 );
 
 export default buildModule("Zeto_AnonNullifier", (m) => {
+  const { zetoLockableLib } = m.useModule(ZetoLockableLibModule);
   const { smtLib, poseidon2, poseidon3 } = m.useModule(SmtLibModule);
   const { verifier } = m.useModule(VerifierModule);
   const { verifier: lockVerifier } = m.useModule(LockVerifierModule);
@@ -112,5 +114,6 @@ export default buildModule("Zeto_AnonNullifier", (m) => {
     smtLib,
     poseidon2,
     poseidon3,
+    zetoLockableLib,
   };
 });

package/ignition/modules/zeto_anon_nullifier_burnable.ts

@@ -20,6 +20,7 @@ import {
   DepositVerifierModule,
   WithdrawNullifierVerifierModule,
   BatchWithdrawNullifierVerifierModule,
+  ZetoLockableLibModule,
 } from "./lib/deps";
 
 const VerifierModule = buildModule(
@@ -104,6 +105,7 @@ const BatchBurnNullifierVerifierModule = buildModule(
 );
 
 export default buildModule("Zeto_AnonNullifier", (m) => {
+  const { zetoLockableLib } = m.useModule(ZetoLockableLibModule);
   const { smtLib, poseidon2, poseidon3 } = m.useModule(SmtLibModule);
   const { verifier } = m.useModule(VerifierModule);
   const { verifier: lockVerifier } = m.useModule(LockVerifierModule);
@@ -134,5 +136,6 @@ export default buildModule("Zeto_AnonNullifier", (m) => {
     smtLib,
     poseidon2,
     poseidon3,
+    zetoLockableLib,
   };
 });

package/ignition/modules/zeto_anon_nullifier_kyc.ts

@@ -20,6 +20,7 @@ import {
   WithdrawNullifierVerifierModule,
   BatchWithdrawNullifierVerifierModule,
   DepositKycVerifierModule,
+  ZetoLockableLibModule,
 } from "./lib/deps";
 
 const VerifierModule = buildModule(
@@ -64,6 +65,7 @@ const BatchLockVerifierModule = buildModule(
 );
 
 export default buildModule("Zeto_AnonNullifierKyc", (m) => {
+  const { zetoLockableLib } = m.useModule(ZetoLockableLibModule);
   const { smtLib, poseidon2, poseidon3 } = m.useModule(SmtLibModule);
   const { verifier } = m.useModule(VerifierModule);
   const { verifier: batchVerifier } = m.useModule(BatchVerifierModule);
@@ -88,5 +90,6 @@ export default buildModule("Zeto_AnonNullifierKyc", (m) => {
     smtLib,
     poseidon2,
     poseidon3,
+    zetoLockableLib,
   };
 });

package/ignition/modules/zeto_anon_nullifier_qurrency.ts

@@ -20,6 +20,7 @@ import {
   DepositVerifierModule,
   WithdrawNullifierVerifierModule,
   BatchWithdrawNullifierVerifierModule,
+  ZetoLockableLibModule,
 } from "./lib/deps";
 
 const VerifierModule = buildModule(
@@ -64,6 +65,7 @@ const BatchVerifierModule = buildModule(
 // );
 
 export default buildModule("Zeto_AnonNullifierQurrency", (m) => {
+  const { zetoLockableLib } = m.useModule(ZetoLockableLibModule);
   const { smtLib, poseidon2, poseidon3, poseidon5, poseidon6 } = m.useModule(SmtLibModule);
   const { verifier } = m.useModule(VerifierModule);
   // const { verifier: lockVerifier } = m.useModule(LockVerifierModule);
@@ -90,5 +92,6 @@ export default buildModule("Zeto_AnonNullifierQurrency", (m) => {
     poseidon3,
     poseidon5,
     poseidon6,
+    zetoLockableLib,
   };
 });

package/ignition/modules/zeto_nf_anon.ts

@@ -15,13 +15,15 @@
 // limitations under the License.
 
 import { buildModule } from "@nomicfoundation/hardhat-ignition/modules";
+import { ZetoLockableLibModule } from "./lib/deps";
 const VerifierModule = buildModule("Groth16Verifier_NfAnon", (m) => {
   const verifier = m.contract("Groth16Verifier_NfAnon", []);
   return { verifier };
 });
 
 export default buildModule("Zeto_NfAnon", (m) => {
+  const { zetoLockableLib } = m.useModule(ZetoLockableLibModule);
   const { verifier } = m.useModule(VerifierModule);
 
-  return { verifier };
+  return { verifier, zetoLockableLib };
 });

package/ignition/modules/zeto_nf_anon_nullifier.ts

@@ -15,7 +15,7 @@
 // limitations under the License.
 
 import { buildModule } from "@nomicfoundation/hardhat-ignition/modules";
-import { SmtLibModule } from "./lib/deps";
+import { SmtLibModule, ZetoLockableLibModule } from "./lib/deps";
 
 const VerifierModule = buildModule(
   "Groth16Verifier_NfAnonNullifierTransfer",
@@ -41,9 +41,10 @@ const LockVerifierModule = buildModule("Groth16Verifier_NfAnon", (m) => {
 });
 
 export default buildModule("Zeto_NfAnonNullifier", (m) => {
+  const { zetoLockableLib } = m.useModule(ZetoLockableLibModule);
   const { smtLib, poseidon2, poseidon3 } = m.useModule(SmtLibModule);
   const { verifier } = m.useModule(VerifierModule);
   const { verifier: lockVerifier } = m.useModule(LockVerifierModule);
 
-  return { verifier, lockVerifier, smtLib, poseidon2, poseidon3 };
+  return { verifier, lockVerifier, smtLib, poseidon2, poseidon3, zetoLockableLib };
 });

package/package.json

@@ -1,13 +1,14 @@
 {
   "name": "@lfdecentralizedtrust/zeto-contracts",
-  "version": "0.5.0",
+  "version": "0.5.1",
   "description": "Zero knowledge proof based UTXO tokens toolkit for fungible or non-fungible assets",
   "main": "index.js",
   "scripts": {
     "prettier": "npx prettier contracts scripts ignition test --check",
     "prettier:fix": "npm run prettier -- --write",
     "pretest": "npm run prettier:fix",
-    "test": "npx hardhat test"
+    "test": "npx hardhat test",
+    "size": "npx hardhat size-contracts"
   },
   "keywords": [],
   "repository": {
@@ -21,6 +22,7 @@
     "@nomicfoundation/hardhat-ignition-ethers": "^0.15.0",
     "@nomicfoundation/hardhat-toolbox": "^5.0.0",
     "circomlibjs": "^0.1.7",
+    "hardhat-contract-sizer": "^2.10.1",
     "maci-crypto": "1.1.1",
     "mlkem": "^2.3.1",
     "prettier": "^3.3.3",

package/scripts/lib/zeto_libraries.ts

@@ -0,0 +1,47 @@
+// Copyright © 2025 Kaleido, Inc.
+//
+// SPDX-License-Identifier: Apache-2.0
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+/** Hardhat library link map for Zeto token implementations. */
+export function withZetoLockableLib(
+  zetoLockableLib: { target: string },
+  extra: Record<string, string> = {},
+): Record<string, string> {
+  return {
+    ZetoLockableLib: zetoLockableLib.target,
+    ...extra,
+  };
+}
+
+export function smtLibraries(deps: {
+  smtLib: { target: string };
+  poseidon2: { target: string };
+  poseidon3: { target: string };
+  poseidon5?: { target: string };
+  poseidon6?: { target: string };
+}): Record<string, string> {
+  const libs: Record<string, string> = {
+    SmtLib: deps.smtLib.target,
+    PoseidonUnit2L: deps.poseidon2.target,
+    PoseidonUnit3L: deps.poseidon3.target,
+  };
+  if (deps.poseidon5) {
+    libs.PoseidonUnit5L = deps.poseidon5.target;
+  }
+  if (deps.poseidon6) {
+    libs.PoseidonUnit6L = deps.poseidon6.target;
+  }
+  return libs;
+}

package/scripts/tokens/Zeto_Anon.ts

@@ -16,6 +16,7 @@
 
 import { ethers, ignition } from "hardhat";
 import zetoModule from "../../ignition/modules/zeto_anon";
+import { withZetoLockableLib } from "../lib/zeto_libraries";
 
 export async function deployDependencies() {
   const [deployer] = await ethers.getSigners();
@@ -26,9 +27,11 @@ export async function deployDependencies() {
     verifier,
     batchVerifier,
     batchWithdrawVerifier,
+    zetoLockableLib,
   } = await ignition.deploy(zetoModule);
   return {
     deployer,
+    libraries: withZetoLockableLib(zetoLockableLib),
     args: [
       "Zeto Anon",
       "ZA",

package/scripts/tokens/Zeto_AnonBurnable.ts

@@ -16,6 +16,7 @@
 
 import { ethers, ignition } from "hardhat";
 import zetoModule from "../../ignition/modules/zeto_anon_burnable";
+import { withZetoLockableLib } from "../lib/zeto_libraries";
 
 export async function deployDependencies() {
   const [deployer] = await ethers.getSigners();
@@ -28,9 +29,11 @@ export async function deployDependencies() {
     batchVerifier,
     batchWithdrawVerifier,
     batchBurnVerifier,
+    zetoLockableLib,
   } = await ignition.deploy(zetoModule);
   return {
     deployer,
+    libraries: withZetoLockableLib(zetoLockableLib),
     args: [
       "Zeto Anon Burnable",
       "ZAB",

package/scripts/tokens/Zeto_AnonEnc.ts

@@ -16,6 +16,7 @@
 
 import { ethers, ignition } from "hardhat";
 import zetoModule from "../../ignition/modules/zeto_anon_enc";
+import { withZetoLockableLib } from "../lib/zeto_libraries";
 
 export async function deployDependencies() {
   const [deployer] = await ethers.getSigners();
@@ -26,9 +27,11 @@ export async function deployDependencies() {
     verifier,
     batchVerifier,
     batchWithdrawVerifier,
+    zetoLockableLib,
   } = await ignition.deploy(zetoModule);
   return {
     deployer,
+    libraries: withZetoLockableLib(zetoLockableLib),
     args: [
       "Zeto Anon Enc",
       "ZAE",

package/scripts/tokens/Zeto_AnonEncNullifier.ts

@@ -16,6 +16,7 @@
 
 import { ethers, ignition } from "hardhat";
 import zetoModule from "../../ignition/modules/zeto_anon_enc_nullifier";
+import { withZetoLockableLib, smtLibraries } from "../lib/zeto_libraries";
 
 export async function deployDependencies() {
   const [deployer] = await ethers.getSigners();
@@ -31,6 +32,7 @@ export async function deployDependencies() {
     smtLib,
     poseidon2,
     poseidon3,
+    zetoLockableLib,
   } = await ignition.deploy(zetoModule);
   return {
     deployer,
@@ -50,10 +52,9 @@ export async function deployDependencies() {
         batchBurnVerifier: "0x0000000000000000000000000000000000000000",
       },
     ],
-    libraries: {
-      SmtLib: smtLib.target,
-      PoseidonUnit2L: poseidon2.target,
-      PoseidonUnit3L: poseidon3.target,
-    },
+    libraries: withZetoLockableLib(
+      zetoLockableLib,
+      smtLibraries({ smtLib, poseidon2, poseidon3 }),
+    ),
   };
 }

package/scripts/tokens/Zeto_AnonEncNullifierKyc.ts

@@ -16,6 +16,7 @@
 
 import { ethers, ignition } from "hardhat";
 import zetoModule from "../../ignition/modules/zeto_anon_enc_nullifier_kyc";
+import { withZetoLockableLib, smtLibraries } from "../lib/zeto_libraries";
 
 export async function deployDependencies() {
   const [deployer] = await ethers.getSigners();
@@ -29,6 +30,7 @@ export async function deployDependencies() {
     smtLib,
     poseidon2,
     poseidon3,
+    zetoLockableLib,
   } = await ignition.deploy(zetoModule);
   return {
     deployer,
@@ -48,10 +50,9 @@ export async function deployDependencies() {
         batchBurnVerifier: "0x0000000000000000000000000000000000000000",
       },
     ],
-    libraries: {
-      SmtLib: smtLib.target,
-      PoseidonUnit2L: poseidon2.target,
-      PoseidonUnit3L: poseidon3.target,
-    },
+    libraries: withZetoLockableLib(
+      zetoLockableLib,
+      smtLibraries({ smtLib, poseidon2, poseidon3 }),
+    ),
   };
 }

package/scripts/tokens/Zeto_AnonEncNullifierNonRepudiation.ts

@@ -16,6 +16,7 @@
 
 import { ethers, ignition } from "hardhat";
 import zetoModule from "../../ignition/modules/zeto_anon_enc_nullifier_non_repudiation";
+import { withZetoLockableLib, smtLibraries } from "../lib/zeto_libraries";
 
 export async function deployDependencies() {
   const [deployer] = await ethers.getSigners();
@@ -29,6 +30,7 @@ export async function deployDependencies() {
     smtLib,
     poseidon2,
     poseidon3,
+    zetoLockableLib,
   } = await ignition.deploy(zetoModule);
   return {
     deployer,
@@ -48,10 +50,9 @@ export async function deployDependencies() {
         batchBurnVerifier: "0x0000000000000000000000000000000000000000",
       },
     ],
-    libraries: {
-      SmtLib: smtLib.target,
-      PoseidonUnit2L: poseidon2.target,
-      PoseidonUnit3L: poseidon3.target,
-    },
+    libraries: withZetoLockableLib(
+      zetoLockableLib,
+      smtLibraries({ smtLib, poseidon2, poseidon3 }),
+    ),
   };
 }

package/scripts/tokens/Zeto_AnonNullifier.ts

@@ -16,6 +16,7 @@
 
 import { ethers, ignition } from "hardhat";
 import zetoModule from "../../ignition/modules/zeto_anon_nullifier";
+import { withZetoLockableLib, smtLibraries } from "../lib/zeto_libraries";
 
 export async function deployDependencies() {
   const [deployer] = await ethers.getSigners();
@@ -31,6 +32,7 @@ export async function deployDependencies() {
     smtLib,
     poseidon2,
     poseidon3,
+    zetoLockableLib,
   } = await ignition.deploy(zetoModule);
   return {
     deployer,
@@ -50,10 +52,9 @@ export async function deployDependencies() {
         batchBurnVerifier: "0x0000000000000000000000000000000000000000",
       },
     ],
-    libraries: {
-      SmtLib: smtLib.target,
-      PoseidonUnit2L: poseidon2.target,
-      PoseidonUnit3L: poseidon3.target,
-    },
+    libraries: withZetoLockableLib(
+      zetoLockableLib,
+      smtLibraries({ smtLib, poseidon2, poseidon3 }),
+    ),
   };
 }

package/scripts/tokens/Zeto_AnonNullifierBurnable.ts

@@ -16,6 +16,7 @@
 
 import { ethers, ignition } from "hardhat";
 import zetoModule from "../../ignition/modules/zeto_anon_nullifier_burnable";
+import { withZetoLockableLib, smtLibraries } from "../lib/zeto_libraries";
 
 export async function deployDependencies() {
   const [deployer] = await ethers.getSigners();
@@ -33,6 +34,7 @@ export async function deployDependencies() {
     smtLib,
     poseidon2,
     poseidon3,
+    zetoLockableLib,
   } = await ignition.deploy(zetoModule);
   return {
     deployer,
@@ -52,10 +54,9 @@ export async function deployDependencies() {
         batchBurnVerifier: batchBurnVerifier.target,
       },
     ],
-    libraries: {
-      SmtLib: smtLib.target,
-      PoseidonUnit2L: poseidon2.target,
-      PoseidonUnit3L: poseidon3.target,
-    },
+    libraries: withZetoLockableLib(
+      zetoLockableLib,
+      smtLibraries({ smtLib, poseidon2, poseidon3 }),
+    ),
   };
 }

package/scripts/tokens/Zeto_AnonNullifierKyc.ts

@@ -16,6 +16,7 @@
 
 import { ethers, ignition } from "hardhat";
 import zetoModule from "../../ignition/modules/zeto_anon_nullifier_kyc";
+import { withZetoLockableLib, smtLibraries } from "../lib/zeto_libraries";
 
 export async function deployDependencies() {
   const [deployer] = await ethers.getSigners();
@@ -31,6 +32,7 @@ export async function deployDependencies() {
     smtLib,
     poseidon2,
     poseidon3,
+    zetoLockableLib,
   } = await ignition.deploy(zetoModule);
   return {
     deployer,
@@ -50,10 +52,9 @@ export async function deployDependencies() {
         batchBurnVerifier: "0x0000000000000000000000000000000000000000",
       },
     ],
-    libraries: {
-      SmtLib: smtLib.target,
-      PoseidonUnit2L: poseidon2.target,
-      PoseidonUnit3L: poseidon3.target,
-    },
+    libraries: withZetoLockableLib(
+      zetoLockableLib,
+      smtLibraries({ smtLib, poseidon2, poseidon3 }),
+    ),
   };
 }

package/scripts/tokens/Zeto_AnonNullifierQurrency.ts

@@ -16,6 +16,7 @@
 
 import { ethers, ignition } from "hardhat";
 import zetoModule from "../../ignition/modules/zeto_anon_nullifier_qurrency";
+import { withZetoLockableLib, smtLibraries } from "../lib/zeto_libraries";
 
 export async function deployDependencies() {
   const [deployer] = await ethers.getSigners();
@@ -33,6 +34,7 @@ export async function deployDependencies() {
     poseidon3,
     poseidon5,
     poseidon6,
+    zetoLockableLib,
   } = await ignition.deploy(zetoModule);
   return {
     deployer,
@@ -52,12 +54,9 @@ export async function deployDependencies() {
         batchBurnVerifier: "0x0000000000000000000000000000000000000000",
       },
     ],
-    libraries: {
-      SmtLib: smtLib.target,
-      PoseidonUnit2L: poseidon2.target,
-      PoseidonUnit3L: poseidon3.target,
-      PoseidonUnit5L: poseidon5.target,
-      PoseidonUnit6L: poseidon6.target,
-    },
+    libraries: withZetoLockableLib(
+      zetoLockableLib,
+      smtLibraries({ smtLib, poseidon2, poseidon3, poseidon5, poseidon6 }),
+    ),
   };
 }

package/scripts/tokens/Zeto_NfAnon.ts

@@ -16,13 +16,15 @@
 
 import { ethers, ignition } from "hardhat";
 import zetoModule from "../../ignition/modules/zeto_nf_anon";
+import { withZetoLockableLib } from "../lib/zeto_libraries";
 
 export async function deployDependencies() {
   const [deployer] = await ethers.getSigners();
 
-  const { verifier } = await ignition.deploy(zetoModule);
+  const { verifier, zetoLockableLib } = await ignition.deploy(zetoModule);
   return {
     deployer,
+    libraries: withZetoLockableLib(zetoLockableLib),
     args: [
       "Zeto NF Anon",
       "ZNFA",

package/scripts/tokens/Zeto_NfAnonNullifier.ts

@@ -16,11 +16,12 @@
 
 import { ethers, ignition } from "hardhat";
 import zetoModule from "../../ignition/modules/zeto_nf_anon_nullifier";
+import { withZetoLockableLib, smtLibraries } from "../lib/zeto_libraries";
 
 export async function deployDependencies() {
   const [deployer] = await ethers.getSigners();
 
-  const { verifier, lockVerifier, smtLib, poseidon2, poseidon3 } =
+  const { verifier, lockVerifier, smtLib, poseidon2, poseidon3, zetoLockableLib } =
     await ignition.deploy(zetoModule);
   return {
     deployer,
@@ -40,10 +41,9 @@ export async function deployDependencies() {
         batchBurnVerifier: "0x0000000000000000000000000000000000000000",
       },
     ],
-    libraries: {
-      SmtLib: smtLib.target,
-      PoseidonUnit2L: poseidon2.target,
-      PoseidonUnit3L: poseidon3.target,
-    },
+    libraries: withZetoLockableLib(
+      zetoLockableLib,
+      smtLibraries({ smtLib, poseidon2, poseidon3 }),
+    ),
   };
 }

package/test/factory.ts

@@ -17,6 +17,8 @@
 import { ethers, network } from "hardhat";
 import { Signer } from "ethers";
 import { expect } from "chai";
+import { getLinkedContractFactory } from "../scripts/lib/common";
+import { withZetoLockableLib } from "../scripts/lib/zeto_libraries";
 
 describe("(factory) Zeto based fungible token with anonymity without encryption or nullifier", function () {
   let deployer: Signer;
@@ -292,7 +294,14 @@ describe("(factory) Zeto based fungible token with anonymity without encryption
 
     // deploy a real Zeto token so that the "implementation" contract can
     // pass the code length check in the ERC1967Proxy contract
-    const Zeto = await ethers.getContractFactory("Zeto_Anon");
+    const ZetoLockableLib = await ethers.getContractFactory("ZetoLockableLib");
+    const zetoLockableLib = await ZetoLockableLib.deploy();
+    await zetoLockableLib.waitForDeployment();
+
+    const Zeto = await getLinkedContractFactory(
+      "Zeto_Anon",
+      withZetoLockableLib(zetoLockableLib),
+    );
     const zeto = await Zeto.deploy();
     await zeto.waitForDeployment();
 

package/test/lib/deploy.ts

@@ -12,8 +12,10 @@ import {
 import fungibilities from "../../scripts/tokens.json";
 import { logger } from "./utils";
 import { ethers } from "hardhat";
+import { assertEip170Compliant } from "./eip170";
 
 export async function deployZeto(tokenName: string) {
+  await assertEip170Compliant(tokenName);
   let zeto, erc20, deployer;
 
   // for testing with public chains, skip deployment if

package/test/lib/eip170.ts

@@ -0,0 +1,23 @@
+import { artifacts } from "hardhat";
+import {
+  EIP170_BYTE_LIMIT,
+  EIP170_EXEMPT_TOKEN_SET,
+} from "../../config/eip170";
+
+/**
+ * Assert a token implementation fits EIP-170 before deploy when it is not exempt.
+ * Hardhat Network runs with allowUnlimitedContractSize enabled (see hardhat.config.ts);
+ * exempt tokens may exceed the limit.
+ */
+export async function assertEip170Compliant(tokenName: string): Promise<void> {
+  if (EIP170_EXEMPT_TOKEN_SET.has(tokenName)) {
+    return;
+  }
+  const artifact = await artifacts.readArtifact(tokenName);
+  const deployedBytes = (artifact.deployedBytecode.length - 2) / 2;
+  if (deployedBytes > EIP170_BYTE_LIMIT) {
+    throw new Error(
+      `${tokenName} deployed bytecode is ${deployedBytes} bytes, exceeding EIP-170 limit ${EIP170_BYTE_LIMIT}`,
+    );
+  }
+}