void CRIFFChunkTreeDlg::OnSave()
{
OPENFILENAME ofn;
PrepareSimpleDialog(&ofn, *this, "*.txt");
ofn.lpstrFilter = "Text files (*.txt)|*.txt||";
ofn.Flags |= OFN_OVERWRITEPROMPT;
if (!GetOpenSaveFileNameUTF8(&ofn, false))
return;
CFileStream* destFile = new CFileStream();
if (STREAM_OK == destFile->Open(ofn.lpstrFile, StreamMode::Write))
{
CACHE* cache = new CACHE(4, 1<<18);
cache->Open(destFile, CACHE_OPEN_WRITE);
cache->Write(cUTF8Hdr, 3);
RenderItem(cache, m_Tree.GetRootItem(), 0);
cache->Close();
delete cache;
destFile->Close();
delete destFile;
}
else
{
MessageBoxHelper::CouldNotOpenFileForWrite(ofn.lpstrFile);
}
}
void CGlobalMshLoader::SaveMeshToFile(const CMesh::TMeshContainer &mesh, const nstring &file)
{
CFileStream sfile;
sfile.Open(file, true, false);
SaveMeshToStream(mesh, &sfile);
sfile.Close();
}
CMesh::TMeshContainer &CGlobalMshLoader::LoadMeshFromFile(const nstring &file)
{
CFileStream sfile;
sfile.Open(file);
LoadMeshFromStream(&sfile);
sfile.Close();
return gMeshBuffer;
}
DWORD CDataServer::LoadFileToBuffer(const nString& FileName, char*& Buffer)
{
CFileStream File;
int BytesRead;
if (File.Open(FileName, SAM_READ, SAP_SEQUENTIAL))
{
int FileSize = File.GetSize();
Buffer = (char*)n_malloc(FileSize);
BytesRead = File.Read(Buffer, FileSize);
File.Close();
}
else
{
Buffer = NULL;
BytesRead = 0;
}
return BytesRead;
}
// Main entry point
int _tmain(int argc, _TCHAR* argv[])
{
// Add command line alias'
cl.AddAlias(L"h", L"?");
cl.AddAlias(L"help", L"?");
cl.AddAlias(L"x", L"extract");
cl.AddAlias(L"o", L"out");
// Parse command line
cl.Parse(GetCommandLine(), true);
// Display logo...
if (!cl.GetSwitch(L"nologo"))
{
wprintf(L"Structured Ini File Compiler v1.1\nCopyright (C) 2007 Topten Software. All Rights Reserved\n\n");
}
// Display help
if (cl.GetSwitch(L"?"))
{
cl.BuildHelp(L"nologo", L"Suppress display of copyright and version info", NULL, clOptional);
cl.BuildHelp(L"?|h|help", L"Display help information", NULL, clOptional);
cl.BuildHelp(L"i", L"Include path", L"path", clMulti|clValue|clOptional);
cl.BuildHelp(L"x|extract", L"Extract (decompile)", NULL, clOptional);
cl.BuildHelp(L"unicode", L"Decompile to unicode", NULL, clOptional);
cl.BuildHelp(L"flat", L"Decompile to flat", NULL, clOptional);
cl.BuildHelp(L"o|out", L"Output File", NULL, clOptional);
cl.BuildHelp(L"InputFile", L"File to be processed", NULL, clPlaced);
cl.BuildHelp(L"crcstamp", L"CRC Stamp Module", NULL, clOptional);
cl.BuildHelp(L"crcverify", L"Check Module CRC Stamp", NULL, clOptional);
wprintf(cl.GetHelp(L"Sinic"));
return 7;
}
// Get input value
CUniString strInputFile;
cl.GetPlacedValue(0, L"InputFile", strInputFile);
strInputFile=QualifyPath(strInputFile);
if (cl.GetSwitch(L"crcverify"))
{
if (!DoesFileExist(strInputFile))
{
wprintf(L"Verifying %s... file doesn't exist, ignoring\n", strInputFile);
return 0;
}
else
{
if (!CheckModuleCRC(strInputFile))
{
CheckModuleCRCWithErrors(strInputFile);
wprintf(L"************************************************************\n");
wprintf(L" Verifying %s... FAILED\n", strInputFile);
wprintf(L"************************************************************\n");
return 7;
}
else
{
wprintf(L"Verifying %s... OK\n", strInputFile);
return 0;
}
}
}
// Module CRC Stamp?
if (cl.GetSwitch(L"crcstamp"))
{
// Work out output file
CUniString strOutputFile;
cl.GetValue(L"out", strOutputFile);
if (strOutputFile.IsEmpty())
strOutputFile=strInputFile;
// Error handling...
cl.CheckNoUnknownArgs();
if (cl.HasError())
{
wprintf(cl.GetErrorMessage());
wprintf(L"\n\n");
return 7;
}
// Open file for read/write
FILE* pFile;
if (_wfopen_s(&pFile, strInputFile, L"rb")!=0)
{
wprintf(L"Error: Can't open file '%s'\n", strInputFile);
return 7;
}
// Get file size
fseek(pFile, 0, SEEK_END);
DWORD dwLength=ftell(pFile);
fseek(pFile, 0, SEEK_SET);
// Check has length
if (dwLength==0)
{
fclose(pFile);
wprintf(L"Error: Zero length file\n");
return 7;
}
// Load file into memory
LPBYTE pFileData=(LPBYTE)malloc(dwLength);
// Read data
if (fread(pFileData, dwLength, 1, pFile)!=1)
{
free(pFileData);
fclose(pFile);
wprintf(L"Error: Error reading file\n");
return 7;
}
// Close the file
fclose(pFile);
if (CountCRCControlBlocks(pFileData, dwLength)>1)
{
free(pFileData);
fclose(pFile);
wprintf(L"Error: File contains multiple CRC Control blocks!\n");
return 7;
}
// Locate CRC Control block
CRCCONTROLBLOCK* pCRCBlock=LocateCRCControlBlock(pFileData, dwLength);
if (!pCRCBlock)
{
free(pFileData);
fclose(pFile);
wprintf(L"Error: File doesn't contain CRC Control block!\n");
return 7;
}
wprintf(L"CRC Control Block located at offset: 0x%.8x\n", reinterpret_cast<LPBYTE>(pCRCBlock)-pFileData);
// Calculate the CRC upto start of control block
pCRCBlock->dwCRC=CalculateModuleCRC(pFileData, dwLength, pCRCBlock);
pCRCBlock->dwModuleLength=dwLength;
wprintf(L"Module length: 0x%.8x\n", dwLength);
wprintf(L"CRC: 0x%.8x\n", pCRCBlock->dwCRC);
pFile=NULL;
if (_wfopen_s(&pFile, strOutputFile, L"wb")!=0)
{
wprintf(L"Error: Can't open output file '%s'\n", strOutputFile);
return 7;
}
if (!CheckModuleCRC(pFileData, dwLength))
{
wprintf(L"Error: double check failed\n");
return 7;
}
// Write CRC stamped
fwrite(pFileData, dwLength, 1, pFile);
// Clean up
free(pFileData);
fclose(pFile);
if (!CheckModuleCRC(strOutputFile))
{
CheckModuleCRCWithErrors(strOutputFile);
return 7;
}
wprintf(L"CRC Stamped %s - OK\n\n", strOutputFile);
return 0;
}
// Compile or decompile
bool bDecompile=cl.GetSwitch(L"extract");
bool bUnicode=cl.GetSwitch(L"unicode");
bool bFlat=cl.GetSwitch(L"flat");
// Get output file
CUniString strOutputFile;
cl.GetValue(L"out", strOutputFile);
if (strOutputFile.IsEmpty())
{
strOutputFile=ChangeFileExtension(strInputFile.IsEmpty() ? L"" : strInputFile, bDecompile ? L"sini" : L"bini");
}
strOutputFile=QualifyPath(strOutputFile);
// Setup file content provider with include paths
CFileContentProvider fp;
CVector<CUniString> vecIncludePaths;
cl.GetMultiValue(L"i", vecIncludePaths);
for (int i=0; i<vecIncludePaths.GetSize(); i++)
{
CUniString str=QualifyPath(vecIncludePaths[i]);
fp.AddIncludePath(str);
}
// Get defines
CVector<CUniString> vecDefines;
cl.GetMultiValue(L"d", vecDefines);
// Error handling...
cl.CheckNoUnknownArgs();
if (cl.HasError())
{
wprintf(cl.GetErrorMessage());
wprintf(L"\n\n");
return 7;
}
if (!bDecompile)
{
// Load the input file
CProfileFile file;
for (int i=0; i<vecDefines.GetSize(); i++)
{
CUniString strName, strValue;
SplitString(vecDefines[i], L"=", strName, strValue);
wprintf(L"#define %s %s\n", strName, strValue);
file.Define(strName, strValue);
}
if (!file.Load(strInputFile, &fp))
{
HRESULT hr=HRESULT_FROM_WIN32(GetLastError());
wprintf(file.GetParseError());
return 7;
}
// Extract encryption key
DWORD dwEncryptionKey=0;
CProfileSection* pSection=file.FindSection(L"Sinic");
if (pSection)
{
// Save the key
dwEncryptionKey=pSection->GetIntValue(L"EncryptionKey", 0);
// Remove the "sinic" section
file.Remove(pSection);
}
// Create output file
CFileStream OutStream;
if (!OutStream.Create(strOutputFile))
{
HRESULT hr=HRESULT_FROM_WIN32(GetLastError());
wprintf(L"Failed to create '%s' - %s (0x%.8x)\n\n", strOutputFile, FormatError(hr), hr);
return 7;
}
// Save as binary file
HRESULT hr;
if (dwEncryptionKey)
{
CAutoPtr<IStream, SRefCounted> spEncStream;
CreateCryptorStream(&OutStream, CCryptorKey(dwEncryptionKey), &spEncStream);
hr=SaveBinaryProfile(file, spEncStream);
}
else
{
hr=SaveBinaryProfile(file, &OutStream);
}
OutStream.Close();
if (FAILED(hr))
{
HRESULT hr=HRESULT_FROM_WIN32(GetLastError());
wprintf(L"Failed to save '%s' - %s (0x%.8x)\n\n", strOutputFile, FormatError(hr), hr);
DeleteFile(strOutputFile);
return 7;
}
}
else
{
// Open stream
CFileStream InStream;
if (!InStream.Open(strInputFile))
{
HRESULT hr=HRESULT_FROM_WIN32(GetLastError());
wprintf(L"Failed to open '%s' - %s (0x%.8x)\n\n", strInputFile, FormatError(hr), hr);
return 7;
}
// Load profile
CProfileFile file;
HRESULT hr=LoadBinaryProfile(file, &InStream);
if (FAILED(hr))
{
if (hr!=E_UNEXPECTED)
{
wprintf(L"Failed to load '%s' - %s (0x%.8x)\n\n", strInputFile, FormatError(hr), hr);
return 7;
}
else
{
wprintf(L"Failed to load '%s' - not a binary profile file\n\n", strInputFile);
return 7;
}
}
// Save as heirarchial
if (!file.Save(strOutputFile, bUnicode, !bFlat))
{
HRESULT hr=HRESULT_FROM_WIN32(GetLastError());
wprintf(L"Failed to save '%s' - %s (0x%.8x)\n\n", strInputFile, FormatError(hr), hr);
return 7;
}
}
wprintf(L"%s - OK\n\n", strOutputFile);
return 0;
}