#!/bin/bash
# by liuhx 2013-Nov-04.
# 設置ftp環境的腳本。ftp的根目錄為只讀，其下的writable目錄為可寫

# 可自定義以下四項
# ftp用戶名
userName="test"
# ftp密碼
password="test"
# ftp根目錄，末尾不要加/
ftp_dir="$HOME/ftp"
# 可寫目錄的目錄名
writable="writable"

# 如果沒有加sudo，提示錯誤并退出
if [ "x$(id -u)" != x0 ]; then 
  echo "Error: please run this script with 'sudo'." 
  exit 1
fi

# 核心工具，vsftpd。 -y是對所有提示都回答yes
sudo apt-get -y install vsftpd
# db-util是用來生成用戶列表數據庫的工具
sudo apt-get -y install db-util

# 以下步驟參考https://help.ubuntu.com/community/vsftpd#The_workshop
# 創建用戶名和密碼的數據庫，以單數行為用戶名，雙數行為密碼記錄
cd /tmp
printf "$userName/n$password/n" > vusers.txt
db_load -T -t hash -f vusers.txt vsftpd-virtual-user.db
sudo cp -f vsftpd-virtual-user.db /etc/
cd /etc
chmod 600 vsftpd-virtual-user.db
if [ ! -e vsftpd.conf.old ]; then
 sudo cp -f vsftpd.conf vsftpd.conf.old
fi

# 創建PAM file。bash的here-document，直接輸出這些內容覆蓋原文件
(sudo cat <<EOF
auth       required     pam_userdb.so db=/etc/vsftpd-virtual-user
account    required     pam_userdb.so db=/etc/vsftpd-virtual-user
session    required     pam_loginuid.so
EOF
) > pam.d/vsftpd.virtual

# 獲取當前的用戶名，不能用whoami或$LOGNAME，否則得到的是root
owner=`who am i| awk '{print $1}'`

# 創建vsftpd的配置文件。
(sudo cat <<EOF
listen=YES
anonymous_enable=NO
local_enable=YES
virtual_use_local_privs=YES
write_enable=YES
local_umask=000
dirmessage_enable=YES
use_localtime=YES
xferlog_enable=YES
connect_from_port_20=YES
chroot_local_user=YES
hide_ids=YES
secure_chroot_dir=/var/run/vsftpd/empty
pam_service_name=vsftpd.virtual
guest_enable=YES
user_sub_token=$USER
rsa_cert_file=/etc/ssl/certs/ssl-cert-snakeoil.pem
rsa_private_key_file=/etc/ssl/private/ssl-cert-snakeoil.key
EOF
) > vsftpd.conf
sudo echo "local_root=$ftp_dir" >> vsftpd.conf
# 虛擬用戶需要映射為本地用戶，設為自己，避免權限問題，但同時也令自己對ftp根目錄不可寫
sudo echo "guest_username=$owner" >> vsftpd.conf

# 設置了每個虛擬用戶只可以瀏覽其根及子目錄（否則可訪問磁盤根目錄），
# 這樣會被要求根目錄不可寫，所以創建一個writable的子目錄
mkdir "$ftp_dir"
mkdir "$ftp_dir/$writable"
sudo chmod a-w "$ftp_dir"
sudo chown -R $owner:$owner $ftp_dir

sudo /etc/init.d/vsftpd restart