Also avoid allocating zero len slice in ProveNextStep

OffchainLabs · Dec 12, 2024 · be3c622 · be3c622
1 parent 155fa7d
commit be3c622
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/validator/server_arb/machine.go b/validator/server_arb/machine.go
@@ -306,8 +306,11 @@ func (m *ArbitratorMachine) ProveNextStep() []byte {
 
 	output := &C.RustBytes{}
 	C.arbitrator_gen_proof(m.ptr, output)
+	defer C.free_rust_bytes(*output)
+	if output.len == 0 {
+		return nil
+	}
 	proofBytes := C.GoBytes(unsafe.Pointer(output.ptr), C.int(output.len))
-	C.free_rust_bytes(*output)
 
 	return proofBytes
 }