init

AztecProtocol · Nov 19, 2024 · 3e4e3d0 · 3e4e3d0
1 parent 9643dcd
commit 3e4e3d0
Show file tree

Hide file tree

Showing 3 changed files with 30 additions and 1 deletion.
diff --git a/yarn-project/pxe/src/database/contracts/contract_artifact_db.ts b/yarn-project/pxe/src/database/contracts/contract_artifact_db.ts
@@ -9,6 +9,7 @@ export interface ContractArtifactDatabase {
    * Adds a new contract artifact to the database or updates an existing one.
    * @param id - Id of the corresponding contract class.
    * @param contract - Contract artifact to add.
+   * @throws - If there are duplicate private function selectors.
    */
   addContractArtifact(id: Fr, contract: ContractArtifact): Promise<void>;
   /**

diff --git a/yarn-project/pxe/src/database/kv_pxe_database.ts b/yarn-project/pxe/src/database/kv_pxe_database.ts
@@ -9,7 +9,7 @@ import {
   SerializableContractInstance,
   computePoint,
 } from '@aztec/circuits.js';
-import { type ContractArtifact } from '@aztec/foundation/abi';
+import { type ContractArtifact, FunctionSelector, FunctionType } from '@aztec/foundation/abi';
 import { toBufferBE } from '@aztec/foundation/bigint-buffer';
 import { Fr } from '@aztec/foundation/fields';
 import {
@@ -128,6 +128,19 @@ export class KVPxeDatabase implements PxeDatabase {
   }
 
   public async addContractArtifact(id: Fr, contract: ContractArtifact): Promise<void> {
+    const privateSelectors = contract.functions
+      .filter(functionArtifact => functionArtifact.functionType === FunctionType.PRIVATE)
+      .map(privateFunctionArtifact =>
+        FunctionSelector.fromNameAndParameters(
+          privateFunctionArtifact.name,
+          privateFunctionArtifact.parameters,
+        ).toString(),
+      );
+
+    if (privateSelectors.length !== new Set(privateSelectors).size) {
+      throw new Error('Repeated function selectors of private functions');
+    }
+
     await this.#contractArtifacts.set(id.toString(), contractArtifactToBuffer(contract));
   }
 

diff --git a/yarn-project/pxe/src/database/pxe_database_test_suite.ts b/yarn-project/pxe/src/database/pxe_database_test_suite.ts
@@ -8,9 +8,11 @@ import {
   computePoint,
 } from '@aztec/circuits.js';
 import { makeHeader } from '@aztec/circuits.js/testing';
+import { FunctionType } from '@aztec/foundation/abi';
 import { randomInt } from '@aztec/foundation/crypto';
 import { Fr, Point } from '@aztec/foundation/fields';
 import { BenchmarkingContractArtifact } from '@aztec/noir-contracts.js/Benchmarking';
+import { TestContractArtifact } from '@aztec/noir-contracts.js/Test';
 
 import { type IncomingNoteDao } from './incoming_note_dao.js';
 import { randomIncomingNoteDao } from './incoming_note_dao.test.js';
@@ -391,6 +393,19 @@ export function describePxeDatabase(getDatabase: () => PxeDatabase) {
         await expect(database.getContractArtifact(id)).resolves.toEqual(artifact);
       });
 
+      it('does not store a contract artifact with a duplicate private function selector', async () => {
+        const artifact = TestContractArtifact;
+        const index = artifact.functions.findIndex(fn => fn.functionType === FunctionType.PRIVATE);
+
+        const copiedFn = structuredClone(artifact.functions[index]);
+        artifact.functions.push(copiedFn);
+
+        const id = Fr.random();
+        await expect(database.addContractArtifact(id, artifact)).rejects.toThrow(
+          'Repeated function selectors of private functions',
+        );
+      });
+
       it('stores a contract instance', async () => {
         const address = AztecAddress.random();
         const instance = SerializableContractInstance.random().withAddress(address);