Gunnar Mills | 6b5b27e | 2017-03-01 21:37:01 -0600 | [diff] [blame] | 1 | #!/bin/bash |
Gunnar Mills | f8f1130 | 2017-03-08 13:11:33 -0600 | [diff] [blame] | 2 | set -eo pipefail |
Gunnar Mills | e822190 | 2017-02-03 15:59:06 -0600 | [diff] [blame] | 3 | |
Saqib Khan | 93433f0 | 2017-03-23 22:24:27 -0500 | [diff] [blame] | 4 | help=$'Generate Tarball with SquashFS image and MANIFEST Script |
Gunnar Mills | e822190 | 2017-02-03 15:59:06 -0600 | [diff] [blame] | 5 | |
| 6 | Generates a SquashFS image from the PNOR image |
Saqib Khan | 93433f0 | 2017-03-23 22:24:27 -0500 | [diff] [blame] | 7 | Creates a MANIFEST for image verification |
| 8 | Packages the SquashFS image and MANIFEST together in a tarball |
Gunnar Mills | e822190 | 2017-02-03 15:59:06 -0600 | [diff] [blame] | 9 | |
Gunnar Mills | c15b02d | 2017-03-03 10:28:37 -0600 | [diff] [blame] | 10 | usage: generate-squashfs [OPTION] <PNOR FILE>... |
Gunnar Mills | e822190 | 2017-02-03 15:59:06 -0600 | [diff] [blame] | 11 | |
| 12 | Options: |
Gunnar Mills | 37751f9 | 2017-02-07 21:05:01 -0600 | [diff] [blame] | 13 | -f, --file <file> Specify destination file. Defaults to |
Michael Tritz | 8f39673 | 2017-06-09 10:55:35 -0500 | [diff] [blame] | 14 | `pwd`/<PNOR FILE>.pnor.squashfs.tar if unspecified. |
| 15 | (For example, "generate-squashfs my.pnor" would |
| 16 | generate `pwd`/my.pnor.squashfs.tar output.) |
Gunnar Mills | e822190 | 2017-02-03 15:59:06 -0600 | [diff] [blame] | 17 | -h, --help Display this help text and exit. |
| 18 | ' |
| 19 | |
Gunnar Mills | d4908a4 | 2017-03-07 08:25:59 -0600 | [diff] [blame] | 20 | let ffs_entry_size=128 |
| 21 | let miscflags_offset=112 |
Michael Tritz | 8f39673 | 2017-06-09 10:55:35 -0500 | [diff] [blame] | 22 | outfile="" |
Gunnar Mills | 6b5b27e | 2017-03-01 21:37:01 -0600 | [diff] [blame] | 23 | declare -a partitions=() |
| 24 | tocfile="pnor.toc" |
Gunnar Mills | 37751f9 | 2017-02-07 21:05:01 -0600 | [diff] [blame] | 25 | |
Gunnar Mills | e822190 | 2017-02-03 15:59:06 -0600 | [diff] [blame] | 26 | while [[ $# -gt 0 ]]; do |
| 27 | key="$1" |
| 28 | case $key in |
Gunnar Mills | 37751f9 | 2017-02-07 21:05:01 -0600 | [diff] [blame] | 29 | -f|--file) |
| 30 | outfile="$2" |
| 31 | shift 2 |
| 32 | ;; |
Gunnar Mills | e822190 | 2017-02-03 15:59:06 -0600 | [diff] [blame] | 33 | -h|--help) |
| 34 | echo "$help" |
| 35 | exit |
| 36 | ;; |
| 37 | *) |
Gunnar Mills | c15b02d | 2017-03-03 10:28:37 -0600 | [diff] [blame] | 38 | pnorfile="$1" |
| 39 | shift 1 |
Gunnar Mills | e822190 | 2017-02-03 15:59:06 -0600 | [diff] [blame] | 40 | ;; |
| 41 | esac |
| 42 | done |
Gunnar Mills | 18f7cdb | 2017-02-07 16:44:19 -0600 | [diff] [blame] | 43 | |
Gunnar Mills | c15b02d | 2017-03-03 10:28:37 -0600 | [diff] [blame] | 44 | if [ ! -f "${pnorfile}" ]; then |
| 45 | echo "Please enter a valid PNOR file." |
| 46 | echo "$help" |
| 47 | exit 1 |
| 48 | fi |
| 49 | |
Michael Tritz | 8f39673 | 2017-06-09 10:55:35 -0500 | [diff] [blame] | 50 | if [[ -z $outfile ]]; then |
| 51 | if [[ ${pnorfile##*.} == "pnor" ]]; then |
| 52 | outfile=`pwd`/${pnorfile##*/}.squashfs.tar |
| 53 | else |
| 54 | outfile=`pwd`/${pnorfile##*/}.pnor.squashfs.tar |
| 55 | fi |
| 56 | else |
| 57 | if [[ $outfile != /* ]]; then |
| 58 | outfile=`pwd`/$outfile |
| 59 | fi |
| 60 | fi |
| 61 | |
Gunnar Mills | f8f1130 | 2017-03-08 13:11:33 -0600 | [diff] [blame] | 62 | scratch_dir=`mktemp -d` |
Gunnar Mills | 6b5b27e | 2017-03-01 21:37:01 -0600 | [diff] [blame] | 63 | |
| 64 | echo "Parsing PNOR TOC..." |
Gunnar Mills | d4908a4 | 2017-03-07 08:25:59 -0600 | [diff] [blame] | 65 | |
| 66 | # Needed to get the READONLY and PRESERVED flags |
| 67 | # TODO: Get READONLY and PRESERVED flags from pflash instead. |
Gunnar Mills | f8f1130 | 2017-03-08 13:11:33 -0600 | [diff] [blame] | 68 | pflash --partition=part --read=${scratch_dir}/part -F ${pnorfile} |
Gunnar Mills | 2f27b74 | 2017-03-20 13:46:30 -0500 | [diff] [blame] | 69 | pflash --partition=VERSION --read=${scratch_dir}/VERSION -F ${pnorfile} |
Gunnar Mills | 6b5b27e | 2017-03-01 21:37:01 -0600 | [diff] [blame] | 70 | { |
Gunnar Mills | 2f27b74 | 2017-03-20 13:46:30 -0500 | [diff] [blame] | 71 | version=$(head -n 1 ${scratch_dir}/VERSION) |
| 72 | echo "version=$version" |
Saqib Khan | 029b825 | 2017-03-22 21:38:07 -0500 | [diff] [blame] | 73 | extended_version=$(echo $(tail -n +2 ${scratch_dir}/VERSION)|tr ' ' ',') |
| 74 | echo "extended_version=$extended_version" |
Gunnar Mills | 6b5b27e | 2017-03-01 21:37:01 -0600 | [diff] [blame] | 75 | while read line; do |
| 76 | if [[ $line == "ID="* ]]; then |
Gunnar Mills | e0ed302 | 2017-03-02 16:16:19 -0600 | [diff] [blame] | 77 | # This line looks like |
| 78 | # "ID=05 MVPD 000d9000..00169000 (actual=00090000) [ECC]" |
Gunnar Mills | 6b5b27e | 2017-03-01 21:37:01 -0600 | [diff] [blame] | 79 | read -r -a fields <<< "$line" |
Gunnar Mills | e0ed302 | 2017-03-02 16:16:19 -0600 | [diff] [blame] | 80 | |
| 81 | # Get any flags attached to end (e.g. [ECC]) |
| 82 | flags="" |
| 83 | for flag in "${fields[@]:4}" |
| 84 | do |
| 85 | flags+=",${flag//[\[\]]/}" |
| 86 | done |
| 87 | |
Gunnar Mills | d4908a4 | 2017-03-07 08:25:59 -0600 | [diff] [blame] | 88 | id=${fields[0]##*=} |
| 89 | offset=$((${ffs_entry_size} * 10#${id} + ${miscflags_offset})) |
| 90 | miscflags=0x$(xxd -p -l 0x1 -seek ${offset} ${scratch_dir}/part) |
| 91 | if ((miscflags & 0x80)); then |
| 92 | flags+=",PRESERVED" |
Adriana Kobylak | 9c86606 | 2017-04-13 16:12:32 -0500 | [diff] [blame] | 93 | elif ((miscflags & 0x40)); then |
Gunnar Mills | d4908a4 | 2017-03-07 08:25:59 -0600 | [diff] [blame] | 94 | flags+=",READONLY" |
Adriana Kobylak | 9c86606 | 2017-04-13 16:12:32 -0500 | [diff] [blame] | 95 | else |
| 96 | flags+=",READWRITE" |
Gunnar Mills | d4908a4 | 2017-03-07 08:25:59 -0600 | [diff] [blame] | 97 | fi |
| 98 | |
Gunnar Mills | e0ed302 | 2017-03-02 16:16:19 -0600 | [diff] [blame] | 99 | # Need the partition ID, name, start location, end location, and flags |
Gunnar Mills | d4908a4 | 2017-03-07 08:25:59 -0600 | [diff] [blame] | 100 | echo "partition${id}=${fields[1]},${fields[2]/../,}${flags}" |
| 101 | |
Gunnar Mills | 6b5b27e | 2017-03-01 21:37:01 -0600 | [diff] [blame] | 102 | # Save the partition name |
| 103 | partitions+=(${fields[1]}) |
| 104 | fi |
Gunnar Mills | 2f27b74 | 2017-03-20 13:46:30 -0500 | [diff] [blame] | 105 | # Don't need part, version, or backup_part partitions |
| 106 | done < <(pflash --info -F ${pnorfile} | grep -v -i "part\|version") |
Gunnar Mills | 6b5b27e | 2017-03-01 21:37:01 -0600 | [diff] [blame] | 107 | } > ${scratch_dir}/${tocfile} |
Gunnar Mills | 37751f9 | 2017-02-07 21:05:01 -0600 | [diff] [blame] | 108 | |
Gunnar Mills | 18f7cdb | 2017-02-07 16:44:19 -0600 | [diff] [blame] | 109 | for partition in "${partitions[@]}"; do |
| 110 | echo "Reading ${partition}..." |
Gunnar Mills | f8f1130 | 2017-03-08 13:11:33 -0600 | [diff] [blame] | 111 | pflash --partition=${partition} \ |
| 112 | --read=${scratch_dir}/${partition} \ |
| 113 | -F ${pnorfile} |
Gunnar Mills | 18f7cdb | 2017-02-07 16:44:19 -0600 | [diff] [blame] | 114 | done |
Gunnar Mills | 37751f9 | 2017-02-07 21:05:01 -0600 | [diff] [blame] | 115 | |
| 116 | echo "Creating SquashFS image..." |
Gunnar Mills | 37751f9 | 2017-02-07 21:05:01 -0600 | [diff] [blame] | 117 | cd "${scratch_dir}" |
Adriana Kobylak | 9c86606 | 2017-04-13 16:12:32 -0500 | [diff] [blame] | 118 | mksquashfs ${tocfile} ${partitions[*]} pnor.xz.squashfs |
Gunnar Mills | 37751f9 | 2017-02-07 21:05:01 -0600 | [diff] [blame] | 119 | |
Saqib Khan | 93433f0 | 2017-03-23 22:24:27 -0500 | [diff] [blame] | 120 | echo "Creating MANIFEST for the image" |
Adriana Kobylak | 9c86606 | 2017-04-13 16:12:32 -0500 | [diff] [blame] | 121 | manifest_location="MANIFEST" |
Saqib Khan | 7254f0e | 2017-04-10 21:45:37 -0500 | [diff] [blame] | 122 | echo -e "purpose=host\nversion=$version\n\ |
| 123 | extended_version=$extended_version" >> $manifest_location |
Saqib Khan | 93433f0 | 2017-03-23 22:24:27 -0500 | [diff] [blame] | 124 | |
| 125 | echo "Generating tarball to contain the SquashFS image and its MANIFEST" |
Adriana Kobylak | 9c86606 | 2017-04-13 16:12:32 -0500 | [diff] [blame] | 126 | tar -cvf $outfile $manifest_location pnor.xz.squashfs |
Saqib Khan | 93433f0 | 2017-03-23 22:24:27 -0500 | [diff] [blame] | 127 | |
| 128 | echo "SquashFSTarball at ${outfile}" |
Gunnar Mills | 37751f9 | 2017-02-07 21:05:01 -0600 | [diff] [blame] | 129 | rm -r "${scratch_dir}" |