56 lines
2.6 KiB
OCaml
56 lines
2.6 KiB
OCaml
|
|
open Blake2
|
|
|
|
type vector = {
|
|
data_in : Bigstring.t list ;
|
|
data_key : Bigstring.t option ;
|
|
data_out : Bigstring.t ;
|
|
}
|
|
|
|
let hex s =
|
|
Cstruct.to_bigarray (Hex.to_cstruct (`Hex s))
|
|
|
|
let vectors = [
|
|
{ data_in = [ hex "000102030405060708090a0b0c0d0e0f101112131415161718191a1b1c1d1e1f202122232425262728292a2b2c2d2e2f303132333435363738393a3b3c3d3e3f404142434445464748494a4b4c4d4e4f505152535455565758595a5b5c5d5e5f" ] ;
|
|
data_key = None ;
|
|
data_out = hex "1c077e279de6548523502b6df800ffdab5e2c3e9442eb838f58c295f3b147cef9d701c41c321283f00c71affa0619310399126295b78dd4d1a74572ef9ed5135" ;
|
|
} ;
|
|
{ data_in = [ hex "000102030405060708090a0b0c0d0e0f101112131415" ; hex "161718"; hex "191a1b1c1d1e1f202122232425262728292a2b2c2d2e2f303132333435363738393a3b3c3d3e3f404142434445464748494a4b4c4d4e4f505152535455565758595a5b5c5d5e5f" ] ;
|
|
data_key = None ;
|
|
data_out = hex "1c077e279de6548523502b6df800ffdab5e2c3e9442eb838f58c295f3b147cef9d701c41c321283f00c71affa0619310399126295b78dd4d1a74572ef9ed5135" ;
|
|
} ;
|
|
{ data_in = [ hex "000102030405060708090a0b0c0d0e0f101112131415161718191a1b1c1d1e1f202122232425262728292a2b2c2d2e2f303132333435363738393a3b3c3d3e3f404142434445464748494a4b4c4d4e4f505152535455565758595a5b5c5d5e5f606162636465666768696a6b6c6d6e6f707172737475767778797a7b7c7d7e7f808182838485868788898a8b8c8d8e8f909192939495969798999a9b9c9d9e9fa0a1a2a3a4a5a6a7a8a9aaabacadaeafb0b1b2b3b4b5b6b7b8b9babbbcbdbebfc0c1c2c3c4c5c6c7c8c9cacbcccdcecfd0d1d2d3d4d5d6d7d8d9dadbdcdddedfe0e1e2e3e4e5e6e7e8e9eaebecedeeeff0f1f2f3" ] ;
|
|
data_key = Some (hex "000102030405060708090a0b0c0d0e0f101112131415161718191a1b1c1d1e1f202122232425262728292a2b2c2d2e2f303132333435363738393a3b3c3d3e3f") ;
|
|
data_out = hex "b39614268fdd8781515e2cfebf89b4d5402bab10c226e6344e6b9ae000fb0d6c79cb2f3ec80e80eaeb1980d2f8698916bd2e9f747236655116649cd3ca23a837" ;
|
|
} ;
|
|
]
|
|
|
|
let test_update { data_in ; data_key ; data_out } =
|
|
let d = Blake2b.init ?key:data_key (Bigstring.length data_out) in
|
|
List.iter (Blake2b.update d) data_in ;
|
|
let Blake2b.Hash h = Blake2b.final d in
|
|
assert Bigstring.(equal data_out h)
|
|
|
|
let test_direct { data_in ; data_key ; data_out } =
|
|
let Blake2b.Hash h =
|
|
Blake2b.direct ?key:data_key
|
|
(Bigstring.concat "" data_in) (Bigstring.length data_out) in
|
|
assert Bigstring.(equal data_out h)
|
|
|
|
let update_tests =
|
|
List.mapi
|
|
(fun i v -> string_of_int i, `Quick, fun () -> test_update v)
|
|
vectors
|
|
|
|
let direct_tests =
|
|
List.mapi
|
|
(fun i v -> string_of_int i, `Quick, fun () -> test_direct v)
|
|
vectors
|
|
|
|
let () =
|
|
Alcotest.run "blake2b" [
|
|
"update", update_tests ;
|
|
"direct", direct_tests ;
|
|
]
|
|
|