LSW/entrypoint.sh at master · MrCyjaneK/LSW · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
#!/bin/bash
if [[ "x$1" == "x..linux" ]];
then
    exec bash
    exit 255
fi

if [[ "x$1" == "x..disk-optimize" ]];
then
    echo "Optimizing disk storage. Resulting disk will be placed in /opt/disk-shrink.img"
    echo "This procedure should only be ran in Dockerfile."
    echo "To speedup the process you may want to run it with --device /dev/kvm"
    echo "or RUN --security=insecure"
    echo "This will reduce amount of time that is required for this operation by at least 10x"
    mkdir $HOME/.cache
    TMPDIR=$HOME/.cache virt-sparsify \
        --convert qcow2 \
        --machine-readable \
        --compress \
        /opt/disk.img \
        /opt/disk-shrink.img | grep -E '^[0-9]+[/\\-][0-9]+$' | tr '/' ' ' | awk '{ print "[DISK OPTIMIZE]: " $1 / $2 *100 "% done" }' || exit 1
    echo "Replacing source disk with optimized one"
    rm /opt/disk.img
    mv /opt/disk-shrink.img /opt/disk.img
    rm -rf $HOME/.cache
    exit 0
fi

if [[ ! -f "/cdrom.iso" ]];
then
    echo "[ISO] Creating cdrom.iso"
    genisoimage -o /cdrom.iso -V LSW -R -J /cdrom
fi

exit_script() {
    echo "[OUTPUTS] Moving C:\\outputs to /outputs"
    mkdir /outputs
    scp -r -P22 user@127.0.0.1:/outputs/'*' /outputs/
    sshpass ssh user@127.0.0.1 -p22 rd /s /q "C:\\outputs"
    sshpass ssh -o StrictHostKeyChecking=no user@127.0.0.1 -p22 shutdown -s -t 1 || (killall -9 qemu-system-x86_64)
    echo "[QEMU] Shutting down guest"
    until ! killall -0 qemu-system-x86_64;
    do
        sleep 1
    done
    echo "[QEMU] Guest offline"
    echo "[ISO] Cleaning up iso"
    rm -f /cdrom.iso
    trap - SIGINT SIGTERM # clear the trap
    exit $EXITCODE
    # kill -- -$$ # Sends SIGTERM to child/sub processes
}
trap exit_script SIGINT SIGTERM

QEMU_KVM_FLAGS=""
kvm-ok && QEMU_KVM_FLAGS="$QEMU_KVM_FLAGS -enable-kvm -cpu host" || QEMU_KVM_FLAGS=""

FORWARD_PORTS="22 $FORWARD_PORTS"


for i in $FORWARD_PORTS;
do
    EXTRA_FWD="$EXTRA_FWD,hostfwd=tcp::$i-:$i"
done


if ! killall -0 qemu-system-x86_64 &>/dev/null;
then
    echo "[QEMU] Booting guest"
    qemu-system-x86_64 \
        $QEMU_KVM_FLAGS \
        -drive file=/opt/disk.img,if=virtio \
        -drive file=/cdrom.iso,media=cdrom \
        -net user"$EXTRA_FWD" \
        -net nic,model=rtl8139 -m "$QEMU_RAM_LIMIT" \
        -daemonize -display none
    echo "[QEMU] Waiting for connection"
    # Timeout added just in case when ssh connection could hang..
    until timeout 30 sshpass ssh -o StrictHostKeyChecking=no user@127.0.0.1 -p22 echo ok &>/dev/null
    do
        sleep 1
    done
fi
if [[ "x$@" == "x" ]];
then
    sshpass ssh -o StrictHostKeyChecking=no user@127.0.0.1 -p22
    EXITCODE=$?
else
    sshpass ssh -o StrictHostKeyChecking=no user@127.0.0.1 -p22 powershell -command "$@"
    EXITCODE=$?
fi

exit_script