#include <ccan/endian/endian.h>
#include <common/hash_u5.h>
#include <string.h>

void hash_u5_init(struct hash_u5 *hu5, const char *hrp)
{
        hu5->buf = 0;
        hu5->num_bits = 0;
        sha256_init(&hu5->hash);
        sha256_update(&hu5->hash, hrp, strlen(hrp));
}

void hash_u5(struct hash_u5 *hu5, const u8 *u5, size_t len)
{
        size_t bits = len * 5;

        while (bits) {
                size_t n = 5;

                if (bits < n)
                        n = bits;

                hu5->buf <<= n;
                hu5->buf |= (*u5 >> (5-n));
                bits -= n;
                hu5->num_bits += n;

                if (n == 5)
                        u5++;

                if (hu5->num_bits >= 32) {
                        be32 be32 = cpu_to_be32(hu5->buf >> (hu5->num_bits-32));
                        sha256_update(&hu5->hash, &be32, sizeof(be32));
                        hu5->num_bits -= 32;
                }
        }
}

void hash_u5_done(struct hash_u5 *hu5, struct sha256 *res)
{
        if (hu5->num_bits) {
                be32 be32 = cpu_to_be32(hu5->buf << (32 - hu5->num_bits));

                sha256_update(&hu5->hash, &be32, (hu5->num_bits + 7) / 8);
        }
        sha256_done(&hu5->hash, res);
}