## mark a section (Learn more)

#include "Shell.h" using namespace std; #if !defined(__APPLE__) bool run = true; #endif #if defined(WIN32) #define WIN32_LEAN_AND_MEAN #include <windows.h> #endif CFStringRef cli_afc_name; COMMAND normal_shell[] = { { "help", sh_help, ">> help <command> - Display help information on <command>. No args lists commands." }, { "cd", n_cd, ">> cd <path> - Change directory to <path>" }, { "lcd", n_lcd, ">> lcd <path> - Change local directory to <path>" }, { "ls", n_ls, ">> ls [path] - List directory at current working path or [path]" }, { "mkdir", n_mkdir, ">> mkdir <path> - Create directory at <path>" }, { "rmdir", n_rmdir, ">> rmdir <path> - Remove directory at <path>" }, { "activate", n_activate, ">> activate <path> - Activate iPhone with .plist file at <path>" }, { "deactivate", n_deactivate, ">> deactivate - Deactivate iPhone" }, { "readvalue", n_readvalue, ">> readvalue <value> - Read <value>. No args lists knownn values." }, { "enterrecovery", n_enterrecovery, ">> enterrecovery - Enter recovery mode. **WARNING: You'll need to restore the iPhone." }, { "reconnect", n_reconnect, ">> reconnect - Exit current shell and reconnect to device." }, { "startservice", n_startservice, ">> startservice <service> - Starts service <service> on the iPhone. No args lists services." }, { "deviceinfo", n_deviceinfo, ">> deviceinfo - Display device info." }, { "getfilesize", n_getfilesize, ">> getfilesize <path> - Display size of file at <path>" }, { "getfile", n_getfile, ">> getfile <path> [localpath] - Get file on iPhone at <path> and write it to [localpath]" }, { "putfile", n_putfile, ">> putfile <localpath> [path] - Put file at <localpath> on iPhone at [path]" }, { "fileinfo", n_fileinfo, ">> fileinfo <path> - Display info for file at <path>" }, { "exit", n_exit, ">> exit - Escape to shell. The other shell, the one whos child i am." }, { "lpwd", n_lpwd, ">> lpwd - Display the current local working directory." }, { "pwd", n_pwd, ">> pwd - Display the current remote working directory." }, { "setafc", n_setafc, ">> setafc <name string> - Set the name of the afc service to use."}, { "run", sh_run, ">> run <path> - runs a script at <path>."}, { (char *)NULL, (shell_funct *)NULL, (char *)NULL } }; COMMAND restore_shell[] = { { "mount", restore_mount, ">> mount <device> <path> - Mount device at path."}, { "partition", restore_partition, ">> partition <device> - Partition device."}, { "erase", restore_erase, ">> erase <device> - Erase device."}, { "ditto", restore_ditto, ">> ditto <path1> <path2> - Copy file at path1 to path2."}, { "umount", restore_umount, ">> umount <path> - Unmount device from path."}, { "foo", erica_foo, ">> erica foo: send message."}, { "FileSystemCheck", restore_filesystemcheck,">> filesystemcheck <device> - Check filesytem on device" }, { "mkdir", restore_mkdir, ">> mkdir <path> - Make directory 'path'."}, { "force", restore_force, ">> force <command> - send cmd to phone."}, { "help", sh_help, ">> help <command> - Display help information on <command>. No args lists commands." }, { "exit", restore_exit, ">> exit - Disconnect and wait for device normal reconnect." }, { "run", sh_run, ">> run <path> - runs a script at <path>."}, { (char *)NULL, (shell_funct *)NULL, (char *)NULL } }; COMMAND recovery_shell[] = { { "restore", recovery_restore, ">> restore - enter restore mode. "}, { "grestore", recovery_grestore, ">> grestore <restorebundle> - enter restore mode interactively."}, { "filecopytophone", recovery_filecopytophone, ">> filecopytophone"}, { "serial", recovery_serial, ">> serial"}, { "bar", erica_bar, ">> bar"}, { "cmd", recovery_cmd, ">> cmd <command> - send command to phone."}, { "exit", recovery_exit, ">> exit - Escape to shell. The other shell, the one whos child i am." }, { "disconnect", recovery_disconnect, ">> disconnect - disconnect from shell and await reconnect." }, { "help", sh_help, ">> help <command> - Display help information on <command>. No args lists commands." }, { "run", sh_run, ">> run <path> - runs a script at <path>."}, { (char *)NULL, (shell_funct *)NULL, (char *)NULL } }; void dfu_connect_callback(am_recovery_device *rdev) { cout << "Detected a DFU connection: How did you do this?" << endl; cout << "Please Report." << endl; cout << flush; } void dfu_disconnect_callback(am_recovery_device *rdev) { cout << "Detected a DFU disconnection: How did you do this?" << endl; cout << "Please Report." << endl; cout << flush; } void recovery_connect_callback(am_recovery_device *rdev) { int retval = AMRestoreEnableFileLogging("restore.log"); ifVerbose cout << "recovery callback: Logging in restore.log: " << retval << endl; ifNotQuiet cout << "recovery callback: Connected in Recovery Mode" << endl; struct shell_state *sh = new shell_state(); sh->dev = NULL; sh->restore_dev = NULL; sh->recovery_dev = rdev; sh->shell_mode = SHELL_RECOVERY; sh->command_array = recovery_shell; sh->remote_path = "#"; sh->local_path = "#"; sh->prompt_string = "(iPHUC Recovery) "; //enter shell ifNotQuiet cout << "recovery callback: Entering shell in Recovery Mode." << endl; int ret = shell(sh); ifNotQuiet cout << "recovery callback: shell returned: " << ret << endl; delete sh; switch(ret) { case SHELL_TERMINATE: ifNotQuiet cout << ">> Nothing left to do. Exiting." << endl; exit(0); default: ifVerbose cout << "recovery_connect_callback: Leaving." << endl; break; } } void recovery_disconnect_callback(am_recovery_device *rdev) { ifNotQuiet cout << endl << ">> Recovery Mode Disconnect." << endl; } void notification(struct am_device_notification_callback_info *info) { struct am_device *dev = info->dev; unsigned int msg = info->msg; int retval; int shell_return_value; // Need more verbosity here. if (msg == ADNCI_MSG_CONNECTED) { ifNotQuiet cout << "notification: iPhone attached." << endl; retval = AMDeviceConnect(dev); if (retval) { if( (getcliflags() & OPT_NORMAL) ) { D("Found restore, but waiting for normal."); return; } ifVerbose cout << "AMDeviceConnect: " << retval << endl; // Check for restore mode ifNotQuiet { cout << "notification: Could not connect." << endl; cout << ">> Attempting to connect to device in Restore mode." << endl; } // build the shell_state struct shell_state *sh = new shell_state(); sh->dev = dev; sh->restore_dev = AMRestoreModeDeviceCreate( 0 , AMDeviceGetConnectionID(dev),0); sh->restore_dev->port = socketForPort(sh->restore_dev, 0xf27e); ifVerbose cout << ">> Restore Mode Port: " << sh->restore_dev->port << endl; sh->shell_mode = SHELL_RESTORE; sh->command_array = restore_shell; sh->remote_path = "#"; sh->local_path = "#"; sh->prompt_string = "(iPHUC Restore) "; //enter shell ifVerbose cout << "notification: Entering shell in Restore Mode." << endl; shell_return_value = shell(sh); delete sh; } else { if( (getcliflags() & OPT_RESTORE) ) { D("Found normal, but waiting for restore."); return; } mach_error_t ret; struct shell_state *sh = new shell_state(); sh->dev = dev; ifVerbose cout << "AMDeviceConnect: " << retval << endl; // Enter normal mode ret = AMDeviceIsPaired(sh->dev); ifVerbose cout << "AMDeviceIsPaired: " << ret << endl; ret = AMDeviceValidatePairing(sh->dev); ifVerbose cout << "AMDeviceValidatePairing: " << ret << endl; ret = AMDeviceStartSession(sh->dev); ifVerbose cout << "AMDeviceStartSession: " << ret << endl; // Start AFC service ret = AMDeviceStartService(sh->dev, cli_afc_name, &(sh->afch), NULL); ifNotQuiet cout << "AMDeviceStartService '" << CFStringGetCStringPtr(cli_afc_name, kCFStringEncodingMacRoman) << "': " << ret << endl; // Open an AFC Connection ret = AFCConnectionOpen(sh->afch, 0, &(sh->conn)); ifVerbose cout << "AFCConnectionOpen: " << ret << endl; /* Turns debug mode on if the environment variable AFCDEBUG is set to a numeric * value, or if the file '/AFCDEBUG' is present and contains a value. */ #if defined(__APPLE__) ifVerbose cout << "AFCPlatformInit: (no retval)" << endl; AFCPlatformInit(); #endif // build the rest of shell_state sh->shell_mode = SHELL_NORMAL; sh->command_array = normal_shell; // get current working directory char *buf = (char *)malloc(sizeof(char)*1024); if( !buf || !(getcwd(buf, 1024)) ) { D("either cwd too long, or out of memory."); if(buf) free(buf); sh->local_path = "/"; } else { sh->local_path = buf; if(buf) free(buf); D("set lpwd: "<< sh->local_path ); } sh->remote_path = "/"; sh->prompt_string = "(iPHUC) "; //enter shell ifVerbose cout << "notification: Entering shell in Normal Mode." << endl; shell_return_value = shell(sh); delete sh; } ifVerbose cout << "notification: Shell returned " << shell_return_value << endl; } else if ( msg == ADNCI_MSG_DISCONNECTED ) { ifNotQuiet cout << endl << "notification: Disconnected. Waiting for suitable device reconnect." << endl; shell_return_value = SHELL_WAIT; } switch (shell_return_value) { case SHELL_TERMINATE: ifNotQuiet cout << ">> Nothing left to do. Exiting." << endl; #if !defined(__APPLE__) run = false; #else exit(0); #endif break; case SHELL_WAIT: ifNotQuiet cout << ">> Waiting for device reconnect." << endl; break; default: ifNotQuiet cout << ">> Shell could not recover. Exiting." << endl; #if !defined(__APPLE__) run = false; #else exit(0); #endif } } int main(int argc, char **argv) { struct am_device_notification *notif; int c; short int cli_flags = 0; mach_error_t retval; while ((c = getopt (argc, argv, "qvs:o:a:drne")) != -1 ) { switch (c) { case 'q': cli_flags = cli_flags | OPT_QUIET; D("Quiet flag set"); break; case 'v': cli_flags = cli_flags | OPT_VERBOSE; D("Verbose flag set"); break; case 's': cli_flags = cli_flags | OPT_SCRIPT; D("Script flag set"); setscriptpath( optarg ); break; case 'o': cli_flags = cli_flags | OPT_ONESHOT; D("Oneshot flag set"); if ( !(cli_flags & OPT_SCRIPT) ) setscriptpath( optarg ); else ifNotQuiet cout << "iphuc: Oneshot flag incompatible with script flag." << endl; break; case 'a': cli_flags = cli_flags | OPT_AFCNAME; D("Afcname flag set: " << optarg); cli_afc_name = CFStringCreateWithCString(NULL, optarg, kCFStringEncodingASCII); break; case 'd': cli_flags = cli_flags | OPT_DEBUG; D("Debug flag set."); break; case 'r': cli_flags = cli_flags | OPT_RECOVERY; D("WaitForRecovery flag set."); break; case 'n': cli_flags = cli_flags | OPT_NORMAL; D("WaitForNormal flag set."); break; case 'e': cli_flags = cli_flags | OPT_RESTORE; D("WaitForRestore flag set."); break; case '?': cout << "getopt: unknown option." << endl; exit(1); default: cout << "getopt: default." << endl; abort(); } } setcliflags( cli_flags ); // default afc "com.apple.afc" if( !cli_afc_name ) { cli_afc_name = AMSVC_AFC; D("Set default afc name."); } ifNotQuiet cout << PACKAGE_STRING; #ifdef HAVE_READLINE_COMPLETION ifNotQuiet cout << " with tab completion."<< endl; #else ifNotQuiet cout << endl; #endif ifNotQuiet cout << ">> By The iPhoneDev Team: " << AUTHOR_NICK_STRING << endl; D("debug mode on."); //Call to SERIOUS_HACKERY ifVerbose cout << "initPrivateFunctions: "; initPrivateFunctions(); ifVerbose cout << endl; //End SERIOUS_HACKERY if( (cli_flags & OPT_RECOVERY) ) { D("skipping AMDeviceNotificationSubscribe"); ifVerbose cout << "iphuc: Waiting for recovery mode callback." << endl; } else { retval = AMDeviceNotificationSubscribe(notification, 0, 0, 0, &notif); ifVerbose cout << "AMDeviceNotificationSubscribe: " << retval << endl; } if( (cli_flags & OPT_NORMAL) || (cli_flags & OPT_RESTORE) ) { D("skipping AMRestoreRegisterForDeviceNotifications"); ifVerbose cout << "iphuc: Waiting for Normal or Restore mode." << endl; } else { unsigned int ret = AMRestoreRegisterForDeviceNotifications( dfu_connect_callback, recovery_connect_callback, dfu_disconnect_callback, recovery_disconnect_callback, 0, NULL); ifVerbose cout << "AMRestoreRegisterForDeviceNotifications: " << ret << endl; if (ret != 0) { ifNotQuiet cout << "Problem registering notification callback. Exiting." << endl; return EXIT_FAILURE; } } ifNotQuiet cout << "CFRunLoop: Waiting for iPhone." << endl; #if defined(__APPLE__) CFRunLoopRun(); ifNotQuiet cout << "main: CFRunLoop returned somehow, exiting. Please report." << endl; return 1; #else while (run) { Sleep(1); } return 1; #endif return 1; } /* -*- mode:c; indent-tabs-mode:nil; c-basic-offset:2; tab-width:2; */

This paste will be private.