vsftpd 自动安装脚本

#!/usr/bin/env python
# -*- coding: utf-8 -*-
__author__ = 'cpy'

import os
import re
import sys
import logging

'''
        @desc : install vsftpd software and configure
        @time : 2018-07-02
'''

reload(sys)
sys.setdefaultencoding('utf-8')


class Config(object):
        """VSFTPD BASE CONFIG"""
        VFTP_CONF = '/etc/vsftpd/vsftpd.conf'
        VIR_CONF = '/etc/vsftpd/vconf'
        VUSER = '/etc/vsftpd/vusers'
        PAMFILE = '/etc/pam.d/vsftpd'
        LOGFILE = '/etc/vsftpd/logs'
        SYSUSER = '/etc/passwd'
        DENYUSER = '/etc/vsftpd/user_list'

        VCONF_CONTENT_TMP = '''anonymous_enable=NO
        #设成YES,允许匿名用户登陆
        allow_writeable_chroot=YES
        #vsftpd 2.3.5之后增强了安全检查，不允许限定在主目录下的用户具有写权限，该命令声明可以具有写权限。
        local_enable=YES
        #允许/禁止本地用户登陆 注意：主要是为虚拟宿主用户，如果该项目设定为NO那么所有虚拟用户将无法访问。
        write_enable=YES
        #设定可以进行写操作。
        local_umask=022
        #设定上传后文件的权限掩码，文件644，文件夹755
        dirmessage_enable=YES
        #设定开启目录标语功能
        xferlog_enable=YES
        #设定开启日志记录功能。
        connect_from_port_20=YES
        #设定端口20进行数据连接
        xferlog_std_format=YES
        #设定日志使用标准的记录格式
        listen=YES
        #开启独立进程vsftpd，不使用超级进程xinetd。设定该Vsftpd服务工作在StandAlone模式下。
        pam_service_name=vsftpd
        #设定，启用pam认证，并指定认证文件名/etc/pam.d/vsftpd
        userlist_enable=YES
        #设定userlist_file中的用户将不得使用FTP
        tcp_wrappers=YES
        #设定支持TCP Wrappers
        chroot_local_user=YES
        #限制所有用户在主目录
        #以下这些是关于Vsftpd虚拟用户支持的重要配置项目。默认Vsftpd.conf中不包含这些设定项目，需要自己手动添加配置
        guest_enable=YES
        #设定启用虚拟用户功能
        guest_username=www
        #指定虚拟用户的宿主用户
        virtual_use_local_privs=YES
        #设定虚拟用户的权限符合他们的宿主用户 
        user_config_dir=/etc/vsftpd/vconf
        #设定虚拟用户个人Vsftp的配置文件存放路径'''

        PAM_CONTENT_TMP = '''auth    sufficient      /lib64/security/pam_userdb.so    db=/etc/vsftpd/virtusers
        account sufficient      /lib64/security/pam_userdb.so    db=/etc/vsftpd/virtusers'''

        VUSER_CONF_CONTENT_TMP = '''local_root=/data/www/virtual/
        #指定虚拟用户的具体主路径
        anonymous_enable=NO
        #设定不允许匿名用户访问
        write_enable=YES
        #设定允许写操作
        local_umask=022
        #设定上传文件权限掩码
        anon_upload_enable=NO
        #设定不允许匿名用户上传
        anon_mkdir_write_enable=NO
        #设定不允许匿名用户建立目录
        idle_session_timeout=600
        #设定空闲连接超时时间
        data_connection_timeout=120
        #设定单次连续传输最大时间
        max_clients=10
        #设定并发客户端访问个数
        max_per_ip=5
        #设定单个客户端的最大线程数，这个配置主要来照顾Flashget、迅雷等多线程下载软件
        local_max_rate=50000
        #设定该用户的最大传输速率，单位b/s'''


class InitLogging(object):
        """ 初始化日志输出配置 """

        def __init__(self):
                self.LOG = Config.LOGFILE

        def logconfig(self):
                if not os.path.exists(self.LOG):
                        BASE_VSFTP_PATH = os.path.split(self.LOG)[0]
                        os.makedirs(BASE_VSFTP_PATH)
                        f = open(self.LOG, 'w')
                        f.close()
                logging.basicConfig(level=logging.INFO, filename=self.LOG, filemode='a',
                                    format='%(asctime)s - %(filename)s[line:%(lineno)d] - %(levelname)s: %(message)s')


class ProgressBar(object):
        def __flush_screen(self, string=''):
                sys.stdout.write(string)
                sys.stdout.flush()

        def progress(self, tasklist=[], width=100):
                # 开始执行
                for taskindex, task in enumerate(tasklist):
                        # 获取每个任务占用的进度条个数
                        scale = int(width / len(tasklist))
                        # 获取每个任务占用任务总数的百分比
                        score = int(100 / len(tasklist))

                        if taskindex == 0:
                                executestring = '
{0:>2}% [{1}]{2}'.format(0, ' ' * width, os.linesep)
                                self.__flush_screen(string=executestring)
                        else:
                                executestring = '
{0:>2}% [{1}{2}]{3}'.format(taskindex * score, '=' * taskindex * scale,
                                                                               ' ' * (width - taskindex * scale), os.linesep)
                                self.__flush_screen(string=executestring)
                        eval(task)
                # 执行结束
                endstring = '
{0:>2}% [{1}] 完成{2}'.format(100, '=' * width, os.linesep)
                self.__flush_screen(string=endstring)


class Install(object):
        """INSTALL VSFTPD AND CONFIGURE"""

        def __init__(self):
                pass

        def __checkvsftp_version(self):
                """check vsftpd version add configuration"""
                faildmsg = u'vsftpd 版本获取失败'
                vsftp_version = os.popen('rpm -qa |grep vsftpd').read().split('-')[1].split('.')
                if vsftp_version:
                        l = [int(i) for i in vsftp_version]
                        if l[0] > 2 or (l[0] == 2 and l[1] > 3) or (l[0] == 2 and l[1] == 3 and l[2] >= 5):
                                return True
                        else:
                                return False
                else:
                        raise RuntimeError(faildmsg)

        def __checkuser(self, username):
                """ Check whether the input mapping user exists in the system  """
                with open(Config.SYSUSER, 'r') as f:
                        list = [username for line in f.readlines() if username == line.split(':')[0]]
                        if list:
                                return True
                        else:
                                return False

        def __deny_users(self):
                """return /etc/vsftpd/user_list deny users"""
                with open(Config.DENYUSER, 'r') as f:
                        denyusers = [line for line in f.readlines() if not re.match('^s*#.+.*$', line)]
                        return denyusers

        def rollback(self, *args):
                """UNINSTALL VSFTPD AND CONFIGURE"""
                if len(args) > 0:
                        if self.__checkuser(username=args[0]):
                                uninstall_command = 'yum -y remove vsftpd && userdel -r %s && rm -rf %s* && rm -rf /etc/vsftpd' % (
                                        args[0], Config.PAMFILE)
                        else:
                                uninstall_command = 'yum -y remove vsftpd && rm -rf %s* && rm -rf /etc/vsftpd' % Config.PAMFILE
                else:
                        uninstall_command = 'yum -y remove vsftpd && rm -rf %s* && rm -rf /etc/vsftpd' % Config.PAMFILE

                if os.system(uninstall_command) != 0:
                        msg = u'卸载vsftpd 发生错误，请手动清除，以免产生垃圾文件'
                        print msg
                        logging.info(msg)
                        raise Exception(msg)

        def input_vuserinfo(self, numbers):
                """Trapping the virtual user information input"""
                vuser_dict = {}
                v_user = raw_input(u'请输入第%d个虚拟用户名:' % numbers)
                v_pass = raw_input(u'请输入第%d个虚拟用户密码:' % numbers)
                vuser_dict['username'] = v_user
                vuser_dict['password'] = v_pass
                return vuser_dict

        def install_vsftpd(self, username):
                install_vsftpd_command = 'yum -y install pam pam-devel db4 db4-tcl vsftpd >>%s' % Config.LOGFILE
                actionmsg = u'开始安装vsftp服务...'
                errormsg = u'vsftpd 或其他依赖项安装失败。'

                logging.info(actionmsg)
                if os.system(install_vsftpd_command) != 0:
                        logging.error(errormsg)
                        self.rollback(username)
                        raise Exception(errormsg)

        def deploy(self, username, parentpath):
                """configure the vsftpd"""
                sucessfulmsg = u'配置成功。'
                msg = u'配置错误，请检查是否已经存在或不允许的参数'
                denyusers = self.__deny_users()
                if username in denyusers:
                        logging.error(msg)
                        self.rollback(username)
                        raise Exception(msg)
                if not self.__checkuser(username=username):
                        os.makedirs(parentpath)
                        os.rmdir(parentpath)
                        create_user_command = 'useradd %s -d %s -s /sbin/nologin >> %s' % (username, parentpath, Config.LOGFILE)
                        if os.system(create_user_command) != 0:
                                logging.error(msg)
                                self.rollback(username)
                                raise Exception(msg)
                else:
                        logging.warning(msg)
                        self.rollback(username)
                        raise Exception(msg)

                back_conf_command = 'cp %s %s.bak >> %s' % (Config.VFTP_CONF, Config.VFTP_CONF, Config.LOGFILE)
                if os.system(back_conf_command) != 0:
                        logging.error(msg)
                        self.rollback(username)
                        raise Exception(msg)
                else:
                        logging.info(sucessfulmsg)

                f = open(Config.VFTP_CONF, 'w')
                f.close()

                list_vconf_content = Config.VCONF_CONTENT_TMP.split(os.linesep)

                for line in list_vconf_content:
                        if re.match('^guest_username=.+$', line.strip()):
                                line = 'guest_username=%s' % username
                        if re.match('^user_config_dir=.+$', line.strip()):
                                line = 'user_config_dir=%s' % Config.VIR_CONF
                        if not self.__checkvsftp_version():
                                if re.match('^allow_writeable_chroot=YES$', line.strip()):
                                        line = '#%s' % line.strip()

                        with open(Config.VFTP_CONF, 'a') as f:
                                f.write(line.strip() + os.linesep)
                logging.info(sucessfulmsg)

        def deploy_vritual_user(self, username, vuser):
                """config vritual users"""
                sucessfulmsg = u'创建虚拟用户配置目录成功'
                faildmsg = u'创建虚拟用户配置目录失败'
                dbsucessfulmsg = u'生成虚拟用户数据文件成功'
                dbfaildmsg = u'生成虚拟用户数据文件失败'

                if not os.path.exists(Config.VIR_CONF):
                        try:
                                os.makedirs(Config.VIR_CONF)
                                logging.info(sucessfulmsg)
                        except Exception:
                                self.rollback(username)
                                raise Exception(faildmsg)
                f = open(Config.VUSER, 'w')
                f.close()

                for v_user in vuser:
                        with open(Config.VUSER, 'a') as f:
                                f.write(v_user.get('username') + os.linesep)
                                f.write(v_user.get('password') + os.linesep)

                create_db_command = 'db_load -T -t hash -f %s %s.db' % (Config.VUSER, Config.VUSER)

                if os.path.exists("%s.db" % Config.VUSER):
                        os.remove('%s.db' % Config.VUSER)

                if os.system(create_db_command) != 0:
                        logging.error(dbfaildmsg)
                        self.rollback(username)
                        raise Exception(dbfaildmsg)
                else:
                        logging.info(dbsucessfulmsg)

        def deploy_pam(self):
                """config virtual user by pam"""
                baksucessfulmsg = u'备份成功'
                bakfaildmssg = u'备份失败'
                msg = u'开始配置pam认证...'
                endmsg = u'配置pam认证完成'
                logging.info(msg)
                back_pam_command = 'cp %s %s.back' % (Config.PAMFILE, Config.PAMFILE)

                if os.path.exists('%s.back' % Config.PAMFILE):
                        os.remove('%s.back' % Config.PAMFILE)

                if os.system(back_pam_command) != 0:
                        logging.warning(bakfaildmssg)
                else:
                        logging.info(baksucessfulmsg)

                f = open(Config.PAMFILE, 'w')
                f.close()

                list_pam_content = Config.PAM_CONTENT_TMP.split(os.linesep)

                for line in list_pam_content:
                        p = re.compile(r'(?P<auth>.+s+.+s+.+s+db=)(?P<vuserfile>.+)')
                        for dt in p.finditer(line):
                                line = dt.groupdict().get('auth') + Config.VUSER

                        with open(Config.PAMFILE, 'a') as f:
                                f.write(line.strip() + os.linesep)
                        logging.info(endmsg)

        def deploy_virtual_config(self, username, parentpath, vuser):
                """config virtual user Configure"""
                sucessfulmsg = u'创建虚拟用户目录或配置成功'
                faildmsg = u'创建虚拟用户目录或配置错误'
                existsmsg = u'已存在，正在尝试删除...'

                list_vuser_config_content = Config.VUSER_CONF_CONTENT_TMP.split(os.linesep)
                for v_user in vuser:
                        vuser_path = os.path.join(parentpath, v_user.get('username'))
                        if os.path.exists(vuser_path):
                                logging.warning(existsmsg)
                                os.rmdir(vuser_path)

                        try:
                                os.makedirs(vuser_path)
                                uid = int(os.popen('id %s -u' % username).read().strip())
                                gid = int(os.popen('id %s -g' % username).read().strip())
                                os.chown(vuser_path, uid, gid)
                                if not os.path.exists(vuser_path):
                                        logging.error(faildmsg)
                                        raise Exception(faildmsg)
                                logging.info(sucessfulmsg)
                        except Exception:
                                logging.error(faildmsg)
                                self.rollback(username)
                                raise Exception(faildmsg)

                        vuser_config = os.path.join(Config.VIR_CONF, v_user.get('username'))
                        if os.path.exists(vuser_path):
                                f = open(vuser_config, 'w')
                                f.close()
                        else:
                                os.makedirs(Config.VIR_CONF)
                                f = open(vuser_config, 'w')
                                f.close()

                        for line in list_vuser_config_content:
                                if re.match('^local_root=.+$', line.strip()):
                                        line = 'local_root=%s' % vuser_path
                                with open(vuser_config, 'a') as f:
                                        f.write(line.strip() + os.linesep)
                logging.info(sucessfulmsg)

        def start_server(self, parentpath):
                start_vsftpd_command = 'service vsftpd start'
                if os.system(start_vsftpd_command) != 0:
                        logging.error(u'启动vsftpd服务失败')
                        raise Exception(u'启动vsftpd服务失败')
                else:
                        logging.info(u'启动vsftpd服务成功')

                print ''
                print u"安装vsftpd 完成！"
                print ''
                print ''
                print u'-----' * 15
                print u'1. 请在%s文件中查看登录的用户名和密码，一行用户名，一行密码' % Config.VUSER
                print u'2. ftp数据存放在%s中' % parentpath
                print u'3. 若FTP无法登陆，请检查主机防火墙是否关闭'
                print u'-----' * 15


if __name__ == '__main__':
        '''初始化日志输出'''
        LOG = InitLogging()
        LOG.logconfig()

        '''安装vsftpd服务'''
        IS = Install()
        username = raw_input(u'请输入vsftp的映射宿主用户名(本机账号)：')
        parentpath = raw_input(u'请输入用于存放ftp数据的目录：')
        usercount = raw_input(u'您需要添加几个虚拟用户：')
        vuserlist = []
        if usercount.isdigit():
                num = 0
                for i in range(int(usercount)):
                        num += 1
                        vd = IS.input_vuserinfo(numbers=num)
                        vuserlist.append(vd)
        else:
                raise RuntimeError(u'您输入的参数不是整型')

        # IS.install_vsftpd(username=username)
        # IS.deploy(username=username, parentpath=parentpath)
        # IS.deploy_vritual_user(username=username, vuser=vuserlist)
        # IS.deploy_pam()
        # IS.deploy_virtual_config(username=username, parentpath=parentpath, vuser=vuserlist)
        # IS.start_server(parentpath=parentpath)

        PB = ProgressBar()
        tasks = ['IS.install_vsftpd(username=username)', 'IS.deploy(username=username, parentpath=parentpath)',
                 'IS.deploy_vritual_user(username=username, vuser=vuserlist)', 'IS.deploy_pam()',
                 'IS.deploy_virtual_config(username=username, parentpath=parentpath, vuser=vuserlist)',
                 'IS.start_server(parentpath=parentpath)']
        PB.progress(tasklist=tasks, width=50)