distrrtgen_cuda code added

[freerainbowtables] / BOINC software / BOINC client apps / distrrtgen_cuda / rcuda_ext.cpp

diff --git a/BOINC software/BOINC client apps/distrrtgen_cuda/rcuda_ext.cpp b/BOINC software/BOINC client apps/distrrtgen_cuda/rcuda_ext.cpp
new file mode 100644 (file)
index 0000000..86f31ec
--- /dev/null
+++ b/BOINC software/BOINC client apps/distrrtgen_cuda/rcuda_ext.cpp
@@ -0,0 +1,85 @@
+//============================================================================
+// Name        : rcuda_ext.cpp
+// Author      : Jan Kyska
+// Version     : 0.9
+// Description : A code to access internals of the CChainWalkContext 
+//               for the CUDA generator of FreeRainbowTables
+//============================================================================ 
+
+#include "rcuda_ext.h"
+#include <algorithm>
+#include <string>
+#include <time.h>
+#include <fstream>
+#include <iostream>
+
+CudaCWCExtender::CudaCWCExtender(CChainWalkContext *cwc) { 
+       this->cwc = cwc; 
+       hash = rcuda::RHASH_UNDEF;
+}
+
+void CudaCWCExtender::Init(void) { 
+       std::string hashName;
+       int ii, jj;
+
+       plainDimVec.clear();
+       plainCharSet.clear();
+
+       hashName = CChainWalkContext::m_sHashRoutineName;
+       std::transform(hashName.begin(), hashName.end(), hashName.begin(), tolower);
+       if(hashName.compare("lm") == 0)
+               hash = rcuda::RHASH_LM;
+       else if(hashName.compare("md4") == 0)
+               hash = rcuda::RHASH_MD4;
+       else if(hashName.compare("md5") == 0)
+               hash = rcuda::RHASH_MD5;
+       else if(hashName.compare("sha1") == 0)
+               hash = rcuda::RHASH_SHA1;
+       else if(hashName.compare("ntlm") == 0)
+               hash = rcuda::RHASH_NTLM;
+       else
+               hash = rcuda::RHASH_UNDEF;
+
+       for(ii = 0; ii < (int)CChainWalkContext::m_vCharset.size(); ii++) {
+               stCharset &chs = CChainWalkContext::m_vCharset[ii];
+               int chSetOffset = plainCharSet.size();
+               plainCharSet.append((char*)chs.m_PlainCharset, chs.m_nPlainCharsetLen);
+               for(jj = 0; jj < chs.m_nPlainLenMax; jj++) {
+                       plainDimVec.push_back((unsigned int)chs.m_nPlainCharsetLen);
+                       plainDimVec.push_back((unsigned int)-1/(unsigned int)chs.m_nPlainCharsetLen);
+                       plainDimVec.push_back((unsigned int)chSetOffset);
+               }
+       }
+}
+
+int CudaCWCExtender::IndexToStartPlain(const uint64 nIndex, std::vector<unsigned char>& stPlain) {
+       int nPlainLen, nCharsetLen;
+       int ii, jj;
+
+       stPlain.clear();
+       stPlain.reserve(0x20);
+       nPlainLen = 0;
+       for(ii = CChainWalkContext::m_nPlainLenMaxTotal - 1; ii >= CChainWalkContext::m_nPlainLenMinTotal - 1; ii--) {
+               if(nIndex >= CChainWalkContext::m_nPlainSpaceUpToX[ii]) {
+                       nPlainLen = ii + 1;
+                       break;
+               }
+       }
+       if(nPlainLen == 0)
+               nPlainLen = CChainWalkContext::m_nPlainLenMinTotal;
+       uint64 nIndexOfX = nIndex - CChainWalkContext::m_nPlainSpaceUpToX[nPlainLen - 1];
+
+       // Slow version, but never mind
+       for(ii = nPlainLen - 1; ii >= 0; ii--) {
+               nCharsetLen = 0;
+               for(jj = 0; jj < (int)CChainWalkContext::m_vCharset.size(); jj++) {
+                       stCharset &chs = CChainWalkContext::m_vCharset[jj];
+                       nCharsetLen += chs.m_nPlainLenMax;
+                       if(ii < nCharsetLen) { // We found the correct charset
+                               stPlain.push_back(nIndexOfX % chs.m_nPlainCharsetLen + 1);
+                               nIndexOfX /= chs.m_nPlainCharsetLen;
+                       }
+               }
+       }
+       return stPlain.size();
+}