Auth.class.php
10.2 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
<?php
// +----------------------------------------------------------------------
// | ThinkPHP [ WE CAN DO IT JUST THINK IT ]
// +----------------------------------------------------------------------
// | Copyright (c) 2011 http://thinkphp.cn All rights reserved.
// +----------------------------------------------------------------------
// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
// +----------------------------------------------------------------------
// | Author: luofei614 <weibo.com/luofei614>
// +----------------------------------------------------------------------
namespace Think;
/**
* 权限认证类
* 功能特性：
* 1，是对规则进行认证，不是对节点进行认证。用户可以把节点当作规则名称实现对节点进行认证。
* $auth=new Auth(); $auth->check('规则名称','用户id')
* 2，可以同时对多条规则进行认证，并设置多条规则的关系（or或者and）
* $auth=new Auth(); $auth->check('规则1,规则2','用户id','and')
* 第三个参数为and时表示，用户需要同时具有规则1和规则2的权限。 当第三个参数为or时，表示用户值需要具备其中一个条件即可。默认为or
* 3，一个用户可以属于多个用户组(think_auth_group_access表 定义了用户所属用户组)。我们需要设置每个用户组拥有哪些规则(think_auth_group 定义了用户组权限)
* 4，支持规则表达式。
* 在think_auth_rule 表中定义一条规则时，如果type为1， condition字段就可以定义规则表达式。 如定义{score}>5 and {score}<100 表示用户的分数在5-100之间时这条规则才会通过。
*/
// 数据库
/*
* -- ----------------------------
* -- think_auth_rule，规则表，
* -- id:主键，name：规则唯一标识, title：规则中文名称 status 状态：为1正常，为0禁用，condition：规则表达式，为空表示存在就验证，不为空表示按照条件验证
* -- ----------------------------
* DROP TABLE IF EXISTS `think_auth_rule`;
* CREATE TABLE `think_auth_rule` (
* `id` mediumint(8) unsigned NOT NULL AUTO_INCREMENT,
* `name` char(80) NOT NULL DEFAULT '',
* `title` char(20) NOT NULL DEFAULT '',
* `type` tinyint(1) NOT NULL DEFAULT '1',
* `status` tinyint(1) NOT NULL DEFAULT '1',
* `condition` char(100) NOT NULL DEFAULT '', # 规则附件条件,满足附加条件的规则,才认为是有效的规则
* PRIMARY KEY (`id`),
* UNIQUE KEY `name` (`name`)
* ) ENGINE=MyISAM DEFAULT CHARSET=utf8;
* -- ----------------------------
* -- think_auth_group 用户组表，
* -- id：主键， title:用户组中文名称， rules：用户组拥有的规则id， 多个规则","隔开，status 状态：为1正常，为0禁用
* -- ----------------------------
* DROP TABLE IF EXISTS `think_auth_group`;
* CREATE TABLE `think_auth_group` (
* `id` mediumint(8) unsigned NOT NULL AUTO_INCREMENT,
* `title` char(100) NOT NULL DEFAULT '',
* `status` tinyint(1) NOT NULL DEFAULT '1',
* `rules` char(80) NOT NULL DEFAULT '',
* PRIMARY KEY (`id`)
* ) ENGINE=MyISAM DEFAULT CHARSET=utf8;
* -- ----------------------------
* -- think_auth_group_access 用户组明细表
* -- uid:用户id，group_id：用户组id
* -- ----------------------------
* DROP TABLE IF EXISTS `think_auth_group_access`;
* CREATE TABLE `think_auth_group_access` (
* `uid` mediumint(8) unsigned NOT NULL,
* `group_id` mediumint(8) unsigned NOT NULL,
* UNIQUE KEY `uid_group_id` (`uid`,`group_id`),
* KEY `uid` (`uid`),
* KEY `group_id` (`group_id`)
* ) ENGINE=MyISAM DEFAULT CHARSET=utf8;
*/
class Auth
{
// 默认配置
protected $_config = array(
'AUTH_ON' => true, // 认证开关
'AUTH_TYPE' => 1, // 认证方式，1为实时认证；2为登录认证。
'AUTH_GROUP' => 'auth_group', // 用户组数据表名
'AUTH_GROUP_ACCESS' => 'auth_group_access', // 用户-用户组关系表
'AUTH_RULE' => 'auth_rule', // 权限规则表
'AUTH_USER' => 'member'
); // 用户信息表
public function __construct()
{
$prefix = C('DB_PREFIX');
$this->_config['AUTH_GROUP'] = $prefix . $this->_config['AUTH_GROUP'];
$this->_config['AUTH_RULE'] = $prefix . $this->_config['AUTH_RULE'];
$this->_config['AUTH_USER'] = $prefix . $this->_config['AUTH_USER'];
$this->_config['AUTH_GROUP_ACCESS'] = $prefix . $this->_config['AUTH_GROUP_ACCESS'];
if (C('AUTH_CONFIG')) {
// 可设置配置项 AUTH_CONFIG, 此配置项为数组。
$this->_config = array_merge($this->_config, C('AUTH_CONFIG'));
}
}
/**
* 检查权限
*
* @param
* name string|array 需要验证的规则列表,支持逗号分隔的权限规则或索引数组
* @param
* uid int 认证用户的id
* @param
* string mode 执行check的模式
* @param
* relation string 如果为 'or' 表示满足任一条规则即通过验证;如果为 'and'则表示需满足所有规则才能通过验证
* @return boolean 通过验证返回true;失败返回false
*/
public function check($name, $uid, $type = 1, $mode = 'url', $relation = 'or')
{
if (! $this->_config['AUTH_ON']) {
return true;
}
$authList = $this->getAuthList($uid, $type); // 获取用户需要验证的所有有效规则列表
if (is_string($name)) {
$name = strtolower($name);
if (strpos($name, ',') !== false) {
$name = explode(',', $name);
} else {
$name = array(
$name
);
}
}
$list = array(); // 保存验证通过的规则名
if ($mode == 'url') {
$REQUEST = unserialize(strtolower(serialize($_REQUEST)));
}
foreach ($authList as $auth) {
$query = preg_replace('/^.+\?/U', '', $auth);
if ($mode == 'url' && $query != $auth) {
parse_str($query, $param); // 解析规则中的param
$intersect = array_intersect_assoc($REQUEST, $param);
$auth = preg_replace('/\?.*$/U', '', $auth);
if (in_array($auth, $name) && $intersect == $param) { // 如果节点相符且url参数满足
$list[] = $auth;
}
} else
if (in_array($auth, $name)) {
$list[] = $auth;
}
}
if ($relation == 'or' && ! empty($list)) {
return true;
}
$diff = array_diff($name, $list);
if ($relation == 'and' && empty($diff)) {
return true;
}
return false;
}
/**
* 根据用户id获取用户组,返回值为数组
*
* @param
* uid int 用户id
* @return array 用户所属的用户组 array(
* array('uid'=>'用户id','group_id'=>'用户组id','title'=>'用户组名称','rules'=>'用户组拥有的规则id,多个,号隔开'),
* ...)
*/
public function getGroups($uid)
{
static $groups = array();
if (isset($groups[$uid])) {
return $groups[$uid];
}
$user_groups = M()->table($this->_config['AUTH_GROUP_ACCESS'] . ' a')
->where("a.uid='$uid' and g.status='1'")
->join($this->_config['AUTH_GROUP'] . " g on a.group_id=g.id")
->field('uid,group_id,title,rules')
->select();
$groups[$uid] = $user_groups ? : array();
return $groups[$uid];
}
/**
* 获得权限列表
*
* @param integer $uid
* 用户id
* @param integer $type
*/
protected function getAuthList($uid, $type)
{
static $_authList = array(); // 保存用户验证通过的权限列表
$t = implode(',', (array) $type);
if (isset($_authList[$uid . $t])) {
return $_authList[$uid . $t];
}
if ($this->_config['AUTH_TYPE'] == 2 && isset($_SESSION['_AUTH_LIST_' . $uid . $t])) {
return $_SESSION['_AUTH_LIST_' . $uid . $t];
}
// 读取用户所属用户组
$groups = $this->getGroups($uid);
$ids = array(); // 保存用户所属用户组设置的所有权限规则id
foreach ($groups as $g) {
$ids = array_merge($ids, explode(',', trim($g['rules'], ',')));
}
$ids = array_unique($ids);
if (empty($ids)) {
$_authList[$uid . $t] = array();
return array();
}
$map = array(
'id' => array(
'in',
$ids
),
'type' => $type,
'status' => 1
);
// 读取用户组所有权限规则
$rules = M()->table($this->_config['AUTH_RULE'])
->where($map)
->field('condition,name')
->select();
// 循环规则，判断结果。
$authList = array();
foreach ($rules as $rule) {
if (! empty($rule['condition'])) {
// 根据condition进行验证
$user = $this->getUserInfo($uid);
// 获取用户信息,一维数组
$command = preg_replace('/\{(\w*?)\}/', '$user[\'\\1\']', $rule['condition']);
// dump($command);//debug
@(eval('$condition=(' . $command . ');'));
if (isset($condition)) {
$authList[] = strtolower($rule['name']);
}
} else {
// 只要存在就记录
$authList[] = strtolower($rule['name']);
}
}
$_authList[$uid . $t] = $authList;
if ($this->_config['AUTH_TYPE'] == 2) {
// 规则列表结果保存到session
$_SESSION['_AUTH_LIST_' . $uid . $t] = $authList;
}
return array_unique($authList);
}
/**
* 获得用户资料,根据自己的情况读取数据库
*/
protected function getUserInfo($uid)
{
static $userinfo = array();
if (! isset($userinfo[$uid])) {
$userinfo[$uid] = M()->where(array(
'uid' => $uid
))
->table($this->_config['AUTH_USER'])
->find();
}
return $userinfo[$uid];
}
}