Team2890/allwpilib
4
0
Fork 0
mirror of https://github.com/wpilibsuite/allwpilib synced 2026-06-19 00:41:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

wpiutil: SHA1: Add method to get raw bytes instead of hex

Browse Source
This commit is contained in:
Peter Johnson
2018-11-18 18:26:01 -08:00
parent 45f4472d42
commit 75cc3cda28
2 changed files with 22 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
wpiutil/src/main/native/cpp/sha1.cpp
View File

@@ -237,7 +237,7 @@ void SHA1::Update(raw_istream& is) {
*/
static void finalize(uint32_t digest[], unsigned char* buffer, size_t& buf_size,
uint64_t& transforms, raw_ostream& os) {
uint64_t& transforms, raw_ostream& os, bool hex) {
/* Total number of hashed bits */
uint64_t total_bits = (transforms * BLOCK_BYTES + buf_size) * 8;
@@ -266,9 +266,16 @@ static void finalize(uint32_t digest[], unsigned char* buffer, size_t& buf_size,
static const char* const LUT = "0123456789abcdef";
for (size_t i = 0; i < 5; i++) {
uint32_t v = digest[i];
os << LUT[(v >> 28) & 0xf] << LUT[(v >> 24) & 0xf] << LUT[(v >> 20) & 0xf]
<< LUT[(v >> 16) & 0xf] << LUT[(v >> 12) & 0xf] << LUT[(v >> 8) & 0xf]
<< LUT[(v >> 4) & 0xf] << LUT[(v >> 0) & 0xf];
if (hex) {
os << LUT[(v >> 28) & 0xf] << LUT[(v >> 24) & 0xf] << LUT[(v >> 20) & 0xf]
<< LUT[(v >> 16) & 0xf] << LUT[(v >> 12) & 0xf] << LUT[(v >> 8) & 0xf]
<< LUT[(v >> 4) & 0xf] << LUT[(v >> 0) & 0xf];
} else {
os.write(static_cast<unsigned char>((v >> 24) & 0xff));
os.write(static_cast<unsigned char>((v >> 16) & 0xff));
os.write(static_cast<unsigned char>((v >> 8) & 0xff));
os.write(static_cast<unsigned char>((v >> 0) & 0xff));
}
}
/* Reset for next run */
@@ -279,7 +286,7 @@ std::string SHA1::Final() {
std::string out;
raw_string_ostream os(out);
finalize(digest, buffer, buf_size, transforms, os);
finalize(digest, buffer, buf_size, transforms, os, true);
return os.str();
}
@@ -287,7 +294,15 @@ std::string SHA1::Final() {
StringRef SHA1::Final(SmallVectorImpl<char>& buf) {
raw_svector_ostream os(buf);
finalize(digest, buffer, buf_size, transforms, os);
finalize(digest, buffer, buf_size, transforms, os, true);
return os.str();
}
StringRef SHA1::RawFinal(SmallVectorImpl<char>& buf) {
raw_svector_ostream os(buf);
finalize(digest, buffer, buf_size, transforms, os, false);
return os.str();
}