Milton D. Miller II | d89d5e0 | 2016-01-20 14:57:54 -0600 | [diff] [blame] | 1 | #!/bin/sh |
| 2 | |
Milton Miller | 54d882e | 2016-02-05 12:07:53 -0600 | [diff] [blame] | 3 | fslist="proc sys dev run" |
Milton D. Miller II | d89d5e0 | 2016-01-20 14:57:54 -0600 | [diff] [blame] | 4 | rodir=run/initramfs/ro |
| 5 | rwdir=run/initramfs/rw |
| 6 | upper=$rwdir/cow |
| 7 | work=$rwdir/work |
| 8 | |
| 9 | cd / |
Milton Miller | 54d882e | 2016-02-05 12:07:53 -0600 | [diff] [blame] | 10 | mkdir -p $fslist |
Milton D. Miller II | d89d5e0 | 2016-01-20 14:57:54 -0600 | [diff] [blame] | 11 | mount dev dev -tdevtmpfs |
| 12 | mount sys sys -tsysfs |
| 13 | mount proc proc -tproc |
| 14 | if ! grep run proc/mounts |
| 15 | then |
| 16 | mount tmpfs run -t tmpfs -o mode=755,nodev |
| 17 | fi |
| 18 | |
| 19 | mkdir -p $rodir $rwdir |
| 20 | |
| 21 | cp -rp init shutdown update whitelist bin sbin usr lib etc var run/initramfs |
| 22 | |
| 23 | # To start a interactive shell with job control at this point, run |
| 24 | # getty 38400 ttyS4 |
| 25 | |
| 26 | findmtd() { |
| 27 | m=$(grep -xl "$1" /sys/class/mtd/*/name) |
| 28 | m=${m%/name} |
| 29 | m=${m##*/} |
| 30 | echo $m |
| 31 | } |
| 32 | |
Andrew Jeffery | acc2c85 | 2016-02-23 23:47:23 +1030 | [diff] [blame] | 33 | blkid_fs_type() { |
| 34 | # Emulate util-linux's `blkid -s TYPE -o value $1` |
| 35 | # Example busybox blkid output: |
| 36 | # # blkid /dev/mtdblock5 |
| 37 | # /dev/mtdblock5: TYPE="squashfs" |
| 38 | # Process output to extract TYPE value "squashfs". |
| 39 | blkid $1 | sed -e 's/^.*TYPE="//' -e 's/".*$//' |
| 40 | } |
| 41 | |
| 42 | probe_fs_type() { |
| 43 | fst=$(blkid_fs_type $1) |
| 44 | echo ${fst:=jffs2} |
| 45 | } |
| 46 | |
Milton Miller | 06ccb1a | 2016-02-05 13:04:29 -0600 | [diff] [blame] | 47 | debug_takeover() { |
| 48 | echo "$@" |
| 49 | test -n "$@" && echo Enter password to try to manually fix. |
| 50 | cat << HERE |
| 51 | After fixing run exit to continue this script, or reboot -f to retry, or |
| 52 | touch /takeover and exit to become PID 1 allowing editing of this script. |
| 53 | HERE |
| 54 | |
| 55 | while ! sulogin && ! test -f /takeover |
| 56 | do |
| 57 | echo getty failed, retrying |
| 58 | done |
| 59 | |
| 60 | # Touch /takeover in the above getty to become pid 1 |
| 61 | if test -e /takeover |
| 62 | then |
| 63 | cat << HERE |
| 64 | |
| 65 | Takeover of init requested. Executing /bin/sh as PID 1. |
| 66 | When finished exec new init or cleanup and run reboot -f. |
| 67 | |
| 68 | Warning: No job control! Shell exit will panic the system! |
| 69 | HERE |
| 70 | export PS1=init#\ |
| 71 | exec /bin/sh |
| 72 | fi |
| 73 | } |
| 74 | |
Milton D. Miller II | bf7bfd4 | 2016-01-27 20:18:16 -0600 | [diff] [blame] | 75 | env=$(findmtd u-boot-env) |
| 76 | if test -n $env |
| 77 | then |
| 78 | ln -s /dev/$env /run/mtd:u-boot-env |
| 79 | cp /run/mtd:u-boot-env /run/fw_env |
| 80 | fi |
| 81 | |
Milton D. Miller II | d89d5e0 | 2016-01-20 14:57:54 -0600 | [diff] [blame] | 82 | rofs=$(findmtd rofs) |
| 83 | rwfs=$(findmtd rwfs) |
| 84 | |
Milton Miller | 54d882e | 2016-02-05 12:07:53 -0600 | [diff] [blame] | 85 | rodev=/dev/mtdblock${rofs#mtd} |
| 86 | rwdev=/dev/mtdblock${rwfs#mtd} |
| 87 | |
Milton Miller | 0b0d5fe | 2016-02-23 21:48:48 -0600 | [diff] [blame] | 88 | # Set to y for yes, anything else for no. |
| 89 | force_rwfst_jffs2=y |
| 90 | flash_images_before_init=n |
| 91 | |
Milton D. Miller II | d89d5e0 | 2016-01-20 14:57:54 -0600 | [diff] [blame] | 92 | rofst=squashfs |
Andrew Jeffery | acc2c85 | 2016-02-23 23:47:23 +1030 | [diff] [blame] | 93 | rwfst=$(probe_fs_type $rwdev) |
Milton Miller | 54d882e | 2016-02-05 12:07:53 -0600 | [diff] [blame] | 94 | roopts=ro |
| 95 | rwopts=rw |
| 96 | |
Milton Miller | 0b0d5fe | 2016-02-23 21:48:48 -0600 | [diff] [blame] | 97 | image=/run/initramfs/image- |
| 98 | trigger=${image}rwfs |
| 99 | |
Milton Miller | 54d882e | 2016-02-05 12:07:53 -0600 | [diff] [blame] | 100 | init=/sbin/init |
Milton Miller | 0b0d5fe | 2016-02-23 21:48:48 -0600 | [diff] [blame] | 101 | fsckbase=/sbin/fsck. |
| 102 | fsck=$fsckbase$rwfst |
Milton Miller | b72114c | 2016-02-06 16:05:06 -0600 | [diff] [blame] | 103 | fsckopts=-a |
Milton D. Miller II | d89d5e0 | 2016-01-20 14:57:54 -0600 | [diff] [blame] | 104 | |
| 105 | echo rofs = $rofs $rofst rwfs = $rwfs $rwfst |
| 106 | |
Milton Miller | 06ccb1a | 2016-02-05 13:04:29 -0600 | [diff] [blame] | 107 | if grep -w debug-init-sh /proc/cmdline |
Milton D. Miller II | d89d5e0 | 2016-01-20 14:57:54 -0600 | [diff] [blame] | 108 | then |
Milton Miller | 06ccb1a | 2016-02-05 13:04:29 -0600 | [diff] [blame] | 109 | debug_takeover "Debug initial shell requested by command line." |
Milton D. Miller II | d89d5e0 | 2016-01-20 14:57:54 -0600 | [diff] [blame] | 110 | fi |
| 111 | |
Milton Miller | 0b0d5fe | 2016-02-23 21:48:48 -0600 | [diff] [blame] | 112 | # If there are images in root move them to run/initramfs/ now. |
| 113 | imagebasename=${image##*/} |
| 114 | if test -n "${imagebasename}" -a "x$flash_images_before_init" = xy && |
| 115 | ls /${imagebasename}* > /dev/null 2>&1 |
| 116 | then |
| 117 | echo "Pending flash updates found." |
| 118 | mv /${imagebasename}* ${image%$imagebasename} |
| 119 | fi |
| 120 | |
| 121 | if grep -w clean-rwfs-filesystem /proc/cmdline |
| 122 | then |
| 123 | echo "Cleaning of read-write overlay filesystem requested." |
| 124 | touch $trigger |
| 125 | fi |
| 126 | |
| 127 | if test "x$force_rwfst_jffs2" = xy -a $rwfst != jffs2 -a ! -f $trigger |
| 128 | then |
| 129 | echo "Converting read-write overlay filesystem to jffs2 forced." |
| 130 | touch $trigger |
| 131 | fi |
| 132 | |
| 133 | if ls $image* > /dev/null 2>&1 |
| 134 | then |
| 135 | if ! test -x /update |
| 136 | then |
| 137 | debug_takeover "Flash update requested but /update missing!" |
| 138 | elif test -f $trigger -a ! -s $trigger |
| 139 | then |
| 140 | echo "Saving selected files from read-write overlay filesystem." |
| 141 | /update && rm -f $image* |
| 142 | echo "Clearing read-write overlay filesystem." |
| 143 | flash_eraseall /dev/$rwfs |
| 144 | echo "Restoring saved files to read-write overlay filesystem." |
| 145 | touch $trigger |
| 146 | /update |
| 147 | rm -rf /save $trigger |
| 148 | else |
| 149 | /update && rm -f $image* |
| 150 | fi |
| 151 | |
| 152 | rwfst=$(probe_fs_type $rwdev) |
| 153 | fsck=$fsckbase$rwfst |
| 154 | fi |
| 155 | |
Milton Miller | b72114c | 2016-02-06 16:05:06 -0600 | [diff] [blame] | 156 | mount $rodev $rodir -t $rofst -o $roopts |
| 157 | |
| 158 | if test -x $rodir$fsck |
| 159 | then |
| 160 | for fs in $fslist |
| 161 | do |
| 162 | mount --bind $fs $rodir/$fs |
| 163 | done |
| 164 | chroot $rodir $fsck $fsckopts $rwdev |
| 165 | rc=$? |
| 166 | for fs in $fslist |
| 167 | do |
| 168 | umount $rodir/$fs |
| 169 | done |
| 170 | if test $rc -gt 1 |
| 171 | then |
| 172 | debug_takeover "fsck of read-write fs on $rwdev failed (rc=$rc)" |
| 173 | fi |
Milton Miller | f81c108 | 2016-02-23 21:00:06 -0600 | [diff] [blame] | 174 | elif test $fsck != /sbin/fsck.jffs2 |
| 175 | then |
Milton Miller | b72114c | 2016-02-06 16:05:06 -0600 | [diff] [blame] | 176 | echo "No '$fsck' in read only fs, skipping fsck." |
| 177 | fi |
| 178 | |
Milton Miller | 06ccb1a | 2016-02-05 13:04:29 -0600 | [diff] [blame] | 179 | if ! mount $rwdev $rwdir -t $rwfst -o $rwopts |
Milton D. Miller II | d89d5e0 | 2016-01-20 14:57:54 -0600 | [diff] [blame] | 180 | then |
Milton Miller | 06ccb1a | 2016-02-05 13:04:29 -0600 | [diff] [blame] | 181 | msg="$(cat)" << HERE |
| 182 | |
| 183 | Mounting read-write $rwdev filesystem failed. Please fix and run |
Andrew Jeffery | ed5fb27 | 2016-02-17 17:19:58 +1030 | [diff] [blame] | 184 | mount $rwdev $rwdir -t $rwfst -o $rwopts |
Milton Miller | 06ccb1a | 2016-02-05 13:04:29 -0600 | [diff] [blame] | 185 | to to continue, or do change nothing to run from RAM for this boot. |
| 186 | HERE |
| 187 | debug_takeover "$msg" |
Milton D. Miller II | d89d5e0 | 2016-01-20 14:57:54 -0600 | [diff] [blame] | 188 | fi |
| 189 | |
Milton D. Miller II | d89d5e0 | 2016-01-20 14:57:54 -0600 | [diff] [blame] | 190 | rm -rf $work |
Milton Miller | b72114c | 2016-02-06 16:05:06 -0600 | [diff] [blame] | 191 | mkdir -p $upper $work |
Milton D. Miller II | d89d5e0 | 2016-01-20 14:57:54 -0600 | [diff] [blame] | 192 | |
| 193 | mount -t overlay -o lowerdir=$rodir,upperdir=$upper,workdir=$work cow /root |
| 194 | |
Milton Miller | 06ccb1a | 2016-02-05 13:04:29 -0600 | [diff] [blame] | 195 | while ! chroot /root /bin/sh -c "test -x '$init' -a -s '$init'" |
| 196 | do |
| 197 | msg="$(cat)" << HERE |
| 198 | |
| 199 | Unable to confirm /sbin/init is an executable non-empty file |
| 200 | in merged file system mounted at /root. |
| 201 | |
| 202 | Change Root test failed! Invoking emergency shell. |
| 203 | HERE |
| 204 | debug_takeover "$msg" |
| 205 | done |
Milton D. Miller II | d89d5e0 | 2016-01-20 14:57:54 -0600 | [diff] [blame] | 206 | |
Milton Miller | 54d882e | 2016-02-05 12:07:53 -0600 | [diff] [blame] | 207 | for f in $fslist |
Milton D. Miller II | d89d5e0 | 2016-01-20 14:57:54 -0600 | [diff] [blame] | 208 | do |
| 209 | mount --move $f root/$f |
| 210 | done |
| 211 | |
Milton Miller | 54d882e | 2016-02-05 12:07:53 -0600 | [diff] [blame] | 212 | # switch_root /root $init |
| 213 | exec chroot /root $init |
Milton D. Miller II | d89d5e0 | 2016-01-20 14:57:54 -0600 | [diff] [blame] | 214 | |