#!/bin/bash
export PATH=/bin:/sbin:/usr/bin:/usr/sbin:$PATH

BIN=$(basename $0)
DEV=$1
CRYPT_DEV="ucrypt_${DEV##*/}"
LOG_FILE="/var/log/usb_crypt.log"

function adddate {
    while IFS= read -r line; do
        echo "$(date +"%Y-%m-%d %H:%m:%S"): $line"
    done
}

if [[ $# -ne 1 ]]; then
	echo "Usage: $BIN <device>" | adddate >>$LOG_FILE 2>&1
	echo "FAILED" >"$UCRYPT_PATH/usb_crypt_shell"
	exit 1
fi

if [[ ! -b $DEV ]]; then
	echo "$DEV not exist" | adddate >>$LOG_FILE 2>&1
	echo "FAILED" >"$UCRYPT_PATH/usb_crypt_shell"
	exit 2
fi

## 卸载usb加密设备
umount_ucrypt $CRYPT_DEV 
if [[ $? -ne 0 ]]; then
	echo  "$DEV umount failed" | adddate >>$LOG_FILE 2>&1
	echo "FAILED" >"$UCRYPT_PATH/usb_crypt_shell"
	exit 3
fi
if [[ -b $DEV ]]; then
	## 参数usb设备的前10M
	dd if=/dev/zero of=$DEV bs=10M count=1 oflag=direct >>/dev/null 2>&1 
	## 写入日志
	log_ucrypt $DEV add 2 
	echo "Erase ucrypt $DEV success" | adddate >>$LOG_FILE 2>&1
	echo "SUCCESS" >"$UCRYPT_PATH/usb_crypt_shell"
else
	log_ucrypt $DEV remove
	echo  "$DEV not exist" | adddate >>$LOG_FILE 2>&1
	echo "FAILED" >"$UCRYPT_PATH/usb_crypt_shell"
	exit 2
fi


