From 8a21fbf0cec3af9c9a7b957fdbffb19addb0b7df Mon Sep 17 00:00:00 2001 From: faketruth Date: Tue, 27 Dec 2011 17:59:08 +0000 Subject: I think I fixed the memory leaks in the converter Updated makefile for Unix I changed how the converter works, now you simply put the .exe in the folder you want to convert and run it, it'll output all the files in ./world git-svn-id: http://mc-server.googlecode.com/svn/trunk@132 0a769ca7-a7f5-676a-18bf-c427514a06d6 --- converter/source/MemoryLeak.h | 18 +++ converter/source/cDeNotch.cpp | 324 ++++++++++++++++++++++-------------------- converter/source/cMakeDir.cpp | 24 ++++ converter/source/cMakeDir.h | 7 + converter/source/cNBTData.cpp | 39 +++-- converter/source/main.cpp | 52 ++++--- converter/source/printdir.c | 5 + 7 files changed, 292 insertions(+), 177 deletions(-) create mode 100644 converter/source/MemoryLeak.h create mode 100644 converter/source/cMakeDir.cpp create mode 100644 converter/source/cMakeDir.h (limited to 'converter/source') diff --git a/converter/source/MemoryLeak.h b/converter/source/MemoryLeak.h new file mode 100644 index 000000000..6ac06a302 --- /dev/null +++ b/converter/source/MemoryLeak.h @@ -0,0 +1,18 @@ +#pragma once + +#ifdef _WIN32 + +#ifdef _DEBUG + +#define _CRTDBG_MAP_ALLOC +#include +#include + +#ifndef DEBUG_NEW +#define DEBUG_NEW new(_NORMAL_BLOCK, __FILE__, __LINE__) +#define new DEBUG_NEW +#endif + +#endif + +#endif diff --git a/converter/source/cDeNotch.cpp b/converter/source/cDeNotch.cpp index c2963f365..89075a92e 100644 --- a/converter/source/cDeNotch.cpp +++ b/converter/source/cDeNotch.cpp @@ -1,3 +1,5 @@ +#include "MemoryLeak.h" + #include "cDeNotch.h" #include #include @@ -9,46 +11,56 @@ #include "cTimer.h" #include "cQuicksort.h" #include "cDeNotch.h" +#include "cMakeDir.h" + #ifdef _WIN32 #include "wdirent.h" #else #include #endif +#ifndef _WIN32 +#define sprintf_s(target, size, ... ) printf(target, __VA_ARGS__ ) +#endif +inline bool fOpenFile( FILE*& a_hFile, const char* a_FileName, const char* a_Mode ) +{ +#ifdef _WIN32 + return fopen_s(&a_hFile, a_FileName, a_Mode ) == 0; +#else + return (a_hFile = fopen(a_FileName, a_Mode )) != 0; +#endif +} -cDeNotch::cDeNotch ( ) { +cDeNotch::cDeNotch ( ) +{ } -int cDeNotch:: Converter ( std::string mcrSource, std::string pakOutput ) { - char SourceFile[128]; - char OutputFile[128]; - +int cDeNotch:: Converter ( std::string mcrSource, std::string pakOutput ) +{ + std::string OutDir = "world/"; FILE* f = 0; - FILE* wf = 0; - #ifdef _WIN32 - sprintf_s(SourceFile, 128, "region\\%s", mcrSource.c_str() ); //replace hard coded file with file array variable - sprintf_s(OutputFile, 128, "world\\%s", pakOutput.c_str() ); //parce x and z from file array variable and place into pak file format - if( fopen_s(&wf, OutputFile, "wb" ) == 0 ) {} else { std::cout << "uhoh!" << std::endl; return false; } //open new pak file for writing - #else - sprintf(SourceFile, "region/%s", mcrSource.c_str() ); //same as above but for linux - sprintf(OutputFile, "world/%s", pakOutput.c_str() ); - if( (wf = fopen(OutputFile, "wb" )) != 0 ) {} else { std::cout << "uhoh!" << std::endl; return false; } - #endif - - - printf ("Now Converting %s to %s\n", mcrSource.c_str(), pakOutput.c_str() ); - if( (f = fopen(SourceFile, "rb" )) != 0 ) { // no error - - char* t_FakeHeader; - t_FakeHeader = new char[1*1024*1024]; //1MB Temp FakeHeader array + FILE* wf = 0; + + std::string SourceFileName = mcrSource; + std::string OutFileName = OutDir + pakOutput; + + cMakeDir::MakeDir( OutDir.c_str() ); + if( !fOpenFile(wf, OutFileName.c_str(), "wb") ) + { + printf("Could not open/create file %s\n", OutFileName.c_str() ); + } + + printf ("Now Converting %s to %s\n", SourceFileName.c_str(), OutFileName.c_str() ); + if( fOpenFile(f, SourceFileName.c_str(), "rb") ) + { + char* t_FakeHeader = new char[1*1024*1024]; //1MB Temp FakeHeader array int t_FakeHeaderSz = -1; //Size of data in array - char* t_CompChunk; - t_CompChunk = new char[5*1024*1024]; //5MB Temp Compressed Chunk Data array - int t_CompChunkSz = -1; //Size of data in array + char* t_CompChunk = new char[5*1024*1024]; //5MB Temp Compressed Chunk Data array + int t_CompChunkSz = -1; //Size of data in array char PakVersion = 1; char ChunkVersion = 1; @@ -68,152 +80,162 @@ int cDeNotch:: Converter ( std::string mcrSource, std::string pakOutput ) { int toffarr[1024]; //loop through notch's header - for( short i = 0; i < 1024 ; ++i ) {//loop through first 4096 bytes of data, 4 bytes at a time + for( short i = 0; i < 1024 ; ++i ) + { //loop through first 4096 bytes of data, 4 bytes at a time //Region files begin with an 8kiB header containing information about which chunks are present in the region file, when they were last updated, and where they can be found. The location in the region file of a chunk at (x, z) (in chunk coordinates) can be found at byte offset 4 * ((x mod 32) + (z mod 32) * 32) in its region file. Its timestamp can be found 4096 bytes later in the file. The remainder of the file consists of data for up to 1024 chunks, interspersed with an arbitrary amount of unused space. //we are only using the first 4096 bytes. We don't need the timestamps right now. - if( fread( &byte1, sizeof(byte1), 1, f) != 1 ) { std::cout << "ERROR 21hs READING FROM FILE " << SourceFile; fclose(f); return false; } - if( fread( &byte2, sizeof(byte2), 1, f) != 1 ) { std::cout << "ERROR ks93 READING FROM FILE " << SourceFile; fclose(f); return false; } - if( fread( &byte3, sizeof(byte3), 1, f) != 1 ) { std::cout << "ERROR 2s5f READING FROM FILE " << SourceFile; fclose(f); return false; }//first three bytes area big-endian representation of the chunk offsets in no particular order. - if( fread( &byte4, sizeof(byte4), 1, f) != 1 ) { std::cout << "ERROR dhj3 READING FROM FILE " << SourceFile; fclose(f); return false; }//we don't need to use this byte right now. + if( fread( &byte1, sizeof(byte1), 1, f) != 1 ) { std::cout << "ERROR 21hs READING FROM FILE " << SourceFileName; fclose(f); return false; } + if( fread( &byte2, sizeof(byte2), 1, f) != 1 ) { std::cout << "ERROR ks93 READING FROM FILE " << SourceFileName; fclose(f); return false; } + if( fread( &byte3, sizeof(byte3), 1, f) != 1 ) { std::cout << "ERROR 2s5f READING FROM FILE " << SourceFileName; fclose(f); return false; }//first three bytes area big-endian representation of the chunk offsets in no particular order. + if( fread( &byte4, sizeof(byte4), 1, f) != 1 ) { std::cout << "ERROR dhj3 READING FROM FILE " << SourceFileName; fclose(f); return false; }//we don't need to use this byte right now. toffset = 4096 * ((byte1*256*256) + (byte2*256) + byte3);//find the chunk offsets using the first three bytes of each long; - toffarr[i] = toffset;//array of chunk offset locatiosn in the fle. + toffarr[i] = toffset;//array of chunk offset locations in the file. } - for ( short i = 0; i < 4096; i++ ) {//loop through next 4096 bytes of the header. + for ( short i = 0; i < 4096; i++ ) + { //loop through next 4096 bytes of the header. //keeping this code here in case we need it later. not using it right now. - if( fread( &trash, sizeof(byte4), 1, f) != 1 ) { std::cout << "ERROR 2jkd READING FROM FILE " << SourceFile; fclose(f); return false; } + if( fread( &trash, sizeof(byte4), 1, f) != 1 ) { std::cout << "ERROR 2jkd READING FROM FILE " << SourceFileName; fclose(f); return false; } } frloc = 8192; //current location of fread is at 4096+ 4096 since we read through and collected important info from the header. cQuicksort Quick; - Quick.quicksort(toffarr, 0, 1023); //sort the array from smallest to larget offset locations so we only have to read through the file once. - - for ( short ia = 0; ia < 1024; ia++ ) {//a region file can hold a maximum of 1024 chunks (32*32) - if (ia < 3500 ) { //only run chunk # 3 - if (toffarr[ia] < 8192) { //offsets of less than 8192 are impossible. 0 means there is no chunk in a particular location. - if (toffarr[ia] > 0) { std::cout << "ERROR 2s31 IN COLLECTED CHUNK OFFSETS " << toffarr[ia]; fclose(f); return false; } //values between 0 and 8192 should be impossible. - //This file does not contain the max 1024 chunks, skip until we get to the first - } else { // found a chunk offset value - //Chunk data begins with a (big-endian) four-byte length field which indicates the exact length of the remaining chunk data in bytes. The following byte indicates the compression scheme used for chunk data, and the remaining (length-1) bytes are the compressed chunk data. - //printf("Working on chunk %i :: %i\n", ia, toffarr[ia]); - if( fread( &byte1, sizeof(byte1), 1, f) != 1 ) { std::cout << "ERROR 2t32 READING FROM FILE " << SourceFile; fclose(f); return false; } - if( fread( &byte2, sizeof(byte2), 1, f) != 1 ) { std::cout << "ERROR 2y51 READING FROM FILE " << SourceFile; fclose(f); return false; } - if( fread( &byte3, sizeof(byte3), 1, f) != 1 ) { std::cout << "ERROR 3424 READING FROM FILE " << SourceFile; fclose(f); return false; } - if( fread( &byte4, sizeof(byte4), 1, f) != 1 ) { std::cout << "ERROR sd22 READING FROM FILE " << SourceFile; fclose(f); return false; } - compdlength = ((byte1*256*256*256) + (byte2*256*256) + (byte3*256) + byte4 - 0); //length of compressed chunk data - if( fread( &byte5, sizeof(byte5), 1, f) != 1 ) { std::cout << "ERROR 2341 READING FROM FILE " << SourceFile; fclose(f); return false; } //compression type, 1 = GZip (RFC1952) (unused in practice) , 2 = Zlib (RFC1950) - - frloc += 5; //moved ahead 5 bytes while reading data. - - char* compBlockData = new char[compdlength]; - if( fread( compBlockData, compdlength, 1, f) != 1 ) { std::cout << "ERROR rf22 READING FROM FILE " << SourceFile; fclose(f); return false; } - frloc = frloc + compdlength; - - uLongf DestSize = 128576;// uncompressed chunks should never be larger than this - - char* BlockData = new char[ DestSize ]; - - int errorcode = uncompress( (Bytef*)BlockData, &DestSize, (Bytef*)compBlockData, compdlength ); //DestSize will update to the actual uncompressed data size after this opperation. - int testr = (int)DestSize; //testing something, can't remember what. - if( errorcode != Z_OK ){ - printf("ERROR: Decompressing chunk data! %i", errorcode ); - switch( errorcode ) + Quick.quicksort(toffarr, 0, 1023); //sort the array from smallest to largest offset locations so we only have to read through the file once. + + for ( short ia = 0; ia < 1024; ia++ ) + { //a region file can hold a maximum of 1024 chunks (32*32) + if (ia < 3500 ) + { //only run chunk # 3 + if (toffarr[ia] < 8192) + { //offsets of less than 8192 are impossible. 0 means there is no chunk in a particular location. + if (toffarr[ia] > 0) { std::cout << "ERROR 2s31 IN COLLECTED CHUNK OFFSETS " << toffarr[ia]; fclose(f); return false; } //values between 0 and 8192 should be impossible. + //This file does not contain the max 1024 chunks, skip until we get to the first + } + else + { // found a chunk offset value + //Chunk data begins with a (big-endian) four-byte length field which indicates the exact length of the remaining chunk data in bytes. The following byte indicates the compression scheme used for chunk data, and the remaining (length-1) bytes are the compressed chunk data. + //printf("Working on chunk %i :: %i\n", ia, toffarr[ia]); + if( fread( &byte1, sizeof(byte1), 1, f) != 1 ) { std::cout << "ERROR 2t32 READING FROM FILE " << SourceFileName; fclose(f); return false; } + if( fread( &byte2, sizeof(byte2), 1, f) != 1 ) { std::cout << "ERROR 2y51 READING FROM FILE " << SourceFileName; fclose(f); return false; } + if( fread( &byte3, sizeof(byte3), 1, f) != 1 ) { std::cout << "ERROR 3424 READING FROM FILE " << SourceFileName; fclose(f); return false; } + if( fread( &byte4, sizeof(byte4), 1, f) != 1 ) { std::cout << "ERROR sd22 READING FROM FILE " << SourceFileName; fclose(f); return false; } + compdlength = ((byte1*256*256*256) + (byte2*256*256) + (byte3*256) + byte4 - 0); //length of compressed chunk data + if( fread( &byte5, sizeof(byte5), 1, f) != 1 ) { std::cout << "ERROR 2341 READING FROM FILE " << SourceFileName; fclose(f); return false; } //compression type, 1 = GZip (RFC1952) (unused in practice) , 2 = Zlib (RFC1950) + + frloc += 5; //moved ahead 5 bytes while reading data. + + char* compBlockData = new char[compdlength]; + if( fread( compBlockData, compdlength, 1, f) != 1 ) { std::cout << "ERROR rf22 READING FROM FILE " << SourceFileName; fclose(f); return false; } + frloc = frloc + compdlength; + + uLongf DestSize = 128576;// uncompressed chunks should never be larger than this + + char* BlockData = new char[ DestSize ]; + + int errorcode = uncompress( (Bytef*)BlockData, &DestSize, (Bytef*)compBlockData, compdlength ); //DestSize will update to the actual uncompressed data size after this opperation. + int testr = (int)DestSize; //testing something, can't remember what. + if( errorcode != Z_OK ){ + printf("ERROR: Decompressing chunk data! %i", errorcode ); + switch( errorcode ) + { + case Z_MEM_ERROR: + printf("Not enough memory"); + break; + case Z_BUF_ERROR: + printf("Not enough room in output buffer"); + break; + case Z_DATA_ERROR: + printf("Input data corrupted or incomplete"); + break; + default: + break; + }; + } + + NumChunks++; + cNBTData* NBTData = new cNBTData(BlockData, (int)DestSize); + NBTData->ParseData(); + //NBTData->PrintData(); + NBTData->OpenCompound(""); + NBTData->OpenCompound("Level"); // You need to open the right compounds before you can access the data in it + + //NBT Data for blocks should look something like this: + //==== STRUCTURED NBT DATA ==== + // COMPOUND ( ) + // COMPOUND + // COMPOUND (Level) + // LIST (Entities) + // LIST (TileEntities) + // INTEGER LastUpdate (0) + // INTEGER xPos (0) + // INTEGER zPos (0) + // BYTE TerrainPopulated (1) + // BYTE ARRAY BlockLight (length: 16384) + // BYTE ARRAY Blocks (length: 32768) + // BYTE ARRAY Data (length: 16384) + // BYTE ARRAY HeightMap (length: 256) + // BYTE ARRAY SkyLight (length: 16384) + //============================= + + int UncompressedChunkSz = (32768+16384+16384+16384); + char* UncompressedChunk = new char[ UncompressedChunkSz ]; + uLongf CompressedSize = compressBound( UncompressedChunkSz ); + char* CompressedChunk = new char[ CompressedSize ]; + int UnChunkArrLoc = 0; + int xPos = NBTData->GetInteger("xPos"); + int zPos = NBTData->GetInteger("zPos"); + //printf("Chunk [%i, %i]\n", xPos, zPos ); + + memcpy( t_FakeHeader + t_FakeHeaderSz + 1, &xPos, sizeof(int) );t_FakeHeaderSz += sizeof(int); + memcpy( t_FakeHeader + t_FakeHeaderSz + 1, &zPos, sizeof(int) );t_FakeHeaderSz += sizeof(int); + + + //TODO: insert json code and add it to chunk data + + memcpy( UncompressedChunk + UnChunkArrLoc, NBTData->GetByteArray("Blocks"), 32768 );UnChunkArrLoc += 32768; + memcpy( UncompressedChunk + UnChunkArrLoc, NBTData->GetByteArray("Data"), 16384 );UnChunkArrLoc += 16384; + memcpy( UncompressedChunk + UnChunkArrLoc, NBTData->GetByteArray("BlockLight"), 16384 );UnChunkArrLoc += 16384; + memcpy( UncompressedChunk + UnChunkArrLoc, NBTData->GetByteArray("SkyLight"), 16384 );UnChunkArrLoc += 16384; + + errorcode = compress2( (Bytef*)CompressedChunk, &CompressedSize, (const Bytef*)UncompressedChunk, UncompressedChunkSz, Z_DEFAULT_COMPRESSION); + if( errorcode != Z_OK ) { - case Z_MEM_ERROR: - printf("Not enough memory"); - break; - case Z_BUF_ERROR: - printf("Not enough room in output buffer"); - break; - case Z_DATA_ERROR: - printf("Input data corrupted or incomplete"); - break; - default: - break; - }; - } + printf("Error compressing data (%i)", errorcode ); + break; + } - NumChunks++; - cNBTData* NBTData = new cNBTData(BlockData, (int)DestSize); - NBTData->ParseData(); - //NBTData->PrintData(); - NBTData->OpenCompound(""); - NBTData->OpenCompound("Level"); // You need to open the right compounds before you can access the data in it - - //NBT Data for blocks should look something like this: - //==== STRUCTURED NBT DATA ==== - // COMPOUND ( ) - // COMPOUND - // COMPOUND (Level) - // LIST (Entities) - // LIST (TileEntities) - // INTEGER LastUpdate (0) - // INTEGER xPos (0) - // INTEGER zPos (0) - // BYTE TerrainPopulated (1) - // BYTE ARRAY BlockLight (length: 16384) - // BYTE ARRAY Blocks (length: 32768) - // BYTE ARRAY Data (length: 16384) - // BYTE ARRAY HeightMap (length: 256) - // BYTE ARRAY SkyLight (length: 16384) - //============================= - - int UncompressedChunkSz = (32768+16384+16384+16384); - char* UncompressedChunk = new char[ UncompressedChunkSz ]; - uLongf CompressedSize = compressBound( UncompressedChunkSz ); - char* CompressedChunk = new char[ CompressedSize ]; - int UnChunkArrLoc = 0; - int xPos = NBTData->GetInteger("xPos"); - int zPos = NBTData->GetInteger("zPos"); - - memcpy( t_FakeHeader + t_FakeHeaderSz + 1, &xPos, sizeof(int) );t_FakeHeaderSz += sizeof(int); - memcpy( t_FakeHeader + t_FakeHeaderSz + 1, &zPos, sizeof(int) );t_FakeHeaderSz += sizeof(int); - - - //todo: inserert json code and add it to chunk data - - memcpy( UncompressedChunk + UnChunkArrLoc, NBTData->GetByteArray("Blocks"), 32768 );UnChunkArrLoc += 32768; - memcpy( UncompressedChunk + UnChunkArrLoc, NBTData->GetByteArray("Data"), 16384 );UnChunkArrLoc += 16384; - memcpy( UncompressedChunk + UnChunkArrLoc, NBTData->GetByteArray("BlockLight"), 16384 );UnChunkArrLoc += 16384; - memcpy( UncompressedChunk + UnChunkArrLoc, NBTData->GetByteArray("SkyLight"), 16384 );UnChunkArrLoc += 16384; - - errorcode = compress2( (Bytef*)CompressedChunk, &CompressedSize, (const Bytef*)UncompressedChunk, UncompressedChunkSz, Z_DEFAULT_COMPRESSION); - if( errorcode != Z_OK ) - { - printf("Error compressing data (%i)", errorcode ); - break; - } + memcpy( t_FakeHeader + t_FakeHeaderSz + 1, &CompressedSize, sizeof(int) );t_FakeHeaderSz += sizeof(int); + memcpy( t_FakeHeader + t_FakeHeaderSz + 1, &UncompressedChunkSz, sizeof(int) );t_FakeHeaderSz += sizeof(int); + memcpy( t_CompChunk + t_CompChunkSz + 1, CompressedChunk, CompressedSize );t_CompChunkSz += CompressedSize; - memcpy( t_FakeHeader + t_FakeHeaderSz + 1, &CompressedSize, sizeof(int) );t_FakeHeaderSz += sizeof(int); - memcpy( t_FakeHeader + t_FakeHeaderSz + 1, &UncompressedChunkSz, sizeof(int) );t_FakeHeaderSz += sizeof(int); - memcpy( t_CompChunk + t_CompChunkSz + 1, CompressedChunk, CompressedSize );t_CompChunkSz += CompressedSize; + NBTData->CloseCompound();// Close the compounds after you're done + NBTData->CloseCompound(); - NBTData->CloseCompound();// Close the compounds after you're done - NBTData->CloseCompound(); + delete [] UncompressedChunk; + delete [] CompressedChunk; + delete [] compBlockData; + delete [] BlockData; - delete [] UncompressedChunk; - delete [] CompressedChunk; - delete [] compBlockData; - delete [] BlockData; - //delete [] NBTData; + while ( (frloc < toffarr[ia+1]) && (ia<1023) ) + { //loop through Notch's junk data until we get to another chunk offset possition to start the loop again + if( fread( &trash, sizeof(byte4), 1, f) != 1 ) { std::cout << "ERROR 2nkd READING FROM FILE " << SourceFileName; fclose(f); return false; } + frloc ++; + } - while ( (frloc < toffarr[ia+1]) && (ia<1023) ) { //loop through Notch's junk data until we get to another chunk offset possition to start the loop again - if( fread( &trash, sizeof(byte4), 1, f) != 1 ) { std::cout << "ERROR 2nkd READING FROM FILE " << SourceFile; fclose(f); return false; } - frloc ++; - } - } + delete NBTData; + } } //only run chunk # 3 - } - fwrite( &PakVersion, sizeof(PakVersion), 1, wf ); - fwrite( &ChunkVersion, sizeof(ChunkVersion), 1, wf ); + + fwrite( &PakVersion, sizeof(PakVersion), 1, wf ); + fwrite( &ChunkVersion, sizeof(ChunkVersion), 1, wf ); fwrite( &NumChunks, sizeof(NumChunks), 1, wf ); - fwrite( t_FakeHeader, t_FakeHeaderSz+1, 1, wf ); - fwrite( t_CompChunk, t_CompChunkSz+1, 1, wf ); - delete [] t_FakeHeader; - delete [] t_CompChunk; + fwrite( t_FakeHeader, t_FakeHeaderSz+1, 1, wf ); + fwrite( t_CompChunk, t_CompChunkSz+1, 1, wf ); + delete [] t_FakeHeader; + delete [] t_CompChunk; - fclose(wf); //close file. + fclose(wf); //close file. fclose(f); //close file. } return true; diff --git a/converter/source/cMakeDir.cpp b/converter/source/cMakeDir.cpp new file mode 100644 index 000000000..60817d5aa --- /dev/null +++ b/converter/source/cMakeDir.cpp @@ -0,0 +1,24 @@ +#include "cMakeDir.h" + +#ifndef _WIN32 +//#include // If something is missing, uncomment some of these! +//#include +//#include +#include // for mkdir +//#include +#else +#include +#endif + +void cMakeDir::MakeDir( const char* a_Directory ) +{ +#ifdef _WIN32 + SECURITY_ATTRIBUTES Attrib; + Attrib.nLength = sizeof(SECURITY_ATTRIBUTES); + Attrib.lpSecurityDescriptor = NULL; + Attrib.bInheritHandle = false; + ::CreateDirectory(a_Directory, &Attrib); +#else + mkdir(a_Directory, S_IRWXU | S_IRWXG | S_IRWXO); +#endif +} \ No newline at end of file diff --git a/converter/source/cMakeDir.h b/converter/source/cMakeDir.h new file mode 100644 index 000000000..378df2a36 --- /dev/null +++ b/converter/source/cMakeDir.h @@ -0,0 +1,7 @@ +#pragma once + +class cMakeDir +{ +public: + static void MakeDir( const char* a_Directory ); +}; \ No newline at end of file diff --git a/converter/source/cNBTData.cpp b/converter/source/cNBTData.cpp index 556960732..cdd88a0c0 100644 --- a/converter/source/cNBTData.cpp +++ b/converter/source/cNBTData.cpp @@ -1,3 +1,5 @@ +#include "MemoryLeak.h" + #include "cNBTData.h" #include // memcpy #include @@ -40,8 +42,10 @@ cNBTData::cNBTData( char* a_Buffer, unsigned int a_BufferSize ) m_ParseFunctions[TAG_ByteArray] = &cNBTData::ParseByteArray; - m_Buffer = a_Buffer; - m_BufferSize = a_BufferSize; + m_BufferSize = a_BufferSize; + m_Buffer = new char[m_BufferSize]; // Make a copy of the buffer + memcpy( m_Buffer, a_Buffer, m_BufferSize ); + m_Index = 0; tm = false; //tm to true will print more information for test mode @@ -479,6 +483,10 @@ void cNBTData::ParseData() } ParseTags(); } + + + delete [] m_Buffer; + m_Buffer = 0; m_BufferSize = 0; } void cNBTData::ParseTags() @@ -682,10 +690,11 @@ void cNBTData::ParseByteArray( bool a_bNamed ) int Length = ReadInt(); std::string String; - char* ByteArray = new char[ Length ]; + char* ByteArray = 0; if( Length > 0 ) { - memcpy( ByteArray, &m_Buffer[ m_Index ], Length ); + ByteArray = new char[ Length ]; + memcpy( ByteArray, &m_Buffer[ m_Index ], Length ); m_Index += Length; } @@ -856,8 +865,9 @@ void cNBTList::Serialize(std::string & a_Buffer) void cNBTData::Clear() { - while( m_CurrentCompound != this ) CloseCompound(); - m_CurrentCompound->Clear(); + while( m_CurrentCompound != this ) CloseCompound(); // Close ALL the compounds!! + + m_CurrentCompound->Clear(); // This recursively clears all compounds if( m_Buffer ) { @@ -885,10 +895,23 @@ void cNBTCompound::Clear() delete itr->second; itr->second = 0; } - m_Lists.clear(); - m_Bytes.clear(); + m_Lists.clear(); + + for( ByteArrayMap::iterator itr = m_ByteArrays.begin(); itr != m_ByteArrays.end(); itr++ ) + { + if( itr->second == 0 ) continue; + delete [] itr->second; + itr->second = 0; + } + m_ByteArrays.clear(); + + // Don't really have to do this, but meh + m_Bytes.clear(); m_Shorts.clear(); m_Integers.clear(); + m_Longs.clear(); + m_Doubles.clear(); + m_Floats.clear(); m_Strings.clear(); } diff --git a/converter/source/main.cpp b/converter/source/main.cpp index 67da82804..6239cbd43 100644 --- a/converter/source/main.cpp +++ b/converter/source/main.cpp @@ -1,3 +1,5 @@ +#include "MemoryLeak.h" + #include #include "cNBTData.h" #include "cTimer.h" @@ -9,32 +11,37 @@ #include #endif -int main () { +int main () +{ + +#ifdef _DEBUG + _CrtSetDbgFlag ( _CRTDBG_ALLOC_MEM_DF | _CRTDBG_LEAK_CHECK_DF ); +#endif cTimer Timer; clock_t progBegin = clock(); //start main program timer std::string dir; DIR* dp; struct dirent *entry; - int found; + int found; std::string entrys; - std::string str2; + std::string str2; std::string str3; - std::string filexPos; + std::string filexPos; std::string filezPos; std::string pak_name; //string* dir_array; int dir_num_files = 0; int ctr = 0; -#ifdef _WIN32 - if(dp = opendir("region\\")){ -#else - if(dp = opendir("region/")){ -#endif - while(entry = readdir(dp)){ + + if(dp = opendir("./")) + { + while(entry = readdir(dp)) + { entrys = entry->d_name; found = entrys.find(".mcr"); - if ( (found!=std::string::npos) ) { + if ( (found!=std::string::npos) ) + { str2 = entrys.substr (2,sizeof(entrys)); filexPos = str2.substr (0,(int)str2.find(".")); str3 = str2.substr ((int)str2.find(".")+1, sizeof(str2)); @@ -47,13 +54,22 @@ int main () { DeNotch.Converter ( entrys, pak_name ); - clock_t end=clock(); - std::cout << "Time to convert chunk: " << double(Timer.diffclock(end,begin)) << " Seconds"<< std::endl; - } - } - closedir(dp); - } + clock_t end=clock(); + std::cout << "Time to convert chunk: " << double(Timer.diffclock(end,begin)) << " Seconds"<< std::endl; + } + } + closedir(dp); + } clock_t progEnd = clock(); //end main program timer std::cout << "Time to complete converter: " << double(Timer.diffclock(progEnd,progBegin)) << " Seconds"<< std::endl; - return 0; + +#ifdef _DEBUG + _CrtDumpMemoryLeaks(); +#endif + +#ifdef _WIN32 + system("PAUSE"); +#endif + + return 0; }; diff --git a/converter/source/printdir.c b/converter/source/printdir.c index 7216413be..d163be380 100644 --- a/converter/source/printdir.c +++ b/converter/source/printdir.c @@ -1,3 +1,5 @@ +#if 0 + /* list contents of a directory */ #include #include @@ -50,3 +52,6 @@ int main (int argc, char *argv[]) { } return EXIT_SUCCESS; } + + +#endif \ No newline at end of file -- cgit v1.2.3