iconhide.xm

//
//  Wallpaper.mm
//  Wallpaper
//
//  Created by Public Nuisance on 6/25/09.
//  Copyright __MyCompanyName__ 2009. All rights reserved.
//
//  MobileSubstrate, libsubstrate.dylib, and substrate.h are
//  created and copyrighted by Jay Freeman a.k.a saurik and 
//  are protected by various means of open source licensing.
//

#include <notify.h>
#include <time.h>

/********************************************************************************************************
// defines
********************************************************************************************************/
#define IH_VERSION "2.2"
#define HIDLIBPATH "/var/mobile/Library/LibHide/hidden.plist"

#define SYS_VER_2(_x) (_x >= 2.0 && _x < 3.0) ? YES : NO
#define SYS_VER_3(_x) (_x >= 3.0 && _x < 4.0) ? YES : NO
#define SYS_VER_4(_x) (_x >= 4.0 && _x < 5.0) ? YES : NO
#define SYS_VER_5(_x) (_x >= 5.0) ? YES : NO

#define _release(object) \
do if (object != nil) { \
[object release]; \
object = nil; \
} while (false)

#define NSLine() NSLog(@"%s %s %d", __FILE__, __FUNCTION__, __LINE__)
#define HookLog(); \
{ \
uint32_t bt; \
__asm__("mov %0, lr": "=r"(bt)); \
NSLog(@"[%@ %s] bt=%x", [[self class] description], sel_getName(sel), bt); \
}

/*******************************************************************************************************
// Declarations (for private classes/methods)
********************************************************************************************************/
@interface SBIcon : UIView
- (void)setShowsImages:(BOOL)images;
@end
@interface SBIcon (Firmware2x3x)
- (id)displayIdentifier;
@end
@interface SBIcon (Firmware4x)
- (id)leafIdentifier;
@end

@interface SBApplicationIcon : SBIcon @end

@interface SBIconController : NSObject @end

@interface SBIconList : UIView @end
@interface SBIconList (Firmware2x3x)
- (BOOL)firstFreeSlotX:(int *)x Y:(int *)y;
- (id)placeIcon:(id)icon atX:(int)x Y:(int)y animate:(BOOL)animate moveNow:(BOOL)now;
@end

@interface SBIconModel : NSObject
- (void)setVisibilityOfIconsWithVisibleTags:(id)visibleTags hiddenTags:(id)tags;
- (void)relayout;
@end
@interface SBIconModel (Firmware2x3x)
@property(readonly, retain) NSMutableArray *iconLists;
- (id)addEmptyIconList;
- (id)iconForDisplayIdentifier:(id)displayIdentifier;
- (id)iconListContainingIcon:(id)icon;
@end
@interface SBIconModel (Firmware4x)
- (void)addIcon:(id)icon;
- (id)leafIconForIdentifier:(id)identifier;
@end

@interface SBPlatformController : NSObject @end

@interface SBSearchView : UIView
- (BOOL)isKeyboardVisible;
@end

@interface SBSearchController : NSObject
+ (id)sharedInstance;
@property(retain, nonatomic) SBSearchView *searchView;
@end

@interface SBSearchModel : NSObject // Firmware4x
- (BOOL)hasQueryString;
@end

@interface SBUIController : NSObject @end

@interface UIApplication (SpringBoard)
- (void)relaunchSpringBoard;
@end

/********************************************************************************************************
// Globals
********************************************************************************************************/
static NSMutableArray*	global_HiddenIconIds = nil;
static NSMutableArray*	global_HiddenSpotlightIconIds = nil;
static bool				global_Rehide = YES;
static float				global_SystemVersion = 3.0;
static SBSearchController*	global_SearchController = nil;
static bool					global_switcherShowing = NO;


#pragma mark -
#pragma mark hidden icon hooks

/********************************************************************************************************
//                                      SBIconModel hooks
********************************************************************************************************/
%hook SBIconModel

/********************************************************************************************************
// 2.x renamed function
********************************************************************************************************/
%group GSBIconModel_Firmware2x

- (BOOL)iconIsVisible:(id)iconId
{
	BOOL isVisible = %orig;
	SBApplicationIcon* icon = (SBApplicationIcon*) iconId;
	
	if(global_HiddenIconIds != nil)
	{
		if([global_HiddenIconIds containsObject:[icon displayIdentifier]])
		{
			isVisible = NO;
		}
	}
	return isVisible;	
}

%end // GSBIconModel_Firmware2x

/********************************************************************************************************
// 3.x renamed function
********************************************************************************************************/
%group GSBIconModel_Firmware3x

- (BOOL)isIconVisible:(id)iconId
{
	BOOL isVisible = YES;
	BOOL isShowingSearch = NO;
	SBApplicationIcon* icon = (SBApplicationIcon*) iconId;

	if(global_switcherShowing == NO)
	{
		Class SBSearchController = objc_getClass("SBSearchController");
		id searchController = [SBSearchController sharedInstance];
		if(searchController != nil)
		{
			SBSearchView* SearchView = [searchController searchView];
			if(SearchView != nil)
			{
				isShowingSearch = [SearchView isKeyboardVisible];
			}
		}

		if(isShowingSearch == NO)
		{
			isVisible = %orig;
			
			if(global_HiddenIconIds != nil)
			{
				//NSLog(@"Icon: %@", [icon displayIdentifier]);
				if([global_HiddenIconIds containsObject:[icon displayIdentifier]])
				{
					isVisible = NO;
				}
			}
		}
		
		// Check for hidden in spotlight key
		else if(global_HiddenSpotlightIconIds != nil)
		{
			if([global_HiddenSpotlightIconIds containsObject:[icon displayIdentifier]])
			{
				isVisible = NO;
			}
		}
	}
	
	return isVisible;
}

%end // GSBIconModel_Firmware3x

/********************************************************************************************************
// 4.x renamed function
********************************************************************************************************/
%group GSBIconModel_Firmware4x

- (BOOL)isIconVisible:(id)iconId
{
	// FIXME: Are the respondsToSelector: calls used in this hook necessary?

	BOOL isVisible = YES;
	BOOL isShowingSearch = NO;
	SBApplicationIcon* icon = (SBApplicationIcon*) iconId;

	if(global_switcherShowing == NO)
	{
		if(global_SystemVersion >= 4.0f)
		{
			//return %orig;
			if(global_SearchController != nil)
			{
				SBSearchView* SearchView = [global_SearchController searchView];
				if(SearchView != nil)
				{
					isShowingSearch = [SearchView isKeyboardVisible];
				}
			}
		}
		else 
		{
			// FIXME: This code is never called (?). What is its purpose?
			Class SBSearchModel = objc_getClass("SBSearchModel");
			id searchModel = [SBSearchModel sharedInstance];
			if(searchModel != nil)
			{
				isShowingSearch = [searchModel hasQueryString];
			}
		}

		if(isShowingSearch == NO)
		{
			isVisible = %orig;
			
			if(global_HiddenIconIds != nil)
			{
				//NSLog(@"LibHide: %@", [icon leafIdentifier]);
				if([icon respondsToSelector:@selector(leafIdentifier)] &&  
				   [global_HiddenIconIds containsObject:[icon leafIdentifier]])
				{
					isVisible = NO;
				}
			}
		}
		
		// Check for hidden in spotlight key
		else if(global_HiddenSpotlightIconIds != nil)
		{
			if([icon respondsToSelector:@selector(leafIdentifier)] &&
			   [global_HiddenSpotlightIconIds containsObject:[icon leafIdentifier]])
			{
				isVisible = NO;
			}
		}
	}
	
	if([icon respondsToSelector:@selector(leafIdentifier)])
	{
		NSString* leafId = [icon leafIdentifier];
		if([leafId isEqualToString: @"com.apple.AdSheet"] || 
			[leafId isEqualToString: @"com.apple.DemoApp"] ||
			[leafId isEqualToString: @"com.apple.iphoneos.iPodOut"] ||
			[leafId isEqualToString: @"com.apple.TrustMe"] ||
			[leafId isEqualToString: @"com.apple.webapp"] ||
			[leafId isEqualToString: @"com.apple.WebSheet"] ||
			[leafId isEqualToString: @"com.apple.nike"] )
		{
			isVisible = %orig;
		}
	}
	
	return isVisible;
}

%end // GSBIconModel_Firmware4x

%end // SBIconModel

/********************************************************************************************************
//                                   SBSearchController hooks
********************************************************************************************************/
%hook SBSearchController

/********************************************************************************************************
********************************************************************************************************/
- (id)init
{
	id newSearchController = %orig;
	
	global_SearchController = newSearchController;
	
	return newSearchController;
}

%end // SBSearchController

/********************************************************************************************************
//                                     SBUIController hooks
********************************************************************************************************/


%hook SBPlatformController
%group GHiddenIcons3x
- (void)setInfo:(NSMutableArray *)arrayOfApps forCapability:(NSString *)Capability {
	if (Capability != nil && [Capability isEqualToString:@"application-display-identifiers"]) {
		NSMutableArray *mutableArray = [NSMutableArray arrayWithArray:arrayOfApps];
		[mutableArray addObjectsFromArray:global_HiddenIconIds];
		
		%orig(mutableArray, Capability);
		return;
	}
	
	%orig;
}
%end

%group GHiddenIcons4x
- (void)setValue:(NSMutableArray *)arrayOfApps forCapability:(NSString *)Capability {
	if (Capability != nil && [Capability isEqualToString:@"application-display-identifiers"]) {
		NSMutableArray *mutableArray = [NSMutableArray arrayWithArray:arrayOfApps];
		[mutableArray addObjectsFromArray:global_HiddenIconIds];
		
		%orig(mutableArray, Capability);
		return;
	}
	
	%orig;
}
%end
%end

/********************************************************************************************************
********************************************************************************************************/
%hook SBUIController
%group GSwitcherHooks4x5x
- (void)_toggleSwitcher
{
	global_switcherShowing = YES;
	%orig;
}
%end

%group GSwitcherHooks4x
- (void)_dismissSwitcher:(double)switcher
{
	%orig;
	global_switcherShowing = NO;
}
%end

%group GSwitcherHooks5x
- (void)dismissSwitcherAnimated:(BOOL)animated {
	%orig;
	global_switcherShowing = NO;
}
%end
%end

#pragma mark -
#pragma mark non-hooks

/********************************************************************************************************
//                                        Non-hooks
********************************************************************************************************/

/********************************************************************************************************
// LoadHiddenIconList
********************************************************************************************************/
int LoadHiddenIconList()
{
	int HiddenIcons = 0;
	
	if([[NSFileManager defaultManager] fileExistsAtPath:@HIDLIBPATH])
	{
		NSMutableDictionary* Dict = [NSMutableDictionary dictionaryWithContentsOfFile:@HIDLIBPATH];

		if(global_HiddenIconIds != nil)
		{
			[global_HiddenIconIds release];
		}
		
		global_HiddenIconIds = [[NSMutableArray alloc] initWithArray:[Dict objectForKey:@"Hidden"]];
		if(global_HiddenIconIds != nil)
		{
			if([global_HiddenIconIds containsObject:@"com.apple.mobileipod"])
			{
				[global_HiddenIconIds removeObject:@"com.apple.mobileipod"];
				[global_HiddenIconIds addObject:@"com.apple.mobileipod-MediaPlayer"];
			}
			
			HiddenIcons = [global_HiddenIconIds count];
		}
		
		// This list is for icons to hide from spotlight. 
		global_HiddenSpotlightIconIds = [[NSMutableArray alloc] initWithArray:[Dict objectForKey:@"Spotlight"]];
	}
	
	NSLog(@"LibHide: Returning %d icons hidden", HiddenIcons);
	
	return HiddenIcons;
}

/********************************************************************************************************
********************************************************************************************************/
void* MSHookIvar(id self, const char *unitName) 
{
    Ivar ivar =  class_getInstanceVariable(object_getClass(self), unitName);
    void *pointer = (ivar == NULL ? NULL : (char*)(self) + ivar_getOffset(ivar));
    return pointer;
}

/********************************************************************************************************
// IconHide_HiddenIconsChanged - Runs when the nofify_post(com.libhide.hiddeniconschanged) is called.
********************************************************************************************************/
static void IconHide_HiddenIconsChanged(CFNotificationCenterRef center,
											  void *observer,
											  CFStringRef name,
											  const void *object,
											  CFDictionaryRef info
											  )
{
	if([[[UIDevice currentDevice] systemVersion] hasPrefix:@"2."])
	{
		UIApplication* theApp = [UIApplication sharedApplication];
		if([theApp respondsToSelector:@selector(relaunchSpringBoard)])
		{
			[theApp relaunchSpringBoard];
		}
	}
	
	else if([[[UIDevice currentDevice] systemVersion] hasPrefix:@"5."])
	{
		LoadHiddenIconList();
		
		Class SBIconModel = objc_getClass("SBIconModel");
		SBIconModel* iconModel = (SBIconModel*)[SBIconModel sharedInstance];
		NSSet** _visibleIconTags = (NSSet**)MSHookIvar(iconModel, "_visibleIconTags");
		NSSet** _hiddenIconTags  = (NSSet**)MSHookIvar(iconModel, "_hiddenIconTags");
		if(*_visibleIconTags != NULL && *_hiddenIconTags != NULL &&
		    _visibleIconTags != NULL && _hiddenIconTags!=NULL)
		{
			global_Rehide = YES;
			NSSet* visibleIconTags = [NSSet setWithSet:*_visibleIconTags];
			NSSet* hiddenIconTags = [NSSet setWithSet:*_hiddenIconTags];
			[iconModel setVisibilityOfIconsWithVisibleTags:visibleIconTags  hiddenTags:hiddenIconTags];
			[iconModel relayout];
		}
	}

	else 
	{
		LoadHiddenIconList();
		
		Class SBIconModel = objc_getClass("SBIconModel");
		SBIconModel* iconModel = (SBIconModel*)[SBIconModel sharedInstance];
		NSSet** _visibleIconTags = (NSSet**)MSHookIvar(iconModel, "_visibleIconTags");
		NSSet** _hiddenIconTags  = (NSSet**)MSHookIvar(iconModel, "_hiddenIconTags");
		if(*_visibleIconTags != NULL && *_hiddenIconTags != NULL)
		{
			global_Rehide = YES;
			NSArray* visibleIconTags = [*_visibleIconTags allObjects];
			NSArray* hiddenIconTags = [*_hiddenIconTags allObjects];
			[iconModel setVisibilityOfIconsWithVisibleTags:visibleIconTags  hiddenTags:hiddenIconTags];
		}
	}

}

/********************************************************************************************************
********************************************************************************************************/
float getSystemVersion()
{
	float Version = 4.0;
	
	if([[NSFileManager defaultManager] fileExistsAtPath:@"/System/Library/CoreServices/SystemVersion.plist"])
	{
		NSMutableDictionary* Dict = [NSMutableDictionary dictionaryWithContentsOfFile:@"/System/Library/CoreServices/SystemVersion.plist"];
		
		NSLog(@"LibHide: Querying system version");
		Version = [[Dict objectForKey:@"ProductVersion"] floatValue];
	}
	
	NSLog(@"LibHide: returning version %f", Version);
	
	return Version;
}

#pragma mark -
#pragma mark dylib initializer

/********************************************************************************************************
// dylib initializer or entry point.
********************************************************************************************************/
%ctor {
	NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];	
	NSLog(@"LibHide: v" IH_VERSION " initializer");
	
	global_SystemVersion = getSystemVersion();
	
	if(LoadHiddenIconList() != 0)
	{
		if (SYS_VER_2(global_SystemVersion) || SYS_VER_3(global_SystemVersion)) {
			%init(GHiddenIcons3x);
			
			if(SYS_VER_2(global_SystemVersion))
				%init(GSBIconModel_Firmware2x);
			
			else if(SYS_VER_3(global_SystemVersion))
				%init(GSBIconModel_Firmware3x);
		}
		
		else {
			%init(GSBIconModel_Firmware4x);
			%init(GHiddenIcons4x);
			
			%init(GSwitcherHooks4x5x);
			if (SYS_VER_4(global_SystemVersion))
				%init(GSwitcherHooks4x);
			else
				%init(GSwitcherHooks5x);
		}
	}

	// Initialize remaining (non-grouped) hooks
	%init;
	
	// Set the rehide flag for setVisibilityOfIcons* function.
	global_Rehide = YES;

	CFNotificationCenterAddObserver(CFNotificationCenterGetDarwinNotifyCenter(),
										 NULL, 
										 &IconHide_HiddenIconsChanged, 
										 CFSTR("com.libhide.hiddeniconschanged"), 
										 NULL, 
										 CFNotificationSuspensionBehaviorDeliverImmediately);	
	
	NSLog(@"LibHide: initializer completed and you're not in safe mode!");

	[pool drain];
}

/* vim: set filetype=objcpp sw=4 ts=4 sts=4 noexpandtab textwidth=80 ff=unix: */