Skip to content

Commit d0711e3

Browse files
ohad-nir-starkwareohad-nir-starkware
committed
implement Blake2sLastBlock opcode in runner
1 parent 4c2fdad commit d0711e3

File tree

5 files changed

+147
-33
lines changed

5 files changed

+147
-33
lines changed

‎CHANGELOG.md‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,8 @@
22

33
#### Upcoming Changes
44

5+
* feat: implement `Blake2sLastBlock` opcode in VM [#1932](https://github.com/lambdaclass/cairo-vm/pull/1932)
6+
57
* feat: implement `Blake2s` opcode in VM [#1927](https://github.com/lambdaclass/cairo-vm/pull/1927)
68

79
* feat: remove `NonZeroReservedBits` from `VirtualMachineError` [#1948](https://github.com/lambdaclass/cairo-vm/pull/1948)

‎cairo_programs/stwo_exclusive_programs/blake2s_opcode_test.cairo‎

Lines changed: 29 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -1,18 +1,24 @@
11
%builtins range_check bitwise
22

33
from starkware.cairo.common.alloc import alloc
4+
from starkware.cairo.common.bool import FALSE, TRUE
45
from starkware.cairo.common.cairo_blake2s.blake2s import STATE_SIZE_FELTS, INPUT_BLOCK_FELTS, _get_sigma
56
from starkware.cairo.common.cairo_blake2s.packed_blake2s import N_PACKED_INSTANCES, blake2s_compress
67
from starkware.cairo.common.cairo_builtins import BitwiseBuiltin
78

89
const COUNTER = 64;
910
const U32_MASK = 0xffffffff;
1011

11-
// Tests the Blake2s opcode runner using a preexisting implementation within the repo as reference.
12-
// The initial state, a random message of 64 bytes and counter are used as input.
12+
// Tests the Blake2s and Blake2sLastBlock opcode runners using a preexisting implementation within the repo as reference.
13+
// The initial state, a random message of 64 bytes and a counter are used as input.
1314
// Both the opcode and the reference implementation are run on the same inputs and then their outputs are compared.
1415
// Before comparing the outputs, it is verified that the opcode runner has written the output to the correct location.
1516
func main{range_check_ptr, bitwise_ptr: BitwiseBuiltin*}() {
17+
run_blake_test(is_last_block=FALSE);
18+
run_blake_test(is_last_block=TRUE);
19+
return ();
20+
}
21+
func run_blake_test{range_check_ptr, bitwise_ptr: BitwiseBuiltin*}(is_last_block: felt) {
1622
alloc_locals;
1723

1824
let (local random_message) = alloc();
@@ -52,7 +58,7 @@ func main{range_check_ptr, bitwise_ptr: BitwiseBuiltin*}() {
5258
h=input_state,
5359
message=random_message,
5460
t0=COUNTER,
55-
f0=0,
61+
f0=is_last_block * U32_MASK,
5662
sigma=sigma,
5763
output=cairo_output,
5864
);
@@ -76,7 +82,8 @@ func main{range_check_ptr, bitwise_ptr: BitwiseBuiltin*}() {
7682
assert bitwise_ptr[7].y = U32_MASK;
7783

7884
// Run the blake2s opcode runner on the same inputs and store its output.
79-
let vm_output = run_blake2s(
85+
let vm_output = run_blake2s_opcode(
86+
is_last_block = is_last_block,
8087
dst=COUNTER,
8188
op0=input_state,
8289
op1=random_message,
@@ -107,7 +114,7 @@ func main{range_check_ptr, bitwise_ptr: BitwiseBuiltin*}() {
107114
return ();
108115
}
109116

110-
// Forces the runner to execute the Blake2s with the given operands.
117+
// Forces the runner to execute the Blake2s or Blake2sLastBlock opcode with the given operands.
111118
// op0 is a pointer to an array of 8 felts as u32 integers of the state.
112119
// op1 is a pointer to an array of 16 felts as u32 integers of the messsage.
113120
// dst is a felt representing a u32 of the counter.
@@ -116,7 +123,8 @@ func main{range_check_ptr, bitwise_ptr: BitwiseBuiltin*}() {
116123
// An instruction encoding is built from offsets -5, -4, -3 and flags which are all 0 except for
117124
// those denoting uses of fp as the base for operand addresses and flag_opcode_blake (16th flag).
118125
// The instruction is then written to [pc] and the runner is forced to execute Blake2s.
119-
func run_blake2s(
126+
func run_blake2s_opcode(
127+
is_last_block: felt,
120128
dst: felt,
121129
op0: felt*,
122130
op1: felt*,
@@ -127,9 +135,9 @@ func run_blake2s(
127135
let offset0 = (2**15)-5;
128136
let offset1 = (2**15)-4;
129137
let offset2 = (2**15)-3;
130-
static_assert dst == [fp -5];
131-
static_assert op0 == [fp -4];
132-
static_assert op1 == [fp -3];
138+
static_assert dst == [fp - 5];
139+
static_assert op0 == [fp - 4];
140+
static_assert op1 == [fp - 3];
133141
134142
// Set the flags for the instruction.
135143
let flag_dst_base_fp = 1;
@@ -147,17 +155,24 @@ func run_blake2s(
147155
let flag_opcode_call = 0;
148156
let flag_opcode_ret = 0;
149157
let flag_opcode_assert_eq = 0;
150-
let flag_opcode_blake2s = 1;
151158
152-
// Build the instruction encoding.
153-
let flag_num = flag_dst_base_fp+flag_op0_base_fp*(2**1)+flag_op1_imm*(2**2)+flag_op1_base_fp*(2**3)+flag_opcode_blake2s*(2**15);
154-
let instruction_num = offset0 + offset1*(2**16) + offset2*(2**32) + flag_num*(2**48);
155-
static_assert instruction_num==9226608988349300731;
159+
let flag_num = flag_dst_base_fp+flag_op0_base_fp*(2**1)+flag_op1_imm*(2**2)+flag_op1_base_fp*(2**3);
160+
let blake2s_opcode_extension_num = 1;
161+
let blake2s_last_block_opcode_extension_num = 2;
162+
let blake2s_instruction_num = offset0 + offset1*(2**16) + offset2*(2**32) + flag_num*(2**48) + blake2s_opcode_extension_num*(2**63);
163+
let blake2s_last_block_instruction_num = offset0 + offset1*(2**16) + offset2*(2**32) + flag_num*(2**48) + blake2s_last_block_opcode_extension_num*(2**63);
164+
static_assert blake2s_instruction_num==9226608988349300731;
165+
static_assert blake2s_last_block_instruction_num==18449981025204076539;
156166
157167
// Write the instruction to [pc] and point [ap] to the designated output.
158168
let (local vm_output) = alloc();
159169
assert [ap] = cast(vm_output, felt);
170+
171+
jmp last_block if is_last_block!=0;
160172
dw 9226608988349300731;
173+
return cast([ap], felt*);
161174
175+
last_block:
176+
dw 18449981025204076539;
162177
return cast([ap], felt*);
163178
}

‎vm/src/types/instruction.rs‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -81,6 +81,7 @@ pub enum Opcode {
8181
pub enum OpcodeExtension {
8282
Stone,
8383
Blake,
84+
BlakeFinalize,
8485
}
8586

8687
impl Instruction {

‎vm/src/vm/decoding/decoder.rs‎

Lines changed: 13 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -102,24 +102,26 @@ pub fn decode_instruction(encoded_instr: u128) -> Result<Instruction, VirtualMac
102102

103103
let opcode_extension = match opcode_extension_num {
104104
0 => OpcodeExtension::Stone,
105-
1 => {
106-
if opcode != Opcode::NOp
107-
|| (op1_addr != Op1Addr::FP && op1_addr != Op1Addr::AP)
108-
|| res != Res::Op1
109-
|| pc_update != PcUpdate::Regular
110-
|| (ap_update_num != 0 && ap_update_num != 2)
111-
{
112-
return Err(VirtualMachineError::InvalidBlake2sFlags(flags & 0x7FFF));
113-
};
114-
OpcodeExtension::Blake
115-
}
105+
1 => OpcodeExtension::Blake,
106+
2 => OpcodeExtension::BlakeFinalize,
116107
_ => {
117108
return Err(VirtualMachineError::InvalidOpcodeExtension(
118109
opcode_extension_num,
119110
))
120111
}
121112
};
122113

114+
if (opcode_extension == OpcodeExtension::Blake
115+
|| opcode_extension == OpcodeExtension::BlakeFinalize)
116+
&& (opcode != Opcode::NOp
117+
|| (op1_addr != Op1Addr::FP && op1_addr != Op1Addr::AP)
118+
|| res != Res::Op1
119+
|| pc_update != PcUpdate::Regular
120+
|| (ap_update_num != 0 && ap_update_num != 2))
121+
{
122+
return Err(VirtualMachineError::InvalidBlake2sFlags(flags & 0x7FFF));
123+
}
124+
123125
let ap_update = match (ap_update_num, opcode == Opcode::Call) {
124126
(0, true) => ApUpdate::Add2,
125127
(0, false) => ApUpdate::Regular,

‎vm/src/vm/vm_core.rs‎

Lines changed: 102 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -445,8 +445,13 @@ impl VirtualMachine {
445445
.memory
446446
.mark_as_accessed(operands_addresses.op1_addr);
447447

448-
if instruction.opcode_extension == OpcodeExtension::Blake {
449-
self.handle_blake2s_instruction(&operands_addresses)?;
448+
if instruction.opcode_extension == OpcodeExtension::Blake
449+
|| instruction.opcode_extension == OpcodeExtension::BlakeFinalize
450+
{
451+
self.handle_blake2s_instruction(
452+
&operands_addresses,
453+
instruction.opcode_extension == OpcodeExtension::BlakeFinalize,
454+
)?;
450455
}
451456

452457
self.update_registers(instruction, operands)?;
@@ -455,7 +460,7 @@ impl VirtualMachine {
455460
Ok(())
456461
}
457462

458-
/// Executes a Blake2s instruction.
463+
/// Executes a Blake2s or Blake2sLastBlock instruction.
459464
/// Expects operands to be RelocatableValue and to point to segments of memory.
460465
/// op0 is expected to point to a sequence of 8 u32 values (state).
461466
/// op1 is expected to point to a sequence of 16 u32 values (message).
@@ -469,6 +474,7 @@ impl VirtualMachine {
469474
fn handle_blake2s_instruction(
470475
&mut self,
471476
operands_addresses: &OperandsAddresses,
477+
is_last_block: bool,
472478
) -> Result<(), VirtualMachineError> {
473479
let counter = self.segments.memory.get_u32(operands_addresses.dst_addr)?;
474480

@@ -493,7 +499,14 @@ impl VirtualMachine {
493499
let ap = self.run_context.get_ap();
494500
let output_address = self.segments.memory.get_relocatable(ap)?;
495501

496-
let new_state = blake2s_compress(&state, &message, counter, 0, 0, 0);
502+
let new_state = blake2s_compress(
503+
&state,
504+
&message,
505+
counter,
506+
0,
507+
if !is_last_block { 0 } else { 0xffffffff },
508+
0,
509+
);
497510

498511
for (i, &val) in new_state.iter().enumerate() {
499512
self.segments.memory.insert_as_accessed(
@@ -4496,7 +4509,7 @@ mod tests {
44964509
};
44974510

44984511
assert_matches!(
4499-
vm.handle_blake2s_instruction(&operands_addresses),
4512+
vm.handle_blake2s_instruction(&operands_addresses, false),
45004513
Err(VirtualMachineError::Memory(MemoryError::UnknownMemoryCell(bx))) if *bx == (0, 7).into()
45014514
);
45024515
}
@@ -4528,7 +4541,7 @@ mod tests {
45284541
};
45294542

45304543
assert_matches!(
4531-
vm.handle_blake2s_instruction(&operands_addresses),
4544+
vm.handle_blake2s_instruction(&operands_addresses, false),
45324545
Err(VirtualMachineError::Memory(MemoryError::UnknownMemoryCell(bx))) if *bx == (0, 8).into()
45334546
);
45344547
}
@@ -4568,7 +4581,7 @@ mod tests {
45684581
};
45694582

45704583
assert_matches!(
4571-
vm.handle_blake2s_instruction(&operands_addresses),
4584+
vm.handle_blake2s_instruction(&operands_addresses, false),
45724585
Err(VirtualMachineError::Memory(MemoryError::InconsistentMemory(bx))) if *bx == ((0, 0).into(),0.into(),1848029226.into())
45734586
);
45744587
}
@@ -4621,7 +4634,10 @@ mod tests {
46214634
ap: 0,
46224635
fp: 0,
46234636
};
4624-
assert_matches!(vm.handle_blake2s_instruction(&operands_addresses), Ok(()));
4637+
assert_matches!(
4638+
vm.handle_blake2s_instruction(&operands_addresses, false),
4639+
Ok(())
4640+
);
46254641

46264642
let state: [u32; 8] = vm
46274643
.get_u32_range((0, 0).into(), 8)
@@ -4647,6 +4663,84 @@ mod tests {
46474663
assert_eq!(new_state, expected_new_state);
46484664
}
46494665

4666+
#[test]
4667+
#[cfg_attr(target_arch = "wasm32", wasm_bindgen_test)]
4668+
fn handle_blake2s_last_block_instruction_ok() {
4669+
let mut vm = vm!();
4670+
vm.segments.memory = memory![
4671+
// State
4672+
((0, 0), 0x6B08E647),
4673+
((0, 1), 0xBB67AE85),
4674+
((0, 2), 0x3C6EF372),
4675+
((0, 3), 0xA54FF53A),
4676+
((0, 4), 0x510E527F),
4677+
((0, 5), 0x9B05688C),
4678+
((0, 6), 0x1F83D9AB),
4679+
((0, 7), 0x5BE0CD19),
4680+
// Message
4681+
((0, 8), 930933030),
4682+
((0, 9), 1766240503),
4683+
((0, 10), 3660871006),
4684+
((0, 11), 388409270),
4685+
((0, 12), 1948594622),
4686+
((0, 13), 3119396969),
4687+
((0, 14), 3924579183),
4688+
((0, 15), 2089920034),
4689+
((0, 16), 3857888532),
4690+
((0, 17), 929304360),
4691+
((0, 18), 1810891574),
4692+
((0, 19), 860971754),
4693+
((0, 20), 1822893775),
4694+
((0, 21), 2008495810),
4695+
((0, 22), 2958962335),
4696+
((0, 23), 2340515744),
4697+
// Counter
4698+
((0, 24), 64),
4699+
// AP
4700+
((1, 0), (0, 25)),
4701+
((2, 0), (0, 0)),
4702+
((2, 1), (0, 8))
4703+
];
4704+
let operands_addresses = OperandsAddresses {
4705+
dst_addr: (0, 24).into(),
4706+
op0_addr: (2, 0).into(),
4707+
op1_addr: (2, 1).into(),
4708+
};
4709+
vm.run_context = RunContext {
4710+
pc: (0, 0).into(),
4711+
ap: 0,
4712+
fp: 0,
4713+
};
4714+
assert_matches!(
4715+
vm.handle_blake2s_instruction(&operands_addresses, true),
4716+
Ok(())
4717+
);
4718+
4719+
let state: [u32; 8] = vm
4720+
.get_u32_range((0, 0).into(), 8)
4721+
.unwrap()
4722+
.try_into()
4723+
.unwrap();
4724+
let message: [u32; 16] = vm
4725+
.get_u32_range((0, 8).into(), 16)
4726+
.unwrap()
4727+
.try_into()
4728+
.unwrap();
4729+
let counter = vm.segments.memory.get_u32((0, 24).into()).unwrap();
4730+
4731+
let expected_new_state: [u32; 8] =
4732+
blake2s_compress(&state, &message, counter, 0, 0xffffffff, 0)
4733+
.try_into()
4734+
.unwrap();
4735+
4736+
let new_state: [u32; 8] = vm
4737+
.get_u32_range((0, 25).into(), 8)
4738+
.unwrap()
4739+
.try_into()
4740+
.unwrap();
4741+
assert_eq!(new_state, expected_new_state);
4742+
}
4743+
46504744
#[test]
46514745
#[cfg_attr(target_arch = "wasm32", wasm_bindgen_test)]
46524746
fn get_traceback_entries_bad_usort() {

0 commit comments

Comments
 (0)