initial importHEADmain

1 files changed, 769 insertions, 0 deletions

diff --git a/src/parsec/t_pack.cpp b/src/parsec/t_pack.cpp
new file mode 100644
index 0000000..9baedc0
--- /dev/null
+++ b/src/parsec/t_pack.cpp
@@ -0,0 +1,769 @@
+/*
+ * PARSEC - Data Package Tools
+ *
+ * $Author: uberlinuxguy $ - $Date: 2004/09/26 03:43:42 $
+ *
+ * Orginally written by:
+ *   Copyright (c) Markus Hadwiger     <msh@parsec.org>   1996-2000
+ *
+ *  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 2 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, write to the Free Software
+ *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+ */ 
+
+// C library
+#include <errno.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <stdarg.h>
+
+// compilation flags/debug support
+#include "config.h"
+#include "debug.h"
+
+// general definitions
+#include "general.h"
+#include "gd_heads.h"
+#include "objstruc.h"
+
+// global externals
+#include "globals.h"
+
+// local module header
+#include "t_pack.h"
+
+// proprietary module headers
+#include "e_getopt.h"
+#include "sys_swap.h"
+
+
+
+// string constants -----------------------------------------------------------
+//
+static char pack_signature[]		= "msh DataPackage";
+static char options_invalid[]		= "invalid command line option.\n";
+static char pack_name_missing[]		= "name of package file missing.\n";
+
+
+// variables settable via command line options --------------------------------
+//
+static char		arg_package_name[ PATH_MAX + 1 ];
+static size_t	arg_package_file_offset = 0;
+static char		arg_file_name[ PATH_MAX + 1 ];
+static char		arg_list_name[ PATH_MAX + 1 ];
+static int		arg_view_files = FALSE;
+
+
+// message output wrappers ----------------------------------------------------
+//
+#define Printf		MSGOUT
+#define PrintfNoNL	MSGPUT
+#define Err_Printf	MSGOUT
+
+
+// exit function that makes sure that messages can be read by the user --------
+//
+PRIVATE
+void Exit( int code )
+{
+	SYSs_MsgPrompt();
+	exit( code );
+}
+
+
+// set name of package --------------------------------------------------------
+//
+PRIVATE
+int Tm_SetPackageName( char **packname )
+{
+	strncpy( arg_package_name, packname[ 0 ], PATH_MAX );
+	arg_package_name[ PATH_MAX ] = 0;
+
+	return TRUE;
+}
+
+
+// set offset of package start in package file --------------------------------
+//
+PRIVATE
+int Tm_SetPackageOffset( int *packofs )
+{
+	arg_package_file_offset = packofs[ 0 ];
+
+	return TRUE;
+}
+
+
+// set file name argument -----------------------------------------------------
+//
+PRIVATE
+int Tm_SetFileName( char **filename )
+{
+	strncpy( arg_file_name, filename[ 0 ], PATH_MAX );
+	arg_file_name[ PATH_MAX ] = 0;
+
+	return TRUE;
+}
+
+
+// set list name argument -----------------------------------------------------
+//
+PRIVATE
+int Tm_SetListName( char **listname )
+{
+	strncpy( arg_list_name, listname[ 0 ], PATH_MAX );
+	arg_list_name[ PATH_MAX ] = 0;
+
+	return TRUE;
+}
+
+
+// set flag to display file names ---------------------------------------------
+//
+PRIVATE
+int Tm_ViewFileList()
+{
+	arg_view_files = TRUE;
+	return TRUE;
+}
+
+
+// data file list item (info read from text file) -----------------------------
+//
+struct list_item_s {
+
+	char	file[ PATH_MAX + 1 ];
+	dword	foffset;
+	dword	flength;
+};
+
+
+// read file list -------------------------------------------------------------
+//
+PRIVATE
+int ReadFileList( const char *filename, list_item_s *listitems, int maxitems )
+{
+	ASSERT( filename != NULL );
+	ASSERT( listitems != NULL );
+	ASSERT( maxitems > 0 );
+
+	FILE *fp = fopen( filename, "r" );
+	if ( fp == NULL ) {
+		Err_Printf( "file list \"%s\" not found.\n", filename );
+		return -1;
+	}
+
+	Printf( "processing file list ..." );
+
+	int pcount   = 32;
+	int numitems = 0;
+
+	static char list_line[ PATH_MAX + 1 ];
+
+	// read file list
+	while ( fgets( list_line, PATH_MAX + 1, fp ) != NULL ) {
+
+		char *scan = strtok( list_line, " \t\n\r" );
+		if ( scan == NULL ) {
+			continue;
+		}
+
+		if ( ( *scan == ';' ) || ( *scan == '#' ) ) {
+			continue;
+		}
+
+		if ( --pcount == 0 ) {
+			PrintfNoNL( "." );
+			pcount = 32;
+		}
+
+		if ( numitems >= maxitems ) {
+			Err_Printf( "\ntoo many files. (%d max)\n", maxitems );
+			fclose( fp );
+			return -1;
+		}
+
+		strncpy( listitems[ numitems ].file, scan, PATH_MAX );
+		listitems[ numitems ].file[ PATH_MAX ] = 0;
+		numitems++;
+	}
+
+	Printf( "\n" );
+	Printf( "file list read.\n" );
+	Printf( "%d items found.\n", numitems );
+
+	fclose( fp );
+	return numitems;
+}
+
+
+// copy string and strip path -------------------------------------------------
+//
+PRIVATE
+void StrcpyStripPath( char *dst, const char *src )
+{
+	ASSERT( dst != NULL );
+	ASSERT( src != NULL );
+
+	int baselen = 0;
+	int pos = 0;
+	for ( pos = strlen( src ) - 1; pos >= 0; pos-- ) {
+
+		if ( src[ pos ] == '\\' )
+			break;
+		if ( src[ pos ] == '/' )
+			break;
+		if ( src[ pos ] == ':' )
+			break;
+
+		if ( ++baselen >= 12 ) {
+			pos--;
+			break;
+		}
+	}
+
+	if ( baselen > 0 ) {
+		strncpy( dst, &src[ pos + 1 ], baselen );
+		dst[ baselen ] = 0;
+	}
+}
+
+
+// write header/file info list into output package ----------------------------
+//
+PRIVATE
+int WritePackageHeader( FILE *fp, list_item_s *listitems, int numitems )
+{
+	ASSERT( fp != NULL );
+	ASSERT( listitems != NULL );
+	ASSERT( numitems > 0 );
+
+	// alloc file info list
+	size_t headersize  = ( numitems + 1 ) * sizeof( pfileinfodisk_s );
+	pfileinfodisk_s *items = (pfileinfodisk_s *) ALLOCMEM( headersize );
+	if ( items == NULL )
+		return FALSE;
+	memset( items, 0, headersize );
+
+	size_t datasize  = 0;
+	size_t curoffset = sizeof( packageheader_s ) + headersize;
+
+	// create file info list
+	int curit = 0;
+	for (  curit = 0; curit < numitems; curit++ ) {
+
+		PrintfNoNL( "." );
+
+		StrcpyStripPath( items[ curit ].file, listitems[ curit ].file );
+
+		FILE *itfp = fopen( listitems[ curit ].file, "rb" );
+		if ( itfp == NULL ) {
+			Err_Printf( "\ndata file \"%s\" not found: %s\n",
+					 listitems[ curit ].file, strerror( errno ) );
+			FREEMEM( items );
+			return FALSE;
+		}
+		if ( fseek( itfp, 0, SEEK_END ) != 0 ) {
+			Err_Printf( "\ndata file \"%s\" access error: %s\n",
+					 listitems[ curit ].file, strerror( errno ) );
+			fclose( itfp );
+			FREEMEM( items );
+			return FALSE;
+		}
+
+		items[ curit ].flength     = (dword) ftell( itfp );
+		listitems[ curit ].flength = items[ curit ].flength;
+
+		items[ curit ].foffset     = curoffset;
+		listitems[ curit ].foffset = items[ curit ].foffset;
+
+		datasize  += items[ curit ].flength;
+		curoffset += items[ curit ].flength;
+
+		fclose( itfp );
+	}
+
+	// create fixed-size header
+	packageheader_s packheader;
+	packheader.numitems   = numitems;
+	packheader.headersize = headersize;
+	packheader.datasize   = datasize;
+	packheader.packsize   = sizeof( packageheader_s ) + headersize + datasize;
+	strcpy( packheader.signature, pack_signature );
+
+	Printf( "\ndata size: %d\n", datasize );
+
+	// swap endianness of fixed-size part of header
+	SYS_SwapPackageHeader( &packheader );
+
+	// write fixed-size header
+	size_t byteswritten = fwrite( &packheader, 1, sizeof( packheader ), fp );
+	if ( byteswritten != sizeof( packheader ) ) {
+		FREEMEM( items );
+		return FALSE;
+	}
+
+	// swap endianness of file info list
+	for ( curit = 0; curit < numitems; curit++ ) {
+		SYS_SwapPFileInfo( &items[ curit ] );
+	}
+
+	// write file info list
+	byteswritten = fwrite( items, 1, headersize, fp );
+	if ( byteswritten != headersize ) {
+		FREEMEM( items );
+		return FALSE;
+	}
+
+	FREEMEM( items );
+	return TRUE;
+}
+
+
+// write actual file data into output package ---------------------------------
+//
+PRIVATE
+int WritePackageData( FILE *fp, list_item_s *listitems, int numitems )
+{
+	ASSERT( fp != NULL );
+	ASSERT( listitems != NULL );
+	ASSERT( numitems > 0 );
+
+	for ( int curit = 0; curit < numitems; curit++ ) {
+
+		PrintfNoNL( "." );
+
+		FILE *itfp = fopen( listitems[ curit ].file, "rb" );
+		if ( itfp == NULL ) {
+			Err_Printf( "\ndata file \"%s\" not found: %s\n",
+					 listitems[ curit ].file, strerror( errno ) );
+			return FALSE;
+		}
+
+		char *filedata = (char *) ALLOCMEM( listitems[ curit ].flength );
+		if ( filedata == NULL ) {
+			fclose( itfp );
+			return FALSE;
+		}
+
+		size_t bytesread = fread( filedata, 1, listitems[ curit ].flength, itfp );
+		if ( bytesread != listitems[ curit ].flength ) {
+			FREEMEM( filedata );
+			fclose( itfp );
+			return FALSE;
+		}
+
+		size_t byteswritten = fwrite( filedata, 1, listitems[ curit ].flength, fp );
+		if ( byteswritten != listitems[ curit ].flength ) {
+			FREEMEM( filedata );
+			fclose( itfp );
+			return FALSE;
+		}
+
+		fclose( itfp );
+		FREEMEM( filedata );
+	}
+
+	return TRUE;
+}
+
+
+
+// write package data ---------------------------------------------------------
+//
+PRIVATE
+int WritePackageFile( const char* packname, list_item_s *listitems, int numitems )
+{
+	ASSERT( packname != NULL );
+	ASSERT( listitems != NULL );
+	ASSERT( numitems > 0 );
+
+	Printf( "creating file package \"%s\" ...", packname );
+	fflush( stdout );
+
+	// create package file
+	FILE *fp = fopen( packname, "wb" );
+	if ( fp == NULL ) {
+		Err_Printf( "\ncannot create output file: %s\n", strerror( errno ) );
+		return FALSE;
+	}
+
+	if ( !WritePackageHeader( fp, listitems, numitems ) ) {
+		fclose( fp );
+		return FALSE;
+	}
+
+	if ( !WritePackageData( fp, listitems, numitems ) ) {
+		fclose( fp );
+		return FALSE;
+	}
+
+	fclose( fp );
+	return TRUE;
+}
+
+
+// critical error exit --------------------------------------------------------
+//
+PRIVATE
+void MakepackError()
+{
+	Err_Printf( "-makepack: critical error.\n" );
+	Exit( EXIT_FAILURE );
+}
+
+
+// create data package --------------------------------------------------------
+//
+PRIVATE
+int CreatePackage( const char* packname, const char *listname )
+{
+	ASSERT( packname != NULL );
+	ASSERT( listname != NULL );
+
+	#define MAX_DATA_FILES 4096
+
+	list_item_s *listitems = (list_item_s *) ALLOCMEM( MAX_DATA_FILES * sizeof( list_item_s ) );
+	if ( listitems == NULL ) {
+		return FALSE;
+	}
+
+	int numitems = ReadFileList( listname, listitems, MAX_DATA_FILES );
+	if ( numitems == -1 ) {
+		FREEMEM( listitems );
+		MakepackError();
+	}
+
+	if ( !WritePackageFile( packname, listitems, numitems ) ) {
+		FREEMEM( listitems );
+		MakepackError();
+	}
+
+	FREEMEM( listitems );
+	return TRUE;
+}
+
+
+// strip extension from supplied file name ------------------------------------
+//
+PRIVATE
+void StripExtension( char *fname )
+{
+	ASSERT( fname != NULL );
+
+	for ( int len = strlen( fname ); len >= 0; len-- ) {
+		if ( fname[ len ] == '.' ) {
+			fname[ len ] = 0;
+			break;
+		}
+		if ( fname[ len ] == '\\' ) {
+			break;
+		}
+		if ( fname[ len ] == '/' ) {
+			break;
+		}
+		if ( fname[ len ] == ':' ) {
+			break;
+		}
+	}
+}
+
+
+// package creation tool main function ----------------------------------------
+//
+void MAKEPACK_main( int argc, char **argv )
+{
+	// clear options of main app
+	OPT_ClearOptions();
+
+	// register local options
+	OPT_RegisterStringOption( "p", "pack", Tm_SetPackageName );
+	OPT_RegisterStringOption( "f", "list", Tm_SetListName );
+
+	// exec all registered command line options
+	if ( !OPT_ExecRegisteredOptions( argc, argv ) ) {
+		Err_Printf( options_invalid );
+		Exit( EXIT_FAILURE );
+	}
+
+	// package name is mandatory
+	if ( arg_package_name[ 0 ] == 0 ) {
+		Err_Printf( pack_name_missing );
+		Exit( EXIT_FAILURE );
+	}
+
+	// list name is optional (use package with different extension by default)
+	if ( arg_list_name[ 0 ] == 0 ) {
+		strcpy( arg_list_name, arg_package_name );
+		StripExtension( arg_list_name );
+		strcat( arg_list_name, ".lst" );
+	}
+
+	// create package
+	CreatePackage( arg_package_name, arg_list_name );
+
+	// end of sub-application
+	Exit( EXIT_SUCCESS );
+}
+
+
+// optional file name parameters ----------------------------------------------
+//
+static char *extract_file_name	= NULL;
+static char *list_file_name		= NULL;
+
+
+// extract data of packaged file(s) -------------------------------------------
+//
+PRIVATE
+int ExtractPackageData( FILE *fp, FILE *listfp, size_t baseofs, int numitems, pfileinfodisk_s *items )
+{
+	ASSERT( fp != NULL );
+	ASSERT( numitems >= 0 );
+	ASSERT( items != NULL );
+
+	int numextracted = 0;
+
+	// write out separate data files
+	for ( int curit = 0; curit < numitems; curit++ ) {
+
+		// extract only specified file if name set
+		if ( extract_file_name != NULL ) {
+			if ( stricmp( items[ curit ].file, extract_file_name ) != 0 ) {
+				continue;
+			}
+		}
+
+		// check for viewing mode (no actual file extraction)
+		if ( arg_view_files ) {
+			Printf( "file %s size %d\n", items[ curit ].file, items[ curit ].flength );
+			continue;
+		}
+
+		char *curdata = (char *) ALLOCMEM( items[ curit ].flength );
+		if ( curdata == NULL ) {
+			return -1;
+		}
+
+		// set read position
+		size_t readpos = baseofs + items[ curit ].foffset;
+		if ( fseek( fp, readpos, SEEK_SET ) != 0 ) {
+			FREEMEM( curdata );
+			return -1;
+		}
+
+		// read packaged file data
+		size_t bytesread = fread( curdata, 1, items[ curit ].flength, fp );
+		if ( bytesread != items[ curit ].flength ) {
+			FREEMEM( curdata );
+			return -1;
+		}
+
+		// display current file and write it to list if open
+		Printf( "*extracting %s\n", items[ curit ].file );
+		if ( listfp != NULL ) {
+			fprintf( listfp, "%s\n", items[ curit ].file );
+		}
+
+		// create output file
+		FILE *wfp = fopen( items[ curit ].file, "wb" );
+		if ( wfp == NULL ) {
+			FREEMEM( curdata );
+			return -1;
+		}
+
+		// write file data to separate file
+		size_t byteswritten = fwrite( curdata, 1, items[ curit ].flength, wfp );
+		if ( byteswritten != items[ curit ].flength ) {
+			FREEMEM( curdata );
+			fclose( wfp );
+			return -1;
+		}
+
+		if ( fclose( wfp ) != 0 ) {
+			FREEMEM( curdata );
+			return -1;
+		}
+
+		FREEMEM( curdata );
+		numextracted++;
+	}
+
+	return numextracted;
+}
+
+
+// critical error exit --------------------------------------------------------
+//
+PRIVATE
+void GetpackError()
+{
+	Err_Printf( "-getpack: critical error.\n" );
+	Exit( EXIT_FAILURE );
+}
+
+
+// extract file(s) from package -----------------------------------------------
+//
+PRIVATE
+int ExtractPackage( const char* packname, size_t baseofs )
+{
+	ASSERT( packname != NULL );
+
+	// open package for reading at specified offset
+	FILE *fp = fopen( packname, "rb" );
+	if ( fp == NULL ) {
+		Err_Printf( "package %s not found.\n", packname );
+		GetpackError();
+	}
+	if ( fseek( fp, baseofs, SEEK_SET ) != 0 ) {
+		fclose( fp );
+		GetpackError();
+	}
+
+	packageheader_s packheader;
+
+	// read fixed-size part of header
+	size_t bytesread = fread( &packheader, 1, sizeof( packheader ), fp );
+	if ( bytesread != sizeof( packheader ) ) {
+		fclose( fp );
+		GetpackError();
+	}
+
+	// file format validation
+	if ( stricmp( packheader.signature, pack_signature ) != 0 ) {
+		Err_Printf( "package %s is invalid.\n", packname );
+		fclose( fp );
+		GetpackError();
+	}
+
+	// swap endianness of fixed-size part of header
+	SYS_SwapPackageHeader( &packheader );
+
+	// fetch size of remaining header and number of contained items
+	dword headersize = packheader.headersize;
+	dword numitems   = packheader.numitems;
+
+	// display number of contained items if in viewing mode
+	if ( arg_view_files ) {
+		Printf( "package contains %d items.\n\n", numitems );
+	}
+
+	// read file list
+	pfileinfodisk_s *items = (pfileinfodisk_s *) ALLOCMEM( headersize );
+	if ( items == NULL ) {
+		fclose( fp );
+		GetpackError();
+	}
+	if ( fread( items, 1, headersize, fp ) != headersize ) {
+		FREEMEM( items );
+		fclose( fp );
+		GetpackError();
+	}
+
+	// swap endianness of file info list
+	for ( unsigned int curit = 0; curit < numitems; curit++ ) {
+		SYS_SwapPFileInfo( &items[ curit ] );
+	}
+
+	// open list output file if specified
+	FILE *listfp = NULL;
+	if ( ( list_file_name != NULL ) && ( extract_file_name == NULL ) ) {
+		listfp = fopen( list_file_name, "w" );
+		fprintf( listfp, "\n" );
+		fprintf( listfp, "; file list of package \"%s\"\n", packname );
+		fprintf( listfp, "\n" );
+	}
+
+	// extract file data
+	Printf( "extracting from package %s ...\n", packname );
+	int numextracted = ExtractPackageData( fp, listfp, baseofs, numitems, items );
+
+	// close/free resources
+	if ( listfp != NULL ) {
+		fprintf( listfp, "\n" );
+		fclose( listfp );
+		listfp = NULL;
+	}
+	fclose( fp );
+	FREEMEM( items );
+
+	if ( numextracted == -1 ) {
+		GetpackError();
+	}
+
+	if ( extract_file_name == NULL ) {
+		Printf( "%d files extracted.\n", numextracted );
+		return TRUE;
+	}
+
+	if ( numextracted < 1 ) {
+		Printf( "file %s not found in package %s\n", extract_file_name, packname );
+		return FALSE;
+	}
+
+	return TRUE;
+}
+
+
+// package extraction tool main function --------------------------------------
+//
+void GETPACK_main( int argc, char **argv )
+{
+	// clear options of main app
+	OPT_ClearOptions();
+
+	// register local options
+	OPT_RegisterStringOption( "p", "pack",   Tm_SetPackageName );
+	OPT_RegisterIntOption(    "o", "offset", Tm_SetPackageOffset );
+	OPT_RegisterStringOption( "f", "file",   Tm_SetFileName );
+	OPT_RegisterStringOption( "l", "list",   Tm_SetListName );
+	OPT_RegisterSetOption(    "v", "view",   Tm_ViewFileList );
+
+	// exec all registered command line options
+	if ( !OPT_ExecRegisteredOptions( argc, argv ) ) {
+		Err_Printf( options_invalid );
+		Exit( EXIT_FAILURE );
+	}
+
+	// package name is mandatory
+	if ( arg_package_name[ 0 ] == 0 ) {
+		Err_Printf( pack_name_missing );
+		Exit( EXIT_FAILURE );
+	}
+
+	// file name is optional (operate on entire package by default)
+	extract_file_name = ( arg_file_name[ 0 ] != 0 ) ? arg_file_name : NULL;
+
+	// list name is optional (do not create a list file by default)
+	list_file_name = ( arg_list_name[ 0 ] != 0 ) ? arg_list_name : NULL;
+
+	// extract one file or all files from package
+	ExtractPackage( arg_package_name, arg_package_file_offset );
+
+	// end of sub-application
+	Exit( EXIT_SUCCESS );
+}
+
+
+// module registration function -----------------------------------------------
+//
+REGISTER_MODULE( T_PACK )
+{
+	OPT_RegisterApplication( "-makepack", MAKEPACK_main );
+	OPT_RegisterApplication( "-getpack",  GETPACK_main );
+}
+
+
+