git-svn-id: http://mc-server.googlecode.com/svn/trunk@36 0a769ca7-a7f5-676a-18bf-c427514a06d6
305 lines
14 KiB
C++
305 lines
14 KiB
C++
#include <iostream>
|
|
#include <fstream>
|
|
#include <string>
|
|
#include <stdio.h>
|
|
//#include <string.h>
|
|
#include <ctype.h>
|
|
#include "zlib.h"
|
|
#include "cNBTData.h"
|
|
#include "timer.h"
|
|
#include "quicksort.h"
|
|
#include <dirent.h>
|
|
//#include "dircont.h"
|
|
|
|
using namespace std;
|
|
|
|
|
|
int main () {
|
|
|
|
string dir;
|
|
DIR* dp;
|
|
struct dirent *entry;
|
|
int found;
|
|
string entrys;
|
|
string str2;
|
|
string str3;
|
|
string filexPos;
|
|
string filezPos;
|
|
string pak_name;
|
|
//string* dir_array;
|
|
int dir_num_files = 0;
|
|
int ctr = 0;
|
|
|
|
if(dp = opendir("region/")){
|
|
while(entry = readdir(dp)){
|
|
entrys = entry->d_name;
|
|
found = entrys.find(".mcr");
|
|
if ( (found!=string::npos) && (entry->d_type==8) ) {
|
|
str2 = entrys.substr (2,sizeof(entrys));
|
|
filexPos = str2.substr (0,(int)str2.find("."));
|
|
str3 = str2.substr ((int)str2.find(".")+1, sizeof(str2));
|
|
filezPos = str3.substr (0,(int)str3.find("."));
|
|
pak_name = "X" + filexPos + "_Z" + filezPos + ".pak";
|
|
//printf ("pak_name: %s\n", pak_name.c_str());
|
|
|
|
// } //moving down to test
|
|
// } //mvong down to test
|
|
// closedir(dp); //moving down to test
|
|
//} //moving down to test
|
|
|
|
char SourceFile[128];
|
|
char OutputFile[128];
|
|
//ifstream file ("region/r.0.0.mcr", ios::in|ios::binary|ios::ate);
|
|
|
|
clock_t begin=clock(); //start execution timer
|
|
|
|
//need to add code to search through region/ directory and load contents of filenames into array.
|
|
//for each file in array do the following:
|
|
FILE* f = 0;
|
|
FILE* wf = 0;
|
|
#ifdef _WIN32
|
|
sprintf_s(SourceFile, 128, "region/%s",entrys.c_str()); //replace hard coded file with file array variable
|
|
sprintf_s(OutputFile, 128, "world/%s",pak_name.c_str()); //parce x and z from file array variable and place into pak file format
|
|
if( fopen_s(&wf, OutputFile, "wb" ) == 0 ) {} else { cout << "uhoh!" << endl; return 0; } //open new pak file for writing
|
|
#else
|
|
sprintf(SourceFile, "region/%s",entrys.c_str()); //same as above but for linux
|
|
sprintf(OutputFile, "world/%s",pak_name.c_str());
|
|
if( (wf = fopen(OutputFile, "wb" )) != 0 ) {} else { cout << "uhoh!" << endl; return 0; }
|
|
#endif
|
|
|
|
|
|
printf ("Now Converting %s to %s\n", entrys.c_str(), pak_name.c_str() );
|
|
if( (f = fopen(SourceFile, "rb" )) != 0 ) { // no error
|
|
|
|
char* t_FakeHeader;
|
|
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 PakVersion = 1;
|
|
char ChunkVersion = 1;
|
|
short NumChunks = 0;
|
|
|
|
unsigned char byte1 = 0;
|
|
unsigned char byte2 = 0;
|
|
unsigned char byte3 = 0;
|
|
unsigned char byte4 = 0;
|
|
unsigned char byte5 = 0;
|
|
unsigned char trash = 0;
|
|
unsigned int frloc = 0;
|
|
|
|
int toffset = 0;
|
|
int compdlength = 0;
|
|
|
|
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
|
|
//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 ) { cout << "ERROR 21hs READING FROM FILE " << SourceFile; fclose(f); return false; }
|
|
if( fread( &byte2, sizeof(byte2), 1, f) != 1 ) { cout << "ERROR ks93 READING FROM FILE " << SourceFile; fclose(f); return false; }
|
|
if( fread( &byte3, sizeof(byte3), 1, f) != 1 ) { 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 ) { cout << "ERROR dhj3 READING FROM FILE " << SourceFile; 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.
|
|
}
|
|
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 ) { cout << "ERROR 2jkd READING FROM FILE " << SourceFile; 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.
|
|
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) { 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 ) { cout << "ERROR 2t32 READING FROM FILE " << SourceFile; fclose(f); return false; }
|
|
if( fread( &byte2, sizeof(byte2), 1, f) != 1 ) { cout << "ERROR 2y51 READING FROM FILE " << SourceFile; fclose(f); return false; }
|
|
if( fread( &byte3, sizeof(byte3), 1, f) != 1 ) { cout << "ERROR 3424 READING FROM FILE " << SourceFile; fclose(f); return false; }
|
|
if( fread( &byte4, sizeof(byte4), 1, f) != 1 ) { 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 ) { cout << "ERROR 2341 READING FROM FILE " << SourceFile; fclose(f); return false; } //compression type, 1 = GZip (RFC1952) (unused in practice) , 2 = Zlib (RFC1950)
|
|
|
|
//printf("byte1: %i\n", byte1);
|
|
//printf("byte2: %i\n", byte2);
|
|
//printf("byte3: %i\n", byte3);
|
|
//printf("byte4: %i\n", byte4);
|
|
//printf("byte5: %i\n", byte5);
|
|
|
|
frloc += 5; //moved ahead 5 bytes while reading data.
|
|
|
|
// TODO - delete [] temparr after you're done with it, now it's a memory leak
|
|
char* compBlockData = new char[compdlength]; //can't get fread to read more than one char at a time into a char array... so that's what I'll do. :( At least it works.
|
|
if( fread( compBlockData, compdlength, 1, f) != 1 ) { cout << "ERROR rf22 READING FROM FILE " << SourceFile; fclose(f); return false; }
|
|
frloc = frloc + compdlength;
|
|
|
|
uLongf DestSize = 98576;// 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;
|
|
};
|
|
}
|
|
|
|
|
|
|
|
//testing of nbtparser.
|
|
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");
|
|
char* xtemppos = (char *)&xPos;
|
|
int zPos = NBTData->GetInteger("zPos");
|
|
char* ztemppos = (char *)&zPos;
|
|
|
|
|
|
for(unsigned int i = 0; i < sizeof(int); i++) {
|
|
t_FakeHeader[t_FakeHeaderSz+1] = xtemppos[i];
|
|
t_FakeHeaderSz ++;
|
|
}
|
|
for(unsigned int i = 0; i < sizeof(int); i++) {
|
|
t_FakeHeader[t_FakeHeaderSz+1] = ztemppos[i];
|
|
t_FakeHeaderSz ++;
|
|
}
|
|
|
|
//todo: inserert json code and add it to chunk data
|
|
for(unsigned int i = 0; i < 32768; i++) {
|
|
UncompressedChunk[ UnChunkArrLoc ] = NBTData->GetByteArray("Blocks")[i];
|
|
UnChunkArrLoc ++;
|
|
}
|
|
for(unsigned int i = 0; i < 16384; i++) {
|
|
UncompressedChunk[ UnChunkArrLoc ] = NBTData->GetByteArray("Data")[i];
|
|
UnChunkArrLoc ++;
|
|
}
|
|
for(unsigned int i = 0; i < 16384; i++) {
|
|
UncompressedChunk[ UnChunkArrLoc ] = NBTData->GetByteArray("BlockLight")[i];
|
|
UnChunkArrLoc ++;
|
|
}
|
|
for(unsigned int i = 0; i < 16384; i++) {
|
|
UncompressedChunk[ UnChunkArrLoc ] = NBTData->GetByteArray("SkyLight")[i];
|
|
UnChunkArrLoc ++;
|
|
}
|
|
|
|
errorcode = compress2( (Bytef*)CompressedChunk, &CompressedSize, (const Bytef*)UncompressedChunk, UncompressedChunkSz, Z_DEFAULT_COMPRESSION);
|
|
if( errorcode != Z_OK )
|
|
{
|
|
printf("Error compressing data (%i)", errorcode );
|
|
break;
|
|
}
|
|
|
|
char* c_sizetemppos = (char *)&CompressedSize;
|
|
char* uc_sizetemppos = (char *)&UncompressedChunkSz;
|
|
|
|
for(unsigned int i = 0; i < sizeof(int); i++) {
|
|
t_FakeHeader[t_FakeHeaderSz+1] = c_sizetemppos[i];
|
|
t_FakeHeaderSz ++;
|
|
}
|
|
for(unsigned int i = 0; i < sizeof(int); i++) {
|
|
t_FakeHeader[t_FakeHeaderSz+1] = uc_sizetemppos[i];
|
|
t_FakeHeaderSz ++;
|
|
}
|
|
for(unsigned int i = 0; i < CompressedSize; i++) {
|
|
t_CompChunk[t_CompChunkSz+1] = CompressedChunk[i];
|
|
t_CompChunkSz ++;
|
|
}
|
|
|
|
|
|
//printf("Coord(X,Z):ChunkSize: %i,%i:%i\n", NBTData->GetInteger("xPos"), NBTData->GetInteger("zPos"), (int)CompressedSize );
|
|
|
|
NBTData->CloseCompound();// Close the compounds after you're done
|
|
NBTData->CloseCompound();
|
|
|
|
//fwrite( BlockData, DestSize, 1, wf );
|
|
|
|
delete [] UncompressedChunk;
|
|
delete [] CompressedChunk;
|
|
delete [] compBlockData;
|
|
delete [] BlockData;
|
|
|
|
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 ) { cout << "ERROR 2nkd READING FROM FILE " << SourceFile; fclose(f); return false; }
|
|
frloc ++;
|
|
}
|
|
|
|
}
|
|
} //only run chunk # 3
|
|
|
|
}
|
|
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;
|
|
|
|
fclose(wf); //close file.
|
|
fclose(f); //close file.
|
|
}
|
|
|
|
clock_t end=clock();
|
|
cout << "Time elapsed: " << double(diffclock(end,begin)) << " ms"<< endl;
|
|
|
|
|
|
} //moving down to test
|
|
} //mvong down to test
|
|
closedir(dp); //moving down to test
|
|
} //moving down to test
|
|
|
|
return 0;
|
|
|
|
|
|
}
|