#!/usr/bin/python2
import os
import subprocess

def read_elf_class(elf_file):
    import struct
    f = open(elf_file, "rb")
    f.seek(4)
    elf_class = f.read(1)
    f.close()
    elf_class = struct.unpack('B', elf_class)[0]
    return elf_class

def exec_chroot_cmd(args, chroot, pre_chroot = [], mount_vfs = True):
    pid = os.fork()
    if pid == 0:
        os.chroot(chroot)
        os.chdir("/")
        myargs = pre_chroot+args
        rc = subprocess.call(myargs)
        if mount_vfs:
            subprocess.call(("mount", "-t", "proc", "proc", "/proc"))
            subprocess.call(("mount", "-t", "devpts", "devpts", "/dev/pts"))
        os._exit(rc)
    else:
        rc_pid, rc = os.waitpid(pid, 0)
        return rc_pid, rc

def spawn_ssh(args, chroot):

    # bind /proc and /dev
    for path in []:
        b_args = ["mount", "--bind", path, os.path.join(chroot, path[1:])]
        subprocess.call(b_args)

    bash_exec = os.path.join(chroot, "bin/bash") # must exists
    elf_class = read_elf_class(bash_exec)
    pre_chroot = []
    if elf_class != 2 and (os.path.basename(chroot).find("arm") == -1): # 32 bit
        pre_chroot = ["linux32"]
    rc_pid, rc = exec_chroot_cmd(args, chroot, pre_chroot = pre_chroot)
    return rc_pid, rc


if __name__ == "__main__":

    start_port = 2000
    chroot_dir = "/sabayon/sources"
    chroots = os.listdir(chroot_dir)
    default_args = ["/usr/sbin/sshd"]
    chroot_map_file = "/sabayon/CHROOTS"

    ch_f = open(chroot_map_file, "w")
    ch_f.write("Sabayon Chroots available via SSH:\n")
    for chroot in chroots:
        if not os.path.isdir(os.path.join(chroot_dir, chroot)):
            continue
        print("starting chroot %s, ssh at localhost:%s" % (chroot, start_port,))
        start_port += 1
        args = default_args + ["-o", "ListenAddress=127.0.0.1:%s" % (start_port,)]
        pid, exit_st = spawn_ssh(args, os.path.join(chroot_dir, chroot))
        if exit_st == 0:
            print("started sshd in chroot: %s" % (chroot,))
        else:
            print("NOT started sshd in chroot: %s, perhaps already started?" % (chroot,))
        ch_f.write("\t%s at: root@localhost -p %s\n" % (chroot, start_port,))

    ch_f.flush()
    ch_f.close()