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) ...@@ -36,15 +36,10 @@ void sha256_init(struct sha256_ctx *ctx)
SHA256_Init(&ctx->c); SHA256_Init(&ctx->c);
} }
void sha256_update_arr(struct sha256_ctx *ctx, const void *p, void sha256_update_bytes(struct sha256_ctx *ctx, const void *p, size_t size)
size_t num, size_t size)
{ {
size_t total = num * size;
/* Don't overflow. */
assert(size == 0 || total / size == num);
check_sha256(ctx); 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) void sha256_done(struct sha256_ctx *ctx, struct sha256 *res)
...@@ -221,15 +216,10 @@ void sha256_init(struct sha256_ctx *ctx) ...@@ -221,15 +216,10 @@ void sha256_init(struct sha256_ctx *ctx)
*ctx = init; *ctx = init;
} }
void sha256_update_arr(struct sha256_ctx *ctx, const void *p, void sha256_update(struct sha256_ctx *ctx, const void *p, size_t size)
size_t num, size_t size)
{ {
size_t total = num * size;
/* Don't overflow. */
assert(size == 0 || total / size == num);
check_sha256(ctx); check_sha256(ctx);
add(ctx, p, total); add(ctx, p, size);
} }
void sha256_done(struct sha256_ctx *ctx, struct sha256 *res) void sha256_done(struct sha256_ctx *ctx, struct sha256 *res)
...@@ -249,69 +239,69 @@ 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 #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; struct sha256_ctx ctx;
sha256_init(&ctx); sha256_init(&ctx);
sha256_update_arr(&ctx, p, num, size); sha256_update(&ctx, p, size);
sha256_done(&ctx, sha); sha256_done(&ctx, sha);
} }
void sha256_u8(struct sha256_ctx *ctx, uint8_t v) 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) 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) 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) 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 */ /* Add as little-endian */
void sha256_le16(struct sha256_ctx *ctx, uint16_t v) void sha256_le16(struct sha256_ctx *ctx, uint16_t v)
{ {
leint16_t lev = cpu_to_le16(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) void sha256_le32(struct sha256_ctx *ctx, uint32_t v)
{ {
leint32_t lev = cpu_to_le32(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) void sha256_le64(struct sha256_ctx *ctx, uint64_t v)
{ {
leint64_t lev = cpu_to_le64(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 */ /* Add as big-endian */
void sha256_be16(struct sha256_ctx *ctx, uint16_t v) void sha256_be16(struct sha256_ctx *ctx, uint16_t v)
{ {
beint16_t bev = cpu_to_be16(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) void sha256_be32(struct sha256_ctx *ctx, uint32_t v)
{ {
beint32_t bev = cpu_to_be32(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) void sha256_be64(struct sha256_ctx *ctx, uint64_t v)
{ {
beint64_t bev = cpu_to_be64(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 { ...@@ -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 * @sha256: the sha256 to fill in
* @p: array or pointer to first element * @p: pointer to memory,
* @num: the number of elements to hash * @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(). * 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 * struct sha256_ctx - structure to store running context for sha256
...@@ -110,16 +110,15 @@ void sha256_init(struct sha256_ctx *ctx); ...@@ -110,16 +110,15 @@ void sha256_init(struct sha256_ctx *ctx);
#endif #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 * @ctx: the sha256_ctx to use
* @p: array or pointer to first element * @p: pointer to memory,
* @num: the number of elements to hash * @size: the number of bytes pointed to by @p
* *
* You can call this multiple times to hash more data, before calling * You can call this multiple times to hash more data, before calling
* sha256_done(). * sha256_done().
*/ */
#define sha256_update(ctx, p, num) \ void sha256_update(struct sha256_ctx *ctx, const void *p, size_t size);
sha256_update_arr((ctx), (p), (num), sizeof(*(p)))
/** /**
* sha256_done - finish SHA256 and return the hash * sha256_done - finish SHA256 and return the hash
...@@ -146,8 +145,4 @@ void sha256_le64(struct sha256_ctx *ctx, uint64_t v); ...@@ -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_be16(struct sha256_ctx *ctx, uint16_t v);
void sha256_be32(struct sha256_ctx *ctx, uint32_t v); void sha256_be32(struct sha256_ctx *ctx, uint32_t v);
void sha256_be64(struct sha256_ctx *ctx, uint64_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 */ #endif /* CCAN_CRYPTO_SHA256_H */
...@@ -33,7 +33,7 @@ static void derive(shachain_index_t from, shachain_index_t to, ...@@ -33,7 +33,7 @@ static void derive(shachain_index_t from, shachain_index_t to,
for (i = ilog64(branches) - 1; i >= 0; i--) { for (i = ilog64(branches) - 1; i >= 0; i--) {
if (((branches >> i) & 1)) { if (((branches >> i) & 1)) {
change_bit(hash->u.u8, i); 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