tf2/tf2_src/utils/vice/vice.cpp

//========= Copyright Valve Corporation, All rights reserved. ============//
//
// Purpose: 
//
// $NoKeywords: $
//
//=============================================================================//
// vice.cpp : Defines the entry point for the console application.
//

#include <stdlib.h>
#include <stdio.h>
#include <malloc.h>
#include <string.h>
#include "tier1/strtools.h"
#include <sys/stat.h>
#include "conio.h"
#include <direct.h>
#include <io.h>
#include "UtlBuffer.h"
#include "tier0/dbg.h"
#include "cmdlib.h"
#include "tier0/icommandline.h"
#include "windows.h"

#include "mathlib/IceKey.h"
#include <filesystem_tools.h>

#define FF_TRYAGAIN 1
#define FF_DONTPROCESS 2

#undef GetCurrentDirectory

static bool g_NoPause = false;
static bool g_Quiet = false;
static bool g_Encrypt = false;
static bool g_Decrypt = false;
static char g_ICEKey[16];
static char g_Extension[16];

static void Pause( void )
{
	if( !g_NoPause )
	{
		printf( "Hit a key to continue\n" );
		getch();
	}
}

static void Exit(const char *msg)
{
	fprintf( stderr, "%s", msg );
	Pause();
	exit( -1 );
}

static void Usage( void )
{
	fprintf( stderr, "Usage: vice [-quiet] [-nopause] [-encrypt key] [-decrypt key] [-newext name] file file2 . . .\n" );
	fprintf( stderr, "-quiet   : don't print anything out, don't pause for input\n" );
	fprintf( stderr, "-nopause : don't pause for input\n" );
	fprintf( stderr, "-encrypt : encrypt files with given key\n" );
	fprintf( stderr, "-decrypt : decypt files with given key\n" );
	fprintf( stderr, "-newext  : new output file extension\n" );
	Pause();
	exit( -1 );
}


bool Process_File( char *pInputBaseName, int maxlen )
{
	Q_FixSlashes( pInputBaseName, '/' );
	// Q_StripExtension( pInputBaseName, pInputBaseName, maxlen );
	
	if( !g_Quiet )
	{
		printf( "input file: %s\n", pInputBaseName );
	}
	
	FileHandle_t f = g_pFullFileSystem->Open(pInputBaseName, "rb", "vice" );
		
	if (!f)
		Error("Could not open input file");
		
	int fileSize = g_pFullFileSystem->Size(f);

	unsigned char *buffer = (unsigned char*)_alloca(fileSize);

	g_pFullFileSystem->Read(buffer, fileSize, f); // read into local buffer
	g_pFullFileSystem->Close( f );	// close file after reading

	IceKey ice( 0 ); // level 0 = 64bit key
	ice.set( (unsigned char*) g_ICEKey ); // set key

	int blockSize = ice.blockSize();

	unsigned char *temp = (unsigned char *)_alloca( fileSize );
	unsigned char *p1 = buffer;
	unsigned char *p2 = temp;
				
	// encrypt data in 8 byte blocks
	int bytesLeft = fileSize;
	while ( bytesLeft >= blockSize )
	{
		if ( g_Encrypt )
		{
			ice.encrypt( p1, p2 );
		}
		else if ( g_Decrypt )
		{
			ice.decrypt( p1, p2 );
		}
		else
		{
			memcpy( p2, p1, blockSize );
		}

		bytesLeft -= blockSize;
		p1+=blockSize;
		p2+=blockSize;
	}

	memcpy( p2, p1, bytesLeft );

	Q_SetExtension( pInputBaseName, g_Extension, maxlen );

	if( !g_Quiet )
	{
		printf( "output file: %s\n", pInputBaseName );
	}

	f = g_pFullFileSystem->Open(pInputBaseName, "wb", "vice" );
	
	if (!f)
		Exit("Could not open output file");
		
	g_pFullFileSystem->Write( temp, fileSize, f ); // read into local buffer
	g_pFullFileSystem->Close( f );	// close file after reading
	
	return TRUE;
}

int main(int argc, char* argv[])
{
	CommandLine()->CreateCmdLine( argc, argv );
	

	if( argc < 2 )
	{
		Usage();
	}
	char *pInputBaseName = NULL;
	int i = 1;
	strcpy( g_Extension, ".dat" );
	while( i < argc )
	{
		if( stricmp( argv[i], "-quiet" ) == 0 )
		{
			i++;
			g_Quiet = true;
			g_NoPause = true; // no point in pausing if we aren't going to print anything out.
		}
		if( stricmp( argv[i], "-nopause" ) == 0 )
		{
			i++;
			g_NoPause = true;
		}
		if( stricmp( argv[i], "-encrypt" ) == 0 )
		{
			g_Encrypt = true;
			i++;

			if ( strlen( argv[i] ) != 8 )
			{
				Exit("Error - ICE key must be a 8 char text.\n");
			}

			Q_strncpy( g_ICEKey, argv[i], sizeof(g_ICEKey) );
			i++;
			
		}
		if( stricmp( argv[i], "-decrypt" ) == 0 )
		{
			g_Decrypt = true;
			i++;

			if ( strlen( argv[i] ) != 8 )
			{
				Exit("Error - ICE key must be a 8 char text.\n");
			}

			Q_strncpy( g_ICEKey, argv[i], sizeof(g_ICEKey) );
			i++;
			
		}
		if( stricmp( argv[i], "-newext" ) == 0 )
		{
			i++;

			if ( strlen( argv[i] ) > 5 )
			{
				Exit("Error - extension must be smaller than 4 chars.\n");
			}

			Q_strncpy( g_Extension, argv[i], sizeof(g_Extension) );
			i++;
			
		}
		else
		{
			break;
		}
	}

	if ( i >= argc )
	{
		Exit("Error - missing files in commandline.\n");
	}

	CmdLib_InitFileSystem( argv[i] );

	g_pFullFileSystem->GetCurrentDirectory( gamedir, sizeof(gamedir) );
	g_pFullFileSystem->AddSearchPath( gamedir, "vice" );


	Q_FixSlashes( gamedir, '/' );

	for( ; i < argc; i++ )
	{
		pInputBaseName = argv[i];
		int maxlen = Q_strlen( pInputBaseName ) + 1;


		if ( strstr( pInputBaseName, "*.") )
		{
			char	search[ MAX_PATH ];
			char	fname[ MAX_PATH ];
			char	ext[_MAX_EXT];

			_splitpath( pInputBaseName, NULL, NULL, fname, ext ); //find extension wanted
			fname[strlen(fname)-1] = 0; // remove *

			sprintf( search, "%s\\*%s", gamedir, ext );

			Q_FixSlashes( search, '/' );

			WIN32_FIND_DATA wfd;
			HANDLE hResult;
			memset(&wfd, 0, sizeof(WIN32_FIND_DATA));
			
			hResult = FindFirstFile( search, &wfd );

			while ( hResult != INVALID_HANDLE_VALUE )
			{
				if ( !strnicmp( fname, wfd.cFileName, strlen(fname) ) )
				{
					if ( !Process_File( wfd.cFileName, sizeof( wfd.cFileName ) ) )
						break;
				}

				if ( !FindNextFile( hResult, &wfd) )
					break;
							
			}
			
			FindClose( hResult );
		}
		else
		{
			Process_File( pInputBaseName, maxlen ); 			
		}
	}

	Pause();
	return 0;
}