X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=BOINC%20software%2FBOINC%20client%20apps%2Fdistrrtgen_cuda%2Frcuda_ext.cpp;fp=BOINC%20software%2FBOINC%20client%20apps%2Fdistrrtgen_cuda%2Frcuda_ext.cpp;h=86f31ecc948d529eeb5bce83857cab7b0127e817;hb=fcc09bb2d8e4fdcca60f465a04c7babf1f5e573c;hp=0000000000000000000000000000000000000000;hpb=b5aba2e58a81ffefe6515f13c7ed34c369691d0f;p=freerainbowtables

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
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();
+}