1
This commit is contained in:
FluorescentCIAAfricanAmerican
277
utils/vice/vice.cpp
Normal file
277
utils/vice/vice.cpp
Normal file
@ -0,0 +1,277 @@
|
||||
//========= 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;
|
||||
}
|
||||
|
||||
41
utils/vice/vice.vpc
Normal file
41
utils/vice/vice.vpc
Normal file
@ -0,0 +1,41 @@
|
||||
//-----------------------------------------------------------------------------
|
||||
// VICE.VPC
|
||||
//
|
||||
// Project Script
|
||||
//-----------------------------------------------------------------------------
|
||||
|
||||
$Macro SRCDIR "..\.."
|
||||
$Macro OUTBINDIR "$SRCDIR\..\game\bin"
|
||||
|
||||
$Include "$SRCDIR\vpc_scripts\source_exe_con_base.vpc"
|
||||
|
||||
$Configuration
|
||||
{
|
||||
$Compiler
|
||||
{
|
||||
$AdditionalIncludeDirectories "$BASE,..\common"
|
||||
}
|
||||
}
|
||||
|
||||
$Project "Vice"
|
||||
{
|
||||
$Folder "Source Files"
|
||||
{
|
||||
$File "..\common\cmdlib.cpp"
|
||||
$File "$SRCDIR\public\filesystem_helpers.cpp"
|
||||
$File "$SRCDIR\public\filesystem_init.cpp"
|
||||
$File "..\common\filesystem_tools.cpp"
|
||||
$File "vice.cpp"
|
||||
}
|
||||
|
||||
$Folder "Header Files"
|
||||
{
|
||||
$File "$SRCDIR\public\mathlib\IceKey.H"
|
||||
}
|
||||
|
||||
$Folder "Link Libraries"
|
||||
{
|
||||
$Lib tier2
|
||||
$Lib mathlib
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user