1#!/bin/bash
2#
3# Copyright (c) 2021 Huawei Device Co., Ltd.
4# Licensed under the Apache License, Version 2.0 (the "License");
5# you may not use this file except in compliance with the License.
6# You may obtain a copy of the License at
7#
8#     http://www.apache.org/licenses/LICENSE-2.0
9#
10# Unless required by applicable law or agreed to in writing, software
11# distributed under the License is distributed on an "AS IS" BASIS,
12# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13# See the License for the specific language governing permissions and
14# limitations under the License.
15set -e
16
17system=$(uname -s)
18ROOTFS_DIR=$1
19FSTYPE=$2
20ROOTFS_IMG=${ROOTFS_DIR}"_"${FSTYPE}".img"
21JFFS2_TOOL=mkfs.jffs2
22WIN_JFFS2_TOOL=mkfs.jffs2.exe
23VFAT_TOOL=mkfs.vfat
24MCOPY_TOOL=mcopy
25EXT4_TOOL=mkfs.ext4
26EXT4_SH=$(dirname $(readlink -f "$0"))/../../../third_party/e2fsprogs/e2fsprogs/contrib/populate-extfs.sh
27INSTALL_E2FSPROGS_SH=$(dirname $(readlink -f "$0"))/../../../third_party/e2fsprogs/install.sh
28INSTALL_E2FSPROGS_PATH=$(dirname $(readlink -f "$0"))/../../../third_party/e2fsprogs
29
30tool_check() {
31local ret='0'
32command -v "$1" >/dev/null 2>&1 || { local ret='1'; }
33if [ "$ret" -ne 0  ]; then
34    echo "$1 tool is not exit, please install it" >&2
35fi
36return 0
37}
38
39if [[ "${ROOTFS_DIR}" = *"rootfs" ]]; then
40    if [ -d "${ROOTFS_DIR}" ]; then
41        chmod -R 755 ${ROOTFS_DIR}
42    fi
43    if [ -f "${ROOTFS_DIR}/bin/init" ] && [ -f "${ROOTFS_DIR}/bin/shell" ]; then
44        chmod 700 ${ROOTFS_DIR}/bin/init 2> /dev/null
45        chmod 700 ${ROOTFS_DIR}/bin/shell 2> /dev/null
46    fi
47fi
48
49CONFIG_PATH=$(dirname $(readlink -f "$0"))/rootfs_linux.config
50
51if [ "${FSTYPE}" = "jffs2" ]; then
52    if [ "${system}" != "Linux" ] ; then
53        tool_check ${WIN_JFFS2_TOOL}
54        ${WIN_JFFS2_TOOL} -q -o ${ROOTFS_IMG} -d ${ROOTFS_DIR} --pagesize=4096
55    else
56        tool_check ${JFFS2_TOOL}
57        if [[ "${ROOTFS_DIR}" = *"rootfs" ]]; then
58            ${JFFS2_TOOL} -q -o ${ROOTFS_IMG} -d ${ROOTFS_DIR} --pagesize=4096 --devtable ${CONFIG_PATH}
59        else
60            ${JFFS2_TOOL} -q -o ${ROOTFS_IMG} -d ${ROOTFS_DIR} --pagesize=4096
61        fi
62    fi
63elif [ "${FSTYPE}" = "vfat" ]; then
64    if [ "${system}" != "Linux" ] ; then
65        echo "Unsupported fs type!" >&2
66    else
67        tool_check ${VFAT_TOOL}
68        tool_check ${MCOPY_TOOL}
69        BLK_SIZE=512
70        CLT_SIZE=2048
71        FAT_TAB_NUM=2
72        CLT_CNT=$(( ${CLT_SIZE} / ${BLK_SIZE} ))
73        if [ $# -eq 3 ]; then
74            IMG_SIZE=$3
75        else
76            FAT32_ITEM_SIZE=4
77            RESV_CNT=38
78            IMG_MIN_SIZE=1048576
79            DIR_SIZE=$(( $(echo $(du -s ${ROOTFS_DIR} | awk '{print $1}')) * 1024 ))
80            IMG_SIZE=$(( ${DIR_SIZE} / (1 - ${FAT_TAB_NUM} * ${FAT32_ITEM_SIZE} / ${CLT_SIZE}) + ${RESV_CNT} * ${BLK_SIZE}))
81            if [ ${IMG_SIZE} -le ${IMG_MIN_SIZE} ]; then
82                IMG_SIZE=${IMG_MIN_SIZE}
83            fi
84        fi
85        IMG_CNT=$(( (${IMG_SIZE} + ${BLK_SIZE} - 1) / ${BLK_SIZE} ))
86        echo mtools_skip_check=1 >> ~/.mtoolsrc
87        dd if=/dev/zero of=${ROOTFS_IMG} count=${IMG_CNT} bs=${BLK_SIZE}
88        ${VFAT_TOOL} ${ROOTFS_IMG} -s ${CLT_CNT} -f ${FAT_TAB_NUM} -S ${BLK_SIZE} > /dev/null
89        ${MCOPY_TOOL} -i ${ROOTFS_IMG} ${ROOTFS_DIR}/* -/ ::/
90    fi
91elif [ "${FSTYPE}" = "ext4" ]; then
92    if [ "${system}" != "Linux" ] ; then
93        echo "Unsupported fs type!" >&2
94    else
95        if [ $# -eq 3 ]; then
96            EMMC_ROOTFS_SIZE=$3
97        else
98            if [[ "${ROOTFS_DIR}" = *"rootfs" ]]; then
99                EMMC_ROOTFS_SIZE=50
100            else
101                EMMC_ROOTFS_SIZE=50
102            fi
103        fi
104        COUNT_SIZE="$(expr ${EMMC_ROOTFS_SIZE} \* 1024 \* 2)"
105        dd if=/dev/zero of=${ROOTFS_IMG} bs=512 count=${COUNT_SIZE}
106        ${EXT4_TOOL} ${ROOTFS_IMG}
107        ${INSTALL_E2FSPROGS_SH} ${INSTALL_E2FSPROGS_PATH}
108        ${EXT4_SH} ${ROOTFS_DIR} ${ROOTFS_IMG} ${CONFIG_PATH}
109    fi
110else
111    echo "Unsupported fs type!" >&2
112fi
113