Commit f989a43b authored by Rusty Russell's avatar Rusty Russell

crypto/sha256: simplify interface.

I misused it by accident: the implied sizeof() is too tricky.
Signed-off-by: default avatarRusty Russell <rusty@rustcorp.com.au>
parent 75d313a8
......@@ -36,15 +36,10 @@ void sha256_init(struct sha256_ctx *ctx)
SHA256_Init(&ctx->c);
}
void sha256_update_arr(struct sha256_ctx *ctx, const void *p,
size_t num, size_t size)
void sha256_update_bytes(struct sha256_ctx *ctx, const void *p, size_t size)
{
size_t total = num * size;
/* Don't overflow. */
assert(size == 0 || total / size == num);
check_sha256(ctx);
SHA256_Update(&ctx->c, p, total);
SHA256_Update(&ctx->c, p, size);
}
void sha256_done(struct sha256_ctx *ctx, struct sha256 *res)
......@@ -221,15 +216,10 @@ void sha256_init(struct sha256_ctx *ctx)
*ctx = init;
}
void sha256_update_arr(struct sha256_ctx *ctx, const void *p,
size_t num, size_t size)
void sha256_update(struct sha256_ctx *ctx, const void *p, size_t size)
{
size_t total = num * size;
/* Don't overflow. */
assert(size == 0 || total / size == num);
check_sha256(ctx);
add(ctx, p, total);
add(ctx, p, size);
}
void sha256_done(struct sha256_ctx *ctx, struct sha256 *res)
......@@ -249,69 +239,69 @@ void sha256_done(struct sha256_ctx *ctx, struct sha256 *res)
}
#endif
void sha256_arr(struct sha256 *sha, const void *p, size_t num, size_t size)
void sha256(struct sha256 *sha, const void *p, size_t size)
{
struct sha256_ctx ctx;
sha256_init(&ctx);
sha256_update_arr(&ctx, p, num, size);
sha256_update(&ctx, p, size);
sha256_done(&ctx, sha);
}
void sha256_u8(struct sha256_ctx *ctx, uint8_t v)
{
sha256_update_arr(ctx, &v, sizeof(v), 1);
sha256_update(ctx, &v, sizeof(v));
}
void sha256_u16(struct sha256_ctx *ctx, uint16_t v)
{
sha256_update_arr(ctx, &v, sizeof(v), 1);
sha256_update(ctx, &v, sizeof(v));
}
void sha256_u32(struct sha256_ctx *ctx, uint32_t v)
{
sha256_update_arr(ctx, &v, sizeof(v), 1);
sha256_update(ctx, &v, sizeof(v));
}
void sha256_u64(struct sha256_ctx *ctx, uint64_t v)
{
sha256_update_arr(ctx, &v, sizeof(v), 1);
sha256_update(ctx, &v, sizeof(v));
}
/* Add as little-endian */
void sha256_le16(struct sha256_ctx *ctx, uint16_t v)
{
leint16_t lev = cpu_to_le16(v);
sha256_update_arr(ctx, &lev, sizeof(lev), 1);
sha256_update(ctx, &lev, sizeof(lev));
}
void sha256_le32(struct sha256_ctx *ctx, uint32_t v)
{
leint32_t lev = cpu_to_le32(v);
sha256_update_arr(ctx, &lev, sizeof(lev), 1);
sha256_update(ctx, &lev, sizeof(lev));
}
void sha256_le64(struct sha256_ctx *ctx, uint64_t v)
{
leint64_t lev = cpu_to_le64(v);
sha256_update_arr(ctx, &lev, sizeof(lev), 1);
sha256_update(ctx, &lev, sizeof(lev));
}
/* Add as big-endian */
void sha256_be16(struct sha256_ctx *ctx, uint16_t v)
{
beint16_t bev = cpu_to_be16(v);
sha256_update_arr(ctx, &bev, sizeof(bev), 1);
sha256_update(ctx, &bev, sizeof(bev));
}
void sha256_be32(struct sha256_ctx *ctx, uint32_t v)
{
beint32_t bev = cpu_to_be32(v);
sha256_update_arr(ctx, &bev, sizeof(bev), 1);
sha256_update(ctx, &bev, sizeof(bev));
}
void sha256_be64(struct sha256_ctx *ctx, uint64_t v)
{
beint64_t bev = cpu_to_be64(v);
sha256_update_arr(ctx, &bev, sizeof(bev), 1);
sha256_update(ctx, &bev, sizeof(bev));
}
......@@ -29,15 +29,15 @@ struct sha256 {
};
/**
* sha256 - return sha256 of an array of bytes.
* sha256 - return sha256 of an object.
* @sha256: the sha256 to fill in
* @p: array or pointer to first element
* @num: the number of elements to hash
* @p: pointer to memory,
* @size: the number of bytes pointed to by @p
*
* The bytes pointed to by @p is SHA256 hashes into @sha256. This is
* The bytes pointed to by @p is SHA256 hashed into @sha256. This is
* equivalent to sha256_init(), sha256_update() then sha256_done().
*/
#define sha256(sha256, p, num) sha256_arr((sha256), (p), (num), sizeof(*(p)))
void sha256(struct sha256 *sha, const void *p, size_t size);
/**
* struct sha256_ctx - structure to store running context for sha256
......@@ -110,16 +110,15 @@ void sha256_init(struct sha256_ctx *ctx);
#endif
/**
* sha256_update - include an array of data in the hash.
* sha256_update - include some memory in the hash.
* @ctx: the sha256_ctx to use
* @p: array or pointer to first element
* @num: the number of elements to hash
* @p: pointer to memory,
* @size: the number of bytes pointed to by @p
*
* You can call this multiple times to hash more data, before calling
* sha256_done().
*/
#define sha256_update(ctx, p, num) \
sha256_update_arr((ctx), (p), (num), sizeof(*(p)))
void sha256_update(struct sha256_ctx *ctx, const void *p, size_t size);
/**
* sha256_done - finish SHA256 and return the hash
......@@ -146,8 +145,4 @@ void sha256_le64(struct sha256_ctx *ctx, uint64_t v);
void sha256_be16(struct sha256_ctx *ctx, uint16_t v);
void sha256_be32(struct sha256_ctx *ctx, uint32_t v);
void sha256_be64(struct sha256_ctx *ctx, uint64_t v);
void sha256_update_arr(struct sha256_ctx *ctx, const void *p,
size_t num, size_t size);
void sha256_arr(struct sha256 *sha, const void *p, size_t num, size_t size);
#endif /* CCAN_CRYPTO_SHA256_H */
......@@ -33,7 +33,7 @@ static void derive(shachain_index_t from, shachain_index_t to,
for (i = ilog64(branches) - 1; i >= 0; i--) {
if (((branches >> i) & 1)) {
change_bit(hash->u.u8, i);
sha256(hash, hash, 1);
sha256(hash, hash, sizeof(*hash));
}
}
}
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment