后台【两步验证】插件

为了防止网站后台工作人员账号密码意外泄露带来的安全问题(如CSDN事件)，可以考虑给后台登录加一个两步验证

一、安装

解压附件中的插件到 /plugins 目录
修改 application/admin/controller/Admin.php 在 protected function _initialize() 方法的第一行添加代码

    /**     * 初始化     * @author 蔡伟明 <314013107@qq.com>     */    protected function _initialize()    {        // 两步验证钩子        $hook_result = \think\Hook::listen('two_step');        if (!empty($hook_result) && true !== $hook_result[0]) {            $current_url = request()->url();    // 当前 url            if (    // 下面三个 URL 不跳转到两步验证页                strpos ( $current_url ,  plugin_url('TwoStep/Admin/verification') ) === false &&    // 两步验证页                strpos ( $current_url ,  plugin_url('TwoStep/Admin/signin') ) === false &&  // 验证两步验证码                strpos ( $current_url ,  url('user/publics/signout') ) === false                    // 登出账号            ){                $this->redirect(plugin_url('TwoStep/Admin/verification'));            }        }        .        .        .        .        .    }

修改 application/admin/controller/Index.php 用下面代码覆盖 public function profile() 方法

    /**     * 个人设置     * @author 蔡伟明 <314013107@qq.com>     */    public function profile()    {        // 保存数据        if ($this->request->isPost()) {            $data = $this->request->post();            $data['nickname'] == '' && $this->error('昵称不能为空');            $data['id'] = UID;            // 如果没有填写密码，则不更新密码            if ($data['password'] == '') {                unset($data['password']);            }            $UserModel = new UserModel();            if ($user = $UserModel->allowField(['nickname', 'email', 'password', 'mobile', 'avatar'])->update($data)) {                // 记录行为                action_log('user_edit', 'admin_user', UID, UID, get_nickname(UID));                $this->success('编辑成功');            } else {                $this->error('编辑失败');            }        }        // 获取数据        $info = UserModel::where('id', UID)->field('password', true)->find();        // 开启两步验证功能        $generate_secret_url = plugin_url('TwoStep/Admin/generateSecret');        $js = <<<EOF    <script type="text/javascript">        $(function(){            $("#is_two_step").change(function() {                 if($('#is_two_step').is(':checked')) {                    $.get("{$generate_secret_url}", function(res){                        if(res.code == 0){                            Dolphin.notify(res.msg, 'danger');                            $("#is_two_step").removeAttr("checked");                        } else {                            $("#two_step_secret").val(res.secret);                            $("#form_group_is_two_step > .col-sm-12").after("<img src='"+res.QRCode+"' id='QRCode' alt='QRCode'>");                        }                    });                } else {                    $("#two_step_secret").val('');                    $("#QRCode").remove();                }            });        });    </script>EOF;        $items = [ // 批量添加表单项            ['static', 'username', '用户名', '不可更改'],            ['text', 'nickname', '昵称', '可以是中文'],            ['text', 'email', '邮箱', ''],            ['password', 'password', '密码', '必填，6-20位'],            ['text', 'mobile', '手机号'],            ['image', 'avatar', '头像']        ];        if (isset($info['two_step_secret'])){   // 两步验证插件已安装            if ($info['two_step_secret']){                array_push ( $items ,  ['switch', 'is_two_step', '两步验证', '关闭后再开启可重新生成二维码', 1] );            }else{                array_push ( $items ,  ['switch', 'is_two_step', '两步验证', '开启后用 <code>Google Authenticator</code> 扫描二维码', 0] );            }            array_push ( $items ,  ['hidden', 'two_step_secret'] );        }else{            $js = '';        }        // 使用ZBuilder快速创建表单        return ZBuilder::make('form')            ->addFormItems($items)            ->setFormData($info) // 设置表单数据            ->setExtraJs($js)            ->fetch();    }