initial commit

extra/pre-build.d/usr/lib/initcpio/hooks/archiso-custom

@@ -0,0 +1,186 @@
+# args: source, newroot, mountpoint
+_mnt_fs() {
+    local img="${1}"
+    local newroot="${2}"
+    local mnt="${3}"
+    local img_fullname="${img##*/}";
+    local img_name="${img_fullname%%.*}"
+    local dm_snap_name="${dm_snap_prefix}_${img_name}"
+    local ro_dev ro_dev_size rw_dev
+
+    ro_dev=$(losetup --find --show --read-only "${img}")
+    echo ${ro_dev} >> /run/archiso/used_block_devices
+    ro_dev_size=$(blockdev --getsz ${ro_dev})
+    if [[ "${cowfile_size}" == "100" ]]; then
+        rw_dev_size=${ro_dev_size}
+    else
+        # size calculation done in this way to avoid integer overflow when ro_dev_size is > 10.2G
+        rw_dev_size=$((ro_dev_size/100*cowfile_size))
+    fi
+
+    if [[ "${cow_persistent}" == "P" ]]; then
+        if [[ -f "/run/archiso/cowspace/${cow_directory}/${img_name}.cow" ]]; then
+            msg ":: Found '/run/archiso/cowspace/${cow_directory}/${img_name}.cow', using as persistent."
+        else
+            msg ":: Creating '/run/archiso/cowspace/${cow_directory}/${img_name}.cow' as persistent."
+            dd of="/run/archiso/cowspace/${cow_directory}/${img_name}.cow" count=0 seek=${rw_dev_size} &> /dev/null
+        fi
+    else
+        if [[ -f "/run/archiso/cowspace/${cow_directory}/${img_name}.cow" ]]; then
+            msg ":: Found '/run/archiso/cowspace/${cow_directory}/${img_name}.cow' but non-persistent requested, removing."
+            rm -f "/run/archiso/cowspace/${cow_directory}/${img_name}.cow"
+        fi
+        msg ":: Creating '/run/archiso/cowspace/${cow_directory}/${img_name}.cow' as non-persistent."
+        dd of="/run/archiso/cowspace/${cow_directory}/${img_name}.cow" count=0 seek=${rw_dev_size} &> /dev/null
+    fi
+
+    rw_dev=$(losetup --find --show "/run/archiso/cowspace/${cow_directory}/${img_name}.cow")
+    echo ${rw_dev} >> /run/archiso/used_block_devices
+
+    echo "0 ${ro_dev_size} snapshot ${ro_dev} ${rw_dev} ${cow_persistent} 8" | dmsetup create ${dm_snap_name}
+
+    _mnt_dev "/dev/mapper/${dm_snap_name}" "${newroot}${mnt}" "-w"
+    echo $(readlink -f /dev/mapper/${dm_snap_name}) >> /run/archiso/used_block_devices
+}
+
+# args: /path/to/image_file, mountpoint
+_mnt_sfs() {
+    local img="${1}"
+    local mnt="${2}"
+    local img_fullname="${img##*/}"
+    local sfs_dev
+
+    if [[ "${copytoram}" == "y" ]]; then
+        msg -n ":: Copying squashfs image to RAM..."
+        if ! cp "${img}" "/run/archiso/copytoram/${img_fullname}" ; then
+            echo "ERROR: while copy '${img}' to '/run/archiso/copytoram/${img_fullname}'"
+            launch_interactive_shell
+        fi
+        img="/run/archiso/copytoram/${img_fullname}"
+        msg "done."
+    fi
+    sfs_dev=$(losetup --find --show --read-only "${img}")
+    echo ${sfs_dev} >> /run/archiso/used_block_devices
+    _mnt_dev "${sfs_dev}" "${mnt}" "-r"
+}
+
+# args: device, mountpoint, flags
+_mnt_dev() {
+    local dev="${1}"
+    local mnt="${2}"
+    local flg="${3}"
+
+    mkdir -p "${mnt}"
+
+    msg ":: Mounting '${dev}' to '${mnt}'"
+
+    while ! poll_device "${dev}" 30; do
+        echo "ERROR: '${dev}' device did not 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
+
+    if mount "${flg}" "${dev}" "${mnt}"; then
+        msg ":: Device '${dev}' mounted successfully."
+    else
+        echo "ERROR; Failed to mount '${dev}'"
+        echo "   Falling back to interactive prompt"
+        echo "   You can try to fix the problem manually, log out when you are finished"
+        launch_interactive_shell
+    fi
+}
+
+_verify_checksum() {
+    local _status
+    cd "/run/archiso/bootmnt/${archisobasedir}/${arch}"
+    md5sum -c airootfs.md5 > /tmp/checksum.log 2>&1
+    _status=$?
+    cd "${OLDPWD}"
+    return ${_status}
+}
+
+run_hook() {
+    [[ -z "${arch}" ]] && arch="$(uname -m)"
+    [[ -z "${cowspace_size}" ]] && cowspace_size="75%"
+    [[ -z "${copytoram_size}" ]] && copytoram_size="75%"
+    [[ -z "${archisobasedir}" ]] && archisobasedir="arch"
+    [[ -z "${dm_snap_prefix}" ]] && dm_snap_prefix="arch"
+    [[ -z "${archisodevice}" ]] && archisodevice="/dev/disk/by-label/${archisolabel}"
+    if [[ -z "${cowfile_size}" ]]; then
+        cowfile_size="100"
+    else
+        cowfile_size=${cowfile_size/%}
+    fi
+
+    if [[ -n "${cow_label}" ]]; then
+        cow_device="/dev/disk/by-label/${cow_label}"
+        [[ -z "${cow_persistent}" ]] && cow_persistent="P"
+    elif [[ -n "${cow_device}" ]]; then
+        [[ -z "${cow_persistent}" ]] && cow_persistent="P"
+    else
+        cow_persistent="N"
+    fi
+
+    [[ -z "${cow_directory}" ]] && cow_directory="persistent_${archisolabel}/${arch}"
+
+    # set mount handler for archiso
+    mount_handler="archiso_mount_handler"
+}
+
+# This function is called normally from init script, but it can be called
+# as chain from other mount handlers.
+# args: /path/to/newroot
+archiso_mount_handler() {
+    local newroot="${1}"
+
+    if ! mountpoint -q "/run/archiso/bootmnt"; then
+        _mnt_dev "${archisodevice}" "/run/archiso/bootmnt" "-r"
+        if [[ "${copytoram}" != "y" ]]; then
+            echo $(readlink -f ${archisodevice}) >> /run/archiso/used_block_devices
+        fi
+    fi
+
+    if [[ "${checksum}" == "y" ]]; then
+        if [[ -f "/run/archiso/bootmnt/${archisobasedir}/${arch}/airootfs.md5" ]]; then
+            msg -n ":: Self-test requested, please wait..."
+            if _verify_checksum; then
+                msg "done. Checksum is OK, continue booting."
+            else
+                echo "ERROR: one or more files are corrupted"
+                echo "see /tmp/checksum.log for details"
+                launch_interactive_shell
+            fi
+        else
+            echo "ERROR: checksum=y option specified but ${archisobasedir}/${arch}/airootfs.md5 not found"
+            launch_interactive_shell
+        fi
+    fi
+
+    if [[ "${copytoram}" == "y" ]]; then
+        msg ":: Mounting /run/archiso/copytoram (tmpfs) filesystem, size=${copytoram_size}"
+        mkdir -p /run/archiso/copytoram
+        mount -t tmpfs -o "size=${copytoram_size}",mode=0755 copytoram /run/archiso/copytoram
+    fi
+
+    if [[ -n "${cow_device}" ]]; then
+        _mnt_dev "${cow_device}" "/run/archiso/cowspace" "-r"
+        echo $(readlink -f ${cow_device}) >> /run/archiso/used_block_devices
+        mount -o remount,rw "/run/archiso/cowspace"
+    else
+        msg ":: Mounting /run/archiso/cowspace (tmpfs) filesystem, size=${cowspace_size}..."
+        mkdir -p /run/archiso/cowspace
+        mount -t tmpfs -o "size=${cowspace_size}",mode=0755 cowspace /run/archiso/cowspace
+    fi
+    mkdir -p "/run/archiso/cowspace/${cow_directory}"
+
+    _mnt_sfs "/run/archiso/bootmnt/${archisobasedir}/${arch}/airootfs.sfs" "/run/archiso/sfs/airootfs"
+   # _mnt_fs "/run/archiso/sfs/airootfs/airootfs.img" "${newroot}" "/"
+    mount --bind "run/archiso/sfs/airootfs" "/new_root"
+
+    if [[ "${copytoram}" == "y" ]]; then
+        umount /run/archiso/bootmnt
+    fi
+}
+
+# vim:ft=sh:ts=4:sw=4:et:

extra/pre-build.d/usr/lib/initcpio/hooks/livecd

@@ -0,0 +1,15 @@
+# vim: set ft=sh:
+
+run_cleanuphook () {
+        msg ":: Mounting aufs on / with tmpfs=rw, ${root}=ro ..."
+        modprobe aufs
+
+	echo "Now attempting aufs mount..."
+
+	mkdir /new_root.hw
+	mount --move /new_root /new_root.hw
+	mkdir /dev/shm
+	mount -t tmpfs none /dev/shm
+	mount -t aufs none /new_root -o dirs=/dev/shm=rw:/new_root.hw=ro -o noatime
+	pkill -9 dropbear #kill SSH (in preparation for the live system starting ssh on port 22
+}

extra/pre-build.d/usr/lib/initcpio/hooks/ssh

@@ -0,0 +1,67 @@
+#!/usr/bin/ash
+
+run_hook ()
+{
+  local line i address netmask gateway dns0 dns1 rootserver rootpath defaultrootpath defaultserver
+
+  : > /ip_opts
+
+  if [ -z "${ip}" -a -n "${nfsaddrs}" ]; then
+    ip="${nfsaddrs}"
+  fi
+
+  if [ -n "${ip}" ]; then
+    # setup network and save some values
+    ipconfig "ip=${ip}" | while read line; do
+      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
+
+    echo "device=$(echo ${ip} | cut -d: -f6)" >> /ip_opts
+  fi
+
+  . /ip_opts
+
+  echo "IP-Config: ${address}/${netmask}"
+  echo "IP-Config: gw: ${gateway}    dns0: ${dns0}    dns1: ${dns1}"
+
+  [ -d /dev/pts ] || mkdir -p /dev/pts
+  mount -t devpts devpts /dev/pts
+
+  echo "Starting dropbear"
+  /usr/sbin/dropbear -E
+}
+
+run_cleanuphook ()
+{
+  pkill -9 dropbear
+}

extra/pre-build.d/usr/lib/initcpio/install/archiso-custom

@@ -0,0 +1,23 @@
+#!/bin/bash
+
+build() {
+    add_module "cdrom"
+    add_module "loop"
+    add_module "dm-snapshot"
+
+    add_runscript
+
+    add_binary /usr/lib/udev/cdrom_id
+    add_binary blockdev
+    add_binary dmsetup
+    add_binary losetup
+    add_binary mountpoint
+    add_binary truncate
+
+    add_file /usr/lib/udev/rules.d/60-cdrom_id.rules
+    add_file /usr/lib/udev/rules.d/10-dm.rules
+    add_file /usr/lib/udev/rules.d/95-dm-notify.rules
+    add_file /usr/lib/initcpio/udev/11-dm-initramfs.rules /usr/lib/udev/rules.d/11-dm-initramfs.rules
+}
+
+# vim: set ft=sh ts=4 sw=4 et:

extra/pre-build.d/usr/lib/initcpio/install/livecd

@@ -0,0 +1,21 @@
+#!/bin/bash
+ 
+build() 
+{
+	add_module 'squashfs'
+	add_module 'loop'
+	add_module 'aufs'
+
+        add_binary "sed"
+        add_binary "pkill"
+
+        add_runscript
+
+}
+ 
+help() 
+{
+        cat <<HELPEOF
+Mount a squashed flat-file directory with AUFS3 on /
+HELPEOF
+}

extra/pre-build.d/usr/lib/initcpio/install/ssh

@@ -0,0 +1,110 @@
+#!/bin/bash
+
+get_fingerprint() {
+  local keyfile="$1"
+  dropbearkey -y -f "${keyfile}" | sed -n '/^Fingerprint:/ {s/Fingerprint: *//; p}'
+}
+
+display_fingerprints() {
+  local keyfile
+
+  for keyfile in "/etc/dropbear/dropbear_dss_host_key" "/etc/dropbear/dropbear_rsa_host_key" ; do
+    if [ ! -r "${keyfile}" ] ; then
+      return 1
+    fi
+    echo "$(basename "${keyfile}") : $(get_fingerprint "${keyfile}")"
+  done
+}
+
+copy_openssh_keys() {
+  local osshrsa="/etc/ssh/ssh_host_rsa_key"
+  local osshdsa="/etc/ssh/ssh_host_dsa_key"
+
+  local dbpre="/etc/dropbear/dropbear_"
+
+  [ -f "$osshrsa" ] && [ -f "$osshdsa" ] || return 1
+
+  dropbearconvert openssh dropbear $osshrsa ${dbpre}rsa_host_key
+  dropbearconvert openssh dropbear $osshdsa ${dbpre}dss_host_key
+}
+
+generate_keys() {
+  local keyfile keytype
+  for keytype in dss rsa ; do
+    keyfile="/etc/dropbear/dropbear_${keytype}_host_key"
+    echo "Generating ${keytype} host key for dropbear ..."
+    dropbearkey -t "${keytype}" -f "${keyfile}"
+  done
+}
+
+make_etc_passwd() {
+  echo 'root:x:0:0:root:/:/bin/ash' > "${TMPDIR}"/passwd
+}
+
+build ()
+{
+  #
+  # Begin real processing
+  #
+
+  # Are we even needed?
+  if [ ! -e "/etc/dropbear/root_key" ]; then
+    echo "There is no root key in /etc/dropbear/root_key existent; exit"
+    return 0
+  fi
+
+  # if TMPDIR is set leave it alone otherwise set
+  [ -z $TMPDIR ] && TMPDIR='/tmp/dropbear_initrd_encrypt'
+
+  # check if TMPDIR exsists if not make it
+  [ -d $TMPDIR ] || mkdir -p $TMPDIR
+
+  umask 0022
+
+  [ -d /etc/dropbear ] && mkdir -p /etc/dropbear
+
+  display_fingerprints || copy_openssh_keys || generate_keys
+
+  [ -e "${TMPDIR}/passwd" ] && ( grep -q -e '^root:' "${TMPDIR}/passwd" ) || make_etc_passwd
+
+  add_checked_modules "/drivers/net/"
+  add_binary "rm"
+  add_binary "dropbear"
+  add_binary "killall"
+
+  echo '/bin/ash' > "${TMPDIR}"/shells
+  add_file "${TMPDIR}/shells" "/etc/shells"
+
+  cat /etc/dropbear/root_key > "${TMPDIR}"/authorized_keys
+
+  add_dir "/.ssh"
+  add_file "${TMPDIR}/authorized_keys" "/.ssh/authorized_keys"
+  add_file "${TMPDIR}/passwd" "/etc/passwd"
+  add_dir "/etc/dropbear"
+  add_file "/etc/dropbear/dropbear_rsa_host_key"
+  add_file "/etc/dropbear/dropbear_dss_host_key"
+  add_file "/lib/libnss_files.so.2"
+  add_binary "ip" "/sbin/ip"
+  add_dir "/var/run"
+
+  touch "${TMPDIR}"/lastlog
+  add_dir "/var/log"
+  add_file "${TMPDIR}/lastlog" "/var/log/lastlog"
+  add_binary "/usr/lib/initcpio/ipconfig" "/bin/ipconfig"
+
+  # cleanup
+  rm "${TMPDIR}/shells"
+  rm "${TMPDIR}/authorized_keys"
+  rm "${TMPDIR}/passwd"
+  rm "${TMPDIR}/lastlog"
+
+  add_runscript
+  
+}
+
+help ()
+{
+    cat<<HELPEOF
+Allow SSH access to an initial environment for debugging.
+HELPEOF
+}