diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/main.cc | 6 | ||||
| -rw-r--r-- | src/sha256.cc | 96 |
2 files changed, 66 insertions, 36 deletions
diff --git a/src/main.cc b/src/main.cc index 9f01c14..a83afee 100644 --- a/src/main.cc +++ b/src/main.cc @@ -1,7 +1,9 @@ -#include<iostream> #include "sha256.h" +#include <iostream> -int main(int argc, char** argv) { +int +main(int argc, char** argv) +{ auto sha = hmacsha256::SHA256(argv[1]); sha.digest(); std::cout << sha.hexdigest(); diff --git a/src/sha256.cc b/src/sha256.cc index 7790f8a..4e03b46 100644 --- a/src/sha256.cc +++ b/src/sha256.cc @@ -1,13 +1,16 @@ #include "sha256.h" -#include<cstring> // used for memset -#include<sstream> // used for stringstream -#include<iomanip> // used for setfill and setw +#include <cstring> // used for memset +#include <iomanip> // used for setfill and setw +#include <sstream> // used for stringstream -hmacsha256::SHA256::SHA256(const uint8_t* data, uint32_t length) { +hmacsha256::SHA256::SHA256(const uint8_t* data, uint32_t length) +{ init(data, length); } -void hmacsha256::SHA256::init(const uint8_t* data, uint32_t length) { +void +hmacsha256::SHA256::init(const uint8_t* data, uint32_t length) +{ message_l_ = bit_len_ = 0; vars_[0] = 0x6a09e667; vars_[1] = 0xbb67ae85; @@ -18,9 +21,9 @@ void hmacsha256::SHA256::init(const uint8_t* data, uint32_t length) { vars_[6] = 0x1f83d9ab; vars_[7] = 0x5be0cd19; - for(size_t i = 0; i < length; ++i) { + for (size_t i = 0; i < length; ++i) { message_[message_l_++] = data[i]; - if(message_l_ == 64) { + if (message_l_ == 64) { transform(); bit_len_ += 512; @@ -29,78 +32,99 @@ void hmacsha256::SHA256::init(const uint8_t* data, uint32_t length) { } } -hmacsha256::SHA256::SHA256(const std::string& data) { - init(reinterpret_cast<const uint8_t*> (data.c_str()), data.size()); +hmacsha256::SHA256::SHA256(const std::string& data) +{ + init(reinterpret_cast<const uint8_t*>(data.c_str()), data.size()); } -uint8_t* hmacsha256::SHA256::digest() { +uint8_t* +hmacsha256::SHA256::digest() +{ uint8_t* hash = new uint8_t[32]; pad(); bigendian(hash); - for(uint32_t i = 0; i < 32; ++i) { + for (uint32_t i = 0; i < 32; ++i) { digest_[i] = hash[i]; } return hash; } -std::string hmacsha256::SHA256::hexdigest() { +std::string +hmacsha256::SHA256::hexdigest() +{ std::stringstream s; s << std::setfill('0') << std::hex; - for(uint8_t i = 0 ; i < 32 ; i++) { - s << std::setw(2) << (unsigned int) digest_[i]; + for (uint8_t i = 0; i < 32; i++) { + s << std::setw(2) << (unsigned int)digest_[i]; } return s.str(); } -uint32_t hmacsha256::SHA256::ch(uint32_t x, uint32_t y, uint32_t z) { +uint32_t +hmacsha256::SHA256::ch(uint32_t x, uint32_t y, uint32_t z) +{ return (x & y) ^ (~x & z); } -uint32_t hmacsha256::SHA256::maj(uint32_t x, uint32_t y, uint32_t z) { +uint32_t +hmacsha256::SHA256::maj(uint32_t x, uint32_t y, uint32_t z) +{ return (x & (y | z)) | (y & z); } -uint32_t hmacsha256::SHA256::rotr(uint32_t x, uint32_t n) { - return (x >> n) | (x << (32-n)); +uint32_t +hmacsha256::SHA256::rotr(uint32_t x, uint32_t n) +{ + return (x >> n) | (x << (32 - n)); } -uint32_t hmacsha256::SHA256::big_sigma0(uint32_t x) { +uint32_t +hmacsha256::SHA256::big_sigma0(uint32_t x) +{ return SHA256::rotr(x, 2) ^ SHA256::rotr(x, 13) ^ SHA256::rotr(x, 22); } -uint32_t hmacsha256::SHA256::big_sigma1(uint32_t x) { +uint32_t +hmacsha256::SHA256::big_sigma1(uint32_t x) +{ return SHA256::rotr(x, 6) ^ SHA256::rotr(x, 11) ^ SHA256::rotr(x, 25); } -uint32_t hmacsha256::SHA256::sigma0(uint32_t x) { +uint32_t +hmacsha256::SHA256::sigma0(uint32_t x) +{ return SHA256::rotr(x, 7) ^ SHA256::rotr(x, 18) ^ (x >> 3); } -uint32_t hmacsha256::SHA256::sigma1(uint32_t x) { +uint32_t +hmacsha256::SHA256::sigma1(uint32_t x) +{ return SHA256::rotr(x, 17) ^ SHA256::rotr(x, 19) ^ (x >> 10); } -void hmacsha256::SHA256::transform() { +void +hmacsha256::SHA256::transform() +{ uint32_t m[64]; uint32_t h[8]; - for(uint8_t i = 0, j = 0; i < 16; ++i, j += 4) { + for (uint8_t i = 0, j = 0; i < 16; ++i, j += 4) { m[i] = (message_[j] << 24) | (message_[j + 1] << 16) | (message_[j + 2] << 8) | (message_[j + 3]); } - for(uint8_t i = 16; i < 64; ++i) { + for (uint8_t i = 16; i < 64; ++i) { m[i] = SHA256::sigma1(m[i - 2]) + m[i - 7] + SHA256::sigma0(m[i - 15]) + m[i - 16]; } - for(uint8_t i = 0; i < 8; ++i) + for (uint8_t i = 0; i < 8; ++i) h[i] = vars_[i]; - for(uint8_t i = 0; i < 64; ++i) { + for (uint8_t i = 0; i < 64; ++i) { auto S1 = SHA256::big_sigma1(h[4]); auto S0 = SHA256::big_sigma0(h[0]); auto t1 = h[7] + S1 + SHA256::ch(h[4], h[5], h[6]) + K[i] + m[i]; @@ -115,13 +139,14 @@ void hmacsha256::SHA256::transform() { h[0] = t1 + t2; } - for(uint8_t i = 0; i < 8; ++i) { - vars_[i]+=h[i]; + for (uint8_t i = 0; i < 8; ++i) { + vars_[i] += h[i]; } - } -void hmacsha256::SHA256::pad() { +void +hmacsha256::SHA256::pad() +{ // start of scanning uint64_t i = message_l_; @@ -132,9 +157,10 @@ void hmacsha256::SHA256::pad() { message_[i++] = 0x80; // append '0' bit, j times - while(i < j) message_[i++] = 0x00; + while (i < j) + message_[i++] = 0x00; - if(message_l_ >= 56) { + if (message_l_ >= 56) { transform(); memset(message_, 0, 56); } @@ -151,7 +177,9 @@ void hmacsha256::SHA256::pad() { transform(); } -void hmacsha256::SHA256::bigendian(uint8_t* hash) { +void +hmacsha256::SHA256::bigendian(uint8_t* hash) +{ for (uint32_t i = 0; i < 4; ++i) { hash[i] = (vars_[0] >> (24 - i * 8)) & 0x000000ff; hash[i + 4] = (vars_[1] >> (24 - i * 8)) & 0x000000ff; |