manager.cpp

#include <stdio.h>
#include "MobileDevice.h"
#include "common.h"

#include "manager.h"

Manager* Manager::_inst = 0;

Manager::Manager()
	:_devices(NULL)
	 , _device_id(NULL), _cmd(CMD_UNKNOWN), _state(S_UNKNOWN)
{
}

Manager * Manager::getInstance()
{
	if (_inst == NULL) {
		_inst = new Manager();
	}
	return _inst;
}

int Manager::parse(int argc, char *argv[])
{
	int i = 1;

	_option = NULL;
	while (i < argc) {
		if (strcmp(argv[i], "-s") == 0) {
			_device_id = argv[i+1];
			i++;
		} else { 
			if (strcmp(argv[i], "devices") == 0) {
				_cmd = CMD_DEVICES;
			} else if (strcmp(argv[i], "install") == 0) {
				_cmd = CMD_INSTALL;
				_arg1 = argv[++i];
			} else if (strcmp(argv[i], "uninstall") == 0) {
				_cmd = CMD_UNINSTALL;
				_arg1 = argv[++i];
			} else if (strcmp(argv[i], "listapp") == 0) {
				_cmd = CMD_LISTAPP;
			} else if (strcmp(argv[i], "logcat") == 0) {
				_cmd = CMD_LOGCAT;
			} else if (strcmp(argv[i], "info") == 0) {
				_cmd = CMD_INFO;
			} else if (strcmp(argv[i], "ls") == 0) {
				_cmd = CMD_LISTDIR;
				if (strcmp(argv[i+1], "-b") == 0) {
					i += 2;
					_option = argv[i];
				}
				_arg1 = argv[++i];
			} else if (strcmp(argv[i], "push") == 0) {
				_cmd = CMD_PUSH;
				if (strcmp(argv[i+1], "-b") == 0) {
					i += 2;
					_option = argv[i];
				}
				_arg1 = argv[++i];
				_arg2 = argv[++i];
			} else if (strcmp(argv[i], "pull") == 0) {
				_cmd = CMD_PULL;
				if (strcmp(argv[i+1], "-b") == 0) {
					i += 2;
					_option = argv[i];
				}
				_arg1 = argv[++i];
				_arg2 = argv[++i];
			} else if (strcmp(argv[i], "rm") == 0) {
				_cmd = CMD_REMOVE;
				if (strcmp(argv[i+1], "-b") == 0) {
					i += 2;
					_option = argv[i];
				}
				_arg1 = argv[++i];
			} else {
				return -1;
			}
			break;
		}
		i++;
	}
	//printf("i:%d, argc:%d\n", i, argc);
	if (argc <= 1 || i >= argc) {
		return -1;
	}
	if (_cmd == CMD_UNKNOWN) {
		return -1;
	}
	/*
	if (_device_id == NULL && _cmd != CMD_DEVICES) {
		return -1;
	}
	*/
	return 0;
}

void Manager::run()
{
    AMDSetLogLevel(5); // otherwise syslog gets flooded with crap

    _devices = CFDictionaryCreateMutable(kCFAllocatorDefault, 0, NULL, NULL);

    int timeout = 1;
	CFRunLoopTimerContext context;
	context.version = 0;
	context.info = this;
	context.retain = NULL;
	context.release = NULL;
	context.copyDescription = NULL;

	CFRunLoopTimerRef timer = CFRunLoopTimerCreate(NULL, CFAbsoluteTimeGetCurrent() + timeout, 0, 0, 0, timeoutCallback, &context);
	CFRunLoopAddTimer(CFRunLoopGetCurrent(), timer, kCFRunLoopCommonModes);
	//printf("[....] Waiting up to %d seconds for iOS device to be connected\n", timeout);

    struct am_device_notification *notify;
    AMDeviceNotificationSubscribe(&deviceCallback, 0, 0, this, &notify);
    CFRunLoopRun();
}

void Manager::connectDevice(struct am_device *amdevice) 
{
	CFStringRef device_id = AMDeviceCopyDeviceIdentifier(amdevice);
	const char *str_device_id = CFStringGetCStringPtr(device_id, CFStringGetSystemEncoding());

	if (_device_id != NULL) {
		if (strcmp(_device_id, str_device_id) == 0) {
			_state = S_RUN_ONCE;
		} else {
			CFRelease(device_id);
			return;
		}
	}

	_state = S_RUN_ONCE;
	Device *device = getDevice(amdevice, true);

	int ret = 0;
	//dispatch
	switch (_cmd) {
	case CMD_DEVICES: {
		if (_device_id == NULL) {
			_state = S_RUN_LOOP;
		}
		printf("[device] %s\n", str_device_id);
		break;
	}
	case CMD_INFO: {
		device->showInfo();
		break;
	}
	case CMD_INSTALL: {
		const char *app_path = _arg1;
		ret = device->install(app_path);
		if (ret == 0) {
			printf("Install success!\n");
		} else {
			printf("Install fail!\n");
		}
		exit(ret);
		break;
	}
	case CMD_UNINSTALL: {
		const char *bundle_id = _arg1;
		ret = device->uninstall(bundle_id);
		if (ret == 0) {
			printf("Uninstall success!\n");
		} else {
			printf("Uninstall fail!\n");
		}
		exit(ret);
		break;
	}
	case CMD_LISTAPP: {
		if (device->listApps() != 0) {
			printf("List applications fail!\n");
		}
		break;
	}
	case CMD_LOGCAT: {
		_state = S_RUN_LOOP;
		device->startLogcat();
		break;
	}
	case CMD_LISTDIR:
	case CMD_PUSH:
	case CMD_PULL:
	case CMD_REMOVE: 
	{
		ret = device->operateFile(_cmd, _option, _arg1, _arg2);
		if (ret != 0) {
			printf("Access file fail!\n");
		}
		exit(ret);
		break;
	}
	default: {}
		break;
	}

	CFRelease(device_id);
}

void Manager::disConnectDevice(struct am_device *amdevice) 
{
	Device *device = getDevice(amdevice);
	if (device != NULL) {
		device->setAlive(false);
		if (_cmd == CMD_LOGCAT) {
			device->stopLogcat();
		}
	}
}

Device * Manager::getDevice(struct am_device *amdevice, bool bcreate)
{
	Device *device = (Device *)CFDictionaryGetValue(_devices, amdevice);
	if (bcreate && device == NULL) {
		device = new Device(amdevice);
		device->setAlive(true);
		CFDictionarySetValue(_devices, amdevice, device);
	}
	return device;
}

bool Manager::isActive()
{
	return (_cmd != CMD_DEVICES && _state != S_UNKNOWN);
}

bool Manager::isRunLoop()
{
	//return (_state == S_RUN_LOOP);
	return (_state != S_RUN_ONCE);
}

void Manager::release()
{
	if (_devices) {
		CFRelease(_devices);
	}
}

void Manager::timeoutCallback(CFRunLoopTimerRef timer, void *arg) 
{
	Manager *manager = (Manager *)arg;

    if (!manager->isActive()) {
        //fprintf(stderr, "Timed out waiting for device.\n");
        exit(-1);
    }
}

void Manager::deviceCallback(struct am_device_notification_callback_info *info, void *arg) 
{
	Manager *manager = (Manager *)arg;

    switch (info->msg) {
        case ADNCI_MSG_CONNECTED: {
            manager->connectDevice(info->dev);
			break;
		}
        case ADNCI_MSG_DISCONNECTED: {
            manager->disConnectDevice(info->dev);
			break;
		}
        default:
            break;
    }

	if (!manager->isRunLoop()) {
		CFRunLoopStop(CFRunLoopGetCurrent());
	}
}