1#!/bin/bash
2#
3# To call this script, make sure mke2fs is somewhere in PATH
4
5function usage() {
6cat<<EOT
7Usage:
8mkuserimg.sh [-s] SRC_DIR OUTPUT_FILE EXT_VARIANT MOUNT_POINT SIZE [-j <journal_size>]
9             [-T TIMESTAMP] [-C FS_CONFIG] [-D PRODUCT_OUT] [-B BLOCK_LIST_FILE]
10             [-d BASE_ALLOC_FILE_IN ] [-A BASE_ALLOC_FILE_OUT ] [-L LABEL]
11             [-i INODES ] [-e ERASE_BLOCK_SIZE] [-o FLASH_BLOCK_SIZE] [FILE_CONTEXTS]
12EOT
13}
14
15BLOCKSIZE=4096
16
17MKE2FS_OPTS=""
18MKE2FS_EXTENDED_OPTS=""
19E2FSDROID_OPTS=""
20
21if [ "$1" = "-s" ]; then
22  MKE2FS_EXTENDED_OPTS+="android_sparse"
23  shift
24else
25  E2FSDROID_OPTS+="-e"
26fi
27
28if [ $# -lt 5 ]; then
29  usage
30  exit 1
31fi
32
33SRC_DIR=$1
34if [ ! -d $SRC_DIR ]; then
35  echo "Can not find directory $SRC_DIR!"
36  exit 2
37fi
38
39OUTPUT_FILE=$2
40EXT_VARIANT=$3
41MOUNT_POINT=$4
42SIZE=$5
43shift; shift; shift; shift; shift
44
45if [ "$1" = "-j" ]; then
46  if [ "$2" = "0" ]; then
47    MKE2FS_OPTS+="-O ^has_journal"
48  else
49    MKE2FS_OPTS+="-J size=$2"
50  fi
51  shift; shift
52fi
53
54if [[ "$1" == "-T" ]]; then
55  E2FSDROID_OPTS+=" -T $2"
56  shift; shift
57fi
58
59if [[ "$1" == "-C" ]]; then
60  E2FSDROID_OPTS+=" -C $2"
61  shift; shift
62fi
63
64if [[ "$1" == "-D" ]]; then
65  E2FSDROID_OPTS+=" -p $2"
66  shift; shift
67fi
68
69if [[ "$1" == "-B" ]]; then
70  E2FSDROID_OPTS+=" -B $2"
71  shift; shift
72fi
73
74if [[ "$1" == "-d" ]]; then
75  E2FSDROID_OPTS+=" -d $2"
76  shift; shift
77fi
78
79if [[ "$1" == "-A" ]]; then
80  E2FSDROID_OPTS+=" -D $2"
81  shift; shift
82fi
83
84if [[ "$1" == "-L" ]]; then
85  MKE2FS_OPTS+=" -L $2"
86  shift; shift
87fi
88
89if [[ "$1" == "-i" ]]; then
90  MKE2FS_OPTS+=" -N $2"
91  shift; shift
92fi
93
94if [[ "$1" == "-e" ]]; then
95  if [[ $MKE2FS_EXTENDED_OPTS ]]; then
96    MKE2FS_EXTENDED_OPTS+=","
97  fi
98  MKE2FS_EXTENDED_OPTS+="stripe_width=$(($2/BLOCKSIZE))"
99  shift; shift
100fi
101
102if [[ "$1" == "-o" ]]; then
103  if [[ $MKE2FS_EXTENDED_OPTS ]]; then
104    MKE2FS_EXTENDED_OPTS+=","
105  fi
106  # stride should be the max of 8kb and the logical block size
107  MKE2FS_EXTENDED_OPTS+="stride=$((($2 > 8192 ? $2 : 8192) / BLOCKSIZE))"
108  shift; shift
109fi
110
111if [[ $MKE2FS_EXTENDED_OPTS ]]; then
112  MKE2FS_OPTS+=" -E $MKE2FS_EXTENDED_OPTS"
113fi
114
115if [[ $1 ]]; then
116  E2FSDROID_OPTS+=" -S $1"
117fi
118
119case $EXT_VARIANT in
120  ext4) ;;
121  *) echo "Only ext4 is supported!"; exit 3 ;;
122esac
123
124if [ -z $MOUNT_POINT ]; then
125  echo "Mount point is required"
126  exit 2
127fi
128
129if [[ ${MOUNT_POINT:0:1} != "/" ]]; then
130  MOUNT_POINT="/"$MOUNT_POINT
131fi
132
133if [ -z $SIZE ]; then
134  echo "Need size of filesystem"
135  exit 2
136fi
137
138# Round down the filesystem length to be a multiple of the block size
139SIZE=$((SIZE / BLOCKSIZE))
140
141# truncate output file since mke2fs will keep verity section in existing file
142cat /dev/null >$OUTPUT_FILE
143
144MAKE_EXT4FS_CMD="mke2fs $MKE2FS_OPTS -t $EXT_VARIANT -b $BLOCKSIZE $OUTPUT_FILE $SIZE"
145echo $MAKE_EXT4FS_CMD
146MKE2FS_CONFIG=./system/extras/ext4_utils/mke2fs.conf $MAKE_EXT4FS_CMD
147if [ $? -ne 0 ]; then
148  exit 4
149fi
150
151E2FSDROID_CMD="e2fsdroid $E2FSDROID_OPTS -f $SRC_DIR -a $MOUNT_POINT $OUTPUT_FILE"
152echo $E2FSDROID_CMD
153$E2FSDROID_CMD
154if [ $? -ne 0 ]; then
155  rm -f $OUTPUT_FILE
156  exit 4
157fi
158