-
Notifications
You must be signed in to change notification settings - Fork 131
Add SHA256 utility implementation #408
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Changes from 2 commits
de74159
8a73b71
471955f
3e3e1b3
ccb7488
1f54004
d868417
a9e987a
fa5aa5a
97aab35
File filter
Filter by extension
Conversations
Jump to
Diff view
Diff view
There are no files selected for viewing
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,96 @@ | ||
| // Copyright 2023 Open Source Robotics Foundation, Inc. | ||
| // | ||
| // Licensed under the Apache License, Version 2.0 (the "License"); | ||
| // you may not use this file except in compliance with the License. | ||
| // You may obtain a copy of the License at | ||
| // | ||
| // http://www.apache.org/licenses/LICENSE-2.0 | ||
| // | ||
| // Unless required by applicable law or agreed to in writing, software | ||
| // distributed under the License is distributed on an "AS IS" BASIS, | ||
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
| // See the License for the specific language governing permissions and | ||
| // limitations under the License. | ||
|
|
||
| /// \file Provides a simple SHA256 algorithm for hashing. | ||
| /// This implementation makes no security guarantees, its use case | ||
| /// is for non-sensitive comparison of message digests | ||
| /// Implementation originally copied from Brad Conte | ||
| /// https://github.com/B-Con/crypto-algorithms/blob/master/sha256.c | ||
|
|
||
| #ifndef RCUTILS__SHA256_H_ | ||
| #define RCUTILS__SHA256_H_ | ||
|
|
||
| #ifdef __cplusplus | ||
| extern "C" | ||
| { | ||
| #endif | ||
|
|
||
| #include "rcutils/macros.h" | ||
| #include "rcutils/types/rcutils_ret.h" | ||
| #include "rcutils/visibility_control.h" | ||
|
|
||
| #define RCUTILS_SHA256_BLOCK_SIZE 32 | ||
|
|
||
| typedef struct | ||
| { | ||
| uint8_t data[64]; | ||
| uint32_t datalen; | ||
| uint64_t bitlen; | ||
| uint32_t state[8]; | ||
| } rcutils_sha256_ctx_t; | ||
|
|
||
| /// Simple SHA256 implementation | ||
| /********************************************************************* | ||
| * Filename: sha256.c | ||
| * Author: Brad Conte (brad AT bradconte.com) | ||
| * Copyright: | ||
| * Disclaimer: This code is presented "as is" without any guarantees. | ||
| * Details: Implementation of the SHA-256 hashing algorithm. | ||
| SHA-256 is one of the three algorithms in the SHA2 | ||
| specification. The others, SHA-384 and SHA-512, are not | ||
| offered in this implementation. | ||
| Algorithm specification can be found here: | ||
| * http://csrc.nist.gov/publications/fips/fips180-2/fips180-2withchangenotice.pdf | ||
| This implementation uses little endian byte order. | ||
| *********************************************************************/ | ||
|
|
||
|
|
||
| /** | ||
| * | ||
| * | ||
| * \param[inout] ctx | ||
| * \return rcutils_ret_t | ||
| */ | ||
| RCUTILS_PUBLIC | ||
| RCUTILS_WARN_UNUSED | ||
| rcutils_ret_t rcutils_sha256_init(rcutils_sha256_ctx_t * ctx); | ||
|
|
||
| /** | ||
| * | ||
| * | ||
| * \param ctx | ||
| * \param data | ||
| * \param len | ||
| * \return rcutils_ret_t | ||
| */ | ||
| RCUTILS_PUBLIC | ||
| RCUTILS_WARN_UNUSED | ||
| rcutils_ret_t rcutils_sha256_update(rcutils_sha256_ctx_t * ctx, const uint8_t data[], size_t len); | ||
|
|
||
| /** | ||
| * | ||
| * | ||
| * \param ctx | ||
| * \param hash | ||
| * \return rcutils_ret_t | ||
| */ | ||
| RCUTILS_PUBLIC | ||
| RCUTILS_WARN_UNUSED | ||
| rcutils_ret_t sha256_final(rcutils_sha256_ctx_t * ctx, uint8_t hash[]); | ||
|
|
||
| #ifdef __cplusplus | ||
| } | ||
| #endif | ||
|
|
||
| #endif // RCUTILS__SHA256_H_ |
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,188 @@ | ||
| // Copyright 2023 Open Source Robotics Foundation, Inc. | ||
| // | ||
| // Licensed under the Apache License, Version 2.0 (the "License"); | ||
| // you may not use this file except in compliance with the License. | ||
| // You may obtain a copy of the License at | ||
| // | ||
| // http://www.apache.org/licenses/LICENSE-2.0 | ||
| // | ||
| // Unless required by applicable law or agreed to in writing, software | ||
| // distributed under the License is distributed on an "AS IS" BASIS, | ||
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
| // See the License for the specific language governing permissions and | ||
| // limitations under the License. | ||
|
|
||
| #include <string.h> | ||
|
|
||
| #include "rcutils/sha256.h" | ||
|
|
||
| static inline uint32_t rotright(uint32_t value, uint8_t bits) | ||
| { | ||
| return (value >> bits) | (value << (32 - bits)); | ||
| } | ||
|
|
||
| static inline uint32_t ch(uint32_t x, uint32_t y, uint32_t z) | ||
| { | ||
| return (x & y) ^ (~x & z); | ||
| } | ||
|
|
||
| static inline uint32_t maj(uint32_t x, uint32_t y, uint32_t z) | ||
| { | ||
| return (x & y) ^ (x & z) ^ (y & z); | ||
| } | ||
|
|
||
| static inline uint32_t ep0(uint32_t x) | ||
| { | ||
| return rotright(x, 2) ^ rotright(x, 13) ^ rotright(x, 22); | ||
| } | ||
|
|
||
| static inline uint32_t ep1(uint32_t x) | ||
| { | ||
| return rotright(x, 6) ^ rotright(x, 11) ^ rotright(x, 25); | ||
| } | ||
|
|
||
| static inline uint32_t sig0(uint32_t x) | ||
| { | ||
| return rotright(x, 7) ^ rotright(x, 18) ^ (x >> 3); | ||
| } | ||
|
|
||
| static inline uint32_t sig1(uint32_t x) | ||
| { | ||
| return rotright(x, 17) ^ rotright(x, 19) ^ (x >> 10); | ||
| } | ||
|
|
||
| static const uint32_t k[64] = { | ||
| 0x428a2f98, 0x71374491, 0xb5c0fbcf, 0xe9b5dba5, 0x3956c25b, 0x59f111f1, 0x923f82a4, 0xab1c5ed5, | ||
| 0xd807aa98, 0x12835b01, 0x243185be, 0x550c7dc3, 0x72be5d74, 0x80deb1fe, 0x9bdc06a7, 0xc19bf174, | ||
| 0xe49b69c1, 0xefbe4786, 0x0fc19dc6, 0x240ca1cc, 0x2de92c6f, 0x4a7484aa, 0x5cb0a9dc, 0x76f988da, | ||
| 0x983e5152, 0xa831c66d, 0xb00327c8, 0xbf597fc7, 0xc6e00bf3, 0xd5a79147, 0x06ca6351, 0x14292967, | ||
| 0x27b70a85, 0x2e1b2138, 0x4d2c6dfc, 0x53380d13, 0x650a7354, 0x766a0abb, 0x81c2c92e, 0x92722c85, | ||
| 0xa2bfe8a1, 0xa81a664b, 0xc24b8b70, 0xc76c51a3, 0xd192e819, 0xd6990624, 0xf40e3585, 0x106aa070, | ||
| 0x19a4c116, 0x1e376c08, 0x2748774c, 0x34b0bcb5, 0x391c0cb3, 0x4ed8aa4a, 0x5b9cca4f, 0x682e6ff3, | ||
| 0x748f82ee, 0x78a5636f, 0x84c87814, 0x8cc70208, 0x90befffa, 0xa4506ceb, 0xbef9a3f7, 0xc67178f2 | ||
| }; | ||
|
|
||
| void sha256_transform(rcutils_sha256_ctx_t * ctx, const uint8_t data[]) | ||
| { | ||
| uint32_t a, b, c, d, e, f, g, h, i, j, t1, t2, m[64]; | ||
|
|
||
| for (i = 0, j = 0; i < 16; ++i, j += 4) { | ||
| m[i] = (data[j] << 24) | (data[j + 1] << 16) | (data[j + 2] << 8) | (data[j + 3]); | ||
| } | ||
| for ( ; i < 64; ++i) { | ||
| m[i] = sig1(m[i - 2]) + m[i - 7] + sig0(m[i - 15]) + m[i - 16]; | ||
| } | ||
|
|
||
| a = ctx->state[0]; | ||
| b = ctx->state[1]; | ||
| c = ctx->state[2]; | ||
| d = ctx->state[3]; | ||
| e = ctx->state[4]; | ||
| f = ctx->state[5]; | ||
| g = ctx->state[6]; | ||
| h = ctx->state[7]; | ||
|
|
||
| for (i = 0; i < 64; ++i) { | ||
| t1 = h + ep1(e) + ch(e, f, g) + k[i] + m[i]; | ||
| t2 = ep0(a) + maj(a, b, c); | ||
| h = g; | ||
| g = f; | ||
| f = e; | ||
| e = d + t1; | ||
| d = c; | ||
| c = b; | ||
| b = a; | ||
| a = t1 + t2; | ||
| } | ||
|
|
||
| ctx->state[0] += a; | ||
| ctx->state[1] += b; | ||
| ctx->state[2] += c; | ||
| ctx->state[3] += d; | ||
| ctx->state[4] += e; | ||
| ctx->state[5] += f; | ||
| ctx->state[6] += g; | ||
| ctx->state[7] += h; | ||
| } | ||
|
|
||
| rcutils_ret_t rcutils_sha256_init(rcutils_sha256_ctx_t * ctx) | ||
| { | ||
| ctx->datalen = 0; | ||
| ctx->bitlen = 0; | ||
| ctx->state[0] = 0x6a09e667; | ||
| ctx->state[1] = 0xbb67ae85; | ||
| ctx->state[2] = 0x3c6ef372; | ||
| ctx->state[3] = 0xa54ff53a; | ||
| ctx->state[4] = 0x510e527f; | ||
| ctx->state[5] = 0x9b05688c; | ||
| ctx->state[6] = 0x1f83d9ab; | ||
| ctx->state[7] = 0x5be0cd19; | ||
|
|
||
| return RCUTILS_RET_OK; | ||
| } | ||
|
|
||
| rcutils_ret_t rcutils_sha256_update(rcutils_sha256_ctx_t * ctx, const uint8_t data[], size_t len) | ||
| { | ||
| uint32_t i; | ||
|
|
||
| for (i = 0; i < len; ++i) { | ||
| ctx->data[ctx->datalen] = data[i]; | ||
| ctx->datalen++; | ||
| if (ctx->datalen == 64) { | ||
| sha256_transform(ctx, ctx->data); | ||
| ctx->bitlen += 512; | ||
| ctx->datalen = 0; | ||
| } | ||
| } | ||
|
Contributor
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. I'm not sure it is worth it, but we could probably optimize this. In particular, we could I'll leave it up to you whether you think that improvement is worth it and you want to try it. This is not a blocking comment from me.
Contributor
Author
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. I gave block-copy implementation a go. The tests still pass so it seems all right. I don't have a performance comparison but maybe it looks better this way. |
||
|
|
||
| return RCUTILS_RET_OK; | ||
| } | ||
|
|
||
| rcutils_ret_t rcutils_sha256_final(rcutils_sha256_ctx_t * ctx, uint8_t hash[]) | ||
| { | ||
| uint32_t i; | ||
|
|
||
| i = ctx->datalen; | ||
|
|
||
| // Pad whatever data is left in the buffer. | ||
| if (ctx->datalen < 56) { | ||
| ctx->data[i++] = 0x80; | ||
| while (i < 56) { | ||
| ctx->data[i++] = 0x00; | ||
| } | ||
|
emersonknapp marked this conversation as resolved.
Outdated
|
||
| } else { | ||
| ctx->data[i++] = 0x80; | ||
| while (i < 64) { | ||
| ctx->data[i++] = 0x00; | ||
| } | ||
|
emersonknapp marked this conversation as resolved.
Outdated
|
||
| sha256_transform(ctx, ctx->data); | ||
| memset(ctx->data, 0, 56); | ||
| } | ||
|
|
||
| // Append to the padding the total message's length in bits and transform. | ||
| ctx->bitlen += ctx->datalen * 8; | ||
| ctx->data[63] = (uint8_t)(ctx->bitlen); | ||
| ctx->data[62] = (uint8_t)(ctx->bitlen >> 8); | ||
| ctx->data[61] = (uint8_t)(ctx->bitlen >> 16); | ||
| ctx->data[60] = (uint8_t)(ctx->bitlen >> 24); | ||
| ctx->data[59] = (uint8_t)(ctx->bitlen >> 32); | ||
| ctx->data[58] = (uint8_t)(ctx->bitlen >> 40); | ||
| ctx->data[57] = (uint8_t)(ctx->bitlen >> 48); | ||
| ctx->data[56] = (uint8_t)(ctx->bitlen >> 56); | ||
| sha256_transform(ctx, ctx->data); | ||
|
|
||
| // Since this implementation uses little endian byte ordering and SHA uses big endian, | ||
| // reverse all the bytes when copying the final state to the output hash. | ||
| for (i = 0; i < 4; ++i) { | ||
| hash[i + 0] = (ctx->state[0] >> (24 - i * 8)) & 0x000000ff; | ||
| hash[i + 4] = (ctx->state[1] >> (24 - i * 8)) & 0x000000ff; | ||
| hash[i + 8] = (ctx->state[2] >> (24 - i * 8)) & 0x000000ff; | ||
| hash[i + 12] = (ctx->state[3] >> (24 - i * 8)) & 0x000000ff; | ||
| hash[i + 16] = (ctx->state[4] >> (24 - i * 8)) & 0x000000ff; | ||
| hash[i + 20] = (ctx->state[5] >> (24 - i * 8)) & 0x000000ff; | ||
| hash[i + 24] = (ctx->state[6] >> (24 - i * 8)) & 0x000000ff; | ||
| hash[i + 28] = (ctx->state[7] >> (24 - i * 8)) & 0x000000ff; | ||
| } | ||
|
|
||
| return RCUTILS_RET_OK; | ||
| } | ||
Uh oh!
There was an error while loading. Please reload this page.