1 #include <ccan/endian/endian.h>
2 #include <common/hash_u5.h>
3 #include <string.h>
4
hash_u5_init(struct hash_u5 * hu5,const char * hrp)5 void hash_u5_init(struct hash_u5 *hu5, const char *hrp)
6 {
7 hu5->buf = 0;
8 hu5->num_bits = 0;
9 sha256_init(&hu5->hash);
10 sha256_update(&hu5->hash, hrp, strlen(hrp));
11 }
12
hash_u5(struct hash_u5 * hu5,const u8 * u5,size_t len)13 void hash_u5(struct hash_u5 *hu5, const u8 *u5, size_t len)
14 {
15 size_t bits = len * 5;
16
17 while (bits) {
18 size_t n = 5;
19
20 if (bits < n)
21 n = bits;
22
23 hu5->buf <<= n;
24 hu5->buf |= (*u5 >> (5-n));
25 bits -= n;
26 hu5->num_bits += n;
27
28 if (n == 5)
29 u5++;
30
31 if (hu5->num_bits >= 32) {
32 be32 be32 = cpu_to_be32(hu5->buf >> (hu5->num_bits-32));
33 sha256_update(&hu5->hash, &be32, sizeof(be32));
34 hu5->num_bits -= 32;
35 }
36 }
37 }
38
hash_u5_done(struct hash_u5 * hu5,struct sha256 * res)39 void hash_u5_done(struct hash_u5 *hu5, struct sha256 *res)
40 {
41 if (hu5->num_bits) {
42 be32 be32 = cpu_to_be32(hu5->buf << (32 - hu5->num_bits));
43
44 sha256_update(&hu5->hash, &be32, (hu5->num_bits + 7) / 8);
45 }
46 sha256_done(&hu5->hash, res);
47 }
48