aboutsummaryrefslogtreecommitdiff
path: root/include
diff options
context:
space:
mode:
Diffstat (limited to 'include')
-rw-r--r--include/salticidae/crypto.h42
-rw-r--r--include/salticidae/msg.h14
2 files changed, 49 insertions, 7 deletions
diff --git a/include/salticidae/crypto.h b/include/salticidae/crypto.h
index 79e9377..772cce1 100644
--- a/include/salticidae/crypto.h
+++ b/include/salticidae/crypto.h
@@ -72,6 +72,48 @@ class SHA256 {
}
};
+class SHA1 {
+ SHA_CTX ctx;
+
+ public:
+ SHA1() { reset(); }
+
+ void reset() {
+ if (!SHA1_Init(&ctx))
+ throw std::runtime_error("openssl SHA1 init error");
+ }
+
+ template<typename T>
+ void update(const T &data) {
+ update(reinterpret_cast<const uint8_t *>(&*data.begin()), data.size());
+ }
+
+ void update(const bytearray_t::const_iterator &it, size_t length) {
+ update(&*it, length);
+ }
+
+ void update(const uint8_t *ptr, size_t length) {
+ if (!SHA1_Update(&ctx, ptr, length))
+ throw std::runtime_error("openssl SHA1 update error");
+ }
+
+ void _digest(bytearray_t &md) {
+ if (!SHA1_Final(&*md.begin(), &ctx))
+ throw std::runtime_error("openssl SHA1 error");
+ }
+
+ void digest(bytearray_t &md) {
+ md.resize(32);
+ _digest(md);
+ }
+
+ bytearray_t digest() {
+ bytearray_t md(32);
+ _digest(md);
+ return std::move(md);
+ }
+};
+
}
#endif
diff --git a/include/salticidae/msg.h b/include/salticidae/msg.h
index 42a9bc2..03eb6dd 100644
--- a/include/salticidae/msg.h
+++ b/include/salticidae/msg.h
@@ -192,19 +192,19 @@ class MsgBase {
#ifndef SALTICIDAE_NOCHECKSUM
uint32_t get_checksum() const {
- static thread_local class SHA256 sha256;
+ static thread_local class SHA1 sha1;
uint32_t res;
bytearray_t tmp;
#ifndef SALTICIDAE_NOCHECK
if (no_payload)
throw std::runtime_error("payload not available");
#endif
- sha256.reset();
- sha256.update(payload);
- sha256.digest(tmp);
- sha256.reset();
- sha256.update(tmp);
- sha256.digest(tmp);
+ sha1.reset();
+ sha1.update(payload);
+ sha1.digest(tmp);
+ //sha256.reset();
+ //sha256.update(tmp);
+ //sha256.digest(tmp);
memmove(&res, &*tmp.begin(), 4);
return res;
}