archiso/hooks/archiso_pxe_nbd


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
94
95
96
97
98
99
100
101

# vim: set ft=sh:
run_hook () {
    local line i net_mac bootif_mac bootif_dev address netmask gateway dns0 dns1 rootserver rootpath filename

    : > /ip_opts

    if [ -n "${ip}" ]; then
        if [ -n "${BOOTIF}" ]; then
            bootif_mac=${BOOTIF#01-}
            bootif_mac=${bootif_mac//-/:}
            for i in /sys/class/net/*/address; do
                read net_mac < ${i}
                if [ "${bootif_mac}" == "${net_mac}" ]; then
                    bootif_dev=${i#/sys/class/net/}
                    bootif_dev=${bootif_dev%/address}
                    break
                fi
            done
            ip="${ip}::${bootif_dev}"
        fi

        # setup network and save some values
        ipconfig "ip=${ip}" | while read line; do
            # echo ":: ${line}"
            if [ "${line#"IP-Config:"}" != "${line}" ]; then
                continue
            fi
            line="$(echo ${line} | sed -e 's/ :/:/g;s/: /=/g')"
            for i in ${line}; do
                case "${i}" in
                    address=*)
                        echo "${i}" >> /ip_opts
                        ;;
                    netmask=*)
                        echo "${i}" >> /ip_opts
                        ;;
                    gateway=*)
                        echo "${i}" >> /ip_opts
                        ;;
                    dns0=*)
                        echo "${i}" >> /ip_opts
                        ;;
                    dns1=*)
                        echo "${i}" >> /ip_opts
                        ;;
                    rootserver=*)
                        echo "${i}" >> /ip_opts
                        ;;
                    rootpath=*)
                        echo "${i}" >> /ip_opts
                        ;;
                esac
            done
        done

        . /ip_opts

        echo "IP-Config: ${address}/${netmask}"
        echo "IP-Config: gw: ${gateway}    dns0: ${dns0}    dns1: ${dns1}"

        nbdserver=${rootserver}

        [[ -z "${archiso_nbd_name}" ]] && archiso_nbd_name="archiso"

        mount_handler="archiso_pxe_nbd_mount_handler"
    fi
}

archiso_pxe_nbd_mount_handler () {
    newroot="${1}"

    # Module autoloading like with loop devices does not work, doing manually...
    modprobe nbd 2> /dev/null
    msg ":: Waiting for boot device..."
    while ! poll_device /dev/nbd0 30; do
        echo "ERROR: boot device didn't show up after 30 seconds..."
        echo "   Falling back to interactive prompt"
        echo "   You can try to fix the problem manually, log out when you are finished"
        launch_interactive_shell
    done

    msg "::: Setup NBD from ${nbdserver} at /dev/nbd0"
    if [ "${copytoram}" = "y" ]; then
        nbd-client ${nbdserver} -N ${archiso_nbd_name} /dev/nbd0
    else
        nbd-client ${nbdserver} -N ${archiso_nbd_name} /dev/nbd0 -persist
    fi

    archisodevice=/dev/nbd0

    archiso_mount_handler ${newroot}

    if [ "${copytoram}" = "y" ]; then
        msg "::: Disconnect NBD from ${nbdserver} at /dev/nbd0"
        nbd-client -d /dev/nbd0
    else
        mkdir -p /run/archiso
        pidof nbd-client > /run/archiso/nbd_client.pid
        cp /archiso_pxe_nbd ${newroot}/etc/rc.d/functions.d/
    fi
}