-
Notifications
You must be signed in to change notification settings - Fork 234
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
4 changed files
with
202 additions
and
31 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
83 changes: 83 additions & 0 deletions
83
yarn-project/acir-simulator/src/avm/opcodes/accrued_substate.test.ts
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,83 @@ | ||
| import { Fr } from '@aztec/foundation/fields'; | ||
|
|
||
| import { mock } from 'jest-mock-extended'; | ||
|
|
||
| import { AvmMachineState } from '../avm_machine_state.js'; | ||
| import { initExecutionEnvironment } from '../fixtures/index.js'; | ||
| import { HostStorage } from '../journal/host_storage.js'; | ||
| import { AvmJournal } from '../journal/journal.js'; | ||
| import { EmitNoteHash, EmitNullifier, EmitUnencryptedLog, SendL2ToL1Message } from './accrued_substate.js'; | ||
| import { StaticCallStorageAlterError } from './storage.js'; | ||
|
|
||
| describe('Accrued Substate', () => { | ||
| let journal: AvmJournal; | ||
| let machineState: AvmMachineState; | ||
|
|
||
| beforeEach(() => { | ||
| const hostStorage = mock<HostStorage>(); | ||
| journal = new AvmJournal(hostStorage); | ||
| machineState = new AvmMachineState([], initExecutionEnvironment()); | ||
| }); | ||
|
|
||
| it('Should append a new note hash correctly', () => { | ||
| const value = new Fr(69n); | ||
| machineState.writeMemory(0, value); | ||
|
|
||
| new EmitNoteHash(0).execute(machineState, journal); | ||
|
|
||
| const journalState = journal.flush(); | ||
| expect(journalState.newNoteHashes).toEqual([value]); | ||
| }); | ||
|
|
||
| it('Should append a new nullifier correctly', () => { | ||
| const value = new Fr(69n); | ||
| machineState.writeMemory(0, value); | ||
|
|
||
| new EmitNullifier(0).execute(machineState, journal); | ||
|
|
||
| const journalState = journal.flush(); | ||
| expect(journalState.newNullifiers).toEqual([value]); | ||
| }); | ||
|
|
||
| it('Should append unencrypted logs correctly', () => { | ||
| const startOffset = 0; | ||
| const length = 2; | ||
|
|
||
| const values = [new Fr(69n), new Fr(420n)]; | ||
| machineState.writeMemoryChunk(0, values); | ||
|
|
||
| new EmitUnencryptedLog(startOffset, length).execute(machineState, journal); | ||
|
|
||
| const journalState = journal.flush(); | ||
| expect(journalState.newLogs).toEqual([values]); | ||
| }); | ||
|
|
||
| it('Should append l1 to l2 messages correctly', () => { | ||
| const startOffset = 0; | ||
| const length = 2; | ||
|
|
||
| const values = [new Fr(69n), new Fr(420n)]; | ||
| machineState.writeMemoryChunk(0, values); | ||
|
|
||
| new SendL2ToL1Message(startOffset, length).execute(machineState, journal); | ||
|
|
||
| const journalState = journal.flush(); | ||
| expect(journalState.newLogs).toEqual([values]); | ||
| }); | ||
|
|
||
| it('All substate instructions should fail within a static call', () => { | ||
| const executionEnvironment = initExecutionEnvironment({ isStaticCall: true }); | ||
| machineState = new AvmMachineState([], executionEnvironment); | ||
|
|
||
| const instructions = [ | ||
| new EmitNoteHash(0), | ||
| new EmitNullifier(0), | ||
| new EmitUnencryptedLog(0, 1), | ||
| new SendL2ToL1Message(0, 1), | ||
| ]; | ||
|
|
||
| for (const instruction of instructions) { | ||
| expect(() => instruction.execute(machineState, journal)).toThrowError(StaticCallStorageAlterError); | ||
| } | ||
| }); | ||
| }); |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,88 @@ | ||
| import { AvmMachineState } from '../avm_machine_state.js'; | ||
| import { AvmJournal } from '../journal/journal.js'; | ||
| import { Instruction } from './instruction.js'; | ||
| import { StaticCallStorageAlterError } from './storage.js'; | ||
|
|
||
| export class EmitNoteHash extends Instruction { | ||
| static type: string = 'EMITNOTEHASH'; | ||
| static numberOfOperands = 1; | ||
|
|
||
| constructor(private noteHashOffset: number) { | ||
| super(); | ||
| } | ||
|
|
||
| execute(machineState: AvmMachineState, journal: AvmJournal): void { | ||
| if (machineState.executionEnvironment.isStaticCall) { | ||
| throw new StaticCallStorageAlterError(); | ||
| } | ||
|
|
||
| const noteHash = machineState.readMemory(this.noteHashOffset); | ||
|
|
||
| journal.writeNoteHash(noteHash); | ||
|
|
||
| this.incrementPc(machineState); | ||
| } | ||
| } | ||
|
|
||
| export class EmitNullifier extends Instruction { | ||
| static type: string = 'EMITNULLIFIER'; | ||
| static numberOfOperands = 1; | ||
|
|
||
| constructor(private nullifierOffset: number) { | ||
| super(); | ||
| } | ||
|
|
||
| execute(machineState: AvmMachineState, journal: AvmJournal): void { | ||
| if (machineState.executionEnvironment.isStaticCall) { | ||
| throw new StaticCallStorageAlterError(); | ||
| } | ||
|
|
||
| const nullifier = machineState.readMemory(this.nullifierOffset); | ||
|
|
||
| journal.writeNullifier(nullifier); | ||
|
|
||
| this.incrementPc(machineState); | ||
| } | ||
| } | ||
|
|
||
| export class EmitUnencryptedLog extends Instruction { | ||
| static type: string = 'EMITUNENCRYPTEDLOG'; | ||
| static numberOfOperands = 2; | ||
|
|
||
| constructor(private logOffset: number, private logSize: number) { | ||
| super(); | ||
| } | ||
|
|
||
| execute(machineState: AvmMachineState, journal: AvmJournal): void { | ||
| if (machineState.executionEnvironment.isStaticCall) { | ||
| throw new StaticCallStorageAlterError(); | ||
| } | ||
|
|
||
| const log = machineState.readMemoryChunk(this.logOffset, this.logSize); | ||
|
|
||
| journal.writeLog(log); | ||
|
|
||
| this.incrementPc(machineState); | ||
| } | ||
| } | ||
|
|
||
| export class SendL2ToL1Message extends Instruction { | ||
| static type: string = 'EMITUNENCRYPTEDLOG'; | ||
| static numberOfOperands = 2; | ||
|
|
||
| constructor(private msgOffset: number, private msgSize: number) { | ||
| super(); | ||
| } | ||
|
|
||
| execute(machineState: AvmMachineState, journal: AvmJournal): void { | ||
| if (machineState.executionEnvironment.isStaticCall) { | ||
| throw new StaticCallStorageAlterError(); | ||
| } | ||
|
|
||
| const msg = machineState.readMemoryChunk(this.msgOffset, this.msgSize); | ||
|
|
||
| journal.writeLog(msg); | ||
|
|
||
| this.incrementPc(machineState); | ||
| } | ||
| } |