@aztec/pxe 0.42.0 → 0.44.0

package/src/note_processor/produce_note_dao.ts

@@ -1,9 +1,12 @@
 import { type L1NotePayload, type TxHash } from '@aztec/circuit-types';
 import { Fr, type PublicKey } from '@aztec/circuits.js';
 import { computeNoteHashNonce, siloNullifier } from '@aztec/circuits.js/hash';
-import { type AcirSimulator } from '@aztec/simulator';
+import { type Logger } from '@aztec/foundation/log';
+import { type AcirSimulator, ContractNotFoundError } from '@aztec/simulator';
 
-import { NoteDao } from '../database/note_dao.js';
+import { DeferredNoteDao } from '../database/deferred_note_dao.js';
+import { IncomingNoteDao } from '../database/incoming_note_dao.js';
+import { OutgoingNoteDao } from '../database/outgoing_note_dao.js';
 
 /**
  * Decodes a note from a transaction that we know was intended for us.
@@ -11,96 +14,207 @@ import { NoteDao } from '../database/note_dao.js';
  * Accepts a set of excluded indices, which are indices that have been assigned a note in the same tx.
  * Inserts the index of the note into the excludedIndices set if the note is successfully decoded.
  *
- * @param publicKey - The public counterpart to the private key to be used in note decryption.
+ * @param ivpkM - The public counterpart to the secret key to be used in the decryption of incoming note logs.
+ * @param ovpkM - The public counterpart to the secret key to be used in the decryption of outgoing note logs.
  * @param payload - An instance of l1NotePayload.
  * @param txHash - The hash of the transaction that created the note. Equivalent to the first nullifier of the transaction.
  * @param newNoteHashes - New note hashes in this transaction, one of which belongs to this note.
  * @param dataStartIndexForTx - The next available leaf index for the note hash tree for this transaction.
  * @param excludedIndices - Indices that have been assigned a note in the same tx. Notes in a tx can have the same l1NotePayload, we need to find a different index for each replicate.
  * @param simulator - An instance of AcirSimulator.
- * @returns an instance of NoteDao, or throws. inserts the index of the note into the excludedIndices set.
+ * @returns An object containing the incoming, outgoing, and deferred notes.
  */
-export async function produceNoteDao(
+export async function produceNoteDaos(
   simulator: AcirSimulator,
-  publicKey: PublicKey,
+  ivpkM: PublicKey | undefined,
+  ovpkM: PublicKey | undefined,
   payload: L1NotePayload,
   txHash: TxHash,
   newNoteHashes: Fr[],
   dataStartIndexForTx: number,
   excludedIndices: Set<number>,
-): Promise<NoteDao> {
-  const { commitmentIndex, nonce, innerNoteHash, siloedNullifier } = await findNoteIndexAndNullifier(
-    simulator,
-    newNoteHashes,
-    txHash,
-    payload,
-    excludedIndices,
-  );
-  const index = BigInt(dataStartIndexForTx + commitmentIndex);
-  excludedIndices?.add(commitmentIndex);
-  return new NoteDao(
-    payload.note,
-    payload.contractAddress,
-    payload.storageSlot,
-    payload.noteTypeId,
-    txHash,
-    nonce,
-    innerNoteHash,
-    siloedNullifier,
-    index,
-    publicKey,
-  );
+  log: Logger,
+): Promise<{
+  incomingNote: IncomingNoteDao | undefined;
+  outgoingNote: OutgoingNoteDao | undefined;
+  incomingDeferredNote: DeferredNoteDao | undefined;
+  outgoingDeferredNote: DeferredNoteDao | undefined;
+}> {
+  if (!ivpkM && !ovpkM) {
+    throw new Error('Both ivpkM and ovpkM are undefined. Cannot create note.');
+  }
+
+  let incomingNote: IncomingNoteDao | undefined;
+  let outgoingNote: OutgoingNoteDao | undefined;
+  let incomingDeferredNote: DeferredNoteDao | undefined;
+  let outgoingDeferredNote: DeferredNoteDao | undefined;
+
+  try {
+    if (ivpkM) {
+      const { noteHashIndex, nonce, innerNoteHash, siloedNullifier } = await findNoteIndexAndNullifier(
+        simulator,
+        newNoteHashes,
+        txHash,
+        payload,
+        excludedIndices,
+        true, // For incoming we compute a nullifier (recipient of incoming is the party that nullifies).
+      );
+      const index = BigInt(dataStartIndexForTx + noteHashIndex);
+      excludedIndices?.add(noteHashIndex);
+
+      incomingNote = new IncomingNoteDao(
+        payload.note,
+        payload.contractAddress,
+        payload.storageSlot,
+        payload.noteTypeId,
+        txHash,
+        nonce,
+        innerNoteHash,
+        siloedNullifier,
+        index,
+        ivpkM,
+      );
+    }
+  } catch (e) {
+    if (e instanceof ContractNotFoundError) {
+      log.warn(e.message);
+
+      if (ivpkM) {
+        incomingDeferredNote = new DeferredNoteDao(
+          ivpkM,
+          payload.note,
+          payload.contractAddress,
+          payload.storageSlot,
+          payload.noteTypeId,
+          txHash,
+          newNoteHashes,
+          dataStartIndexForTx,
+        );
+      }
+    } else {
+      log.error(`Could not process note because of "${e}". Discarding note...`);
+    }
+  }
+
+  try {
+    if (ovpkM) {
+      if (incomingNote) {
+        // Incoming note is defined meaning that this PXE has both the incoming and outgoing keys. We can skip computing
+        // note hash and note index since we already have them in the incoming note.
+        outgoingNote = new OutgoingNoteDao(
+          payload.note,
+          payload.contractAddress,
+          payload.storageSlot,
+          payload.noteTypeId,
+          txHash,
+          incomingNote.nonce,
+          incomingNote.innerNoteHash,
+          incomingNote.index,
+          ovpkM,
+        );
+      } else {
+        const { noteHashIndex, nonce, innerNoteHash } = await findNoteIndexAndNullifier(
+          simulator,
+          newNoteHashes,
+          txHash,
+          payload,
+          excludedIndices,
+          false, // For outgoing we do not compute a nullifier.
+        );
+        const index = BigInt(dataStartIndexForTx + noteHashIndex);
+        excludedIndices?.add(noteHashIndex);
+        outgoingNote = new OutgoingNoteDao(
+          payload.note,
+          payload.contractAddress,
+          payload.storageSlot,
+          payload.noteTypeId,
+          txHash,
+          nonce,
+          innerNoteHash,
+          index,
+          ovpkM,
+        );
+      }
+    }
+  } catch (e) {
+    if (e instanceof ContractNotFoundError) {
+      log.warn(e.message);
+
+      if (ovpkM) {
+        outgoingDeferredNote = new DeferredNoteDao(
+          ovpkM,
+          payload.note,
+          payload.contractAddress,
+          payload.storageSlot,
+          payload.noteTypeId,
+          txHash,
+          newNoteHashes,
+          dataStartIndexForTx,
+        );
+      }
+    } else {
+      log.error(`Could not process note because of "${e}". Discarding note...`);
+    }
+  }
+
+  return {
+    incomingNote,
+    outgoingNote,
+    incomingDeferredNote,
+    outgoingDeferredNote,
+  };
 }
 
 /**
- * Find the index of the note in the note hash tree by computing the note hash with different nonce and see which
- * commitment for the current tx matches this value.
- * Compute a nullifier for a given l1NotePayload.
- * The nullifier is calculated using the private key of the account,
- * contract address, and the note associated with the l1NotePayload.
- * This method assists in identifying spent commitments in the private state.
- * @param commitments - Commitments in the tx. One of them should be the note's commitment.
- * @param txHash - First nullifier in the tx.
- * @param l1NotePayload - An instance of l1NotePayload.
+ * Finds nonce, index, inner hash and siloed nullifier for a given note.
+ * @dev Finds the index in the note hash tree by computing the note hash with different nonce and see which hash for
+ * the current tx matches this value.
+ * @remarks This method assists in identifying spent notes in the note hash tree.
+ * @param noteHashes - Note hashes in the tx. One of them should correspond to the note we are looking for
+ * @param txHash - Hash of a tx the note was emitted in.
+ * @param l1NotePayload - The note payload.
  * @param excludedIndices - Indices that have been assigned a note in the same tx. Notes in a tx can have the same
  * l1NotePayload. We need to find a different index for each replicate.
- * @returns Information for a decrypted note, including the index of its commitment, nonce, inner note
- * hash, and the siloed nullifier. Throw if cannot find the nonce for the note.
+ * @param computeNullifier - A flag indicating whether to compute the nullifier or just return 0.
+ * @returns Nonce, index, inner hash and siloed nullifier for a given note.
+ * @throws If cannot find the nonce for the note.
  */
 async function findNoteIndexAndNullifier(
   simulator: AcirSimulator,
-  commitments: Fr[],
+  noteHashes: Fr[],
   txHash: TxHash,
   { contractAddress, storageSlot, noteTypeId, note }: L1NotePayload,
   excludedIndices: Set<number>,
+  computeNullifier: boolean,
 ) {
-  let commitmentIndex = 0;
+  let noteHashIndex = 0;
   let nonce: Fr | undefined;
   let innerNoteHash: Fr | undefined;
   let siloedNoteHash: Fr | undefined;
   let innerNullifier: Fr | undefined;
   const firstNullifier = Fr.fromBuffer(txHash.toBuffer());
 
-  for (; commitmentIndex < commitments.length; ++commitmentIndex) {
-    if (excludedIndices.has(commitmentIndex)) {
+  for (; noteHashIndex < noteHashes.length; ++noteHashIndex) {
+    if (excludedIndices.has(noteHashIndex)) {
       continue;
     }
 
-    const commitment = commitments[commitmentIndex];
-    if (commitment.equals(Fr.ZERO)) {
+    const noteHash = noteHashes[noteHashIndex];
+    if (noteHash.equals(Fr.ZERO)) {
       break;
     }
 
-    const expectedNonce = computeNoteHashNonce(firstNullifier, commitmentIndex);
-    ({ innerNoteHash, siloedNoteHash, innerNullifier } = await simulator.computeNoteHashAndNullifier(
+    const expectedNonce = computeNoteHashNonce(firstNullifier, noteHashIndex);
+    ({ innerNoteHash, siloedNoteHash, innerNullifier } = await simulator.computeNoteHashAndOptionallyANullifier(
       contractAddress,
       expectedNonce,
       storageSlot,
       noteTypeId,
+      computeNullifier,
       note,
     ));
 
-    if (commitment.equals(siloedNoteHash)) {
+    if (noteHash.equals(siloedNoteHash)) {
       nonce = expectedNonce;
       break;
     }
@@ -109,11 +223,11 @@ async function findNoteIndexAndNullifier(
   if (!nonce) {
     // NB: this used to warn the user that a decrypted log didn't match any notes.
     // This was previously fine as we didn't chop transient note logs, but now we do (#1641 complete).
-    throw new Error('Cannot find a matching commitment for the note.');
+    throw new Error('Cannot find a matching note hash for the note.');
   }
 
   return {
-    commitmentIndex,
+    noteHashIndex,
     nonce,
     innerNoteHash: innerNoteHash!,
     siloedNullifier: siloNullifier(contractAddress, innerNullifier!),

package/src/pxe_http/pxe_http_server.ts

@@ -18,6 +18,7 @@ import {
   UnencryptedL2BlockL2Logs,
 } from '@aztec/circuit-types';
 import { FunctionSelector } from '@aztec/circuits.js';
+import { NoteSelector } from '@aztec/foundation/abi';
 import { AztecAddress } from '@aztec/foundation/aztec-address';
 import { EthAddress } from '@aztec/foundation/eth-address';
 import { Fr, GrumpkinScalar, Point } from '@aztec/foundation/fields';
@@ -49,6 +50,7 @@ export function createPXERpcServer(pxeService: PXE): JsonRpcServer {
       L2Block,
       TxEffect,
       LogId,
+      NoteSelector,
     },
     { SimulatedTx, Tx, TxReceipt, EncryptedNoteL2BlockL2Logs, UnencryptedL2BlockL2Logs, NullifierMembershipWitness },
     ['start', 'stop'],

package/src/pxe_service/create_pxe_service.ts

@@ -5,6 +5,7 @@ import { createDebugLogger } from '@aztec/foundation/log';
 import { KeyStore } from '@aztec/key-store';
 import { AztecLmdbStore } from '@aztec/kv-store/lmdb';
 import { initStoreForRollup } from '@aztec/kv-store/utils';
+import { getCanonicalAuthRegistry } from '@aztec/protocol-contracts/auth-registry';
 import { getCanonicalClassRegisterer } from '@aztec/protocol-contracts/class-registerer';
 import { getCanonicalGasToken } from '@aztec/protocol-contracts/gas-token';
 import { getCanonicalInstanceDeployer } from '@aztec/protocol-contracts/instance-deployer';
@@ -67,6 +68,7 @@ export async function createPXEService(
     getCanonicalMultiCallEntrypointContract(),
     getCanonicalGasToken(),
     getCanonicalKeyRegistry(),
+    getCanonicalAuthRegistry(),
   ]) {
     await server.registerContract(contract);
   }