250 lines · bash
1#!/bin/sh2# Copyright (C) Martin Schlemmer <azarah@nosferatu.za.org>3# Copyright (C) 2006 Sam Ravnborg <sam@ravnborg.org>4#5# Released under the terms of the GNU GPL6#7# Generate a cpio packed initramfs. It uses gen_init_cpio to generate8# the cpio archive.9# This script assumes that gen_init_cpio is located in usr/ directory10 11# error out on errors12set -e13 14usage() {15cat << EOF16Usage:17$0 [-o <file>] [-l <dep_list>] [-u <uid>] [-g <gid>] {-d | <cpio_source>} ...18 -o <file> Create initramfs file named <file> by using gen_init_cpio19 -l <dep_list> Create dependency list named <dep_list>20 -u <uid> User ID to map to user ID 0 (root).21 <uid> is only meaningful if <cpio_source> is a22 directory. "squash" forces all files to uid 0.23 -g <gid> Group ID to map to group ID 0 (root).24 <gid> is only meaningful if <cpio_source> is a25 directory. "squash" forces all files to gid 0.26 -d <date> Use date for all file mtime values27 <cpio_source> File list or directory for cpio archive.28 If <cpio_source> is a .cpio file it will be used29 as direct input to initramfs.30 31All options except -o and -l may be repeated and are interpreted32sequentially and immediately. -u and -g states are preserved across33<cpio_source> options so an explicit "-u 0 -g 0" is required34to reset the root/group mapping.35EOF36}37 38# awk style field access39# $1 - field number; rest is argument string40field() {41 shift $1 ; echo $142}43 44filetype() {45 local argv1="$1"46 47 # symlink test must come before file test48 if [ -L "${argv1}" ]; then49 echo "slink"50 elif [ -f "${argv1}" ]; then51 echo "file"52 elif [ -d "${argv1}" ]; then53 echo "dir"54 elif [ -b "${argv1}" -o -c "${argv1}" ]; then55 echo "nod"56 elif [ -p "${argv1}" ]; then57 echo "pipe"58 elif [ -S "${argv1}" ]; then59 echo "sock"60 else61 echo "invalid"62 fi63 return 064}65 66print_mtime() {67 local my_mtime="0"68 69 if [ -e "$1" ]; then70 my_mtime=$(find "$1" -printf "%T@\n" | sort -r | head -n 1)71 fi72 73 echo "# Last modified: ${my_mtime}" >> $cpio_list74 echo "" >> $cpio_list75}76 77list_parse() {78 if [ -z "$dep_list" -o -L "$1" ]; then79 return80 fi81 echo "$1" | sed 's/:/\\:/g; s/$/ \\/' >> $dep_list82}83 84# for each file print a line in following format85# <filetype> <name> <path to file> <octal mode> <uid> <gid>86# for links, devices etc the format differs. See gen_init_cpio for details87parse() {88 local location="$1"89 local name="/${location#${srcdir}}"90 # change '//' into '/'91 name=$(echo "$name" | sed -e 's://*:/:g')92 local mode="$2"93 local uid="$3"94 local gid="$4"95 local ftype=$(filetype "${location}")96 # remap uid/gid to 0 if necessary97 [ "$root_uid" = "squash" ] && uid=0 || [ "$uid" -eq "$root_uid" ] && uid=098 [ "$root_gid" = "squash" ] && gid=0 || [ "$gid" -eq "$root_gid" ] && gid=099 local str="${mode} ${uid} ${gid}"100 101 [ "${ftype}" = "invalid" ] && return 0102 [ "${location}" = "${srcdir}" ] && return 0103 104 case "${ftype}" in105 "file")106 str="${ftype} ${name} ${location} ${str}"107 ;;108 "nod")109 local dev="`LC_ALL=C ls -l "${location}"`"110 local maj=`field 5 ${dev}`111 local min=`field 6 ${dev}`112 maj=${maj%,}113 114 [ -b "${location}" ] && dev="b" || dev="c"115 116 str="${ftype} ${name} ${str} ${dev} ${maj} ${min}"117 ;;118 "slink")119 local target=`readlink "${location}"`120 str="${ftype} ${name} ${target} ${str}"121 ;;122 *)123 str="${ftype} ${name} ${str}"124 ;;125 esac126 127 echo "${str}" >> $cpio_list128 129 return 0130}131 132unknown_option() {133 printf "ERROR: unknown option \"$arg\"\n" >&2134 printf "If the filename validly begins with '-', " >&2135 printf "then it must be prefixed\n" >&2136 printf "by './' so that it won't be interpreted as an option." >&2137 printf "\n" >&2138 usage >&2139 exit 1140}141 142header() {143 printf "\n#####################\n# $1\n" >> $cpio_list144}145 146# process one directory (incl sub-directories)147dir_filelist() {148 header "$1"149 150 srcdir=$(echo "$1" | sed -e 's://*:/:g')151 dirlist=$(find "${srcdir}" -printf "%p %m %U %G\n" | LC_ALL=C sort)152 153 # If $dirlist is only one line, then the directory is empty154 if [ "$(echo "${dirlist}" | wc -l)" -gt 1 ]; then155 print_mtime "$1"156 157 echo "${dirlist}" | \158 while read x; do159 list_parse $x160 parse $x161 done162 fi163}164 165input_file() {166 source="$1"167 if [ -f "$1" ]; then168 # If a regular file is specified, assume it is in169 # gen_init_cpio format170 header "$1"171 print_mtime "$1" >> $cpio_list172 cat "$1" >> $cpio_list173 if [ -n "$dep_list" ]; then174 echo "$1 \\" >> $dep_list175 cat "$1" | while read type dir file perm ; do176 if [ "$type" = "file" ]; then177 echo "$file \\" >> $dep_list178 fi179 done180 fi181 elif [ -d "$1" ]; then182 # If a directory is specified then add all files in it to fs183 dir_filelist "$1"184 else185 echo " ${prog}: Cannot open '$1'" >&2186 exit 1187 fi188}189 190prog=$0191root_uid=0192root_gid=0193dep_list=194timestamp=195cpio_list=$(mktemp ${TMPDIR:-/tmp}/cpiolist.XXXXXX)196output="/dev/stdout"197 198trap "rm -f $cpio_list" EXIT199 200while [ $# -gt 0 ]; do201 arg="$1"202 shift203 case "$arg" in204 "-l") # files included in initramfs - used by kbuild205 dep_list="$1"206 echo "deps_initramfs := \\" > $dep_list207 shift208 ;;209 "-o") # generate cpio image named $1210 output="$1"211 shift212 ;;213 "-u") # map $1 to uid=0 (root)214 root_uid="$1"215 [ "$root_uid" = "-1" ] && root_uid=$(id -u || echo 0)216 shift217 ;;218 "-g") # map $1 to gid=0 (root)219 root_gid="$1"220 [ "$root_gid" = "-1" ] && root_gid=$(id -g || echo 0)221 shift222 ;;223 "-d") # date for file mtimes224 timestamp="$(date -d"$1" +%s || :)"225 if test -n "$timestamp"; then226 timestamp="-t $timestamp"227 fi228 shift229 ;;230 "-h")231 usage232 exit 0233 ;;234 *)235 case "$arg" in236 "-"*)237 unknown_option238 ;;239 *) # input file/dir - process it240 input_file "$arg"241 ;;242 esac243 ;;244 esac245done246 247# If output_file is set we will generate cpio archive248# we are careful to delete tmp files249usr/gen_init_cpio $timestamp $cpio_list > $output250