vieter/src/util.v

34 lines
824 B
Coq
Raw Normal View History

2022-01-18 21:41:09 +01:00
module util
import os
import crypto.md5
import crypto.sha256
// hash_file returns the md5 & sha256 hash of a given file
2022-01-18 22:52:14 +01:00
// TODO actually implement sha256
2022-01-18 21:41:09 +01:00
pub fn hash_file(path &string) ?(string, string) {
file := os.open(path) or { return error('Failed to open file.') }
mut md5sum := md5.new()
mut sha256sum := sha256.new()
2022-01-18 22:52:14 +01:00
buf_size := int(1_000_000)
2022-01-18 21:41:09 +01:00
mut buf := []byte{len: buf_size}
mut bytes_left := os.file_size(path)
for bytes_left > 0 {
// TODO check if just breaking here is safe
bytes_read := file.read(mut buf) or { break }
bytes_left -= u64(bytes_read)
if bytes_left > buf_size {
2022-01-18 22:52:14 +01:00
// For now we'll assume that this always works
md5sum.write(buf) or {}
// sha256sum.write(buf) or {}
2022-01-18 21:41:09 +01:00
}
}
2022-01-18 22:52:14 +01:00
// return md5sum.sum(buf).hex(), sha256sum.sum(buf).hex()
return md5sum.sum(buf).hex(), ''
2022-01-18 21:41:09 +01:00
}