|
|
|
/*==LICENSE==*
|
|
|
|
|
|
|
|
CyanWorlds.com Engine - MMOG client, server and tools
|
|
|
|
Copyright (C) 2011 Cyan Worlds, Inc.
|
|
|
|
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
|
|
it under the terms of the GNU General Public License as published by
|
|
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
|
|
(at your option) any later version.
|
|
|
|
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
GNU General Public License for more details.
|
|
|
|
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
|
|
|
|
Additional permissions under GNU GPL version 3 section 7
|
|
|
|
|
|
|
|
If you modify this Program, or any covered work, by linking or
|
|
|
|
combining it with any of RAD Game Tools Bink SDK, Autodesk 3ds Max SDK,
|
|
|
|
NVIDIA PhysX SDK, Microsoft DirectX SDK, OpenSSL library, Independent
|
|
|
|
JPEG Group JPEG library, Microsoft Windows Media SDK, or Apple QuickTime SDK
|
|
|
|
(or a modified version of those libraries),
|
|
|
|
containing parts covered by the terms of the Bink SDK EULA, 3ds Max EULA,
|
|
|
|
PhysX SDK EULA, DirectX SDK EULA, OpenSSL and SSLeay licenses, IJG
|
|
|
|
JPEG Library README, Windows Media SDK EULA, or QuickTime SDK EULA, the
|
|
|
|
licensors of this Program grant you additional
|
|
|
|
permission to convey the resulting work. Corresponding Source for a
|
|
|
|
non-source form of such a combination shall include the source code for
|
|
|
|
the parts of OpenSSL and IJG JPEG Library used as well as that of the covered
|
|
|
|
work.
|
|
|
|
|
|
|
|
You can contact Cyan Worlds, Inc. by email legal@cyan.com
|
|
|
|
or by snail mail at:
|
|
|
|
Cyan Worlds, Inc.
|
|
|
|
14617 N Newport Hwy
|
|
|
|
Mead, WA 99021
|
|
|
|
|
|
|
|
*==LICENSE==*/
|
|
|
|
#include "plSecureStream.h"
|
|
|
|
#include "hsWindows.h"
|
|
|
|
|
|
|
|
#include "plFileUtils.h"
|
|
|
|
#include "hsSTLStream.h"
|
|
|
|
|
|
|
|
#include <time.h>
|
|
|
|
|
|
|
|
#if !HS_BUILD_FOR_WIN32
|
|
|
|
#define INVALID_HANDLE_VALUE 0
|
|
|
|
#endif
|
|
|
|
|
|
|
|
// our default encryption key
|
|
|
|
const uint32_t plSecureStream::kDefaultKey[4] = { 0x6c0a5452, 0x3827d0f, 0x3a170b92, 0x16db7fc2 };
|
|
|
|
|
|
|
|
static const int kEncryptChunkSize = 8;
|
|
|
|
|
|
|
|
static const char* kMagicString = "notthedroids";
|
|
|
|
static const int kMagicStringLen = 12;
|
|
|
|
|
|
|
|
static const int kFileStartOffset = kMagicStringLen + sizeof(uint32_t);
|
|
|
|
|
|
|
|
static const int kMaxBufferedFileSize = 10*1024;
|
|
|
|
|
|
|
|
const char plSecureStream::kKeyFilename[] = "encryption.key";
|
|
|
|
const wchar_t plSecureStream::kWKeyFilename[] = L"encryption.key";
|
|
|
|
|
|
|
|
plSecureStream::plSecureStream(bool deleteOnExit, uint32_t* key) :
|
|
|
|
fRef(INVALID_HANDLE_VALUE),
|
|
|
|
fActualFileSize(0),
|
|
|
|
fBufferedStream(false),
|
|
|
|
fRAMStream(nil),
|
|
|
|
fWriteFileName(nil),
|
|
|
|
fOpenMode(kOpenFail),
|
|
|
|
fDeleteOnExit(deleteOnExit)
|
|
|
|
{
|
|
|
|
if (key)
|
|
|
|
memcpy(&fKey, key, sizeof(kDefaultKey));
|
|
|
|
else
|
|
|
|
memcpy(&fKey, &kDefaultKey, sizeof(kDefaultKey));
|
|
|
|
}
|
|
|
|
|
|
|
|
plSecureStream::plSecureStream(hsStream* base, uint32_t* key) :
|
|
|
|
fRef(INVALID_HANDLE_VALUE),
|
|
|
|
fActualFileSize(0),
|
|
|
|
fBufferedStream(false),
|
|
|
|
fRAMStream(nil),
|
|
|
|
fWriteFileName(nil),
|
|
|
|
fOpenMode(kOpenFail),
|
|
|
|
fDeleteOnExit(false)
|
|
|
|
{
|
|
|
|
if (key)
|
|
|
|
memcpy(&fKey, key, sizeof(kDefaultKey));
|
|
|
|
else
|
|
|
|
memcpy(&fKey, &kDefaultKey, sizeof(kDefaultKey));
|
|
|
|
Open(base);
|
|
|
|
}
|
|
|
|
|
|
|
|
plSecureStream::~plSecureStream()
|
|
|
|
{
|
|
|
|
}
|
|
|
|
|
|
|
|
//
|
|
|
|
// XXTEA
|
|
|
|
// http://www-users.cs.york.ac.uk/~matthew/TEA/
|
|
|
|
//
|
|
|
|
// A potential weakness in this implementation is the fact that a known value
|
|
|
|
// (length of the original file) is written at the start of the encrypted file. -Colin
|
|
|
|
//
|
|
|
|
|
|
|
|
#define MX (z>>5 ^ y<<2) + (y>>3 ^ z<<4) ^ (sum^y) + (fKey[p&3^e]^z)
|
|
|
|
|
|
|
|
void plSecureStream::IEncipher(uint32_t* const v, uint32_t n)
|
|
|
|
{
|
|
|
|
register unsigned long y=v[0], z=v[n-1], e, delta=0x9E3779B9;
|
|
|
|
register unsigned long q = 6 + 52/n, p, sum = 0;
|
|
|
|
|
|
|
|
while (q-- > 0)
|
|
|
|
{
|
|
|
|
sum += delta;
|
|
|
|
e = (sum >> 2) & 3;
|
|
|
|
for (p = 0; p < n - 1; p++)
|
|
|
|
{
|
|
|
|
y = v[p + 1];
|
|
|
|
v[p] += MX;
|
|
|
|
z = v[p];
|
|
|
|
}
|
|
|
|
y = v[0];
|
|
|
|
v[n - 1] += MX;
|
|
|
|
z = v[n - 1];
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void plSecureStream::IDecipher(uint32_t* const v, uint32_t n)
|
|
|
|
{
|
|
|
|
register unsigned long y=v[0], z=v[n-1], e, delta=0x9E3779B9;
|
|
|
|
register unsigned long q = 6 + 52/n, p, sum = q * delta;
|
|
|
|
|
|
|
|
while (sum > 0)
|
|
|
|
{
|
|
|
|
e = (sum >> 2) & 3;
|
|
|
|
for (p = n - 1; p > 0; p--)
|
|
|
|
{
|
|
|
|
z = v[p - 1];
|
|
|
|
v[p] -= MX;
|
|
|
|
y = v[p];
|
|
|
|
}
|
|
|
|
z = v[n - 1];
|
|
|
|
v[0] -= MX;
|
|
|
|
y = v[0];
|
|
|
|
sum -= delta;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
bool plSecureStream::Open(const char* name, const char* mode)
|
|
|
|
{
|
|
|
|
wchar_t* wName = hsStringToWString(name);
|
|
|
|
wchar_t* wMode = hsStringToWString(mode);
|
|
|
|
bool ret = Open(wName, wMode);
|
|
|
|
delete [] wName;
|
|
|
|
delete [] wMode;
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool plSecureStream::Open(const wchar_t* name, const wchar_t* mode)
|
|
|
|
{
|
|
|
|
if (wcscmp(mode, L"rb") == 0)
|
|
|
|
{
|
|
|
|
#if HS_BUILD_FOR_WIN32
|
|
|
|
if (fDeleteOnExit)
|
|
|
|
{
|
|
|
|
fRef = CreateFileW(name,
|
|
|
|
GENERIC_READ, // open for reading
|
|
|
|
0, // no one can open the file until we're done
|
|
|
|
NULL, // default security
|
|
|
|
OPEN_EXISTING, // only open existing files (no creation)
|
|
|
|
FILE_FLAG_DELETE_ON_CLOSE, // delete the file from disk when we close the handle
|
|
|
|
NULL); // no template
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
fRef = CreateFileW(name,
|
|
|
|
GENERIC_READ, // open for reading
|
|
|
|
0, // no one can open the file until we're done
|
|
|
|
NULL, // default security
|
|
|
|
OPEN_EXISTING, // only open existing files (no creation)
|
|
|
|
FILE_ATTRIBUTE_NORMAL, // normal file attributes
|
|
|
|
NULL); // no template
|
|
|
|
}
|
|
|
|
|
|
|
|
fPosition = 0;
|
|
|
|
|
|
|
|
if (fRef == INVALID_HANDLE_VALUE)
|
|
|
|
return false;
|
|
|
|
|
|
|
|
// Make sure our special magic string is there
|
|
|
|
if (!ICheckMagicString(fRef))
|
|
|
|
{
|
|
|
|
CloseHandle(fRef);
|
|
|
|
fRef = INVALID_HANDLE_VALUE;
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
DWORD numBytesRead;
|
|
|
|
ReadFile(fRef, &fActualFileSize, sizeof(uint32_t), &numBytesRead, NULL);
|
|
|
|
#elif HS_BUILD_FOR_UNIX
|
|
|
|
const char* cname = hsWStringToString(name);
|
|
|
|
fRef = fopen(cname, "rb");
|
|
|
|
delete[] cname;
|
|
|
|
|
|
|
|
fPosition = 0;
|
|
|
|
|
|
|
|
if (fRef == INVALID_HANDLE_VALUE)
|
|
|
|
return false;
|
|
|
|
|
|
|
|
if (!ICheckMagicString(fRef))
|
|
|
|
{
|
|
|
|
fclose(fRef);
|
|
|
|
fRef = INVALID_HANDLE_VALUE;
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
#endif
|
|
|
|
|
|
|
|
// The encrypted stream is inefficient if you do reads smaller than
|
|
|
|
// 8 bytes. Since we do a lot of those, any file under a size threshold
|
|
|
|
// is buffered in memory
|
|
|
|
if (fActualFileSize <= kMaxBufferedFileSize)
|
|
|
|
IBufferFile();
|
|
|
|
|
|
|
|
fOpenMode = kOpenRead;
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
else if (wcscmp(mode, L"wb") == 0)
|
|
|
|
{
|
|
|
|
fRAMStream = new hsVectorStream;
|
|
|
|
fWriteFileName = new wchar_t[wcslen(name) + 1];
|
|
|
|
wcscpy(fWriteFileName, name);
|
|
|
|
fPosition = 0;
|
|
|
|
|
|
|
|
fOpenMode = kOpenWrite;
|
|
|
|
fBufferedStream = true;
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
hsAssert(0, "Unsupported open mode");
|
|
|
|
fOpenMode = kOpenFail;
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
bool plSecureStream::Open(hsStream* stream)
|
|
|
|
{
|
|
|
|
uint32_t pos = stream->GetPosition();
|
|
|
|
stream->Rewind();
|
|
|
|
if (!ICheckMagicString(stream))
|
|
|
|
return false;
|
|
|
|
|
|
|
|
fActualFileSize = stream->ReadLE32();
|
|
|
|
uint32_t trimSize = kMagicStringLen + sizeof(uint32_t) + fActualFileSize;
|
|
|
|
fRAMStream = new hsRAMStream;
|
|
|
|
while (!stream->AtEnd())
|
|
|
|
{
|
|
|
|
// Don't write out any garbage
|
|
|
|
uint32_t size;
|
|
|
|
if ((trimSize - stream->GetPosition()) < kEncryptChunkSize)
|
|
|
|
size = (trimSize - stream->GetPosition());
|
|
|
|
else
|
|
|
|
size = kEncryptChunkSize;
|
|
|
|
|
|
|
|
uint8_t buf[kEncryptChunkSize];
|
|
|
|
stream->Read(kEncryptChunkSize, &buf);
|
|
|
|
IDecipher((uint32_t*)&buf, kEncryptChunkSize / sizeof(uint32_t));
|
|
|
|
fRAMStream->Write(size, &buf);
|
|
|
|
}
|
|
|
|
|
|
|
|
stream->SetPosition(pos);
|
|
|
|
fRAMStream->Rewind();
|
|
|
|
fPosition = 0;
|
|
|
|
fBufferedStream = true;
|
|
|
|
fOpenMode = kOpenRead;
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool plSecureStream::Close()
|
|
|
|
{
|
|
|
|
int rtn = false;
|
|
|
|
|
|
|
|
if (fOpenMode == kOpenWrite)
|
|
|
|
{
|
|
|
|
fRAMStream->Rewind();
|
|
|
|
rtn = IWriteEncrypted(fRAMStream, fWriteFileName);
|
|
|
|
}
|
|
|
|
if (fRef != INVALID_HANDLE_VALUE)
|
|
|
|
{
|
|
|
|
#if HS_BUILD_FOR_WIN32
|
|
|
|
rtn = CloseHandle(fRef);
|
|
|
|
#elif HS_BUILD_FOR_UNIX
|
|
|
|
rtn = fclose(fRef);
|
|
|
|
#endif
|
|
|
|
fRef = INVALID_HANDLE_VALUE;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (fRAMStream)
|
|
|
|
{
|
|
|
|
delete fRAMStream;
|
|
|
|
fRAMStream = nil;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (fWriteFileName)
|
|
|
|
{
|
|
|
|
delete [] fWriteFileName;
|
|
|
|
fWriteFileName = nil;
|
|
|
|
}
|
|
|
|
|
|
|
|
fActualFileSize = 0;
|
|
|
|
fBufferedStream = false;
|
|
|
|
fOpenMode = kOpenFail;
|
|
|
|
|
|
|
|
return rtn;
|
|
|
|
}
|
|
|
|
|
|
|
|
uint32_t plSecureStream::IRead(uint32_t bytes, void* buffer)
|
|
|
|
{
|
|
|
|
if (fRef == INVALID_HANDLE_VALUE)
|
|
|
|
return 0;
|
|
|
|
uint32_t numItems;
|
|
|
|
#if HS_BUILD_FOR_WIN32
|
|
|
|
bool success = (ReadFile(fRef, buffer, bytes, (LPDWORD)&numItems, NULL) != 0);
|
|
|
|
#elif HS_BUILD_FOR_UNIX
|
|
|
|
numItems = fread(buffer, bytes, 1, fRef);
|
|
|
|
bool success = numItems != 0;
|
|
|
|
#endif
|
|
|
|
fBytesRead += numItems;
|
|
|
|
fPosition += numItems;
|
|
|
|
if ((unsigned)numItems < bytes)
|
|
|
|
{
|
|
|
|
if (success)
|
|
|
|
{
|
|
|
|
// EOF ocurred
|
|
|
|
char str[128];
|
|
|
|
sprintf(str, "Hit EOF on Windows read, only read %d out of requested %d bytes\n", numItems, bytes);
|
|
|
|
hsDebugMessage(str, 0);
|
|
|
|
}
|
|
|
|
else
|
|
|
|
{
|
|
|
|
hsDebugMessage("Error on Windows read", GetLastError());
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return numItems;
|
|
|
|
}
|
|
|
|
|
|
|
|
void plSecureStream::IBufferFile()
|
|
|
|
{
|
|
|
|
fRAMStream = new hsVectorStream;
|
|
|
|
char buf[1024];
|
|
|
|
while (!AtEnd())
|
|
|
|
{
|
|
|
|
uint32_t numRead = Read(1024, buf);
|
|
|
|
fRAMStream->Write(numRead, buf);
|
|
|
|
}
|
|
|
|
fRAMStream->Rewind();
|
|
|
|
|
|
|
|
fBufferedStream = true;
|
|
|
|
#if HS_BUILD_FOR_WIN32
|
|
|
|
CloseHandle(fRef);
|
|
|
|
#elif HS_BUILD_FOR_UNIX
|
|
|
|
fclose(fRef);
|
|
|
|
#endif
|
|
|
|
fRef = INVALID_HANDLE_VALUE;
|
|
|
|
fPosition = 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool plSecureStream::AtEnd()
|
|
|
|
{
|
|
|
|
if (fBufferedStream)
|
|
|
|
return fRAMStream->AtEnd();
|
|
|
|
else
|
|
|
|
return (GetPosition() == fActualFileSize);
|
|
|
|
}
|
|
|
|
|
|
|
|
void plSecureStream::Skip(uint32_t delta)
|
|
|
|
{
|
|
|
|
if (fBufferedStream)
|
|
|
|
{
|
|
|
|
fRAMStream->Skip(delta);
|
|
|
|
fPosition = fRAMStream->GetPosition();
|
|
|
|
}
|
|
|
|
else if (fRef != INVALID_HANDLE_VALUE)
|
|
|
|
{
|
|
|
|
fBytesRead += delta;
|
|
|
|
fPosition += delta;
|
|
|
|
#if HS_BUILD_FOR_WIN32
|
|
|
|
SetFilePointer(fRef, delta, 0, FILE_CURRENT);
|
|
|
|
#elif HS_BUILD_FOR_UNIX
|
|
|
|
fseek(fRef, delta, SEEK_CUR);
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void plSecureStream::Rewind()
|
|
|
|
{
|
|
|
|
if (fBufferedStream)
|
|
|
|
{
|
|
|
|
fRAMStream->Rewind();
|
|
|
|
fPosition = fRAMStream->GetPosition();
|
|
|
|
}
|
|
|
|
else if (fRef != INVALID_HANDLE_VALUE)
|
|
|
|
{
|
|
|
|
fBytesRead = 0;
|
|
|
|
fPosition = 0;
|
|
|
|
#if HS_BUILD_FOR_WIN32
|
|
|
|
SetFilePointer(fRef, kFileStartOffset, 0, FILE_BEGIN);
|
|
|
|
#elif HS_BUILD_FOR_UNIX
|
|
|
|
fseek(fRef, kFileStartOffset, SEEK_SET);
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
void plSecureStream::FastFwd()
|
|
|
|
{
|
|
|
|
if (fBufferedStream)
|
|
|
|
{
|
|
|
|
fRAMStream->FastFwd();
|
|
|
|
fPosition = fRAMStream->GetPosition();
|
|
|
|
}
|
|
|
|
else if (fRef != INVALID_HANDLE_VALUE)
|
|
|
|
{
|
|
|
|
#if HS_BUILD_FOR_WIN32
|
|
|
|
fBytesRead = fPosition = SetFilePointer(fRef, kFileStartOffset + fActualFileSize, 0, FILE_BEGIN);
|
|
|
|
#elif HS_BUILD_FOR_UNIX
|
|
|
|
fBytesRead = fPosition = fseek(fRef, 0, SEEK_END);
|
|
|
|
#endif
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
uint32_t plSecureStream::GetEOF()
|
|
|
|
{
|
|
|
|
return fActualFileSize;
|
|
|
|
}
|
|
|
|
|
|
|
|
uint32_t plSecureStream::Read(uint32_t bytes, void* buffer)
|
|
|
|
{
|
|
|
|
if (fBufferedStream)
|
|
|
|
{
|
|
|
|
uint32_t numRead = fRAMStream->Read(bytes, buffer);
|
|
|
|
fPosition = fRAMStream->GetPosition();
|
|
|
|
return numRead;
|
|
|
|
}
|
|
|
|
|
|
|
|
uint32_t startPos = fPosition;
|
|
|
|
|
|
|
|
// Offset into the first buffer (0 if we are aligned on a chunk, which means no extra block read)
|
|
|
|
uint32_t startChunkPos = startPos % kEncryptChunkSize;
|
|
|
|
// Amount of data in the partial first chunk (0 if we're aligned)
|
|
|
|
uint32_t startAmt = (startChunkPos != 0) ? hsMinimum(kEncryptChunkSize - startChunkPos, bytes) : 0;
|
|
|
|
|
|
|
|
uint32_t totalNumRead = IRead(bytes, buffer);
|
|
|
|
|
|
|
|
uint32_t numMidChunks = (totalNumRead - startAmt) / kEncryptChunkSize;
|
|
|
|
uint32_t endAmt = (totalNumRead - startAmt) % kEncryptChunkSize;
|
|
|
|
|
|
|
|
// If the start position is in the middle of a chunk we need to rewind and
|
|
|
|
// read that whole chunk in and decrypt it.
|
|
|
|
if (startChunkPos != 0)
|
|
|
|
{
|
|
|
|
// Move to the start of this chunk
|
|
|
|
SetPosition(startPos-startChunkPos);
|
|
|
|
|
|
|
|
// Read in the chunk and decrypt it
|
|
|
|
char buf[kEncryptChunkSize];
|
|
|
|
uint32_t numRead = IRead(kEncryptChunkSize, &buf);
|
|
|
|
IDecipher((uint32_t*)&buf, kEncryptChunkSize / sizeof(uint32_t));
|
|
|
|
|
|
|
|
// Copy the relevant portion to the output buffer
|
|
|
|
memcpy(buffer, &buf[startChunkPos], startAmt);
|
|
|
|
|
|
|
|
SetPosition(startPos+totalNumRead);
|
|
|
|
}
|
|
|
|
|
|
|
|
if (numMidChunks != 0)
|
|
|
|
{
|
|
|
|
uint32_t* bufferPos = (uint32_t*)(((char*)buffer)+startAmt);
|
|
|
|
for (int i = 0; i < numMidChunks; i++)
|
|
|
|
{
|
|
|
|
// Decrypt chunk
|
|
|
|
IDecipher(bufferPos, kEncryptChunkSize / sizeof(uint32_t));
|
|
|
|
bufferPos += (kEncryptChunkSize / sizeof(uint32_t));
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (endAmt != 0)
|
|
|
|
{
|
|
|
|
// Read in the final chunk and decrypt it
|
|
|
|
char buf[kEncryptChunkSize];
|
|
|
|
SetPosition(startPos + startAmt + numMidChunks*kEncryptChunkSize);
|
|
|
|
uint32_t numRead = IRead(kEncryptChunkSize, &buf);
|
|
|
|
IDecipher((uint32_t*)&buf, kEncryptChunkSize / sizeof(uint32_t));
|
|
|
|
|
|
|
|
memcpy(((char*)buffer)+totalNumRead-endAmt, &buf, endAmt);
|
|
|
|
|
|
|
|
SetPosition(startPos+totalNumRead);
|
|
|
|
}
|
|
|
|
|
|
|
|
// If we read into the padding at the end, update the total read to not include that
|
|
|
|
if (totalNumRead > 0 && startPos + totalNumRead > fActualFileSize)
|
|
|
|
{
|
|
|
|
totalNumRead -= (startPos + totalNumRead) - fActualFileSize;
|
|
|
|
SetPosition(fActualFileSize);
|
|
|
|
}
|
|
|
|
|
|
|
|
return totalNumRead;
|
|
|
|
}
|
|
|
|
|
|
|
|
uint32_t plSecureStream::Write(uint32_t bytes, const void* buffer)
|
|
|
|
{
|
|
|
|
if (fOpenMode != kOpenWrite)
|
|
|
|
{
|
|
|
|
hsAssert(0, "Trying to write to a read stream");
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
|
|
|
|
return fRAMStream->Write(bytes, buffer);
|
|
|
|
}
|
|
|
|
|
|
|
|
bool plSecureStream::IWriteEncrypted(hsStream* sourceStream, const wchar_t* outputFile)
|
|
|
|
{
|
|
|
|
hsUNIXStream outputStream;
|
|
|
|
|
|
|
|
if (!outputStream.Open(outputFile, L"wb"))
|
|
|
|
return false;
|
|
|
|
|
|
|
|
outputStream.Write(kMagicStringLen, kMagicString);
|
|
|
|
|
|
|
|
// Save some space to write the file size at the end
|
|
|
|
outputStream.WriteLE32(0);
|
|
|
|
|
|
|
|
// Write out all the full size encrypted blocks we can
|
|
|
|
char buf[kEncryptChunkSize];
|
|
|
|
uint32_t amtRead;
|
|
|
|
while ((amtRead = sourceStream->Read(kEncryptChunkSize, &buf)) == kEncryptChunkSize)
|
|
|
|
{
|
|
|
|
IEncipher((uint32_t*)&buf, kEncryptChunkSize / sizeof(uint32_t));
|
|
|
|
outputStream.Write(kEncryptChunkSize, &buf);
|
|
|
|
}
|
|
|
|
|
|
|
|
// Pad with random data and write out the final partial block, if there is one
|
|
|
|
if (amtRead > 0)
|
|
|
|
{
|
|
|
|
static bool seededRand = false;
|
|
|
|
if (!seededRand)
|
|
|
|
{
|
|
|
|
seededRand = true;
|
|
|
|
srand((unsigned int)time(nil));
|
|
|
|
}
|
|
|
|
|
|
|
|
for (int i = amtRead; i < kEncryptChunkSize; i++)
|
|
|
|
buf[i] = rand();
|
|
|
|
|
|
|
|
IEncipher((uint32_t*)&buf, kEncryptChunkSize / sizeof(uint32_t));
|
|
|
|
|
|
|
|
outputStream.Write(kEncryptChunkSize, &buf);
|
|
|
|
}
|
|
|
|
|
|
|
|
// Write the original file size at the start
|
|
|
|
uint32_t actualSize = sourceStream->GetPosition();
|
|
|
|
outputStream.Rewind();
|
|
|
|
outputStream.Skip(kMagicStringLen);
|
|
|
|
outputStream.WriteLE32(actualSize);
|
|
|
|
|
|
|
|
outputStream.Close();
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool plSecureStream::FileEncrypt(const char* fileName, uint32_t* key /* = nil */)
|
|
|
|
{
|
|
|
|
wchar_t* wFilename = hsStringToWString(fileName);
|
|
|
|
bool ret = FileEncrypt(wFilename, key);
|
|
|
|
delete [] wFilename;
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool plSecureStream::FileEncrypt(const wchar_t* fileName, uint32_t* key /* = nil */)
|
|
|
|
{
|
|
|
|
hsUNIXStream sIn;
|
|
|
|
if (!sIn.Open(fileName))
|
|
|
|
return false;
|
|
|
|
|
|
|
|
// Don't double encrypt any files
|
|
|
|
if (ICheckMagicString(sIn.GetFILE()))
|
|
|
|
{
|
|
|
|
sIn.Close();
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
sIn.Rewind();
|
|
|
|
|
|
|
|
plSecureStream sOut(false, key);
|
|
|
|
bool wroteEncrypted = sOut.IWriteEncrypted(&sIn, L"crypt.dat");
|
|
|
|
|
|
|
|
sIn.Close();
|
|
|
|
sOut.Close();
|
|
|
|
|
|
|
|
if (wroteEncrypted)
|
|
|
|
{
|
|
|
|
plFileUtils::RemoveFile(fileName);
|
|
|
|
plFileUtils::FileMove(L"crypt.dat", fileName);
|
|
|
|
}
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool plSecureStream::FileDecrypt(const char* fileName, uint32_t* key /* = nil */)
|
|
|
|
{
|
|
|
|
wchar_t* wFilename = hsStringToWString(fileName);
|
|
|
|
bool ret = FileDecrypt(wFilename, key);
|
|
|
|
delete [] wFilename;
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool plSecureStream::FileDecrypt(const wchar_t* fileName, uint32_t* key /* = nil */)
|
|
|
|
{
|
|
|
|
plSecureStream sIn(false, key);
|
|
|
|
if (!sIn.Open(fileName))
|
|
|
|
return false;
|
|
|
|
|
|
|
|
hsUNIXStream sOut;
|
|
|
|
if (!sOut.Open(L"crypt.dat", L"wb"))
|
|
|
|
{
|
|
|
|
sIn.Close();
|
|
|
|
return false;
|
|
|
|
}
|
|
|
|
|
|
|
|
char buf[1024];
|
|
|
|
|
|
|
|
while (!sIn.AtEnd())
|
|
|
|
{
|
|
|
|
uint32_t numRead = sIn.Read(sizeof(buf), buf);
|
|
|
|
sOut.Write(numRead, buf);
|
|
|
|
}
|
|
|
|
|
|
|
|
sIn.Close();
|
|
|
|
sOut.Close();
|
|
|
|
|
|
|
|
plFileUtils::RemoveFile(fileName);
|
|
|
|
plFileUtils::FileMove(L"crypt.dat", fileName);
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool plSecureStream::ICheckMagicString(hsStream* s)
|
|
|
|
{
|
|
|
|
char magicString[kMagicStringLen+1];
|
|
|
|
s->Read(kMagicStringLen, &magicString);
|
|
|
|
magicString[kMagicStringLen] = '\0';
|
|
|
|
return (strcmp(magicString, kMagicString) == 0);
|
|
|
|
}
|
|
|
|
|
|
|
|
bool plSecureStream::ICheckMagicString(hsFD fp)
|
|
|
|
{
|
|
|
|
char magicString[kMagicStringLen+1];
|
|
|
|
#ifdef HS_BUILD_FOR_WIN32
|
|
|
|
DWORD numread;
|
|
|
|
ReadFile(fp, &magicString, kMagicStringLen, &numread, NULL);
|
|
|
|
#elif HS_BUILD_FOR_UNIX
|
|
|
|
fread(&magicString, kMagicStringLen, 1, fp);
|
|
|
|
#endif
|
|
|
|
magicString[kMagicStringLen] = '\0';
|
|
|
|
return (strcmp(magicString, kMagicString) == 0);
|
|
|
|
}
|
|
|
|
|
|
|
|
bool plSecureStream::IsSecureFile(const char* fileName)
|
|
|
|
{
|
|
|
|
wchar_t* wFilename = hsStringToWString(fileName);
|
|
|
|
bool ret = IsSecureFile(wFilename);
|
|
|
|
delete [] wFilename;
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool plSecureStream::IsSecureFile(const wchar_t* fileName)
|
|
|
|
{
|
|
|
|
hsFD fp = INVALID_HANDLE_VALUE;
|
|
|
|
|
|
|
|
#if HS_BUILD_FOR_WIN32
|
|
|
|
fp = CreateFileW(fileName,
|
|
|
|
GENERIC_READ, // open for reading
|
|
|
|
0, // no one can open the file until we're done
|
|
|
|
NULL, // default security
|
|
|
|
OPEN_EXISTING, // only open existing files (no creation)
|
|
|
|
FILE_ATTRIBUTE_NORMAL, // normal file attributes
|
|
|
|
NULL); // no template
|
|
|
|
#elif HS_BUILD_FOR_UNIX
|
|
|
|
const char* cfile = hsWStringToString(fileName);
|
|
|
|
fp = fopen(cfile, "rb");
|
|
|
|
delete[] cfile;
|
|
|
|
#endif
|
|
|
|
|
|
|
|
if (fp == INVALID_HANDLE_VALUE)
|
|
|
|
return false;
|
|
|
|
|
|
|
|
bool isEncrypted = ICheckMagicString(fp);
|
|
|
|
|
|
|
|
#if HS_BUILD_FOR_WIN32
|
|
|
|
CloseHandle(fp);
|
|
|
|
#elif HS_BUILD_FOR_UNIX
|
|
|
|
fclose(fp);
|
|
|
|
#endif
|
|
|
|
|
|
|
|
return isEncrypted;
|
|
|
|
}
|
|
|
|
|
|
|
|
hsStream* plSecureStream::OpenSecureFile(const char* fileName, const uint32_t flags /* = kRequireEncryption */, uint32_t* key /* = nil */)
|
|
|
|
{
|
|
|
|
wchar_t* wFilename = hsStringToWString(fileName);
|
|
|
|
hsStream* ret = OpenSecureFile(wFilename, flags, key);
|
|
|
|
delete [] wFilename;
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
hsStream* plSecureStream::OpenSecureFile(const wchar_t* fileName, const uint32_t flags /* = kRequireEncryption */, uint32_t* key /* = nil */)
|
|
|
|
{
|
|
|
|
bool requireEncryption = flags & kRequireEncryption;
|
|
|
|
#ifndef PLASMA_EXTERNAL_RELEASE
|
|
|
|
requireEncryption = false;
|
|
|
|
#endif
|
|
|
|
|
|
|
|
bool deleteOnExit = flags & kDeleteOnExit;
|
|
|
|
bool isEncrypted = IsSecureFile(fileName);
|
|
|
|
|
|
|
|
hsStream* s = nil;
|
|
|
|
if (isEncrypted)
|
|
|
|
s = new plSecureStream(deleteOnExit, key);
|
|
|
|
else if (!requireEncryption) // If this isn't an external release, let them use unencrypted data
|
|
|
|
s = new hsUNIXStream;
|
|
|
|
|
|
|
|
if (s)
|
|
|
|
s->Open(fileName, L"rb");
|
|
|
|
return s;
|
|
|
|
}
|
|
|
|
|
|
|
|
hsStream* plSecureStream::OpenSecureFileWrite(const char* fileName, uint32_t* key /* = nil */)
|
|
|
|
{
|
|
|
|
wchar_t* wFilename = hsStringToWString(fileName);
|
|
|
|
hsStream* ret = OpenSecureFileWrite(wFilename, key);
|
|
|
|
delete [] wFilename;
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
hsStream* plSecureStream::OpenSecureFileWrite(const wchar_t* fileName, uint32_t* key /* = nil */)
|
|
|
|
{
|
|
|
|
hsStream* s = nil;
|
|
|
|
#ifdef PLASMA_EXTERNAL_RELEASE
|
|
|
|
s = new plSecureStream(false, key);
|
|
|
|
#else
|
|
|
|
s = new hsUNIXStream;
|
|
|
|
#endif
|
|
|
|
|
|
|
|
s->Open(fileName, L"wb");
|
|
|
|
return s;
|
|
|
|
}
|
|
|
|
|
|
|
|
//// GetSecureEncryptionKey //////////////////////////////////////////////////
|
|
|
|
|
|
|
|
bool plSecureStream::GetSecureEncryptionKey(const char* filename, uint32_t* key, unsigned length)
|
|
|
|
{
|
|
|
|
wchar_t* wFilename = hsStringToWString(filename);
|
|
|
|
bool ret = GetSecureEncryptionKey(wFilename, key, length);
|
|
|
|
delete [] wFilename;
|
|
|
|
return ret;
|
|
|
|
}
|
|
|
|
|
|
|
|
bool plSecureStream::GetSecureEncryptionKey(const wchar_t* filename, uint32_t* key, unsigned length)
|
|
|
|
{
|
|
|
|
// looks for an encryption key file in the same directory, and reads it
|
|
|
|
std::wstring sFilename = filename;
|
|
|
|
|
|
|
|
// grab parent directory
|
|
|
|
size_t loc = sFilename.rfind(L"\\");
|
|
|
|
if (loc == std::wstring::npos)
|
|
|
|
loc = sFilename.rfind(L"/");
|
|
|
|
|
|
|
|
std::wstring sDir;
|
|
|
|
if (loc != std::wstring::npos)
|
|
|
|
sDir = sFilename.substr(0, loc);
|
|
|
|
else // no directory
|
|
|
|
sDir = L"./";
|
|
|
|
if ((sDir[sDir.length()-1] != L'/') && (sDir[sDir.length()-1] != L'\\'))
|
|
|
|
sDir += L'/'; // add the slash, if it doesn't has one
|
|
|
|
|
|
|
|
// now add the key filename
|
|
|
|
std::wstring keyFile = sDir + kWKeyFilename;
|
|
|
|
|
|
|
|
if (plFileUtils::FileExists(keyFile.c_str()))
|
|
|
|
{
|
|
|
|
// file exists, read from it
|
|
|
|
hsUNIXStream file;
|
|
|
|
file.Open(keyFile.c_str(), L"rb");
|
|
|
|
|
|
|
|
unsigned bytesToRead = length * sizeof(uint32_t);
|
|
|
|
uint8_t* buffer = (uint8_t*)malloc(bytesToRead);
|
|
|
|
unsigned bytesRead = file.Read(bytesToRead, buffer);
|
|
|
|
|
|
|
|
file.Close();
|
|
|
|
|
|
|
|
unsigned memSize = min(bytesToRead, bytesRead);
|
|
|
|
memcpy(key, buffer, memSize);
|
|
|
|
free(buffer);
|
|
|
|
|
|
|
|
return true;
|
|
|
|
}
|
|
|
|
|
|
|
|
// file doesn't exist, use default key
|
|
|
|
unsigned memSize = min(length, arrsize(plSecureStream::kDefaultKey));
|
|
|
|
memSize *= sizeof(uint32_t);
|
|
|
|
memcpy(key, plSecureStream::kDefaultKey, memSize);
|
|
|
|
|
|
|
|
return false;
|
|
|
|
}
|