#include <CoreFoundation/CoreFoundation.h>

#include <CoreFoundation/CFPriv.h>

#include <sys/types.h>

#include <sys/stat.h>

#include <fcntl.h>

#include <dlfcn.h>

#include <unistd.h>

#include <objc/runtime.h>

#include "CydiaSubstrate.h"

#include "Log.hpp"

#include "Environment.hpp"

#define ForSaurik 0

static char MSWatch[PATH_MAX];

static char MSStack[SIGSTKSZ];

static void MSAction(int sig, siginfo_t *info, void *uap) {

open(MSWatch, O_CREAT | O_RDWR, 0644);

raise(sig);

}

#define Libraries_ "/Library/MobileSubstrate/DynamicLibraries"

#define Safety_ "/Library/Frameworks/CydiaSubstrate.framework/MobileSafety.dylib"

extern "C" char ***_NSGetArgv(void);

MSInitialize {

#ifndef __arm__

if (dlopen("/System/Library/Frameworks/Security.framework/Security", RTLD_LAZY | RTLD_NOLOAD) == NULL)

return;

#endif

#if 1

CFBundleRef bundle(CFBundleGetMainBundle());

CFStringRef identifier(bundle == NULL ? NULL : CFBundleGetIdentifier(bundle));

#else

#endif

char *argv0(**_NSGetArgv());

char *slash(strrchr(argv0, '/'));

slash = slash == NULL ? argv0 : slash + 1;

Class (*NSClassFromString)(CFStringRef) = reinterpret_cast<Class (*)(CFStringRef)>(dlsym(RTLD_DEFAULT, "NSClassFromString"));

MSLog(MSLogLevelNotice, "MS:Notice: Installing: %@ [%s] (%.2f)", identifier, slash, kCFCoreFoundationVersionNumber);

CFURLRef home(CFCopyHomeDirectoryURLForUser(NULL));

if (home == NULL) {

MSLog(MSLogLevelError, "MS:Error: Unable to Copy HOME");

return;

}

char watch[PATH_MAX];

CFURLGetFileSystemRepresentation(home, TRUE, reinterpret_cast<UInt8 *>(watch), sizeof(watch));

CFRelease(home);

const char *dead(NULL);

if (identifier != NULL && CFEqual(identifier, CFSTR("com.apple.springboard")))

dead = "com.saurik.MobileSubstrate.SpringBoard.Dead.dat";

if (dead != NULL) {

sprintf(MSWatch, "%s/Library/Preferences/%s", watch, dead);

if (access(MSWatch, R_OK) == 0) {

MSClearEnvironment();

MSLog(MSLogLevelWarning, "MS:Warning: Deactivating Substrate");

if (unlink(MSWatch) == -1)

MSLog(MSLogLevelError, "MS:Error: Cannot Clear: %s", strerror(errno));

return;

}

}

bool safe(false);

const char *dat(NULL);

if (identifier != NULL && CFEqual(identifier, CFSTR("com.apple.springboard")))

dat = "com.saurik.mobilesubstrate.dat";

if (identifier == NULL && (strcmp(slash, "CommCenter") == 0 || strcmp(slash, "CommCenterClassic") == 0))

dat = "com.saurik.MobileSubstrate.CommCenter.Safe.dat";

if (dat != NULL) {

strcat(watch, "/Library/Preferences/");

strcat(watch, dat);

if (access(watch, R_OK) == 0) {

MSClearEnvironment();

MSLog(MSLogLevelWarning, "MS:Warning: Entering Safe Mode");

if (unlink(watch) == -1)

MSLog(MSLogLevelError, "MS:Error: Cannot Clear: %s", strerror(errno));

safe = true;

}

stack_t stack;

stack.ss_size = sizeof(MSStack);

stack.ss_flags = 0;

stack.ss_sp = MSStack;

bool stacked;

if (sigaltstack(&stack, NULL) != -1)

stacked = true;

else {

stacked = false;

MSLog(MSLogLevelWarning, "MS:Error: Cannot Stack: %s", strerror(errno));

}

struct sigaction action;

memset(&action, 0, sizeof(action));

action.sa_sigaction = &MSAction;

action.sa_flags = SA_SIGINFO | SA_RESETHAND;

if (stacked)

action.sa_flags |= SA_ONSTACK;

sigemptyset(&action.sa_mask);

struct sigaction old;

#define HookSignal(signum) \

sigaction(signum, NULL, &old); { \

sigaction(signum, &action, NULL); \

}

HookSignal(SIGQUIT)

HookSignal(SIGILL)

HookSignal(SIGTRAP)

HookSignal(SIGABRT)

HookSignal(SIGEMT)

HookSignal(SIGFPE)

HookSignal(SIGBUS)

HookSignal(SIGSEGV)

HookSignal(SIGSYS)

}

if (dead == NULL && dat == NULL)

MSWatch[0] = '\0';

else if (dead == NULL || !safe)

strcpy(MSWatch, watch);

CFURLRef libraries(CFURLCreateFromFileSystemRepresentation(kCFAllocatorDefault, reinterpret_cast<const UInt8 *>(Libraries_), sizeof(Libraries_) - 1, TRUE));

CFBundleRef folder(CFBundleCreate(kCFAllocatorDefault, libraries));

CFRelease(libraries);

if (folder == NULL)

return;

CFArrayRef dylibs(CFBundleCopyResourceURLsOfType(folder, CFSTR("dylib"), NULL));

CFRelease(folder);

for (CFIndex i(0), count(CFArrayGetCount(dylibs)); i != count; ++i) {

CFURLRef dylib(reinterpret_cast<CFURLRef>(CFArrayGetValueAtIndex(dylibs, i)));

char path[PATH_MAX];

CFURLGetFileSystemRepresentation(dylib, TRUE, reinterpret_cast<UInt8 *>(path), sizeof(path));

size_t length(strlen(path));

memcpy(path + length - 5, "plist", 5);

CFURLRef plist(CFURLCreateFromFileSystemRepresentation(kCFAllocatorDefault, reinterpret_cast<UInt8 *>(path), length, FALSE));

CFDataRef data;

if (!CFURLCreateDataAndPropertiesFromResource(kCFAllocatorDefault, plist, &data, NULL, NULL, NULL))

data = NULL;

CFRelease(plist);

CFDictionaryRef meta(NULL);

if (data != NULL) {

CFStringRef error(NULL);

meta = reinterpret_cast<CFDictionaryRef>(CFPropertyListCreateFromXMLData(kCFAllocatorDefault, data, kCFPropertyListImmutable, &error));

CFRelease(data);

if (meta == NULL && error != NULL) {

MSLog(MSLogLevelError, "MS:Error: Corrupt PropertyList: %@", dylib);

continue;

}

}

// XXX: eventually this should become false: dylibs must have a filter

bool load(!safe);

if (meta != NULL) {

if (CFDictionaryRef filter = reinterpret_cast<CFDictionaryRef>(CFDictionaryGetValue(meta, CFSTR("Filter")))) {

load = true;

int value(0);

if (CFNumberRef flags = reinterpret_cast<CFNumberRef>(CFDictionaryGetValue(filter, CFSTR("Flags")))) {

if (CFGetTypeID(flags) != CFNumberGetTypeID() || !CFNumberGetValue(flags, kCFNumberIntType, &value)) {

MSLog(MSLogLevelError, "MS:Error: Unable to Read Flags: %@", flags);

load = false;

goto release;

}

}

#define MSFlagWhenSafe (1 << 0)

#define MSFlagNotNoSafe (1 << 1)

if ((value & MSFlagWhenSafe) == 0 && safe) {

load = false;

goto release;

}

if ((value & MSFlagNotNoSafe) != 0 && !safe) {

load = false;

goto release;

}

if (CFArrayRef version = reinterpret_cast<CFArrayRef>(CFDictionaryGetValue(filter, CFSTR("CoreFoundationVersion")))) {

load = false;

if (CFIndex count = CFArrayGetCount(version)) {

if (count > 2) {

MSLog(MSLogLevelError, "MS:Error: Invalid CoreFoundationVersion: %@", version);

goto release;

}

CFNumberRef number;

double value;

number = reinterpret_cast<CFNumberRef>(CFArrayGetValueAtIndex(version, 0));

if (CFGetTypeID(number) != CFNumberGetTypeID() || !CFNumberGetValue(number, kCFNumberDoubleType, &value)) {

MSLog(MSLogLevelError, "MS:Error: Unable to Read CoreFoundationVersion[0]: %@", number);

goto release;

}

if (value > kCFCoreFoundationVersionNumber)

goto release;

if (count != 1) {

number = reinterpret_cast<CFNumberRef>(CFArrayGetValueAtIndex(version, 1));

if (CFGetTypeID(number) != CFNumberGetTypeID() || !CFNumberGetValue(number, kCFNumberDoubleType, &value)) {

MSLog(MSLogLevelError, "MS:Error: Unable to Read CoreFoundationVersion[1]: %@", number);

goto release;

}

if (value <= kCFCoreFoundationVersionNumber)

goto release;

}

}

load = true;

}

bool any;

if (CFStringRef mode = reinterpret_cast<CFStringRef>(CFDictionaryGetValue(filter, CFSTR("Mode"))))

any = CFEqual(mode, CFSTR("Any"));

else

any = false;

if (any)

load = false;

if (CFArrayRef executables = reinterpret_cast<CFArrayRef>(CFDictionaryGetValue(filter, CFSTR("Executables")))) {

if (!any)

load = false;

CFStringRef name(CFStringCreateWithCStringNoCopy(kCFAllocatorDefault, slash, kCFStringEncodingUTF8, kCFAllocatorNull));

for (CFIndex i(0), count(CFArrayGetCount(executables)); i != count; ++i) {

CFStringRef executable(reinterpret_cast<CFStringRef>(CFArrayGetValueAtIndex(executables, i)));

if (CFEqual(executable, name)) {

if (ForSaurik)

MSLog(MSLogLevelNotice, "MS:Notice: Found: %@", name);

load = true;

break;

}

}

CFRelease(name);

if (!any && !load)

goto release;

}

if (CFArrayRef bundles = reinterpret_cast<CFArrayRef>(CFDictionaryGetValue(filter, CFSTR("Bundles")))) {

if (!any)

load = false;

for (CFIndex i(0), count(CFArrayGetCount(bundles)); i != count; ++i) {

CFStringRef bundle(reinterpret_cast<CFStringRef>(CFArrayGetValueAtIndex(bundles, i)));

if (CFBundleGetBundleWithIdentifier(bundle) != NULL) {

if (ForSaurik)

MSLog(MSLogLevelNotice, "MS:Notice: Found: %@", bundle);

load = true;

break;

}

}

if (!any && !load)

goto release;

}

if (CFArrayRef classes = reinterpret_cast<CFArrayRef>(CFDictionaryGetValue(filter, CFSTR("Classes")))) {

if (!any)

load = false;

if (NSClassFromString != NULL)

for (CFIndex i(0), count(CFArrayGetCount(classes)); i != count; ++i) {

CFStringRef _class(reinterpret_cast<CFStringRef>(CFArrayGetValueAtIndex(classes, i)));

if (NSClassFromString(_class) != NULL) {

if (ForSaurik)

MSLog(MSLogLevelNotice, "MS:Notice: Found: %@", _class);

load = true;

break;

}

}

if (!any && !load)

goto release;

}

}

release:

CFRelease(meta);

}

if (!load)

continue;

memcpy(path + length - 5, "dylib", 5);

MSLog(MSLogLevelNotice, "MS:Notice: Loading: %s", path);

if (MSWatch[0] != '\0') {

int fd(open(MSWatch, O_CREAT | O_RDWR, 0644));

if (fd == -1)

MSLog(MSLogLevelError, "MS:Error: Cannot Set: %s", strerror(errno));

else if (close(fd) == -1)

MSLog(MSLogLevelError, "MS:Error: Cannot Close: %s", strerror(errno));

}

void *handle(dlopen(path, RTLD_LAZY | RTLD_GLOBAL));

if (MSWatch[0] != '\0')

if (unlink(MSWatch) == -1)

MSLog(MSLogLevelError, "MS:Error: Cannot Reset: %s", strerror(errno));

if (handle == NULL) {

MSLog(MSLogLevelError, "MS:Error: %s", dlerror());

continue;

}

}

if (!safe)

setenv(SubstrateSafeMode_, "0", false);

if (false) {

MSLog(MSLogLevelNotice, "MobileSubstrate fell asleep... I'll wake him up in 10 seconds ;P");

sleep(10);

}

}