X-Git-Url: https://git.dogcows.com/gitweb?p=chaz%2Fthecheat;a=blobdiff_plain;f=SearchContext.m;fp=SearchContext.m;h=704e267dd5d4da3d758b3fe250098b3ed207659d;hp=0000000000000000000000000000000000000000;hb=d27548f80fe411fda2ee69c74a24eab4292267e9;hpb=e8d51183acdd2410a38dcf8f0efbf7c30cd6c581

diff --git a/SearchContext.m b/SearchContext.m
new file mode 100644
index 0000000..704e267
--- /dev/null
+++ b/SearchContext.m
@@ -0,0 +1,295 @@
+//
+//  SearchContext.m
+//  The Cheat
+//
+//  Created by Chaz McGarvey on 12/4/04.
+//  Copyright 2004 Chaz McGarvey. All rights reserved.
+//
+
+#import "SearchContext.h"
+
+
+@implementation SearchContext
+
+
+#pragma mark Initialization
+
+/*
+ * There isn't really a designated initializer because the initialization of the types
+ * of searches vary way too much.
+ */
+
+- (id)initWithPID:(pid_t)pid searchOperator:(TCSearchOperator)op value:(Variable *)val
+{
+	unsigned valueSize;
+	
+	if ( [super init] ) {
+		value = [val retain];
+		if ( !value ) {
+			[self release];
+			return nil;
+		}
+		valueSize = [value valueSize];
+		process = pid;
+		_variableType = [value type];
+		_integerSign = [value integerSign];
+		_operator = op;
+		_searchType = TCGivenValue;
+		compareFunc = [self compareFunction];
+		
+		// allocate the memory objects which will be used during the search
+		regionCount = VMCountRegionsWithAttributes( process, VMREGION_READABLE | VMREGION_WRITABLE );
+		addresses = TCMakeArray( TC_BUFFER_SIZE / sizeof(TCAddress), sizeof(TCAddress) );
+		values = TCMakeArray( TC_BUFFER_SIZE / valueSize, valueSize );
+		regions = TCMakeArray( 0, sizeof(TCAddress) );
+		perRegion = TCMakeArray( 0, sizeof(unsigned) );
+		addressPtr = TCArrayBytes( addresses );
+		valuePtr = TCArrayBytes( values );
+		
+		ChazLog( @"SearchContext: varType=%i, intSign=%i, op=%i, value=%@", _variableType, _integerSign, _operator, [value stringValue] );
+	}
+	return self;
+}
+
+- (id)initWithLastContext:(SearchContext *)context searchOperator:(TCSearchOperator)op
+{
+	unsigned valueSize;
+	
+	if ( [super init] ) {
+		if ( !context ) {
+			[self release];
+			return nil;
+		}
+		valueSize = TCArrayElementSize(context->values);
+		process = context->process;
+		_variableType = [context variableType];
+		_integerSign = [context integerSign];
+		_operator = op;
+		_searchType = TCLastValue;
+		compareFunc = [self compareFunction];
+		
+		regionCount = TCArrayElementCount( context->regions );
+		addresses = TCMakeArray( TC_BUFFER_SIZE / sizeof(TCAddress), sizeof(TCAddress) );
+		values = TCMakeArray( TC_BUFFER_SIZE / valueSize, valueSize );
+		regions = TCMakeArray( 0, sizeof(TCAddress) );
+		perRegion = TCMakeArray( 0, sizeof(unsigned) );
+		lastAddresses = context->addresses;
+		lastValues = context->values;
+		lastRegions = context->regions;
+		lastPerRegion = context->perRegion;
+		addressPtr = TCArrayBytes( addresses );
+		valuePtr = TCArrayBytes( values );
+		lastAddressPtr = TCArrayBytes( lastAddresses );
+		lastValuePtr = TCArrayBytes( lastValues );
+		lastRegionPtr = TCArrayBytes( lastRegions );
+		lastPerRegionPtr = TCArrayBytes( lastPerRegion );
+		
+		ChazLog( @"SearchContext: varType=%i, intSign=%i, op=%i", _variableType, _integerSign, _operator );
+	}
+	return self;
+}
+
+- (id)initWithLastContext:(SearchContext *)context searchOperator:(TCSearchOperator)op value:(Variable *)val
+{
+	unsigned valueSize;
+	
+	if ( [super init] ) {
+		if ( !context || !val || ([val type] != [context variableType]) // and search values can't be bigger than the last time.
+			 || (([context variableType] == TCString) && ([val valueSize] > TCArrayElementSize(context->values))) ) {
+			[self release];
+			return nil;
+		}
+		value = [val retain];
+		valueSize = [value valueSize];
+		process = context->process;
+		_variableType = [context variableType];
+		_integerSign = [context integerSign];
+		_operator = op;
+		_searchType = TCGivenValue;
+		compareFunc = [self compareFunction];
+		
+		regionCount = TCArrayElementCount( context->regions );
+		addresses = TCMakeArray( TC_BUFFER_SIZE / sizeof(TCAddress), sizeof(TCAddress) );
+		values = TCMakeArray( TC_BUFFER_SIZE / valueSize, valueSize );
+		regions = TCMakeArray( 0, sizeof(TCAddress) );
+		perRegion = TCMakeArray( 0, sizeof(unsigned) );
+		lastAddresses = context->addresses;
+		lastValues = context->values;
+		lastRegions = context->regions;
+		lastPerRegion = context->perRegion;
+		addressPtr = TCArrayBytes( addresses );
+		valuePtr = TCArrayBytes( values );
+		lastAddressPtr = TCArrayBytes( lastAddresses );
+		lastValuePtr = TCArrayBytes( lastValues );
+		lastRegionPtr = TCArrayBytes( lastRegions );
+		lastPerRegionPtr = TCArrayBytes( lastPerRegion );
+		
+		ChazLog( @"SearchContext: varType=%i, intSign=%i, op=%i, value=%@", _variableType, _integerSign, _operator, [value stringValue] );
+	}
+	return self;
+}
+
+- (void)dealloc
+{
+	ChazLog( @"SearchContext %p dealloc", self );
+	TCReleaseArray( addresses );
+	TCReleaseArray( values );
+	TCReleaseArray( regions );
+	TCReleaseArray( perRegion );
+	
+	if ( buffer ) {
+		free( buffer );
+	}
+	[value release];
+	
+	[super dealloc];
+}
+
+
+#pragma mark Accessors
+
+- (TCVariableType)variableType
+{
+	return _variableType;
+}
+
+- (TCIntegerSign)integerSign
+{
+	return _integerSign;
+}
+
+- (TCSearchOperator)searchOperator
+{
+	return _operator;
+}
+
+
+- (BOOL (*)(void const *, void const *))compareFunction
+{
+	// here begins a very pretty collection of switch and if statements. enjoy!
+	switch ( _operator ) {
+		case TCEqual:
+			switch ( _variableType ) {
+				case TCFloat: return EqualFloat;
+				case TCDouble: return EqualDouble;
+			}
+			if ( _integerSign == TCSigned ) {
+				switch ( _variableType ) {
+					case TCInt64: return EqualInt64;
+					case TCInt32: return EqualInt32;
+					case TCInt16: return EqualInt16;
+					case TCInt8: return EqualInt8;
+				}
+			}
+			else {
+				switch ( _variableType ) {
+					case TCInt64: return EqualUInt64;
+					case TCInt32: return EqualUInt32;
+					case TCInt16: return EqualUInt16;
+					case TCInt8: return EqualUInt8;
+				}
+			}
+			break;
+		case TCNotEqual:
+			switch ( _variableType ) {
+				case TCFloat: return NotEqualFloat;
+				case TCDouble: return NotEqualDouble;
+			}
+			if ( _integerSign == TCSigned ) {
+				switch ( _variableType ) {
+					case TCInt64: return NotEqualInt64;
+					case TCInt32: return NotEqualInt32;
+					case TCInt16: return NotEqualInt16;
+					case TCInt8: return NotEqualInt8;
+				}
+			}
+			else {
+				switch ( _variableType ) {
+					case TCInt64: return NotEqualUInt64;
+					case TCInt32: return NotEqualUInt32;
+					case TCInt16: return NotEqualUInt16;
+					case TCInt8: return NotEqualUInt8;
+				}
+			}
+			break;
+		case TCLessThan:
+			switch ( _variableType ) {
+				case TCFloat: return LessThanFloat;
+				case TCDouble: return LessThanDouble;
+			}
+			if ( _integerSign == TCSigned ) {
+				switch ( _variableType ) {
+					case TCInt64: return LessThanInt64;
+					case TCInt32: return LessThanInt32;
+					case TCInt16: return LessThanInt16;
+					case TCInt8: return LessThanInt8;
+				}
+			}
+			else {
+				switch ( _variableType ) {
+					case TCInt64: return LessThanUInt64;
+					case TCInt32: return LessThanUInt32;
+					case TCInt16: return LessThanUInt16;
+					case TCInt8: return LessThanUInt8;
+				}
+			}
+			break;
+		case TCGreaterThan:
+			switch ( _variableType ) {
+				case TCFloat: return GreaterThanFloat;
+				case TCDouble: return GreaterThanDouble;
+			}
+			if ( _integerSign == TCSigned ) {
+				switch ( _variableType ) {
+					case TCInt64: return GreaterThanInt64;
+					case TCInt32: return GreaterThanInt32;
+					case TCInt16: return GreaterThanInt16;
+					case TCInt8: return GreaterThanInt8;
+				}
+			}
+			else {
+				switch ( _variableType ) {
+					case TCInt64: return GreaterThanUInt64;
+					case TCInt32: return GreaterThanUInt32;
+					case TCInt16: return GreaterThanUInt16;
+					case TCInt8: return GreaterThanUInt8;
+				}
+			}
+			break;
+	}
+	return NULL;
+}
+
+- (int (*)(id, unsigned))iterationFunction
+{
+	if ( _searchType == TCGivenValue ) {
+		if ( !lastAddresses ) {
+			if ( _variableType == TCString ) {
+				return SearchStringIteration;
+			}
+			else {
+				return SearchIteration;
+			}
+		}
+		else {
+			if ( _variableType == TCString ) {
+				return SearchStringIterationAgain;
+			}
+			else {
+				return SearchIterationAgain;
+			}
+		}
+	}
+	else if ( _searchType == TCLastValue ) {
+		if ( _variableType == TCString ) {
+			return SearchStringIterationLastValue;
+		}
+		else {
+			return SearchIterationLastValue;
+		}
+	}
+	return NULL;
+}
+
+
+@end