Modified source engine (2017) developed by valve and leaked in 2020. Not for commercial purporses
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

278 lines
5.5 KiB

5 years ago
//========= 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;
}