From 06b80d8cde946b4df418153fb501f0adcd29cf82 Mon Sep 17 00:00:00 2001 From: Bazsalanszky Date: Tue, 6 Dec 2022 20:34:54 +0100 Subject: [PATCH] Updated zkp scheme --- CCTF_Solutions_main/hash.zok | 5 ++--- CCTF_Solutions_main/root.zok | 3 +-- 2 files changed, 3 insertions(+), 5 deletions(-) diff --git a/CCTF_Solutions_main/hash.zok b/CCTF_Solutions_main/hash.zok index d80a46e..ee842b5 100644 --- a/CCTF_Solutions_main/hash.zok +++ b/CCTF_Solutions_main/hash.zok @@ -1,11 +1,10 @@ import "hashes/sha256/sha256Padded.zok" as sha256; from "utils/casts.zok" import cast; -def main(public u32[5] address,private u8[64] flag) { +def main(public u32[5] address,private u8[64] flag) -> u32[8] { u8[20] addr8 = cast(address); - u8[84] forHash = [...addr8,...flag]; u32[8] genHash = sha256(flag); log("Hash: {} {} {} {} {} {} {} {}",genHash[0],genHash[1],genHash[2],genHash[3],genHash[4],genHash[5],genHash[6],genHash[7]); //assert(genHash == hash); - return; + return genHash; } diff --git a/CCTF_Solutions_main/root.zok b/CCTF_Solutions_main/root.zok index 43ac1a2..e1a4e35 100644 --- a/CCTF_Solutions_main/root.zok +++ b/CCTF_Solutions_main/root.zok @@ -3,9 +3,8 @@ from "utils/casts.zok" import cast; def main(public u32[8] hash,public u32[5] address,private u8[64] flag) { u8[20] addr8 = cast(address); - u8[84] forHash = [...addr8,...flag]; u32[8] genHash = sha256(flag); log("Hash: {} {} {} {} {} {} {} {}",genHash[0],genHash[1],genHash[2],genHash[3],genHash[4],genHash[5],genHash[6],genHash[7]); assert(genHash == hash); - return; + return sha256([...addr8,...sha256(flag)]); } \ No newline at end of file