|
|
#include "Shell.h"
using namespace std;
#if !defined(__APPLE__)
bool run = true;
#endif
#if defined(WIN32)
#define WIN32_LEAN_AND_MEAN
#include
#endif
CFStringRef cli_afc_name;
COMMAND normal_shell[] =
{
{ "help", sh_help, ">> help - Display help information on . No args lists commands." },
{ "cd", n_cd, ">> cd - Change directory to " },
{ "lcd", n_lcd, ">> lcd - Change local directory to " },
{ "ls", n_ls, ">> ls [path] - List directory at current working path or [path]" },
{ "mkdir", n_mkdir, ">> mkdir - Create directory at " },
{ "rmdir", n_rmdir, ">> rmdir - Remove directory at " },
{ "activate", n_activate, ">> activate - Activate iPhone with .plist file at " },
{ "deactivate", n_deactivate, ">> deactivate - Deactivate iPhone" },
{ "readvalue", n_readvalue, ">> readvalue - Read . 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 - Starts service on the iPhone. No args lists services." },
{ "deviceinfo", n_deviceinfo, ">> deviceinfo - Display device info." },
{ "getfilesize", n_getfilesize, ">> getfilesize - Display size of file at " },
{ "getfile", n_getfile, ">> getfile [localpath] - Get file on iPhone at and write it to [localpath]" },
{ "putfile", n_putfile, ">> putfile [path] - Put file at on iPhone at [path]" },
{ "fileinfo", n_fileinfo, ">> fileinfo - Display info for file at " },
{ "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 - Set the name of the afc service to use."},
{ "run", sh_run, ">> run - runs a script at ."},
{ (char *)NULL, (shell_funct *)NULL, (char *)NULL }
};
COMMAND restore_shell[] =
{
{ "mount", restore_mount, ">> mount - Mount device at path."},
{ "partition", restore_partition, ">> partition - Partition device."},
{ "erase", restore_erase, ">> erase - Erase device."},
{ "ditto", restore_ditto, ">> ditto - Copy file at path1 to path2."},
{ "umount", restore_umount, ">> umount - Unmount device from path."},
{ "foo", erica_foo, ">> erica foo: send message."},
{ "FileSystemCheck", restore_filesystemcheck,">> filesystemcheck - Check filesytem on device" },
{ "mkdir", restore_mkdir, ">> mkdir - Make directory 'path'."},
{ "force", restore_force, ">> force - send cmd to phone."},
{ "help", sh_help, ">> help - Display help information on . No args lists commands." },
{ "exit", restore_exit, ">> exit - Disconnect and wait for device normal reconnect." },
{ "run", sh_run, ">> run - runs a script at ."},
{ (char *)NULL, (shell_funct *)NULL, (char *)NULL }
};
COMMAND recovery_shell[] =
{
{ "restore", recovery_restore, ">> restore - enter restore mode. "},
{ "grestore", recovery_grestore, ">> grestore - enter restore mode interactively."},
{ "filecopytophone", recovery_filecopytophone, ">> filecopytophone"},
{ "serial", recovery_serial, ">> serial"},
{ "bar", erica_bar, ">> bar"},
{ "cmd", recovery_cmd, ">> cmd - 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 - Display help information on . No args lists commands." },
{ "run", sh_run, ">> run - runs a script at ."},
{ (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, ¬if);
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; */
|
Pastie
Support Pastie
or read my
Pastie << self
blog
