script/genuzip: don't cache mkuzip output
[CDN/Mosi.git] / script / gentree
1 #!/bin/sh
2
3 # Boilerplate
4 _root="$(dirname "${0}")"
5 . "${_root}/lib/env.sh"
6
7 # Load needed modules
8 want root ansi log ask
9
10 targets="prepwork admin packages patch overlay prepboot preptmp prepetc prepvar imgboot imgconf imgetc imgvar custom"
11
12 pebkac() {
13   [ "${*}" ] && printf "${*}\n\n"
14   echo "Usage: $(basename "${0}") <-b basedir> <-t stagedir> <-i pkgsdir> <-p patchdir>"
15   echo "        <-r rootdir> <-l logfile> [-h] <targets>"
16   echo '  -b basedir    Basedir for automagic defaults'
17   echo '  -t stagedir   Staging directory name (Default: ${base}/stage)'
18   echo '  -i pkgsdir    Directory holding packages to install (Default: ${base}/pkg)'
19   echo '  -p patchdir   Directory holding patches to apply (Default: ${base}/patch)'
20   echo '  -o overlaydir Directory holding an overlay tree (Default: ${base}/overlay)'
21   echo '  -r rootdir    Directory holding the virgin source tree (Default: ${root})'
22   echo '  -l logfile    File to hold stderr spam (Default: ${stage}/gentree.log'
23   echo '  -h            Hello! >^-^<'
24   echo ''
25   echo 'Available targets:'
26   for target in ${targets}
27   do
28     echo "  ${target}"
29   done
30   exit 1
31 }
32
33 while getopts "b:t:i:p:o:r:l:h" opt
34 do
35   case "${opt}" in
36     b) base="${OPTARG}" ;;
37     t) stage="${OPTARG}" ;;
38     i) pkgs="${OPTARG}" ;;
39     p) patch="${OPTARG}" ;;
40     o) overlay="${OPTARG}" ;;
41     r) root="${OPTARG}" ;;
42     l) logfile="${OPTARG}" ;;
43     h) pebkac ;;
44     [?]) pebkac "Unrecognized option ${opt}" ;;
45   esac
46 done
47 shift $(( $OPTIND - 1 ))
48
49 sequence="${*:-${targets}}"
50
51 base="${base:-/usr/home/cyberleo/world}"
52 base="$(realpath "${base}")"
53 stage="${stage:-${base}/tree}"
54 pkgs="${pkgs:-${base}/pkg}"
55 patch="${patch:-${base}/patch}"
56 overlay="${overlay:-${base}/overlay}"
57 root="${root:-${root}}"
58 logfile="${logfile:-${stage}/gentree.log}"
59
60 mkdir -p "${stage}"
61 stage="$(realpath "${stage}")"
62 sroot="${stage}/root"
63 sboot="${stage}/boot"
64 sconf="${stage}/conf"
65
66 exec 2>>"${logfile}"
67 _log_to_stderr=yes
68
69 # Helper functions
70 onelink() {
71   # Make sure the provided file(s) have only one link!
72   while [ -n "${1}" ]
73   do
74     if [ -f "${1}" -a "$(stat -f '%l' "${1}")" -gt 1 ]
75     then
76       cp -p "${1}" "${1}.tmp" && mv "${1}.tmp" "${1}" || err "breaklink failed"
77     fi
78     shift
79   done
80 }
81
82 # Build steps
83 do_prepwork() {
84   log Prepare workspace
85   if [ -d "${stage}" -a \( -d "${sroot}" -o -d "${sboot}" -o -d "${sconf}" \) ]
86   then
87     yn n " ${a_yellow}*${a_normal} Workspace already exists. Delete? [y/N]" || err Aborting
88     chk rm -Rf "${sroot}" "${sboot}" "${sconf}"
89   fi
90   chk mkdir -p "${sroot}"
91   # Eliminate schg, because it interferes with hardlinks
92   chk chflags -R noschg "${root}/lib"
93   chk chflags -R noschg "${root}/libexec"
94   chk chflags -R noschg "${root}/sbin"
95   chk chflags -R noschg "${root}/usr"
96   ( cd "${root}" && find . | cpio -p --link "${sroot}" ) || chk
97 }
98
99 do_admin() {
100   log Create an emergency user admin/admin
101   # delink passwd to ensure it doesn't get patched in-place
102   chk onelink "${sroot}/etc/passwd" "${sroot}/etc/master.passwd"
103   echo '$1$2rXOWsK/$eiBHA6K7xL96DZbcY24YR0' | chk chroot "${sroot}" /usr/sbin/pw useradd admin -u 999 -g wheel -G operator -c Administrator -d /usr/home/admin -m -s /bin/csh -H 0
104 }
105
106 do_packages() {
107   if [ -d "${pkgs}" -a "$(ls -1 "${pkgs}" | wc -l)" -gt 0 ]
108   then
109     count="$(ls -1 "${pkgs}" | wc -l)"
110     log Install ${count} packages from "${pkgs}"
111     chk mkdir -p "${sroot}/pkg"
112     ( cd "${pkgs}" && find . | cpio -p --link "${sroot}/pkg" ) || chk
113     chk chroot "${sroot}" /bin/sh -c 'cd /pkg; exec pkg_add -F *'
114     chk rm -Rf "${sroot}/pkg"
115   fi
116 }
117
118 do_patch() {
119   log Apply patches from "${patch}"
120   for file in "${patch}"/*
121   do
122     note "... $(basename "${file}")"
123     ( cd "${sroot}" && patch < "${file}" ) || chk
124     # Remove .orig files
125     sed -e '/^+++ /!d; s/^+++ //; s/[   ]*[0-9: .+-]*$//' "${file}" | while read target
126     do
127       rm -f "${sroot}/${target}.orig"
128     done || chk
129   done
130 }
131
132 do_overlay() {
133   log Apply overlay from "${overlay}"
134   [ -d "${overlay}" ] || return
135   ( cd "${overlay}" && find . | cpio -p "${sroot}" ) || chk
136 }
137
138 do_prepboot() {
139   log Prepare /boot
140   chk mv "${sroot}/boot/boot" "${sroot}/boot/boot.blk"
141   chk ln -sf . "${sroot}/boot/boot"
142 }
143
144 do_preptmp() {
145   log Prepare /tmp
146   ( cd "${sroot}/tmp" && find . | cpio -p --link ../var/tmp ) || chk
147   chk rm -Rf "${sroot}/tmp"
148   chk ln -sf var/tmp "${sroot}/tmp"
149 }
150
151 do_prepetc() {
152   log Prepare /etc
153   chk mkdir -p "${sroot}/etc/local"
154   chk mkdir -p "${sroot}/usr/local/etc" # Silence warnings
155   ( cd "${sroot}/usr/local/etc" && find . | cpio -p --link ../../../etc/local ) || chk
156   chk rm -Rf "${sroot}/usr/local/etc"
157   chk ln -sf ../../etc/local "${sroot}/usr/local/etc"
158 }
159
160 do_prepvar() {
161   log Prepare /var
162   # Nothing necessary here
163 }
164
165 do_imgboot() {
166   log Create boot imgsrc
167   chk mv "${sroot}/boot" "${stage}"
168   chk mkdir -p "${sroot}/boot"
169   chk rm -f "${sboot}/kernel"/*.symbols
170   # Gzipped kernel is okay
171   chk onelink "${sboot}/kernel/kernel"
172   chk gzip -9 "${sboot}/kernel/kernel"
173   # Compress all files in /boot/kernel
174   # Loader cannot handle gzipped modules. Decompress the required modules
175   # kldload cannot handle gzipped modules either
176   #find "${sboot}/kernel" -type f | xargs gzip -9f || chk
177   #cat "${sboot}/loader.conf" | grep '_load=' | sed -e 's/^\(.*\)_load=.*$/\1/' | while read mod
178   #do
179   #  [ -f "${sboot}/kernel/${mod}.ko.gz" ] && gunzip "${sboot}/kernel/${mod}.ko.gz" || chk
180   #done
181   #
182   # Instead: put all modules in the root image, except those needed to boot the kernel
183   chk mkdir -p "${sroot}/boot/boot"
184   chk mkdir -p "${sroot}/boot/kernel"
185
186   # Link all modules into the root fs
187   ( cd "${sboot}/kernel" && find . -name '*.ko' -o -name 'linker.hints' | cpio -p --link "${sroot}/boot/kernel" ) || chk
188
189   # Remove all modules from the root fs that are preloaded by the loader
190   cat "${sboot}/loader.conf" | grep '_load=' | sed -e 's#^\(.*\)_load=.*$#'"${sroot}/boot/kernel/"'\1.ko#' | xargs rm -f
191
192   # Remove all modules from the boot fs that are present in the root fs
193   ( cd "${sroot}/boot/kernel" && ls -1 ) | sed -e 's#^#'"${sboot}/kernel/"'#' | xargs rm -f
194
195   # Link the preloaded modules from the boot fs to the root fs, to provide a homogenous view
196   ( cd "${sboot}/kernel" && ls -1 ) | while read mod
197   do
198     ln -sf "../boot/kernel/${mod}" "${sroot}/boot/kernel/${mod}"
199   done
200 }
201
202 do_imgconf() {
203   log Create conf imgsrc
204   chk mkdir -p "${sroot}/conf"
205   echo "ufs:/dev/ufs/conf" > "${sroot}/conf/diskless_remount" || chk
206   chk mkdir -p "${sconf}/backup"
207   chk mkdir -p "${sconf}/base/etc"
208   chk mkdir -p "${sconf}/base/var"
209   chk mkdir -p "${sconf}/default/etc"
210   chk mkdir -p "${sconf}/default/var"
211 }
212
213 do_imgetc() {
214   log Create etc imgsrc
215   chk touch "${sroot}/etc/diskless"
216   chk sleep 1 # Make sure diskless is at least one second older than everything in custom
217   chk mv "${sroot}/etc" "${stage}"
218   chk mkdir -p "${sroot}/etc"
219   chk cp -p "${stage}/etc/rc" "${stage}/etc/rc.subr" "${stage}/etc/rc.initdiskless" "${stage}/etc/login.conf.db" "${stage}/etc/diskless" "${sroot}/etc"
220   echo "10240" > "${sconf}/base/etc/md_size" || chk
221   ( cd "${stage}" && find etc | cpio -o ) | gzip -9 > "${sconf}/base/etc.cpio.gz" || chk
222   chk rm -Rf "${stage}/etc"
223 }
224
225 do_imgvar() {
226   log Create var imgsrc
227   chk mv "${sroot}/var" "${stage}"
228   chk mkdir -p "${sroot}/var"
229   echo "131072" > "${sconf}/base/var/md_size" || chk
230   ( cd "${stage}" && find var | cpio -o ) | gzip -9 > "${sconf}/base/var.cpio.gz" || chk
231   chk rm -Rf "${stage}/var"
232 }
233
234 do_custom() {
235   log Patch in custom config
236   ( cd "${base}/conf" && find . | cpio -p --link "${sconf}/default" ) || chk
237   # Make sure files in default are newer than the tagfile, so they will be caught by saveconfig
238   find "${sconf}/default" -type f -print0 | xargs -0 touch
239 }
240
241 for step in ${sequence}
242 do
243   echo "${targets}" | grep -q "${step}" || err Unrecognized target "${step}"
244   do_${step}
245 done