backport rcracki_mt trunk from rcracki.sourceforge.net

[freerainbowtables] / Client Applications / rcracki_mt / RTI2Reader.cpp

diff --git a/Client Applications/rcracki_mt/RTI2Reader.cpp b/Client Applications/rcracki_mt/RTI2Reader.cpp
index abd855076517797a322f22c81be219e749963112..62b96d72d6b82c9256ae22c09947506a5ad3e52e 100644 (file)
--- a/Client Applications/rcracki_mt/RTI2Reader.cpp
+++ b/Client Applications/rcracki_mt/RTI2Reader.cpp
@@ -48,7 +48,11 @@ RTI2Reader::RTI2Reader(string Filename)
        long len = GetFileLen(pFileIndex);
        fseek(pFileIndex, 0, SEEK_SET);
 
-       m_pIndex = new unsigned char[len];
+       m_pIndex = new (nothrow) unsigned char[len];
+       if(m_pIndex == NULL) {
+               printf("Error allocating %ld MB memory for index in RTI2Reader::RTI2Reader()", len / (1024 * 1024));
+               exit(-2);
+       }
        if(fread(m_pIndex, 1, len, pFileIndex) != (unsigned long)len)
        {
                printf("Error while reading index file");
@@ -81,7 +85,7 @@ RTI2Reader::~RTI2Reader(void)
 
 }
 
-unsigned int RTI2Reader::GetChainsLeft()
+uint32 RTI2Reader::GetChainsLeft()
 {
        long len = GetFileLen(m_pFile);
        return len / m_chainsizebytes - m_chainPosition;
@@ -101,6 +105,7 @@ int RTI2Reader::ReadChains(unsigned int &numChains, RainbowChainO *pData)
        
        while(true) // Fast forward to current position
        {
+               /// XXX
                // ALERT: Possible problem here if m_indexrowsizebytes > 1 as pNumChains is a unsigned char.
                unsigned int NumChainsInRow = (unsigned int)*(pNumChains + indexRow * m_indexrowsizebytes);
                if(m_indexrowsizebytes > 1)