Adding hook parameter to execute custom scripts.
[live-boot-grml.git] / scripts / live
1 #!/bin/sh
2
3 # set -e
4
5 export PATH="/root/usr/bin:/root/usr/sbin:/root/bin:/root/sbin:/usr/bin:/usr/sbin:/bin:/sbin"
6
7 echo "/root/lib" >> /etc/ld.so.conf
8 echo "/root/usr/lib" >> /etc/ld.so.conf
9
10 mountpoint="/live/image"
11 LIVE_MEDIA_PATH="live"
12
13 root_persistence="live-rw"
14 home_persistence="home-rw"
15 root_snapshot_label="live-sn"
16 home_snapshot_label="home-sn"
17
18 USERNAME="user"
19 USERFULLNAME="Live user"
20 HOSTNAME="host"
21
22 mkdir -p "${mountpoint}"
23
24 [ -f /etc/live.conf ] && . /etc/live.conf
25 export USERNAME USERFULLNAME HOSTNAME
26
27 . /scripts/live-helpers
28
29 if [ ! -f /live.vars ]
30 then
31         touch /live.vars
32 fi
33
34 Arguments ()
35 {
36         PRESEEDS=""
37
38         for ARGUMENT in $(cat /proc/cmdline)
39         do
40                 case "${ARGUMENT}" in
41                         access=*)
42                                 ACCESS="${ARGUMENT#access=}"
43                                 export ACCESS
44                                 ;;
45
46                         console=*)
47                                 DEFCONSOLE="${ARGUMENT#*=}"
48                                 export DEFCONSOLE
49                                 ;;
50
51                         debug)
52                                 DEBUG="Yes"
53                                 export DEBUG
54
55                                 set -x
56                                 ;;
57
58                         fetch=*)
59                                 FETCH="${ARGUMENT#fetch=}"
60                                 export FETCH
61                                 ;;
62
63                         hook=*)
64                                 HOOK="${ARGUMENT#hook=}"
65                                 export HOOK
66                                 ;;
67
68                         hostname=*)
69                                 HOSTNAME="${ARGUMENT#hostname=}"
70                                 LIVECONF="changed"
71                                 export HOSTNAME LIVECONF
72                                 ;;
73
74                         username=*)
75                                 USERNAME="${ARGUMENT#username=}"
76                                 LIVECONF="changed"
77                                 export USERNAME LIVECONF
78                                 ;;
79
80                         userfullname=*)
81                                 USERFULLNAME="${ARGUMENT#userfullname=}"
82                                 LIVECONF="changed"
83                                 export USERFULLNAME LIVECONF
84                                 ;;
85
86                         ignore_uuid)
87                                 IGNORE_UUID="Yes"
88                                 export IGNORE_UUID
89                                 ;;
90
91                         ip=*)
92                                 STATICIP="${ARGUMENT#ip=}"
93
94                                 if [ -z "${STATICIP}" ]
95                                 then
96                                         STATICIP="frommedia"
97                                 fi
98
99                                 export STATICIP
100                                 ;;
101
102                         keyb=*|kbd-chooser/method=*)
103                                 KBD="${ARGUMENT#*=}"
104                                 export KBD
105                                 ;;
106
107                         klayout=*|console-setup/layoutcode=*)
108                                 KLAYOUT="${ARGUMENT#*=}"
109                                 export KLAYOUT
110                                 ;;
111
112                         kvariant=*|console-setup/variantcode=*)
113                                 KVARIANT="${ARGUMENT#*=}"
114                                 export KVARIANT
115                                 ;;
116
117                         kmodel=*|console-setup/modelcode=*)
118                                 KMODEL="${ARGUMENT#*=}"
119                                 export KMODEL
120                                 ;;
121
122                         koptions=*)
123                                 KOPTIONS="${ARGUMENT#koptions=}"
124                                 export KOPTIONS
125                                 ;;
126
127                         live-getty)
128                                 LIVE_GETTY="1"
129                                 export LIVE_GETTY
130                                 ;;
131
132                         live-media=*|bootfrom=*)
133                                 LIVE_MEDIA="${ARGUMENT#*=}"
134                                 export LIVE_MEDIA
135                                 ;;
136
137                         live-media-encryption=*|encryption=*)
138                                 LIVE_MEDIA_ENCRYPTION="${ARGUMENT#*=}"
139                                 export LIVE_MEDIA_ENCRYPTION
140                                 ;;
141
142                         live-media-offset=*)
143                                 LIVE_MEDIA_OFFSET="${ARGUMENT#live-media-offset=}"
144                                 export LIVE_MEDIA_OFFSET
145                                 ;;
146
147                         live-media-path=*)
148                                 LIVE_MEDIA_PATH="${ARGUMENT#live-media-path=}"
149                                 export LIVE_MEDIA_PATH
150                                 ;;
151
152                         live-media-timeout=*)
153                                 LIVE_MEDIA_TIMEOUT="${ARGUMENT#live-media-timeout=}"
154                                 export LIVE_MEDIA_TIMEOUT
155                                 ;;
156
157                         locale=*|debian-installer/locale=*)
158                                 LOCALE="${ARGUMENT#*=}"
159                                 export LOCALE
160                                 ;;
161
162                         module=*)
163                                 MODULE="${ARGUMENT#module=}"
164                                 export MODULE
165                                 ;;
166
167                         netboot=*)
168                                 NETBOOT="${ARGUMENT#netboot=}"
169                                 export NETBOOT
170                                 ;;
171
172                         nfsopts=*)
173                                 NFSOPTS="${ARGUMENT#nfsopts=}"
174                                 export NFSOPTS
175                                 ;;
176
177                         nfscow=*)
178                                 NFS_COW="${ARGUMENT#nfscow=}"
179                                 export NFS_COW
180                                 ;;
181
182                         noaccessibility)
183                                 NOACCESSIBILITY="Yes"
184                                 export NOACCESSIBILITY
185                                 ;;
186
187                         noapparmor)
188                                 NOAPPARMOR="Yes"
189                                 export NOAPPARMOR
190                                 ;;
191
192                         noaptcdrom)
193                                 NOAPTCDROM="Yes"
194                                 export NOAPTCDROM
195                                 ;;
196
197                         noautologin)
198                                 NOAUTOLOGIN="Yes"
199                                 export NOAUTOLOGIN
200                                 ;;
201
202                         noxautologin)
203                                 NOXAUTOLOGIN="Yes"
204                                 export NOXAUTOLOGIN
205                                 ;;
206
207                         noconsolekeyboard)
208                                 NOCONSOLEKEYBOARD="Yes"
209                                 export NOCONSOLEKEYBOARD
210                                 ;;
211
212                         nofastboot)
213                                 NOFASTBOOT="Yes"
214                                 export NOFASTBOOT
215                                 ;;
216
217                         nofstab)
218                                 NOFSTAB="Yes"
219                                 export NOFSTAB
220                                 ;;
221
222                         nognomepanel)
223                                 NOGNOMEPANEL="Yes"
224                                 export NOGNOMEPANEL
225                                 ;;
226
227                         nohosts)
228                                 NOHOSTS="Yes"
229                                 export NOHOSTS
230                                 ;;
231
232                         nokpersonalizer)
233                                 NOKPERSONALIZER="Yes"
234                                 export NOKPERSONALIZER
235                                 ;;
236
237                         nokwallet)
238                                 NOKWALLET="Yes"
239                                 export NOKWALLET
240                                 ;;
241
242                         nolanguageselector)
243                                 NOLANGUAGESELECTOR="Yes"
244                                 export NOLANGUAGESELECTOR
245                                 ;;
246
247                         nolocales)
248                                 NOLOCALES="Yes"
249                                 export NOLOCALES
250                                 ;;
251
252                         nonetworking)
253                                 NONETWORKING="Yes"
254                                 export NONETWORKING
255                                 ;;
256
257                         nopowermanagement)
258                                 NOPOWERMANAGEMENT="Yes"
259                                 export NOPOWERMANAGEMENT
260                                 ;;
261
262                         noprogramcrashes)
263                                 NOPROGRAMCRASHES="Yes"
264                                 export NOPROGRAMCRASHES
265                                 ;;
266
267                         norestrictedmanager)
268                                 NORESTRICTEDMANAGER="Yes"
269                                 export NORESTRICTEDMANAGER
270                                 ;;
271
272                         nosudo)
273                                 NOSUDO="Yes"
274                                 export NOSUDO
275                                 ;;
276
277                         noswap)
278                                 NOSWAP="Yes"
279                                 export NOSWAP
280                                 ;;
281
282                         noupdatenotifier)
283                                 NOUPDATENOTIFIER="Yes"
284                                 export NOUPDATENOTIFIER
285                                 ;;
286
287                         nouser)
288                                 NOUSER="Yes"
289                                 export NOUSER
290                                 ;;
291
292                         noxautoconfig)
293                                 NOXAUTOCONFIG="Yes"
294                                 export NOXAUTOCONFIG
295                                 ;;
296
297                         noxscreensaver)
298                                 NOXSCREENSAVER="Yes"
299                                 export NOXSCREENSAVER
300                                 ;;
301
302                         persistent)
303                                 PERSISTENT="Yes"
304                                 export PERSISTENT
305                                 ;;
306
307                         nopersistent)
308                                 NOPERSISTENT="Yes"
309                                 export NOPERSISTENT
310                                 ;;
311
312                         preseed/file=*|file=*)
313                                 LOCATION="${ARGUMENT#*=}"
314                                 export LOCATION
315                                 ;;
316
317                         nopreseed)
318                                 NOPRESEED="Yes"
319                                 export NOPRESEED
320                                 ;;
321
322                         url=*)
323                                 location="${ARGUMENT#url=}"
324
325                                 mount -n -o bind /sys /root/sys
326                                 mount -n -o bind /proc /root/proc
327                                 mount -n -o bind /dev /root/dev
328
329                                 mkdir -p /root/var/run/network
330                                 chroot /root ifup -a
331                                 chroot /root wget -P /tmp "${location}"
332                                 chroot /root ifdown -a
333
334                                 umount /root/sys
335                                 umount /root/proc
336                                 umount /root/dev
337
338                                 LOCATION="/tmp/$(basename "${location}")"
339                                 ;;
340
341                         */*=*)
342                                 question="${ARGUMENT%%=*}"
343                                 value="${ARGUMENT#*=}"
344                                 PRESEEDS="${PRESEEDS}\"${question}=${value}\" "
345                                 export PRESEEDS
346                                 ;;
347
348                         showmounts)
349                                 SHOWMOUNTS="Yes"
350                                 export SHOWMOUNTS
351                                 ;;
352
353                         textonly)
354                                 TEXTONLY="Yes"
355                                 export TEXTONLY
356                                 ;;
357
358                         timezone=*)
359                                 TIMEZONE="${ARGUMENT#timezone=}"
360                                 export TIMEZONE
361                                 ;;
362
363                         notimezone)
364                                 NOTIMEZONE="Yes"
365                                 export NOTIMEZONE
366                                 ;;
367
368                         todisk=*)
369                                 TODISK="${ARGUMENT#todisk=}"
370                                 export TODISK
371                                 ;;
372
373                         toram)
374                                 TORAM="Yes"
375                                 export TORAM
376                                 ;;
377
378                         toram=*)
379                                 TORAM="Yes"
380                                 MODULETORAM="${ARGUMENT#toram=}"
381                                 export TORAM MODULETORAM
382                                 ;;
383
384                         exposedroot)
385                                 EXPOSED_ROOT="Yes"
386                                 export EXPOSED_ROOT
387                                 ;;
388
389                         union=*)
390                                 UNIONTYPE="${ARGUMENT#union=}"
391                                 export UNIONTYPE
392                                 ;;
393
394                         utc=*)
395                                 UTC="${ARGUMENT#utc=}"
396                                 export UTC
397                                 ;;
398
399                         xdebconf)
400                                 XDEBCONF="Yes"
401                                 export XDEBCONF
402                                 ;;
403
404                         xdriver=*)
405                                 XDRIVER="${ARGUMENT#xdriver=}"
406                                 export XDRIVER
407                                 ;;
408
409                         xvideomode=*)
410                                 XVIDEOMODE="${ARGUMENT#xvideomode=}"
411                                 export XVIDEOMODE
412                                 ;;
413                 esac
414         done
415
416         # sort of compatibility with netboot.h from linux docs
417         if [ -z "${NETBOOT}" ]
418         then
419                 if [ "${ROOT}" = "/dev/nfs" ]
420                 then
421                         NETBOOT="nfs"
422                         export NETBOOT
423                 elif [ "${ROOT}" = "/dev/cifs" ]
424                 then
425                         NETBOOT="cifs"
426                         export NETBOOT
427                 fi
428         fi
429
430         if [ -z "${MODULE}" ]
431         then
432                 MODULE="filesystem"
433                 export MODULE
434         fi
435
436         if [ -z "${UNIONTYPE}" ]
437         then
438                 UNIONTYPE="unionfs"
439                 export UNIONTYPE
440         fi
441 }
442
443 is_live_path ()
444 {
445         DIRECTORY="${1}"
446
447         if [ -d "${DIRECTORY}"/"${LIVE_MEDIA_PATH}" ]
448         then
449                 for FILESYSTEM in squashfs ext2 ext3 xfs dir jffs2
450                 do
451                         if [ "$(echo ${DIRECTORY}/${LIVE_MEDIA_PATH}/*.${FILESYSTEM})" != "${DIRECTORY}/${LIVE_MEDIA_PATH}/*.${FILESYSTEM}" ]
452                         then
453                                 return 0
454                         fi
455                 done
456         fi
457
458         return 1
459 }
460
461 matches_uuid ()
462 {
463         if [ "${IGNORE_UUID}" ] || [ ! -e /conf/uuid.conf ]
464         then
465                 return 0
466         fi
467
468         path="${1}"
469         uuid="$(cat /conf/uuid.conf)"
470
471         for try_uuid_file in "${mountpoint}/.disk/casper-uuid"*
472         do
473                 [ -e "${try_uuid_file}" ] || continue
474
475                 try_uuid="$(cat "${try_uuid_file}")"
476
477                 if [ "${uuid}" = "${try_uuid}" ]
478                 then
479                         return 0
480                 fi
481         done
482
483         return 1
484 }
485
486 get_backing_device ()
487 {
488         case "${1}" in
489                 *.squashfs|*.ext2|*.ext3|*.jffs2)
490                         echo $(setup_loop "${1}" "loop" "/sys/block/loop*" '0' "${LIVE_MEDIA_ENCRYPTION}")
491                         ;;
492
493                 *.dir)
494                         echo "directory"
495                         ;;
496
497                 *)
498                         panic "Unrecognized live filesystem: ${1}"
499                         ;;
500         esac
501 }
502
503 match_files_in_dir ()
504 {
505         # Does any files match pattern ${1} ?
506         local pattern="${1}"
507
508         if [ "$(echo ${pattern})" != "${pattern}" ]
509         then
510                 return 0
511         fi
512
513         return 1
514 }
515
516 mount_images_in_directory ()
517 {
518         directory="${1}"
519         rootmnt="${2}"
520         mac="${3}"
521
522
523         if match_files_in_dir "${directory}/${LIVE_MEDIA_PATH}/*.squashfs" ||
524                 match_files_in_dir "${directory}/${LIVE_MEDIA_PATH}/*.ext2" ||
525                 match_files_in_dir "${directory}/${LIVE_MEDIA_PATH}/*.ext3" ||
526                 match_files_in_dir "${directory}/${LIVE_MEDIA_PATH}/*.jffs2" ||
527                 match_files_in_dir "${directory}/${LIVE_MEDIA_PATH}/*.dir"
528         then
529                 [ -n "${mac}" ] && adddirectory="${directory}/${LIVE_MEDIA_PATH}/${mac}"
530                 setup_unionfs "${directory}/${LIVE_MEDIA_PATH}" "${rootmnt}" "${adddirectory}"
531         else
532                 :
533         fi
534 }
535
536 is_nice_device ()
537 {
538         sysfs_path="${1#/sys}"
539
540         if /lib/udev/path_id "${sysfs_path}" | grep -E -q "ID_PATH=(usb|pci-[^-]*-[ide|scsi|usb])"
541         then
542                 return 0
543         fi
544
545         return 1
546 }
547
548 is_supported_fs ()
549 {
550         # FIXME: do something better like the scan of supported filesystems
551         fstype="${1}"
552
553         case ${fstype} in
554                 vfat|iso9660|udf|ext2|ext3|ntfs|jffs2)
555                         return 0
556                         ;;
557         esac
558
559         return 1
560 }
561
562 copy_live_to ()
563 {
564         copyfrom="${1}"
565         copytodev="${2}"
566         copyto="${copyfrom}_swap"
567
568         if [ -z "${MODULETORAM}" ]
569         then
570                 size=$(fs_size "" ${copyfrom} "used")
571         else
572                 MODULETORAMFILE="${copyfrom}/${LIVE_MEDIA_PATH}/${MODULETORAM}"
573
574                 if [ -f "${MODULETORAMFILE}" ]
575                 then
576                         size=$( expr $(ls -la ${MODULETORAMFILE} | awk '{print $5}') / 1024 + 5000 )
577                 else
578                         log_warning_msg "Error: toram-module ${MODULETORAM} (${MODULETORAMFILE}) could not be read."
579                         return 1
580                 fi
581         fi
582
583         if [ "${copytodev}" = "ram" ]
584         then
585                 # copying to ram:
586                 freespace=$( expr $(awk '/MemFree/{print $2}' /proc/meminfo) + $( cat /proc/meminfo | grep Cached | head -n 1 | awk '/Cached/{print $2}' - ) )
587                 mount_options="-o size=${size}k"
588                 free_string="memory"
589                 fstype="tmpfs"
590                 dev="/dev/shm"
591         else
592                 # it should be a writable block device
593                 if [ -b "${copytodev}" ]
594                 then
595                         dev="${copytodev}"
596                         free_string="space"
597                         fstype=$(get_fstype "${dev}")
598                         freespace=$(fs_size "${dev}")
599                 else
600                         [ "$quiet" != "y" ] && log_warning_msg "${copytodev} is not a block device."
601                         return 1
602                 fi
603         fi
604
605         if [ "${freespace}" -lt "${size}" ]
606         then
607                 [ "${quiet}" != "y" ] && log_warning_msg "Not enough free ${free_string} (${freespace}k free, ${size}k needed) to copy live media in ${copytodev}."
608                 return 1
609         fi
610
611         # begin copying (or uncompressing)
612         mkdir "${copyto}"
613         echo "mount -t ${fstype} ${mount_options} ${dev} ${copyto}"
614         mount -t "${fstype}" ${mount_options} "${dev}" "${copyto}"
615
616         if [ "${extension}" = "tgz" ]
617         then
618                 cd "${copyto}"
619                 tar zxf "${copyfrom}/${LIVE_MEDIA_PATH}/$(basename ${FETCH})"
620                 rm -f "${copyfrom}/${LIVE_MEDIA_PATH}/$(basename ${FETCH})"
621                 mount -r --move "${copyto}" "${rootmnt}"
622                 cd "${OLDPWD}"
623         else
624                 if [ -n "${MODULETORAMFILE}" ]
625                 then
626                         cp ${MODULETORAMFILE} ${copyto} # copy only the filesystem module
627                 else
628                         cp -a ${copyfrom}/* ${copyto}   # "cp -a" from busybox also copies hidden files
629                 fi
630
631                 livefs_root
632                 umount ${copyfrom}
633                 mount -r --move ${copyto} ${copyfrom}
634         fi
635
636         rmdir ${copyto}
637         return 0
638 }
639
640 do_netmount ()
641 {
642         rc=1
643
644         modprobe -q af_packet # For DHCP
645
646         udevtrigger
647         udevsettle
648
649         ipconfig ${DEVICE} | tee /netboot.config
650
651         # source relevant ipconfig output
652         OLDHOSTNAME=${HOSTNAME}
653         . /tmp/net-${DEVICE}.conf
654         [ -z ${HOSTNAME} ] && HOSTNAME=${OLDHOSTNAME}
655         export HOSTNAME
656
657         if [ "${NFSROOT}" = "auto" ]
658         then
659                 NFSROOT=${ROOTSERVER}:${ROOTPATH}
660         fi
661
662         if [ -n "${FETCH}" ] && do_httpmount
663         then
664                 rc=0
665                 return ${rc}
666         fi
667
668         if [ "${NFSROOT#*:}" = "${NFSROOT}" ] && [ "$NETBOOT" != "cifs" ]
669         then
670                 NFSROOT=${ROOTSERVER}:${NFSROOT}
671         fi
672
673         [ "${quiet}" != "y" ] && log_begin_msg "Trying netboot from ${NFSROOT}"
674
675         if [ "${NETBOOT}" != "nfs" ] && do_cifsmount
676         then
677                 rc=0
678         elif do_nfsmount
679         then
680                 NETBOOT="nfs"
681                 export NETBOOT
682                 rc=0
683         fi
684
685         [ "${quiet}" != "y" ] && log_end_msg
686         return ${rc}
687 }
688
689 do_httpmount ()
690 {
691         rc=1
692         extension=$(echo "${FETCH}" | sed 's/\(.*\)\.\(.*\)/\2/')
693
694         case "${extension}" in
695                 squashfs|tgz|tar)
696                         [ "${quiet}" != "y" ] && log_begin_msg "Trying wget ${FETCH} -O ${mountpoint}/$(basename ${FETCH})"
697                         mkdir -p "${mountpoint}/${LIVE_MEDIA_PATH}"
698                         wget "${FETCH}" -O "${mountpoint}/${LIVE_MEDIA_PATH}/$(basename ${FETCH})"
699                         [ ${?} -eq 0 ] && rc=0
700                         [ "${extension}" = "tgz" ] && live_dest="ram"
701                         ;;
702
703                 *)
704                         [ "${quiet}" != "y" ] && log_begin_msg "Unrecognized archive extension for ${FETCH}"
705         esac
706
707         return ${rc}
708 }
709
710 do_nfsmount ()
711 {
712         rc=1
713
714         modprobe -q nfs
715
716         if [ -z "${NFSOPTS}" ]
717         then
718                 NFSOPTS=""
719         fi
720
721         [ "${quiet}" != "y" ] && log_begin_msg "Trying nfsmount -o nolock -o ro ${NFSOPTS} ${NFSROOT} ${mountpoint}"
722
723         # FIXME: This for loop is an ugly HACK round an nfs bug
724         for i in 0 1 2 3 4 5 6 7 8 9 a b c d e f 10 11 12 13
725         do
726                 nfsmount -o nolock -o ro ${NFSOPTS} "${NFSROOT}" "${mountpoint}" && rc=0 && break
727                 sleep 1
728         done
729
730         return ${rc}
731 }
732
733 do_cifsmount ()
734 {
735         rc=1
736
737         if [ -x "/sbin/mount.cifs" ]
738         then
739                 if [ -z "${NFSOPTS}" ]
740                 then
741                         CIFSOPTS="-ouser=root,password="
742                 else
743                         CIFSOPTS="${NFSOPTS}"
744                 fi
745
746                 [ "${quiet}" != "y" ] && log_begin_msg "Trying mount.cifs ${NFSROOT} ${mountpoint} ${CIFSOPTS}"
747                 modprobe -q cifs
748
749                 if mount.cifs "${NFSROOT}" "${mountpoint}" "${CIFSOPTS}"
750                 then
751                         rc=0
752                 fi
753         fi
754
755         return ${rc}
756 }
757
758 do_snap_copy ()
759 {
760         fromdev="${1}"
761         todir="${2}"
762         snap_type="${3}"
763         size=$(fs_size "${fromdev}" "" "used")
764
765         if [ -b "${fromdev}" ]
766         then
767                 # look for free mem
768                 if [ -n "${HOMEMOUNTED}" -a "${snap_type}" = "HOME" ]
769                 then
770                         todev=$(cat /proc/mounts | grep -s " $(base_path ${todir}) " | awk '{print $1}' )
771                         freespace=$(df -k  | grep -s ${todev} | awk '{print $4}')
772                 else
773                         freespace=$( expr $(awk '/MemFree/{print $2}' /proc/meminfo) + $( cat /proc/meminfo | grep Cached | head -n 1 | awk '/Cached/{print $2}' - ))
774                 fi
775
776                 tomount="/mnt/tmpsnap"
777
778                 if [ ! -d "${tomount}" ]
779                 then
780                         mkdir -p "${tomount}"
781                 fi
782
783                 fstype=$(get_fstype "${fromdev}")
784
785                 if [ -n "${fstype}" ]
786                 then
787                         # Copying stuff...
788                         mount -t "${fstype}" -o ro,noatime "${fromdev}" "${tomount}"
789                         cp -a "${tomount}"/* ${todir}
790                         umount "${tomount}"
791                 else
792                         log_warning_msg "Unrecognized fstype: ${fstype} on ${fromdev}:${snap_type}"
793                 fi
794
795                 rmdir "${tomount}"
796
797                 if echo ${fromdev} | grep -qs loop
798                 then
799                         losetup -d "${fromdev}"
800                 fi
801
802                 return 0
803         else
804                 return 1
805
806                 [ "${quiet}" != "y" ] && log_warning_msg "Unable to find the snapshot ${snap_type} medium"
807         fi
808 }
809
810 try_snap ()
811 {
812         # Look for ${snap_label}.* in block devices and copy the contents to ${snap_mount}
813         # and remember the device and filename for resync on exit in live-initramfs.init
814
815         snap_label="${1}"
816         snap_mount="${2}"
817         snap_type="${3}"
818         snapdata=$(find_files "${snap_label}.squashfs ${snap_label}.cpio.gz ${snap_label}.ext2 ${snap_label}.ext3 ${snap_label}.jffs2")
819
820         if [ ! -z "${snapdata}" ]
821         then
822                 snapdev="$(echo ${snapdata} | cut -f1 -d ' ')"
823                 snapback="$(echo ${snapdata} | cut -f2 -d ' ')"
824                 snapfile="$(echo ${snapdata} | cut -f3 -d ' ')"
825
826                 if echo "${snapfile}" | grep -qs '\(squashfs\|ext2\|ext3\|jffs2\)'
827                 then
828                         # squashfs, jffs2 or ext2/ext3 snapshot
829                         dev=$(get_backing_device "${snapback}/${snapfile}")
830
831                         if ! do_snap_copy "${dev}" "${snap_mount}" "${snap_type}"
832                         then
833                                 log_warning_msg "Impossible to include the ${snapfile} Snapshot"
834                                 return 1
835                         fi
836                 else
837                         # cpio.gz snapshot
838                         if ! (cd "${snap_mount}" && zcat "${snapback}/${snapfile}" | cpio -i -u -d 2>/dev/null)
839                         then
840                                 log_warning_msg "Impossible to include the ${snapfile} Snapshot"
841                                 return 1
842                         fi
843                 fi
844
845                 umount "${snapback}"
846         else
847                 dev=$(find_cow_device "${snap_label}")
848
849                 if [ -b ${dev} ]
850                 then
851                         if echo "${dev}" | grep -qs loop
852                         then
853                                 # strange things happens, user confused?
854                                 snaploop=$( losetup ${dev} | awk '{print $3}' | tr -d '()' )
855                                 snapfile=$(basename ${snaploop})
856                                 snapdev=$(cat /proc/mounts | awk '{print $2,$1}' | grep -es "^$( dirname ${snaploop} )" | cut -f2 -d ' ')
857                         else
858                                 snapdev="${dev}"
859                         fi
860
861                         if ! do_snap_copy "${dev}" "${snap_mount}" "${snap_type}"
862                         then
863                                 log_warning_msg "Impossible to include the ${snap_label} Snapshot"
864                                 return 1
865                         else
866                                 if [ -n "${snapfile}" ]
867                                 then
868                                         # it was a loop device, user confused
869                                         umount ${snapdev}
870                                 fi
871                         fi
872                 else
873                         log_warning_msg "Impossible to include the ${snap_label} Snapshot"
874                         return 1
875                 fi
876         fi
877
878         echo "export ${snap_type}SNAP="${snap_mount}":${snapdev}:${snapfile}" >> /etc/live.conf # for resync on reboot/halt
879         return 0
880 }
881
882 setup_unionfs ()
883 {
884         image_directory="${1}"
885         rootmnt="${2}"
886         addimage_directory="${3}"
887         modprobe -q -b ${UNIONTYPE}
888
889         # run-init can't deal with images in a subdir, but we're going to
890         # move all of these away before it runs anyway.  No, we're not,
891         # put them in / since move-mounting them into / breaks mono and
892         # some other apps.
893
894         croot="/"
895
896         # Let's just mount the read-only file systems first
897         rofsstring=""
898         rofslist=""
899         minor_kernel_version=$(uname -r|cut -c 5-|sed 's/[^0-9].*//')
900
901         if [ "${NETBOOT}" = "nfs" ] && [ "${minor_kernel_version}" -lt 22 ]
902         then
903                 # go aroung a bug in nfs-unionfs locking for unionfs <= 1.4
904                 roopt="nfsro"
905         elif [ "${UNIONTYPE}" = "aufs" ]
906         then
907                 roopt="rr"
908         else
909                 roopt="ro"
910         fi
911
912         # Read image names from ${MODULE}.module if it exists
913         if [ -e "${image_directory}/filesystem.${MODULE}.module" ]
914         then
915                 for IMAGE in $(cat ${image_directory}/filesystem.${MODULE}.module)
916                 do
917                         image_string="${image_string} ${image_directory}/${IMAGE}"
918                 done
919         elif [ -e "${image_directory}/${MODULE}.module" ]
920         then
921                 for IMAGE in $(cat ${image_directory}/${MODULE}.module)
922                 do
923                         image_string="${image_string} ${image_directory}/${IMAGE}"
924                 done
925         else
926                 # ${MODULE}.module does not exist, create a list of images
927                 for FILESYSTEM in squashfs ext2 ext3 xfs jffs2 dir
928                 do
929                         for IMAGE in "${image_directory}"/*."${FILESYSTEM}"
930                         do
931                                 if [ -e "${IMAGE}" ]
932                                 then
933                                         image_string="${image_string} ${IMAGE}"
934                                 fi
935                         done
936                 done
937
938                 if [ -n "${addimage_directory}" ] && [ -d "${addimage_directory}" ]
939                 then
940                         for FILESYSTEM in squashfs ext2 ext3 xfs jffs2 dir
941                         do
942                                 for IMAGE in "${addimage_directory}"/*."${FILESYSTEM}"
943                                 do
944                                         if [ -e "${IMAGE}" ]
945                                         then
946                                                 image_string="${image_string} ${IMAGE}"
947                                         fi
948                                 done
949                         done
950                 fi
951
952                 # Now sort the list
953                 image_string="$(echo ${image_string} | sed -e 's/ /\n/g' | sort )"
954         fi
955
956         [ -n "${MODULETORAMFILE}" ] && image_string="${image_directory}/$(basename ${MODULETORAMFILE})"
957
958         mkdir -p "${croot}"
959
960         for image in ${image_string}
961         do
962                 imagename=$(basename "${image}")
963
964                 if [ -d "${image}" ]
965                 then
966                         # it is a plain directory: do nothing
967                         rofsstring="${image}=${roopt}:${rofsstring}"
968                         rofslist="${image} ${rofslist}"
969                 elif [ -f "${image}" ]
970                 then
971                         backdev=$(get_backing_device "${image}")
972                         fstype=$(get_fstype "${backdev}")
973
974                         if [ "${fstype}" = "unknown" ]
975                         then
976                                 panic "Unknown file system type on ${backdev} (${image})"
977                         fi
978
979                         mkdir -p "${croot}/${imagename}"
980                         echo "debug: Can not mount backdev ${backdev} (image = ${image}) on croot/imagename ${croot}/${imagename}"
981                         mount -t "${fstype}" -o ro,noatime "${backdev}" "${croot}/${imagename}" || panic "Can not mount ${backdev} (${image}) on ${croot}/${imagename}" && rofsstring="${croot}/${imagename}=${roopt}:${rofsstring}" && rofslist="${croot}/${imagename} ${rofslist}"
982                 fi
983         done
984
985         rofsstring=${rofsstring%:}
986
987         mkdir -p /cow
988
989         # Looking for "${root_persistence}" device or file
990         if [ -n "${PERSISTENT}" ]
991         then
992                 cowprobe=$(find_cow_device "${root_persistence}")
993
994                 if [ -b "${cowprobe}" ]
995                 then
996                         cowdevice=${cowprobe}
997                         cow_fstype=$(get_fstype "${cowprobe}")
998                 else
999                         [ "${quiet}" != "y" ] && log_warning_msg "Unable to find the persistent medium"
1000                         cowdevice="tmpfs"
1001                         cow_fstype="tmpfs"
1002                 fi
1003         elif [ -n "${NFS_COW}" ]
1004         then
1005                 # check if there are any nfs options
1006                 if echo ${NFS_COW}|grep -q ','
1007                 then
1008                         nfs_cow_opts="-o nolock,$(echo ${NFS_COW}|cut -d, -f2-)"
1009                         nfs_cow=$(echo ${NFS_COW}|cut -d, -f1)
1010                 else
1011                         nfs_cow_opts="-o nolock"
1012                         nfs_cow=${NFS_COW}
1013                 fi
1014                 mac="$(get_mac)"
1015                 if [ -n "${mac}" ]
1016                 then
1017                         cowdevice=$(echo ${nfs_cow}|sed "s/client_mac_address/${mac}/")
1018                         cow_fstype="nfs"
1019                 else
1020                         panic "unable to determine mac address"
1021                 fi
1022         else
1023                 cowdevice="tmpfs"
1024                 cow_fstype="tmpfs"
1025         fi
1026
1027         if [ "${cow_fstype}" = "nfs" ]
1028         then
1029                 [ "${quiet}" != "y" ] && log_begin_msg \
1030                         "Trying nfsmount ${nfs_cow_opts} ${cowdevice} /cow"
1031                 nfsmount ${nfs_cow_opts} ${cowdevice} /cow || \
1032                         panic "Can not mount ${cowdevice} on /cow"
1033         else
1034                 mount ${cowdevice} -t ${cow_fstype} -o rw,noatime /cow || \
1035                         panic "Can not mount ${cowdevice} on /cow"
1036         fi
1037
1038         rofscount=$(echo ${rofslist} |wc -w)
1039
1040         if [ -n "${EXPOSED_ROOT}" ]
1041         then
1042                 if [ ${rofscount} -ne 1 ]
1043                 then
1044                         panic "only one RO file system supported with exposedroot: ${rofslist}"
1045                 fi
1046                 exposedrootfs=${rofslist%% }
1047
1048                 mount --bind ${exposedrootfs} ${rootmnt} || \
1049                         panic "bind mount of ${exposedrootfs} failed"
1050
1051                 cow_dirs='/var/tmp /var/lock /var/run /var/log /var/spool
1052                         /home /var/lib/live'
1053
1054                 for dir in ${cow_dirs}; do
1055                         mkdir -p /cow${dir}
1056                         mount -t ${UNIONTYPE} \
1057                                 -o rw,noatime,dirs=/cow${dir}=rw:${exposedrootfs}${dir}=ro \
1058                                 ${UNIONTYPE} "${rootmnt}${dir}" || \
1059                                 panic "mount ${UNIONTYPE} on ${rootmnt}${dir} failed with option \
1060                                         rw,noatime,dirs=/cow${dir}=rw:${exposedrootfs}${dir}=ro"
1061                 done
1062         else
1063                 mount -t ${UNIONTYPE} -o noatime,dirs=/cow=rw:${rofsstring} \
1064                         ${UNIONTYPE} "${rootmnt}" || panic "mount ${UNIONTYPE} on \
1065                         ${rootmnt} failed with option noatime,dirs=/cow=rw:${rofsstring}"
1066         fi
1067
1068         # tmpfs file systems
1069         mkdir -p "${rootmnt}/live"
1070         mount -t tmpfs tmpfs ${rootmnt}/live
1071
1072         # Adding other custom mounts
1073         if [ -n "${PERSISTENT}" ]
1074         then
1075                 # directly mount /home
1076                 # FIXME: add a custom mounts configurable system
1077                 homecow=$(find_cow_device "${home_persistence}" )
1078
1079                 if [ -b "${homecow}" ]
1080                 then
1081                         mount -t $(get_fstype "${homecow}") -o rw,noatime "${homecow}" "${rootmnt}/home"
1082                         export HOMEMOUNTED=1 # used to proper calculate free space in do_snap_copy()
1083                 else
1084                         [ "${quiet}" != "y" ] && log_warning_msg "Unable to find the persistent home medium"
1085                 fi
1086
1087                 # Look for other snapshots to copy in
1088                 try_snap "${root_snapshot_label}" "${rootmnt}" "ROOT"
1089                 try_snap "${home_snapshot_label}" "${rootmnt}/home" "HOME"
1090         fi
1091
1092         if [ -n "${SHOWMOUNTS}" ]
1093         then
1094                 for d in ${rofslist}
1095                 do
1096                         mkdir -p "${rootmnt}/live/${d##*/}"
1097
1098                         case d in
1099                                 *.dir)
1100                                         # do nothing # mount -o bind "${d}" "${rootmnt}/live/${d##*/}"
1101                                         ;;
1102
1103                                 *)
1104                                         mount --move "${d}" "${rootmnt}/live/${d##*/}"
1105                                         ;;
1106                         esac
1107                 done
1108         fi
1109
1110         # shows cow fs on /cow for use by live-snapshot
1111         mkdir -p "${rootmnt}/live/cow"
1112         mount -o move /cow "${rootmnt}/live/cow"
1113 }
1114
1115 check_dev ()
1116 {
1117         sysdev="${1}"
1118         devname="${2}"
1119         skip_uuid_check="${3}"
1120
1121         if [ -z "${devname}" ]
1122         then
1123                 devname=$(sys2dev "${sysdev}")
1124         fi
1125
1126         if [ -n "${LIVE_MEDIA_OFFSET}" ]
1127         then
1128                 loopdevname=$(setup_loop "${devname}" "loop" "/sys/block/loop*" "${LIVE_MEDIA_OFFSET}" '')
1129                 devname="${loopdevname}"
1130         fi
1131
1132         fstype=$(get_fstype "${devname}")
1133
1134         if is_supported_fs ${fstype}
1135         then
1136                 mount -t ${fstype} -o ro,noatime "${devname}" ${mountpoint} || continue
1137
1138                 if is_live_path ${mountpoint} && \
1139                         ([ "${skip_uuid_check}" ] || matches_uuid ${mountpoint})
1140                 then
1141                         echo ${mountpoint}
1142                         return 0
1143                 else
1144                         umount ${mountpoint}
1145                 fi
1146         fi
1147
1148         if [ -n "${LIVE_MEDIA_OFFSET}" ]
1149         then
1150                 losetup -d "${loopdevname}"
1151         fi
1152
1153         return 1
1154 }
1155
1156 find_livefs ()
1157 {
1158         timeout="${1}"
1159
1160         # first look at the one specified in the command line
1161         if [ ! -z "${LIVE_MEDIA}" ]
1162         then
1163                 if check_dev "null" "${LIVE_MEDIA}" "skip_uuid_check"
1164                 then
1165                         return 0
1166                 fi
1167         fi
1168
1169         # don't start autodetection before timeout has expired
1170         if [ -n "${LIVE_MEDIA_TIMEOUT}" ]
1171         then
1172                 if [ "${timeout}" -lt "${LIVE_MEDIA_TIMEOUT}" ]
1173                 then
1174                         return 1
1175                 fi
1176         fi
1177
1178         # or do the scan of block devices
1179         for sysblock in $(echo /sys/block/* | tr ' ' '\n' | grep -v loop | grep -v ram | grep -v 'dm-')
1180         do
1181                 devname=$(sys2dev "${sysblock}")
1182                 fstype=$(get_fstype "${devname}")
1183
1184                 if /lib/udev/cdrom_id ${devname} > /dev/null
1185                 then
1186                         if check_dev "null" "${devname}"
1187                         then
1188                                 return 0
1189                         fi
1190                 elif is_nice_device "${sysblock}"
1191                 then
1192                         for dev in $(subdevices "${sysblock}")
1193                         do
1194                                 if check_dev "${dev}"
1195                                 then
1196                                         return 0
1197                                 fi
1198                         done
1199                 elif [ "${fstype}" = "squashfs" -o \
1200                         "${fstype}" = "ext2" -o \
1201                         "${fstype}" = "ext3" -o \
1202                         "${fstype}" = "jffs2" ]
1203                 then
1204                         # This is an ugly hack situation, the block device has
1205                         # an image directly on it.  It's hopefully
1206                         # live-initramfs, so take it and run with it.
1207                         ln -s "${devname}" "${devname}.${fstype}"
1208                         echo "${devname}.${fstype}"
1209                         return 0
1210                 fi
1211         done
1212
1213         return 1
1214 }
1215
1216 set_usplash_timeout ()
1217 {
1218         if [ -x /sbin/usplash_write ]
1219         then
1220                 /sbin/usplash_write "TIMEOUT 120"
1221         fi
1222 }
1223
1224 mountroot ()
1225 {
1226     
1227         if [ -x /scripts/local-top/cryptroot ]; then
1228             /scripts/local-top/cryptroot
1229         fi
1230
1231         exec 6>&1
1232         exec 7>&2
1233         exec > live.log
1234         exec 2>&1
1235
1236         Arguments
1237
1238         set_usplash_timeout
1239         [ "${quiet}" != "y" ] && log_begin_msg "Running /scripts/live-premount"
1240         run_scripts /scripts/live-premount
1241         [ "${quiet}" != "y" ] && log_end_msg
1242
1243         # Needed here too because some things (*cough* udev *cough*)
1244         # changes the timeout
1245
1246         set_usplash_timeout
1247
1248         if [ ! -z "${NETBOOT}" ] || [ ! -z "${FETCH}" ]
1249         then
1250                 if do_netmount
1251                 then
1252                         livefs_root="${mountpoint}"
1253                 else
1254                         panic "Unable to find a live file system on the network"
1255                 fi
1256         else
1257                 # Scan local devices for the image
1258                 for i in 0 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
1259                 do
1260                         livefs_root=$(find_livefs ${i})
1261
1262                         if [ -n "${livefs_root}" ]
1263                         then
1264                                 break
1265                         fi
1266
1267                         sleep 1
1268                 done
1269         fi
1270
1271         if [ -z "${livefs_root}" ]
1272         then
1273                 panic "Unable to find a medium containing a live file system"
1274         fi
1275
1276         if [ "${TORAM}" ]
1277         then
1278                 live_dest="ram"
1279         elif [ "${TODISK}" ]
1280         then
1281                 live_dest="${TODISK}"
1282         fi
1283
1284         if [ "${live_dest}" ]
1285         then
1286                 log_begin_msg "Copying live media to ${live_dest}"
1287                 copy_live_to "${livefs_root}" "${live_dest}"
1288                 log_end_msg
1289         fi
1290
1291         if [ -n "${MODULETORAMFILE}" ]
1292         then
1293                 setup_unionfs "${livefs_root}" "${rootmnt}"
1294         else
1295                 mac="$(get_mac)"
1296                 mac="$(echo ${mac} | sed 's/-//g')"
1297                 mount_images_in_directory "${livefs_root}" "${rootmnt}" "${mac}"
1298         fi
1299
1300         log_end_msg
1301
1302         maybe_break live-bottom
1303         [ "${quiet}" != "y" ] && log_begin_msg "Running /scripts/live-bottom"
1304
1305         run_scripts /scripts/live-bottom
1306         [ "${quiet}" != "y" ] && log_end_msg
1307
1308         exec 1>&6 6>&-
1309         exec 2>&7 7>&-
1310         cp live.log "${rootmnt}/var/log/"
1311 }