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

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

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

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

# 以下步驟參考https://help.ubuntu.com/community/vsftpd#The_workshop
# 創(chuàng)建用戶名和密碼的數(shù)據(jù)庫，以單數(shù)行為用戶名，雙數(shù)行為密碼記錄
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

# 創(chuàng)建PAM file。bash的here-document，直接輸出這些內(nèi)容覆蓋原文件
(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

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

# 創(chuàng)建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
# 虛擬用戶需要映射為本地用戶，設(shè)為自己，避免權(quán)限問題，但同時(shí)也令自己對(duì)ftp根目錄不可寫
sudo echo "guest_username=$owner" >> vsftpd.conf

# 設(shè)置了每個(gè)虛擬用戶只可以瀏覽其根及子目錄（否則可訪問磁盤根目錄），
# 這樣會(huì)被要求根目錄不可寫，所以創(chuàng)建一個(gè)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