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
22YAFFS2_TOOL=mkyaffs2image100
23WIN_JFFS2_TOOL=mkfs.jffs2.exe
24VFAT_TOOL=mkfs.vfat
25MCOPY_TOOL=mcopy
26CONFIG_PATH=$(dirname $(readlink -f "$0"))/rootfs_liteos.config
27
28tool_check() {
29local ret='0'
30command -v "$1" >/dev/null 2>&1 || { local ret='1'; }
31if [ "$ret" -ne 0  ]; then
32    echo "$1 tool is not exit, please install it" >&2
33fi
34return 0
35}
36
37if [[ "${ROOTFS_DIR}" = *"rootfs" ]]; then
38    if [ -d "${ROOTFS_DIR}" ]; then
39        chmod -R 755 ${ROOTFS_DIR}
40    fi
41    if [ -f "${ROOTFS_DIR}/bin/init" ] && [ -f "${ROOTFS_DIR}/bin/shell" ]; then
42        chmod 700 ${ROOTFS_DIR}/bin/init 2> /dev/null
43        chmod 700 ${ROOTFS_DIR}/bin/shell 2> /dev/null
44    fi
45fi
46
47if [ "${FSTYPE}" = "jffs2" ]; then
48    if [ "${system}" != "Linux" ]; then
49        tool_check ${WIN_JFFS2_TOOL}
50        ${WIN_JFFS2_TOOL} -q -o ${ROOTFS_IMG} -d ${ROOTFS_DIR} --pagesize=4096
51    else
52        tool_check ${JFFS2_TOOL}
53        if [[ "${ROOTFS_DIR}" = *"rootfs" ]]; then
54            if [ -f "${ROOTFS_DIR}/bin/foundation" ] && [ -f "${ROOTFS_DIR}/bin/apphilogcat" ]; then
55                ${JFFS2_TOOL} -q -o ${ROOTFS_IMG} -d ${ROOTFS_DIR} --pagesize=4096 --devtable ${CONFIG_PATH}
56            else
57                ${JFFS2_TOOL} -q -o ${ROOTFS_IMG} -d ${ROOTFS_DIR} --pagesize=4096
58            fi
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}" = yaffs2 ]; then
92    tool_check ${YAFFS2_TOOL}
93    ${YAFFS2_TOOL}  ${ROOTFS_DIR} ${ROOTFS_IMG} 2k 4bit
94else
95    echo "Unsupported fs type!" >&2
96fi
97