pub use blake2b_rs::{Blake2b, Blake2bBuilder};
use includedir_codegen::Compression;
use std::{
env,
fs::File,
io::{BufWriter, Read, Write},
path::Path,
};
const PATH_PREFIX: &str = "build/";
const BUF_SIZE: usize = 8 * 1024;
const CKB_HASH_PERSONALIZATION: &[u8] = b"ckb-default-hash";
const BINARIES: &[(&str, &str)] = &[
(
"generator",
"e004500bd74bf0ecb8c5f69e392cc65409e9c8c427bc327110a83e614e892546",
),
(
"validator",
"6a946971979c019fe5096108267779775a141c9647936053b58358caa87bf5a2",
),
];
fn main() {
let mut bundled = includedir_codegen::start("BUNDLED_CELL");
let out_path = Path::new(&env::var("OUT_DIR").unwrap()).join("code_hashes.rs");
let mut out_file = BufWriter::new(File::create(&out_path).expect("create code_hashes.rs"));
let mut errors = Vec::new();
for (name, expected_hash) in BINARIES {
let path = format!("{}{}", PATH_PREFIX, name);
let mut buf = [0u8; BUF_SIZE];
bundled
.add_file(&path, Compression::Gzip)
.expect("add files to resource bundle");
let mut blake2b = new_blake2b();
let mut fd = File::open(&path).expect("open file");
loop {
let read_bytes = fd.read(&mut buf).expect("read file");
if read_bytes > 0 {
blake2b.update(&buf[..read_bytes]);
} else {
break;
}
}
let mut hash = [0u8; 32];
blake2b.finalize(&mut hash);
let actual_hash = faster_hex::hex_string(&hash).unwrap();
if expected_hash != &actual_hash {
errors.push((name, expected_hash, actual_hash));
continue;
}
write!(
&mut out_file,
"pub const {}: [u8; 32] = {:?};\n",
format!("CODE_HASH_{}", name.to_uppercase()),
hash
)
.expect("write to code_hashes.rs");
}
if !errors.is_empty() {
for (name, expected, actual) in errors.into_iter() {
eprintln!("{}: expect {}, actual {}", name, expected, actual);
}
panic!("not all hashes are right");
}
bundled.build("bundled.rs").expect("build resource bundle");
}
pub fn new_blake2b() -> Blake2b {
Blake2bBuilder::new(32)
.personal(CKB_HASH_PERSONALIZATION)
.build()
}