adb.c

#include <pthread.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <stdio.h>
#include <string.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <stdlib.h>
#include <errno.h>
#include <sys/mount.h>
#include <sys/klog.h>
#include <linux/loop.h>

#include "adb.h"
#include "util.h"
#include "multirom.h"
#include "log.h"

static pthread_t adb_thread;
static volatile int run_thread = 0;
static pid_t adb_pid = -1;

static char * const ENV[] = {
    "PATH=/sbin:/bin:/usr/bin:/usr/sbin:/mrom_bin",
    "LD_LIBRARY_PATH=.:/sbin",
    "ANDROID_ROOT=/system",
    "ANDROID_DATA=/data",
    "EXTERNAL_STORAGE=/sdcard",
    "ANDROID_PROPERTY_WORKSPACE=8,49152",

    NULL
};

static void *adb_thread_work(void *c)
{
    adb_init_usb();

    if(adb_init_busybox() < 0)
        return NULL;

    adb_init_fs();

    while(run_thread)
    {
        adb_pid = fork();
        if(adb_pid == 0) // child
        {
            setsid();
            umask(077);
            setpgid(0, getpid());

            static char * const cmd[] = { adbd_path, NULL };
            execve(cmd[0], cmd, ENV);
            exit(0);
        }
        else
        {
            int status = 0;
            while(waitpid(adb_pid, &status, WNOHANG) == 0)
                usleep(300000);
        }
        usleep(300000);
    }

    adb_cleanup();

    return NULL;
}

void adb_init(void)
{
    INFO("Starting adbd\n");
    if(run_thread)
        return;

    run_thread = 1;
    pthread_create(&adb_thread, NULL, adb_thread_work, NULL);
}

void adb_quit(void)
{
    if(!run_thread)
        return;

    INFO("Stopping adbd\n");

    run_thread = 0;

    if(adb_pid != -1)
    {
        kill(adb_pid, 9);
        adb_pid = -1;
    }

    pthread_join(adb_thread, NULL);
}

void adb_init_usb(void)
{
    write_file("/sys/class/android_usb/android0/enable", "0");

    char cmdline[1024];
    char serial[64] = { 0 };
    static const char *tag = "androidboot.serialno=";
    if(multirom_get_cmdline(cmdline, sizeof(cmdline)) >= 0)
    {
        char *start = strstr(cmdline, tag);
        if(start)
        {
            start += strlen(tag);
            char *end = strchr(start, ' ');
            if(end && end-start < (int)sizeof(serial))
                strncpy(serial, start, end-serial);
        }
    }

    write_file("/sys/class/android_usb/android0/idVendor", "18d1");
    write_file("/sys/class/android_usb/android0/idProduct", "4e42");
    write_file("/sys/class/android_usb/android0/functions", "adb");
    write_file("/sys/class/android_usb/android0/iManufacturer", "unknown");
    write_file("/sys/class/android_usb/android0/iProduct", "Nexus 7");
    write_file("/sys/class/android_usb/android0/iSerial", serial);

    write_file("/sys/class/android_usb/android0/enable", "1");
}

int adb_init_busybox(void)
{
    mkdir("/mrom_bin", 0777);

    copy_file(busybox_path, "/mrom_bin/busybox");
    chmod("/mrom_bin/busybox", 0755);

    static const char *install_cmd[] = {
        "/mrom_bin/busybox", "--install", "/mrom_bin/", NULL
    };

    if(run_cmd((char**)install_cmd) != 0)
    {
        ERROR("adb: failed to --install busybox\n");
        return -1;
    }

    mkdir("/dev/pts", 0666);
    if(mount("devpts", "/dev/pts", "devpts", 0, NULL) < 0)
    {
        ERROR("Failed to mount devpts: %d (%s)\n", errno, strerror(errno));
        return -1;
    }

    return 0;
}

void adb_init_fs(void)
{
    mkdir("/sdcard", 0777);
    if(strstr(adbd_path, "/realdata/media/0/multirom"))
        mount("/realdata/media/0/", "/sdcard/", "auto", MS_BIND, "");
    else
        mount("/realdata/media/", "/sdcard/", "auto", MS_BIND, "");
}

void adb_cleanup(void)
{
    if(umount("/sdcard") >= 0)
        remove_dir("/sdcard");

    remove_dir("/mrom_bin");

    umount("/dev/pts");
    rmdir("/dev/pts");
}