diff options
Diffstat (limited to 'src/modules')
| -rw-r--r-- | src/modules/m_hash.h | 4 | ||||
| -rw-r--r-- | src/modules/m_md5.cpp | 6 | ||||
| -rw-r--r-- | src/modules/m_sha256.cpp | 2 |
3 files changed, 6 insertions, 6 deletions
diff --git a/src/modules/m_hash.h b/src/modules/m_hash.h index 6351cf04f..5686169dd 100644 --- a/src/modules/m_hash.h +++ b/src/modules/m_hash.h @@ -54,9 +54,9 @@ class HashRequest : public Request } /** Get data to be hashed */ - const char* GetHashData() + std::string& GetHashData() { - return tohash.c_str(); + return tohash; } /** Get keys (IVs) to be used */ diff --git a/src/modules/m_md5.cpp b/src/modules/m_md5.cpp index 402007177..6096f9afa 100644 --- a/src/modules/m_md5.cpp +++ b/src/modules/m_md5.cpp @@ -245,11 +245,11 @@ class ModuleMD5 : public Module } - void GenHash(const char* src, char* dest, const char* xtab, unsigned int* ikey) + void GenHash(const char* src, char* dest, const char* xtab, unsigned int* ikey, size_t srclen) { unsigned char bytes[16]; - MyMD5((char*)bytes, (void*)src, strlen(src), ikey); + MyMD5((char*)bytes, (void*)src, srclen, ikey); for (int i = 0; i < 16; i++) { @@ -293,7 +293,7 @@ class ModuleMD5 : public Module else if (strcmp("SUM", request->GetId()) == 0) { static char data[MAXBUF]; - GenHash((const char*)MD5->GetHashData(), data, chars ? chars : "0123456789abcdef", key); + GenHash(MD5->GetHashData().data(), data, chars ? chars : "0123456789abcdef", key, MD5->GetHashData().length()); return data; } else if (strcmp("NAME", request->GetId()) == 0) diff --git a/src/modules/m_sha256.cpp b/src/modules/m_sha256.cpp index 70727b301..aee1cb12a 100644 --- a/src/modules/m_sha256.cpp +++ b/src/modules/m_sha256.cpp @@ -285,7 +285,7 @@ class ModuleSHA256 : public Module else if (strcmp("SUM", request->GetId()) == 0) { static char data[MAXBUF]; - SHA256((const char*)SHA->GetHashData(), data, strlen(SHA->GetHashData()), chars ? chars : "0123456789abcdef", key); + SHA256((const char*)SHA->GetHashData().data(), data, SHA->GetHashData().length(), chars ? chars : "0123456789abcdef", key); return data; } else if (strcmp("NAME", request->GetId()) == 0) |