projects/linux·blob master · refs

250 lines · 5.8 KiB

1#!/bin/sh
2# Copyright (C) Martin Schlemmer <[email protected]>
3# Copyright (C) 2006 Sam Ravnborg <[email protected]>
4#
5# Released under the terms of the GNU GPL
6#
7# Generate a cpio packed initramfs. It uses gen_init_cpio to generate
8# the cpio archive.
9# This script assumes that gen_init_cpio is located in usr/ directory
10
11# error out on errors
12set -e
13
14usage() {
15cat << EOF
16Usage:
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_cpio
19 -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 a
22 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 a
25 directory. "squash" forces all files to gid 0.
26 -d <date> Use date for all file mtime values
27 <cpio_source> File list or directory for cpio archive.
28 If <cpio_source> is a .cpio file it will be used
29 as direct input to initramfs.
30
31All options except -o and -l may be repeated and are interpreted
32sequentially and immediately. -u and -g states are preserved across
33<cpio_source> options so an explicit "-u 0 -g 0" is required
34to reset the root/group mapping.
35EOF
36}
37
38# awk style field access
39# $1 - field number; rest is argument string
40field() {
41 shift $1 ; echo $1
42}
43
44filetype() {
45 local argv1="$1"
46
47 # symlink test must come before file test
48 if [ -L "${argv1}" ]; then
49 echo "slink"
50 elif [ -f "${argv1}" ]; then
51 echo "file"
52 elif [ -d "${argv1}" ]; then
53 echo "dir"
54 elif [ -b "${argv1}" -o -c "${argv1}" ]; then
55 echo "nod"
56 elif [ -p "${argv1}" ]; then
57 echo "pipe"
58 elif [ -S "${argv1}" ]; then
59 echo "sock"
60 else
61 echo "invalid"
62 fi
63 return 0
64}
65
66print_mtime() {
67 local my_mtime="0"
68
69 if [ -e "$1" ]; then
70 my_mtime=$(find "$1" -printf "%T@\n" | sort -r | head -n 1)
71 fi
72
73 echo "# Last modified: ${my_mtime}" >> $cpio_list
74 echo "" >> $cpio_list
75}
76
77list_parse() {
78 if [ -z "$dep_list" -o -L "$1" ]; then
79 return
80 fi
81 echo "$1" | sed 's/:/\\:/g; s/$/ \\/' >> $dep_list
82}
83
84# for each file print a line in following format
85# <filetype> <name> <path to file> <octal mode> <uid> <gid>
86# for links, devices etc the format differs. See gen_init_cpio for details
87parse() {
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 necessary
97 [ "$root_uid" = "squash" ] && uid=0 || [ "$uid" -eq "$root_uid" ] && uid=0
98 [ "$root_gid" = "squash" ] && gid=0 || [ "$gid" -eq "$root_gid" ] && gid=0
99 local str="${mode} ${uid} ${gid}"
100
101 [ "${ftype}" = "invalid" ] && return 0
102 [ "${location}" = "${srcdir}" ] && return 0
103
104 case "${ftype}" in
105 "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 esac
126
127 echo "${str}" >> $cpio_list
128
129 return 0
130}
131
132unknown_option() {
133 printf "ERROR: unknown option \"$arg\"\n" >&2
134 printf "If the filename validly begins with '-', " >&2
135 printf "then it must be prefixed\n" >&2
136 printf "by './' so that it won't be interpreted as an option." >&2
137 printf "\n" >&2
138 usage >&2
139 exit 1
140}
141
142header() {
143 printf "\n#####################\n# $1\n" >> $cpio_list
144}
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 empty
154 if [ "$(echo "${dirlist}" | wc -l)" -gt 1 ]; then
155 print_mtime "$1"
156
157 echo "${dirlist}" | \
158 while read x; do
159 list_parse $x
160 parse $x
161 done
162 fi
163}
164
165input_file() {
166 source="$1"
167 if [ -f "$1" ]; then
168 # If a regular file is specified, assume it is in
169 # gen_init_cpio format
170 header "$1"
171 print_mtime "$1" >> $cpio_list
172 cat "$1" >> $cpio_list
173 if [ -n "$dep_list" ]; then
174 echo "$1 \\" >> $dep_list
175 cat "$1" | while read type dir file perm ; do
176 if [ "$type" = "file" ]; then
177 echo "$file \\" >> $dep_list
178 fi
179 done
180 fi
181 elif [ -d "$1" ]; then
182 # If a directory is specified then add all files in it to fs
183 dir_filelist "$1"
184 else
185 echo " ${prog}: Cannot open '$1'" >&2
186 exit 1
187 fi
188}
189
190prog=$0
191root_uid=0
192root_gid=0
193dep_list=
194timestamp=
195cpio_list=$(mktemp ${TMPDIR:-/tmp}/cpiolist.XXXXXX)
196# gen_init_cpio writes to stdout by default
197output=""
198
199trap "rm -f $cpio_list" EXIT
200
201while [ $# -gt 0 ]; do
202 arg="$1"
203 shift
204 case "$arg" in
205 "-l") # files included in initramfs - used by kbuild
206 dep_list="$1"
207 echo "deps_initramfs := \\" > $dep_list
208 shift
209 ;;
210 "-o") # generate cpio image named $1
211 output="-o $1"
212 shift
213 ;;
214 "-u") # map $1 to uid=0 (root)
215 root_uid="$1"
216 [ "$root_uid" = "-1" ] && root_uid=$(id -u || echo 0)
217 shift
218 ;;
219 "-g") # map $1 to gid=0 (root)
220 root_gid="$1"
221 [ "$root_gid" = "-1" ] && root_gid=$(id -g || echo 0)
222 shift
223 ;;
224 "-d") # date for file mtimes
225 timestamp="$(date -d"$1" +%s || :)"
226 if test -n "$timestamp"; then
227 timestamp="-t $timestamp"
228 fi
229 shift
230 ;;
231 "-h")
232 usage
233 exit 0
234 ;;
235 *)
236 case "$arg" in
237 "-"*)
238 unknown_option
239 ;;
240 *) # input file/dir - process it
241 input_file "$arg"
242 ;;
243 esac
244 ;;
245 esac
246done
247
248# If output_file is set we will generate cpio archive
249# we are careful to delete tmp files
250usr/gen_init_cpio $output $timestamp $cpio_list